|
diff --git a/cgit.c b/cgit.c index 2c933dc..e6b4526 100644 --- a/ cgit.c+++ b/ cgit.c |
|
@@ -166,152 +166,151 @@ static void cgit_print_repo_page(struct cacheitem *item) |
166 | cgit_print_commit(ctx.qry.sha1); |
166 | cgit_print_commit(ctx.qry.sha1); |
167 | break; |
167 | break; |
168 | case CMD_REFS: |
168 | case CMD_REFS: |
169 | cgit_print_refs(); |
169 | cgit_print_refs(); |
170 | break; |
170 | break; |
171 | case CMD_TAG: |
171 | case CMD_TAG: |
172 | cgit_print_tag(ctx.qry.sha1); |
172 | cgit_print_tag(ctx.qry.sha1); |
173 | break; |
173 | break; |
174 | case CMD_DIFF: |
174 | case CMD_DIFF: |
175 | cgit_print_diff(ctx.qry.sha1, ctx.qry.sha2, ctx.qry.path); |
175 | cgit_print_diff(ctx.qry.sha1, ctx.qry.sha2, ctx.qry.path); |
176 | break; |
176 | break; |
177 | default: |
177 | default: |
178 | cgit_print_error("Invalid request"); |
178 | cgit_print_error("Invalid request"); |
179 | } |
179 | } |
180 | cgit_print_docend(); |
180 | cgit_print_docend(); |
181 | } |
181 | } |
182 | |
182 | |
183 | static void cgit_fill_cache(struct cacheitem *item, int use_cache) |
183 | static void cgit_fill_cache(struct cacheitem *item, int use_cache) |
184 | { |
184 | { |
185 | int stdout2; |
185 | int stdout2; |
186 | |
186 | |
187 | item->st.st_mtime = time(NULL); |
187 | item->st.st_mtime = time(NULL); |
188 | |
188 | |
189 | if (use_cache) { |
189 | if (use_cache) { |
190 | stdout2 = chk_positive(dup(STDOUT_FILENO), |
190 | stdout2 = chk_positive(dup(STDOUT_FILENO), |
191 | "Preserving STDOUT"); |
191 | "Preserving STDOUT"); |
192 | chk_zero(close(STDOUT_FILENO), "Closing STDOUT"); |
192 | chk_zero(close(STDOUT_FILENO), "Closing STDOUT"); |
193 | chk_positive(dup2(item->fd, STDOUT_FILENO), "Dup2(cachefile)"); |
193 | chk_positive(dup2(item->fd, STDOUT_FILENO), "Dup2(cachefile)"); |
194 | } |
194 | } |
195 | |
195 | |
196 | if (ctx.repo) |
196 | if (ctx.repo) |
197 | cgit_print_repo_page(item); |
197 | cgit_print_repo_page(item); |
198 | else |
198 | else |
199 | cgit_print_repolist(item); |
199 | cgit_print_repolist(item); |
200 | |
200 | |
201 | if (use_cache) { |
201 | if (use_cache) { |
202 | chk_zero(close(STDOUT_FILENO), "Close redirected STDOUT"); |
202 | chk_zero(close(STDOUT_FILENO), "Close redirected STDOUT"); |
203 | chk_positive(dup2(stdout2, STDOUT_FILENO), |
203 | chk_positive(dup2(stdout2, STDOUT_FILENO), |
204 | "Restoring original STDOUT"); |
204 | "Restoring original STDOUT"); |
205 | chk_zero(close(stdout2), "Closing temporary STDOUT"); |
205 | chk_zero(close(stdout2), "Closing temporary STDOUT"); |
206 | } |
206 | } |
207 | } |
207 | } |
208 | |
208 | |
209 | static void cgit_check_cache(struct cacheitem *item) |
209 | static void cgit_check_cache(struct cacheitem *item) |
210 | { |
210 | { |
211 | int i = 0; |
211 | int i = 0; |
212 | |
212 | |
213 | top: |
213 | top: |
214 | if (++i > ctx.cfg.max_lock_attempts) { |
214 | if (++i > ctx.cfg.max_lock_attempts) { |
215 | die("cgit_refresh_cache: unable to lock %s: %s", |
215 | die("cgit_refresh_cache: unable to lock %s: %s", |
216 | item->name, strerror(errno)); |
216 | item->name, strerror(errno)); |
217 | } |
217 | } |
218 | if (!cache_exist(item)) { |
218 | if (!cache_exist(item)) { |
219 | if (!cache_lock(item)) { |
219 | if (!cache_lock(item)) { |
220 | sleep(1); |
220 | sleep(1); |
221 | goto top; |
221 | goto top; |
222 | } |
222 | } |
223 | if (!cache_exist(item)) { |
223 | if (!cache_exist(item)) { |
224 | cgit_fill_cache(item, 1); |
224 | cgit_fill_cache(item, 1); |
225 | cache_unlock(item); |
225 | cache_unlock(item); |
226 | } else { |
226 | } else { |
227 | cache_cancel_lock(item); |
227 | cache_cancel_lock(item); |
228 | } |
228 | } |
229 | } else if (cache_expired(item) && cache_lock(item)) { |
229 | } else if (cache_expired(item) && cache_lock(item)) { |
230 | if (cache_expired(item)) { |
230 | if (cache_expired(item)) { |
231 | cgit_fill_cache(item, 1); |
231 | cgit_fill_cache(item, 1); |
232 | cache_unlock(item); |
232 | cache_unlock(item); |
233 | } else { |
233 | } else { |
234 | cache_cancel_lock(item); |
234 | cache_cancel_lock(item); |
235 | } |
235 | } |
236 | } |
236 | } |
237 | } |
237 | } |
238 | |
238 | |
239 | static void cgit_print_cache(struct cacheitem *item) |
239 | static void cgit_print_cache(struct cacheitem *item) |
240 | { |
240 | { |
241 | static char buf[4096]; |
241 | static char buf[4096]; |
242 | ssize_t i; |
242 | ssize_t i; |
243 | |
243 | |
244 | int fd = open(item->name, O_RDONLY); |
244 | int fd = open(item->name, O_RDONLY); |
245 | if (fd<0) |
245 | if (fd<0) |
246 | die("Unable to open cached file %s", item->name); |
246 | die("Unable to open cached file %s", item->name); |
247 | |
247 | |
248 | while((i=read(fd, buf, sizeof(buf))) > 0) |
248 | while((i=read(fd, buf, sizeof(buf))) > 0) |
249 | write(STDOUT_FILENO, buf, i); |
249 | write(STDOUT_FILENO, buf, i); |
250 | |
250 | |
251 | close(fd); |
251 | close(fd); |
252 | } |
252 | } |
253 | |
253 | |
254 | static void cgit_parse_args(int argc, const char **argv) |
254 | static void cgit_parse_args(int argc, const char **argv) |
255 | { |
255 | { |
256 | int i; |
256 | int i; |
257 | |
257 | |
258 | for (i = 1; i < argc; i++) { |
258 | for (i = 1; i < argc; i++) { |
259 | if (!strncmp(argv[i], "--cache=", 8)) { |
259 | if (!strncmp(argv[i], "--cache=", 8)) { |
260 | ctx.cfg.cache_root = xstrdup(argv[i]+8); |
260 | ctx.cfg.cache_root = xstrdup(argv[i]+8); |
261 | } |
261 | } |
262 | if (!strcmp(argv[i], "--nocache")) { |
262 | if (!strcmp(argv[i], "--nocache")) { |
263 | ctx.cfg.nocache = 1; |
263 | ctx.cfg.nocache = 1; |
264 | } |
264 | } |
265 | if (!strncmp(argv[i], "--query=", 8)) { |
265 | if (!strncmp(argv[i], "--query=", 8)) { |
266 | ctx.qry.raw = xstrdup(argv[i]+8); |
266 | ctx.qry.raw = xstrdup(argv[i]+8); |
267 | } |
267 | } |
268 | if (!strncmp(argv[i], "--repo=", 7)) { |
268 | if (!strncmp(argv[i], "--repo=", 7)) { |
269 | ctx.qry.repo = xstrdup(argv[i]+7); |
269 | ctx.qry.repo = xstrdup(argv[i]+7); |
270 | } |
270 | } |
271 | if (!strncmp(argv[i], "--page=", 7)) { |
271 | if (!strncmp(argv[i], "--page=", 7)) { |
272 | ctx.qry.page = xstrdup(argv[i]+7); |
272 | ctx.qry.page = xstrdup(argv[i]+7); |
273 | } |
273 | } |
274 | if (!strncmp(argv[i], "--head=", 7)) { |
274 | if (!strncmp(argv[i], "--head=", 7)) { |
275 | ctx.qry.head = xstrdup(argv[i]+7); |
275 | ctx.qry.head = xstrdup(argv[i]+7); |
276 | ctx.qry.has_symref = 1; |
276 | ctx.qry.has_symref = 1; |
277 | } |
277 | } |
278 | if (!strncmp(argv[i], "--sha1=", 7)) { |
278 | if (!strncmp(argv[i], "--sha1=", 7)) { |
279 | ctx.qry.sha1 = xstrdup(argv[i]+7); |
279 | ctx.qry.sha1 = xstrdup(argv[i]+7); |
280 | ctx.qry.has_sha1 = 1; |
280 | ctx.qry.has_sha1 = 1; |
281 | } |
281 | } |
282 | if (!strncmp(argv[i], "--ofs=", 6)) { |
282 | if (!strncmp(argv[i], "--ofs=", 6)) { |
283 | ctx.qry.ofs = atoi(argv[i]+6); |
283 | ctx.qry.ofs = atoi(argv[i]+6); |
284 | } |
284 | } |
285 | } |
285 | } |
286 | } |
286 | } |
287 | |
287 | |
288 | int main(int argc, const char **argv) |
288 | int main(int argc, const char **argv) |
289 | { |
289 | { |
290 | struct cacheitem item; |
290 | struct cacheitem item; |
291 | const char *cgit_config_env = getenv("CGIT_CONFIG"); |
291 | const char *cgit_config_env = getenv("CGIT_CONFIG"); |
292 | |
292 | |
293 | cgit_prepare_context(&ctx); |
293 | cgit_prepare_context(&ctx); |
294 | htmlfd = STDOUT_FILENO; |
| |
295 | item.st.st_mtime = time(NULL); |
294 | item.st.st_mtime = time(NULL); |
296 | cgit_repolist.length = 0; |
295 | cgit_repolist.length = 0; |
297 | cgit_repolist.count = 0; |
296 | cgit_repolist.count = 0; |
298 | cgit_repolist.repos = NULL; |
297 | cgit_repolist.repos = NULL; |
299 | |
298 | |
300 | cgit_read_config(cgit_config_env ? cgit_config_env : CGIT_CONFIG, |
299 | cgit_read_config(cgit_config_env ? cgit_config_env : CGIT_CONFIG, |
301 | cgit_global_config_cb); |
300 | cgit_global_config_cb); |
302 | if (getenv("SCRIPT_NAME")) |
301 | if (getenv("SCRIPT_NAME")) |
303 | ctx.cfg.script_name = xstrdup(getenv("SCRIPT_NAME")); |
302 | ctx.cfg.script_name = xstrdup(getenv("SCRIPT_NAME")); |
304 | if (getenv("QUERY_STRING")) |
303 | if (getenv("QUERY_STRING")) |
305 | ctx.qry.raw = xstrdup(getenv("QUERY_STRING")); |
304 | ctx.qry.raw = xstrdup(getenv("QUERY_STRING")); |
306 | cgit_parse_args(argc, argv); |
305 | cgit_parse_args(argc, argv); |
307 | cgit_parse_query(ctx.qry.raw, cgit_querystring_cb); |
306 | cgit_parse_query(ctx.qry.raw, cgit_querystring_cb); |
308 | if (!cgit_prepare_cache(&item)) |
307 | if (!cgit_prepare_cache(&item)) |
309 | return 0; |
308 | return 0; |
310 | if (ctx.cfg.nocache) { |
309 | if (ctx.cfg.nocache) { |
311 | cgit_fill_cache(&item, 0); |
310 | cgit_fill_cache(&item, 0); |
312 | } else { |
311 | } else { |
313 | cgit_check_cache(&item); |
312 | cgit_check_cache(&item); |
314 | cgit_print_cache(&item); |
313 | cgit_print_cache(&item); |
315 | } |
314 | } |
316 | return 0; |
315 | return 0; |
317 | } |
316 | } |
|