Lars Hjemli | 7640d90 | 2006-12-10 22:41:14 +0100 | [diff] [blame] | 1 | /* cache.c: cache management |
| 2 | * |
| 3 | * Copyright (C) 2006 Lars Hjemli |
| 4 | * |
| 5 | * Licensed under GNU General Public License v2 |
| 6 | * (see COPYING for full license text) |
| 7 | */ |
| 8 | |
Lars Hjemli | 25105d7 | 2006-12-10 22:31:36 +0100 | [diff] [blame] | 9 | #include "cgit.h" |
| 10 | |
| 11 | const int NOLOCK = -1; |
| 12 | |
Lars Hjemli | 2c2047f | 2007-01-12 00:24:35 +0100 | [diff] [blame] | 13 | char *cache_safe_filename(const char *unsafe) |
| 14 | { |
Lars Hjemli | 72fa5c6 | 2007-05-18 03:54:15 +0200 | [diff] [blame] | 15 | static char buf[4][PATH_MAX]; |
| 16 | static int bufidx; |
| 17 | char *s; |
Lars Hjemli | 2c2047f | 2007-01-12 00:24:35 +0100 | [diff] [blame] | 18 | char c; |
| 19 | |
Lars Hjemli | 72fa5c6 | 2007-05-18 03:54:15 +0200 | [diff] [blame] | 20 | bufidx++; |
| 21 | bufidx &= 3; |
| 22 | s = buf[bufidx]; |
| 23 | |
Lars Hjemli | 2c2047f | 2007-01-12 00:24:35 +0100 | [diff] [blame] | 24 | while(unsafe && (c = *unsafe++) != 0) { |
Lars Hjemli | 72fa5c6 | 2007-05-18 03:54:15 +0200 | [diff] [blame] | 25 | if (c == '/' || c == ' ' || c == '&' || c == '|' || |
Lars Hjemli | 2c2047f | 2007-01-12 00:24:35 +0100 | [diff] [blame] | 26 | c == '>' || c == '<' || c == '.') |
| 27 | c = '_'; |
| 28 | *s++ = c; |
| 29 | } |
| 30 | *s = '\0'; |
Lars Hjemli | 72fa5c6 | 2007-05-18 03:54:15 +0200 | [diff] [blame] | 31 | return buf[bufidx]; |
Lars Hjemli | 2c2047f | 2007-01-12 00:24:35 +0100 | [diff] [blame] | 32 | } |
| 33 | |
Lars Hjemli | f5069d8 | 2006-12-11 09:57:58 +0100 | [diff] [blame] | 34 | int cache_exist(struct cacheitem *item) |
| 35 | { |
Lars Hjemli | 25105d7 | 2006-12-10 22:31:36 +0100 | [diff] [blame] | 36 | if (stat(item->name, &item->st)) { |
| 37 | item->st.st_mtime = 0; |
| 38 | return 0; |
| 39 | } |
| 40 | return 1; |
| 41 | } |
| 42 | |
| 43 | int cache_create_dirs() |
| 44 | { |
| 45 | char *path; |
| 46 | |
Lars Hjemli | b228d4f | 2008-02-16 13:07:13 +0100 | [diff] [blame] | 47 | path = fmt("%s", ctx.cfg.cache_root); |
Lars Hjemli | 7c849d9 | 2006-12-16 13:55:58 +0100 | [diff] [blame] | 48 | if (mkdir(path, S_IRWXU) && errno!=EEXIST) |
| 49 | return 0; |
| 50 | |
Lars Hjemli | d1f3bbe | 2008-02-16 13:56:09 +0100 | [diff] [blame] | 51 | if (!ctx.repo) |
Lars Hjemli | 25105d7 | 2006-12-10 22:31:36 +0100 | [diff] [blame] | 52 | return 0; |
| 53 | |
Lars Hjemli | b228d4f | 2008-02-16 13:07:13 +0100 | [diff] [blame] | 54 | path = fmt("%s/%s", ctx.cfg.cache_root, |
Lars Hjemli | d1f3bbe | 2008-02-16 13:56:09 +0100 | [diff] [blame] | 55 | cache_safe_filename(ctx.repo->url)); |
Lars Hjemli | 30ccdca | 2007-05-18 03:00:54 +0200 | [diff] [blame] | 56 | |
Lars Hjemli | 25105d7 | 2006-12-10 22:31:36 +0100 | [diff] [blame] | 57 | if (mkdir(path, S_IRWXU) && errno!=EEXIST) |
| 58 | return 0; |
| 59 | |
Lars Hjemli | d14d77f | 2008-02-16 11:53:40 +0100 | [diff] [blame] | 60 | if (ctx.qry.page) { |
Lars Hjemli | b228d4f | 2008-02-16 13:07:13 +0100 | [diff] [blame] | 61 | path = fmt("%s/%s/%s", ctx.cfg.cache_root, |
Lars Hjemli | d1f3bbe | 2008-02-16 13:56:09 +0100 | [diff] [blame] | 62 | cache_safe_filename(ctx.repo->url), |
Lars Hjemli | d14d77f | 2008-02-16 11:53:40 +0100 | [diff] [blame] | 63 | ctx.qry.page); |
Lars Hjemli | 25105d7 | 2006-12-10 22:31:36 +0100 | [diff] [blame] | 64 | if (mkdir(path, S_IRWXU) && errno!=EEXIST) |
| 65 | return 0; |
| 66 | } |
| 67 | return 1; |
| 68 | } |
| 69 | |
Lars Hjemli | 318d106 | 2006-12-11 12:10:12 +0100 | [diff] [blame] | 70 | int cache_refill_overdue(const char *lockfile) |
| 71 | { |
| 72 | struct stat st; |
| 73 | |
| 74 | if (stat(lockfile, &st)) |
| 75 | return 0; |
| 76 | else |
Lars Hjemli | b228d4f | 2008-02-16 13:07:13 +0100 | [diff] [blame] | 77 | return (time(NULL) - st.st_mtime > ctx.cfg.cache_max_create_time); |
Lars Hjemli | 318d106 | 2006-12-11 12:10:12 +0100 | [diff] [blame] | 78 | } |
| 79 | |
Lars Hjemli | 25105d7 | 2006-12-10 22:31:36 +0100 | [diff] [blame] | 80 | int cache_lock(struct cacheitem *item) |
| 81 | { |
Lars Hjemli | 318d106 | 2006-12-11 12:10:12 +0100 | [diff] [blame] | 82 | int i = 0; |
Lars Hjemli | 58d04f6 | 2006-12-12 10:16:41 +0100 | [diff] [blame] | 83 | char *lockfile = xstrdup(fmt("%s.lock", item->name)); |
Lars Hjemli | 25105d7 | 2006-12-10 22:31:36 +0100 | [diff] [blame] | 84 | |
Lars Hjemli | 318d106 | 2006-12-11 12:10:12 +0100 | [diff] [blame] | 85 | top: |
Lars Hjemli | b228d4f | 2008-02-16 13:07:13 +0100 | [diff] [blame] | 86 | if (++i > ctx.cfg.max_lock_attempts) |
Lars Hjemli | 318d106 | 2006-12-11 12:10:12 +0100 | [diff] [blame] | 87 | die("cache_lock: unable to lock %s: %s", |
| 88 | item->name, strerror(errno)); |
| 89 | |
| 90 | item->fd = open(lockfile, O_WRONLY|O_CREAT|O_EXCL, S_IRUSR|S_IWUSR); |
| 91 | |
Lars Hjemli | 25105d7 | 2006-12-10 22:31:36 +0100 | [diff] [blame] | 92 | if (item->fd == NOLOCK && errno == ENOENT && cache_create_dirs()) |
| 93 | goto top; |
Lars Hjemli | 318d106 | 2006-12-11 12:10:12 +0100 | [diff] [blame] | 94 | |
| 95 | if (item->fd == NOLOCK && errno == EEXIST && |
| 96 | cache_refill_overdue(lockfile) && !unlink(lockfile)) |
Lars Hjemli | 25105d7 | 2006-12-10 22:31:36 +0100 | [diff] [blame] | 97 | goto top; |
Lars Hjemli | 318d106 | 2006-12-11 12:10:12 +0100 | [diff] [blame] | 98 | |
Lars Hjemli | 58d04f6 | 2006-12-12 10:16:41 +0100 | [diff] [blame] | 99 | free(lockfile); |
Lars Hjemli | 25105d7 | 2006-12-10 22:31:36 +0100 | [diff] [blame] | 100 | return (item->fd > 0); |
| 101 | } |
| 102 | |
| 103 | int cache_unlock(struct cacheitem *item) |
| 104 | { |
| 105 | close(item->fd); |
| 106 | return (rename(fmt("%s.lock", item->name), item->name) == 0); |
| 107 | } |
| 108 | |
Lars Hjemli | fbaf117 | 2006-12-11 22:53:50 +0100 | [diff] [blame] | 109 | int cache_cancel_lock(struct cacheitem *item) |
| 110 | { |
| 111 | return (unlink(fmt("%s.lock", item->name)) == 0); |
| 112 | } |
| 113 | |
Lars Hjemli | 25105d7 | 2006-12-10 22:31:36 +0100 | [diff] [blame] | 114 | int cache_expired(struct cacheitem *item) |
| 115 | { |
| 116 | if (item->ttl < 0) |
| 117 | return 0; |
| 118 | return item->st.st_mtime + item->ttl * 60 < time(NULL); |
| 119 | } |