mirror of git://git.psyc.eu/libpsyc
139 lines
3.1 KiB
C
139 lines
3.1 KiB
C
/*
|
|
This file is part of libpsyc.
|
|
Copyright (C) 2011,2012 Carlo v. Loesch, Gabor X Toth, Mathias L. Baumann,
|
|
and other contributing authors.
|
|
|
|
libpsyc is free software: you can redistribute it and/or modify it under the
|
|
terms of the GNU Affero General Public License as published by the Free
|
|
Software Foundation, either version 3 of the License, or (at your option) any
|
|
later version. As a special exception, libpsyc is distributed with additional
|
|
permissions to link libpsyc libraries with non-AGPL works.
|
|
|
|
libpsyc is distributed in the hope that it will be useful, but WITHOUT
|
|
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
|
FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
|
|
details.
|
|
|
|
You should have received a copy of the GNU Affero General Public License and
|
|
the linking exception along with libpsyc in a COPYING file.
|
|
*/
|
|
|
|
#include <ctype.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <stdint.h>
|
|
#include <string.h>
|
|
#include <assert.h>
|
|
#include <unistd.h>
|
|
#include <getopt.h>
|
|
#include <sys/socket.h>
|
|
|
|
#include <json/json.h>
|
|
#include <json/json_object.h>
|
|
#include <json/json_tokener.h>
|
|
|
|
#include "test.c"
|
|
|
|
// cmd line args
|
|
char *filename, *port = "4441";
|
|
uint8_t verbose, stats;
|
|
uint8_t multiple, single, no_render, quiet, progress;
|
|
size_t count = 1, recv_buf_size;
|
|
|
|
int exit_code;
|
|
|
|
json_object *obj;
|
|
json_tokener *tok;
|
|
enum json_tokener_error error;
|
|
|
|
void
|
|
test_init (int i)
|
|
{
|
|
tok = json_tokener_new();
|
|
}
|
|
|
|
int
|
|
test_input (int i, char *recvbuf, size_t nbytes)
|
|
{
|
|
size_t cursor = 0;
|
|
int r, ret = 0;
|
|
|
|
json_tokener_reset(tok);
|
|
|
|
do {
|
|
obj = json_tokener_parse_ex(tok, recvbuf + cursor, nbytes - cursor);
|
|
cursor += tok->char_offset;
|
|
|
|
if (verbose)
|
|
printf("#%d\n", tok->err);
|
|
|
|
switch (tok->err) {
|
|
case json_tokener_continue:
|
|
return 0;
|
|
|
|
case json_tokener_success:
|
|
if (!no_render) {
|
|
const char *str = json_object_to_json_string(obj);
|
|
if (!quiet) {
|
|
size_t len = strlen(str);
|
|
if (filename) {
|
|
r = write(1, str, len);
|
|
r = write(1, "\n", 1);
|
|
} else {
|
|
send(i, str, len, 0);
|
|
send(i, "\n", 1, 0);
|
|
}
|
|
}
|
|
}
|
|
|
|
json_object_put(obj);
|
|
|
|
if (verbose)
|
|
printf("# Done parsing.\n");
|
|
else if (progress)
|
|
r = write(1, ".", 1);
|
|
if ((filename && !multiple) || (!filename && single))
|
|
return -1;
|
|
break;
|
|
|
|
default:
|
|
printf("parse error\n");
|
|
exit_code = tok->err;
|
|
return -1;
|
|
}
|
|
} while (cursor < nbytes);
|
|
|
|
return ret;
|
|
}
|
|
|
|
int
|
|
main (int argc, char **argv)
|
|
{
|
|
int c;
|
|
|
|
while ((c = getopt (argc, argv, "f:p:b:c:mnqsvPSh")) != -1) {
|
|
switch (c) {
|
|
CASE_f CASE_p CASE_b CASE_c
|
|
CASE_m CASE_n CASE_q CASE_s
|
|
CASE_v CASE_S CASE_P
|
|
case 'h':
|
|
printf(HELP_FILE("test_json", "mnqSsvP")
|
|
HELP_PORT("test_json", "nqsvP")
|
|
HELP_f HELP_p HELP_b HELP_c
|
|
HELP_m HELP_n HELP_q HELP_S
|
|
HELP_s HELP_v HELP_P HELP_h,
|
|
port, RECV_BUF_SIZE);
|
|
exit(0);
|
|
case '?': exit(-1);
|
|
default: abort();
|
|
}
|
|
}
|
|
|
|
if (filename)
|
|
test_file(filename, count, recv_buf_size);
|
|
else
|
|
test_server(port, count, recv_buf_size);
|
|
|
|
return exit_code;
|
|
}
|