aboutsummaryrefslogtreecommitdiff
path: root/libssh/socket.c
diff options
context:
space:
mode:
authorAndreas Schneider <mail@cynapses.org>2009-04-14 14:24:41 +0000
committerAndreas Schneider <mail@cynapses.org>2009-04-14 14:24:41 +0000
commitf6e3f1d985dfe542ff4b5a67e69deed52f690730 (patch)
tree7c11fc43f79dba02e5f408da38166e291094d254 /libssh/socket.c
parent7eeb47ffad7dfb236589d5782d213d45a5701d0b (diff)
downloadlibssh-f6e3f1d985dfe542ff4b5a67e69deed52f690730.tar.gz
libssh-f6e3f1d985dfe542ff4b5a67e69deed52f690730.tar.xz
libssh-f6e3f1d985dfe542ff4b5a67e69deed52f690730.zip
Add more error checks to ssh_socket_nonblocking_flush().
git-svn-id: svn+ssh://svn.berlios.de/svnroot/repos/libssh/trunk@469 7dcaeef0-15fb-0310-b436-a5af3365683c
Diffstat (limited to 'libssh/socket.c')
-rw-r--r--libssh/socket.c93
1 files changed, 59 insertions, 34 deletions
diff --git a/libssh/socket.c b/libssh/socket.c
index 3bd3ed3..b493eb8 100644
--- a/libssh/socket.c
+++ b/libssh/socket.c
@@ -601,44 +601,69 @@ int ssh_socket_poll(struct socket *s, int *writeable, int *except) {
/** \internal
* \brief nonblocking flush of the output buffer
*/
-int ssh_socket_nonblocking_flush(struct socket *s){
- int except, can_write;
- int w;
- SSH_SESSION *session=s->session;
- enter_function();
- ssh_socket_poll(s,&can_write,&except); /* internally sets data_to_write */
- if(!ssh_socket_is_open(s)){
- session->alive=0;
- // FIXME use ssh_socket_get_errno
- ssh_set_error(session,SSH_FATAL,"Writing packet : error on socket (or connection closed): %s",strerror(errno));
- leave_function();
- return SSH_ERROR;
+int ssh_socket_nonblocking_flush(struct socket *s) {
+ SSH_SESSION *session = s->session;
+ int except;
+ int can_write;
+ int w;
+
+ enter_function();
+
+ /* internally sets data_to_write */
+ if (ssh_socket_poll(s, &can_write, &except) < 0) {
+ leave_function();
+ return SSH_ERROR;
+ }
+
+ if (!ssh_socket_is_open(s)) {
+ session->alive = 0;
+ /* FIXME use ssh_socket_get_errno */
+ ssh_set_error(session, SSH_FATAL,
+ "Writing packet: error on socket (or connection closed): %s",
+ strerror(errno));
+
+ leave_function();
+ return SSH_ERROR;
+ }
+
+ while(s->data_to_write && buffer_get_rest_len(s->out_buffer) > 0) {
+ if (ssh_socket_is_open(s)) {
+ w = ssh_socket_unbuffered_write(s, buffer_get_rest(s->out_buffer),
+ buffer_get_rest_len(s->out_buffer));
+ } else {
+ /* write failed */
+ w =- 1;
}
- while(s->data_to_write && buffer_get_rest_len(s->out_buffer)>0){
- if(ssh_socket_is_open(s)){
- w=ssh_socket_unbuffered_write(s,buffer_get_rest(s->out_buffer),
- buffer_get_rest_len(s->out_buffer));
- } else
- w=-1; /* write failed */
- if(w<0){
- session->alive=0;
- ssh_socket_close(s);
- // FIXME use ssh_socket_get_errno()
- ssh_set_error(session,SSH_FATAL,"Writing packet : error on socket (or connection closed): %s",
- strerror(errno));
- leave_function();
- return SSH_ERROR;
- }
- buffer_pass_bytes(s->out_buffer,w);
- /* refresh the socket status */
- ssh_socket_poll(session->socket,&can_write,&except);
+
+ if (w < 0) {
+ session->alive = 0;
+ ssh_socket_close(s);
+ /* FIXME use ssh_socket_get_errno() */
+ ssh_set_error(session, SSH_FATAL,
+ "Writing packet: error on socket (or connection closed): %s",
+ strerror(errno));
+
+ leave_function();
+ return SSH_ERROR;
}
- if(buffer_get_rest_len(s->out_buffer)>0){
- leave_function();
- return SSH_AGAIN; /* there is data pending */
+
+ buffer_pass_bytes(s->out_buffer, w);
+ /* refresh the socket status */
+ if (ssh_socket_poll(session->socket, &can_write, &except) < 0) {
+ leave_function();
+ return SSH_ERROR;
}
+ }
+
+ /* Is there some data pending? */
+ if (buffer_get_rest_len(s->out_buffer) > 0) {
leave_function();
- return SSH_OK; // all data written
+ return SSH_AGAIN;
+ }
+
+ /* all data written */
+ leave_function();
+ return SSH_OK;
}