Annotation of embedaddon/libpdel/io/timeout_fp.c, revision 1.1
1.1 ! misho 1:
! 2: /*
! 3: * Copyright (c) 2001-2002 Packet Design, LLC.
! 4: * All rights reserved.
! 5: *
! 6: * Subject to the following obligations and disclaimer of warranty,
! 7: * use and redistribution of this software, in source or object code
! 8: * forms, with or without modifications are expressly permitted by
! 9: * Packet Design; provided, however, that:
! 10: *
! 11: * (i) Any and all reproductions of the source or object code
! 12: * must include the copyright notice above and the following
! 13: * disclaimer of warranties; and
! 14: * (ii) No rights are granted, in any manner or form, to use
! 15: * Packet Design trademarks, including the mark "PACKET DESIGN"
! 16: * on advertising, endorsements, or otherwise except as such
! 17: * appears in the above copyright notice or in the software.
! 18: *
! 19: * THIS SOFTWARE IS BEING PROVIDED BY PACKET DESIGN "AS IS", AND
! 20: * TO THE MAXIMUM EXTENT PERMITTED BY LAW, PACKET DESIGN MAKES NO
! 21: * REPRESENTATIONS OR WARRANTIES, EXPRESS OR IMPLIED, REGARDING
! 22: * THIS SOFTWARE, INCLUDING WITHOUT LIMITATION, ANY AND ALL IMPLIED
! 23: * WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE,
! 24: * OR NON-INFRINGEMENT. PACKET DESIGN DOES NOT WARRANT, GUARANTEE,
! 25: * OR MAKE ANY REPRESENTATIONS REGARDING THE USE OF, OR THE RESULTS
! 26: * OF THE USE OF THIS SOFTWARE IN TERMS OF ITS CORRECTNESS, ACCURACY,
! 27: * RELIABILITY OR OTHERWISE. IN NO EVENT SHALL PACKET DESIGN BE
! 28: * LIABLE FOR ANY DAMAGES RESULTING FROM OR ARISING OUT OF ANY USE
! 29: * OF THIS SOFTWARE, INCLUDING WITHOUT LIMITATION, ANY DIRECT,
! 30: * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, PUNITIVE, OR CONSEQUENTIAL
! 31: * DAMAGES, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES, LOSS OF
! 32: * USE, DATA OR PROFITS, HOWEVER CAUSED AND UNDER ANY THEORY OF
! 33: * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
! 34: * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
! 35: * THE USE OF THIS SOFTWARE, EVEN IF PACKET DESIGN IS ADVISED OF
! 36: * THE POSSIBILITY OF SUCH DAMAGE.
! 37: *
! 38: * Author: Archie Cobbs <archie@freebsd.org>
! 39: */
! 40:
! 41: #include <sys/types.h>
! 42: #include <sys/time.h>
! 43:
! 44: #include <stdio.h>
! 45: #include <stdlib.h>
! 46: #include <stdarg.h>
! 47: #include <syslog.h>
! 48: #include <string.h>
! 49: #include <unistd.h>
! 50: #include <errno.h>
! 51: #include <poll.h>
! 52:
! 53: #include "structs/structs.h"
! 54: #include "structs/type/array.h"
! 55:
! 56: #include "io/timeout_fp.h"
! 57: #include "util/typed_mem.h"
! 58:
! 59: #define MEM_TYPE "timeout_fp"
! 60:
! 61: struct fdt_info {
! 62: int fd;
! 63: int timeout;
! 64: };
! 65:
! 66: typedef int fun_reader(void *cookie, char *buf, int len);
! 67: typedef int fun_writer(void *cookie, const char *buf, int len);
! 68:
! 69: /*
! 70: * Internal functions
! 71: */
! 72: static int timeout_fp_readwrite(struct fdt_info *fdt,
! 73: char *buf, int len, int wr);
! 74: static int timeout_fp_close(void *cookie);
! 75:
! 76: static fun_reader timeout_fp_read;
! 77: static fun_writer timeout_fp_write;
! 78:
! 79: /*
! 80: * Same as fdopen(3) but with a timeout.
! 81: */
! 82: FILE *
! 83: timeout_fdopen(int fd, const char *mode, int timeout)
! 84: {
! 85: fun_reader *reader = NULL;
! 86: fun_writer *writer = NULL;
! 87: struct fdt_info *fdt;
! 88: const char *s;
! 89: FILE *fp;
! 90:
! 91: /* If no timeout, fall back to normal case */
! 92: if (timeout <= 0)
! 93: return (fdopen(fd, mode));
! 94:
! 95: /* Create the info structure */
! 96: if ((fdt = MALLOC(MEM_TYPE, sizeof(*fdt))) == NULL)
! 97: return (NULL);
! 98: memset(fdt, 0, sizeof(*fdt));
! 99: fdt->fd = fd;
! 100: fdt->timeout = timeout;
! 101:
! 102: /* Get reader & writer functions */
! 103: for (s = mode; *s != '\0'; s++) {
! 104: switch (*s) {
! 105: case 'r':
! 106: reader = timeout_fp_read;
! 107: break;
! 108: case 'w':
! 109: case 'a':
! 110: writer = timeout_fp_write;
! 111: break;
! 112: case '+':
! 113: reader = timeout_fp_read;
! 114: writer = timeout_fp_write;
! 115: break;
! 116: }
! 117: }
! 118:
! 119: /* Get FILE * wrapper */
! 120: if ((fp = funopen(fdt, reader, writer,
! 121: NULL, timeout_fp_close)) == NULL) {
! 122: FREE(MEM_TYPE, fdt);
! 123: return (NULL);
! 124: }
! 125:
! 126: /* Done */
! 127: return (fp);
! 128: }
! 129:
! 130: static int
! 131: timeout_fp_read(void *cookie, char *buf, int len)
! 132: {
! 133: struct fdt_info *const fdt = cookie;
! 134:
! 135: return (timeout_fp_readwrite(fdt, buf, len, 0));
! 136: }
! 137:
! 138: static int
! 139: timeout_fp_write(void *cookie, const char *buf, int len)
! 140: {
! 141: struct fdt_info *const fdt = cookie;
! 142:
! 143: return (timeout_fp_readwrite(fdt, (char *)buf, len, 1));
! 144: }
! 145:
! 146: /*
! 147: * Do the threaded read or write depending on the value of 'wr'.
! 148: */
! 149: static int
! 150: timeout_fp_readwrite(struct fdt_info *fdt, char *buf, int len, int wr)
! 151: {
! 152: struct pollfd pfd;
! 153: int nfds;
! 154:
! 155: /* Set up read/write poll(2) event */
! 156: memset(&pfd, 0, sizeof(pfd));
! 157: pfd.fd = fdt->fd;
! 158: pfd.events = wr ? POLLWRNORM : POLLRDNORM;
! 159:
! 160: /* Wait for readability or writability */
! 161: if ((nfds = poll(&pfd, 1, fdt->timeout * 1000)) == -1)
! 162: return (-1);
! 163:
! 164: /* Check for timeout */
! 165: if (nfds == 0) {
! 166: errno = ETIMEDOUT;
! 167: return (-1);
! 168: }
! 169:
! 170: /* Do I/O */
! 171: return (wr ? write(fdt->fd, buf, len) : read(fdt->fd, buf, len));
! 172: }
! 173:
! 174: /*
! 175: * Closer for timeout_fp streams.
! 176: */
! 177: static int
! 178: timeout_fp_close(void *cookie)
! 179: {
! 180: struct fdt_info *const fdt = cookie;
! 181: int r;
! 182:
! 183: r = close(fdt->fd);
! 184: FREE(MEM_TYPE, fdt);
! 185: return (r);
! 186: }
! 187:
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to timeout_fp.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / libpdel / io