[BACK]Return to sftp-server.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / ssh

Diff for /src/usr.bin/ssh/sftp-server.c between version 1.131 and 1.132

version 1.131, 2021/11/08 21:32:49 version 1.132, 2021/11/14 03:25:10
Line 25 
Line 25 
 #include <dirent.h>  #include <dirent.h>
 #include <errno.h>  #include <errno.h>
 #include <fcntl.h>  #include <fcntl.h>
   #include <poll.h>
 #include <stdlib.h>  #include <stdlib.h>
 #include <stdio.h>  #include <stdio.h>
 #include <string.h>  #include <string.h>
Line 1677 
Line 1678 
 int  int
 sftp_server_main(int argc, char **argv, struct passwd *user_pw)  sftp_server_main(int argc, char **argv, struct passwd *user_pw)
 {  {
         fd_set *rset, *wset;          int i, r, in, out, ch, skipargs = 0, log_stderr = 0;
         int i, r, in, out, max, ch, skipargs = 0, log_stderr = 0;          ssize_t len, olen;
         ssize_t len, olen, set_size;  
         SyslogFacility log_facility = SYSLOG_FACILITY_AUTH;          SyslogFacility log_facility = SYSLOG_FACILITY_AUTH;
         char *cp, *homedir = NULL, uidstr[32], buf[4*4096];          char *cp, *homedir = NULL, uidstr[32], buf[4*4096];
         long mask;          long mask;
Line 1779 
Line 1779 
         in = STDIN_FILENO;          in = STDIN_FILENO;
         out = STDOUT_FILENO;          out = STDOUT_FILENO;
   
         max = 0;  
         if (in > max)  
                 max = in;  
         if (out > max)  
                 max = out;  
   
         if ((iqueue = sshbuf_new()) == NULL)          if ((iqueue = sshbuf_new()) == NULL)
                 fatal_f("sshbuf_new failed");                  fatal_f("sshbuf_new failed");
         if ((oqueue = sshbuf_new()) == NULL)          if ((oqueue = sshbuf_new()) == NULL)
                 fatal_f("sshbuf_new failed");                  fatal_f("sshbuf_new failed");
   
         rset = xcalloc(howmany(max + 1, NFDBITS), sizeof(fd_mask));  
         wset = xcalloc(howmany(max + 1, NFDBITS), sizeof(fd_mask));  
   
         if (homedir != NULL) {          if (homedir != NULL) {
                 if (chdir(homedir) != 0) {                  if (chdir(homedir) != 0) {
                         error("chdir to \"%s\" failed: %s", homedir,                          error("chdir to \"%s\" failed: %s", homedir,
Line 1800 
Line 1791 
                 }                  }
         }          }
   
         set_size = howmany(max + 1, NFDBITS) * sizeof(fd_mask);  
         for (;;) {          for (;;) {
                 memset(rset, 0, set_size);                  struct pollfd pfd[2];
                 memset(wset, 0, set_size);  
   
                   memset(pfd, 0, sizeof pfd);
                   pfd[0].fd = pfd[1].fd = -1;
                   pfd[0].events = POLLIN;
                   pfd[1].events = POLLOUT;
   
                 /*                  /*
                  * Ensure that we can read a full buffer and handle                   * Ensure that we can read a full buffer and handle
                  * the worst-case length packet it can generate,                   * the worst-case length packet it can generate,
Line 1813 
Line 1807 
                 if ((r = sshbuf_check_reserve(iqueue, sizeof(buf))) == 0 &&                  if ((r = sshbuf_check_reserve(iqueue, sizeof(buf))) == 0 &&
                     (r = sshbuf_check_reserve(oqueue,                      (r = sshbuf_check_reserve(oqueue,
                     SFTP_MAX_MSG_LENGTH)) == 0)                      SFTP_MAX_MSG_LENGTH)) == 0)
                         FD_SET(in, rset);                          pfd[0].fd = in;
                 else if (r != SSH_ERR_NO_BUFFER_SPACE)                  else if (r != SSH_ERR_NO_BUFFER_SPACE)
                         fatal_fr(r, "reserve");                          fatal_fr(r, "reserve");
   
                 olen = sshbuf_len(oqueue);                  olen = sshbuf_len(oqueue);
                 if (olen > 0)                  if (olen > 0)
                         FD_SET(out, wset);                          pfd[1].fd = out;
   
                 if (select(max+1, rset, wset, NULL, NULL) == -1) {                  if (poll(pfd, 2, -1) == -1) {
                         if (errno == EINTR)                          if (errno == EINTR)
                                 continue;                                  continue;
                         error("select: %s", strerror(errno));                          error("poll: %s", strerror(errno));
                         sftp_server_cleanup_exit(2);                          sftp_server_cleanup_exit(2);
                 }                  }
   
                 /* copy stdin to iqueue */                  /* copy stdin to iqueue */
                 if (FD_ISSET(in, rset)) {                  if (pfd[0].revents & POLLIN) {
                         len = read(in, buf, sizeof buf);                          len = read(in, buf, sizeof buf);
                         if (len == 0) {                          if (len == 0) {
                                 debug("read eof");                                  debug("read eof");
Line 1841 
Line 1835 
                                 fatal_fr(r, "sshbuf_put");                                  fatal_fr(r, "sshbuf_put");
                 }                  }
                 /* send oqueue to stdout */                  /* send oqueue to stdout */
                 if (FD_ISSET(out, wset)) {                  if (pfd[1].revents & POLLOUT) {
                         len = write(out, sshbuf_ptr(oqueue), olen);                          len = write(out, sshbuf_ptr(oqueue), olen);
                         if (len == -1) {                          if (len == -1) {
                                 error("write: %s", strerror(errno));                                  error("write: %s", strerror(errno));
Legend:
Removed from v.1.131  
changed lines
  Added in v.1.132