diff options
Diffstat (limited to 'tests/test_security_plain.cpp')
-rw-r--r-- | tests/test_security_plain.cpp | 150 |
1 files changed, 0 insertions, 150 deletions
diff --git a/tests/test_security_plain.cpp b/tests/test_security_plain.cpp deleted file mode 100644 index 74973fd..0000000 --- a/tests/test_security_plain.cpp +++ /dev/null @@ -1,150 +0,0 @@ -/* - Copyright (c) 2007-2013 Contributors as noted in the AUTHORS file - - This file is part of 0MQ. - - 0MQ is free software; you can redistribute it and/or modify it under - the terms of the GNU Lesser General Public License as published by - the Free Software Foundation; either version 3 of the License, or - (at your option) any later version. - - 0MQ is distributed in the hope that it will be useful, - but WITHOUT ANY WARRANTY; without even the implied warranty of - MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - GNU Lesser General Public License for more details. - - You should have received a copy of the GNU Lesser General Public License - along with this program. If not, see <http://www.gnu.org/licenses/>. -*/ - -#include "testutil.hpp" - -static void -zap_handler (void *ctx) -{ - // Create and bind ZAP socket - void *zap = zmq_socket (ctx, ZMQ_REP); - assert (zap); - int rc = zmq_bind (zap, "inproc://zeromq.zap.01"); - assert (rc == 0); - - // Process ZAP requests forever - while (true) { - char *version = s_recv (zap); - if (!version) - break; // Terminating - char *sequence = s_recv (zap); - char *domain = s_recv (zap); - char *address = s_recv (zap); - char *identity = s_recv (zap); - char *mechanism = s_recv (zap); - char *username = s_recv (zap); - char *password = s_recv (zap); - - assert (streq (version, "1.0")); - assert (streq (mechanism, "PLAIN")); - assert (streq (identity, "IDENT")); - - s_sendmore (zap, version); - s_sendmore (zap, sequence); - if (streq (username, "admin") - && streq (password, "password")) { - s_sendmore (zap, "200"); - s_sendmore (zap, "OK"); - s_sendmore (zap, "anonymous"); - s_send (zap, ""); - } - else { - s_sendmore (zap, "400"); - s_sendmore (zap, "Invalid username or password"); - s_sendmore (zap, ""); - s_send (zap, ""); - } - free (version); - free (sequence); - free (domain); - free (address); - free (identity); - free (mechanism); - free (username); - free (password); - } - rc = zmq_close (zap); - assert (rc == 0); -} - -int main (void) -{ - setup_test_environment(); - void *ctx = zmq_ctx_new (); - assert (ctx); - - // Spawn ZAP handler - void *zap_thread = zmq_threadstart (&zap_handler, ctx); - - // Server socket will accept connections - void *server = zmq_socket (ctx, ZMQ_DEALER); - assert (server); - int rc = zmq_setsockopt (server, ZMQ_IDENTITY, "IDENT", 6); - assert (rc == 0); - int as_server = 1; - rc = zmq_setsockopt (server, ZMQ_PLAIN_SERVER, &as_server, sizeof (int)); - assert (rc == 0); - rc = zmq_bind (server, "tcp://127.0.0.1:9998"); - assert (rc == 0); - - char username [256]; - char password [256]; - - // Check PLAIN security with correct username/password - void *client = zmq_socket (ctx, ZMQ_DEALER); - assert (client); - strcpy (username, "admin"); - rc = zmq_setsockopt (client, ZMQ_PLAIN_USERNAME, username, strlen (username)); - assert (rc == 0); - strcpy (password, "password"); - rc = zmq_setsockopt (client, ZMQ_PLAIN_PASSWORD, password, strlen (password)); - assert (rc == 0); - rc = zmq_connect (client, "tcp://localhost:9998"); - assert (rc == 0); - bounce (server, client); - rc = zmq_close (client); - assert (rc == 0); - - // Check PLAIN security with badly configured client (as_server) - // This will be caught by the plain_server class, not passed to ZAP - client = zmq_socket (ctx, ZMQ_DEALER); - assert (client); - as_server = 1; - rc = zmq_setsockopt (client, ZMQ_PLAIN_SERVER, &as_server, sizeof (int)); - assert (rc == 0); - rc = zmq_connect (client, "tcp://localhost:9998"); - assert (rc == 0); - expect_bounce_fail (server, client); - close_zero_linger (client); - - // Check PLAIN security -- failed authentication - client = zmq_socket (ctx, ZMQ_DEALER); - assert (client); - strcpy (username, "wronguser"); - strcpy (password, "wrongpass"); - rc = zmq_setsockopt (client, ZMQ_PLAIN_USERNAME, username, strlen (username)); - assert (rc == 0); - rc = zmq_setsockopt (client, ZMQ_PLAIN_PASSWORD, password, strlen (password)); - assert (rc == 0); - rc = zmq_connect (client, "tcp://localhost:9998"); - assert (rc == 0); - expect_bounce_fail (server, client); - close_zero_linger (client); - - // Shutdown - rc = zmq_close (server); - assert (rc == 0); - rc = zmq_ctx_term (ctx); - assert (rc == 0); - - // Wait until ZAP handler terminates - zmq_threadclose (zap_thread); - - return 0; -} |