diff options
Diffstat (limited to 'common/tio.c')
-rw-r--r-- | common/tio.c | 320 |
1 files changed, 165 insertions, 155 deletions
diff --git a/common/tio.c b/common/tio.c index 4012364..f4a769a 100644 --- a/common/tio.c +++ b/common/tio.c @@ -76,17 +76,17 @@ struct tio_fileinfo { }; /* build a timeval for comparison to when the operation should be finished */ -static inline void tio_get_deadline(struct timeval *deadline,int timeout) +static inline void tio_get_deadline(struct timeval *deadline, int timeout) { - if (gettimeofday(deadline,NULL)) + if (gettimeofday(deadline, NULL)) { /* just blank it in case of errors */ - deadline->tv_sec=0; - deadline->tv_usec=0; + deadline->tv_sec = 0; + deadline->tv_usec = 0; return; } - deadline->tv_sec+=timeout/1000; - deadline->tv_usec+=(timeout%1000)*1000; + deadline->tv_sec += timeout / 1000; + deadline->tv_usec += (timeout % 1000) * 1000; } /* update the timeout to the value that is remaining before deadline @@ -95,62 +95,63 @@ static inline int tio_time_remaining(const struct timeval *deadline) { struct timeval tv; /* get the current time */ - if (gettimeofday(&tv,NULL)) + if (gettimeofday(&tv, NULL)) { /* 1 second default if gettimeofday() is broken */ return 1000; } /* calculate time remaining in miliseconds */ - return (deadline->tv_sec-tv.tv_sec)*1000 + (deadline->tv_usec-tv.tv_usec)/1000; + return (deadline->tv_sec - tv.tv_sec) * 1000 + + (deadline->tv_usec - tv.tv_usec) / 1000; } /* open a new TFILE based on the file descriptor */ -TFILE *tio_fdopen(int fd,int readtimeout,int writetimeout, - size_t initreadsize,size_t maxreadsize, - size_t initwritesize,size_t maxwritesize) +TFILE *tio_fdopen(int fd, int readtimeout, int writetimeout, + size_t initreadsize, size_t maxreadsize, + size_t initwritesize, size_t maxwritesize) { struct tio_fileinfo *fp; - fp=(struct tio_fileinfo *)malloc(sizeof(struct tio_fileinfo)); - if (fp==NULL) + fp = (struct tio_fileinfo *)malloc(sizeof(struct tio_fileinfo)); + if (fp == NULL) return NULL; - fp->fd=fd; + fp->fd = fd; /* initialize read buffer */ - fp->readbuffer.buffer=(uint8_t *)malloc(initreadsize); - if (fp->readbuffer.buffer==NULL) + fp->readbuffer.buffer = (uint8_t *)malloc(initreadsize); + if (fp->readbuffer.buffer == NULL) { free(fp); return NULL; } - fp->readbuffer.size=initreadsize; - fp->readbuffer.maxsize=maxreadsize; - fp->readbuffer.start=0; - fp->readbuffer.len=0; + fp->readbuffer.size = initreadsize; + fp->readbuffer.maxsize = maxreadsize; + fp->readbuffer.start = 0; + fp->readbuffer.len = 0; /* initialize write buffer */ - fp->writebuffer.buffer=(uint8_t *)malloc(initwritesize); - if (fp->writebuffer.buffer==NULL) + fp->writebuffer.buffer = (uint8_t *)malloc(initwritesize); + if (fp->writebuffer.buffer == NULL) { free(fp->readbuffer.buffer); free(fp); return NULL; } - fp->writebuffer.size=initwritesize; - fp->writebuffer.maxsize=maxwritesize; - fp->writebuffer.start=0; - fp->writebuffer.len=0; + fp->writebuffer.size = initwritesize; + fp->writebuffer.maxsize = maxwritesize; + fp->writebuffer.start = 0; + fp->writebuffer.len = 0; /* initialize other attributes */ - fp->readtimeout=readtimeout; - fp->writetimeout=writetimeout; - fp->read_resettable=0; + fp->readtimeout = readtimeout; + fp->writetimeout = writetimeout; + fp->read_resettable = 0; #ifdef DEBUG_TIO_STATS - fp->byteswritten=0; - fp->bytesread=0; + fp->byteswritten = 0; + fp->bytesread = 0; #endif /* DEBUG_TIO_STATS */ return fp; } /* wait for any activity on the specified file descriptor using the specified deadline */ -static int tio_wait(TFILE *fp,int readfd,const struct timeval *deadline) +static int tio_wait(TFILE *fp, int readfd, const struct timeval *deadline) { int timeout; struct pollfd fds[1]; @@ -158,38 +159,38 @@ static int tio_wait(TFILE *fp,int readfd,const struct timeval *deadline) while (1) { /* figure out the time we need to wait */ - if ((timeout=tio_time_remaining(deadline))<0) + if ((timeout = tio_time_remaining(deadline)) < 0) { - errno=ETIME; + errno = ETIME; return -1; } /* wait for activity */ if (readfd) { - fds[0].fd=fp->fd; - fds[0].events=POLLIN; + fds[0].fd = fp->fd; + fds[0].events = POLLIN; /* santiy check for moving clock */ - if (timeout>fp->readtimeout) - timeout=fp->readtimeout; + if (timeout > fp->readtimeout) + timeout = fp->readtimeout; } else { - fds[0].fd=fp->fd; - fds[0].events=POLLOUT; + fds[0].fd = fp->fd; + fds[0].events = POLLOUT; /* santiy check for moving clock */ - if (timeout>fp->writetimeout) - timeout=fp->writetimeout; + if (timeout > fp->writetimeout) + timeout = fp->writetimeout; } - rv=poll(fds,1,timeout); - if (rv>0) + rv = poll(fds, 1, timeout); + if (rv > 0) return 0; /* we have activity */ - else if (rv==0) + else if (rv == 0) { /* no file descriptors were available within the specified time */ - errno=ETIME; + errno = ETIME; return -1; } - else if (errno!=EINTR) + else if (errno != EINTR) /* some error ocurred */ return -1; /* we just try again on EINTR */ @@ -206,87 +207,88 @@ int tio_read(TFILE *fp, void *buf, size_t count) size_t newsz; size_t len; /* have a more convenient storage type for the buffer */ - uint8_t *ptr=(uint8_t *)buf; + uint8_t *ptr = (uint8_t *)buf; /* build a time by which we should be finished */ - tio_get_deadline(&deadline,fp->readtimeout); + tio_get_deadline(&deadline, fp->readtimeout); /* loop until we have returned all the needed data */ while (1) { /* check if we have enough data in the buffer */ if (fp->readbuffer.len >= count) { - if (count>0) + if (count > 0) { - if (ptr!=NULL) - memcpy(ptr,fp->readbuffer.buffer+fp->readbuffer.start,count); + if (ptr != NULL) + memcpy(ptr, fp->readbuffer.buffer + fp->readbuffer.start, count); /* adjust buffer position */ - fp->readbuffer.start+=count; - fp->readbuffer.len-=count; + fp->readbuffer.start += count; + fp->readbuffer.len -= count; } return 0; } /* empty what we have and continue from there */ - if (fp->readbuffer.len>0) + if (fp->readbuffer.len > 0) { - if (ptr!=NULL) + if (ptr != NULL) { - memcpy(ptr,fp->readbuffer.buffer+fp->readbuffer.start,fp->readbuffer.len); - ptr+=fp->readbuffer.len; + memcpy(ptr, fp->readbuffer.buffer + fp->readbuffer.start, + fp->readbuffer.len); + ptr += fp->readbuffer.len; } - count-=fp->readbuffer.len; - fp->readbuffer.start+=fp->readbuffer.len; - fp->readbuffer.len=0; + count -= fp->readbuffer.len; + fp->readbuffer.start += fp->readbuffer.len; + fp->readbuffer.len = 0; } /* after this point until the read fp->readbuffer.len is 0 */ if (!fp->read_resettable) { /* the stream is not resettable, re-use the buffer */ - fp->readbuffer.start=0; + fp->readbuffer.start = 0; } - else if (fp->readbuffer.start>=(fp->readbuffer.size-4)) + else if (fp->readbuffer.start >= (fp->readbuffer.size - 4)) { /* buffer is running empty, try to grow buffer */ - if (fp->readbuffer.size<fp->readbuffer.maxsize) + if (fp->readbuffer.size < fp->readbuffer.maxsize) { - newsz=fp->readbuffer.size*2; - if (newsz>fp->readbuffer.maxsize) - newsz=fp->readbuffer.maxsize; - tmp=realloc(fp->readbuffer.buffer,newsz); - if (tmp!=NULL) + newsz = fp->readbuffer.size * 2; + if (newsz > fp->readbuffer.maxsize) + newsz = fp->readbuffer.maxsize; + tmp = realloc(fp->readbuffer.buffer, newsz); + if (tmp != NULL) { - fp->readbuffer.buffer=tmp; - fp->readbuffer.size=newsz; + fp->readbuffer.buffer = tmp; + fp->readbuffer.size = newsz; } } /* if buffer still does not contain enough room, clear resettable */ - if (fp->readbuffer.start>=(fp->readbuffer.size-4)) + if (fp->readbuffer.start >= (fp->readbuffer.size - 4)) { - fp->readbuffer.start=0; - fp->read_resettable=0; + fp->readbuffer.start = 0; + fp->read_resettable = 0; } } /* wait until we have input */ - if (tio_wait(fp,1,&deadline)) + if (tio_wait(fp, 1, &deadline)) return -1; /* read the input in the buffer */ - len=fp->readbuffer.size-fp->readbuffer.start; + len = fp->readbuffer.size - fp->readbuffer.start; #ifdef SSIZE_MAX - if (len>SSIZE_MAX) - len=SSIZE_MAX; + if (len > SSIZE_MAX) + len = SSIZE_MAX; #endif /* SSIZE_MAX */ - rv=read(fp->fd,fp->readbuffer.buffer+fp->readbuffer.start,len); + rv = read(fp->fd, fp->readbuffer.buffer + fp->readbuffer.start, len); /* check for errors */ - if (rv==0) + if (rv == 0) { - errno=ECONNRESET; + errno = ECONNRESET; return -1; } - else if ((rv<0)&&(errno!=EINTR)&&(errno!=EAGAIN)) - return -1; /* something went wrong with the read */ + else if ((rv < 0) && (errno != EINTR) && (errno != EAGAIN)) + return -1; /* something went wrong with the read */ /* skip the read part in the buffer */ - fp->readbuffer.len=rv; + fp->readbuffer.len = rv; #ifdef DEBUG_TIO_STATS - fp->bytesread+=rv; + fp->bytesread += rv; #endif /* DEBUG_TIO_STATS */ } } @@ -294,7 +296,7 @@ int tio_read(TFILE *fp, void *buf, size_t count) /* Read and discard the specified number of bytes from the stream. */ int tio_skip(TFILE *fp, size_t count) { - return tio_read(fp,NULL,count); + return tio_read(fp, NULL, count); } /* Read all available data from the stream and empty the read buffer. */ @@ -304,35 +306,35 @@ int tio_skipall(TFILE *fp) int rv; size_t len; /* clear the read buffer */ - fp->readbuffer.start=0; - fp->readbuffer.len=0; - fp->read_resettable=0; + fp->readbuffer.start = 0; + fp->readbuffer.len = 0; + fp->read_resettable = 0; /* read until we can't read no more */ - len=fp->readbuffer.size; + len = fp->readbuffer.size; #ifdef SSIZE_MAX - if (len>SSIZE_MAX) - len=SSIZE_MAX; + if (len > SSIZE_MAX) + len = SSIZE_MAX; #endif /* SSIZE_MAX */ while (1) { /* see if any data is available */ - fds[0].fd=fp->fd; - fds[0].events=POLLIN; - rv=poll(fds,1,0); + fds[0].fd = fp->fd; + fds[0].events = POLLIN; + rv = poll(fds, 1, 0); /* check the poll() result */ - if (rv==0) + if (rv == 0) return 0; /* no file descriptor ready */ - if ((rv<0)&&((errno==EINTR)||(errno==EAGAIN))) + if ((rv < 0) && ((errno == EINTR) || (errno == EAGAIN))) continue; /* interrupted, try again */ - if (rv<0) + if (rv < 0) return -1; /* something went wrong */ /* read data from the stream */ - rv=read(fp->fd,fp->readbuffer.buffer,len); - if (rv==0) + rv = read(fp->fd, fp->readbuffer.buffer, len); + if (rv == 0) return 0; /* end-of-file */ - if ((rv<0)&&(errno==EWOULDBLOCK)) + if ((rv < 0) && (errno == EWOULDBLOCK)) return 0; /* we've ready everything we can without blocking */ - if ((rv<0)&&(errno!=EINTR)&&(errno!=EAGAIN)) + if ((rv < 0) && (errno != EINTR) && (errno != EAGAIN)) return -1; /* something went wrong with the read */ } } @@ -344,46 +346,50 @@ static int tio_writebuf(TFILE *fp) int rv; /* write the buffer */ #ifdef MSG_NOSIGNAL - rv=send(fp->fd,fp->writebuffer.buffer+fp->writebuffer.start,fp->writebuffer.len,MSG_NOSIGNAL); + rv = send(fp->fd, fp->writebuffer.buffer + fp->writebuffer.start, + fp->writebuffer.len, MSG_NOSIGNAL); #else /* not MSG_NOSIGNAL */ /* on platforms that cannot use send() with masked signals, we change the signal mask and change it back after the write (note that there is a race condition here) */ - struct sigaction act,oldact; + struct sigaction act, oldact; /* set up sigaction */ - memset(&act,0,sizeof(struct sigaction)); - act.sa_sigaction=NULL; - act.sa_handler=SIG_IGN; + memset(&act, 0, sizeof(struct sigaction)); + act.sa_sigaction = NULL; + act.sa_handler = SIG_IGN; sigemptyset(&act.sa_mask); - act.sa_flags=SA_RESTART; + act.sa_flags = SA_RESTART; /* ignore SIGPIPE */ - if (sigaction(SIGPIPE,&act,&oldact)!=0) + if (sigaction(SIGPIPE, &act, &oldact) != 0) return -1; /* error setting signal handler */ /* write the buffer */ - rv=write(fp->fd,fp->writebuffer.buffer+fp->writebuffer.start,fp->writebuffer.len); + rv = write(fp->fd, fp->writebuffer.buffer + fp->writebuffer.start, + fp->writebuffer.len); /* restore the old handler for SIGPIPE */ - if (sigaction(SIGPIPE,&oldact,NULL)!=0) + if (sigaction(SIGPIPE, &oldact, NULL) != 0) return -1; /* error restoring signal handler */ #endif /* check for errors */ - if ((rv==0)||((rv<0)&&(errno!=EINTR)&&(errno!=EAGAIN))) + if ((rv == 0) || ((rv < 0) && (errno != EINTR) && (errno != EAGAIN))) return -1; /* something went wrong with the write */ /* skip the written part in the buffer */ - if (rv>0) + if (rv > 0) { - fp->writebuffer.start+=rv; - fp->writebuffer.len-=rv; + fp->writebuffer.start += rv; + fp->writebuffer.len -= rv; #ifdef DEBUG_TIO_STATS - fp->byteswritten+=rv; + fp->byteswritten += rv; #endif /* DEBUG_TIO_STATS */ /* reset start if len is 0 */ - if (fp->writebuffer.len==0) - fp->writebuffer.start=0; + if (fp->writebuffer.len == 0) + fp->writebuffer.start = 0; /* move contents of the buffer to the front if it will save enough room */ - if (fp->writebuffer.start>=(fp->writebuffer.size/4)) + if (fp->writebuffer.start >= (fp->writebuffer.size / 4)) { - memmove(fp->writebuffer.buffer,fp->writebuffer.buffer+fp->writebuffer.start,fp->writebuffer.len); - fp->writebuffer.start=0; + memmove(fp->writebuffer.buffer, + fp->writebuffer.buffer + fp->writebuffer.start, + fp->writebuffer.len); + fp->writebuffer.start = 0; } } return 0; @@ -394,12 +400,12 @@ int tio_flush(TFILE *fp) { struct timeval deadline; /* build a time by which we should be finished */ - tio_get_deadline(&deadline,fp->writetimeout); + tio_get_deadline(&deadline, fp->writetimeout); /* loop until we have written our buffer */ while (fp->writebuffer.len > 0) { /* wait until we can write */ - if (tio_wait(fp,0,&deadline)) + if (tio_wait(fp, 0, &deadline)) return -1; /* write one block */ if (tio_writebuf(fp)) @@ -415,15 +421,15 @@ static int tio_flush_nonblock(TFILE *fp) struct pollfd fds[1]; int rv; /* see if we can write without blocking */ - fds[0].fd=fp->fd; - fds[0].events=POLLOUT; - rv=poll(fds,1,0); + fds[0].fd = fp->fd; + fds[0].events = POLLOUT; + rv = poll(fds, 1, 0); /* check if any file descriptors were ready (timeout) or we were interrupted */ - if ((rv==0)||((rv<0)&&(errno==EINTR))) + if ((rv == 0) || ((rv < 0) && (errno == EINTR))) return 0; /* any other errors? */ - if (rv<0) + if (rv < 0) return -1; /* so file descriptor will accept writes */ return tio_writebuf(fp); @@ -434,44 +440,46 @@ int tio_write(TFILE *fp, const void *buf, size_t count) size_t fr; uint8_t *tmp; size_t newsz; - const uint8_t *ptr=(const uint8_t *)buf; + const uint8_t *ptr = (const uint8_t *)buf; /* keep filling the buffer until we have bufferred everything */ - while (count>0) + while (count > 0) { /* figure out free size in buffer */ - fr=fp->writebuffer.size-(fp->writebuffer.start+fp->writebuffer.len); + fr = fp->writebuffer.size - (fp->writebuffer.start + fp->writebuffer.len); if (count <= fr) { /* the data fits in the buffer */ - memcpy(fp->writebuffer.buffer+fp->writebuffer.start+fp->writebuffer.len,ptr,count); - fp->writebuffer.len+=count; + memcpy(fp->writebuffer.buffer + fp->writebuffer.start + + fp->writebuffer.len, ptr, count); + fp->writebuffer.len += count; return 0; } else if (fr > 0) { /* fill the buffer with data that will fit */ - memcpy(fp->writebuffer.buffer+fp->writebuffer.start+fp->writebuffer.len,ptr,fr); - fp->writebuffer.len+=fr; - ptr+=fr; - count-=fr; + memcpy(fp->writebuffer.buffer + fp->writebuffer.start + + fp->writebuffer.len, ptr, fr); + fp->writebuffer.len += fr; + ptr += fr; + count -= fr; } /* try to flush some of the data that is in the buffer */ if (tio_flush_nonblock(fp)) return -1; /* if we have room now, try again */ - if (fp->writebuffer.size>(fp->writebuffer.start+fp->writebuffer.len)) + if (fp->writebuffer.size > (fp->writebuffer.start + fp->writebuffer.len)) continue; /* try to grow the buffer */ - if (fp->writebuffer.size<fp->writebuffer.maxsize) + if (fp->writebuffer.size < fp->writebuffer.maxsize) { - newsz=fp->writebuffer.size*2; - if (newsz>fp->writebuffer.maxsize) - newsz=fp->writebuffer.maxsize; - tmp=realloc(fp->writebuffer.buffer,newsz); - if (tmp!=NULL) + newsz = fp->writebuffer.size * 2; + if (newsz > fp->writebuffer.maxsize) + newsz = fp->writebuffer.maxsize; + tmp = realloc(fp->writebuffer.buffer, newsz); + if (tmp != NULL) { - fp->writebuffer.buffer=tmp; - fp->writebuffer.size=newsz; + fp->writebuffer.buffer = tmp; + fp->writebuffer.size = newsz; continue; /* try again */ } } @@ -486,14 +494,15 @@ int tio_close(TFILE *fp) { int retv; /* write any buffered data */ - retv=tio_flush(fp); + retv = tio_flush(fp); #ifdef DEBUG_TIO_STATS /* dump statistics to stderr */ - fprintf(stderr,"DEBUG_TIO_STATS READ=%d WRITTEN=%d\n",fp->bytesread,fp->byteswritten); + fprintf(stderr, "DEBUG_TIO_STATS READ=%d WRITTEN=%d\n", fp->bytesread, + fp->byteswritten); #endif /* DEBUG_TIO_STATS */ /* close file descriptor */ if (close(fp->fd)) - retv=-1; + retv = -1; /* free any allocated buffers */ free(fp->readbuffer.buffer); free(fp->writebuffer.buffer); @@ -506,13 +515,14 @@ int tio_close(TFILE *fp) void tio_mark(TFILE *fp) { /* move any data in the buffer to the start of the buffer */ - if ((fp->readbuffer.start>0)&&(fp->readbuffer.len>0)) + if ((fp->readbuffer.start > 0) && (fp->readbuffer.len > 0)) { - memmove(fp->readbuffer.buffer,fp->readbuffer.buffer+fp->readbuffer.start,fp->readbuffer.len); - fp->readbuffer.start=0; + memmove(fp->readbuffer.buffer, + fp->readbuffer.buffer + fp->readbuffer.start, fp->readbuffer.len); + fp->readbuffer.start = 0; } /* mark the stream as resettable */ - fp->read_resettable=1; + fp->read_resettable = 1; } int tio_reset(TFILE *fp) @@ -521,7 +531,7 @@ int tio_reset(TFILE *fp) if (!fp->read_resettable) return -1; /* reset the buffer */ - fp->readbuffer.len+=fp->readbuffer.start; - fp->readbuffer.start=0; + fp->readbuffer.len += fp->readbuffer.start; + fp->readbuffer.start = 0; return 0; } |