libpsyc/test/test_psyc_speed.c

#include <ctype.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <assert.h>
#include <unistd.h>
#include <getopt.h>
#include <sys/socket.h>

#define __INLINE_PSYC_PARSE
#include "../src/parse.c"
//#include <psyc/parse.h>

#include "test.c"

// max size for routing & entity header
#define ROUTING_LINES 16
#define ENTITY_LINES 32

// cmd line args
char *filename, *port = "4440";
uint8_t verbose, stats;
uint8_t routing_only;
size_t count = 1, recv_buf_size;

PsycParseState parser;

void
test_init (int i)
{
    psyc_parse_state_init(&parser, routing_only
			  ? PSYC_PARSE_ROUTING_ONLY : PSYC_PARSE_ALL);
}

int
test_input (int i, char *recvbuf, size_t nbytes)
{
    char oper;
    PsycString name, value;
    int ret;

    psyc_parse_buffer_set(&parser, recvbuf, nbytes);

    for (;;) {
	ret = psyc_parse(&parser, &oper, &name, &value);
	if (ret == PSYC_PARSE_COMPLETE || ret < 0)
	    return -1;
    }
}

int
main (int argc, char **argv)
{
    int c;
    while ((c = getopt (argc, argv, "f:p:b:c:rsh")) != -1) {
	switch (c) {
	CASE_f CASE_p CASE_b CASE_c CASE_r CASE_s
	case 'h':
	    printf(HELP_FILE("test_psyc_speed", "rs")
		   HELP_PORT("test_psyc_speed", "rs")
		   HELP_f HELP_p HELP_b HELP_c
		   HELP_r HELP_s 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 0;
}