Version 0.1.7 from FTP
[asterisk/asterisk.git] / loader.c
index af52f65..b216a07 100755 (executable)
--- a/loader.c
+++ b/loader.c
@@ -3,7 +3,7 @@
  *
  * Module Loader
  * 
- * Copyright (C) 1999, Adtran Inc. and Linux Support Services, LLC
+ * Copyright (C) 1999, Mark Spencer
  *
  * Mark Spencer <markster@linux-support.net>
  *
 #include <asterisk/config.h>
 #include <asterisk/logger.h>
 #include <dlfcn.h>
+#include <asterisk/md5.h>
 #define __USE_GNU
 #include <pthread.h>
 #include "asterisk.h"
 
+static char expected_key[] =
+{ 0x8e, 0x93, 0x22, 0x83, 0xf5, 0xc3, 0xc0, 0x75,
+  0xff, 0x8b, 0xa9, 0xbe, 0x7c, 0x43, 0x74, 0x63 };
+
 struct module {
        int (*load_module)(void);
        int (*unload_module)(void);
        int (*usecount)(void);
        char *(*description)(void);
+       char *(*key)(void);
+       int (*reload)(void);
        void *lib;
        char resource[256];
        struct module *next;
 };
 
+static int printdigest(unsigned char *d)
+{
+       int x;
+       char buf[256];
+       char buf2[16];
+       snprintf(buf, sizeof(buf), "Unexpected signature:");
+       for (x=0;x<16;x++) {
+               snprintf(buf2, sizeof(buf2), " %02x", *(d++));
+               strcat(buf, buf2);
+       }
+       strcat(buf, "\n");
+       ast_log(LOG_DEBUG, buf);
+       return 0;
+}
+
+static int key_matches(char *key1, char *key2)
+{
+       int match = 1;
+       int x;
+       for (x=0;x<16;x++) {
+               match &= (key1[x] == key2[x]);
+       }
+       return match;
+}
+
+static int verify_key(char *key)
+{
+       struct MD5Context c;
+       char digest[16];
+       MD5Init(&c);
+       MD5Update(&c, key, strlen(key));
+       MD5Final(digest, &c);
+       if (key_matches(expected_key, digest))
+               return 0;
+       printdigest(digest);
+       return -1;
+}
+
 static struct loadupdate {
        int (*updater)(void);
        struct loadupdate *next;
@@ -91,9 +136,39 @@ int ast_load_resource(char *resource_name)
        static char fn[256];
        int errors=0;
        int res;
-       struct module *m = malloc(sizeof(struct module));
+       struct module *m;
+       int flags=0;
+       char *val;
+       char *key;
+       int o;
+       struct ast_config *cfg;
+       /* Keep the module file parsing silent */
+       o = option_verbose;
+       option_verbose = 0;
+       cfg = ast_load(AST_MODULE_CONFIG);
+       option_verbose = o;
+       if (cfg) {
+               if ((val = ast_variable_retrieve(cfg, "global", resource_name))
+                               && ast_true(val))
+                       flags |= RTLD_GLOBAL;
+               ast_destroy(cfg);
+       }
+       
+       if (pthread_mutex_lock(&modlock))
+               ast_log(LOG_WARNING, "Failed to lock\n");
+       m = module_list;
+       while(m) {
+               if (!strcasecmp(m->resource, resource_name)) {
+                       ast_log(LOG_WARNING, "Module '%s' already exists\n", resource_name);
+                       pthread_mutex_unlock(&modlock);
+                       return -1;
+               }
+               m = m->next;
+       }
+       m = malloc(sizeof(struct module));      
        if (!m) {
                ast_log(LOG_WARNING, "Out of memory\n");
+               pthread_mutex_unlock(&modlock);
                return -1;
        }
        strncpy(m->resource, resource_name, sizeof(m->resource));
@@ -102,10 +177,11 @@ int ast_load_resource(char *resource_name)
        } else {
                snprintf(fn, sizeof(fn), "%s/%s", AST_MODULE_DIR, resource_name);
        }
-       m->lib = dlopen(fn, RTLD_NOW  | RTLD_GLOBAL);
+       m->lib = dlopen(fn, RTLD_NOW | flags);
        if (!m->lib) {
                ast_log(LOG_WARNING, "%s\n", dlerror());
                free(m);
+               pthread_mutex_unlock(&modlock);
                return -1;
        }
        m->load_module = dlsym(m->lib, "load_module");
@@ -128,29 +204,51 @@ int ast_load_resource(char *resource_name)
                ast_log(LOG_WARNING, "No description in module %s\n", fn);
                errors++;
        }
+       m->key = dlsym(m->lib, "key");
+       if (!m->key) {
+               ast_log(LOG_WARNING, "No key routine in module %s\n", fn);
+               errors++;
+       }
+       m->reload = dlsym(m->lib, "reload");
+       if (m->key && !(key = m->key())) {
+               ast_log(LOG_WARNING, "Key routine returned NULL in module %s\n", fn);
+               errors++;
+       } else
+               key = NULL;
+       if (key && verify_key(key)) {
+               ast_log(LOG_WARNING, "Unexpected key returned by module %s\n", fn);
+               errors++;
+       }
        if (errors) {
                ast_log(LOG_WARNING, "%d error(s) loading module %s, aborted\n", errors, fn);
                dlclose(m->lib);
                free(m);
+               pthread_mutex_unlock(&modlock);
                return -1;
        }
-       if (option_verbose) 
-               ast_verbose( " => (%s)\n", m->description());
-       if ((res = m->load_module())) {
-               ast_log(LOG_WARNING, "%s: load_module failed, returning %d\n", m->resource, fn, res);
-               free(m);
-               return -1;
+       if (!fully_booted) {
+               if (option_verbose) 
+                       ast_verbose( " => (%s)\n", m->description());
+               if (option_console && !option_verbose)
+                       ast_verbose( ".");
+       } else {
+               if (option_verbose)
+                       ast_verbose(VERBOSE_PREFIX_1 "Loaded %s => (%s)\n", fn, m->description());
        }
-       if (pthread_mutex_lock(&modlock))
-               ast_log(LOG_WARNING, "Failed to lock\n");
        m->next = module_list;
+       
        module_list = m;
        pthread_mutex_unlock(&modlock);
+       if ((res = m->load_module())) {
+               ast_log(LOG_WARNING, "%s: load_module failed, returning %d\n", m->resource, res);
+               ast_unload_resource(resource_name, 0);
+               return -1;
+       }
        ast_update_use_count();
        return 0;
 }      
 
-int ast_resource_exists(char *resource)
+static int ast_resource_exists(char *resource)
 {
        struct module *m;
        if (pthread_mutex_lock(&modlock))