XenDomainWatcher::XenDomainWatcher( LogHelper *logHelper )
: xsh_( NULL ), xci_( NULL ), introduceToken_( "introduce" ), releaseToken_( "release" ), logHelper_( logHelper )
{
xsh_ = xs_open( 0 );
if ( !xsh_ )
throw Exception( "xs_open() failed" );
if ( !xs_watch( xsh_, "@introduceDomain", introduceToken_.c_str() ) ) {
xs_close( xsh_ );
throw Exception( "xs_watch() failed" );
}
if ( !xs_watch( xsh_, "@releaseDomain", releaseToken_.c_str() ) ) {
xs_unwatch( xsh_, "@introduceDomain", introduceToken_.c_str() );
xs_close( xsh_ );
throw Exception( "xs_watch() failed" );
}
xci_ = xc_interface_open( NULL, NULL, 0 );
if ( !xci_ ) {
xs_unwatch( xsh_, "@introduceDomain", introduceToken_.c_str() );
xs_unwatch( xsh_, "@releaseDomain", releaseToken_.c_str() );
xs_close( xsh_ );
throw Exception( "xc_interface_init() failed" );
}
}
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, "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 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;
}
char *get_vm_name(int dom, int *target_dom)
{
struct xs_handle *xs;
char buf[64];
char *name;
char *target_dom_str;
unsigned int len = 0;
xs = xs_open(0);
if (!xs) {
perror("xs_daemon_open");
exit(1);
}
snprintf(buf, sizeof(buf), "/local/domain/%d/target", dom);
target_dom_str = xs_read(xs, 0, buf, &len);
if (target_dom_str) {
errno = 0;
*target_dom = strtol(target_dom_str, (char **) NULL, 10);
if (errno != 0) {
perror("strtol");
exit(1);
}
} else
*target_dom = dom;
snprintf(buf, sizeof(buf), "/local/domain/%d/name", *target_dom);
name = xs_read(xs, 0, buf, &len);
if (!name) {
perror("xs_read domainname");
exit(1);
}
xs_close(xs);
return name;
}
int libvchan_wait(libvchan_t *ctrl) {
int ret = -2; /* invalid, so can be distinguished from real
libxenvchan_wait return code */
struct xs_handle *xs;
if (ctrl->xenvchan->is_server && libxenvchan_is_open(ctrl->xenvchan) == 2) {
/* In case of vchan server waiting for a client, we'll not receive any
* notification if the remote domain dies before connecting. Because of
* that, check periodically if remote domain is still alive while
* waiting for a connection. Actually this doesn't cover all the cases
* - if remote domain is still alive, but remote process dies before
* connecting, we'll also not receive any notification. But this, in
* most cases, can be solved by application using libvchan.
*
* During normal operation this shouldn't be long - in most cases vchan
* client will connect almost instantly. So this sleep(10) loop will
* not hurt. Alternativelly it could be implemented with
* xs_watch("@releaseDomain"), but such approach will slow down most
* common execution path (xs_open+xs_watch even if client connects
* right away).
*/
while (ret == -2 && libxenvchan_is_open(ctrl->xenvchan) == 2) {
fd_set rd_set;
struct timeval tv = { 10, 0 };
int vchan_fd = libxenvchan_fd_for_select(ctrl->xenvchan);
FD_ZERO(&rd_set);
FD_SET(vchan_fd, &rd_set);
switch (select(vchan_fd+1, &rd_set, NULL, NULL, &tv)) {
case 0:
if (!libvchan__check_domain_alive(ctrl->xc_handle, ctrl->remote_domain))
return -1;
break;
case 1:
/* break the loop */
ret = -1;
break;
default:
if (errno == EINTR)
break;
perror("select");
return -1;
}
}
}
ret = libxenvchan_wait(ctrl->xenvchan);
if (ctrl->xs_path) {
/* remove xenstore entry at first client connection */
xs = xs_open(0);
if (xs) {
/* if xenstore connection failed just do not remove entries, but do
* not abort whole function, especially still free the memory
*/
xs_rm(xs, 0, ctrl->xs_path);
xs_close(xs);
}
free(ctrl->xs_path);
ctrl->xs_path = NULL;
}
return ret;
}
XenDomainWatcher::~XenDomainWatcher()
{
xs_unwatch( xsh_, "@introduceDomain", introduceToken_.c_str() );
xs_unwatch( xsh_, "@releaseDomain", releaseToken_.c_str() );
xs_close( xsh_ );
xc_interface_close( xci_ );
}
static void *worker (void *ctx_)
#endif
{
void *s;
int rc;
int i;
xs_msg_t msg;
s = xs_socket (ctx_, XS_REP);
if (!s) {
printf ("error in xs_socket: %s\n", xs_strerror (errno));
exit (1);
}
rc = xs_connect (s, "inproc://lat_test");
if (rc == -1) {
printf ("error in xs_connect: %s\n", xs_strerror (errno));
exit (1);
}
rc = xs_msg_init (&msg);
if (rc != 0) {
printf ("error in xs_msg_init: %s\n", xs_strerror (errno));
exit (1);
}
for (i = 0; i != roundtrip_count; i++) {
rc = xs_recvmsg (s, &msg, 0);
if (rc < 0) {
printf ("error in xs_recvmsg: %s\n", xs_strerror (errno));
exit (1);
}
rc = xs_sendmsg (s, &msg, 0);
if (rc < 0) {
printf ("error in xs_sendmsg: %s\n", xs_strerror (errno));
exit (1);
}
}
rc = xs_msg_close (&msg);
if (rc != 0) {
printf ("error in xs_msg_close: %s\n", xs_strerror (errno));
exit (1);
}
rc = xs_close (s);
if (rc != 0) {
printf ("error in xs_close: %s\n", xs_strerror (errno));
exit (1);
}
#if defined XS_HAVE_WINDOWS
return 0;
#else
return NULL;
#endif
}
/*
* Abort current xenstore transaction if any
* and shutdown connection to xenstore
*/
static void shutdown_xs(void)
{
abort_xs();
if (xs != NULL)
{
xs_close(xs);
xs = NULL;
}
}
static void *worker (void *ctx_)
#endif
{
void *s;
int rc;
int i;
xs_msg_t msg;
s = xs_socket (ctx_, XS_PUSH);
if (!s) {
printf ("error in xs_socket: %s\n", xs_strerror (errno));
exit (1);
}
rc = xs_connect (s, "inproc://thr_test");
if (rc == -1) {
printf ("error in xs_connect: %s\n", xs_strerror (errno));
exit (1);
}
for (i = 0; i != message_count; i++) {
rc = xs_msg_init_size (&msg, message_size);
if (rc != 0) {
printf ("error in xs_msg_init_size: %s\n", xs_strerror (errno));
exit (1);
}
#if defined XS_MAKE_VALGRIND_HAPPY
memset (xs_msg_data (&msg), 0, message_size);
#endif
rc = xs_sendmsg (s, &msg, 0);
if (rc < 0) {
printf ("error in xs_sendmsg: %s\n", xs_strerror (errno));
exit (1);
}
rc = xs_msg_close (&msg);
if (rc != 0) {
printf ("error in xs_msg_close: %s\n", xs_strerror (errno));
exit (1);
}
}
rc = xs_close (s);
if (rc != 0) {
printf ("error in xs_close: %s\n", xs_strerror (errno));
exit (1);
}
#if defined XS_HAVE_WINDOWS
return 0;
#else
return NULL;
#endif
}
/**
* xs_destroy - prepare to shutdown a transport
* @xprt: doomed transport
*
*/
static void xs_destroy(struct rpc_xprt *xprt)
{
dprintk("RPC: xs_destroy xprt %p\n", xprt);
cancel_delayed_work(&xprt->connect_worker);
flush_scheduled_work();
xprt_disconnect(xprt);
xs_close(xprt);
kfree(xprt->slot);
}
void closeIVCLibrary(libIVC_t *iface)
{
xc_evtchn_close(iface->ec);
xs_close(iface->xs);
xc_interface_close(iface->xc);
xc_gnttab_close(iface->gt);
#ifdef HAVE_XENSTORE_H
if(iface->gs) xc_gntshr_close(iface->gs);
#endif
free(iface);
}
static void shutdown_stress_worker (void *s_)
{
int rc;
rc = xs_connect (s_, "tcp://127.0.0.1:5560");
assert (rc == 0);
// Start closing the socket while the connecting process is underway.
rc = xs_close (s_);
assert (rc == 0);
}
/**
* xs_udp_connect_worker - set up a UDP socket
* @args: RPC transport to connect
*
* Invoked by a work queue tasklet.
*/
static void xs_udp_connect_worker(void *args)
{
struct rpc_xprt *xprt = (struct rpc_xprt *) args;
struct socket *sock = xprt->sock;
int err, status = -EIO;
if (xprt->shutdown || xprt->addr.sin_port == 0)
goto out;
dprintk("RPC: xs_udp_connect_worker for xprt %p\n", xprt);
/* Start by resetting any existing state */
xs_close(xprt);
if ((err = sock_create_kern(PF_INET, SOCK_DGRAM, IPPROTO_UDP, &sock)) < 0) {
dprintk("RPC: can't create UDP transport socket (%d).\n", -err);
goto out;
}
xs_reclassify_socket(sock);
if (xs_bind(xprt, sock)) {
sock_release(sock);
goto out;
}
if (!xprt->inet) {
struct sock *sk = sock->sk;
write_lock_bh(&sk->sk_callback_lock);
sk->sk_user_data = xprt;
xprt->old_data_ready = sk->sk_data_ready;
xprt->old_state_change = sk->sk_state_change;
xprt->old_write_space = sk->sk_write_space;
sk->sk_data_ready = xs_udp_data_ready;
sk->sk_write_space = xs_udp_write_space;
sk->sk_no_check = UDP_CSUM_NORCV;
sk->sk_allocation = GFP_ATOMIC;
xprt_set_connected(xprt);
/* Reset to new socket */
xprt->sock = sock;
xprt->inet = sk;
write_unlock_bh(&sk->sk_callback_lock);
}
xs_udp_do_set_buffer_size(xprt);
status = 0;
out:
xprt_wake_pending_tasks(xprt, status);
xprt_clear_connecting(xprt);
}
/**
* xs_destroy - prepare to shutdown a transport
* @xprt: doomed transport
*
*/
static void xs_destroy(struct rpc_xprt *xprt)
{
dprintk("RPC: xs_destroy xprt %p\n", xprt);
if (!cancel_delayed_work(&xprt->connect_worker))
wait_on_bit(&xprt->state, XPRT_CONNECTING,
xs_wait_bit_uninterruptible, TASK_UNINTERRUPTIBLE);
xprt_disconnect(xprt);
xs_close(xprt);
kfree(xprt->slot);
}
void polltimeo_worker(void *ctx_)
{
// Worker thread connects after delay of 1 second. Then it waits
// for 1 more second, so that async connect has time to succeed.
sleep (1);
void *sc = xs_socket (ctx_, XS_PUSH);
assert (sc);
int rc = xs_connect (sc, "inproc://timeout_test");
assert (rc == 0);
sleep (1);
rc = xs_close (sc);
assert (rc == 0);
}
int
xenstore_uninit(void)
{
xs_close(xs);
if (is_xenstore_cleaned_up == 0) {
if (xenstore_cleanup())
return -1;
is_xenstore_cleaned_up = 1;
}
free(dompath);
dompath = NULL;
return 0;
}
static int store_dev_info(int domid, CharDriverState *cs, const char *string)
{
struct xs_handle *xs = NULL;
char *path = NULL;
char *newpath = NULL;
char *pts = NULL;
int ret = -1;
/* Only continue if we're talking to a pty. */
if (strncmp(cs->filename, "pty:", 4)) {
return 0;
}
pts = cs->filename + 4;
/* We now have everything we need to set the xenstore entry. */
xs = xs_open(0);
if (xs == NULL) {
fprintf(stderr, "Could not contact XenStore\n");
goto out;
}
path = xs_get_domain_path(xs, domid);
if (path == NULL) {
fprintf(stderr, "xs_get_domain_path() error\n");
goto out;
}
newpath = realloc(path, (strlen(path) + strlen(string) +
strlen("/tty") + 1));
if (newpath == NULL) {
fprintf(stderr, "realloc error\n");
goto out;
}
path = newpath;
strcat(path, string);
strcat(path, "/tty");
if (!xs_write(xs, XBT_NULL, path, pts, strlen(pts))) {
fprintf(stderr, "xs_write for '%s' fail", string);
goto out;
}
ret = 0;
out:
free(path);
xs_close(xs);
return ret;
}
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;
}
static uint32_t
vbd_read_tapdisk_pid(uint32_t domid, uint32_t vbdid)
{
// XenStore tapdisk pid path
static const char* PATH_FMT = "/local/domain/0/backend/vbd3/%u/%u/kthread-pid";
// Local variables
char *path; // Path storage
unsigned int len; // Temp variable
void *value; // Value returned by xs_read
uint32_t retvalue = 0; // Value converted to integer
// Open Xenstore connection
struct xs_handle *handle = xs_open(XS_OPEN_READONLY);
if (!handle) {
perror("xs_open");
goto err;
}
// Format tapdisk pid path
if (asprintf(&path, PATH_FMT, domid, vbdid) < 0) {
perror("asprintf");
goto asperr;
}
// Read value
value = xs_read(handle, XBT_NULL, path, &len);
// We don't print error here, as we would always report invalid cdrom entry,
// for which there is no tapdisk pid
if (!value) {
goto readerr;
}
// Convert to int
retvalue = atoi((const char *)value);
free(value);
readerr:
free(path);
asperr:
xs_close(handle);
err:
return retvalue;
}
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 libvchan_close(libvchan_t *ctrl) {
struct xs_handle *xs;
libxenvchan_close(ctrl->xenvchan);
if (ctrl->xs_path) {
/* remove xenstore entry in case of no client connected */
xs = xs_open(0);
if (xs) {
/* if xenstore connection failed just do not remove entries, but do
* not abort whole function, especially still free the memory
*/
xs_rm(xs, 0, ctrl->xs_path);
xs_close(xs);
}
free(ctrl->xs_path);
}
free(ctrl);
}
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;
}
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 [])
{
#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;
}
int zmq_close (void *s)
{
return xs_close (s);
}
/**
* xs_tcp_connect_worker - connect a TCP socket to a remote endpoint
* @args: RPC transport to connect
*
* Invoked by a work queue tasklet.
*/
static void xs_tcp_connect_worker(void *args)
{
struct rpc_xprt *xprt = (struct rpc_xprt *)args;
struct socket *sock = xprt->sock;
int err, status = -EIO;
if (xprt->shutdown || xprt->addr.sin_port == 0)
goto out;
dprintk("RPC: xs_tcp_connect_worker for xprt %p\n", xprt);
if (!xprt->sock) {
/* start from scratch */
if ((err = sock_create_kern(PF_INET, SOCK_STREAM, IPPROTO_TCP, &sock)) < 0) {
dprintk("RPC: can't create TCP transport socket (%d).\n", -err);
goto out;
}
xs_reclassify_socket(sock);
if (xs_bind(xprt, sock)) {
sock_release(sock);
goto out;
}
} else
/* "close" the socket, preserving the local port */
xs_tcp_reuse_connection(xprt);
if (!xprt->inet) {
struct sock *sk = sock->sk;
write_lock_bh(&sk->sk_callback_lock);
sk->sk_user_data = xprt;
xprt->old_data_ready = sk->sk_data_ready;
xprt->old_state_change = sk->sk_state_change;
xprt->old_write_space = sk->sk_write_space;
sk->sk_data_ready = xs_tcp_data_ready;
sk->sk_state_change = xs_tcp_state_change;
sk->sk_write_space = xs_tcp_write_space;
sk->sk_allocation = GFP_ATOMIC;
/* socket options */
sk->sk_userlocks |= SOCK_BINDPORT_LOCK;
sock_reset_flag(sk, SOCK_LINGER);
tcp_sk(sk)->linger2 = 0;
tcp_sk(sk)->nonagle |= TCP_NAGLE_OFF;
xprt_clear_connected(xprt);
/* Reset to new socket */
xprt->sock = sock;
xprt->inet = sk;
write_unlock_bh(&sk->sk_callback_lock);
}
/* Tell the socket layer to start connecting... */
xprt->stat.connect_count++;
xprt->stat.connect_start = jiffies;
status = kernel_connect(sock, (struct sockaddr *) &xprt->addr,
sizeof(xprt->addr), O_NONBLOCK);
dprintk("RPC: %p connect status %d connected %d sock state %d\n",
xprt, -status, xprt_connected(xprt), sock->sk->sk_state);
if (status < 0) {
switch (status) {
case -EINPROGRESS:
case -EALREADY:
goto out_clear;
case -ECONNREFUSED:
case -ECONNRESET:
/* retry with existing socket, after a delay */
break;
case -ENETUNREACH:
status = -ENOTCONN;
break;
default:
/* get rid of existing socket, and retry */
xs_close(xprt);
break;
}
}
out:
xprt_wake_pending_tasks(xprt, status);
out_clear:
xprt_clear_connecting(xprt);
}
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)
{
int opt;
xc_interface *xch;
struct xs_handle *xsh;
char buf[16];
int rv, fd;
while ( (opt = getopt_long(argc, argv, "", options, NULL)) != -1 )
{
switch ( opt )
{
case 'k':
kernel = optarg;
break;
case 'm':
memory = strtol(optarg, NULL, 10);
break;
case 'f':
flask = optarg;
break;
case 'r':
ramdisk = optarg;
break;
case 'p':
param = optarg;
break;
case 'n':
name = optarg;
break;
default:
usage();
return 2;
}
}
if ( optind != argc || !kernel || !memory )
{
usage();
return 2;
}
xch = xc_interface_open(NULL, NULL, 0);
if ( !xch )
{
fprintf(stderr, "xc_interface_open() failed\n");
return 1;
}
rv = check_domain(xch);
if ( !rv )
rv = build(xch);
else if ( rv > 0 )
fprintf(stderr, "xenstore domain already present.\n");
xc_interface_close(xch);
if ( rv )
return 1;
rv = gen_stub_json_config(domid);
if ( rv )
return 3;
xsh = xs_open(0);
if ( !xsh )
{
fprintf(stderr, "xs_open() failed.\n");
return 3;
}
snprintf(buf, 16, "%d", domid);
do_xs_write(xsh, "/tool/xenstored/domid", buf);
do_xs_write_dom(xsh, "domid", buf);
do_xs_write_dom(xsh, "name", name);
snprintf(buf, 16, "%d", memory * 1024);
do_xs_write_dom(xsh, "memory/target", buf);
do_xs_write_dom(xsh, "memory/static-max", buf);
xs_close(xsh);
fd = creat("/var/run/xenstored.pid", 0666);
if ( fd < 0 )
{
fprintf(stderr, "Creating /var/run/xenstored.pid failed\n");
return 3;
}
rv = snprintf(buf, 16, "domid:%d\n", domid);
rv = write(fd, buf, rv);
close(fd);
if ( rv < 0 )
{
fprintf(stderr, "Writing domid to /var/run/xenstored.pid failed\n");
return 3;
}
return 0;
}
