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