|
diff --git a/cache.c b/cache.c index 6847202..a996109 100644 --- a/ cache.c+++ b/ cache.c |
|
@@ -1,209 +1,209 @@ |
1 | /* cache.c: cache management |
1 | /* cache.c: cache management |
2 | * |
2 | * |
3 | * Copyright (C) 2006 Lars Hjemli |
3 | * Copyright (C) 2006 Lars Hjemli |
4 | * |
4 | * |
5 | * Licensed under GNU General Public License v2 |
5 | * Licensed under GNU General Public License v2 |
6 | * (see COPYING for full license text) |
6 | * (see COPYING for full license text) |
7 | * |
7 | * |
8 | * |
8 | * |
9 | * The cache is just a directory structure where each file is a cache slot, |
9 | * The cache is just a directory structure where each file is a cache slot, |
10 | * and each filename is based on the hash of some key (e.g. the cgit url). |
10 | * and each filename is based on the hash of some key (e.g. the cgit url). |
11 | * Each file contains the full key followed by the cached content for that |
11 | * Each file contains the full key followed by the cached content for that |
12 | * key. |
12 | * key. |
13 | * |
13 | * |
14 | */ |
14 | */ |
15 | |
15 | |
16 | #include "cgit.h" |
16 | #include "cgit.h" |
17 | #include "cache.h" |
17 | #include "cache.h" |
18 | |
18 | |
19 | #define CACHE_BUFSIZE (1024 * 4) |
19 | #define CACHE_BUFSIZE (1024 * 4) |
20 | |
20 | |
21 | struct cache_slot { |
21 | struct cache_slot { |
22 | const char *key; |
22 | const char *key; |
23 | int keylen; |
23 | int keylen; |
24 | int ttl; |
24 | int ttl; |
25 | cache_fill_fn fn; |
25 | cache_fill_fn fn; |
26 | void *cbdata; |
26 | void *cbdata; |
27 | int cache_fd; |
27 | int cache_fd; |
28 | int lock_fd; |
28 | int lock_fd; |
29 | const char *cache_name; |
29 | const char *cache_name; |
30 | const char *lock_name; |
30 | const char *lock_name; |
31 | int match; |
31 | int match; |
32 | struct stat cache_st; |
32 | struct stat cache_st; |
33 | struct stat lock_st; |
33 | struct stat lock_st; |
34 | int bufsize; |
34 | int bufsize; |
35 | char buf[CACHE_BUFSIZE]; |
35 | char buf[CACHE_BUFSIZE]; |
36 | }; |
36 | }; |
37 | |
37 | |
38 | /* Open an existing cache slot and fill the cache buffer with |
38 | /* Open an existing cache slot and fill the cache buffer with |
39 | * (part of) the content of the cache file. Return 0 on success |
39 | * (part of) the content of the cache file. Return 0 on success |
40 | * and errno otherwise. |
40 | * and errno otherwise. |
41 | */ |
41 | */ |
42 | static int open_slot(struct cache_slot *slot) |
42 | static int open_slot(struct cache_slot *slot) |
43 | { |
43 | { |
44 | char *bufz; |
44 | char *bufz; |
45 | int bufkeylen = -1; |
45 | int bufkeylen = -1; |
46 | |
46 | |
47 | slot->cache_fd = open(slot->cache_name, O_RDONLY); |
47 | slot->cache_fd = open(slot->cache_name, O_RDONLY); |
48 | if (slot->cache_fd == -1) |
48 | if (slot->cache_fd == -1) |
49 | return errno; |
49 | return errno; |
50 | |
50 | |
51 | if (fstat(slot->cache_fd, &slot->cache_st)) |
51 | if (fstat(slot->cache_fd, &slot->cache_st)) |
52 | return errno; |
52 | return errno; |
53 | |
53 | |
54 | slot->bufsize = read(slot->cache_fd, slot->buf, sizeof(slot->buf)); |
54 | slot->bufsize = xread(slot->cache_fd, slot->buf, sizeof(slot->buf)); |
55 | if (slot->bufsize < 0) |
55 | if (slot->bufsize < 0) |
56 | return errno; |
56 | return errno; |
57 | |
57 | |
58 | bufz = memchr(slot->buf, 0, slot->bufsize); |
58 | bufz = memchr(slot->buf, 0, slot->bufsize); |
59 | if (bufz) |
59 | if (bufz) |
60 | bufkeylen = bufz - slot->buf; |
60 | bufkeylen = bufz - slot->buf; |
61 | |
61 | |
62 | slot->match = bufkeylen == slot->keylen && |
62 | slot->match = bufkeylen == slot->keylen && |
63 | !memcmp(slot->key, slot->buf, bufkeylen + 1); |
63 | !memcmp(slot->key, slot->buf, bufkeylen + 1); |
64 | |
64 | |
65 | return 0; |
65 | return 0; |
66 | } |
66 | } |
67 | |
67 | |
68 | /* Close the active cache slot */ |
68 | /* Close the active cache slot */ |
69 | static int close_slot(struct cache_slot *slot) |
69 | static int close_slot(struct cache_slot *slot) |
70 | { |
70 | { |
71 | int err = 0; |
71 | int err = 0; |
72 | if (slot->cache_fd > 0) { |
72 | if (slot->cache_fd > 0) { |
73 | if (close(slot->cache_fd)) |
73 | if (close(slot->cache_fd)) |
74 | err = errno; |
74 | err = errno; |
75 | else |
75 | else |
76 | slot->cache_fd = -1; |
76 | slot->cache_fd = -1; |
77 | } |
77 | } |
78 | return err; |
78 | return err; |
79 | } |
79 | } |
80 | |
80 | |
81 | /* Print the content of the active cache slot (but skip the key). */ |
81 | /* Print the content of the active cache slot (but skip the key). */ |
82 | static int print_slot(struct cache_slot *slot) |
82 | static int print_slot(struct cache_slot *slot) |
83 | { |
83 | { |
84 | ssize_t i, j = 0; |
84 | ssize_t i; |
85 | |
85 | |
86 | i = lseek(slot->cache_fd, slot->keylen + 1, SEEK_SET); |
86 | i = lseek(slot->cache_fd, slot->keylen + 1, SEEK_SET); |
87 | if (i != slot->keylen + 1) |
87 | if (i != slot->keylen + 1) |
88 | return errno; |
88 | return errno; |
89 | |
89 | |
90 | while((i=read(slot->cache_fd, slot->buf, sizeof(slot->buf))) > 0) |
90 | while((i = xread(slot->cache_fd, slot->buf, sizeof(slot->buf))) > 0) |
91 | j = write(STDOUT_FILENO, slot->buf, i); |
91 | i = xwrite(STDOUT_FILENO, slot->buf, i); |
92 | |
92 | |
93 | if (j < 0) |
93 | if (i < 0) |
94 | return errno; |
94 | return errno; |
95 | else |
95 | else |
96 | return 0; |
96 | return 0; |
97 | } |
97 | } |
98 | |
98 | |
99 | /* Check if the slot has expired */ |
99 | /* Check if the slot has expired */ |
100 | static int is_expired(struct cache_slot *slot) |
100 | static int is_expired(struct cache_slot *slot) |
101 | { |
101 | { |
102 | if (slot->ttl < 0) |
102 | if (slot->ttl < 0) |
103 | return 0; |
103 | return 0; |
104 | else |
104 | else |
105 | return slot->cache_st.st_mtime + slot->ttl*60 < time(NULL); |
105 | return slot->cache_st.st_mtime + slot->ttl*60 < time(NULL); |
106 | } |
106 | } |
107 | |
107 | |
108 | /* Check if the slot has been modified since we opened it. |
108 | /* Check if the slot has been modified since we opened it. |
109 | * NB: If stat() fails, we pretend the file is modified. |
109 | * NB: If stat() fails, we pretend the file is modified. |
110 | */ |
110 | */ |
111 | static int is_modified(struct cache_slot *slot) |
111 | static int is_modified(struct cache_slot *slot) |
112 | { |
112 | { |
113 | struct stat st; |
113 | struct stat st; |
114 | |
114 | |
115 | if (stat(slot->cache_name, &st)) |
115 | if (stat(slot->cache_name, &st)) |
116 | return 1; |
116 | return 1; |
117 | return (st.st_ino != slot->cache_st.st_ino || |
117 | return (st.st_ino != slot->cache_st.st_ino || |
118 | st.st_mtime != slot->cache_st.st_mtime || |
118 | st.st_mtime != slot->cache_st.st_mtime || |
119 | st.st_size != slot->cache_st.st_size); |
119 | st.st_size != slot->cache_st.st_size); |
120 | } |
120 | } |
121 | |
121 | |
122 | /* Close an open lockfile */ |
122 | /* Close an open lockfile */ |
123 | static int close_lock(struct cache_slot *slot) |
123 | static int close_lock(struct cache_slot *slot) |
124 | { |
124 | { |
125 | int err = 0; |
125 | int err = 0; |
126 | if (slot->lock_fd > 0) { |
126 | if (slot->lock_fd > 0) { |
127 | if (close(slot->lock_fd)) |
127 | if (close(slot->lock_fd)) |
128 | err = errno; |
128 | err = errno; |
129 | else |
129 | else |
130 | slot->lock_fd = -1; |
130 | slot->lock_fd = -1; |
131 | } |
131 | } |
132 | return err; |
132 | return err; |
133 | } |
133 | } |
134 | |
134 | |
135 | /* Create a lockfile used to store the generated content for a cache |
135 | /* Create a lockfile used to store the generated content for a cache |
136 | * slot, and write the slot key + \0 into it. |
136 | * slot, and write the slot key + \0 into it. |
137 | * Returns 0 on success and errno otherwise. |
137 | * Returns 0 on success and errno otherwise. |
138 | */ |
138 | */ |
139 | static int lock_slot(struct cache_slot *slot) |
139 | static int lock_slot(struct cache_slot *slot) |
140 | { |
140 | { |
141 | slot->lock_fd = open(slot->lock_name, O_RDWR|O_CREAT|O_EXCL, |
141 | slot->lock_fd = open(slot->lock_name, O_RDWR|O_CREAT|O_EXCL, |
142 | S_IRUSR|S_IWUSR); |
142 | S_IRUSR|S_IWUSR); |
143 | if (slot->lock_fd == -1) |
143 | if (slot->lock_fd == -1) |
144 | return errno; |
144 | return errno; |
145 | if (write(slot->lock_fd, slot->key, slot->keylen + 1) < 0) |
145 | if (xwrite(slot->lock_fd, slot->key, slot->keylen + 1) < 0) |
146 | return errno; |
146 | return errno; |
147 | return 0; |
147 | return 0; |
148 | } |
148 | } |
149 | |
149 | |
150 | /* Release the current lockfile. If `replace_old_slot` is set the |
150 | /* Release the current lockfile. If `replace_old_slot` is set the |
151 | * lockfile replaces the old cache slot, otherwise the lockfile is |
151 | * lockfile replaces the old cache slot, otherwise the lockfile is |
152 | * just deleted. |
152 | * just deleted. |
153 | */ |
153 | */ |
154 | static int unlock_slot(struct cache_slot *slot, int replace_old_slot) |
154 | static int unlock_slot(struct cache_slot *slot, int replace_old_slot) |
155 | { |
155 | { |
156 | int err; |
156 | int err; |
157 | |
157 | |
158 | if (replace_old_slot) |
158 | if (replace_old_slot) |
159 | err = rename(slot->lock_name, slot->cache_name); |
159 | err = rename(slot->lock_name, slot->cache_name); |
160 | else |
160 | else |
161 | err = unlink(slot->lock_name); |
161 | err = unlink(slot->lock_name); |
162 | |
162 | |
163 | if (err) |
163 | if (err) |
164 | return errno; |
164 | return errno; |
165 | |
165 | |
166 | return 0; |
166 | return 0; |
167 | } |
167 | } |
168 | |
168 | |
169 | /* Generate the content for the current cache slot by redirecting |
169 | /* Generate the content for the current cache slot by redirecting |
170 | * stdout to the lock-fd and invoking the callback function |
170 | * stdout to the lock-fd and invoking the callback function |
171 | */ |
171 | */ |
172 | static int fill_slot(struct cache_slot *slot) |
172 | static int fill_slot(struct cache_slot *slot) |
173 | { |
173 | { |
174 | int tmp; |
174 | int tmp; |
175 | |
175 | |
176 | /* Preserve stdout */ |
176 | /* Preserve stdout */ |
177 | tmp = dup(STDOUT_FILENO); |
177 | tmp = dup(STDOUT_FILENO); |
178 | if (tmp == -1) |
178 | if (tmp == -1) |
179 | return errno; |
179 | return errno; |
180 | |
180 | |
181 | /* Redirect stdout to lockfile */ |
181 | /* Redirect stdout to lockfile */ |
182 | if (dup2(slot->lock_fd, STDOUT_FILENO) == -1) |
182 | if (dup2(slot->lock_fd, STDOUT_FILENO) == -1) |
183 | return errno; |
183 | return errno; |
184 | |
184 | |
185 | /* Generate cache content */ |
185 | /* Generate cache content */ |
186 | slot->fn(slot->cbdata); |
186 | slot->fn(slot->cbdata); |
187 | |
187 | |
188 | /* Restore stdout */ |
188 | /* Restore stdout */ |
189 | if (dup2(tmp, STDOUT_FILENO) == -1) |
189 | if (dup2(tmp, STDOUT_FILENO) == -1) |
190 | return errno; |
190 | return errno; |
191 | |
191 | |
192 | /* Close the temporary filedescriptor */ |
192 | /* Close the temporary filedescriptor */ |
193 | if (close(tmp)) |
193 | if (close(tmp)) |
194 | return errno; |
194 | return errno; |
195 | |
195 | |
196 | return 0; |
196 | return 0; |
197 | } |
197 | } |
198 | |
198 | |
199 | /* Crude implementation of 32-bit FNV-1 hash algorithm, |
199 | /* Crude implementation of 32-bit FNV-1 hash algorithm, |
200 | * see http://www.isthe.com/chongo/tech/comp/fnv/ for details |
200 | * see http://www.isthe.com/chongo/tech/comp/fnv/ for details |
201 | * about the magic numbers. |
201 | * about the magic numbers. |
202 | */ |
202 | */ |
203 | #define FNV_OFFSET 0x811c9dc5 |
203 | #define FNV_OFFSET 0x811c9dc5 |
204 | #define FNV_PRIME 0x01000193 |
204 | #define FNV_PRIME 0x01000193 |
205 | |
205 | |
206 | unsigned long hash_str(const char *str) |
206 | unsigned long hash_str(const char *str) |
207 | { |
207 | { |
208 | unsigned long h = FNV_OFFSET; |
208 | unsigned long h = FNV_OFFSET; |
209 | unsigned char *s = (unsigned char *)str; |
209 | unsigned char *s = (unsigned char *)str; |
|