src/server/safs/init.c

/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright 2013 Olaf Wintermann. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *   1. Redistributions of source code must retain the above copyright
 *      notice, this list of conditions and the following disclaimer.
 *
 *   2. Redistributions in binary form must reproduce the above copyright
 *      notice, this list of conditions and the following disclaimer in the
 *      documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#include <dlfcn.h> 
#include <cx/string.h>

#include "../daemon/func.h"
#include "../daemon/log.h"
#include "../daemon/config.h"

#include "init.h"

int init_test(pblock *pb, Session *sn, Request *rq) {
    printf("init-test\n");
    return REQ_PROCEED;
}

int load_modules(pblock *pb, Session *sn, Request *rq) {
    char *shlib = pblock_findval("shlib", pb);
    char *funcs = pblock_findval("funcs", pb);

    if(shlib == NULL || funcs == NULL) {
        log_ereport(LOG_MISCONFIG, "load-modules: missing parameters");
        if(!shlib && funcs) {
            log_ereport(
                    LOG_MISCONFIG,
                    "load-modules: missing shlib parameter");
        } else if(!shlib && !funcs) {
            log_ereport(
                    LOG_MISCONFIG,
                    "load-modules: missing funcs parameter");
        } else {
            log_ereport(
                    LOG_MISCONFIG,
                    "load-modules: missing shlib and funcs parameters");
        }
    }

    // load lib
    void *lib = dlopen(shlib, RTLD_GLOBAL | RTLD_NOW);
    if(lib == NULL) {
        char *dlerr = dlerror();
        fprintf(stderr, "Cannot load library %s %s\n", shlib, dlerr);
        
        log_ereport(LOG_FAILURE, "Cannot load library %s: %s", shlib, dlerr);
        
        return REQ_ABORTED;
    }

    // load function symbols
    pool_handle_t *init_pool = cfg_get_init_pool();
    int b = 0;
    for(int i=0;;i++) {
        if(funcs[i] == '-') {
            funcs[i] = '_';
        }
        if(funcs[i] == ','  || funcs[i] == 0) {
            if(funcs[i] == 0) {
                b = 1;
            }

            // TODO: although this works fine, is not really clean to just
            //       destroy the string with random 0-bytes
            //       maybe use cxstr here
            funcs[i] = 0;

            // we have a function name         
            void *sym = dlsym(lib, funcs);
            if(sym == NULL) {
                fprintf(stderr, "Cannot load symbol %s\n", funcs);
                dlclose(lib);
                return REQ_ABORTED;
            }
            struct FuncStruct fc;
            ZERO(&fc, sizeof(struct FuncStruct));
            fc.func = (FuncPtr)sym;
            fc.name = pool_strdup(init_pool, funcs);
            add_function(&fc);

            if(b) {
                break;
            }

            funcs += i + 1;
            i = 0;
        }
    }
    
    // if exists, execute nsapi_module_init
    void *sym = dlsym(lib, "nsapi_module_init");
    if(sym) {
        /*
         * We remove shlib and funcs from the pblock. The module init function
         * gets all remaining parameters.
         */
        pblock_remove("shlib", pb);
        pblock_remove("funcs", pb);
        
        FuncPtr init_func = (FuncPtr)sym;
        int ret = init_func(pb, NULL, NULL);
        if(ret != REQ_PROCEED && ret != REQ_NOACTION) {
            log_ereport(
                    LOG_FAILURE,
                    "nsapi_module_init for module %s failed",
                    shlib);
            return REQ_ABORTED;
        }
    }
    
    return REQ_PROCEED;
}