int XS_TEST_MAIN ()
{
fprintf (stderr, "reqrep_ipc test running...\n");
void *ctx = xs_init ();
assert (ctx);
void *sb = xs_socket (ctx, XS_REP);
assert (sb);
int rc = xs_bind (sb, "ipc:///tmp/tester");
assert (rc == 0);
void *sc = xs_socket (ctx, XS_REQ);
assert (sc);
rc = xs_connect (sc, "ipc:///tmp/tester");
assert (rc == 0);
bounce (sb, sc);
rc = xs_close (sc);
assert (rc == 0);
rc = xs_close (sb);
assert (rc == 0);
rc = xs_term (ctx);
assert (rc == 0);
return 0 ;
}
int XS_TEST_MAIN ()
{
fprintf (stderr, "pair_tcp test running...\n");
void *ctx = xs_init ();
assert (ctx);
void *sb = xs_socket (ctx, XS_PAIR);
assert (sb);
int rc = xs_bind (sb, "tcp://127.0.0.1:5560");
assert (rc == 0);
void *sc = xs_socket (ctx, XS_PAIR);
assert (sc);
rc = xs_connect (sc, "tcp://127.0.0.1:5560");
assert (rc == 0);
bounce (sb, sc);
rc = xs_close (sc);
assert (rc == 0);
rc = xs_close (sb);
assert (rc == 0);
rc = xs_term (ctx);
assert (rc == 0);
return 0 ;
}
int XS_TEST_MAIN ()
{
fprintf (stderr, "hwm test running...\n");
void *ctx = xs_init ();
assert (ctx);
// Create pair of socket, each with high watermark of 2. Thus the total
// buffer space should be 4 messages.
void *sb = xs_socket (ctx, XS_PULL);
assert (sb);
int hwm = 2;
int rc = xs_setsockopt (sb, XS_RCVHWM, &hwm, sizeof (hwm));
assert (rc == 0);
rc = xs_bind (sb, "inproc://a");
assert (rc != -1);
void *sc = xs_socket (ctx, XS_PUSH);
assert (sc);
rc = xs_setsockopt (sc, XS_SNDHWM, &hwm, sizeof (hwm));
assert (rc == 0);
rc = xs_connect (sc, "inproc://a");
assert (rc != -1);
// Try to send 10 messages. Only 4 should succeed.
for (int i = 0; i < 10; i++)
{
rc = xs_send (sc, NULL, 0, XS_DONTWAIT);
if (i < 4)
assert (rc == 0);
else
assert (rc < 0 && errno == EAGAIN);
}
// There should be now 4 messages pending, consume them.
for (int i = 0; i != 4; i++) {
rc = xs_recv (sb, NULL, 0, 0);
assert (rc == 0);
}
// Now it should be possible to send one more.
rc = xs_send (sc, NULL, 0, 0);
assert (rc == 0);
// Consume the remaining message.
rc = xs_recv (sb, NULL, 0, 0);
assert (rc == 0);
rc = xs_close (sc);
assert (rc == 0);
rc = xs_close (sb);
assert (rc == 0);
rc = xs_term (ctx);
assert (rc == 0);
return 0;
}
int XS_TEST_MAIN ()
{
fprintf (stderr, "msg_flags test running...\n");
// Create the infrastructure
void *ctx = xs_init ();
assert (ctx);
void *sb = xs_socket (ctx, XS_XREP);
assert (sb);
int rc = xs_bind (sb, "inproc://a");
assert (rc == 0);
void *sc = xs_socket (ctx, XS_XREQ);
assert (sc);
rc = xs_connect (sc, "inproc://a");
assert (rc == 0);
// Send 2-part message.
rc = xs_send (sc, "A", 1, XS_SNDMORE);
assert (rc == 1);
rc = xs_send (sc, "B", 1, 0);
assert (rc == 1);
// Identity comes first.
xs_msg_t msg;
rc = xs_msg_init (&msg);
assert (rc == 0);
rc = xs_recvmsg (sb, &msg, 0);
assert (rc >= 0);
int more;
size_t more_size = sizeof (more);
rc = xs_getmsgopt (&msg, XS_MORE, &more, &more_size);
assert (rc == 0);
assert (more == 1);
// Then the first part of the message body.
rc = xs_recvmsg (sb, &msg, 0);
assert (rc == 1);
more_size = sizeof (more);
rc = xs_getmsgopt (&msg, XS_MORE, &more, &more_size);
assert (rc == 0);
assert (more == 1);
// And finally, the second part of the message body.
rc = xs_recvmsg (sb, &msg, 0);
assert (rc == 1);
more_size = sizeof (more);
rc = xs_getmsgopt (&msg, XS_MORE, &more, &more_size);
assert (rc == 0);
assert (more == 0);
// Deallocate the infrastructure.
rc = xs_close (sc);
assert (rc == 0);
rc = xs_close (sb);
assert (rc == 0);
rc = xs_term (ctx);
assert (rc == 0);
return 0 ;
}
int main(int argc, char** argv)
{
xs_init();
char* p = xs_malloc(100);
xs_fini();
return 0;
}
static int __init xenbus_init(void)
{
int err = 0;
if (!xen_domain())
return -ENODEV;
if (xen_hvm_domain()) {
uint64_t v = 0;
err = hvm_get_parameter(HVM_PARAM_STORE_EVTCHN, &v);
if (err)
goto out_error;
xen_store_evtchn = (int)v;
err = hvm_get_parameter(HVM_PARAM_STORE_PFN, &v);
if (err)
goto out_error;
xen_store_mfn = (unsigned long)v;
xen_store_interface = ioremap(xen_store_mfn << PAGE_SHIFT, PAGE_SIZE);
} else {
xen_store_evtchn = xen_start_info->store_evtchn;
xen_store_mfn = xen_start_info->store_mfn;
if (xen_store_evtchn)
xenstored_ready = 1;
else {
err = xenstored_local_init();
if (err)
goto out_error;
}
xen_store_interface = mfn_to_virt(xen_store_mfn);
}
/* Initialize the interface to xenstore. */
err = xs_init();
if (err) {
printk(KERN_WARNING
"XENBUS: Error initializing xenstore comms: %i\n", err);
goto out_error;
}
#ifdef CONFIG_XEN_COMPAT_XENFS
/*
* Create xenfs mountpoint in /proc for compatibility with
* utilities that expect to find "xenbus" under "/proc/xen".
*/
proc_mkdir("xen", NULL);
#endif
out_error:
return err;
}
int XS_TEST_MAIN ()
{
void *ctx;
void *s1;
void *s2;
int i;
int j;
int rc;
int io_threads;
void *threads [THREAD_COUNT];
fprintf (stderr, "shutdown_stress test running...\n");
for (j = 0; j != 10; j++) {
// Check the shutdown with many parallel I/O threads.
ctx = xs_init ();
assert (ctx);
io_threads = 7;
rc = xs_setctxopt (ctx, XS_IO_THREADS, &io_threads,
sizeof (io_threads));
assert (rc == 0);
s1 = xs_socket (ctx, XS_PUB);
assert (s1);
rc = xs_bind (s1, "tcp://127.0.0.1:5560");
assert (rc == 0);
for (i = 0; i != THREAD_COUNT; i++) {
s2 = xs_socket (ctx, XS_SUB);
assert (s2);
threads [i] = thread_create (shutdown_stress_worker, s2);
assert (threads [i]);
}
for (i = 0; i != THREAD_COUNT; i++)
thread_join (threads [i]);
rc = xs_close (s1);
assert (rc == 0);
rc = xs_term (ctx);
assert (rc == 0);
}
return 0;
}
int XS_TEST_MAIN ()
{
fprintf (stderr, "polltimeo test running...\n");
void *ctx = xs_init ();
assert (ctx);
// Create a disconnected socket.
void *sb = xs_socket (ctx, XS_PULL);
assert (sb);
int rc = xs_bind (sb, "inproc://timeout_test");
assert (rc == 0);
// Check whether timeout is honoured.
xs_pollitem_t pi;
pi.socket = sb;
pi.events = XS_POLLIN;
void *watch = xs_stopwatch_start ();
rc = xs_poll (&pi, 1, 500);
assert (rc == 0);
unsigned long elapsed = xs_stopwatch_stop (watch) / 1000;
#if !defined _WIN32 || !defined _DEBUG
assert (elapsed > 440 && elapsed < 550);
#endif
// Check whether connection during the wait doesn't distort the timeout.
void *thread = thread_create (polltimeo_worker, ctx);
assert (thread);
watch = xs_stopwatch_start ();
rc = xs_poll (&pi, 1, 2000);
assert (rc == 0);
elapsed = xs_stopwatch_stop (watch) / 1000;
#if !defined _WIN32 || !defined _DEBUG
assert (elapsed > 1900 && elapsed < 2100);
#endif
thread_join (thread);
// Clean-up.
rc = xs_close (sb);
assert (rc == 0);
rc = xs_term (ctx);
assert (rc == 0);
return 0 ;
}
void *zmq_init (int io_threads)
{
void *ctx = xs_init ();
if (!ctx)
return NULL;
// Crossroads don't allow for zero I/O threads.
if (io_threads < 1)
io_threads = 1;
int rc = xs_setctxopt (ctx, XS_IO_THREADS, &io_threads,
sizeof (io_threads));
if (rc != 0) {
xs_term (ctx);
return NULL;
}
return ctx;
}
int XS_TEST_MAIN ()
{
fprintf (stderr, "linger test running...\n");
// Create socket.
void *ctx = xs_init ();
assert (ctx);
void *s = xs_socket (ctx, XS_PUSH);
assert (s);
// Set linger to 0.1 second.
int linger = 100;
int rc = xs_setsockopt (s, XS_LINGER, &linger, sizeof (int));
// Connect to non-existent endpoing.
assert (rc == 0);
rc = xs_connect (s, "tcp://127.0.0.1:5560");
assert (rc == 0);
// Send a message.
rc = xs_send (s, "r", 1, 0);
assert (rc == 1);
// Close the socket.
rc = xs_close (s);
assert (rc == 0);
// Terminate the context. This should take 0.1 second.
void *watch = xs_stopwatch_start ();
rc = xs_term (ctx);
assert (rc == 0);
int ms = (int) xs_stopwatch_stop (watch) / 1000;
#if !defined _WIN32 || !defined _DEBUG
assert (ms > 50 && ms < 150);
#endif
return 0;
}
static int
xenbus_probe(device_t dev)
{
int err = 0;
DPRINTK("");
/* Initialize the interface to xenstore. */
err = xs_init();
if (err) {
log(LOG_WARNING,
"XENBUS: Error initializing xenstore comms: %i\n", err);
return (ENXIO);
}
err = gnttab_init();
if (err) {
log(LOG_WARNING,
"XENBUS: Error initializing grant table: %i\n", err);
return (ENXIO);
}
device_set_desc(dev, "Xen Devices");
return (0);
}
static int __init xenbus_init(void)
{
int err = 0;
uint64_t v = 0;
xen_store_domain_type = XS_UNKNOWN;
if (!xen_domain())
return -ENODEV;
xenbus_ring_ops_init();
if (xen_pv_domain())
xen_store_domain_type = XS_PV;
if (xen_hvm_domain())
xen_store_domain_type = XS_HVM;
if (xen_hvm_domain() && xen_initial_domain())
xen_store_domain_type = XS_LOCAL;
if (xen_pv_domain() && !xen_start_info->store_evtchn)
xen_store_domain_type = XS_LOCAL;
if (xen_pv_domain() && xen_start_info->store_evtchn)
xenstored_ready = 1;
switch (xen_store_domain_type) {
case XS_LOCAL:
err = xenstored_local_init();
if (err)
goto out_error;
xen_store_interface = gfn_to_virt(xen_store_gfn);
break;
case XS_PV:
xen_store_evtchn = xen_start_info->store_evtchn;
xen_store_gfn = xen_start_info->store_mfn;
xen_store_interface = gfn_to_virt(xen_store_gfn);
break;
case XS_HVM:
err = hvm_get_parameter(HVM_PARAM_STORE_EVTCHN, &v);
if (err)
goto out_error;
xen_store_evtchn = (int)v;
err = hvm_get_parameter(HVM_PARAM_STORE_PFN, &v);
if (err)
goto out_error;
xen_store_gfn = (unsigned long)v;
xen_store_interface =
xen_remap(xen_store_gfn << XEN_PAGE_SHIFT,
XEN_PAGE_SIZE);
break;
default:
pr_warn("Xenstore state unknown\n");
break;
}
/* Initialize the interface to xenstore. */
err = xs_init();
if (err) {
pr_warn("Error initializing xenstore comms: %i\n", err);
goto out_error;
}
if ((xen_store_domain_type != XS_LOCAL) &&
(xen_store_domain_type != XS_UNKNOWN))
xen_resume_notifier_register(&xenbus_resume_nb);
#ifdef CONFIG_XEN_COMPAT_XENFS
/*
* Create xenfs mountpoint in /proc for compatibility with
* utilities that expect to find "xenbus" under "/proc/xen".
*/
proc_mkdir("xen", NULL);
#endif
out_error:
return err;
}
int main (int argc, char *argv [])
{
#if defined XS_HAVE_WINDOWS
HANDLE local_thread;
#else
pthread_t local_thread;
#endif
void *ctx;
void *s;
int rc;
int i;
xs_msg_t msg;
void *watch;
unsigned long elapsed;
double latency;
if (argc != 3) {
printf ("usage: inproc_lat <message-size> <roundtrip-count>\n");
return 1;
}
message_size = atoi (argv [1]);
roundtrip_count = atoi (argv [2]);
ctx = xs_init ();
if (!ctx) {
printf ("error in xs_init: %s\n", xs_strerror (errno));
return -1;
}
s = xs_socket (ctx, XS_REQ);
if (!s) {
printf ("error in xs_socket: %s\n", xs_strerror (errno));
return -1;
}
rc = xs_bind (s, "inproc://lat_test");
if (rc == -1) {
printf ("error in xs_bind: %s\n", xs_strerror (errno));
return -1;
}
#if defined XS_HAVE_WINDOWS
local_thread = (HANDLE) _beginthreadex (NULL, 0,
worker, ctx, 0 , NULL);
if (local_thread == 0) {
printf ("error in _beginthreadex\n");
return -1;
}
#else
rc = pthread_create (&local_thread, NULL, worker, ctx);
if (rc != 0) {
printf ("error in pthread_create: %s\n", xs_strerror (rc));
return -1;
}
#endif
rc = xs_msg_init_size (&msg, message_size);
if (rc != 0) {
printf ("error in xs_msg_init_size: %s\n", xs_strerror (errno));
return -1;
}
memset (xs_msg_data (&msg), 0, message_size);
printf ("message size: %d [B]\n", (int) message_size);
printf ("roundtrip count: %d\n", (int) roundtrip_count);
watch = xs_stopwatch_start ();
for (i = 0; i != roundtrip_count; i++) {
rc = xs_sendmsg (s, &msg, 0);
if (rc < 0) {
printf ("error in xs_sendmsg: %s\n", xs_strerror (errno));
return -1;
}
rc = xs_recvmsg (s, &msg, 0);
if (rc < 0) {
printf ("error in xs_recvmsg: %s\n", xs_strerror (errno));
return -1;
}
if (xs_msg_size (&msg) != message_size) {
printf ("message of incorrect size received\n");
return -1;
}
}
elapsed = xs_stopwatch_stop (watch);
rc = xs_msg_close (&msg);
if (rc != 0) {
printf ("error in xs_msg_close: %s\n", xs_strerror (errno));
return -1;
}
latency = (double) elapsed / (roundtrip_count * 2);
#if defined XS_HAVE_WINDOWS
DWORD rc2 = WaitForSingleObject (local_thread, INFINITE);
if (rc2 == WAIT_FAILED) {
printf ("error in WaitForSingleObject\n");
return -1;
}
BOOL rc3 = CloseHandle (local_thread);
if (rc3 == 0) {
printf ("error in CloseHandle\n");
return -1;
}
#else
rc = pthread_join (local_thread, NULL);
if (rc != 0) {
printf ("error in pthread_join: %s\n", xs_strerror (rc));
return -1;
}
#endif
printf ("average latency: %.3f [us]\n", (double) latency);
rc = xs_close (s);
if (rc != 0) {
printf ("error in xs_close: %s\n", xs_strerror (errno));
return -1;
}
rc = xs_term (ctx);
if (rc != 0) {
printf ("error in xs_term: %s\n", xs_strerror (errno));
return -1;
}
return 0;
}
static int __init xenbus_init(void)
{
int err = 0;
enum xenstore_init usage = UNKNOWN;
uint64_t v = 0;
if (!xen_domain())
return -ENODEV;
xenbus_ring_ops_init();
if (xen_pv_domain())
usage = PV;
if (xen_hvm_domain())
usage = HVM;
if (xen_hvm_domain() && xen_initial_domain())
usage = LOCAL;
if (xen_pv_domain() && !xen_start_info->store_evtchn)
usage = LOCAL;
if (xen_pv_domain() && xen_start_info->store_evtchn)
xenstored_ready = 1;
switch (usage) {
case LOCAL:
err = xenstored_local_init();
if (err)
goto out_error;
xen_store_interface = mfn_to_virt(xen_store_mfn);
break;
case PV:
xen_store_evtchn = xen_start_info->store_evtchn;
xen_store_mfn = xen_start_info->store_mfn;
xen_store_interface = mfn_to_virt(xen_store_mfn);
break;
case HVM:
err = hvm_get_parameter(HVM_PARAM_STORE_EVTCHN, &v);
if (err)
goto out_error;
xen_store_evtchn = (int)v;
err = hvm_get_parameter(HVM_PARAM_STORE_PFN, &v);
if (err)
goto out_error;
xen_store_mfn = (unsigned long)v;
xen_store_interface =
xen_remap(xen_store_mfn << PAGE_SHIFT, PAGE_SIZE);
break;
default:
pr_warn("Xenstore state unknown\n");
break;
}
/* Initialize the interface to xenstore. */
err = xs_init();
if (err) {
printk(KERN_WARNING
"XENBUS: Error initializing xenstore comms: %i\n", err);
goto out_error;
}
#ifdef CONFIG_XEN_COMPAT_XENFS
/*
* Create xenfs mountpoint in /proc for compatibility with
* utilities that expect to find "xenbus" under "/proc/xen".
*/
proc_mkdir("xen", NULL);
#endif
out_error:
return err;
}
static int __init xenbus_probe_init(void)
{
int err = 0;
DPRINTK("");
err = -ENODEV;
if (!xen_domain())
goto out_error;
/* Register ourselves with the kernel bus subsystem */
err = bus_register(&xenbus_frontend.bus);
if (err)
goto out_error;
err = xenbus_backend_bus_register();
if (err)
goto out_unreg_front;
/*
* Domain0 doesn't have a store_evtchn or store_mfn yet.
*/
if (xen_initial_domain()) {
/* dom0 not yet supported */
} else {
xenstored_ready = 1;
xen_store_evtchn = xen_start_info->store_evtchn;
xen_store_mfn = xen_start_info->store_mfn;
}
xen_store_interface = mfn_to_virt(xen_store_mfn);
/* Initialize the interface to xenstore. */
err = xs_init();
if (err) {
printk(KERN_WARNING
"XENBUS: Error initializing xenstore comms: %i\n", err);
goto out_unreg_back;
}
if (!xen_initial_domain())
xenbus_probe(NULL);
#ifdef CONFIG_XEN_COMPAT_XENFS
/*
* Create xenfs mountpoint in /proc for compatibility with
* utilities that expect to find "xenbus" under "/proc/xen".
*/
proc_mkdir("xen", NULL);
#endif
return 0;
out_unreg_back:
xenbus_backend_bus_unregister();
out_unreg_front:
bus_unregister(&xenbus_frontend.bus);
out_error:
return err;
}
static int __init xenbus_init(void)
{
int err = 0;
DPRINTK("");
err = -ENODEV;
if (!xen_domain())
goto out_error;
/* Register ourselves with the kernel bus subsystem */
err = bus_register(&xenbus_frontend.bus);
if (err)
goto out_error;
err = xenbus_backend_bus_register();
if (err)
goto out_unreg_front;
/*
* Domain0 doesn't have a store_evtchn or store_mfn yet.
*/
if (xen_initial_domain()) {
/* dom0 not yet supported */
} else {
if (xen_hvm_domain()) {
uint64_t v = 0;
err = hvm_get_parameter(HVM_PARAM_STORE_EVTCHN, &v);
if (err)
goto out_error;
xen_store_evtchn = (int)v;
err = hvm_get_parameter(HVM_PARAM_STORE_PFN, &v);
if (err)
goto out_error;
xen_store_mfn = (unsigned long)v;
xen_store_interface = ioremap(xen_store_mfn << PAGE_SHIFT, PAGE_SIZE);
} else {
xen_store_evtchn = xen_start_info->store_evtchn;
xen_store_mfn = xen_start_info->store_mfn;
xen_store_interface = mfn_to_virt(xen_store_mfn);
xenstored_ready = 1;
}
}
/* Initialize the interface to xenstore. */
err = xs_init();
if (err) {
printk(KERN_WARNING
"XENBUS: Error initializing xenstore comms: %i\n", err);
goto out_unreg_back;
}
#ifdef CONFIG_XEN_COMPAT_XENFS
/*
* Create xenfs mountpoint in /proc for compatibility with
* utilities that expect to find "xenbus" under "/proc/xen".
*/
proc_mkdir("xen", NULL);
#endif
return 0;
out_unreg_back:
xenbus_backend_bus_unregister();
out_unreg_front:
bus_unregister(&xenbus_frontend.bus);
out_error:
return err;
}
int XS_TEST_MAIN ()
{
int rc;
char buf [32];
fprintf (stderr, "survey test running...\n");
// Create the basic infrastructure.
void *ctx = xs_init ();
assert (ctx);
void *xsurveyor = xs_socket (ctx, XS_XSURVEYOR);
assert (xsurveyor);
rc = xs_bind (xsurveyor, "inproc://a");
assert (rc != -1);
void *xrespondent = xs_socket (ctx, XS_XRESPONDENT);
assert (xrespondent);
rc = xs_bind (xrespondent, "inproc://b");
assert (rc != -1);
void *surveyor = xs_socket (ctx, XS_SURVEYOR);
assert (surveyor);
rc = xs_connect (surveyor, "inproc://b");
assert (rc != -1);
void *respondent1 = xs_socket (ctx, XS_RESPONDENT);
assert (respondent1);
rc = xs_connect (respondent1, "inproc://a");
assert (rc != -1);
void *respondent2 = xs_socket (ctx, XS_RESPONDENT);
assert (respondent2);
rc = xs_connect (respondent2, "inproc://a");
assert (rc != -1);
// Send the survey.
rc = xs_send (surveyor, "ABC", 3, 0);
assert (rc == 3);
// Forward the survey through the intermediate device.
// Survey consist of identity (4 bytes), survey ID (4 bytes) and the body.
rc = xs_recv (xrespondent, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xsurveyor, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xrespondent, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xsurveyor, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xrespondent, buf, sizeof (buf), 0);
assert (rc == 3);
rc = xs_send (xsurveyor, buf, 3, 0);
assert (rc == 3);
// Respondent 1 responds to the survey.
rc = xs_recv (respondent1, buf, sizeof (buf), 0);
assert (rc == 3);
rc = xs_send (respondent1, "DE", 2, 0);
assert (rc == 2);
// Forward the response through the intermediate device.
rc = xs_recv (xsurveyor, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xrespondent, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xsurveyor, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xrespondent, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xsurveyor, buf, sizeof (buf), 0);
assert (rc == 2);
rc = xs_send (xrespondent, buf, 2, 0);
assert (rc == 2);
// Surveyor gets the response.
rc = xs_recv (surveyor, buf, sizeof (buf), 0);
assert (rc == 2);
// Respondent 2 responds to the survey.
rc = xs_recv (respondent2, buf, sizeof (buf), 0);
assert (rc == 3);
rc = xs_send (respondent2, "FGHI", 4, 0);
assert (rc == 4);
// Forward the response through the intermediate device.
rc = xs_recv (xsurveyor, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xrespondent, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xsurveyor, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xrespondent, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xsurveyor, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xrespondent, buf, 4, 0);
assert (rc == 4);
// Surveyor gets the response.
rc = xs_recv (surveyor, buf, sizeof (buf), 0);
assert (rc == 4);
// Now let's test whether survey timeout works as expected.
int timeout = 100;
rc = xs_setsockopt (surveyor, XS_SURVEY_TIMEOUT, &timeout, sizeof (int));
assert (rc == 0);
rc = xs_send (surveyor, "ABC", 3, 0);
assert (rc == 3);
void *watch = xs_stopwatch_start ();
rc = xs_recv (surveyor, buf, sizeof (buf), 0);
assert (rc == - 1 && errno == EAGAIN);
unsigned long elapsed = xs_stopwatch_stop (watch) / 1000;
time_assert (elapsed, (unsigned long) timeout);
// Test whether responses for old surveys are discarded. First,
// initiate new survey.
rc = xs_setsockopt (surveyor, XS_SURVEY_TIMEOUT, &timeout, sizeof (int));
assert (rc == 0);
rc = xs_send (surveyor, "DE", 2, 0);
assert (rc == 2);
// Read, process and reply to the old survey.
rc = xs_recv (xrespondent, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xrespondent, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xrespondent, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xrespondent, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xrespondent, buf, sizeof (buf), 0);
assert (rc == 3);
rc = xs_send (xrespondent, buf, 3, 0);
assert (rc == 3);
// Read, process and reply to the new survey.
rc = xs_recv (xrespondent, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xrespondent, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xrespondent, buf, sizeof (buf), 0);
assert (rc == 4);
rc = xs_send (xrespondent, buf, 4, XS_SNDMORE);
assert (rc == 4);
rc = xs_recv (xrespondent, buf, sizeof (buf), 0);
assert (rc == 2);
rc = xs_send (xrespondent, buf, 2, 0);
assert (rc == 2);
// Get the response and check it's the response to the new survey and
// that response to the old survey was silently discarded.
rc = xs_recv (surveyor, buf, sizeof (buf), 0);
assert (rc == 2);
rc = xs_close (respondent2);
assert (rc == 0);
rc = xs_close (respondent1);
assert (rc == 0);
rc = xs_close (surveyor);
assert (rc == 0);
rc = xs_close (xrespondent);
assert (rc == 0);
rc = xs_close (xsurveyor);
assert (rc == 0);
rc = xs_term (ctx);
assert (rc == 0);
return 0 ;
}
int XS_TEST_MAIN ()
{
fprintf (stderr, "sub_forward test running...\n");
void *ctx = xs_init ();
assert (ctx);
// First, create an intermediate device.
void *xpub = xs_socket (ctx, XS_XPUB);
assert (xpub);
int rc = xs_bind (xpub, "tcp://127.0.0.1:5560");
assert (rc == 0);
void *xsub = xs_socket (ctx, XS_XSUB);
assert (xsub);
rc = xs_bind (xsub, "tcp://127.0.0.1:5561");
assert (rc == 0);
// Create a publisher.
void *pub = xs_socket (ctx, XS_PUB);
assert (pub);
rc = xs_connect (pub, "tcp://127.0.0.1:5561");
assert (rc == 0);
// Create a subscriber.
void *sub = xs_socket (ctx, XS_SUB);
assert (sub);
rc = xs_connect (sub, "tcp://127.0.0.1:5560");
assert (rc == 0);
// Subscribe for all messages.
rc = xs_setsockopt (sub, XS_SUBSCRIBE, "", 0);
assert (rc == 0);
// Pass the subscription upstream through the device.
char buff [32];
rc = xs_recv (xpub, buff, sizeof (buff), 0);
assert (rc >= 0);
rc = xs_send (xsub, buff, rc, 0);
assert (rc >= 0);
// Wait a bit till the subscription gets to the publisher.
sleep (1);
// Send an empty message.
rc = xs_send (pub, NULL, 0, 0);
assert (rc == 0);
// Pass the message downstream through the device.
rc = xs_recv (xsub, buff, sizeof (buff), 0);
assert (rc >= 0);
rc = xs_send (xpub, buff, rc, 0);
assert (rc >= 0);
// Receive the message in the subscriber.
rc = xs_recv (sub, buff, sizeof (buff), 0);
assert (rc == 0);
// Clean up.
rc = xs_close (xpub);
assert (rc == 0);
rc = xs_close (xsub);
assert (rc == 0);
rc = xs_close (pub);
assert (rc == 0);
rc = xs_close (sub);
assert (rc == 0);
rc = xs_term (ctx);
assert (rc == 0);
return 0 ;
}
int main (int argc, char *argv [])
{
#if defined XS_HAVE_WINDOWS
HANDLE local_thread;
#else
pthread_t local_thread;
#endif
void *ctx;
void *s;
int rc;
int i;
xs_msg_t msg;
void *watch;
unsigned long elapsed;
unsigned long throughput;
double megabits;
if (argc != 3) {
printf ("usage: thread_thr <message-size> <message-count>\n");
return 1;
}
message_size = atoi (argv [1]);
message_count = atoi (argv [2]);
ctx = xs_init ();
if (!ctx) {
printf ("error in xs_init: %s\n", xs_strerror (errno));
return -1;
}
s = xs_socket (ctx, XS_PULL);
if (!s) {
printf ("error in xs_socket: %s\n", xs_strerror (errno));
return -1;
}
rc = xs_bind (s, "inproc://thr_test");
if (rc == -1) {
printf ("error in xs_bind: %s\n", xs_strerror (errno));
return -1;
}
#if defined XS_HAVE_WINDOWS
local_thread = (HANDLE) _beginthreadex (NULL, 0,
worker, ctx, 0 , NULL);
if (local_thread == 0) {
printf ("error in _beginthreadex\n");
return -1;
}
#else
rc = pthread_create (&local_thread, NULL, worker, ctx);
if (rc != 0) {
printf ("error in pthread_create: %s\n", xs_strerror (rc));
return -1;
}
#endif
rc = xs_msg_init (&msg);
if (rc != 0) {
printf ("error in xs_msg_init: %s\n", xs_strerror (errno));
return -1;
}
printf ("message size: %d [B]\n", (int) message_size);
printf ("message count: %d\n", (int) message_count);
rc = xs_recvmsg (s, &msg, 0);
if (rc < 0) {
printf ("error in xs_recvmsg: %s\n", xs_strerror (errno));
return -1;
}
if (xs_msg_size (&msg) != message_size) {
printf ("message of incorrect size received\n");
return -1;
}
watch = xs_stopwatch_start ();
for (i = 0; i != message_count - 1; i++) {
rc = xs_recvmsg (s, &msg, 0);
if (rc < 0) {
printf ("error in xs_recvmsg: %s\n", xs_strerror (errno));
return -1;
}
if (xs_msg_size (&msg) != message_size) {
printf ("message of incorrect size received\n");
return -1;
}
}
elapsed = xs_stopwatch_stop (watch);
if (elapsed == 0)
elapsed = 1;
rc = xs_msg_close (&msg);
if (rc != 0) {
printf ("error in xs_msg_close: %s\n", xs_strerror (errno));
return -1;
}
#if defined XS_HAVE_WINDOWS
DWORD rc2 = WaitForSingleObject (local_thread, INFINITE);
if (rc2 == WAIT_FAILED) {
printf ("error in WaitForSingleObject\n");
return -1;
}
BOOL rc3 = CloseHandle (local_thread);
if (rc3 == 0) {
printf ("error in CloseHandle\n");
return -1;
}
#else
rc = pthread_join (local_thread, NULL);
if (rc != 0) {
printf ("error in pthread_join: %s\n", xs_strerror (rc));
return -1;
}
#endif
rc = xs_close (s);
if (rc != 0) {
printf ("error in xs_close: %s\n", xs_strerror (errno));
return -1;
}
rc = xs_term (ctx);
if (rc != 0) {
printf ("error in xs_term: %s\n", xs_strerror (errno));
return -1;
}
throughput = (unsigned long)
((double) message_count / (double) elapsed * 1000000);
megabits = (double) (throughput * message_size * 8) / 1000000;
printf ("mean throughput: %d [msg/s]\n", (int) throughput);
printf ("mean throughput: %.3f [Mb/s]\n", (double) megabits);
return 0;
}
void
process_args(nmsgtool_ctx *c) {
char *t;
FILE *fp_pidfile = NULL;
nmsg_msgmod_t mod = NULL;
if (c->help)
usage(NULL);
if (c->version) {
#ifdef HAVE_LIBXS
fprintf(stderr, "%s: version %s\n", argv_program, PACKAGE_VERSION);
#else /* HAVE_LIBXS */
fprintf(stderr, "%s: version %s (without libxs support)\n",
argv_program, PACKAGE_VERSION);
#endif /* HAVE_LIBXS */
exit(EXIT_SUCCESS);
}
if (c->endline == NULL)
c->endline_str = strdup("\n");
else
c->endline_str = unescape(c->endline);
if (c->mtu == 0)
c->mtu = NMSG_WBUFSZ_JUMBO;
if (c->vname == NULL && c->mname != NULL)
c->vname = "base";
if (c->vname != NULL) {
if (c->mname == NULL)
usage("-V requires -T");
c->vid = nmsg_msgmod_vname_to_vid(c->vname);
if (c->vid == 0)
usage("invalid vendor ID");
if (c->debug >= 2)
fprintf(stderr, "%s: input vendor = %s\n",
argv_program, c->vname);
}
if (c->mname != NULL) {
if (c->vname == NULL)
usage("-T requires -V");
c->msgtype = nmsg_msgmod_mname_to_msgtype(c->vid, c->mname);
if (c->msgtype == 0)
usage("invalid message type");
if (c->debug >= 2)
fprintf(stderr, "%s: input msgtype = %s\n",
argv_program, c->mname);
}
if (c->debug < 1)
c->debug = 1;
if (c->debug > 0)
nmsg_io_set_debug(c->io, c->debug);
if (c->count > 0)
nmsg_io_set_count(c->io, c->count);
if (c->interval > 0)
nmsg_io_set_interval(c->io, c->interval);
if (c->mirror == true)
nmsg_io_set_output_mode(c->io, nmsg_io_output_mode_mirror);
/* bpf string */
if (c->bpfstr == NULL) {
t = getenv("NMSG_BPF");
if (t != NULL)
c->bpfstr = strdup(t);
}
/* kicker command */
if (c->kicker == NULL) {
t = getenv("NMSG_KICKER");
if (t != NULL)
c->kicker = strdup(t);
}
/* set source, operator, group */
if (c->set_source_str != NULL) {
c->set_source = (unsigned) strtoul(c->set_source_str, &t, 0);
if (*t != '\0')
usage("invalid source ID");
if (c->debug >= 2)
fprintf(stderr, "%s: nmsg source set to %#.08x\n",
argv_program, c->set_source);
}
if (c->set_operator_str != NULL) {
c->set_operator = nmsg_alias_by_value(nmsg_alias_operator,
c->set_operator_str);
if (c->set_operator == 0)
usage("unknown operator name");
if (c->debug >= 2)
fprintf(stderr, "%s: nmsg operator set to '%s' (%u)\n",
argv_program,
c->set_operator_str,
c->set_operator);
}
if (c->set_group_str != NULL) {
c->set_group = nmsg_alias_by_value(nmsg_alias_group,
c->set_group_str);
if (c->set_group == 0)
usage("unknown group name");
if (c->debug >= 2)
fprintf(stderr, "%s: nmsg group set to '%s' (%u)\n",
argv_program,
c->set_group_str,
c->set_group);
}
/* get source, operator, group */
if (c->get_source_str != NULL) {
c->get_source = (unsigned) strtoul(c->get_source_str, &t, 0);
if (*t != '\0')
usage("invalid filter source ID");
if (c->debug >= 2)
fprintf(stderr, "%s: nmsg source filter set to "
"%#.08x\n",
argv_program, c->get_source);
}
if (c->get_operator_str != NULL) {
c->get_operator = nmsg_alias_by_value(nmsg_alias_operator,
c->get_operator_str);
if (c->get_operator == 0)
usage("unknown filter operator name");
if (c->debug >= 2)
fprintf(stderr, "%s: nmsg filter operator set to "
"'%s' (%u)\n",
argv_program,
c->get_operator_str,
c->get_operator);
}
if (c->get_group_str != NULL) {
c->get_group = nmsg_alias_by_value(nmsg_alias_group,
c->get_group_str);
if (c->get_group == 0)
usage("unknown filter group name");
if (c->debug >= 2)
fprintf(stderr, "%s: nmsg filter group set to "
"'%s' (%u)\n",
argv_program,
c->get_group_str,
c->get_group);
}
/* -V, -T sanity check */
if (ARGV_ARRAY_COUNT(c->r_pres) > 0 ||
ARGV_ARRAY_COUNT(c->r_pcapfile) > 0 ||
ARGV_ARRAY_COUNT(c->r_pcapif) > 0)
{
if (c->vname == NULL || c->mname == NULL)
usage("reading presentation or pcap data requires "
"-V, -T");
mod = nmsg_msgmod_lookup(c->vid, c->msgtype);
if (mod == NULL)
usage("unknown msgmod");
}
#define process_args_loop(arry, func) do { \
for (int i = 0; i < ARGV_ARRAY_COUNT(arry); i++) \
func(c, *ARGV_ARRAY_ENTRY_P(arry, char *, i)); \
} while(0)
#define process_args_loop_mod(arry, func, mod) do { \
for (int i = 0; i < ARGV_ARRAY_COUNT(arry); i++) \
func(c, mod, *ARGV_ARRAY_ENTRY_P(arry, char *, i)); \
} while(0)
/* pcap interface inputs */
process_args_loop_mod(c->r_pcapif, add_pcapif_input, mod);
/* open pidfile if necessary */
if (c->pidfile != NULL)
fp_pidfile = pidfile_open(c->pidfile);
else
fp_pidfile = NULL;
/* drop privileges */
if (c->username != NULL)
droproot(c, fp_pidfile);
/* pcap file inputs */
process_args_loop_mod(c->r_pcapfile, add_pcapfile_input, mod);
/* XS context */
if (ARGV_ARRAY_COUNT(c->r_xsock) > 0 ||
ARGV_ARRAY_COUNT(c->w_xsock) > 0 ||
ARGV_ARRAY_COUNT(c->r_xchannel) > 0)
{
#ifdef HAVE_LIBXS
c->xs_ctx = xs_init();
if (c->xs_ctx == NULL) {
fprintf(stderr, "%s: xs_init() failed: %s\n",
argv_program, strerror(errno));
exit(EXIT_FAILURE);
}
#else /* HAVE_LIBXS */
fprintf(stderr, "%s: Error: compiled without libxs support\n",
argv_program);
exit(EXIT_FAILURE);
#endif /* HAVE_LIBXS */
}
/* nmsg inputs and outputs */
process_args_loop(c->r_sock, add_sock_input);
process_args_loop(c->w_sock, add_sock_output);
process_args_loop(c->r_xsock, add_xsock_input);
process_args_loop(c->w_xsock, add_xsock_output);
process_args_loop(c->r_nmsg, add_file_input);
process_args_loop(c->w_nmsg, add_file_output);
for (int i = 0; i < ARGV_ARRAY_COUNT(c->r_channel); i++) {
char *ch;
char **alias = NULL;
int num_aliases;
ch = *ARGV_ARRAY_ENTRY_P(c->r_channel, char *, i);
if (c->debug >= 2)
fprintf(stderr, "%s: looking up channel '%s'\n", argv_program, ch);
num_aliases = nmsg_chalias_lookup(ch, &alias);
if (num_aliases <= 0)
usage("channel alias lookup failed");
for (int j = 0; j < num_aliases; j++) {
if (strstr(alias[j], "://"))
usage("channel alias appears to be an xchannel");
add_sock_input(c, alias[j]);
}
nmsg_chalias_free(&alias);
}
for (int i = 0; i < ARGV_ARRAY_COUNT(c->r_xchannel); i++) {
char *ch;
char **alias = NULL;
int num_aliases;
ch = *ARGV_ARRAY_ENTRY_P(c->r_xchannel, char *, i);
if (c->debug >= 2)
fprintf(stderr, "%s: looking up xchannel '%s'\n", argv_program, ch);
num_aliases = nmsg_chalias_lookup(ch, &alias);
if (num_aliases <= 0)
usage("xchannel alias lookup failed");
for (int j = 0; j < num_aliases; j++)
add_xsock_input(c, alias[j]);
nmsg_chalias_free(&alias);
}
/* pres inputs and outputs */
process_args_loop_mod(c->r_pres, add_pres_input, mod);
process_args_loop(c->w_pres, add_pres_output);
#undef process_args_loop
#undef process_args_loop_mod
/* validation */
if (c->n_inputs == 0)
usage("no data sources specified");
if (c->n_outputs == 0) {
/* implicit "-o -" */
add_pres_output(c, "-");
}
/* daemonize if necessary */
if (c->daemon) {
if (!daemonize()) {
fprintf(stderr, "nmsgtool: unable to daemonize: %s\n",
strerror(errno));
exit(EXIT_FAILURE);
}
}
/* write pidfile if necessary */
if (c->pidfile != NULL && fp_pidfile != NULL)
pidfile_write(fp_pidfile);
}
int XS_TEST_MAIN ()
{
fprintf (stderr, "reqrep_device test running...\n");
void *ctx = xs_init ();
errno_assert (ctx);
// Create a req/rep device.
void *xreq = xs_socket (ctx, XS_XREQ);
errno_assert (xreq);
int rc = xs_bind (xreq, "tcp://127.0.0.1:5560");
errno_assert (rc != -1);
void *xrep = xs_socket (ctx, XS_XREP);
errno_assert (xrep);
rc = xs_bind (xrep, "tcp://127.0.0.1:5561");
errno_assert (rc != -1);
// Create a worker.
void *rep = xs_socket (ctx, XS_REP);
errno_assert (rep);
rc = xs_connect (rep, "tcp://127.0.0.1:5560");
errno_assert (rc != -1);
// Create a client.
void *req = xs_socket (ctx, XS_REQ);
errno_assert (req);
rc = xs_connect (req, "tcp://127.0.0.1:5561");
errno_assert (rc != -1);
// Send a request.
rc = xs_send (req, "ABC", 3, XS_SNDMORE);
errno_assert (rc == 3);
rc = xs_send (req, "DEF", 3, 0);
errno_assert (rc == 3);
// Pass the request through the device.
for (int i = 0; i != 4; i++) {
xs_msg_t msg;
rc = xs_msg_init (&msg);
errno_assert (rc == 0);
rc = xs_recvmsg (xrep, &msg, 0);
errno_assert (rc >= 0);
int rcvmore;
size_t sz = sizeof (rcvmore);
rc = xs_getsockopt (xrep, XS_RCVMORE, &rcvmore, &sz);
errno_assert (rc == 0);
rc = xs_sendmsg (xreq, &msg, rcvmore ? XS_SNDMORE : 0);
errno_assert (rc >= 0);
}
// Receive the request.
char buff [3];
rc = xs_recv (rep, buff, 3, 0);
errno_assert (rc == 3);
assert (memcmp (buff, "ABC", 3) == 0);
int rcvmore;
size_t sz = sizeof (rcvmore);
rc = xs_getsockopt (rep, XS_RCVMORE, &rcvmore, &sz);
errno_assert (rc == 0);
assert (rcvmore);
rc = xs_recv (rep, buff, 3, 0);
errno_assert (rc == 3);
assert (memcmp (buff, "DEF", 3) == 0);
rc = xs_getsockopt (rep, XS_RCVMORE, &rcvmore, &sz);
errno_assert (rc == 0);
assert (!rcvmore);
// Send the reply.
rc = xs_send (rep, "GHI", 3, XS_SNDMORE);
errno_assert (rc == 3);
rc = xs_send (rep, "JKL", 3, 0);
errno_assert (rc == 3);
// Pass the reply through the device.
for (int i = 0; i != 4; i++) {
xs_msg_t msg;
rc = xs_msg_init (&msg);
errno_assert (rc == 0);
rc = xs_recvmsg (xreq, &msg, 0);
errno_assert (rc >= 0);
rc = xs_getsockopt (xreq, XS_RCVMORE, &rcvmore, &sz);
errno_assert (rc == 0);
rc = xs_sendmsg (xrep, &msg, rcvmore ? XS_SNDMORE : 0);
errno_assert (rc >= 0);
}
// Receive the reply.
rc = xs_recv (req, buff, 3, 0);
errno_assert (rc == 3);
assert (memcmp (buff, "GHI", 3) == 0);
rc = xs_getsockopt (req, XS_RCVMORE, &rcvmore, &sz);
errno_assert (rc == 0);
assert (rcvmore);
rc = xs_recv (req, buff, 3, 0);
errno_assert (rc == 3);
assert (memcmp (buff, "JKL", 3) == 0);
rc = xs_getsockopt (req, XS_RCVMORE, &rcvmore, &sz);
errno_assert (rc == 0);
assert (!rcvmore);
// Clean up.
rc = xs_close (req);
errno_assert (rc == 0);
rc = xs_close (rep);
errno_assert (rc == 0);
rc = xs_close (xrep);
errno_assert (rc == 0);
rc = xs_close (xreq);
errno_assert (rc == 0);
rc = xs_term (ctx);
errno_assert (rc == 0);
return 0 ;
}
int main (int argc, char *argv [])
{
const char *bind_to;
int message_count;
size_t message_size;
void *ctx;
void *s;
int rc;
int i;
xs_msg_t msg;
void *watch;
unsigned long elapsed;
unsigned long throughput;
double megabits;
if (argc != 4) {
printf ("usage: local_thr <bind-to> <message-size> <message-count>\n");
return 1;
}
bind_to = argv [1];
message_size = atoi (argv [2]);
message_count = atoi (argv [3]);
ctx = xs_init ();
if (!ctx) {
printf ("error in xs_init: %s\n", xs_strerror (errno));
return -1;
}
s = xs_socket (ctx, XS_PULL);
if (!s) {
printf ("error in xs_socket: %s\n", xs_strerror (errno));
return -1;
}
// Add your socket options here.
// For example XS_RATE, XS_RECOVERY_IVL and XS_MCAST_LOOP for PGM.
rc = xs_bind (s, bind_to);
if (rc == -1) {
printf ("error in xs_bind: %s\n", xs_strerror (errno));
return -1;
}
rc = xs_msg_init (&msg);
if (rc != 0) {
printf ("error in xs_msg_init: %s\n", xs_strerror (errno));
return -1;
}
rc = xs_recvmsg (s, &msg, 0);
if (rc < 0) {
printf ("error in xs_recvmsg: %s\n", xs_strerror (errno));
return -1;
}
if (xs_msg_size (&msg) != message_size) {
printf ("message of incorrect size received\n");
return -1;
}
watch = xs_stopwatch_start ();
for (i = 0; i != message_count - 1; i++) {
rc = xs_recvmsg (s, &msg, 0);
if (rc < 0) {
printf ("error in xs_recvmsg: %s\n", xs_strerror (errno));
return -1;
}
if (xs_msg_size (&msg) != message_size) {
printf ("message of incorrect size received\n");
return -1;
}
}
elapsed = xs_stopwatch_stop (watch);
if (elapsed == 0)
elapsed = 1;
rc = xs_msg_close (&msg);
if (rc != 0) {
printf ("error in xs_msg_close: %s\n", xs_strerror (errno));
return -1;
}
throughput = (unsigned long)
((double) message_count / (double) elapsed * 1000000);
megabits = (double) (throughput * message_size * 8) / 1000000;
printf ("message size: %d [B]\n", (int) message_size);
printf ("message count: %d\n", (int) message_count);
printf ("mean throughput: %d [msg/s]\n", (int) throughput);
printf ("mean throughput: %.3f [Mb/s]\n", (double) megabits);
rc = xs_close (s);
if (rc != 0) {
printf ("error in xs_close: %s\n", xs_strerror (errno));
return -1;
}
rc = xs_term (ctx);
if (rc != 0) {
printf ("error in xs_term: %s\n", xs_strerror (errno));
return -1;
}
return 0;
}
int main (int argc, char *argv [])
{
const char *connect_to;
int roundtrip_count;
size_t message_size;
void *ctx;
void *s;
int rc;
int i;
xs_msg_t msg;
void *watch;
unsigned long elapsed;
double latency;
if (argc != 4) {
printf ("usage: remote_lat <connect-to> <message-size> "
"<roundtrip-count>\n");
return 1;
}
connect_to = argv [1];
message_size = atoi (argv [2]);
roundtrip_count = atoi (argv [3]);
ctx = xs_init ();
if (!ctx) {
printf ("error in xs_init: %s\n", xs_strerror (errno));
return -1;
}
s = xs_socket (ctx, XS_REQ);
if (!s) {
printf ("error in xs_socket: %s\n", xs_strerror (errno));
return -1;
}
rc = xs_connect (s, connect_to);
if (rc == -1) {
printf ("error in xs_connect: %s\n", xs_strerror (errno));
return -1;
}
rc = xs_msg_init_size (&msg, message_size);
if (rc != 0) {
printf ("error in xs_msg_init_size: %s\n", xs_strerror (errno));
return -1;
}
memset (xs_msg_data (&msg), 0, message_size);
watch = xs_stopwatch_start ();
for (i = 0; i != roundtrip_count; i++) {
rc = xs_sendmsg (s, &msg, 0);
if (rc < 0) {
printf ("error in xs_sendmsg: %s\n", xs_strerror (errno));
return -1;
}
rc = xs_recvmsg (s, &msg, 0);
if (rc < 0) {
printf ("error in xs_recvmsg: %s\n", xs_strerror (errno));
return -1;
}
if (xs_msg_size (&msg) != message_size) {
printf ("message of incorrect size received\n");
return -1;
}
}
elapsed = xs_stopwatch_stop (watch);
rc = xs_msg_close (&msg);
if (rc != 0) {
printf ("error in xs_msg_close: %s\n", xs_strerror (errno));
return -1;
}
latency = (double) elapsed / (roundtrip_count * 2);
printf ("message size: %d [B]\n", (int) message_size);
printf ("roundtrip count: %d\n", (int) roundtrip_count);
printf ("average latency: %.3f [us]\n", (double) latency);
rc = xs_close (s);
if (rc != 0) {
printf ("error in xs_close: %s\n", xs_strerror (errno));
return -1;
}
rc = xs_term (ctx);
if (rc != 0) {
printf ("error in xs_term: %s\n", xs_strerror (errno));
return -1;
}
return 0;
}
