libpsyc/test/test_json.c

#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;
}