shinys000114/libssh
1
0
Fork 0
mirror of https://git.libssh.org/projects/libssh.git synced 2026-02-12 03:00:26 +09:00
Code Issues Packages Projects Releases Wiki Activity

Improve channel_write().

Browse Source 
git-svn-id: svn+ssh://svn.berlios.de/svnroot/repos/libssh/trunk@689 7dcaeef0-15fb-0310-b436-a5af3365683c
This commit is contained in:
Andreas Schneider
2009-05-04 06:30:12 +00:00
parent 17c146391c
commit 9345ba7030
1 changed files with 82 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files

151
libssh/channels.c
View File

@@ -847,82 +847,95 @@ error:
return rc; return rc;
} }
/** \brief blocking write on channel /**
* \param channel channel * @brief Blocking write on channel.
* \param data pointer to data to write *
* \param len length of data * @param channel The channel to write to.
* \return number of bytes written on success\n *
* SSH_ERROR on error * @param data A pointer to the data to write.
* \see channel_read() *
* @param len The length of the buffer to write to.
*
* @return The number of bytes written, SSH_ERROR on error.
*
* @see channel_read()
*/ */
int channel_write(CHANNEL *channel, const void *data, u32 len) { int channel_write(CHANNEL *channel, const void *data, u32 len) {
SSH_SESSION *session=channel->session; SSH_SESSION *session = channel->session;
int effectivelen; int origlen = len;
int origlen=len; int effectivelen;
enter_function();
if(channel->local_eof){ enter_function();
ssh_set_error(session,SSH_REQUEST_DENIED,"Can't write to channel %d:%d"
" after EOF was sent",channel->local_channel,channel->remote_channel); if (channel->local_eof) {
leave_function(); ssh_set_error(session, SSH_REQUEST_DENIED,
return -1; "Can't write to channel %d:%d after EOF was sent",
} channel->local_channel,
if(!channel->open || channel->delayed_close){ channel->remote_channel);
ssh_set_error(session,SSH_REQUEST_DENIED,"Remote channel is closed"); leave_function();
leave_function(); return -1;
return -1; }
}
if (channel->open == 0 || channel->delayed_close != 0) {
ssh_set_error(session, SSH_REQUEST_DENIED, "Remote channel is closed");
leave_function();
return -1;
}
#ifdef HAVE_SSH1 #ifdef HAVE_SSH1
if(channel->version==1){ if (channel->version == 1) {
int err = channel_write1(channel,data,len); int rc = channel_write1(channel, data, len);
leave_function(); leave_function();
return err; return rc;
} }
#endif #endif
while(len >0){
if(channel->remote_window<len){
ssh_log(session, SSH_LOG_PROTOCOL,
"Remote window is %d bytes. going to write %d bytes",
channel->remote_window,len);
ssh_log(session, SSH_LOG_PROTOCOL,
"Waiting for a growing window message...");
// wonder what happens when the channel window is zero
while(channel->remote_window==0){
// parse every incoming packet
packet_wait(channel->session,0,0);
}
effectivelen=len>channel->remote_window?channel->remote_window:len;
} else
effectivelen=len;
if (buffer_add_u8(session->out_buffer,SSH2_MSG_CHANNEL_DATA) < 0) {
goto error;
}
if (buffer_add_u32(session->out_buffer,htonl(channel->remote_channel)) < 0) {
goto error;
}
if (buffer_add_u32(session->out_buffer,htonl(effectivelen)) < 0) {
goto error;
}
if (buffer_add_data(session->out_buffer,data,effectivelen) < 0) {
goto error;
}
if (packet_send(session) != SSH_OK) { while (len > 0) {
leave_function(); if (channel->remote_window < len) {
return SSH_ERROR; ssh_log(session, SSH_LOG_PROTOCOL,
} "Remote window is %d bytes. going to write %d bytes",
ssh_log(session, SSH_LOG_RARE, channel->remote_window,
"channel_write wrote %d bytes", effectivelen); len);
channel->remote_window-=effectivelen; ssh_log(session, SSH_LOG_PROTOCOL,
len -= effectivelen; "Waiting for a growing window message...");
data+=effectivelen; /* What happens when the channel window is zero? */
while(channel->remote_window == 0) {
/* parse every incoming packet */
packet_wait(channel->session, 0, 0);
}
effectivelen = len > channel->remote_window ? channel->remote_window : len;
} else {
effectivelen = len;
} }
leave_function();
return origlen;
error:
buffer_free(session->out_buffer);
leave_function(); if (buffer_add_u8(session->out_buffer, SSH2_MSG_CHANNEL_DATA) < 0 ||
return SSH_ERROR; buffer_add_u32(session->out_buffer,
htonl(channel->remote_channel)) < 0 ||
buffer_add_u32(session->out_buffer, htonl(effectivelen)) < 0 ||
buffer_add_data(session->out_buffer, data, effectivelen) < 0) {
goto error;
}
if (packet_send(session) != SSH_OK) {
leave_function();
return SSH_ERROR;
}
ssh_log(session, SSH_LOG_RARE,
"channel_write wrote %d bytes", effectivelen);
channel->remote_window -= effectivelen;
len -= effectivelen;
data += effectivelen;
}
leave_function();
return origlen;
error:
buffer_free(session->out_buffer);
leave_function();
return SSH_ERROR;
} }
/** \brief returns if the channel is open or not /** \brief returns if the channel is open or not