Annotation of libaitsched/example/test.c, revision 1.15

1.2       misho       1: #include <stdio.h>
1.7       misho       2: #include <stdlib.h>
                      3: #include <string.h>
1.2       misho       4: #include <unistd.h>
                      5: #include <fcntl.h>
1.5       misho       6: #include <signal.h>
1.2       misho       7: #include <sys/types.h>
                      8: #include <sys/stat.h>
                      9: #include <sys/socket.h>
                     10: #include <netinet/in.h>
1.7       misho      11: #include "../inc/config.h"
1.2       misho      12: #include <aitsched.h>
                     13: 
1.14      misho      14: intptr_t Kill[1];
1.7       misho      15: #ifdef AIO_SUPPORT
                     16: struct iovec iv[3], wiv[3], riv[3];
                     17: #endif
1.2       misho      18: 
                     19: void *event(sched_task_t *arg)
                     20: {
                     21:        printf("Event::\n");
1.10      misho      22:        taskExit(arg, NULL);
1.2       misho      23: }
                     24: 
1.8       misho      25: void *regular(sched_task_t *arg)
1.2       misho      26: {
1.8       misho      27:        printf("Task::\n");
1.10      misho      28:        taskExit(arg, NULL);
1.2       misho      29: }
                     30: 
                     31: void *timer(sched_task_t *arg)
                     32: {
1.6       misho      33:        printf("Timer %p sec::\n", TASK_ARG(arg));
1.10      misho      34:        taskExit(arg, NULL);
1.2       misho      35: }
                     36: 
                     37: void *r(sched_task_t *arg)
                     38: {
1.12      misho      39:        printf("read:: %ld bytes wait\n", (long) TASK_RET(arg));
1.14      misho      40:        Kill[0]++;
1.10      misho      41:        taskExit(arg, NULL);
1.2       misho      42: }
                     43: 
                     44: void *w(sched_task_t *arg)
                     45: {
1.12      misho      46:        printf("write:: %ld bytes wait\n", (long) TASK_RET(arg));
1.10      misho      47:        taskExit(arg, NULL);
1.2       misho      48: }
                     49: 
                     50: void *once(sched_task_t *arg)
                     51: {
                     52:        printf("once::\n");
1.10      misho      53:        taskExit(arg, NULL);
1.2       misho      54: }
                     55: 
1.7       misho      56: #ifdef AIO_SUPPORT
                     57: void *aioread(sched_task_t *arg);
                     58: void *aiowrite(sched_task_t *arg)
                     59: {
                     60:        char *ole = malloc(BUFSIZ);
                     61: 
                     62:        printf("AIO write[%d]:: %d bytes\n%p\n", TASK_FD(arg), (int) TASK_DATLEN(arg), 
                     63:                        TASK_DATA(arg));
                     64:        free(TASK_DATA(arg));
                     65: 
                     66:        memset(ole, 0, BUFSIZ);
                     67:        schedAIORead(TASK_ROOT(arg), aioread, NULL, TASK_FD(arg), ole, BUFSIZ - 1, -1);
1.10      misho      68:        taskExit(arg, NULL);
1.7       misho      69: }
                     70: 
                     71: void *aioread(sched_task_t *arg)
                     72: {
                     73:        char *ole = malloc(BUFSIZ);
                     74:        int len;
                     75: 
                     76:        printf("AIO read[%d]:: %d bytes\n%s\n-------\n", TASK_FD(arg), (int) TASK_DATLEN(arg), 
                     77:                        (char*) TASK_DATA(arg));
                     78: 
                     79:        if (TASK_ARG(arg)) {
                     80:                len = strlcpy(ole, "++++++BAHURA OR CULTURE .... A CULTURE OR BAHURA :-)\n", BUFSIZ);
                     81:                printf("sched Write len=%d %p\n", len, ole);
                     82:                schedAIOWrite(TASK_ROOT(arg), aiowrite, TASK_ARG(arg), TASK_FD(arg), ole,  len, -1);
                     83:                                
                     84:        }
                     85:        free(TASK_DATA(arg));
1.10      misho      86:        taskExit(arg, NULL);
1.7       misho      87: }
                     88: 
                     89: void *aiobulkread(sched_task_t *arg)
                     90: {
                     91:        struct iovec *iv = TASK_DATA(arg);
                     92:        register int i;
                     93: 
                     94:        printf("aioBULKread::\n");
                     95:        for (i = 0; i < 3; i++) {
                     96:                printf("%d) rlen[%d]=%s\n---\n", i, iv[i].iov_len, (char*) iv[i].iov_base);
                     97:                free(iv[i].iov_base);
                     98:        }
                     99: 
1.10      misho     100:        taskExit(arg, NULL);
1.7       misho     101: }
                    102: 
                    103: void *aiobulkwrite(sched_task_t *arg)
                    104: {
                    105:        struct iovec *iv = TASK_DATA(arg);
                    106:        register int i;
                    107: 
                    108:        printf("aioBULKwrite::\n");
                    109:        for (i = 0; i < 3; i++) {
                    110:                printf("%d) wlen=%d\n", i, iv[i].iov_len);
                    111:                free(iv[i].iov_base);
                    112:        }
                    113: 
1.10      misho     114:        taskExit(arg, NULL);
1.7       misho     115: }
                    116: #endif
                    117: 
1.9       misho     118: void *thr(sched_task_t *arg)
                    119: {
1.10      misho     120:        printf("tid (%lx):: %s\n", TASK_VAL(arg), __func__);
1.9       misho     121:        taskExit(arg, 42);
                    122: }
                    123: 
1.10      misho     124: void *thr4kill(sched_task_t *arg)
                    125: {
                    126:        char blah[BUFSIZ];
                    127: 
1.14      misho     128:        printf("THREAD!!! tid (%lx):: %s\n", TASK_VAL(arg), __func__);
1.10      misho     129: 
                    130:        read(0, blah, sizeof blah);
                    131:        printf("never see!!! (%lx):: %s (%d == %d)\n", TASK_VAL(arg), (char*) TASK_ARG(arg), TASK_TYPE(arg), taskTHREAD);
                    132:        taskExit(arg, 0);
                    133: }
                    134: 
1.5       misho     135: void sig(int s)
                    136: {
                    137:        switch (s) {
                    138:                case SIGTERM:
1.15    ! misho     139:                case SIGINT:
        !           140:                        printf("I'm in switch case %d\n", s);
1.14      misho     141:                        Kill[0]++;
1.5       misho     142:                        break;
1.15    ! misho     143:                case SIGHUP:
        !           144:                        printf("Test SIGHUP\n");
        !           145:                        break;
1.5       misho     146:        }
                    147: }
                    148: 
1.15    ! misho     149: void *sigt(sched_task_t *arg)
        !           150: {
        !           151:        int s = TASK_VAL(arg);
        !           152: 
        !           153:        printf("Received signal #%d\n", s);
        !           154: 
        !           155:        sig(s);
        !           156: 
        !           157:        schedSignalSelf(arg);
        !           158:        taskExit(arg, NULL);
        !           159: }
        !           160: 
1.2       misho     161: int
                    162: main(int argc, char **argv)
                    163: {
                    164:        sched_root_task_t *root;
1.7       misho     165:        int f, fd;
1.2       misho     166:        struct sockaddr_in sin;
1.5       misho     167:        struct timespec ts = { 20, 0 };
1.15    ! misho     168:        struct timespec p = { 0, 10000000 };
        !           169: //     struct sigaction sa;
1.10      misho     170:        sched_task_t *t;
1.7       misho     171: #ifdef AIO_SUPPORT
                    172:        char *ole = malloc(BUFSIZ);
                    173:        register int i;
                    174: #endif
1.5       misho     175: 
1.15    ! misho     176:        /*
1.5       misho     177:        sa.sa_handler = sig;
                    178:        sigemptyset(&sa.sa_mask);
                    179:        sigaction(SIGTERM, &sa, NULL);
1.15    ! misho     180:        */
1.2       misho     181: 
                    182:        f = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
                    183:        if (f == -1)
                    184:                return 1;
1.14      misho     185: #ifndef __linux__
1.2       misho     186:        sin.sin_len = sizeof sin;
1.14      misho     187: #endif
1.2       misho     188:        sin.sin_family = AF_INET;
                    189:        sin.sin_port = htons(2345);
                    190:        sin.sin_addr.s_addr = INADDR_ANY;
                    191:        if (bind(f, (struct sockaddr*) &sin, sizeof sin) == -1)
                    192:                return 1;
                    193: 
1.7       misho     194:        fd = open("test_aio.dat", O_CREAT | O_RDWR, 0644);
                    195:        if (fd == -1)
                    196:                return 1;
                    197:        printf("fd=%d\n", fd);
                    198: 
1.2       misho     199:        root = schedBegin();
                    200:        if (!root) {
                    201:                printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
                    202:                return 1;
                    203:        }
                    204: 
1.6       misho     205:        if (!schedTimer(root, timer, (void*) (intptr_t) ts.tv_sec, ts, NULL, 0)) {
1.5       misho     206:                printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
                    207:                return 4;
                    208:        } else
                    209:                ts.tv_sec = 15;
1.6       misho     210:        if (!schedTimer(root, timer, (void*) (intptr_t) ts.tv_sec, ts, NULL, 0)) {
1.5       misho     211:                printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
                    212:                return 4;
                    213:        } else
                    214:                ts.tv_sec = 10;
                    215: 
1.4       misho     216:        if (!schedEvent(root, event, "piuk", 1234, NULL, 0)) {
1.2       misho     217:                printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
                    218:                return 2;
                    219:        }
                    220: 
1.8       misho     221:        if (!schedTask(root, regular, "piuk", 1111, NULL, 0)) {
1.2       misho     222:                printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
                    223:                return 3;
                    224:        }
                    225: 
1.6       misho     226:        if (!schedTimer(root, timer, (void*) (intptr_t) ts.tv_sec, ts, NULL, 0)) {
1.2       misho     227:                printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
                    228:                return 4;
                    229:        }
                    230: 
1.14      misho     231:        if (!(t = schedRead(root, r, "rrr_test", f, NULL, 0))) {
                    232:                printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
                    233:                return 5;
                    234:        }
1.4       misho     235:        if (!schedRead(root, r, "rrr", f, NULL, 0)) {
1.2       misho     236:                printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
                    237:                return 5;
                    238:        }
1.14      misho     239:        schedCancel(t);
1.2       misho     240: 
1.4       misho     241:        if (!schedWrite(root, w, "www", f, NULL, 0)) {
1.2       misho     242:                printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
                    243:                return 6;
                    244:        }
                    245: 
1.13      misho     246:        if (!(t = schedThread(root, thr4kill, "0aaaa", 0, NULL, 0))) {
1.10      misho     247:                printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
                    248:                return 7;
                    249:        }
1.13      misho     250:        if (!schedThread(root, thr, "mdaaaa this is thread task", 8192, NULL, 0)) {
1.9       misho     251:                printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
1.14      misho     252:                printf("stack is too small\n");
1.9       misho     253:        }
1.13      misho     254:        if (!schedThread(root, thr, "mdaaaa this is thread task -detached", 131072, NULL, 0)) {
1.9       misho     255:                printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
                    256:                return 7;
                    257:        }
1.13      misho     258:        if (!schedThread(root, thr, "mdaaaa this is thread task -j", 0, NULL, 0)) {
1.9       misho     259:                printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
                    260:                return 7;
                    261:        }
1.14      misho     262:        printf("~~~try to cancel tid = %lx\n", TASK_VAL(t));
1.10      misho     263:        schedCancel(t);
1.13      misho     264:        if (!schedThread(root, thr, "mdaaaa this is thread task -j2", 0, NULL, 0)) {
1.10      misho     265:                printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
                    266:                return 7;
                    267:        }
1.13      misho     268:        if (!(t = schedThread(root, thr4kill, "0aaaa", 0, NULL, 0))) {
1.10      misho     269:                printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
                    270:                return 7;
                    271:        }
1.13      misho     272:        if (!schedThread(root, thr, "mdaaaa this is thread task -j3", 4096, NULL, 0)) {
1.10      misho     273:                printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
1.14      misho     274:                printf("stack is too small\n");
1.10      misho     275:        }
                    276:        sleep(1);
                    277:        schedCancel(t);
1.9       misho     278: 
1.7       misho     279: #ifdef AIO_SUPPORT
                    280:        memset(ole, 0, BUFSIZ);
                    281:        if (!schedAIORead(root, aioread, (void*) f, fd, ole, BUFSIZ - 1, 0))
                    282:                printf("Warning:: #%d - %s\n", sched_GetErrno(), sched_GetError());
                    283: 
                    284: 
                    285:        iv[0].iov_len = 5;
                    286:        iv[1].iov_len = 2;
                    287:        iv[2].iov_len = 50;
                    288:        for (i = 0; i < 3; i++)
                    289:                iv[i].iov_base = malloc(iv[i].iov_len);
                    290:        if (!schedLIORead(root, aiobulkread, NULL, fd, iv, 3, 0))
                    291:                printf("Warning:: #%d - %s\n", sched_GetErrno(), sched_GetError());
                    292:        fsync(fd);
                    293:        for (i = 0; i < 3; i++) {
                    294:                wiv[i].iov_len = 100;
                    295:                wiv[i].iov_base = malloc(wiv[i].iov_len);
                    296:        }
                    297:        strlcpy(wiv[0].iov_base, "12345678900000000000000000\n", wiv[0].iov_len);
                    298:        wiv[0].iov_len = strlen(wiv[0].iov_base) + 1;
                    299:        strlcpy(wiv[1].iov_base, "222222222222222222222222\n", wiv[1].iov_len);
                    300:        wiv[1].iov_len = strlen(wiv[1].iov_base) + 1;
                    301:        strlcpy(wiv[2].iov_base, "333\n", wiv[2].iov_len);
                    302:        wiv[2].iov_len = strlen(wiv[2].iov_base) + 1;
                    303:        if (!schedLIOWrite(root, aiobulkwrite, NULL, fd, wiv, 3, 0))
                    304:                printf("Warning:: #%d - %s\n", sched_GetErrno(), sched_GetError());
                    305: 
                    306:        for (i = 0; i < 3; i++) {
                    307:                riv[i].iov_len = 5;
                    308:                riv[i].iov_base = malloc(riv[i].iov_len + 1);
                    309:                memset(riv[i].iov_base, 0, riv[i].iov_len + 1);
                    310:        }
                    311:        if (!schedLIORead(root, aiobulkread, NULL, fd, riv, 3, 0))
                    312:                printf("Warning:: #%d - %s\n", sched_GetErrno(), sched_GetError());
                    313: #endif
                    314: 
1.4       misho     315:        schedCallOnce(root, once, "000000", 42, NULL, 0);
1.2       misho     316: 
1.15    ! misho     317:        printf("add signals\n");
        !           318:        schedSignal(root, sigt, NULL, SIGHUP, NULL, 0);
        !           319:        schedSignal(root, sigt, NULL, SIGTERM, NULL, 0);
        !           320:        schedSignal(root, sigt, NULL, SIGINT, NULL, 0);
        !           321: 
        !           322:        schedPolling(root, &p, NULL);
        !           323:        schedSignalDispatch(root, 42);
1.14      misho     324:        schedRun(root, Kill);
1.15    ! misho     325:        schedSignalDispatch(root, 0);
1.2       misho     326:        schedEnd(&root);
1.14      misho     327:        sleep(1);
1.2       misho     328: 
1.7       misho     329: #ifdef AIO_SUPPORT
                    330:        for (i = 0; i < 3; i++)
                    331:                free(iv[i].iov_base);
                    332: #endif
                    333: 
                    334:        close(fd);
1.2       misho     335:        close(f);
                    336:        return 0;
                    337: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>