Annotation of libaitrpc/src/lists.c, revision 1.9

1.2       misho       1: /*************************************************************************
                      2: * (C) 2010 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com>
                      3: *  by Michael Pounov <misho@openbsd-bg.org>
                      4: *
                      5: * $Author: misho $
1.9     ! misho       6: * $Id: lists.c,v 1.8.2.2 2012/05/11 08:49:43 misho Exp $
1.2       misho       7: *
                      8: **************************************************************************
                      9: The ELWIX and AITNET software is distributed under the following
                     10: terms:
                     11: 
                     12: All of the documentation and software included in the ELWIX and AITNET
                     13: Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>
                     14: 
1.7       misho      15: Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012
1.2       misho      16:        by Michael Pounov <misho@elwix.org>.  All rights reserved.
                     17: 
                     18: Redistribution and use in source and binary forms, with or without
                     19: modification, are permitted provided that the following conditions
                     20: are met:
                     21: 1. Redistributions of source code must retain the above copyright
                     22:    notice, this list of conditions and the following disclaimer.
                     23: 2. Redistributions in binary form must reproduce the above copyright
                     24:    notice, this list of conditions and the following disclaimer in the
                     25:    documentation and/or other materials provided with the distribution.
                     26: 3. All advertising materials mentioning features or use of this software
                     27:    must display the following acknowledgement:
                     28: This product includes software developed by Michael Pounov <misho@elwix.org>
                     29: ELWIX - Embedded LightWeight unIX and its contributors.
                     30: 4. Neither the name of AITNET nor the names of its contributors
                     31:    may be used to endorse or promote products derived from this software
                     32:    without specific prior written permission.
                     33: 
                     34: THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND
                     35: ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
                     36: IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
                     37: ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
                     38: FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
                     39: DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
                     40: OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
                     41: HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
                     42: LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
                     43: OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
                     44: SUCH DAMAGE.
                     45: */
                     46: #include "global.h"
                     47: 
                     48: 
                     49: /*
1.7       misho      50:  * rpc_srv_allocVars() - Allocate array for call variables
                     51:  *
1.2       misho      52:  * @call = RPC function call
1.6       misho      53:  * @varnum = Number of variables, if ==0 return already allocated variables number
1.4       misho      54:  * return: -1 error, !=-1 return varnum value
1.2       misho      55:  */
1.6       misho      56: static inline int
1.4       misho      57: rpc_srv_allocVars(rpc_func_t * __restrict call, int varnum)
1.2       misho      58: {
1.4       misho      59:        if (!call || varnum < 0) {
1.7       misho      60:                rpc_SetErr(EINVAL, "Invalid parameter can`t allocate variables for RPC call");
1.2       misho      61:                return -1;
1.5       misho      62:        }
                     63: 
                     64:        if (varnum) {
1.6       misho      65:                call->func_vars = io_allocVars(varnum);
1.5       misho      66:                if (!call->func_vars)
                     67:                        return -1;
                     68:        }
                     69: 
                     70:        return io_arraySize(call->func_vars);
1.2       misho      71: }
                     72: 
                     73: /*
1.7       misho      74:  * rpc_srv_getVars() - Get variables array for RPC call
                     75:  *
1.2       misho      76:  * @call = RPC function call
1.4       misho      77:  * @vars = Returned variables array, may be NULL
1.2       misho      78:  * return: -1 error, !=-1 Number of returned variables
                     79:  */
                     80: inline int
1.5       misho      81: rpc_srv_getVars(rpc_func_t * __restrict call, array_t ** __restrict vars)
1.2       misho      82: {
                     83:        if (!call) {
1.7       misho      84:                rpc_SetErr(EINVAL, "Invalid parameter can`t get variables");
1.2       misho      85:                return -1;
                     86:        }
                     87: 
1.4       misho      88:        if (vars)
                     89:                *vars = call->func_vars;
1.5       misho      90:        return io_arraySize(call->func_vars);
1.2       misho      91: }
                     92: 
1.7       misho      93: /* --------------------------------------------------------- */
1.2       misho      94: 
                     95: /*
1.7       misho      96:  * rpc_srv_registerCall() - Register call to RPC server
                     97:  *
1.2       misho      98:  * @srv = RPC Server instance
                     99:  * @csModule = Module name, if NULL self binary
                    100:  * @csFunc = Function name
                    101:  * @args = Number of return function arguments, use for restriction case!
                    102:  * return: -1 error or 0 register ok
                    103:  */
                    104: int
1.7       misho     105: rpc_srv_registerCall(rpc_srv_t * __restrict srv, const char *csModule, 
                    106:                const char *csFunc, u_short args)
1.2       misho     107: {
                    108:        rpc_func_t *func;
                    109: 
                    110:        if (!srv || !csFunc) {
1.7       misho     111:                rpc_SetErr(EINVAL, "Invalid parameter can`t register function");
1.2       misho     112:                return -1;
                    113:        }
                    114:        if (!(func = malloc(sizeof(rpc_func_t)))) {
                    115:                LOGERR;
                    116:                return -1;
1.7       misho     117:        } else
1.2       misho     118:                memset(func, 0, sizeof(rpc_func_t));
1.7       misho     119: 
                    120:        /* calculate hashes */
                    121:        if (rpc_calcHashes(func, csModule, csFunc) == -1) {
                    122:                AIT_FREE_VAL(&func->func_name);
                    123:                AIT_FREE_VAL(&func->func_file);
                    124:                free(func);
                    125:                return -1;
1.2       misho     126:        }
                    127: 
                    128:        func->func_parent = srv;
                    129: 
1.7       misho     130:        /* allocate return variables */
1.4       misho     131:        if (args > 0 && rpc_srv_allocVars(func, args) == -1) {
1.7       misho     132:                AIT_FREE_VAL(&func->func_name);
                    133:                AIT_FREE_VAL(&func->func_file);
1.2       misho     134:                free(func);
                    135:                return -1;
                    136:        }
                    137: 
1.7       misho     138:        /* add to list of functions */
1.2       misho     139:        func->func_next = srv->srv_funcs;
                    140:        srv->srv_funcs = func;
                    141:        return 0;
                    142: }
                    143: 
                    144: /*
1.7       misho     145:  * rpc_srv_unregisterCall() - Unregister call from RPC server
                    146:  *
1.2       misho     147:  * @srv = RPC Server instance
                    148:  * @csModule = Module name, if NULL self binary
                    149:  * @csFunc = Function name
                    150:  * return: -1 error, 0 not found call, 1 unregister ok
                    151:  */
                    152: int
                    153: rpc_srv_unregisterCall(rpc_srv_t * __restrict srv, const char *csModule, const char *csFunc)
                    154: {
                    155:        rpc_func_t func, *f, *curr;
                    156: 
                    157:        if (!srv || !csFunc) {
1.7       misho     158:                rpc_SetErr(EINVAL, "Invalid parameter can`t unregister function");
1.2       misho     159:                return -1;
                    160:        } else
1.7       misho     161:                memset(&func, 0, sizeof func);
1.2       misho     162: 
1.7       misho     163:        /* calculate hashes */
                    164:        if (rpc_calcHashes(&func, csModule, csFunc) == -1)
                    165:                return -1;
1.2       misho     166: 
                    167:        f = rpc_srv_getCall(srv, func.func_tag, func.func_hash);
1.7       misho     168:        AIT_FREE_VAL(&func.func_name);
                    169:        AIT_FREE_VAL(&func.func_file);
1.9     ! misho     170:        if (!f)                 /* not found element for unregister */
1.2       misho     171:                return 0;
                    172: 
1.7       misho     173:        /* remove from list of functions */
1.5       misho     174:        if (srv->srv_funcs == f)        /* if is 1st element */
1.2       misho     175:                srv->srv_funcs = srv->srv_funcs->func_next;
1.5       misho     176:        else {
1.2       misho     177:                for (curr = srv->srv_funcs; curr->func_next != f; curr = curr->func_next);
                    178:                curr->func_next = curr->func_next->func_next;
                    179:        }
1.6       misho     180:        io_freeVars(&f->func_vars);
1.7       misho     181:        AIT_FREE_VAL(&f->func_name);
                    182:        AIT_FREE_VAL(&f->func_file);
1.5       misho     183:        free(f);
1.2       misho     184: 
                    185:        return 1;
                    186: }
                    187: 
                    188: /*
1.7       misho     189:  * rpc_srv_getCall()  - Get registered call from RPC server
                    190:  *
1.2       misho     191:  * @srv = RPC Server instance
                    192:  * @tag = tag for function
                    193:  * @hash = hash for function
                    194:  * return: NULL not found call, !=NULL return call
                    195:  */
                    196: inline rpc_func_t *
                    197: rpc_srv_getCall(rpc_srv_t * __restrict srv, uint16_t tag, uint32_t hash)
                    198: {
                    199:        rpc_func_t *f;
                    200: 
                    201:        if (!srv) {
1.7       misho     202:                rpc_SetErr(EINVAL, "Invalid parameter can`t get function");
1.2       misho     203:                return NULL;
                    204:        }
                    205: 
                    206:        for (f = srv->srv_funcs; f; f = f->func_next)
                    207:                if (f->func_tag == tag && f->func_hash == hash)
                    208:                        break;
                    209: 
                    210:        return f;
                    211: }
                    212: 
                    213: /*
1.7       misho     214:  * rpc_srv_getFunc() - Get registered call from RPC server by Name
                    215:  *
1.2       misho     216:  * @srv = RPC Server instance
                    217:  * @csModule = Module name, if NULL self binary
                    218:  * @csFunc = Function name
                    219:  * return: NULL not found call, !=NULL return call
                    220:  */
                    221: rpc_func_t *
                    222: rpc_srv_getFunc(rpc_srv_t * __restrict srv, const char *csModule, const char *csFunc)
                    223: {
1.7       misho     224:        rpc_func_t func, *f = NULL;
1.2       misho     225: 
                    226:        if (!srv || !csFunc) {
1.7       misho     227:                rpc_SetErr(EINVAL, "Invalid parameter can`t get function");
1.2       misho     228:                return NULL;
                    229:        } else
1.7       misho     230:                memset(&func, 0, sizeof(rpc_func_t));
1.2       misho     231: 
1.7       misho     232:        /* calculate hashes */
                    233:        if (rpc_calcHashes(&func, csModule, csFunc) == -1)
                    234:                return NULL;
1.2       misho     235: 
1.7       misho     236:        f = rpc_srv_getCall(srv, func.func_tag, func.func_hash);
                    237: 
                    238:        AIT_FREE_VAL(&func.func_name);
                    239:        AIT_FREE_VAL(&func.func_file);
                    240:        return f;
1.2       misho     241: }
                    242: 
                    243: // ---------------------------------------------------------
                    244: 
                    245: /*
1.7       misho     246:  * rpc_srv_getBLOB() - Get registered BLOB 
                    247:  *
1.2       misho     248:  * @srv = RPC Server instance
                    249:  * @var = hash for variable
                    250:  * return: NULL not found, !=NULL return blob var
                    251:  */
                    252: inline rpc_blob_t *
                    253: rpc_srv_getBLOB(rpc_srv_t * __restrict srv, uint32_t var)
                    254: {
                    255:        rpc_blob_t *b;
                    256: 
                    257:        if (!srv) {
1.7       misho     258:                rpc_SetErr(EINVAL, "Invalid parameter can`t get BLOB variable");
1.2       misho     259:                return NULL;
                    260:        }
                    261: 
                    262:        for (b = srv->srv_blob.blobs; b; b = b->blob_next) {
                    263:                if (b->blob_var == var)
                    264:                        break;
                    265:        }
                    266: 
                    267:        return b;
                    268: }
                    269: 
                    270: /*
1.7       misho     271:  * rpc_srv_registerBLOB() - Register new BLOB to server
                    272:  *
1.2       misho     273:  * @srv = RPC Server instance
                    274:  * @len = BLOB length
                    275:  * return: NULL error or new registered BLOB
                    276:  */
                    277: rpc_blob_t *
                    278: rpc_srv_registerBLOB(rpc_srv_t * __restrict srv, size_t len)
                    279: {
                    280:        rpc_blob_t *blob = NULL;
                    281: 
                    282:        if (!srv || !len) {
1.7       misho     283:                rpc_SetErr(EINVAL, "Invalid parameter can`t register BLOB variable");
1.2       misho     284:                return blob;
                    285:        }
                    286: 
                    287:        blob = rpc_srv_blobCreate(srv, len);
                    288:        if (blob) {
                    289:                blob->blob_next = srv->srv_blob.blobs;
                    290:                srv->srv_blob.blobs = blob;
                    291:        }
                    292: 
                    293:        return blob;
                    294: }
                    295: 
                    296: /*
1.7       misho     297:  * rpc_srv_unregisterBLOB() - Unregister BLOB from server
                    298:  *
1.2       misho     299:  * @srv = RPC Server instance
                    300:  * @var = BLOB Variable for unregister
                    301:  * return: -1 error, 0 not found call, 1 unregister ok
                    302:  */
                    303: int
                    304: rpc_srv_unregisterBLOB(rpc_srv_t * __restrict srv, uint32_t var)
                    305: {
                    306:        rpc_blob_t *b, *curr;
                    307: 
                    308:        if (!srv) {
1.7       misho     309:                rpc_SetErr(EINVAL, "Invalid parameter can`t unregister BLOB variable");
1.2       misho     310:                return -1;
                    311:        }
                    312: 
                    313:        b = rpc_srv_getBLOB(srv, var);
1.3       misho     314:        if (!b)                         /* not found element for unregister */
1.2       misho     315:                return 0;
1.5       misho     316:        /* if BLOB is unmapped force to unmap object */
                    317:        if (b->blob_data)
                    318:                rpc_srv_blobUnmap(b);
1.2       misho     319: 
1.3       misho     320:        if (srv->srv_blob.blobs == b) { /* if is 1st element */
1.2       misho     321:                srv->srv_blob.blobs = srv->srv_blob.blobs->blob_next;
                    322:        } else {
                    323:                for (curr = srv->srv_blob.blobs; curr->blob_next != b; curr = curr->blob_next);
                    324:                curr->blob_next = curr->blob_next->blob_next;
                    325:        }
                    326:        rpc_srv_blobFree(srv, b);
                    327:        free(b);
                    328: 
                    329:        return 1;
                    330: }
1.7       misho     331: 
                    332: /*
                    333:  * rpc_calcHashes() - Calculate hashes for RPC call
                    334:  *
                    335:  * @func = function
                    336:  * @csModule = Module name, if NULL self binary
                    337:  * @csFunc = Function name
                    338:  * return: -1 error or 0 ok
                    339:  */
                    340: int
                    341: rpc_calcHashes(rpc_func_t * __restrict func, const char *csModule, const char *csFunc)
                    342: {
                    343:        char *str = NULL;
                    344:        int len = 0;
                    345: 
                    346:        assert(func && csFunc);
                    347: 
                    348:        /* set function name */
                    349:        AIT_SET_STR(&func->func_name, csFunc);
                    350:        len = strlen(csFunc) + 3;       /* extra 3 bytes, because add string "__" and 0 */
                    351:        /* set module name if exists */
                    352:        if (csModule) {
                    353:                AIT_SET_STR(&func->func_file, csModule);
                    354:                len += strlen(csModule);
                    355:        }
                    356:        /* align len to 2 */
                    357:        len = io_align(len, 1);
                    358: 
                    359:        /* prepare hash source string */
                    360:        str = malloc(len);
                    361:        if (!str) {
                    362:                LOGERR;
                    363:                return -1;
                    364:        } else {
                    365:                memset(str, 0, len);
                    366:                if (csModule)
                    367:                        strlcpy((char*) str, csModule, len);
                    368:                strlcat((char*) str, "__", len);
                    369:                strlcat((char*) str, csFunc, len);
                    370:        }
                    371: 
                    372:        func->func_tag = crcFletcher16((u_short*) str, len / 2);
                    373:        func->func_hash = hash_fnv((char*) str, len);
                    374: 
                    375:        free(str);
                    376:        return len;
                    377: }
                    378: 

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>