struct cache_slot {
const char *key;
- int keylen;
+ size_t keylen;
int ttl;
cache_fill_fn fn;
int cache_fd;
int lock_fd;
+ int stdout_fd;
const char *cache_name;
const char *lock_name;
int match;
static int open_slot(struct cache_slot *slot)
{
char *bufz;
- int bufkeylen = -1;
+ ssize_t bufkeylen = -1;
slot->cache_fd = open(slot->cache_name, O_RDONLY);
if (slot->cache_fd == -1)
if (bufz)
bufkeylen = bufz - slot->buf;
- slot->match = bufkeylen == slot->keylen &&
- !memcmp(slot->key, slot->buf, bufkeylen + 1);
+ if (slot->key)
+ slot->match = bufkeylen == slot->keylen &&
+ !memcmp(slot->key, slot->buf, bufkeylen + 1);
return 0;
}
else
err = unlink(slot->lock_name);
+ /* Restore stdout and close the temporary FD. */
+ if (slot->stdout_fd >= 0) {
+ dup2(slot->stdout_fd, STDOUT_FILENO);
+ close(slot->stdout_fd);
+ slot->stdout_fd = -1;
+ }
+
if (err)
return errno;
*/
static int fill_slot(struct cache_slot *slot)
{
- int tmp;
-
/* Preserve stdout */
- tmp = dup(STDOUT_FILENO);
- if (tmp == -1)
+ slot->stdout_fd = dup(STDOUT_FILENO);
+ if (slot->stdout_fd == -1)
return errno;
/* Redirect stdout to lockfile */
/* Generate cache content */
slot->fn();
- /* update stat info */
- if (fstat(slot->lock_fd, &slot->cache_st))
+ /* Make sure any buffered data is flushed to the file */
+ if (fflush(stdout))
return errno;
- /* Restore stdout */
- if (dup2(tmp, STDOUT_FILENO) == -1)
- return errno;
-
- /* Close the temporary filedescriptor */
- if (close(tmp))
+ /* update stat info */
+ if (fstat(slot->lock_fd, &slot->cache_st))
return errno;
return 0;
/* If the cache slot does not exist (or its key doesn't match the
* current key), lets try to create a new cache slot for this
* request. If this fails (for whatever reason), lets just generate
- * the content without caching it and fool the caller to belive
+ * the content without caching it and fool the caller to believe
* everything worked out (but print a warning on stdout).
*/
strbuf_addstr(&lockname, ".lock");
slot.fn = fn;
slot.ttl = ttl;
+ slot.stdout_fd = -1;
slot.cache_name = filename.buf;
slot.lock_name = lockname.buf;
slot.key = key;
static char *sprintftime(const char *format, time_t time)
{
static char buf[64];
- struct tm *tm;
+ struct tm tm;
if (!time)
return NULL;
- tm = gmtime(&time);
- strftime(buf, sizeof(buf)-1, format, tm);
+ gmtime_r(&time, &tm);
+ strftime(buf, sizeof(buf)-1, format, &tm);
return buf;
}