throughput/main.c

#include <string.h>

#include <math.h>

#include <stdlib.h>

#include <time.h>

#include <strophe.h>

xmpp_conn_t *conn1, *conn2;

unsigned long mincount, maxcount;

unsigned long count = 0;

unsigned long sum_squares = 0;

unsigned long samples = 0; /* Current number of samples taken */

unsigned long discard_samples = 5;

unsigned long max_samples = 20; /* The number of samples to do, if specified by user */

time_t lasttime; /* Time of the last sample we took */

char* jid1 = "test1@localhost";

char* jid2 = "test2@localhost";

char* pass1 = "test";

char* pass2 = "test";

xmpp_stanza_t *bstanza;

int message_handler(xmpp_conn_t * const conn, xmpp_stanza_t * const stanza, void * const userdata)

{

	xmpp_ctx_t *ctx = (xmpp_ctx_t*)userdata;

	if(!xmpp_stanza_get_child_by_name(stanza, "body")) return 1;

	if(!strcmp(xmpp_stanza_get_attribute(stanza, "type"), "error")) return 1;

	if(conn == conn2)

	{

		count++;

		if(time(NULL) != lasttime)

		{

			if(lasttime > 0)

			{

				printf("%ld stanzas/sec\n", count);

				if(count > maxcount)

				{

					printf("%ld beats old maximum of %ld\n", count, maxcount);

					maxcount = count;

				}

				if(samples >= discard_samples)

					sum_squares += count*count;

				samples++;

				if(samples >= max_samples)

				{

					printf("----\nMax: %ld\nRMS: %.2Lf\n----\n", 

						maxcount,

						sqrtl(sum_squares/(samples-discard_samples))

					);

					xmpp_stop(ctx);

					return 0;

				}

			}

			count = 0;

			lasttime = time(NULL);

		}

		xmpp_send(conn1, bstanza);

		//fprintf(stderr, "C: %d\n", count);

	}

	return 1;

}

/* define a handler for connection events */

void conn_handler(xmpp_conn_t * const conn, const xmpp_conn_event_t status, 

		  const int error, xmpp_stream_error_t * const stream_error,

		  void * const userdata)

{

    xmpp_ctx_t *ctx = (xmpp_ctx_t *)userdata;

    if (status == XMPP_CONN_CONNECT) {

	xmpp_stanza_t* pres;

	fprintf(stderr, "DEBUG: connected\n");

	xmpp_handler_add(conn,message_handler, NULL, "message", NULL, ctx);

	/* Send initial <presence/> so that we appear online to contacts */

	pres = xmpp_stanza_new(ctx);

	xmpp_stanza_set_name(pres, "presence");

	xmpp_send(conn, pres);

	xmpp_stanza_release(pres);

	if(conn == conn2) 

	{

		count = 0;

		lasttime = 0;

		xmpp_send(conn1, bstanza);

	}

    }

    else {

	fprintf(stderr, "DEBUG: disconnected\n");

	xmpp_stop(ctx);

    }

}

int main(int argc, char *argv[])

{

    xmpp_ctx_t *ctx;

    xmpp_log_t *log = NULL;

    xmpp_stanza_t *body, *text;

    printf("Loading...\n");

    /* init library */

    xmpp_initialize();

    /* create a context */

#ifdef LOGGING

    	log = xmpp_get_default_logger(XMPP_LEVEL_DEBUG); /* pass NULL instead to silence output */

#endif

    ctx = xmpp_ctx_new(NULL, log);

    /* create a connection */

    conn1 = xmpp_conn_new(ctx);

    conn2 = xmpp_conn_new(ctx);

    /* setup authentication information */

    xmpp_conn_set_jid(conn1, jid1);

    xmpp_conn_set_pass(conn1, pass1);

    xmpp_conn_set_jid(conn2, jid2);

    xmpp_conn_set_pass(conn2, pass2);

	bstanza = xmpp_stanza_new(ctx);

	xmpp_stanza_set_name(bstanza, "message");

	xmpp_stanza_set_type(bstanza, "chat");

	xmpp_stanza_set_attribute(bstanza, "to", jid2);

	body = xmpp_stanza_new(ctx);

	xmpp_stanza_set_name(body, "body");

	text = xmpp_stanza_new(ctx);

	xmpp_stanza_set_text(text, "Hello");

	xmpp_stanza_add_child(body, text);

	xmpp_stanza_add_child(bstanza, body);

	xmpp_stanza_release(text);

	xmpp_stanza_release(body);

    /* initiate connection */

    xmpp_connect_client(conn1, "localhost", 5222, conn_handler, ctx);

    xmpp_connect_client(conn2, "localhost", 5222, conn_handler, ctx);

    printf("Connecting...\n");

    /* enter the event loop - 

       our connect handler will trigger an exit */

    xmpp_run(ctx);

    /* release our connection and context */

    xmpp_conn_release(conn1);

    xmpp_conn_release(conn2);

    xmpp_ctx_free(ctx);

    /* final shutdown of the library */

    xmpp_shutdown();

    return 0;

}