int payload_start_array(struct payload * po, char * key) {
if(!payload_add_key(po, key))
return 0;
adjust_payload_len(po, sizeof("[ "));
po->bufused += sprintf(po->json_buf + po->bufused, "[ ");
return 1;
}
void payload_new_integer(struct payload * po, char * key, long long val) {
if(!payload_add_key(po, key))
return;
adjust_payload_len(po, sizeof("INT64_MAX, "));
po->bufused += sprintf(po->json_buf + po->bufused,
"%lli, ", val);
}
void payload_new_double(struct payload * po, char * key, double val) {
if(!payload_add_key(po, key))
return;
adjust_payload_len(po, DBL_MAX_10_EXP + sizeof(", "));
po->bufused += snprintf(po->json_buf + po->bufused,
DBL_MAX_10_EXP, "%f", val) - 1;
po->bufused += sprintf(po->json_buf + po->bufused,
", ");
}
void payload_new_boolean(struct payload * po, char * key, int val) {
if(!payload_add_key(po, key))
return;
adjust_payload_len(po, sizeof("false, "));
if(val > 0)
po->bufused += sprintf(po->json_buf + po->bufused,
"true, ");
else
po->bufused += sprintf(po->json_buf + po->bufused,
"false, ");
}
void payload_new_timestamp(struct payload * po,
char* key, struct timeval * tv) {
if(!payload_add_key(po, key))
return;
adjust_payload_len(po, sizeof("{ }, "));
po->bufused += sprintf(po->json_buf + po->bufused, "{ ");
payload_new_integer(po, "tv_sec", tv->tv_sec);
payload_new_integer(po, "tv_usec", tv->tv_usec);
po->bufused -= 2;
po->bufused += sprintf(po->json_buf + (po->bufused),
" }, ");
}
void payload_finalize(struct payload * po) {
size_t offset = po->bufused;
if(offset > 2)
offset -= 2;
adjust_payload_len(po, sizeof("] "));
if(po->json_buf[0] == '[')
sprintf(po->json_buf + offset, " ]");
else
sprintf(po->json_buf + offset, " }");
if(offset == 2)
po->bufused += 2;
}
int payload_add_key(struct payload * po, char * key) {
if(key == NULL)
return 1;
size_t keylen = strlen(key);
if(po->use_hash) {
unsigned int hashval = hash_output_key(key, keylen);
if(!(po->hashed_keys[WORD_OFFSET(hashval)] & (1 << BIT_OFFSET(hashval))))
return 0;
}
adjust_payload_len(po, keylen + sizeof("\"\": "));
po->bufused += sprintf(po->json_buf + po->bufused,
"\"%s\": ", key);
return 1;
}
int payload_add_key(struct payload * po, char * key) {
if(key == NULL)
return 1;
size_t keylen = strlen(key);
unsigned char hash = keylen;
int i;
if(po->keys) {
for(i = keylen; i > 0;)
hash = mixtable[hash ^ key[i--]];
struct keybucket * b = po->keys[hash];
while(b && strcmp(b->key, key) != 0)
b = b->next;
if(!b)
return 0;
}
adjust_payload_len(po, keylen + sizeof("\"\": "));
po->bufused += sprintf(po->json_buf + po->bufused,
"\"%s\": ", key);
return 1;
}
void payload_end_object(struct payload * po) {
adjust_payload_len(po, sizeof(", "));
if(*(po->json_buf + po->bufused - 2) != '{')
po->bufused -= 2;
po->bufused += sprintf(po->json_buf + po->bufused, " }, ");
}
void payload_new_string(struct payload * po, char * key, char * val) {
if(!payload_add_key(po, key))
return;
if(val == NULL) {
adjust_payload_len(po, sizeof("null, "));
po->bufused += sprintf(po->json_buf + po->bufused,
"null, ");
return;
}
size_t len = 0;
char * ptr = val, *out;
char * save;
unsigned char token;
while((token=*ptr) && ++len) {
if(strchr("\"\\\b\f\n\r\t",token))
len++;
else if(token < 32 || (token > 127 && token < 160))
len += 5;
ptr++;
}
adjust_payload_len(po, len + sizeof("\"\", "));
ptr = val;
po->bufused += sprintf(po->json_buf + po->bufused, "\"");
out = po->json_buf + po->bufused;
save = out;
while(*ptr != '\0') {
if ((unsigned char)*ptr>31 && *ptr!='\"' && *ptr!='\\')
*(out++)=*ptr++;
else
{
*(out++) = '\\';
switch (token=*ptr++)
{
case '\\': *(out++)='\\'; break;
case '\"': *(out++)='\"'; break;
case '\b': *(out++)='b'; break;
case '\f': *(out++)='f'; break;
case '\n': *(out++)='n'; break;
case '\r': *(out++)='r'; break;
case '\t': *(out++)='t'; break;
default:
sprintf(out,"u%04x",token);
out += 5;
break; /* escape and print */
}
}
}
*out = '\0';
if(po->keep_auxdata) {
if(key && strcmp(key, "type") == 0)
po->type = strdup(save);
else if(key && strcmp(key, "host_name") == 0)
po->host_name = strdup(save);
else if(key && strcmp(key, "service_description") == 0)
po->service_description = strdup(save);
else if(key && strcmp(key, "pong_target") == 0)
po->pong_target = strdup(save);
}
po->bufused += out - save;
po->bufused += sprintf(po->json_buf + po->bufused, "\", ");
}
