int zre_msg_send_whisper ( void *output, uint16_t sequence, zframe_t *content) { zre_msg_t *self = zre_msg_new (ZRE_MSG_WHISPER); zre_msg_sequence_set (self, sequence); zre_msg_content_set (self, zframe_dup (content)); return zre_msg_send (&self, output); }
int zre_msg_send_shout ( void *output, uint16_t sequence, char *group, zframe_t *content) { zre_msg_t *self = zre_msg_new (ZRE_MSG_SHOUT); zre_msg_sequence_set (self, sequence); zre_msg_group_set (self, group); zre_msg_content_set (self, zframe_dup (content)); return zre_msg_send (&self, output); }
static int agent_recv_from_api (agent_t *self) { // Get the whole message off the pipe in one go zmsg_t *request = zmsg_recv (self->pipe); char *command = zmsg_popstr (request); if (!command) return -1; // Interrupted if (streq (command, "WHISPER")) { // Get peer to send message to char *identity = zmsg_popstr (request); zre_peer_t *peer = (zre_peer_t *) zhash_lookup (self->peers, identity); // Send frame on out to peer's mailbox, drop message // if peer doesn't exist (may have been destroyed) if (peer) { zre_msg_t *msg = zre_msg_new (ZRE_MSG_WHISPER); zre_msg_content_set (msg, zmsg_pop (request)); zre_peer_send (peer, &msg); } free (identity); } else if (streq (command, "SHOUT")) { // Get group to send message to char *name = zmsg_popstr (request); zre_group_t *group = (zre_group_t *) zhash_lookup (self->peer_groups, name); if (group) { zre_msg_t *msg = zre_msg_new (ZRE_MSG_SHOUT); zre_msg_group_set (msg, name); zre_msg_content_set (msg, zmsg_pop (request)); zre_group_send (group, &msg); } free (name); } else if (streq (command, "JOIN")) { char *name = zmsg_popstr (request); zre_group_t *group = (zre_group_t *) zhash_lookup (self->own_groups, name); if (!group) { // Only send if we're not already in group group = zre_group_new (name, self->own_groups); zre_msg_t *msg = zre_msg_new (ZRE_MSG_JOIN); zre_msg_group_set (msg, name); // Update status before sending command zre_msg_status_set (msg, ++(self->status)); zhash_foreach (self->peers, s_peer_send, msg); zre_msg_destroy (&msg); zre_log_info (self->log, ZRE_LOG_MSG_EVENT_JOIN, NULL, name); } free (name); } else if (streq (command, "LEAVE")) { char *name = zmsg_popstr (request); zre_group_t *group = (zre_group_t *) zhash_lookup (self->own_groups, name); if (group) { // Only send if we are actually in group zre_msg_t *msg = zre_msg_new (ZRE_MSG_LEAVE); zre_msg_group_set (msg, name); // Update status before sending command zre_msg_status_set (msg, ++(self->status)); zhash_foreach (self->peers, s_peer_send, msg); zre_msg_destroy (&msg); zhash_delete (self->own_groups, name); zre_log_info (self->log, ZRE_LOG_MSG_EVENT_LEAVE, NULL, name); } free (name); } else if (streq (command, "SET")) { char *name = zmsg_popstr (request); char *value = zmsg_popstr (request); zhash_update (self->headers, name, value); free (name); free (value); } else if (streq (command, "PUBLISH")) { char *logical = zmsg_popstr (request); char *physical = zmsg_popstr (request); // Virtual filename must start with slash assert (logical [0] == '/'); // We create symbolic link pointing to real file char *symlink = malloc (strlen (logical) + 3); sprintf (symlink, "%s.ln", logical + 1); fmq_file_t *file = fmq_file_new (self->fmq_outbox, symlink); int rc = fmq_file_output (file); assert (rc == 0); fprintf (fmq_file_handle (file), "%s\n", physical); fmq_file_destroy (&file); free (symlink); free (logical); free (physical); } else if (streq (command, "RETRACT")) { char *logical = zmsg_popstr (request); // Logical filename must start with slash assert (logical [0] == '/'); // We create symbolic link pointing to real file char *symlink = malloc (strlen (logical) + 3); sprintf (symlink, "%s.ln", logical + 1); fmq_file_t *file = fmq_file_new (self->fmq_outbox, symlink); fmq_file_remove (file); free (symlink); free (logical); } free (command); zmsg_destroy (&request); return 0; }
int zre_msg_test (bool verbose) { printf (" * zre_msg: "); // Simple create/destroy test zre_msg_t *self = zre_msg_new (0); assert (self); zre_msg_destroy (&self); // Create pair of sockets we can send through zctx_t *ctx = zctx_new (); assert (ctx); void *output = zsocket_new (ctx, ZMQ_DEALER); assert (output); zsocket_bind (output, "inproc://selftest"); void *input = zsocket_new (ctx, ZMQ_ROUTER); assert (input); zsocket_connect (input, "inproc://selftest"); // Encode/send/decode and verify each message type self = zre_msg_new (ZRE_MSG_HELLO); zre_msg_sequence_set (self, 123); zre_msg_ipaddress_set (self, "Life is short but Now lasts for ever"); zre_msg_mailbox_set (self, 123); zre_msg_groups_append (self, "Name: %s", "Brutus"); zre_msg_groups_append (self, "Age: %d", 43); zre_msg_status_set (self, 123); zre_msg_headers_insert (self, "Name", "Brutus"); zre_msg_headers_insert (self, "Age", "%d", 43); zre_msg_send (&self, output); self = zre_msg_recv (input); assert (self); assert (zre_msg_sequence (self) == 123); assert (streq (zre_msg_ipaddress (self), "Life is short but Now lasts for ever")); assert (zre_msg_mailbox (self) == 123); assert (zre_msg_groups_size (self) == 2); assert (streq (zre_msg_groups_first (self), "Name: Brutus")); assert (streq (zre_msg_groups_next (self), "Age: 43")); assert (zre_msg_status (self) == 123); assert (zre_msg_headers_size (self) == 2); assert (streq (zre_msg_headers_string (self, "Name", "?"), "Brutus")); assert (zre_msg_headers_number (self, "Age", 0) == 43); zre_msg_destroy (&self); self = zre_msg_new (ZRE_MSG_WHISPER); zre_msg_sequence_set (self, 123); zre_msg_content_set (self, zframe_new ("Captcha Diem", 12)); zre_msg_send (&self, output); self = zre_msg_recv (input); assert (self); assert (zre_msg_sequence (self) == 123); assert (zframe_streq (zre_msg_content (self), "Captcha Diem")); zre_msg_destroy (&self); self = zre_msg_new (ZRE_MSG_SHOUT); zre_msg_sequence_set (self, 123); zre_msg_group_set (self, "Life is short but Now lasts for ever"); zre_msg_content_set (self, zframe_new ("Captcha Diem", 12)); zre_msg_send (&self, output); self = zre_msg_recv (input); assert (self); assert (zre_msg_sequence (self) == 123); assert (streq (zre_msg_group (self), "Life is short but Now lasts for ever")); assert (zframe_streq (zre_msg_content (self), "Captcha Diem")); zre_msg_destroy (&self); self = zre_msg_new (ZRE_MSG_JOIN); zre_msg_sequence_set (self, 123); zre_msg_group_set (self, "Life is short but Now lasts for ever"); zre_msg_status_set (self, 123); zre_msg_send (&self, output); self = zre_msg_recv (input); assert (self); assert (zre_msg_sequence (self) == 123); assert (streq (zre_msg_group (self), "Life is short but Now lasts for ever")); assert (zre_msg_status (self) == 123); zre_msg_destroy (&self); self = zre_msg_new (ZRE_MSG_LEAVE); zre_msg_sequence_set (self, 123); zre_msg_group_set (self, "Life is short but Now lasts for ever"); zre_msg_status_set (self, 123); zre_msg_send (&self, output); self = zre_msg_recv (input); assert (self); assert (zre_msg_sequence (self) == 123); assert (streq (zre_msg_group (self), "Life is short but Now lasts for ever")); assert (zre_msg_status (self) == 123); zre_msg_destroy (&self); self = zre_msg_new (ZRE_MSG_PING); zre_msg_sequence_set (self, 123); zre_msg_send (&self, output); self = zre_msg_recv (input); assert (self); assert (zre_msg_sequence (self) == 123); zre_msg_destroy (&self); self = zre_msg_new (ZRE_MSG_PING_OK); zre_msg_sequence_set (self, 123); zre_msg_send (&self, output); self = zre_msg_recv (input); assert (self); assert (zre_msg_sequence (self) == 123); zre_msg_destroy (&self); zctx_destroy (&ctx); printf ("OK\n"); return 0; }