Files: f87af6a0bf824e4980c8e1b25a7a1307183bbe3e / ccdl.c
4704 bytesRaw
1 | |
2 | |
3 | |
4 | |
5 | |
6 | |
7 | |
8 | |
9 | |
10 | |
11 | |
12 | |
13 | static void lib_init(struct lib *, const char *src_fname, const char *sym); |
14 | static int lib_build(struct lib *); |
15 | static void *lib_get(struct lib *); |
16 | static void lib_deinit(struct lib *); |
17 | |
18 | static int ccdl_rebuild(struct ccdl *); |
19 | static struct lib *ccdl_get_old_lib(struct ccdl *); |
20 | static int ccdl_watch_run(struct ccdl *); |
21 | |
22 | static const char *my_mktemp() |
23 | { |
24 | const char *tmpdir = getenv("TMPDIR"); |
25 | if (!tmpdir) tmpdir = "/tmp"; |
26 | static char buf[256]; |
27 | ssize_t len = snprintf(buf, sizeof(buf), "%s/ccdl-XXXXXX", tmpdir); |
28 | if (len < 0) return NULL; |
29 | if (mkstemp(buf) < 0) return NULL; |
30 | return buf; |
31 | } |
32 | |
33 | static void lib_init(struct lib *lib, const char *src_fname, const char *sym_name) |
34 | { |
35 | lib->src_fname = src_fname; |
36 | lib->sym_name = sym_name; |
37 | lib->handle = NULL; |
38 | lib->sym = NULL; |
39 | } |
40 | |
41 | static int compile(const char *src, const char *lib) |
42 | { |
43 | if (!src || !lib) return -1; |
44 | int child = fork(); |
45 | if (child < 0) return -1; |
46 | if (child == 0) { |
47 | char *cc = getenv("CC"); |
48 | if (!cc) cc = "cc"; |
49 | char *const argv[] = { |
50 | cc, (char *)src, "-o", (char *)lib, |
51 | "-fPIC", "-shared", "-lm", |
52 | NULL}; |
53 | if (execvp(argv[0], argv)) err(1, "execvp"); |
54 | exit(0); |
55 | } else { |
56 | int wstatus; |
57 | if (waitpid(child, &wstatus, 0) < 0) { |
58 | return -1; |
59 | } else if (WIFEXITED(wstatus) && WEXITSTATUS(wstatus) == 0) { |
60 | return 0; |
61 | } else { |
62 | return -1; |
63 | } |
64 | } |
65 | } |
66 | |
67 | static int lib_build(struct lib *lib) |
68 | { |
69 | void *handle = lib->handle; |
70 | if (!handle) { |
71 | const char *src_fname = lib->src_fname; |
72 | if (!src_fname) return -1; |
73 | const char *lib_fname = my_mktemp(); |
74 | if (!lib_fname) { |
75 | warn("mktemp"); |
76 | return -1; |
77 | } |
78 | if (compile(src_fname, lib_fname)) return -1; |
79 | dlerror(); |
80 | handle = dlopen(lib_fname, RTLD_NOW); |
81 | char *err = dlerror(); |
82 | if (unlink(lib_fname) < 0) warn("unlink"); |
83 | if (err) { |
84 | warnx("dlopen: %s", err); |
85 | return -1; |
86 | } |
87 | lib->handle = handle; |
88 | } |
89 | const char *sym_name = lib->sym_name; |
90 | if (!sym_name) return -1; |
91 | void *sym = dlsym(handle, sym_name); |
92 | if (!sym) return -1; |
93 | lib->sym = sym; |
94 | return 0; |
95 | } |
96 | |
97 | static void *lib_get(struct lib *lib) |
98 | { |
99 | if (!lib->sym) lib_build(lib); |
100 | return lib->sym; |
101 | } |
102 | |
103 | static void lib_deinit(struct lib *lib) |
104 | { |
105 | if (lib->handle) { |
106 | dlclose(lib->handle); |
107 | lib->handle = NULL; |
108 | lib->sym = NULL; |
109 | } |
110 | } |
111 | |
112 | void ccdl_init(struct ccdl *ccdl, const char *src_fname, const char *sym_name) |
113 | { |
114 | lib_init(&ccdl->libs[0], src_fname, sym_name); |
115 | lib_init(&ccdl->libs[1], src_fname, sym_name); |
116 | ccdl->lib = &ccdl->libs[0]; |
117 | } |
118 | |
119 | static struct lib *ccdl_get_old_lib(struct ccdl *ccdl) |
120 | { |
121 | return ccdl->lib == &ccdl->libs[1] ? &ccdl->libs[0] : &ccdl->libs[1]; |
122 | } |
123 | |
124 | void *ccdl_get(struct ccdl *ccdl) |
125 | { |
126 | struct lib *old_lib = ccdl_get_old_lib(ccdl); |
127 | if (old_lib) { |
128 | lib_deinit(old_lib); |
129 | } |
130 | return lib_get(ccdl->lib); |
131 | } |
132 | |
133 | static int ccdl_watch_run(struct ccdl *ccdl) |
134 | { |
135 | int ifd = inotify_init(); |
136 | int wd = 0; |
137 | if (ifd < 0) { |
138 | warn("inotify_init"); |
139 | return -1; |
140 | } |
141 | if (!ccdl || !ccdl->lib || !ccdl->lib->src_fname) { |
142 | warn("missing lib filename"); |
143 | return -1; |
144 | } |
145 | wd = inotify_add_watch(ifd, ccdl->lib->src_fname, IN_MODIFY); |
146 | if (wd < 0) { |
147 | warn("watch"); |
148 | return -1; |
149 | } |
150 | |
151 | while (1) { |
152 | struct inotify_event ev; |
153 | ssize_t len = read(ifd, &ev, sizeof ev); |
154 | //printf("readed %zd %u %u %u\n", len, sizeof ev, ev.len, ev.mask); |
155 | if (len < 0) { |
156 | warn("read"); |
157 | return -1; |
158 | } |
159 | if ((size_t)len < sizeof ev) { |
160 | warn("small read"); |
161 | continue; |
162 | } |
163 | if (ev.len > 0) { |
164 | warn("unexpected name"); |
165 | return -1; |
166 | } |
167 | if (ev.mask & IN_MODIFY) { |
168 | } else if (ev.mask & IN_DELETE_SELF) { |
169 | printf("deleted\n"); |
170 | } else if (ev.mask & IN_IGNORED) { |
171 | printf("ignored\n"); |
172 | // file is being replaced. replace the watch |
173 | wd = inotify_add_watch(ifd, ccdl->lib->src_fname, IN_MODIFY); |
174 | if (wd < 0) { |
175 | warn("inotify_add_watch"); |
176 | return -1; |
177 | } |
178 | } else { |
179 | printf("unknown mask: %u\n", ev.mask); |
180 | } |
181 | (void)ccdl_rebuild(ccdl); |
182 | } |
183 | |
184 | return 0; |
185 | } |
186 | |
187 | static void *ccdl_watch_run_thread(void *ccdl) |
188 | { |
189 | return (void *)(intptr_t)ccdl_watch_run(ccdl); |
190 | } |
191 | |
192 | int ccdl_watch(struct ccdl *ccdl) |
193 | { |
194 | pthread_t *id = &ccdl->watcher_thread; |
195 | if (pthread_create(id, NULL, ccdl_watch_run_thread, ccdl)) return -1; |
196 | return 0; |
197 | } |
198 | |
199 | static int ccdl_rebuild(struct ccdl *ccdl) |
200 | { |
201 | struct lib *new_lib = ccdl_get_old_lib(ccdl); |
202 | lib_deinit(new_lib); |
203 | if (lib_build(new_lib) < 0) return -1; |
204 | ccdl->lib = new_lib; |
205 | return 0; |
206 | } |
207 | |
208 | int ccdl_deinit(struct ccdl *ccdl) |
209 | { |
210 | pthread_t thread = ccdl->watcher_thread; |
211 | if (thread) { |
212 | void *retval; |
213 | pthread_join(thread, &retval); |
214 | return (intptr_t)retval; |
215 | } |
216 | return 0; |
217 | } |
218 | |
219 |
Built with git-ssb-web