Commit 4ea7d018 authored by evoskuil's avatar evoskuil

Problem: send_failure() util name undergeneralized (for read success).

parent 8c165ad1
...@@ -580,7 +580,7 @@ int zmq::curve_server_t::send_zap_request (const uint8_t *key) ...@@ -580,7 +580,7 @@ int zmq::curve_server_t::send_zap_request (const uint8_t *key)
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Version frame // Version frame
rc = msg.init_size (3); rc = msg.init_size (3);
...@@ -589,7 +589,7 @@ int zmq::curve_server_t::send_zap_request (const uint8_t *key) ...@@ -589,7 +589,7 @@ int zmq::curve_server_t::send_zap_request (const uint8_t *key)
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Request ID frame // Request ID frame
rc = msg.init_size (1); rc = msg.init_size (1);
...@@ -598,7 +598,7 @@ int zmq::curve_server_t::send_zap_request (const uint8_t *key) ...@@ -598,7 +598,7 @@ int zmq::curve_server_t::send_zap_request (const uint8_t *key)
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Domain frame // Domain frame
rc = msg.init_size (options.zap_domain.length ()); rc = msg.init_size (options.zap_domain.length ());
...@@ -607,7 +607,7 @@ int zmq::curve_server_t::send_zap_request (const uint8_t *key) ...@@ -607,7 +607,7 @@ int zmq::curve_server_t::send_zap_request (const uint8_t *key)
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Address frame // Address frame
rc = msg.init_size (peer_address.length ()); rc = msg.init_size (peer_address.length ());
...@@ -616,7 +616,7 @@ int zmq::curve_server_t::send_zap_request (const uint8_t *key) ...@@ -616,7 +616,7 @@ int zmq::curve_server_t::send_zap_request (const uint8_t *key)
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Identity frame // Identity frame
rc = msg.init_size (options.identity_size); rc = msg.init_size (options.identity_size);
...@@ -625,7 +625,7 @@ int zmq::curve_server_t::send_zap_request (const uint8_t *key) ...@@ -625,7 +625,7 @@ int zmq::curve_server_t::send_zap_request (const uint8_t *key)
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Mechanism frame // Mechanism frame
rc = msg.init_size (5); rc = msg.init_size (5);
...@@ -634,7 +634,7 @@ int zmq::curve_server_t::send_zap_request (const uint8_t *key) ...@@ -634,7 +634,7 @@ int zmq::curve_server_t::send_zap_request (const uint8_t *key)
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Credentials frame // Credentials frame
rc = msg.init_size (crypto_box_PUBLICKEYBYTES); rc = msg.init_size (crypto_box_PUBLICKEYBYTES);
...@@ -642,7 +642,7 @@ int zmq::curve_server_t::send_zap_request (const uint8_t *key) ...@@ -642,7 +642,7 @@ int zmq::curve_server_t::send_zap_request (const uint8_t *key)
memcpy (msg.data (), key, crypto_box_PUBLICKEYBYTES); memcpy (msg.data (), key, crypto_box_PUBLICKEYBYTES);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
return 0; return 0;
} }
...@@ -661,12 +661,12 @@ int zmq::curve_server_t::receive_and_process_zap_reply () ...@@ -661,12 +661,12 @@ int zmq::curve_server_t::receive_and_process_zap_reply ()
for (int i = 0; i < 7; i++) { for (int i = 0; i < 7; i++) {
rc = session->read_zap_msg (&msg [i]); rc = session->read_zap_msg (&msg [i]);
if (rc == -1) if (rc == -1)
return send_failure (msg); return close_and_return (msg, -1);
if ((msg [i].flags () & msg_t::more) == (i < 6? 0: msg_t::more)) { if ((msg [i].flags () & msg_t::more) == (i < 6? 0: msg_t::more)) {
// Temporary support for security debugging // Temporary support for security debugging
puts ("CURVE I: ZAP handler sent incomplete reply message"); puts ("CURVE I: ZAP handler sent incomplete reply message");
errno = EPROTO; errno = EPROTO;
return send_failure (msg); return close_and_return (msg, -1);
} }
} }
...@@ -675,7 +675,7 @@ int zmq::curve_server_t::receive_and_process_zap_reply () ...@@ -675,7 +675,7 @@ int zmq::curve_server_t::receive_and_process_zap_reply ()
// Temporary support for security debugging // Temporary support for security debugging
puts ("CURVE I: ZAP handler sent malformed reply message"); puts ("CURVE I: ZAP handler sent malformed reply message");
errno = EPROTO; errno = EPROTO;
return send_failure (msg); return close_and_return (msg, -1);
} }
// Version frame // Version frame
...@@ -683,7 +683,7 @@ int zmq::curve_server_t::receive_and_process_zap_reply () ...@@ -683,7 +683,7 @@ int zmq::curve_server_t::receive_and_process_zap_reply ()
// Temporary support for security debugging // Temporary support for security debugging
puts ("CURVE I: ZAP handler sent bad version number"); puts ("CURVE I: ZAP handler sent bad version number");
errno = EPROTO; errno = EPROTO;
return send_failure (msg); return close_and_return (msg, -1);
} }
// Request id frame // Request id frame
...@@ -691,7 +691,7 @@ int zmq::curve_server_t::receive_and_process_zap_reply () ...@@ -691,7 +691,7 @@ int zmq::curve_server_t::receive_and_process_zap_reply ()
// Temporary support for security debugging // Temporary support for security debugging
puts ("CURVE I: ZAP handler sent bad request ID"); puts ("CURVE I: ZAP handler sent bad request ID");
errno = EPROTO; errno = EPROTO;
return send_failure (msg); return close_and_return (msg, -1);
} }
// Status code frame // Status code frame
...@@ -699,7 +699,7 @@ int zmq::curve_server_t::receive_and_process_zap_reply () ...@@ -699,7 +699,7 @@ int zmq::curve_server_t::receive_and_process_zap_reply ()
// Temporary support for security debugging // Temporary support for security debugging
puts ("CURVE I: ZAP handler rejected client authentication"); puts ("CURVE I: ZAP handler rejected client authentication");
errno = EACCES; errno = EACCES;
return send_failure (msg); return close_and_return (msg, -1);
} }
// Save status code // Save status code
...@@ -713,7 +713,7 @@ int zmq::curve_server_t::receive_and_process_zap_reply () ...@@ -713,7 +713,7 @@ int zmq::curve_server_t::receive_and_process_zap_reply ()
msg [6].size (), true); msg [6].size (), true);
if (rc != 0) if (rc != 0)
return send_failure (msg); return close_and_return (msg, -1);
// Close all reply frames // Close all reply frames
for (int i = 0; i < 7; i++) { for (int i = 0; i < 7; i++) {
......
...@@ -161,7 +161,7 @@ int zmq::gssapi_server_t::send_zap_request () ...@@ -161,7 +161,7 @@ int zmq::gssapi_server_t::send_zap_request ()
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Version frame // Version frame
rc = msg.init_size (3); rc = msg.init_size (3);
...@@ -170,7 +170,7 @@ int zmq::gssapi_server_t::send_zap_request () ...@@ -170,7 +170,7 @@ int zmq::gssapi_server_t::send_zap_request ()
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Request ID frame // Request ID frame
rc = msg.init_size (1); rc = msg.init_size (1);
...@@ -179,7 +179,7 @@ int zmq::gssapi_server_t::send_zap_request () ...@@ -179,7 +179,7 @@ int zmq::gssapi_server_t::send_zap_request ()
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Domain frame // Domain frame
rc = msg.init_size (options.zap_domain.length ()); rc = msg.init_size (options.zap_domain.length ());
...@@ -188,7 +188,7 @@ int zmq::gssapi_server_t::send_zap_request () ...@@ -188,7 +188,7 @@ int zmq::gssapi_server_t::send_zap_request ()
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Address frame // Address frame
rc = msg.init_size (peer_address.length ()); rc = msg.init_size (peer_address.length ());
...@@ -197,7 +197,7 @@ int zmq::gssapi_server_t::send_zap_request () ...@@ -197,7 +197,7 @@ int zmq::gssapi_server_t::send_zap_request ()
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Identity frame // Identity frame
rc = msg.init_size (options.identity_size); rc = msg.init_size (options.identity_size);
...@@ -206,7 +206,7 @@ int zmq::gssapi_server_t::send_zap_request () ...@@ -206,7 +206,7 @@ int zmq::gssapi_server_t::send_zap_request ()
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Mechanism frame // Mechanism frame
rc = msg.init_size (6); rc = msg.init_size (6);
...@@ -215,7 +215,7 @@ int zmq::gssapi_server_t::send_zap_request () ...@@ -215,7 +215,7 @@ int zmq::gssapi_server_t::send_zap_request ()
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Principal frame // Principal frame
gss_buffer_desc principal; gss_buffer_desc principal;
...@@ -227,7 +227,7 @@ int zmq::gssapi_server_t::send_zap_request () ...@@ -227,7 +227,7 @@ int zmq::gssapi_server_t::send_zap_request ()
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
gss_release_buffer(&min_stat, &principal); gss_release_buffer(&min_stat, &principal);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
return 0; return 0;
} }
...@@ -246,35 +246,35 @@ int zmq::gssapi_server_t::receive_and_process_zap_reply () ...@@ -246,35 +246,35 @@ int zmq::gssapi_server_t::receive_and_process_zap_reply ()
for (int i = 0; i < 7; i++) { for (int i = 0; i < 7; i++) {
rc = session->read_zap_msg (&msg [i]); rc = session->read_zap_msg (&msg [i]);
if (rc == -1) if (rc == -1)
return send_failure (msg); return close_and_return (msg, -1);
if ((msg [i].flags () & msg_t::more) == (i < 6? 0: msg_t::more)) { if ((msg [i].flags () & msg_t::more) == (i < 6? 0: msg_t::more)) {
errno = EPROTO; errno = EPROTO;
return send_failure (msg); return close_and_return (msg, -1);
} }
} }
// Address delimiter frame // Address delimiter frame
if (msg [0].size () > 0) { if (msg [0].size () > 0) {
errno = EPROTO; errno = EPROTO;
return send_failure (msg); return close_and_return (msg, -1);
} }
// Version frame // Version frame
if (msg [1].size () != 3 || memcmp (msg [1].data (), "1.0", 3)) { if (msg [1].size () != 3 || memcmp (msg [1].data (), "1.0", 3)) {
errno = EPROTO; errno = EPROTO;
return send_failure (msg); return close_and_return (msg, -1);
} }
// Request id frame // Request id frame
if (msg [2].size () != 1 || memcmp (msg [2].data (), "1", 1)) { if (msg [2].size () != 1 || memcmp (msg [2].data (), "1", 1)) {
errno = EPROTO; errno = EPROTO;
return send_failure (msg); return close_and_return (msg, -1);
} }
// Status code frame // Status code frame
if (msg [3].size () != 3 || memcmp (msg [3].data (), "200", 3)) { if (msg [3].size () != 3 || memcmp (msg [3].data (), "200", 3)) {
errno = EACCES; errno = EACCES;
return send_failure (msg); return close_and_return (msg, -1);
} }
// Save user id // Save user id
...@@ -285,7 +285,7 @@ int zmq::gssapi_server_t::receive_and_process_zap_reply () ...@@ -285,7 +285,7 @@ int zmq::gssapi_server_t::receive_and_process_zap_reply ()
msg [6].size (), true); msg [6].size (), true);
if (rc != 0) if (rc != 0)
return send_failure (msg); return close_and_return (msg, -1);
// Close all reply frames // Close all reply frames
for (int i = 0; i < 7; i++) { for (int i = 0; i < 7; i++) {
......
...@@ -247,19 +247,21 @@ namespace zmq ...@@ -247,19 +247,21 @@ namespace zmq
} u; } u;
}; };
inline int send_failure (zmq::msg_t *msg) inline int close_and_return (zmq::msg_t *msg, int echo)
{ {
// Since we abort on close failure we preserve errno for success case.
int err = errno;
const int rc = msg->close (); const int rc = msg->close ();
errno_assert (rc == 0); errno_assert (rc == 0);
return -1; errno = err;
return echo;
} }
inline int send_failure (zmq::msg_t msg[], int count) inline int close_and_return (zmq::msg_t msg [], int count, int echo)
{ {
for (int i = 0; i < count; i++) for (int i = 0; i < count; i++)
send_failure (&msg [i]); close_and_return (&msg [i], 0);
return echo;
return -1;
} }
} }
......
...@@ -225,7 +225,7 @@ int zmq::null_mechanism_t::send_zap_request () ...@@ -225,7 +225,7 @@ int zmq::null_mechanism_t::send_zap_request ()
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Version frame // Version frame
rc = msg.init_size (3); rc = msg.init_size (3);
...@@ -234,7 +234,7 @@ int zmq::null_mechanism_t::send_zap_request () ...@@ -234,7 +234,7 @@ int zmq::null_mechanism_t::send_zap_request ()
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Request id frame // Request id frame
rc = msg.init_size (1); rc = msg.init_size (1);
...@@ -243,7 +243,7 @@ int zmq::null_mechanism_t::send_zap_request () ...@@ -243,7 +243,7 @@ int zmq::null_mechanism_t::send_zap_request ()
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Domain frame // Domain frame
rc = msg.init_size (options.zap_domain.length ()); rc = msg.init_size (options.zap_domain.length ());
...@@ -252,7 +252,7 @@ int zmq::null_mechanism_t::send_zap_request () ...@@ -252,7 +252,7 @@ int zmq::null_mechanism_t::send_zap_request ()
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Address frame // Address frame
rc = msg.init_size (peer_address.length ()); rc = msg.init_size (peer_address.length ());
...@@ -261,7 +261,7 @@ int zmq::null_mechanism_t::send_zap_request () ...@@ -261,7 +261,7 @@ int zmq::null_mechanism_t::send_zap_request ()
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Identity frame // Identity frame
rc = msg.init_size (options.identity_size); rc = msg.init_size (options.identity_size);
...@@ -270,7 +270,7 @@ int zmq::null_mechanism_t::send_zap_request () ...@@ -270,7 +270,7 @@ int zmq::null_mechanism_t::send_zap_request ()
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Mechanism frame // Mechanism frame
rc = msg.init_size (4); rc = msg.init_size (4);
...@@ -278,7 +278,7 @@ int zmq::null_mechanism_t::send_zap_request () ...@@ -278,7 +278,7 @@ int zmq::null_mechanism_t::send_zap_request ()
memcpy (msg.data (), "NULL", 4); memcpy (msg.data (), "NULL", 4);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
return 0; return 0;
} }
...@@ -297,12 +297,12 @@ int zmq::null_mechanism_t::receive_and_process_zap_reply () ...@@ -297,12 +297,12 @@ int zmq::null_mechanism_t::receive_and_process_zap_reply ()
for (int i = 0; i < 7; i++) { for (int i = 0; i < 7; i++) {
rc = session->read_zap_msg (&msg [i]); rc = session->read_zap_msg (&msg [i]);
if (rc == -1) if (rc == -1)
return send_failure (msg); return close_and_return (msg, -1);
if ((msg [i].flags () & msg_t::more) == (i < 6? 0: msg_t::more)) { if ((msg [i].flags () & msg_t::more) == (i < 6? 0: msg_t::more)) {
// Temporary support for security debugging // Temporary support for security debugging
puts ("NULL I: ZAP handler sent incomplete reply message"); puts ("NULL I: ZAP handler sent incomplete reply message");
errno = EPROTO; errno = EPROTO;
return send_failure (msg); return close_and_return (msg, -1);
} }
} }
...@@ -311,7 +311,7 @@ int zmq::null_mechanism_t::receive_and_process_zap_reply () ...@@ -311,7 +311,7 @@ int zmq::null_mechanism_t::receive_and_process_zap_reply ()
// Temporary support for security debugging // Temporary support for security debugging
puts ("NULL I: ZAP handler sent malformed reply message"); puts ("NULL I: ZAP handler sent malformed reply message");
errno = EPROTO; errno = EPROTO;
return send_failure (msg); return close_and_return (msg, -1);
} }
// Version frame // Version frame
...@@ -319,7 +319,7 @@ int zmq::null_mechanism_t::receive_and_process_zap_reply () ...@@ -319,7 +319,7 @@ int zmq::null_mechanism_t::receive_and_process_zap_reply ()
// Temporary support for security debugging // Temporary support for security debugging
puts ("NULL I: ZAP handler sent bad version number"); puts ("NULL I: ZAP handler sent bad version number");
errno = EPROTO; errno = EPROTO;
return send_failure (msg); return close_and_return (msg, -1);
} }
// Request id frame // Request id frame
...@@ -327,7 +327,7 @@ int zmq::null_mechanism_t::receive_and_process_zap_reply () ...@@ -327,7 +327,7 @@ int zmq::null_mechanism_t::receive_and_process_zap_reply ()
// Temporary support for security debugging // Temporary support for security debugging
puts ("NULL I: ZAP handler sent bad request ID"); puts ("NULL I: ZAP handler sent bad request ID");
errno = EPROTO; errno = EPROTO;
return send_failure (msg); return close_and_return (msg, -1);
} }
// Status code frame // Status code frame
...@@ -335,7 +335,7 @@ int zmq::null_mechanism_t::receive_and_process_zap_reply () ...@@ -335,7 +335,7 @@ int zmq::null_mechanism_t::receive_and_process_zap_reply ()
// Temporary support for security debugging // Temporary support for security debugging
puts ("NULL I: ZAP handler rejected client authentication"); puts ("NULL I: ZAP handler rejected client authentication");
errno = EPROTO; errno = EPROTO;
return send_failure (msg); return close_and_return (msg, -1);
} }
// Save status code // Save status code
...@@ -349,7 +349,7 @@ int zmq::null_mechanism_t::receive_and_process_zap_reply () ...@@ -349,7 +349,7 @@ int zmq::null_mechanism_t::receive_and_process_zap_reply ()
msg [6].size (), true); msg [6].size (), true);
if (rc != 0) if (rc != 0)
return send_failure (msg); return close_and_return (msg, -1);
// Close all reply frames // Close all reply frames
for (int i = 0; i < 7; i++) { for (int i = 0; i < 7; i++) {
......
...@@ -289,7 +289,7 @@ int zmq::plain_server_t::send_zap_request (const std::string &username, ...@@ -289,7 +289,7 @@ int zmq::plain_server_t::send_zap_request (const std::string &username,
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Version frame // Version frame
rc = msg.init_size (3); rc = msg.init_size (3);
...@@ -298,7 +298,7 @@ int zmq::plain_server_t::send_zap_request (const std::string &username, ...@@ -298,7 +298,7 @@ int zmq::plain_server_t::send_zap_request (const std::string &username,
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Request id frame // Request id frame
rc = msg.init_size (1); rc = msg.init_size (1);
...@@ -307,7 +307,7 @@ int zmq::plain_server_t::send_zap_request (const std::string &username, ...@@ -307,7 +307,7 @@ int zmq::plain_server_t::send_zap_request (const std::string &username,
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Domain frame // Domain frame
rc = msg.init_size (options.zap_domain.length ()); rc = msg.init_size (options.zap_domain.length ());
...@@ -316,7 +316,7 @@ int zmq::plain_server_t::send_zap_request (const std::string &username, ...@@ -316,7 +316,7 @@ int zmq::plain_server_t::send_zap_request (const std::string &username,
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Address frame // Address frame
rc = msg.init_size (peer_address.length ()); rc = msg.init_size (peer_address.length ());
...@@ -325,7 +325,7 @@ int zmq::plain_server_t::send_zap_request (const std::string &username, ...@@ -325,7 +325,7 @@ int zmq::plain_server_t::send_zap_request (const std::string &username,
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Identity frame // Identity frame
rc = msg.init_size (options.identity_size); rc = msg.init_size (options.identity_size);
...@@ -334,7 +334,7 @@ int zmq::plain_server_t::send_zap_request (const std::string &username, ...@@ -334,7 +334,7 @@ int zmq::plain_server_t::send_zap_request (const std::string &username,
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Mechanism frame // Mechanism frame
rc = msg.init_size (5); rc = msg.init_size (5);
...@@ -343,7 +343,7 @@ int zmq::plain_server_t::send_zap_request (const std::string &username, ...@@ -343,7 +343,7 @@ int zmq::plain_server_t::send_zap_request (const std::string &username,
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Username frame // Username frame
rc = msg.init_size (username.length ()); rc = msg.init_size (username.length ());
...@@ -352,7 +352,7 @@ int zmq::plain_server_t::send_zap_request (const std::string &username, ...@@ -352,7 +352,7 @@ int zmq::plain_server_t::send_zap_request (const std::string &username,
msg.set_flags (msg_t::more); msg.set_flags (msg_t::more);
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
// Password frame // Password frame
rc = msg.init_size (password.length ()); rc = msg.init_size (password.length ());
...@@ -360,7 +360,7 @@ int zmq::plain_server_t::send_zap_request (const std::string &username, ...@@ -360,7 +360,7 @@ int zmq::plain_server_t::send_zap_request (const std::string &username,
memcpy (msg.data (), password.c_str (), password.length ()); memcpy (msg.data (), password.c_str (), password.length ());
rc = session->write_zap_msg (&msg); rc = session->write_zap_msg (&msg);
if (rc != 0) if (rc != 0)
return send_failure (&msg); return close_and_return (&msg, -1);
return 0; return 0;
} }
...@@ -379,12 +379,12 @@ int zmq::plain_server_t::receive_and_process_zap_reply () ...@@ -379,12 +379,12 @@ int zmq::plain_server_t::receive_and_process_zap_reply ()
for (int i = 0; i < 7; i++) { for (int i = 0; i < 7; i++) {
rc = session->read_zap_msg (&msg [i]); rc = session->read_zap_msg (&msg [i]);
if (rc == -1) if (rc == -1)
return send_failure (msg); return close_and_return (msg, -1);
if ((msg [i].flags () & msg_t::more) == (i < 6? 0: msg_t::more)) { if ((msg [i].flags () & msg_t::more) == (i < 6? 0: msg_t::more)) {
// Temporary support for security debugging // Temporary support for security debugging
puts ("PLAIN I: ZAP handler sent incomplete reply message"); puts ("PLAIN I: ZAP handler sent incomplete reply message");
errno = EPROTO; errno = EPROTO;
return send_failure (msg); return close_and_return (msg, -1);
} }
} }
...@@ -393,7 +393,7 @@ int zmq::plain_server_t::receive_and_process_zap_reply () ...@@ -393,7 +393,7 @@ int zmq::plain_server_t::receive_and_process_zap_reply ()
// Temporary support for security debugging // Temporary support for security debugging
puts ("PLAIN I: ZAP handler sent malformed reply message"); puts ("PLAIN I: ZAP handler sent malformed reply message");
errno = EPROTO; errno = EPROTO;
return send_failure (msg); return close_and_return (msg, -1);
} }
// Version frame // Version frame
...@@ -401,7 +401,7 @@ int zmq::plain_server_t::receive_and_process_zap_reply () ...@@ -401,7 +401,7 @@ int zmq::plain_server_t::receive_and_process_zap_reply ()
// Temporary support for security debugging // Temporary support for security debugging
puts ("PLAIN I: ZAP handler sent bad version number"); puts ("PLAIN I: ZAP handler sent bad version number");
errno = EPROTO; errno = EPROTO;
return send_failure (msg); return close_and_return (msg, -1);
} }
// Request id frame // Request id frame
...@@ -409,7 +409,7 @@ int zmq::plain_server_t::receive_and_process_zap_reply () ...@@ -409,7 +409,7 @@ int zmq::plain_server_t::receive_and_process_zap_reply ()
// Temporary support for security debugging // Temporary support for security debugging
puts ("PLAIN I: ZAP handler sent bad request ID"); puts ("PLAIN I: ZAP handler sent bad request ID");
errno = EPROTO; errno = EPROTO;
return send_failure (msg); return close_and_return (msg, -1);
} }
// Status code frame // Status code frame
...@@ -417,7 +417,7 @@ int zmq::plain_server_t::receive_and_process_zap_reply () ...@@ -417,7 +417,7 @@ int zmq::plain_server_t::receive_and_process_zap_reply ()
// Temporary support for security debugging // Temporary support for security debugging
puts ("PLAIN I: ZAP handler rejected client authentication"); puts ("PLAIN I: ZAP handler rejected client authentication");
errno = EACCES; errno = EACCES;
return send_failure (msg); return close_and_return (msg, -1);
} }
// Save status code // Save status code
...@@ -431,7 +431,7 @@ int zmq::plain_server_t::receive_and_process_zap_reply () ...@@ -431,7 +431,7 @@ int zmq::plain_server_t::receive_and_process_zap_reply ()
msg [6].size (), true); msg [6].size (), true);
if (rc != 0) if (rc != 0)
return send_failure (msg); return close_and_return (msg, -1);
// Close all reply frames // Close all reply frames
for (int i = 0; i < 7; i++) { for (int i = 0; i < 7; i++) {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment