Annotation of libaitsched/example/test.c, revision 1.14.30.3
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.14.30.3! 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.14.30.1 misho 143: case SIGHUP:
144: printf("Test SIGHUP\n");
145: break;
1.5 misho 146: }
147: }
148:
1.14.30.1 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.14.30.3! misho 168: struct timespec p = { 0, 10000000 };
1.14.30.2 misho 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.14.30.2 misho 176: /*
1.5 misho 177: sa.sa_handler = sig;
178: sigemptyset(&sa.sa_mask);
179: sigaction(SIGTERM, &sa, NULL);
1.14.30.2 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.14.30.1 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:
1.14.30.3! misho 322: schedPolling(root, &p, NULL);
1.14.30.1 misho 323: schedSignalDispatch(root, 42);
1.14 misho 324: schedRun(root, Kill);
1.14.30.1 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>