diff options
Diffstat (limited to 'examples/sshnetcat.c')
| -rw-r--r-- | examples/sshnetcat.c | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/examples/sshnetcat.c b/examples/sshnetcat.c index 21b9242..cad777b 100644 --- a/examples/sshnetcat.c +++ b/examples/sshnetcat.c @@ -125,8 +125,14 @@ static void select_loop(ssh_session session,ssh_channel channel){ channel_free(channel); channel=channels[0]=NULL; - } else - write(1,buffer,lus); + } else { + ret = write(1, buffer, lus); + if (ret < 0) { + fprintf(stderr, "Error writing to stdin: %s", + strerror(errno)); + return; + } + } } while(channel && channel_is_open(channel) && channel_poll(channel,1)){ /* stderr */ lus=channel_read(channel,buffer,sizeof(buffer),1); @@ -141,7 +147,12 @@ static void select_loop(ssh_session session,ssh_channel channel){ channel_free(channel); channel=channels[0]=NULL; } else - write(2,buffer,lus); + ret = write(2, buffer, lus); + if (ret < 0) { + fprintf(stderr, "Error writing to stderr: %s", + strerror(errno)); + return; + } } } if(channel && channel_is_closed(channel)){ |