src/mod_magnet_cache.c

   1 #include "first.h"
   2
   3 #include "mod_magnet_cache.h"
   4 #include "base.h"
   5 #include "stat_cache.h"
   6
   7 #include <stdlib.h>
   8 #include <time.h>
   9
  10 #include <lualib.h>
  11 #include <lauxlib.h>
  12
  13 static script *script_init() {
  14         script *sc;
  15
  16         sc = calloc(1, sizeof(*sc));
  17         sc->name = buffer_init();
  18         sc->etag = buffer_init();
  19
  20         return sc;
  21 }
  22
  23 static void script_free(script *sc) {
  24         if (!sc) return;
  25
  26         lua_pop(sc->L, 1); /* the function copy */
  27
  28         buffer_free(sc->name);
  29         buffer_free(sc->etag);
  30
  31         lua_close(sc->L);
  32
  33         free(sc);
  34 }
  35
  36 script_cache *script_cache_init() {
  37         script_cache *p;
  38
  39         p = calloc(1, sizeof(*p));
  40
  41         return p;
  42 }
  43
  44 void script_cache_free(script_cache *p) {
  45         size_t i;
  46
  47         if (!p) return;
  48
  49         for (i = 0; i < p->used; i++) {
  50                 script_free(p->ptr[i]);
  51         }
  52
  53         free(p->ptr);
  54
  55         free(p);
  56 }
  57
  58 lua_State *script_cache_get_script(server *srv, connection *con, script_cache *cache, buffer *name) {
  59         size_t i;
  60         script *sc = NULL;
  61         stat_cache_entry *sce;
  62
  63         for (i = 0; i < cache->used; i++) {
  64                 sc = cache->ptr[i];
  65
  66                 if (buffer_is_equal(name, sc->name)) {
  67                         sc->last_used = time(NULL);
  68
  69                         /* oops, the script failed last time */
  70
  71                         if (lua_gettop(sc->L) == 0) break;
  72                         force_assert(lua_gettop(sc->L) == 1);
  73
  74                         if (HANDLER_ERROR == stat_cache_get_entry(srv, con, sc->name, &sce)) {
  75                                 lua_pop(sc->L, 1); /* pop the old function */
  76                                 break;
  77                         }
  78
  79                         stat_cache_etag_get(sce, con->etag_flags);
  80                         if (!buffer_is_equal(sce->etag, sc->etag)) {
  81                                 /* the etag is outdated, reload the function */
  82                                 lua_pop(sc->L, 1);
  83                                 break;
  84                         }
  85
  86                         force_assert(lua_isfunction(sc->L, -1));
  87
  88                         return sc->L;
  89                 }
  90
  91                 sc = NULL;
  92         }
  93
  94         /* if the script was script already loaded but either got changed or
  95          * failed to load last time */
  96         if (sc == NULL) {
  97                 sc = script_init();
  98
  99                 if (cache->used == cache->size) {
 100                         cache->size += 16;
 101                         cache->ptr = realloc(cache->ptr, cache->size * sizeof(*(cache->ptr)));
 102                 }
 103
 104                 cache->ptr[cache->used++] = sc;
 105
 106                 buffer_copy_buffer(sc->name, name);
 107
 108                 sc->L = luaL_newstate();
 109                 luaL_openlibs(sc->L);
 110         }
 111
 112         sc->last_used = time(NULL);
 113
 114         if (0 != luaL_loadfile(sc->L, name->ptr)) {
 115                 /* oops, an error, return it */
 116                 return sc->L;
 117         }
 118
 119         if (HANDLER_GO_ON == stat_cache_get_entry(srv, con, sc->name, &sce)) {
 120                 buffer_copy_buffer(sc->etag, stat_cache_etag_get(sce, con->etag_flags));
 121         }
 122
 123         force_assert(lua_isfunction(sc->L, -1));
 124
 125         return sc->L;
 126 }