libpsyc/test/test_json_glib.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 <glib-object.h>
#include <json-glib/json-glib.h>

#include "test.c"

// cmd line args
char *filename, *port = "4442";
uint8_t verbose, stats;
uint8_t no_render, quiet, progress;
size_t count = 1, recv_buf_size;

int exit_code;

JsonParser *parser;
JsonGenerator *generator;

void
test_init (int i)
{
    g_type_init();
    parser = json_parser_new();
    generator = json_generator_new();
}

int
test_input (int i, char *recvbuf, size_t nbytes)
{
    JsonNode *root;
    GError *error = NULL;
    char *str;
    size_t len;
    int r, ret;

    ret = json_parser_load_from_data(parser, recvbuf, nbytes, &error);

    if (!ret) {
	printf("Parse error\n");
	exit(1);
    }

    root = json_parser_get_root(parser);

    if (!no_render) {
	json_generator_set_root(generator, root);
	str = json_generator_to_data(generator, &len);;
	if (!quiet) {
	    if (filename) {
		r = write(1, str, len);
		r = write(1, "\n", 1);
	    } else {
		send(i, str, len, 0);
		send(i, "\n", 1, 0);
	    }
	}
    }

    if (verbose)
	printf("# Done parsing.\n");
    else if (progress)
	r = write(1, ".", 1);

    return ret;
}

int main (int argc, char **argv) {
    int c;

    while ((c = getopt (argc, argv, "f:p:b:c:nqsvPh")) != -1) {
	switch (c) {
	CASE_f CASE_p CASE_b
	CASE_c CASE_n CASE_q
	CASE_s CASE_v CASE_P
	case 'h':
	    printf(HELP_FILE("test_json_glib", "mnqSsvP")
		   HELP_PORT("test_json_glib", "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;
}