| #include <stdio.h> |
| #include <stddef.h> |
| #include "coroutine.h" |
| #include "generator.h" |
| #include "asleep.h" |
| #include "task.h" |
|
| #include <dirent.h> |
| #include <string.h> |
| #include <stdlib.h> |
| #include <time.h> |
|
| #define DEMO_STACK_SIZE (8192*sizeof(void *)) |
|
| void *yield_files(void *param){ |
| bool domore = true; |
| char *path = param; |
| DIR *d; |
| struct dirent *dir; |
| int pathlen = strlen(path); |
| d = opendir(path); |
| if (d) { |
| while (domore && (dir = readdir(d)) != NULL) { |
| int blklen = pathlen + 1 + strlen(dir->d_name) + 1; |
| char *r = malloc(blklen); |
| snprintf(r, blklen, "%s/%s", path, dir->d_name); |
| domore = Generator_Yield(r); |
| if (domore && dir->d_type == DT_DIR) { |
| if (strcmp(dir->d_name, ".") != 0 && strcmp(dir->d_name, "..") != 0) { |
| r = malloc(blklen); |
| snprintf(r, blklen, "%s/%s", path, dir->d_name); |
| domore = yield_files(r); |
| free(r); |
| } |
| } |
| } |
| closedir(d); |
| } |
|
| return (void *)domore; |
| } |
|
| void *gentest(void *param){ |
| Generator gen; |
| Generator_ctor(&gen, DEMO_STACK_SIZE, yield_files, (char *)param); |
| int count = 0; |
| while(Generator_Next(&gen, ¶m)){ |
| printf("%d) %s\n", count, (char *)param); |
| free(param); |
| if (++count>16000) break; |
| } |
| Generator_dtor(&gen); |
|
| return param; |
| } |
|
| int main(int argc, char *argv[]) { |
| (void)argc; |
| (void)argv; |
|
| Coroutine_Run(DEMO_STACK_SIZE, gentest, "..", NULL); |
|
| return 0; |
| } |
|
jonroach