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);
}
int auparse_feed(auparse_state_t *au, const char *data, size_t data_len)
{
if (databuf_append(&au->databuf, data, data_len) < 0)
return -1;
consume_feed(au, 0);
return 0;
}
static int databuf_strcat(DataBuf *db, const char *str)
{
size_t str_len;
DATABUF_VALIDATE(db);
if (str == NULL) return 0;
// +1 so the data append also copies the NULL terminator
str_len = strlen(str) + 1;
// If there is a NULL terminator exclude it so the subsequent
// data append produces a proper string concatenation
if (db->len > 0) {
char *last_char = databuf_end(db) - 1;
if (*last_char == 0) {
db->len--; // backup over NULL terminator
}
}
// Copy string and NULL terminator
databuf_append(db, str, str_len);
DATABUF_VALIDATE(db);
return 1;
}
/* 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;
}
