static void try_security_mechanism (client_t *self) { char *login = fmq_config_resolve (self->config, "security/plain/login", "guest"); char *password = fmq_config_resolve (self->config, "security/plain/password", ""); zframe_t *frame = fmq_sasl_plain_encode (login, password); fmq_msg_mechanism_set (self->request, "PLAIN"); fmq_msg_response_set (self->request, frame); }
// -------------------------------------------------------------------------- // Self test of this class int fmq_sasl_test (bool verbose) { printf (" * fmq_sasl: "); // @selftest zframe_t *frame = fmq_sasl_plain_encode ("Happy", "Harry"); char *login, *password; int rc = fmq_sasl_plain_decode (frame, &login, &password); assert (rc == 0); assert (streq (login, "Happy")); assert (streq (password, "Harry")); zframe_destroy (&frame); free (login); free (password); // @end printf ("OK\n"); return 0; }
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; }