Annotation of libaitcfg/src/pq.c, revision 1.5

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.5     ! misho       6: * $Id: pq.c,v 1.4.4.1 2014/01/29 23:46:35 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.5     ! misho      15: Copyright 2004 - 2014
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: static inline struct tagUser *
                     50: _selectPasswd(pwd_root_t * __restrict pwd, u_int uid, const char *csName)
                     51: {
                     52:        struct tagUser fu;
                     53: 
                     54:        if (!pwd)
                     55:                return NULL;
                     56:        else
                     57:                memset(&fu, 0, sizeof fu);
                     58: 
                     59:        if (csName) {
1.4       misho      60:                ait_setlikeVar(&fu.usr_name, string, strlen(csName) + 1, csName);
1.2       misho      61:                return RB_FIND(tagPWD, pwd, &fu);
                     62:        }
                     63: 
                     64:        return (struct tagUser*) cfg_findPasswdBy(pwd, PWD_CRIT_UID, uid);
                     65: }
                     66: 
                     67: /* --------------------------------------------------------------- */
                     68: 
                     69: /*
                     70:  * cfg_findPasswdBy() - Find user by criteria position in list
                     71:  *
                     72:  * @pwd = Password root
                     73:  * @criteria = Search criteria [PWD_CRIT_NAME|PWD_CRIT_UID|PWD_CRIT_GID]
                     74:  * @arg1 = Username | UID | GID
                     75:  * return: NULL not found item or error and !=NULL found item
                     76:  */
                     77: const struct tagUser *
                     78: cfg_findPasswdBy(pwd_root_t * __restrict pwd, int criteria, ...)
                     79: {
                     80:        struct tagUser *u;
                     81:        va_list lst;
                     82:        ait_val_t v;
                     83: 
                     84:        if (!pwd)
                     85:                return NULL;
                     86:        else
                     87:                AIT_INIT_VAL(&v);
                     88: 
                     89:        va_start(lst, criteria);
                     90:        switch (criteria) {
                     91:                case PWD_CRIT_NAME:
                     92:                        AIT_SET_STR(&v, va_arg(lst, char*));
                     93:                        break;
                     94:                case PWD_CRIT_UID:
                     95:                case PWD_CRIT_GID:
                     96:                        AIT_SET_U32(&v, va_arg(lst, u_int));
                     97:                        break;
                     98:                default:
                     99:                        return NULL;
                    100:        }
                    101:        va_end(lst);
                    102: 
                    103:        SLIST_FOREACH(u, pwd, usr_next)
                    104:                switch (criteria) {
                    105:                        case PWD_CRIT_NAME:
1.4       misho     106:                                if (!ait_cmpVar(&u->usr_name, &v)) {
1.2       misho     107:                                        AIT_FREE_VAL(&v);
                    108:                                        return u;
                    109:                                }
                    110:                                break;
                    111:                        case PWD_CRIT_UID:
                    112:                                if ((u_int) AIT_RAW(&u->usr_uid) == AIT_GET_U32(&v)) {
                    113:                                        AIT_FREE_VAL(&v);
                    114:                                        return u;
                    115:                                }
                    116:                                break;
                    117:                        case PWD_CRIT_GID:
                    118:                                if ((u_int) AIT_RAW(&u->usr_gid) == AIT_GET_U32(&v)) {
                    119:                                        AIT_FREE_VAL(&v);
                    120:                                        return u;
                    121:                                }
                    122:                                break;
                    123:                }
                    124: 
                    125:        AIT_FREE_VAL(&v);
                    126:        return NULL;
                    127: }
                    128: 
                    129: /*
                    130:  * cfg_unsetPasswd() - Unset item from passwords and free resources
                    131:  *
                    132:  * @pwd = Password root
                    133:  * @criteria = Search criteria [PWD_CRIT_NAME|PWD_CRIT_UID]
                    134:  * @arg1 = Username | UID
                    135:  * return: 0 item not found, -1 error or 1 removed item
                    136:  */
                    137: int
                    138: cfg_unsetPasswd(pwd_root_t * __restrict pwd, int criteria, ...)
                    139: {
                    140:        struct tagUser *u, *u2;
                    141:        va_list lst;
                    142:        u_int n = 0;
                    143:        char *name = NULL;
                    144: 
                    145:        if (!pwd)
                    146:                return -1;
                    147: 
                    148:        va_start(lst, criteria);
                    149:        switch (criteria) {
                    150:                case PWD_CRIT_UID:
                    151:                        n = va_arg(lst, u_int);
                    152:                        break;
                    153:                case PWD_CRIT_NAME:
                    154:                        name = va_arg(lst, char*);
                    155:                        if (name)
                    156:                                break;
                    157:                default:
                    158:                        va_end(lst);
                    159:                        return -1;
                    160:        }
                    161:        va_end(lst);
                    162: 
                    163:        u = _selectPasswd(pwd, n, name);
                    164:        if (!u)
                    165:                return 0;
                    166: 
                    167:        PWD_LOCK(pwd);
                    168:        RB_REMOVE(tagPWD, pwd, u);
                    169:        SLIST_REMOVE(pwd, u, tagUser, usr_next);
                    170: 
                    171:        /* if duplicates exists, then update r/b tree */
                    172:        SLIST_FOREACH(u2, pwd, usr_next)
                    173:                if (!AIT_ISEMPTY(&u2->usr_name) && 
                    174:                                !strcmp(AIT_GET_STR(&u->usr_name), AIT_GET_STR(&u2->usr_name))) {
                    175:                        RB_INSERT(tagPWD, pwd, u2);
                    176:                        break;
                    177:                }
                    178:        PWD_UNLOCK(pwd);
                    179: 
                    180:        AIT_FREE_VAL(&u->usr_name);
                    181:        AIT_FREE_VAL(&u->usr_pass);
                    182:        AIT_FREE_VAL(&u->usr_uid);
                    183:        AIT_FREE_VAL(&u->usr_gid);
                    184:        AIT_FREE_VAL(&u->usr_class);
                    185:        AIT_FREE_VAL(&u->usr_change);
                    186:        AIT_FREE_VAL(&u->usr_expire);
                    187:        AIT_FREE_VAL(&u->usr_realm);
                    188:        AIT_FREE_VAL(&u->usr_home);
                    189:        AIT_FREE_VAL(&u->usr_shell);
1.4       misho     190:        e_free(u);
1.2       misho     191:        return 1;
                    192: }
                    193: 
                    194: /*
                    195:  * cfg_setPasswd() - Set item in password or adding new item if not exists
                    196:  *
                    197:  * @cfg = Password root
1.3       misho     198:  * @fields = Following parameters are continuous to certain field
1.2       misho     199:  * @csName = Username
                    200:  * @arg1 = Password
                    201:  * @arg2 = UID
                    202:  * @arg3 = GID
                    203:  * @arg4 = Login class
1.3       misho     204:  * @arg5 = Change date
1.2       misho     205:  * @arg6 = Expire date
                    206:  * @arg7 = Realm
                    207:  * @arg8 = Home dir
                    208:  * @arg9 = Shell
                    209:  * return: 0 nothing changed, -1 error, 1 found and updated item or 2 added new item
                    210:  */
                    211: int
1.3       misho     212: cfg_setPasswd(pwd_root_t * __restrict pwd, passwd_attr_t fields, const char *csName, ...) 
1.2       misho     213: {
                    214:        struct tagUser *u;
                    215:        register int i;
                    216:        va_list lst;
                    217: 
                    218:        if (!pwd || !csName)
                    219:                return -1;
                    220: 
                    221:        u = _selectPasswd(pwd, 0, csName);
                    222:        if (!u) {
                    223:                /* adding new element */
1.4       misho     224:                u = e_malloc(sizeof(struct tagUser));
1.2       misho     225:                if (!u) {
1.4       misho     226:                        cfg_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
1.2       misho     227:                        return -1;
                    228:                } else {
                    229:                        memset(u, 0, sizeof(struct tagUser));
1.3       misho     230:                        if (fields >= Username && fields <= Shell)
                    231:                                u->usr_fields = (int) fields;
1.2       misho     232:                        else
                    233:                                u->usr_fields = PWD_MAX_FIELDS - 1;
                    234: 
                    235:                        PWD_LOCK(pwd);
                    236:                        SLIST_INSERT_HEAD(pwd, u, usr_next);
                    237:                        PWD_UNLOCK(pwd);
                    238:                }
                    239: 
                    240:                va_start(lst, csName);
                    241:                for (i = 0; i < (u->usr_fields + 1); i++)
                    242:                        switch (i) {
                    243:                                case 0:
                    244:                                        AIT_SET_STR(&u->usr_name, csName);
                    245:                                        break;
                    246:                                case 1:
                    247:                                        AIT_SET_STR(&u->usr_pass, va_arg(lst, char*));
                    248:                                        break;
                    249:                                case 2:
                    250:                                        AIT_SET_U32(&u->usr_uid, va_arg(lst, u_int));
                    251:                                        break;
                    252:                                case 3:
                    253:                                        AIT_SET_U32(&u->usr_gid, va_arg(lst, u_int));
                    254:                                        break;
                    255:                                case 4:
                    256:                                        AIT_SET_STR(&u->usr_class, va_arg(lst, char*));
                    257:                                        break;
                    258:                                case 5:
                    259:                                        AIT_SET_U32(&u->usr_change, va_arg(lst, u_int));
                    260:                                        break;
                    261:                                case 6:
                    262:                                        AIT_SET_U32(&u->usr_expire, va_arg(lst, u_int));
                    263:                                        break;
                    264:                                case 7:
                    265:                                        AIT_SET_STR(&u->usr_realm, va_arg(lst, char*));
                    266:                                        break;
                    267:                                case 8:
                    268:                                        AIT_SET_STR(&u->usr_home, va_arg(lst, char*));
                    269:                                        break;
                    270:                                case 9:
                    271:                                        AIT_SET_STR(&u->usr_shell, va_arg(lst, char*));
                    272:                                        break;
                    273:                        }
                    274:                va_end(lst);
                    275: 
                    276:                AIT_KEY(&u->usr_name) = crcFletcher16(AIT_GET_LIKE(&u->usr_name, u_short*), 
1.4       misho     277:                                E_ALIGN(AIT_LEN(&u->usr_name) - 1, 2) / 2);
1.2       misho     278: 
                    279:                PWD_LOCK(pwd);
                    280:                RB_INSERT(tagPWD, pwd, u);
                    281:                PWD_UNLOCK(pwd);
                    282:                return 2;
                    283:        } else {
                    284:                AIT_FREE_VAL(&u->usr_pass);
                    285:                AIT_FREE_VAL(&u->usr_uid);
                    286:                AIT_FREE_VAL(&u->usr_gid);
                    287:                AIT_FREE_VAL(&u->usr_class);
                    288:                AIT_FREE_VAL(&u->usr_change);
                    289:                AIT_FREE_VAL(&u->usr_expire);
                    290:                AIT_FREE_VAL(&u->usr_realm);
                    291:                AIT_FREE_VAL(&u->usr_home);
                    292:                AIT_FREE_VAL(&u->usr_shell);
                    293: 
                    294:                /* Update element */
                    295:                va_start(lst, csName);
                    296:                for (i = 1; i < (u->usr_fields + 1); i++)
                    297:                        switch (i) {
                    298:                                case 1:
                    299:                                        AIT_SET_STR(&u->usr_pass, va_arg(lst, char*));
                    300:                                        break;
                    301:                                case 2:
                    302:                                        AIT_SET_U32(&u->usr_uid, va_arg(lst, u_int));
                    303:                                        break;
                    304:                                case 3:
                    305:                                        AIT_SET_U32(&u->usr_gid, va_arg(lst, u_int));
                    306:                                        break;
                    307:                                case 4:
                    308:                                        AIT_SET_STR(&u->usr_class, va_arg(lst, char*));
                    309:                                        break;
                    310:                                case 5:
                    311:                                        AIT_SET_U32(&u->usr_change, va_arg(lst, u_int));
                    312:                                        break;
                    313:                                case 6:
                    314:                                        AIT_SET_U32(&u->usr_expire, va_arg(lst, u_int));
                    315:                                        break;
                    316:                                case 7:
                    317:                                        AIT_SET_STR(&u->usr_realm, va_arg(lst, char*));
                    318:                                        break;
                    319:                                case 8:
                    320:                                        AIT_SET_STR(&u->usr_home, va_arg(lst, char*));
                    321:                                        break;
                    322:                                case 9:
                    323:                                        AIT_SET_STR(&u->usr_shell, va_arg(lst, char*));
                    324:                                        break;
                    325:                        }
                    326:                va_end(lst);
                    327:                return 1;
                    328:        }
                    329: 
                    330:        /* Nothing happens ... found & values is equal! */
                    331:        return 0;
                    332: }
                    333: 
                    334: /*
                    335:  * cfg_getPasswd() - Get item from passwords and return structure from it
                    336:  *
                    337:  * @pwd = Password root
                    338:  * @criteria = Search criteria [PWD_CRIT_NAME|PWD_CRIT_UID]
                    339:  * @arg1 = Username | UID
                    340:  * return: NULL item not found, !=NULL structure found
                    341:  */
1.4       misho     342: const struct tagUser *
1.2       misho     343: cfg_getPasswd(pwd_root_t * __restrict pwd, int criteria, ...)
                    344: {
                    345:        struct tagUser *u;
                    346:        va_list lst;
                    347:        char *str;
                    348: 
                    349:        if (!pwd)
                    350:                return NULL;
                    351: 
                    352:        va_start(lst, criteria);
                    353:        switch (criteria) {
                    354:                case PWD_CRIT_NAME:
                    355:                        str = va_arg(lst, char*);
                    356:                        if (!str)
                    357:                                u = NULL;
                    358:                        else
                    359:                                u = _selectPasswd(pwd, 0, str);
                    360:                        break;
                    361:                case PWD_CRIT_UID:
                    362:                        u = _selectPasswd(pwd, va_arg(lst, u_int), NULL);
                    363:                        break;
                    364:                default:
                    365:                        u = NULL;
                    366:                        break;
                    367:        }
                    368:        va_end(lst);
                    369: 
                    370:        return u;
                    371: }

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