diff options
-rw-r--r-- | include/libssh/socket.h | 1 | ||||
-rw-r--r-- | src/client.c | 2 | ||||
-rw-r--r-- | src/socket.c | 27 |
3 files changed, 22 insertions, 8 deletions
diff --git a/include/libssh/socket.h b/include/libssh/socket.h index 527f52ed..c95a2c61 100644 --- a/include/libssh/socket.h +++ b/include/libssh/socket.h @@ -62,6 +62,7 @@ int ssh_socket_pollcallback(struct ssh_poll_handle_struct *p, socket_t fd, int r struct ssh_poll_handle_struct * ssh_socket_get_poll_handle_in(ssh_socket s); struct ssh_poll_handle_struct * ssh_socket_get_poll_handle_out(ssh_socket s); +void ssh_socket_set_connecting(ssh_socket s, socket_t fd); int ssh_socket_connect(ssh_socket s, const char *host, int port, const char *bind_addr); #endif /* SOCKET_H_ */ diff --git a/src/client.c b/src/client.c index b3c3d93e..cc3c57b9 100644 --- a/src/client.c +++ b/src/client.c @@ -665,7 +665,7 @@ int ssh_connect(ssh_session session) { session->socket_callbacks.exception=ssh_socket_exception_callback; session->socket_callbacks.userdata=session; if (session->fd != SSH_INVALID_SOCKET) { - ssh_socket_set_fd(session->socket, session->fd); + ssh_socket_set_connecting(session->socket, session->fd); ret=SSH_OK; #ifndef _WIN32 } else if (session->ProxyCommand != NULL){ diff --git a/src/socket.c b/src/socket.c index 062739ae..7eb080f9 100644 --- a/src/socket.c +++ b/src/socket.c @@ -701,6 +701,25 @@ int ssh_socket_get_status(ssh_socket s) { return r; } + +/** + * @internal + * @brief Set a socket into the connecting state + * @param s socket handle + * @param fd file descriptor + */ + +void ssh_socket_set_connecting(ssh_socket s, int fd){ + ssh_socket_set_fd(s,fd); + s->state=SSH_SOCKET_CONNECTING; + /* POLLOUT is the event to wait for in a nonblocking connect */ + ssh_poll_set_events(ssh_socket_get_poll_handle_in(s),POLLOUT); +#ifdef _WIN32 + ssh_poll_add_events(ssh_socket_get_poll_handle_in(s),POLLWRNORM); +#endif +} + + /** * @internal * @brief Launches a socket connection @@ -729,13 +748,7 @@ int ssh_socket_connect(ssh_socket s, const char *host, int port, const char *bin ssh_log(session,SSH_LOG_PROTOCOL,"Nonblocking connection socket: %d",fd); if(fd == SSH_INVALID_SOCKET) return SSH_ERROR; - ssh_socket_set_fd(s,fd); - s->state=SSH_SOCKET_CONNECTING; - /* POLLOUT is the event to wait for in a nonblocking connect */ - ssh_poll_set_events(ssh_socket_get_poll_handle_in(s),POLLOUT); -#ifdef _WIN32 - ssh_poll_add_events(ssh_socket_get_poll_handle_in(s),POLLWRNORM); -#endif + ssh_socket_set_connecting(s,fd); leave_function(); return SSH_OK; } |