[BACK]Return to file.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / sndiod

Annotation of src/usr.bin/sndiod/file.c, Revision 1.7

1.7     ! ratchov     1: /*     $OpenBSD: file.c,v 1.6 2015/02/16 06:26:24 ratchov Exp $        */
1.1       ratchov     2: /*
                      3:  * Copyright (c) 2008-2012 Alexandre Ratchov <alex@caoua.org>
                      4:  *
                      5:  * Permission to use, copy, modify, and distribute this software for any
                      6:  * purpose with or without fee is hereby granted, provided that the above
                      7:  * copyright notice and this permission notice appear in all copies.
                      8:  *
                      9:  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
                     10:  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
                     11:  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
                     12:  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
                     13:  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
                     14:  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
                     15:  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
                     16:  */
                     17: /*
                     18:  * non-blocking file i/o module: each file can be read or written (or
                     19:  * both). To achieve non-blocking io, we simply use the poll() syscall
                     20:  * in an event loop and dispatch events to sub-modules.
                     21:  *
                     22:  * the module also provides trivial timeout implementation,
                     23:  * derived from:
                     24:  *
                     25:  *     anoncvs@moule.caoua.org:/midish
                     26:  *
                     27:  *             midish/timo.c rev 1.18
                     28:  *             midish/mdep.c rev 1.71
                     29:  *
                     30:  * A timeout is used to schedule the call of a routine (the callback)
                     31:  * there is a global list of timeouts that is processed inside the
                     32:  * event loop. Timeouts work as follows:
                     33:  *
                     34:  *     first the timo structure must be initialized with timo_set()
                     35:  *
                     36:  *     then the timeout is scheduled (only once) with timo_add()
                     37:  *
                     38:  *     if the timeout expires, the call-back is called; then it can
                     39:  *     be scheduled again if needed. It's OK to reschedule it again
                     40:  *     from the callback
                     41:  *
                     42:  *     the timeout can be aborted with timo_del(), it is OK to try to
                     43:  *     abort a timout that has expired
                     44:  *
                     45:  */
                     46:
                     47: #include <sys/types.h>
                     48:
                     49: #include <err.h>
                     50: #include <errno.h>
                     51: #include <fcntl.h>
                     52: #include <poll.h>
                     53: #include <signal.h>
                     54: #include <stdio.h>
                     55: #include <stdlib.h>
                     56: #include <time.h>
                     57:
                     58: #include "file.h"
                     59: #include "utils.h"
                     60:
                     61: #define MAXFDS 100
1.7     ! ratchov    62: #define TIMER_MSEC 5
1.4       ratchov    63:
                     64: void timo_update(unsigned int);
                     65: void timo_init(void);
                     66: void timo_done(void);
1.1       ratchov    67:
                     68: struct timespec file_ts;
                     69: struct file *file_list;
                     70: struct timo *timo_queue;
                     71: unsigned int timo_abstime;
                     72: int file_slowaccept = 0, file_nfds;
                     73: #ifdef DEBUG
                     74: long long file_wtime, file_utime;
                     75: #endif
                     76:
                     77: /*
                     78:  * initialise a timeout structure, arguments are callback and argument
                     79:  * that will be passed to the callback
                     80:  */
                     81: void
                     82: timo_set(struct timo *o, void (*cb)(void *), void *arg)
                     83: {
                     84:        o->cb = cb;
                     85:        o->arg = arg;
                     86:        o->set = 0;
                     87: }
                     88:
                     89: /*
                     90:  * schedule the callback in 'delta' 24-th of microseconds. The timeout
                     91:  * must not be already scheduled
                     92:  */
                     93: void
                     94: timo_add(struct timo *o, unsigned int delta)
                     95: {
                     96:        struct timo **i;
                     97:        unsigned int val;
                     98:        int diff;
                     99:
                    100: #ifdef DEBUG
                    101:        if (o->set) {
                    102:                log_puts("timo_add: already set\n");
                    103:                panic();
                    104:        }
                    105:        if (delta == 0) {
                    106:                log_puts("timo_add: zero timeout is evil\n");
                    107:                panic();
                    108:        }
                    109: #endif
                    110:        val = timo_abstime + delta;
                    111:        for (i = &timo_queue; *i != NULL; i = &(*i)->next) {
                    112:                diff = (*i)->val - val;
                    113:                if (diff > 0) {
                    114:                        break;
                    115:                }
                    116:        }
                    117:        o->set = 1;
                    118:        o->val = val;
                    119:        o->next = *i;
                    120:        *i = o;
                    121: }
                    122:
                    123: /*
                    124:  * abort a scheduled timeout
                    125:  */
                    126: void
                    127: timo_del(struct timo *o)
                    128: {
                    129:        struct timo **i;
                    130:
                    131:        for (i = &timo_queue; *i != NULL; i = &(*i)->next) {
                    132:                if (*i == o) {
                    133:                        *i = o->next;
                    134:                        o->set = 0;
                    135:                        return;
                    136:                }
                    137:        }
                    138: #ifdef DEBUG
                    139:        if (log_level >= 4)
                    140:                log_puts("timo_del: not found\n");
                    141: #endif
                    142: }
                    143:
                    144: /*
                    145:  * routine to be called by the timer when 'delta' 24-th of microsecond
                    146:  * elapsed. This routine updates time referece used by timeouts and
                    147:  * calls expired timeouts
                    148:  */
                    149: void
                    150: timo_update(unsigned int delta)
                    151: {
                    152:        struct timo *to;
                    153:        int diff;
                    154:
                    155:        /*
                    156:         * update time reference
                    157:         */
                    158:        timo_abstime += delta;
                    159:
                    160:        /*
                    161:         * remove from the queue and run expired timeouts
                    162:         */
                    163:        while (timo_queue != NULL) {
                    164:                /*
                    165:                 * there is no overflow here because + and - are
                    166:                 * modulo 2^32, they are the same for both signed and
                    167:                 * unsigned integers
                    168:                 */
                    169:                diff = timo_queue->val - timo_abstime;
                    170:                if (diff > 0)
                    171:                        break;
                    172:                to = timo_queue;
                    173:                timo_queue = to->next;
                    174:                to->set = 0;
                    175:                to->cb(to->arg);
                    176:        }
                    177: }
                    178:
                    179: /*
                    180:  * initialize timeout queue
                    181:  */
                    182: void
                    183: timo_init(void)
                    184: {
                    185:        timo_queue = NULL;
                    186:        timo_abstime = 0;
                    187: }
                    188:
                    189: /*
                    190:  * destroy timeout queue
                    191:  */
                    192: void
                    193: timo_done(void)
                    194: {
                    195: #ifdef DEBUG
                    196:        if (timo_queue != NULL) {
                    197:                log_puts("timo_done: timo_queue not empty!\n");
                    198:                panic();
                    199:        }
                    200: #endif
                    201:        timo_queue = (struct timo *)0xdeadbeef;
                    202: }
                    203:
                    204: #ifdef DEBUG
                    205: void
                    206: file_log(struct file *f)
                    207: {
1.5       ratchov   208:        static char *states[] = { "ini", "zom" };
1.1       ratchov   209:
                    210:        log_puts(f->ops->name);
                    211:        if (log_level >= 3) {
                    212:                log_puts("(");
                    213:                log_puts(f->name);
                    214:                log_puts("|");
                    215:                log_puts(states[f->state]);
                    216:                log_puts(")");
                    217:        }
                    218: }
                    219: #endif
                    220:
                    221: struct file *
                    222: file_new(struct fileops *ops, void *arg, char *name, unsigned int nfds)
                    223: {
                    224:        struct file *f;
                    225:
                    226:        if (file_nfds + nfds > MAXFDS) {
                    227: #ifdef DEBUG
                    228:                if (log_level >= 1) {
                    229:                        log_puts(name);
                    230:                        log_puts(": too many polled files\n");
                    231:                }
                    232: #endif
                    233:                return NULL;
                    234:        }
                    235:        f = xmalloc(sizeof(struct file));
                    236:        f->nfds = nfds;
                    237:        f->ops = ops;
                    238:        f->arg = arg;
                    239:        f->name = name;
                    240:        f->state = FILE_INIT;
                    241:        f->next = file_list;
                    242:        file_list = f;
                    243: #ifdef DEBUG
                    244:        if (log_level >= 3) {
                    245:                file_log(f);
                    246:                log_puts(": created\n");
                    247:        }
                    248: #endif
                    249:        file_nfds += f->nfds;
                    250:        return f;
                    251: }
                    252:
                    253: void
                    254: file_del(struct file *f)
                    255: {
                    256: #ifdef DEBUG
                    257:        if (f->state == FILE_ZOMB) {
                    258:                log_puts("bad state in file_del()\n");
                    259:                panic();
                    260:        }
                    261: #endif
                    262:        file_nfds -= f->nfds;
                    263:        f->state = FILE_ZOMB;
                    264: #ifdef DEBUG
                    265:        if (log_level >= 3) {
                    266:                file_log(f);
                    267:                log_puts(": destroyed\n");
                    268:        }
                    269: #endif
                    270: }
                    271:
                    272: int
                    273: file_poll(void)
                    274: {
                    275:        struct pollfd pfds[MAXFDS];
                    276:        struct file *f, **pf;
                    277:        struct timespec ts;
                    278: #ifdef DEBUG
                    279:        struct timespec sleepts;
                    280:        struct timespec ts0, ts1;
                    281:        long us;
1.6       ratchov   282:        int i, n, nfds;
1.1       ratchov   283: #endif
                    284:        long long delta_nsec;
1.6       ratchov   285:        int revents, res, immed;
1.1       ratchov   286:
                    287:        /*
                    288:         * cleanup zombies
                    289:         */
                    290:        pf = &file_list;
                    291:        while ((f = *pf) != NULL) {
                    292:                if (f->state == FILE_ZOMB) {
                    293:                        *pf = f->next;
                    294:                        xfree(f);
                    295:                } else
                    296:                        pf = &f->next;
                    297:        }
                    298:
                    299:        if (file_list == NULL && timo_queue == NULL) {
                    300: #ifdef DEBUG
                    301:                if (log_level >= 3)
                    302:                        log_puts("nothing to do...\n");
                    303: #endif
                    304:                return 0;
                    305:        }
                    306:
                    307:        log_flush();
                    308: #ifdef DEBUG
                    309:        if (log_level >= 4)
                    310:                log_puts("poll:");
                    311: #endif
                    312:        nfds = 0;
1.6       ratchov   313:        immed = 0;
1.1       ratchov   314:        for (f = file_list; f != NULL; f = f->next) {
                    315: #ifdef DEBUG
                    316:                if (log_level >= 4) {
                    317:                        log_puts(" ");
                    318:                        file_log(f);
                    319:                }
                    320: #endif
                    321:                n = f->ops->pollfd(f->arg, pfds + nfds);
                    322:                if (n == 0) {
                    323:                        f->pfd = NULL;
                    324:                        continue;
                    325:                }
1.6       ratchov   326:                if (n < 0) {
                    327:                        immed = 1;
                    328:                        n = 0;
                    329:                }
1.1       ratchov   330:                f->pfd = pfds + nfds;
                    331:                nfds += n;
                    332: #ifdef DEBUG
                    333:                if (log_level >= 4) {
                    334:                        log_puts("=");
                    335:                        for (i = 0; i < n; i++) {
                    336:                                if (i > 0)
                    337:                                        log_puts(",");
                    338:                                log_putx(f->pfd[i].events);
                    339:                        }
                    340:                }
                    341: #endif
                    342:        }
                    343: #ifdef DEBUG
                    344:        if (log_level >= 4)
                    345:                log_puts("\n");
                    346: #endif
                    347:
                    348: #ifdef DEBUG
                    349:        clock_gettime(CLOCK_MONOTONIC, &sleepts);
                    350:        file_utime += 1000000000LL * (sleepts.tv_sec - file_ts.tv_sec);
                    351:        file_utime += sleepts.tv_nsec - file_ts.tv_nsec;
                    352: #endif
1.6       ratchov   353:        if (!immed) {
1.7     ! ratchov   354:                res = poll(pfds, nfds, TIMER_MSEC);
1.6       ratchov   355:                if (res < 0 && errno != EINTR)
                    356:                        err(1, "poll");
                    357: #ifdef DEBUG
                    358:                if (log_level >= 4 && res >= 0) {
                    359:                        log_puts("poll: return:");
                    360:                        for (i = 0; i < nfds; i++) {
                    361:                                log_puts(" ");
                    362:                                log_putx(pfds[i].revents);
                    363:                        }
                    364:                        log_puts("\n");
1.1       ratchov   365:                }
                    366: #endif
1.6       ratchov   367:        } else
                    368:                res = 0;
1.1       ratchov   369:        clock_gettime(CLOCK_MONOTONIC, &ts);
                    370: #ifdef DEBUG
                    371:        file_wtime += 1000000000LL * (ts.tv_sec - sleepts.tv_sec);
                    372:        file_wtime += ts.tv_nsec - sleepts.tv_nsec;
                    373: #endif
                    374:        delta_nsec = 1000000000LL * (ts.tv_sec - file_ts.tv_sec);
                    375:        delta_nsec += ts.tv_nsec - file_ts.tv_nsec;
                    376: #ifdef DEBUG
                    377:        if (delta_nsec < 0)
                    378:                log_puts("file_poll: negative time interval\n");
                    379: #endif
                    380:        file_ts = ts;
                    381:        if (delta_nsec >= 0 && delta_nsec < 1000000000LL)
                    382:                timo_update(delta_nsec / 1000);
                    383:        else {
                    384:                if (log_level >= 2)
                    385:                        log_puts("ignored huge clock delta\n");
                    386:        }
1.6       ratchov   387:        if (!immed && res <= 0)
1.1       ratchov   388:                return 1;
                    389:
                    390:        for (f = file_list; f != NULL; f = f->next) {
                    391:                if (f->pfd == NULL)
                    392:                        continue;
                    393: #ifdef DEBUG
                    394:                clock_gettime(CLOCK_MONOTONIC, &ts0);
                    395: #endif
1.3       ratchov   396:                revents = (f->state != FILE_ZOMB) ?
                    397:                    f->ops->revents(f->arg, f->pfd) : 0;
1.1       ratchov   398:                if ((revents & POLLHUP) && (f->state != FILE_ZOMB))
                    399:                        f->ops->hup(f->arg);
                    400:                if ((revents & POLLIN) && (f->state != FILE_ZOMB))
                    401:                        f->ops->in(f->arg);
                    402:                if ((revents & POLLOUT) && (f->state != FILE_ZOMB))
                    403:                        f->ops->out(f->arg);
                    404: #ifdef DEBUG
                    405:                clock_gettime(CLOCK_MONOTONIC, &ts1);
                    406:                us = 1000000L * (ts1.tv_sec - ts0.tv_sec);
                    407:                us += (ts1.tv_nsec - ts0.tv_nsec) / 1000;
                    408:                if (log_level >= 4 || (log_level >= 3 && us >= 5000)) {
                    409:                        file_log(f);
                    410:                        log_puts(": processed in ");
                    411:                        log_putu(us);
                    412:                        log_puts("us\n");
                    413:                }
                    414: #endif
                    415:        }
                    416:        return 1;
                    417: }
                    418:
                    419: void
                    420: filelist_init(void)
                    421: {
                    422:        sigset_t set;
                    423:
                    424:        sigemptyset(&set);
                    425:        (void)sigaddset(&set, SIGPIPE);
                    426:        if (sigprocmask(SIG_BLOCK, &set, NULL))
                    427:                err(1, "sigprocmask");
                    428:        file_list = NULL;
                    429:        if (clock_gettime(CLOCK_MONOTONIC, &file_ts) < 0) {
                    430:                perror("clock_gettime");
                    431:                exit(1);
                    432:        }
                    433:        log_sync = 0;
                    434:        timo_init();
                    435: }
                    436:
                    437: void
                    438: filelist_done(void)
                    439: {
                    440: #ifdef DEBUG
                    441:        struct file *f;
                    442:
                    443:        if (file_list != NULL) {
                    444:                for (f = file_list; f != NULL; f = f->next) {
                    445:                        file_log(f);
                    446:                        log_puts(" not closed\n");
                    447:                }
                    448:                panic();
                    449:        }
                    450:        log_sync = 1;
                    451:        log_flush();
                    452: #endif
                    453:        timo_done();
                    454: }