static void
server_destroy (server_t **self_p)
{
assert (self_p);
if (*self_p) {
server_t *self = *self_p;
zsocket_destroy (self->ctx, self->dealer);
fmq_msg_destroy (&self->request);
fmq_msg_destroy (&self->reply);
free (self->endpoint);
free (self);
*self_p = NULL;
}
}
static void
server_message (client_t *self)
{
if (self->reply)
fmq_msg_destroy (&self->reply);
self->reply = fmq_msg_recv (self->dealer);
if (!self->reply)
return; // Interrupted; do nothing
if (fmq_msg_id (self->reply) == FMQ_MSG_SRSLY)
client_execute (self, srsly_event);
else
if (fmq_msg_id (self->reply) == FMQ_MSG_RTFM)
client_execute (self, rtfm_event);
else
if (fmq_msg_id (self->reply) == FMQ_MSG_ORLY)
client_execute (self, orly_event);
else
if (fmq_msg_id (self->reply) == FMQ_MSG_OHAI_OK)
client_execute (self, ohai_ok_event);
else
if (fmq_msg_id (self->reply) == FMQ_MSG_CHEEZBURGER)
client_execute (self, cheezburger_event);
else
if (fmq_msg_id (self->reply) == FMQ_MSG_HUGZ)
client_execute (self, hugz_event);
else
if (fmq_msg_id (self->reply) == FMQ_MSG_ICANHAZ_OK)
client_execute (self, icanhaz_ok_event);
// Any input from server counts as activity
self->expires_at = zclock_time () + self->heartbeat * 2;
}
static void
client_destroy (client_t **self_p)
{
assert (self_p);
if (*self_p) {
client_t *self = *self_p;
zframe_destroy (&self->address);
fmq_msg_destroy (&self->request);
fmq_msg_destroy (&self->reply);
free (self->hashkey);
while (zlist_size (self->patches)) {
fmq_patch_t *patch = (fmq_patch_t *) zlist_pop (self->patches);
fmq_patch_destroy (&patch);
}
zlist_destroy (&self->patches);
free (self);
*self_p = NULL;
}
}
static void
client_destroy (client_t **self_p)
{
assert (self_p);
if (*self_p) {
client_t *self = *self_p;
fmq_config_destroy (&self->config);
fmq_msg_destroy (&self->request);
fmq_msg_destroy (&self->reply);
// Destroy subscriptions
while (zlist_size (self->subs)) {
sub_t *sub = (sub_t *) zlist_pop (self->subs);
sub_destroy (&sub);
}
zlist_destroy (&self->subs);
free (self);
*self_p = NULL;
}
}
static void
server_client_message (server_t *self)
{
fmq_msg_t *request = fmq_msg_recv (self->router);
if (!request)
return; // Interrupted; do nothing
char *hashkey = zframe_strhex (fmq_msg_address (request));
client_t *client = zhash_lookup (self->clients, hashkey);
if (client == NULL) {
client = client_new (fmq_msg_address (request));
client->heartbeat = self->heartbeat;
client->router = self->router;
zhash_insert (self->clients, hashkey, client);
zhash_freefn (self->clients, hashkey, client_free);
}
free (hashkey);
if (client->request)
fmq_msg_destroy (&client->request);
client->request = request;
// Any input from client counts as heartbeat
client->heartbeat_at = zclock_time () + client->heartbeat;
// Any input from client counts as activity
client->expires_at = zclock_time () + client->heartbeat * 3;
if (fmq_msg_id (request) == FMQ_MSG_OHAI)
server_client_execute (self, client, ohai_event);
else
if (fmq_msg_id (request) == FMQ_MSG_YARLY)
server_client_execute (self, client, yarly_event);
else
if (fmq_msg_id (request) == FMQ_MSG_ICANHAZ)
server_client_execute (self, client, icanhaz_event);
else
if (fmq_msg_id (request) == FMQ_MSG_NOM)
server_client_execute (self, client, nom_event);
else
if (fmq_msg_id (request) == FMQ_MSG_HUGZ)
server_client_execute (self, client, hugz_event);
else
if (fmq_msg_id (request) == FMQ_MSG_KTHXBAI)
server_client_execute (self, client, kthxbai_event);
}
static void
client_restart (client_t *self, char *endpoint)
{
// Reconnect to new endpoint if specified
if (endpoint) {
if (self->dealer)
zsocket_destroy (self->ctx, self->dealer);
self->dealer = zsocket_new (self->ctx, ZMQ_DEALER);
zmq_connect (self->dealer, endpoint);
}
// Clear out any previous request data
fmq_msg_destroy (&self->request);
self->request = fmq_msg_new (0);
// Restart dialog state machine from zero
self->state = start_state;
self->expires_at = 0;
// Application hook to reinitialize dialog
// Provides us with an event to kick things off
initialize_the_client (self);
client_execute (self, self->next_event);
}
int
fmq_server_test (bool verbose)
{
printf (" * fmq_server: ");
fflush (stdout);
zctx_t *ctx = zctx_new ();
fmq_server_t *self;
void *dealer = zsocket_new (ctx, ZMQ_DEALER);
zsocket_set_rcvtimeo (dealer, 2000);
zsocket_connect (dealer, "tcp://localhost:5670");
fmq_msg_t *request, *reply;
// Run selftest using '' configuration
self = fmq_server_new ();
assert (self);
int port = fmq_server_bind (self, "tcp://*:5670");
assert (port == 5670);
request = fmq_msg_new (FMQ_MSG_OHAI);
fmq_msg_send (&request, dealer);
reply = fmq_msg_recv (dealer);
assert (reply);
assert (fmq_msg_id (reply) == FMQ_MSG_SRSLY);
fmq_msg_destroy (&reply);
request = fmq_msg_new (FMQ_MSG_ICANHAZ);
fmq_msg_send (&request, dealer);
reply = fmq_msg_recv (dealer);
assert (reply);
assert (fmq_msg_id (reply) == FMQ_MSG_RTFM);
fmq_msg_destroy (&reply);
request = fmq_msg_new (FMQ_MSG_NOM);
fmq_msg_send (&request, dealer);
reply = fmq_msg_recv (dealer);
assert (reply);
assert (fmq_msg_id (reply) == FMQ_MSG_RTFM);
fmq_msg_destroy (&reply);
request = fmq_msg_new (FMQ_MSG_HUGZ);
fmq_msg_send (&request, dealer);
reply = fmq_msg_recv (dealer);
assert (reply);
assert (fmq_msg_id (reply) == FMQ_MSG_RTFM);
fmq_msg_destroy (&reply);
fmq_server_destroy (&self);
// Run selftest using 'anonymous.cfg' configuration
self = fmq_server_new ();
assert (self);
fmq_server_configure (self, "anonymous.cfg");
port = fmq_server_bind (self, "tcp://*:5670");
assert (port == 5670);
request = fmq_msg_new (FMQ_MSG_OHAI);
fmq_msg_send (&request, dealer);
reply = fmq_msg_recv (dealer);
assert (reply);
assert (fmq_msg_id (reply) == FMQ_MSG_OHAI_OK);
fmq_msg_destroy (&reply);
request = fmq_msg_new (FMQ_MSG_NOM);
fmq_msg_send (&request, dealer);
request = fmq_msg_new (FMQ_MSG_HUGZ);
fmq_msg_send (&request, dealer);
reply = fmq_msg_recv (dealer);
assert (reply);
assert (fmq_msg_id (reply) == FMQ_MSG_HUGZ_OK);
fmq_msg_destroy (&reply);
request = fmq_msg_new (FMQ_MSG_YARLY);
fmq_msg_send (&request, dealer);
reply = fmq_msg_recv (dealer);
assert (reply);
assert (fmq_msg_id (reply) == FMQ_MSG_RTFM);
fmq_msg_destroy (&reply);
fmq_server_destroy (&self);
// Run selftest using 'server_test.cfg' configuration
self = fmq_server_new ();
assert (self);
fmq_server_configure (self, "server_test.cfg");
port = fmq_server_bind (self, "tcp://*:5670");
assert (port == 5670);
request = fmq_msg_new (FMQ_MSG_OHAI);
fmq_msg_send (&request, dealer);
reply = fmq_msg_recv (dealer);
assert (reply);
assert (fmq_msg_id (reply) == FMQ_MSG_ORLY);
fmq_msg_destroy (&reply);
request = fmq_msg_new (FMQ_MSG_YARLY);
fmq_msg_set_mechanism (request, "PLAIN");
fmq_msg_set_response (request, fmq_sasl_plain_encode ("guest", "guest"));
fmq_msg_send (&request, dealer);
reply = fmq_msg_recv (dealer);
assert (reply);
assert (fmq_msg_id (reply) == FMQ_MSG_OHAI_OK);
fmq_msg_destroy (&reply);
request = fmq_msg_new (FMQ_MSG_NOM);
fmq_msg_send (&request, dealer);
request = fmq_msg_new (FMQ_MSG_HUGZ);
fmq_msg_send (&request, dealer);
reply = fmq_msg_recv (dealer);
assert (reply);
assert (fmq_msg_id (reply) == FMQ_MSG_HUGZ_OK);
fmq_msg_destroy (&reply);
reply = fmq_msg_recv (dealer);
assert (reply);
assert (fmq_msg_id (reply) == FMQ_MSG_HUGZ);
fmq_msg_destroy (&reply);
fmq_server_destroy (&self);
zctx_destroy (&ctx);
// @end
// No clean way to wait for a background thread to exit
// Under valgrind this will randomly show as leakage
// Reduce this by giving server thread time to exit
zclock_sleep (200);
printf ("OK\n");
return 0;
}
void
fmq_msg_test (bool verbose)
{
printf (" * fmq_msg:");
if (verbose)
printf ("\n");
// @selftest
// Simple create/destroy test
fmq_msg_t *self = fmq_msg_new ();
assert (self);
fmq_msg_destroy (&self);
// Create pair of sockets we can send through
// We must bind before connect if we wish to remain compatible with ZeroMQ < v4
zsock_t *output = zsock_new (ZMQ_DEALER);
assert (output);
int rc = zsock_bind (output, "inproc://selftest-fmq_msg");
assert (rc == 0);
zsock_t *input = zsock_new (ZMQ_ROUTER);
assert (input);
rc = zsock_connect (input, "inproc://selftest-fmq_msg");
assert (rc == 0);
// Encode/send/decode and verify each message type
int instance;
self = fmq_msg_new ();
fmq_msg_set_id (self, FMQ_MSG_OHAI);
// Send twice
fmq_msg_send (self, output);
fmq_msg_send (self, output);
for (instance = 0; instance < 2; instance++) {
fmq_msg_recv (self, input);
assert (fmq_msg_routing_id (self));
}
fmq_msg_set_id (self, FMQ_MSG_OHAI_OK);
// Send twice
fmq_msg_send (self, output);
fmq_msg_send (self, output);
for (instance = 0; instance < 2; instance++) {
fmq_msg_recv (self, input);
assert (fmq_msg_routing_id (self));
}
fmq_msg_set_id (self, FMQ_MSG_ICANHAZ);
fmq_msg_set_path (self, "Life is short but Now lasts for ever");
zhash_t *icanhaz_options = zhash_new ();
zhash_insert (icanhaz_options, "Name", "Brutus");
fmq_msg_set_options (self, &icanhaz_options);
zhash_t *icanhaz_cache = zhash_new ();
zhash_insert (icanhaz_cache, "Name", "Brutus");
fmq_msg_set_cache (self, &icanhaz_cache);
// Send twice
fmq_msg_send (self, output);
fmq_msg_send (self, output);
for (instance = 0; instance < 2; instance++) {
fmq_msg_recv (self, input);
assert (fmq_msg_routing_id (self));
assert (streq (fmq_msg_path (self), "Life is short but Now lasts for ever"));
zhash_t *options = fmq_msg_get_options (self);
assert (zhash_size (options) == 2);
assert (streq ((char *) zhash_first (options), "Brutus"));
assert (streq ((char *) zhash_cursor (options), "Name"));
zhash_destroy (&options);
if (instance == 1)
zhash_destroy (&icanhaz_options);
zhash_t *cache = fmq_msg_get_cache (self);
assert (zhash_size (cache) == 2);
assert (streq ((char *) zhash_first (cache), "Brutus"));
assert (streq ((char *) zhash_cursor (cache), "Name"));
zhash_destroy (&cache);
if (instance == 1)
zhash_destroy (&icanhaz_cache);
}
fmq_msg_set_id (self, FMQ_MSG_ICANHAZ_OK);
// Send twice
fmq_msg_send (self, output);
fmq_msg_send (self, output);
for (instance = 0; instance < 2; instance++) {
fmq_msg_recv (self, input);
assert (fmq_msg_routing_id (self));
}
fmq_msg_set_id (self, FMQ_MSG_NOM);
fmq_msg_set_credit (self, 123);
fmq_msg_set_sequence (self, 123);
// Send twice
fmq_msg_send (self, output);
fmq_msg_send (self, output);
for (instance = 0; instance < 2; instance++) {
fmq_msg_recv (self, input);
assert (fmq_msg_routing_id (self));
assert (fmq_msg_credit (self) == 123);
assert (fmq_msg_sequence (self) == 123);
}
fmq_msg_set_id (self, FMQ_MSG_CHEEZBURGER);
fmq_msg_set_sequence (self, 123);
fmq_msg_set_operation (self, 123);
fmq_msg_set_filename (self, "Life is short but Now lasts for ever");
fmq_msg_set_offset (self, 123);
fmq_msg_set_eof (self, 123);
zhash_t *cheezburger_headers = zhash_new ();
zhash_insert (cheezburger_headers, "Name", "Brutus");
fmq_msg_set_headers (self, &cheezburger_headers);
zchunk_t *cheezburger_chunk = zchunk_new ("Captcha Diem", 12);
fmq_msg_set_chunk (self, &cheezburger_chunk);
// Send twice
fmq_msg_send (self, output);
fmq_msg_send (self, output);
for (instance = 0; instance < 2; instance++) {
fmq_msg_recv (self, input);
assert (fmq_msg_routing_id (self));
assert (fmq_msg_sequence (self) == 123);
assert (fmq_msg_operation (self) == 123);
assert (streq (fmq_msg_filename (self), "Life is short but Now lasts for ever"));
assert (fmq_msg_offset (self) == 123);
assert (fmq_msg_eof (self) == 123);
zhash_t *headers = fmq_msg_get_headers (self);
assert (zhash_size (headers) == 2);
assert (streq ((char *) zhash_first (headers), "Brutus"));
assert (streq ((char *) zhash_cursor (headers), "Name"));
zhash_destroy (&headers);
if (instance == 1)
zhash_destroy (&cheezburger_headers);
assert (memcmp (zchunk_data (fmq_msg_chunk (self)), "Captcha Diem", 12) == 0);
if (instance == 1)
zchunk_destroy (&cheezburger_chunk);
}
fmq_msg_set_id (self, FMQ_MSG_HUGZ);
// Send twice
fmq_msg_send (self, output);
fmq_msg_send (self, output);
for (instance = 0; instance < 2; instance++) {
fmq_msg_recv (self, input);
assert (fmq_msg_routing_id (self));
}
fmq_msg_set_id (self, FMQ_MSG_HUGZ_OK);
// Send twice
fmq_msg_send (self, output);
fmq_msg_send (self, output);
for (instance = 0; instance < 2; instance++) {
fmq_msg_recv (self, input);
assert (fmq_msg_routing_id (self));
}
fmq_msg_set_id (self, FMQ_MSG_KTHXBAI);
// Send twice
fmq_msg_send (self, output);
fmq_msg_send (self, output);
for (instance = 0; instance < 2; instance++) {
fmq_msg_recv (self, input);
assert (fmq_msg_routing_id (self));
}
fmq_msg_set_id (self, FMQ_MSG_SRSLY);
fmq_msg_set_reason (self, "Life is short but Now lasts for ever");
// Send twice
fmq_msg_send (self, output);
fmq_msg_send (self, output);
for (instance = 0; instance < 2; instance++) {
fmq_msg_recv (self, input);
assert (fmq_msg_routing_id (self));
assert (streq (fmq_msg_reason (self), "Life is short but Now lasts for ever"));
}
fmq_msg_set_id (self, FMQ_MSG_RTFM);
fmq_msg_set_reason (self, "Life is short but Now lasts for ever");
// Send twice
fmq_msg_send (self, output);
fmq_msg_send (self, output);
for (instance = 0; instance < 2; instance++) {
fmq_msg_recv (self, input);
assert (fmq_msg_routing_id (self));
assert (streq (fmq_msg_reason (self), "Life is short but Now lasts for ever"));
}
fmq_msg_destroy (&self);
zsock_destroy (&input);
zsock_destroy (&output);
// @end
printf ("OK\n");
}
