Annotation of libaitsched/example/test.c, revision 1.13.16.2
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:
14: intptr_t Kill;
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.2 misho 40: Kill++;
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:
128: printf("tid (%lx):: %s\n", TASK_VAL(arg), __func__);
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:
139: Kill++;
140: break;
141: }
142: }
143:
1.2 misho 144: int
145: main(int argc, char **argv)
146: {
147: sched_root_task_t *root;
1.7 misho 148: int f, fd;
1.2 misho 149: struct sockaddr_in sin;
1.5 misho 150: struct timespec ts = { 20, 0 };
151: // struct timespec p = { 0, 10000000 };
152: struct sigaction sa;
1.10 misho 153: sched_task_t *t;
1.7 misho 154: #ifdef AIO_SUPPORT
155: char *ole = malloc(BUFSIZ);
156: register int i;
157: #endif
1.5 misho 158:
159: sa.sa_handler = sig;
160: sigemptyset(&sa.sa_mask);
161: sigaction(SIGTERM, &sa, NULL);
1.2 misho 162:
163: f = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
164: if (f == -1)
165: return 1;
1.13.16.1 misho 166: #ifndef __linux__
1.2 misho 167: sin.sin_len = sizeof sin;
1.13.16.1 misho 168: #endif
1.2 misho 169: sin.sin_family = AF_INET;
170: sin.sin_port = htons(2345);
171: sin.sin_addr.s_addr = INADDR_ANY;
172: if (bind(f, (struct sockaddr*) &sin, sizeof sin) == -1)
173: return 1;
174:
1.7 misho 175: fd = open("test_aio.dat", O_CREAT | O_RDWR, 0644);
176: if (fd == -1)
177: return 1;
178: printf("fd=%d\n", fd);
179:
1.2 misho 180: root = schedBegin();
181: if (!root) {
182: printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
183: return 1;
184: }
185:
1.6 misho 186: if (!schedTimer(root, timer, (void*) (intptr_t) ts.tv_sec, ts, NULL, 0)) {
1.5 misho 187: printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
188: return 4;
189: } else
190: ts.tv_sec = 15;
1.6 misho 191: if (!schedTimer(root, timer, (void*) (intptr_t) ts.tv_sec, ts, NULL, 0)) {
1.5 misho 192: printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
193: return 4;
194: } else
195: ts.tv_sec = 10;
196:
1.4 misho 197: if (!schedEvent(root, event, "piuk", 1234, NULL, 0)) {
1.2 misho 198: printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
199: return 2;
200: }
201:
1.8 misho 202: if (!schedTask(root, regular, "piuk", 1111, NULL, 0)) {
1.2 misho 203: printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
204: return 3;
205: }
206:
1.6 misho 207: if (!schedTimer(root, timer, (void*) (intptr_t) ts.tv_sec, ts, NULL, 0)) {
1.2 misho 208: printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
209: return 4;
210: }
211:
1.4 misho 212: if (!schedRead(root, r, "rrr", f, NULL, 0)) {
1.2 misho 213: printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
214: return 5;
215: }
216:
1.4 misho 217: if (!schedWrite(root, w, "www", f, NULL, 0)) {
1.2 misho 218: printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
219: return 6;
220: }
221:
1.13 misho 222: if (!(t = schedThread(root, thr4kill, "0aaaa", 0, NULL, 0))) {
1.10 misho 223: printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
224: return 7;
225: }
1.13 misho 226: if (!schedThread(root, thr, "mdaaaa this is thread task", 8192, NULL, 0)) {
1.9 misho 227: printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
1.13.16.2! misho 228: printf("stack is too small\n");
1.9 misho 229: }
1.13 misho 230: if (!schedThread(root, thr, "mdaaaa this is thread task -detached", 131072, NULL, 0)) {
1.9 misho 231: printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
232: return 7;
233: }
1.13 misho 234: if (!schedThread(root, thr, "mdaaaa this is thread task -j", 0, NULL, 0)) {
1.9 misho 235: printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
236: return 7;
237: }
1.10 misho 238: printf("try to cancel tid = %lx\n", TASK_VAL(t));
239: schedCancel(t);
1.13 misho 240: if (!schedThread(root, thr, "mdaaaa this is thread task -j2", 0, NULL, 0)) {
1.10 misho 241: printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
242: return 7;
243: }
1.13 misho 244: if (!(t = schedThread(root, thr4kill, "0aaaa", 0, NULL, 0))) {
1.10 misho 245: printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
246: return 7;
247: }
1.13 misho 248: if (!schedThread(root, thr, "mdaaaa this is thread task -j3", 4096, NULL, 0)) {
1.10 misho 249: printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError());
1.13.16.2! misho 250: printf("stack is too small\n");
1.10 misho 251: }
252: sleep(1);
253: schedCancel(t);
1.9 misho 254:
1.7 misho 255: #ifdef AIO_SUPPORT
256: memset(ole, 0, BUFSIZ);
257: if (!schedAIORead(root, aioread, (void*) f, fd, ole, BUFSIZ - 1, 0))
258: printf("Warning:: #%d - %s\n", sched_GetErrno(), sched_GetError());
259:
260:
261: iv[0].iov_len = 5;
262: iv[1].iov_len = 2;
263: iv[2].iov_len = 50;
264: for (i = 0; i < 3; i++)
265: iv[i].iov_base = malloc(iv[i].iov_len);
266: if (!schedLIORead(root, aiobulkread, NULL, fd, iv, 3, 0))
267: printf("Warning:: #%d - %s\n", sched_GetErrno(), sched_GetError());
268: fsync(fd);
269: for (i = 0; i < 3; i++) {
270: wiv[i].iov_len = 100;
271: wiv[i].iov_base = malloc(wiv[i].iov_len);
272: }
273: strlcpy(wiv[0].iov_base, "12345678900000000000000000\n", wiv[0].iov_len);
274: wiv[0].iov_len = strlen(wiv[0].iov_base) + 1;
275: strlcpy(wiv[1].iov_base, "222222222222222222222222\n", wiv[1].iov_len);
276: wiv[1].iov_len = strlen(wiv[1].iov_base) + 1;
277: strlcpy(wiv[2].iov_base, "333\n", wiv[2].iov_len);
278: wiv[2].iov_len = strlen(wiv[2].iov_base) + 1;
279: if (!schedLIOWrite(root, aiobulkwrite, NULL, fd, wiv, 3, 0))
280: printf("Warning:: #%d - %s\n", sched_GetErrno(), sched_GetError());
281:
282: for (i = 0; i < 3; i++) {
283: riv[i].iov_len = 5;
284: riv[i].iov_base = malloc(riv[i].iov_len + 1);
285: memset(riv[i].iov_base, 0, riv[i].iov_len + 1);
286: }
287: if (!schedLIORead(root, aiobulkread, NULL, fd, riv, 3, 0))
288: printf("Warning:: #%d - %s\n", sched_GetErrno(), sched_GetError());
289: #endif
290:
1.4 misho 291: schedCallOnce(root, once, "000000", 42, NULL, 0);
1.2 misho 292:
1.5 misho 293: // schedPolling(root, &p, NULL);
1.2 misho 294: schedRun(root, &Kill);
295: schedEnd(&root);
296:
1.7 misho 297: #ifdef AIO_SUPPORT
298: for (i = 0; i < 3; i++)
299: free(iv[i].iov_base);
300: #endif
301:
302: close(fd);
1.2 misho 303: close(f);
304: return 0;
305: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>