int main(int argc, char **argv)
{
size_t size = 0;
DataBuf buf;
char *data;
assert(databuf_init(&buf, size, DATABUF_FLAG_STRING));
databuf_print(&buf, 1, "after init size=%d", size);
#if 1
data = "a";
assert(databuf_strcat(&buf, data));
databuf_print(&buf, 1, "after strcat(%s)", data);
data = "bb";
assert(databuf_strcat(&buf, data));
databuf_print(&buf, 1, "after strcat(%s)", data);
data = "ccc";
assert(databuf_strcat(&buf, data));
databuf_print(&buf, 1, "after strcat(%s)", data);
#endif
data = databuf_export(&buf);
printf("concatenation=\"%s\"\n", data);
free(data);
#if 0
assert(databuf_init(&buf, size, 0));
databuf_print(&buf, 1, "after init size=%d", size);
size = 8;
data = make_data(size, "a");
assert(databuf_append(&buf, data, size));
databuf_print(&buf, 1, "after append size=%d", size);
assert(databuf_append(&buf, data, size));
free(data);
databuf_print(&buf, 1, "after append size=%d", size);
assert(databuf_advance(&buf, 4));
databuf_print(&buf, 1, "after databuf_advance(%d", 4);
assert(databuf_compress(&buf));
databuf_print(&buf, 1, "after compress");
size = 5;
data = make_data(size, "b");
assert(databuf_append(&buf, data, size));
free(data);
databuf_print(&buf, 1, "after append size=%d", size);
size = 7;
data = make_data(size, "c");
assert(databuf_append(&buf, data, size));
free(data);
databuf_print(&buf, 1, "after append size=%d", size);
databuf_free(&buf);
#endif
exit(0);
}
/* General functions that affect operation of the library */
auparse_state_t *auparse_init(ausource_t source, const void *b)
{
char **tmp, **bb = (char **)b, *buf = (char *)b;
int n, i;
size_t size, len;
auparse_state_t *au = malloc(sizeof(auparse_state_t));
if (au == NULL) {
errno = ENOMEM;
return NULL;
}
au->le = NULL;
/*
* Set up the List of List events base structure
*/
au->au_lo = calloc(sizeof(au_lol), 1);
if (au->au_lo == NULL) {
free(au);
errno = ENOMEM;
return NULL;
}
au_lol_clear(au->au_lo, 0); // python doesn't call auparse_destroy
if (au_lol_create(au->au_lo) == NULL) {
free(au->au_lo);
free(au);
errno = ENOMEM;
return NULL;
}
au->au_ready = 0;
au->in = NULL;
au->source_list = NULL;
databuf_init(&au->databuf, 0, 0);
au->callback = NULL;
au->callback_user_data = NULL;
au->callback_user_data_destroy = NULL;
switch (source)
{
case AUSOURCE_LOGS:
if (setup_log_file_array(au))
goto bad_exit;
break;
case AUSOURCE_FILE:
if (b == NULL)
goto bad_exit;
if (access(b, R_OK))
goto bad_exit;
tmp = malloc(2*sizeof(char *));
tmp[0] = strdup(b);
tmp[1] = NULL;
au->source_list = tmp;
break;
case AUSOURCE_FILE_ARRAY:
if (bb == NULL)
goto bad_exit;
n = 0;
while (bb[n]) {
if (access(bb[n], R_OK))
goto bad_exit;
n++;
}
tmp = malloc((n+1)*sizeof(char *));
for (i=0; i<n; i++)
tmp[i] = strdup(bb[i]);
tmp[n] = NULL;
au->source_list = tmp;
break;
case AUSOURCE_BUFFER:
if (buf == NULL)
goto bad_exit;
len = strlen(buf);
if (databuf_init(&au->databuf, len,
DATABUF_FLAG_PRESERVE_HEAD) < 0)
goto bad_exit;
if (databuf_append(&au->databuf, buf, len) < 0)
goto bad_exit;
break;
case AUSOURCE_BUFFER_ARRAY:
if (bb == NULL)
goto bad_exit;
size = 0;
for (n = 0; (buf = bb[n]); n++) {
len = strlen(bb[n]);
if (bb[n][len-1] != '\n') {
size += len + 1;
} else {
size += len;
}
}
if (databuf_init(&au->databuf, size,
DATABUF_FLAG_PRESERVE_HEAD) < 0)
goto bad_exit;
for (n = 0; (buf = bb[n]); n++) {
len = strlen(buf);
if (databuf_append(&au->databuf, buf, len) < 0)
goto bad_exit;
}
break;
case AUSOURCE_DESCRIPTOR:
n = (long)b;
au->in = fdopen(n, "rm");
break;
case AUSOURCE_FILE_POINTER:
au->in = (FILE *)b;
break;
case AUSOURCE_FEED:
if (databuf_init(&au->databuf, 0, 0) < 0) goto bad_exit;
break;
default:
errno = EINVAL;
goto bad_exit;
break;
}
au->source = source;
au->list_idx = 0;
au->line_number = 0;
au->next_buf = NULL;
au->off = 0;
au->cur_buf = NULL;
au->line_pushed = 0;
au->parse_state = EVENT_EMPTY;
au->expr = NULL;
au->find_field = NULL;
au->search_where = AUSEARCH_STOP_EVENT;
au->escape_mode = AUPARSE_ESC_TTY;
au->message_mode = MSG_QUIET;
au->debug_message = DBG_NO;
au->tmp_translation = NULL;
init_normalizer(&au->norm_data);
return au;
bad_exit:
databuf_free(&au->databuf);
/* Free list of events list (au_lo) structure */
au_lol_clear(au->au_lo, 0);
free(au->au_lo);
free(au);
return NULL;
}
/*
* Returns a file descriptor to the incoming port
* on success, or -1 on error.
*/
static int
startup_chunker(
char * filename,
off_t use,
off_t chunksize,
struct databuf * db)
{
int infd, outfd;
char *tmp_filename, *pc;
in_port_t data_port;
int data_socket;
int result;
struct addrinfo *res;
data_port = 0;
if ((result = resolve_hostname("localhost", 0, &res, NULL) != 0)) {
errstr = newvstrallocf(errstr, _("could not resolve localhost: %s"),
gai_strerror(result));
return -1;
}
data_socket = stream_server(res->ai_family, &data_port, 0,
STREAM_BUFSIZE, 0);
if (res) freeaddrinfo(res);
if(data_socket < 0) {
errstr = vstrallocf(_("error creating stream server: %s"), strerror(errno));
return -1;
}
putresult(PORT, "%d\n", data_port);
infd = stream_accept(data_socket, CONNECT_TIMEOUT, 0, STREAM_BUFSIZE);
aclose(data_socket);
if(infd == -1) {
errstr = vstrallocf(_("error accepting stream: %s"), strerror(errno));
return -1;
}
tmp_filename = vstralloc(filename, ".tmp", NULL);
pc = strrchr(tmp_filename, '/');
g_assert(pc != NULL);
*pc = '\0';
mkholdingdir(tmp_filename);
*pc = '/';
if ((outfd = open(tmp_filename, O_RDWR|O_CREAT|O_TRUNC, 0600)) < 0) {
int save_errno = errno;
char *m = vstrallocf(_("holding file \"%s\": %s"),
tmp_filename,
strerror(errno));
errstr = quote_string(m);
amfree(m);
amfree(tmp_filename);
aclose(infd);
if(save_errno == ENOSPC) {
putresult(NO_ROOM, "%s %lld\n",
handle, (long long)use);
return -2;
} else {
return -1;
}
}
amfree(tmp_filename);
databuf_init(db, outfd, filename, use, chunksize);
db->filename_seq++;
return infd;
}
