rosapps/net/ncftp/sio/SSend.c

   1 #include "syshdrs.h"
   2
   3 #ifndef NO_SIGNALS
   4 extern volatile Sjmp_buf gNetTimeoutJmp;
   5 extern volatile Sjmp_buf gPipeJmp;
   6 #endif
   7
   8 #ifndef NO_SIGNALS
   9
  10 int
  11 SSend(int sfd, char *buf0, size_t size, int fl, int tlen)
  12 {
  13         volatile int nleft;
  14         char *volatile buf = buf0;
  15         int nwrote, tleft;
  16         vsio_sigproc_t sigalrm, sigpipe;
  17         time_t done, now;
  18
  19         if (SSetjmp(gNetTimeoutJmp) != 0) {
  20                 alarm(0);
  21                 (void) SSignal(SIGALRM, (sio_sigproc_t) sigalrm);
  22                 (void) SSignal(SIGPIPE, (sio_sigproc_t) sigpipe);
  23                 nwrote = size - nleft;
  24                 if (nwrote > 0)
  25                         return (nwrote);
  26                 errno = ETIMEDOUT;
  27                 return (kTimeoutErr);
  28         }
  29
  30         if (SSetjmp(gPipeJmp) != 0) {
  31                 alarm(0);
  32                 (void) SSignal(SIGALRM, (sio_sigproc_t) sigalrm);
  33                 (void) SSignal(SIGPIPE, (sio_sigproc_t) sigpipe);
  34                 nwrote = size - nleft;
  35                 if (nwrote > 0)
  36                         return (nwrote);
  37                 errno = EPIPE;
  38                 return (kBrokenPipeErr);
  39         }
  40
  41         sigalrm = (vsio_sigproc_t) SSignal(SIGALRM, SIOHandler);
  42         sigpipe = (vsio_sigproc_t) SSignal(SIGPIPE, SIOHandler);
  43
  44         nleft = (int) size;
  45         time(&now);
  46         done = now + tlen;
  47         forever {
  48                 tleft = (int) (done - now);
  49                 if (tleft < 1) {
  50                         nwrote = size - nleft;
  51                         if (nwrote == 0) {
  52                                 nwrote = kTimeoutErr;
  53                                 errno = ETIMEDOUT;
  54                         }
  55                         goto done;
  56                 }
  57                 (void) alarm((unsigned int) tleft);
  58                 nwrote = send(sfd, buf, size, fl);
  59                 (void) alarm(0);
  60                 if (nwrote < 0) {
  61                         if (errno != EINTR) {
  62                                 nwrote = size - nleft;
  63                                 if (nwrote == 0)
  64                                         nwrote = -1;
  65                                 goto done;
  66                         } else {
  67                                 errno = 0;
  68                                 nwrote = 0;
  69                                 /* Try again. */
  70                         }
  71                 }
  72                 nleft -= nwrote;
  73                 if (nleft <= 0)
  74                         break;
  75                 buf += nwrote;
  76                 time(&now);
  77         }
  78         nwrote = size - nleft;
  79
  80 done:
  81         (void) SSignal(SIGALRM, (sio_sigproc_t) sigalrm);
  82         (void) SSignal(SIGPIPE, (sio_sigproc_t) sigpipe);
  83
  84         return (nwrote);
  85 }       /* SSend */
  86
  87 #else
  88
  89 int
  90 SSend(int sfd, char *buf0, size_t size, int fl, int tlen)
  91 {
  92         int nleft;
  93         char *buf = buf0;
  94         int nwrote, tleft;
  95         time_t done, now;
  96         fd_set ss;
  97         struct timeval tv;
  98         int result;
  99
 100         nleft = (int) size;
 101         time(&now);
 102         done = now + tlen;
 103         forever {
 104                 tleft = (int) (done - now);
 105                 if (tleft < 1) {
 106                         nwrote = size - nleft;
 107                         if (nwrote == 0) {
 108                                 nwrote = kTimeoutErr;
 109                                 errno = ETIMEDOUT;
 110                                 SETWSATIMEOUTERR
 111                         }
 112                         goto done;
 113                 }
 114
 115
 116                 /* Unfortunately this doesn't help when the
 117                  * send buffer fills during the time we're
 118                  * writing to it, so you could still be
 119                  * blocked after breaking this loop and starting
 120                  * the write.
 121                  */
 122
 123                 forever {
 124                         errno = 0;
 125                         FD_ZERO(&ss);
 126                         FD_SET(sfd, &ss);
 127                         tv.tv_sec = tlen;
 128                         tv.tv_usec = 0;
 129                         result = select(sfd + 1, NULL, SELECT_TYPE_ARG234 &ss, NULL, SELECT_TYPE_ARG5 &tv);
 130                         if (result == 1) {
 131                                 /* ready */
 132                                 break;
 133                         } else if (result == 0) {
 134                                 /* timeout */
 135                                 nwrote = size - nleft;
 136                                 if (nwrote > 0)
 137                                         return (nwrote);
 138                                 errno = ETIMEDOUT;
 139                                 SETWSATIMEOUTERR
 140                                 return (kTimeoutErr);
 141                         } else if (errno != EINTR) {
 142                                 return (-1);
 143                         }
 144                 }
 145
 146                 nwrote = send(sfd, buf, size, fl);
 147
 148                 if (nwrote < 0) {
 149                         if (errno != EINTR) {
 150                                 nwrote = size - nleft;
 151                                 if (nwrote == 0)
 152                                         nwrote = -1;
 153                                 goto done;
 154                         } else {
 155                                 errno = 0;
 156                                 nwrote = 0;
 157                                 /* Try again. */
 158                         }
 159                 }
 160                 nleft -= nwrote;
 161                 if (nleft <= 0)
 162                         break;
 163                 buf += nwrote;
 164                 time(&now);
 165         }
 166         nwrote = size - nleft;
 167
 168 done:
 169         return (nwrote);
 170 }       /* SSend */
 171
 172 #endif