/*	File:       restart.c	Contains:   Universal Internet Communication Interface restarting library calls				from _Unix System Programming_, Robbins & Robbins	Copyright:  (c) 2003 by Xten Networks, Inc., all rights reserved.*/#include <errno.h>#include <fcntl.h>#include <limits.h>#include <string.h>//#include <sys/select.h>#include <sys/time.h>#include <sys/wait.h>#include "restart.h"#define BLKSIZE PIPE_BUF#define MILLION 1000000L#define D_MILLION 1000000.0/* Private functions */static int gettimeout(struct timeval end,                               struct timeval *timeoutp) {   gettimeofday(timeoutp, NULL);   timeoutp->tv_sec = end.tv_sec - timeoutp->tv_sec;   timeoutp->tv_usec = end.tv_usec - timeoutp->tv_usec;   if (timeoutp->tv_usec >= MILLION) {      timeoutp->tv_sec++;      timeoutp->tv_usec -= MILLION;   }   if (timeoutp->tv_usec < 0) {      timeoutp->tv_sec--;      timeoutp->tv_usec += MILLION;   }   if ((timeoutp->tv_sec < 0) ||       ((timeoutp->tv_sec == 0) && (timeoutp->tv_usec == 0))) {      errno = ETIME;      return -1;   }   return 0;}/* Restart versions of traditional functions */int r_close(int fildes) {   int retval;   while (retval = close(fildes), retval == -1 && errno == EINTR) ;   return retval;}int r_dup2(int fildes, int fildes2) {   int retval;   while (retval = dup2(fildes, fildes2), retval == -1 && errno == EINTR) ;   return retval;}int r_open2(const char *path, int oflag) {   int retval;   while (retval = open(path, oflag), retval == -1 && errno == EINTR) ;   return retval;}int r_open3(const char *path, int oflag, mode_t mode) {   int retval;   while (retval = open(path, oflag, mode), retval == -1 && errno == EINTR) ;   return retval;}ssize_t r_read(int fd, void *buf, size_t size) {   ssize_t retval;   while (retval = read(fd, buf, size), retval == -1 && errno == EINTR) ;   return retval;}pid_t r_wait(int *stat_loc) {   pid_t retval;   while (((retval = wait(stat_loc)) == -1) && (errno == EINTR)) ;   return retval;}pid_t r_waitpid(pid_t pid, int *stat_loc, int options) {   pid_t retval;   while (((retval = waitpid(pid, stat_loc, options)) == -1) &&           (errno == EINTR)) ;   return retval;}ssize_t r_write(int fd, const void *buf, size_t size) {   char *bufp;   size_t bytestowrite;   ssize_t byteswritten;   size_t totalbytes;   //for (bufp = reinterpret_cast<char*>(buf), bytestowrite = size, totalbytes = 0;   for (bufp = (char*)buf, bytestowrite = size, totalbytes = 0;        bytestowrite > 0;        bufp += byteswritten, bytestowrite -= byteswritten) {      byteswritten = write(fd, bufp, bytestowrite);      if ((byteswritten) == -1 && (errno != EINTR))         return -1;      if (byteswritten == -1)         byteswritten = 0;      totalbytes += byteswritten;   }   return totalbytes;}/* Utility functions */struct timeval add2currenttime(double seconds) {   struct timeval newtime;   gettimeofday(&newtime, NULL);   newtime.tv_sec += (int)seconds;   newtime.tv_usec += (int)((seconds - (int)seconds)*D_MILLION + 0.5);   if (newtime.tv_usec >= MILLION) {      newtime.tv_sec++;      newtime.tv_usec -= MILLION;   }   return newtime;}int copyfile(int fromfd, int tofd) {   int bytesread;   int totalbytes = 0;   while ((bytesread = readwrite(fromfd, tofd)) > 0)      totalbytes += bytesread;   return totalbytes;}ssize_t readblock(int fd, void *buf, size_t size) {   char *bufp;   ssize_t bytesread;   size_t bytestoread;   size_t totalbytes;    //for (bufp = reinterpret_cast<char*>(buf), bytestoread = size, totalbytes = 0;   for (bufp = (char*)buf, bytestoread = size, totalbytes = 0;        bytestoread > 0;        bufp += bytesread, bytestoread -= bytesread) {      bytesread = read(fd, bufp, bytestoread);      if ((bytesread == 0) && (totalbytes == 0))         return 0;      if (bytesread == 0) {         errno = EINVAL;         return -1;      }        if ((bytesread) == -1 && (errno != EINTR))         return -1;      if (bytesread == -1)         bytesread = 0;      totalbytes += bytesread;   }   return totalbytes;}int readline(int fd, char *buf, int nbytes) {   int numread = 0;   int returnval;   while (numread < nbytes - 1) {      returnval = read(fd, buf + numread, 1);      if ((returnval == -1) && (errno == EINTR))         continue;      if ((returnval == 0) && (numread == 0))         return 0;      if (returnval == 0)         break;      if (returnval == -1)         return -1;      numread++;      if (buf[numread-1] == '\n') {         buf[numread] = '\0';         return numread;      }     }      errno = EINVAL;   return -1;}ssize_t readtimed(int fd, void *buf, size_t nbyte, double seconds) {   struct timeval timedone;   timedone = add2currenttime(seconds);   if (waitfdtimed(fd, timedone) == -1)      return (ssize_t)(-1);   return r_read(fd, buf, nbyte);}int readwrite(int fromfd, int tofd) {   char buf[BLKSIZE];   int bytesread;   if ((bytesread = r_read(fromfd, buf, BLKSIZE)) < 0)      return -1;   if (bytesread == 0)      return 0;   if (r_write(tofd, buf, bytesread) < 0)      return -1;   return bytesread;}int readwriteblock(int fromfd, int tofd, char *buf, int size) {   int bytesread;   bytesread = readblock(fromfd, buf, size);   if (bytesread != size)         /* can only be 0 or -1 */      return bytesread;   return r_write(tofd, buf, size);}int waitfdtimed(int fd, struct timeval end) {   fd_set readset;   int retval;   struct timeval timeout;    if ((fd < 0) || (fd >= FD_SETSIZE)) {      errno = EINVAL;      return -1;   }     FD_ZERO(&readset);   FD_SET(fd, &readset);   if (gettimeout(end, &timeout) == -1)      return -1;   while (((retval = select(fd+1, &readset, NULL, NULL, &timeout)) == -1)           && (errno == EINTR)) {      if (gettimeout(end, &timeout) == -1)         return -1;      FD_ZERO(&readset);      FD_SET(fd, &readset);   }   if (retval == 0) {      errno = ETIME;      return -1;   }   if (retval == -1)      return -1;   return 0;}int setblock(int fd) {   int fdflags;   if ((fdflags = fcntl(fd, F_GETFL, 0)) == -1)      return -1;   fdflags &= ~O_NONBLOCK;   if (fcntl(fd, F_SETFL, fdflags) == -1)      return -1;   return 0;}int setnonblock(int fd) {   int fdflags;   if ((fdflags = fcntl(fd, F_GETFL, 0)) == -1)      return -1;   fdflags |= O_NONBLOCK;   if (fcntl(fd, F_SETFL, fdflags) == -1)      return -1;   return 0;}