3party/libzmq
0
0
Fork 0
mirror of https://github.com/zeromq/libzmq.git synced 2024-12-27 07:31:03 +08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #4030 from sigiesec/test-coverage-3

Browse Source 
Problem: code duplicated between handshake and out_event
This commit is contained in:
Luca Boccassi 2020-09-04 16:24:28 +01:00 committed by GitHub
commit 04ac547c20
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 14 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
src/wss_engine.cpp
View File

@ -133,6 +133,11 @@ void zmq::wss_engine_t::out_event ()
if (_established)
return ws_engine_t::out_event ();
do_handshake ();
}
bool zmq::wss_engine_t::do_handshake ()
{
int rc = gnutls_handshake (_tls_session);
reset_pollout ();
@ -140,42 +145,28 @@ void zmq::wss_engine_t::out_event ()
if (rc == GNUTLS_E_SUCCESS) {
start_ws_handshake ();
_established = true;
return;
return false;
} else if (rc == GNUTLS_E_AGAIN) {
int direction = gnutls_record_get_direction (_tls_session);
if (direction == 1)
set_pollout ();
return;
return false;
} else if (rc == GNUTLS_E_INTERRUPTED
|| rc == GNUTLS_E_WARNING_ALERT_RECEIVED) {
return;
return false;
} else {
error (zmq::i_engine::connection_error);
return;
return false;
}
return true;
}
bool zmq::wss_engine_t::handshake ()
{
if (!_established) {
int rc = gnutls_handshake (_tls_session);
if (rc == GNUTLS_E_SUCCESS) {
start_ws_handshake ();
_established = true;
return false;
} else if (rc == GNUTLS_E_AGAIN) {
int direction = gnutls_record_get_direction (_tls_session);
if (direction == 1)
set_pollout ();
return false;
} else if (rc == GNUTLS_E_INTERRUPTED
|| rc == GNUTLS_E_WARNING_ALERT_RECEIVED) {
return false;
} else {
error (zmq::i_engine::connection_error);
if (!do_handshake ()) {
return false;
}
}

3
src/wss_engine.hpp
View File

@ -57,8 +57,9 @@ class wss_engine_t : public ws_engine_t
int read (void *data, size_t size_);
int write (const void *data_, size_t size_);
private:
bool do_handshake ();
bool _established;
gnutls_certificate_credentials_t _tls_client_cred;
gnutls_session_t _tls_session;