rosapps/net/ncftp/sio/SRecvfrom.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 SRecvfrom(int sfd, char *const buf, size_t size, int fl, struct sockaddr_in *const fromAddr, int tlen)
  12 {
  13         int nread, tleft;
  14         vsio_sigproc_t sigalrm, sigpipe;
  15         time_t done, now;
  16         int alen;
  17
  18         if (SSetjmp(gNetTimeoutJmp) != 0) {
  19                 alarm(0);
  20                 (void) SSignal(SIGALRM, (sio_sigproc_t) sigalrm);
  21                 (void) SSignal(SIGPIPE, (sio_sigproc_t) sigpipe);
  22                 errno = ETIMEDOUT;
  23                 return (kTimeoutErr);
  24         }
  25
  26         if (SSetjmp(gPipeJmp) != 0) {
  27                 alarm(0);
  28                 (void) SSignal(SIGALRM, (sio_sigproc_t) sigalrm);
  29                 (void) SSignal(SIGPIPE, (sio_sigproc_t) sigpipe);
  30                 errno = EPIPE;
  31                 return (kBrokenPipeErr);
  32         }
  33
  34         sigalrm = (vsio_sigproc_t) SSignal(SIGALRM, SIOHandler);
  35         sigpipe = (vsio_sigproc_t) SSignal(SIGPIPE, SIOHandler);
  36
  37         time(&now);
  38         done = now + tlen;
  39         tleft = (int) (done - now);
  40         forever {
  41                 alen = sizeof(struct sockaddr_in);
  42                 (void) alarm((unsigned int) tleft);
  43                 nread = recvfrom(sfd, buf, size, fl,
  44                         (struct sockaddr *) fromAddr, &alen);
  45                 (void) alarm(0);
  46                 if (nread >= 0)
  47                         break;
  48                 if (errno != EINTR)
  49                         break;          /* Fatal error. */
  50                 errno = 0;
  51                 time(&now);
  52                 tleft = (int) (done - now);
  53                 if (tleft < 1) {
  54                         nread = kTimeoutErr;
  55                         errno = ETIMEDOUT;
  56                         break;
  57                 }
  58         }
  59
  60         (void) SSignal(SIGALRM, (sio_sigproc_t) sigalrm);
  61         (void) SSignal(SIGPIPE, (sio_sigproc_t) sigpipe);
  62
  63         return (nread);
  64 }       /* SRecvfrom */
  65
  66 #else
  67
  68 int
  69 SRecvfrom(int sfd, char *const buf, size_t size, int fl, struct sockaddr_in *const fromAddr, int tlen)
  70 {
  71         int nread, tleft;
  72         fd_set ss;
  73         struct timeval tv;
  74         int result;
  75         time_t done, now;
  76         int alen;
  77
  78         time(&now);
  79         done = now + tlen;
  80         tleft = (int) (done - now);
  81         nread = 0;
  82         forever {
  83                 alen = sizeof(struct sockaddr_in);
  84
  85                 forever {
  86                         errno = 0;
  87                         FD_ZERO(&ss);
  88                         FD_SET(sfd, &ss);
  89                         tv.tv_sec = tleft;
  90                         tv.tv_usec = 0;
  91                         result = select(sfd + 1, SELECT_TYPE_ARG234 &ss, NULL, NULL, SELECT_TYPE_ARG5 &tv);
  92                         if (result == 1) {
  93                                 /* ready */
  94                                 break;
  95                         } else if (result == 0) {
  96                                 /* timeout */
  97                                 errno = ETIMEDOUT;
  98                                 SETWSATIMEOUTERR
  99                                 return (kTimeoutErr);
 100                         } else if (errno != EINTR) {
 101                                 return (-1);
 102                         }
 103                 }
 104
 105                 nread = recvfrom(sfd, buf, size, fl,
 106                         (struct sockaddr *) fromAddr, &alen);
 107
 108                 if (nread >= 0)
 109                         break;
 110                 if (errno != EINTR)
 111                         break;          /* Fatal error. */
 112                 errno = 0;
 113                 time(&now);
 114                 tleft = (int) (done - now);
 115                 if (tleft < 1) {
 116                         nread = kTimeoutErr;
 117                         errno = ETIMEDOUT;
 118                         SETWSATIMEOUTERR
 119                         break;
 120                 }
 121         }
 122
 123         return (nread);
 124 }       /* SRecvfrom */
 125
 126 #endif
 127