/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */
/*
 * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 * Copyright 2021 Joyent, Inc.
 */

#include <secdb.h>
#include <exec_attr.h>
#include "ldap_common.h"


/* exec_attr attributes filters */
#define ISWILD(x)               (x == NULL) ? "*" : x
#define _EXEC_NAME              "cn"
#define _EXEC_POLICY            "SolarisKernelSecurityPolicy"
#define _EXEC_TYPE              "SolarisProfileType"
#define _EXEC_RES1              "SolarisAttrRes1"
#define _EXEC_RES2              "SolarisAttrRes2"
#define _EXEC_ID                "SolarisProfileId"
#define _EXEC_ATTRS             "SolarisAttrKeyValue"
#define _EXEC_GETEXECNAME       "(&(objectClass=SolarisExecAttr)(cn=%s)"\
                                "(SolarisKernelSecurityPolicy=%s)"\
                                "(SolarisProfileType=%s))"
#define _EXEC_GETEXECNAME_SSD   "(&(%%s)(cn=%s)"\
                                "(SolarisKernelSecurityPolicy=%s)"\
                                "(SolarisProfileType=%s))"
#define _EXEC_GETEXECID         "(&(objectClass=SolarisExecAttr)"\
                                "(SolarisProfileId=%s)"\
                                "(SolarisKernelSecurityPolicy=%s)"\
                                "(SolarisProfileType=%s))"
#define _EXEC_GETEXECID_SSD     "(&(%%s)"\
                                "(SolarisProfileId=%s)"\
                                "(SolarisKernelSecurityPolicy=%s)"\
                                "(SolarisProfileType=%s))"
#define _EXEC_GETEXECNAMEID     "(&(objectClass=SolarisExecAttr)(cn=%s)"\
                                "(SolarisProfileId=%s)"\
                                "(SolarisKernelSecurityPolicy=%s)"\
                                "(SolarisProfileType=%s))"
#define _EXEC_GETEXECNAMEID_SSD "(&(%%s)(cn=%s)"\
                                "(SolarisProfileId=%s)"\
                                "(SolarisKernelSecurityPolicy=%s)"\
                                "(SolarisProfileType=%s))"


/* from libnsl */
extern int _doexeclist(nss_XbyY_args_t *);
extern char *_exec_wild_id(char *, const char *);
extern void _exec_cleanup(nss_status_t, nss_XbyY_args_t *);


static const char *exec_attrs[] = {
        _EXEC_NAME,
        _EXEC_POLICY,
        _EXEC_TYPE,
        _EXEC_RES1,
        _EXEC_RES2,
        _EXEC_ID,
        _EXEC_ATTRS,
        (char *)NULL
};


#ifdef  DEBUG
static void
_print_execstr(execstr_t *exec)
{

        (void) fprintf(stdout, "      exec-name: [%s]\n", exec->name);
        if (exec->policy != (char *)NULL) {
                (void) fprintf(stdout, "      policy: [%s]\n", exec->policy);
        }
        if (exec->type != (char *)NULL) {
                (void) fprintf(stdout, "      type: [%s]\n", exec->type);
        }
        if (exec->res1 != (char *)NULL) {
                (void) fprintf(stdout, "      res1: [%s]\n", exec->res1);
        }
        if (exec->res2 != (char *)NULL) {
                (void) fprintf(stdout, "      res2: [%s]\n", exec->res2);
        }
        if (exec->id != (char *)NULL) {
                (void) fprintf(stdout, "      id: [%s]\n", exec->id);
        }
        if (exec->attr != (char *)NULL) {
                (void) fprintf(stdout, "      attr: [%s]\n", exec->attr);
        }
        if (exec->next != (execstr_t *)NULL) {
                (void) fprintf(stdout, "      next: [%s]\n", exec->next->name);
                (void) fprintf(stdout, "\n");
                _print_execstr(exec->next);
        }
}
#endif  /* DEBUG */


static int
_exec_ldap_exec2ent(ns_ldap_entry_t *entry, nss_XbyY_args_t *argp)
{

        int                     i;
        unsigned long           len = 0L;
        int                     buflen = (int)0;
        char                    *nullstring = (char *)NULL;
        char                    *buffer = (char *)NULL;
        char                    *ceiling = (char *)NULL;
        execstr_t               *exec = (execstr_t *)NULL;
        ns_ldap_attr_t          *attrptr;

        buffer = argp->buf.buffer;
        buflen = (size_t)argp->buf.buflen;
        (void) memset(argp->buf.buffer, 0, buflen);
        exec = (execstr_t *)(argp->buf.result);
        ceiling = buffer + buflen;
        exec->name = (char *)NULL;
        exec->policy = (char *)NULL;
        exec->type = (char *)NULL;
        exec->res1 = (char *)NULL;
        exec->res2 = (char *)NULL;
        exec->id = (char *)NULL;
        exec->attr = (char *)NULL;

        for (i = 0; i < entry->attr_count; i++) {
                attrptr = entry->attr_pair[i];
                if (attrptr == NULL) {
                        return ((int)NSS_STR_PARSE_PARSE);
                }
                if (strcasecmp(attrptr->attrname, _EXEC_NAME) == 0) {
                        if ((attrptr->attrvalue[0] == NULL) ||
                            (len = strlen(attrptr->attrvalue[0])) < 1) {
                                return ((int)NSS_STR_PARSE_PARSE);
                        }
                        exec->name = buffer;
                        buffer += len + 1;
                        if (buffer >= ceiling) {
                                return ((int)NSS_STR_PARSE_ERANGE);
                        }
                        (void) strcpy(exec->name, attrptr->attrvalue[0]);
                        continue;
                }
                if (strcasecmp(attrptr->attrname, _EXEC_POLICY) == 0) {
                        if ((attrptr->attrvalue[0] == NULL) ||
                            (len = strlen(attrptr->attrvalue[0])) < 1) {
                                exec->policy = nullstring;
                        } else {
                                exec->policy = buffer;
                                buffer += len + 1;
                                if (buffer >= ceiling) {
                                        return ((int)NSS_STR_PARSE_ERANGE);
                                }
                                (void) strcpy(exec->policy,
                                    attrptr->attrvalue[0]);
                        }
                        continue;
                }
                if (strcasecmp(attrptr->attrname, _EXEC_TYPE) == 0) {
                        if ((attrptr->attrvalue[0] == NULL) ||
                            (len = strlen(attrptr->attrvalue[0])) < 1) {
                                exec->type = nullstring;
                        } else {
                                exec->type = buffer;
                                buffer += len + 1;
                                if (buffer >= ceiling) {
                                        return ((int)NSS_STR_PARSE_ERANGE);
                                }
                                (void) strcpy(exec->type,
                                    attrptr->attrvalue[0]);
                        }
                        continue;
                }
                if (strcasecmp(attrptr->attrname, _EXEC_RES1) == 0) {
                        if ((attrptr->attrvalue[0] == NULL) ||
                            (len = strlen(attrptr->attrvalue[0])) < 1) {
                                exec->res1 = nullstring;
                        } else {
                                exec->res1 = buffer;
                                buffer += len + 1;
                                if (buffer >= ceiling) {
                                        return ((int)NSS_STR_PARSE_ERANGE);
                                }
                                (void) strcpy(exec->res1,
                                    attrptr->attrvalue[0]);
                        }
                        continue;
                }
                if (strcasecmp(attrptr->attrname, _EXEC_RES2) == 0) {
                        if ((attrptr->attrvalue[0] == NULL) ||
                            (len = strlen(attrptr->attrvalue[0])) < 1) {
                                exec->res2 = nullstring;
                        } else {
                                exec->res2 = buffer;
                                buffer += len + 1;
                                if (buffer >= ceiling) {
                                        return ((int)NSS_STR_PARSE_ERANGE);
                                }
                                (void) strcpy(exec->res2,
                                    attrptr->attrvalue[0]);
                        }
                        continue;
                }
                if (strcasecmp(attrptr->attrname, _EXEC_ID) == 0) {
                        if ((attrptr->attrvalue[0] == NULL) ||
                            (len = strlen(attrptr->attrvalue[0])) < 1) {
                                exec->id = nullstring;
                        } else {
                                exec->id = buffer;
                                buffer += len + 1;
                                if (buffer >= ceiling) {
                                        return ((int)NSS_STR_PARSE_ERANGE);
                                }
                                (void) strcpy(exec->id, attrptr->attrvalue[0]);
                        }
                        continue;
                }
                if (strcasecmp(attrptr->attrname, _EXEC_ATTRS) == 0) {
                        if ((attrptr->attrvalue[0] == NULL) ||
                            (len = strlen(attrptr->attrvalue[0])) < 1) {
                                exec->attr = nullstring;
                        } else {
                                exec->attr = buffer;
                                buffer += len + 1;
                                if (buffer >= ceiling) {
                                        return ((int)NSS_STR_PARSE_ERANGE);
                                }
                                (void) strcpy(exec->attr,
                                    attrptr->attrvalue[0]);
                        }
                        continue;
                }
        }

        exec->next = (execstr_t *)NULL;

#ifdef  DEBUG
        (void) fprintf(stdout, "\n[getexecattr.c: _exec_ldap_exec2ent]\n");
        _print_execstr(exec);
#endif  /* DEBUG */

        return ((int)NSS_STR_PARSE_SUCCESS);
}


/*
 * place the results from ldap object structure into the file format
 * returns NSS_STR_PARSE_{SUCCESS, ERANGE, PARSE}
 */
static int
_nss_ldap_exec2str(ldap_backend_ptr be, nss_XbyY_args_t *argp)
{
        int                     status = NSS_STR_PARSE_SUCCESS;
        ns_ldap_result_t        *result = be->result;
        int                     len;
        char                    *buffer, **name, **policy, **type;
        char                    **res1, **res2, **id, **attr;
        char                    *policy_str, *type_str, *res1_str, *res2_str;
        char                    *id_str, *attr_str;

        if (result == NULL)
                return (NSS_STR_PARSE_PARSE);

        (void) memset(argp->buf.buffer, 0, argp->buf.buflen);

        name = __ns_ldap_getAttr(result->entry, _EXEC_NAME);
        if (name == NULL || name[0] == NULL ||
            (strlen(name[0]) < 1)) {
                status = NSS_STR_PARSE_PARSE;
                goto result_exec2str;
        }

        policy = __ns_ldap_getAttr(result->entry, _EXEC_POLICY);

        if (policy == NULL || policy[0] == NULL)
                policy_str = _NO_VALUE;
        else
                policy_str = policy[0];

        type = __ns_ldap_getAttr(result->entry, _EXEC_TYPE);
        if (type == NULL || type[0] == NULL)
                type_str = _NO_VALUE;
        else
                type_str = type[0];

        res1 = __ns_ldap_getAttr(result->entry, _EXEC_RES1);
        if (res1 == NULL || res1[0] == NULL)
                res1_str = _NO_VALUE;
        else
                res1_str = res1[0];

        res2 = __ns_ldap_getAttr(result->entry, _EXEC_RES2);
        if (res2 == NULL || res2[0] == NULL)
                res2_str = _NO_VALUE;
        else
                res2_str = res2[0];

        id = __ns_ldap_getAttr(result->entry, _EXEC_ID);
        if (id == NULL || id[0] == NULL)
                id_str = _NO_VALUE;
        else
                id_str = id[0];

        attr = __ns_ldap_getAttr(result->entry, _EXEC_ATTRS);
        if (attr == NULL || attr[0] == NULL)
                attr_str = _NO_VALUE;
        else
                attr_str = attr[0];

        /* 7 = 6 ':' + 1 '\0' */
        len = strlen(name[0]) + strlen(policy_str) + strlen(type_str) +
            strlen(res1_str) + strlen(res2_str) + strlen(id_str) +
            strlen(attr_str) + 7;

        if (len > argp->buf.buflen) {
                status = NSS_STR_PARSE_ERANGE;
                goto  result_exec2str;
        }
        if (argp->buf.result != NULL) {
                if ((be->buffer = calloc(1, len)) == NULL) {
                        status = NSS_STR_PARSE_PARSE;
                        goto result_exec2str;
                }
                buffer = be->buffer;
        } else
                buffer = argp->buf.buffer;

        (void) snprintf(buffer, len, "%s:%s:%s:%s:%s:%s:%s",
            name[0], policy_str, type_str, res1_str,
            res2_str, id_str, attr_str);
        /* The front end marshaller does not need the trailing null */
        if (argp->buf.result != NULL)
                be->buflen = strlen(buffer);
result_exec2str:
        (void) __ns_ldap_freeResult(&be->result);
        return (status);
}


static nss_status_t
_exec_process_val(ldap_backend_ptr be, nss_XbyY_args_t *argp)
{
        int                     status;
        nss_status_t            nss_stat = NSS_UNAVAIL;
        ns_ldap_attr_t          *attrptr;
        ns_ldap_entry_t         *entry;
        ns_ldap_result_t        *result = be->result;
        _priv_execattr  *_priv_exec = (_priv_execattr *)(argp->key.attrp);

        argp->returnval = NULL;
        attrptr = getattr(result, 0);
        if (attrptr == NULL) {
                (void) __ns_ldap_freeResult(&be->result);
                return (nss_stat);
        }
        for (entry = result->entry; entry != NULL; entry = entry->next) {
                status = _exec_ldap_exec2ent(entry, argp);
                switch (status) {
                case NSS_STR_PARSE_SUCCESS:
                        argp->returnval = argp->buf.result;
                        nss_stat = NSS_SUCCESS;
                        if (IS_GET_ALL(_priv_exec->search_flag)) {
                                if (_doexeclist(argp) == 0) {
                                        nss_stat = NSS_UNAVAIL;
                                }
                        }
                        break;
                case NSS_STR_PARSE_ERANGE:
                        argp->erange = 1;
                        nss_stat = NSS_NOTFOUND;
                        break;
                case NSS_STR_PARSE_PARSE:
                        nss_stat = NSS_NOTFOUND;
                        break;
                default:
                        nss_stat = NSS_UNAVAIL;
                        break;
                }

                if (IS_GET_ONE(_priv_exec->search_flag) ||
                    (nss_stat != NSS_SUCCESS)) {
                        break;
                }
        }

        return (nss_stat);
}


/*
 * Check if we have either an exact match or a wild-card entry for that id.
 */
static nss_status_t
get_wild(ldap_backend_ptr be, nss_XbyY_args_t *argp, int getby_flag)
{
        char            *dup_id = NULL;
        char            *wild_id;
        char            searchfilter[SEARCHFILTERLEN];
        char            userdata[SEARCHFILTERLEN];
        char            name[SEARCHFILTERLEN];
        char            id[SEARCHFILTERLEN];
        int             ret;
        nss_status_t    nss_stat = NSS_NOTFOUND;
        _priv_execattr  *_priv_exec = (_priv_execattr *)(argp->key.attrp);
        const char      *policy = _priv_exec->policy;
        const char      *type = _priv_exec->type;

        if ((policy != NULL && strpbrk(policy, "*()\\") != NULL) ||
            type != NULL && strpbrk(type, "*()\\") != NULL)
                return ((nss_status_t)NSS_NOTFOUND);

        if (_priv_exec->id != NULL)
                dup_id = strdup(_priv_exec->id);

        switch (getby_flag) {
        case NSS_DBOP_EXECATTR_BYNAMEID:
                if (_ldap_filter_name(name, _priv_exec->name,
                    sizeof (name)) != 0)
                        goto go_out;
                break;
        }

        wild_id = dup_id;
        do {
                if (wild_id != NULL) {
                        if (_ldap_filter_name(id, wild_id, sizeof (id)) != 0)
                                goto go_out;
                } else
                        (void) strlcpy(id, "*", sizeof (id));

                switch (getby_flag) {
                case NSS_DBOP_EXECATTR_BYID:
                        ret = snprintf(searchfilter, sizeof (searchfilter),
                            _EXEC_GETEXECID, id, ISWILD(policy), ISWILD(type));
                        if (ret >= sizeof (searchfilter) || ret < 0)
                                goto go_out;
                        ret = snprintf(userdata, sizeof (userdata),
                            _EXEC_GETEXECID_SSD, id, ISWILD(policy),
                            ISWILD(type));
                        if (ret >= sizeof (userdata) || ret < 0)
                                goto go_out;
                        break;

                case NSS_DBOP_EXECATTR_BYNAMEID:
                        ret = snprintf(searchfilter, sizeof (searchfilter),
                            _EXEC_GETEXECNAMEID, name, id,
                            ISWILD(policy), ISWILD(type));
                        if (ret >= sizeof (searchfilter) || ret < 0)
                                goto go_out;
                        ret = snprintf(userdata, sizeof (userdata),
                            _EXEC_GETEXECNAMEID_SSD, name, id,
                            ISWILD(policy), ISWILD(type));
                        if (ret >= sizeof (userdata) || ret < 0)
                                goto go_out;
                        break;

                default:
                        goto go_out;
                }
                nss_stat = _nss_ldap_nocb_lookup(be, argp, _EXECATTR,
                    searchfilter, NULL, _merge_SSD_filter, userdata);
                if (nss_stat == NSS_SUCCESS)
                        break;
        } while ((wild_id = _exec_wild_id(wild_id, type)) != NULL);

go_out:
        free(dup_id);

        return (nss_stat);
}

static nss_status_t
exec_attr_process_val(ldap_backend_ptr be, nss_XbyY_args_t *argp)
{
        _priv_execattr  *_priv_exec = (_priv_execattr *)(argp->key.attrp);
        int             stat, nss_stat = NSS_SUCCESS;

        if (IS_GET_ONE(_priv_exec->search_flag)) {
                /* ns_ldap_entry_t -> file format */
                stat = (*be->ldapobj2str)(be, argp);

                if (stat == NSS_STR_PARSE_SUCCESS) {
                        if (argp->buf.result != NULL) {
                                /* file format -> execstr_t */
                                stat = (*argp->str2ent)(be->buffer,
                                    be->buflen,
                                    argp->buf.result,
                                    argp->buf.buffer,
                                    argp->buf.buflen);
                                if (stat == NSS_STR_PARSE_SUCCESS) {
                                        argp->returnval = argp->buf.result;
                                        argp->returnlen = 1; /* irrelevant */
                                        nss_stat = NSS_SUCCESS;
                                } else {
                                        argp->returnval = NULL;
                                        argp->returnlen = 0;
                                        nss_stat = NSS_NOTFOUND;
                                }
                        } else {
                                /* return file format in argp->buf.buffer */
                                argp->returnval = argp->buf.buffer;
                                argp->returnlen = strlen(argp->buf.buffer);
                                nss_stat = NSS_SUCCESS;
                        }
                } else {
                        argp->returnval = NULL;
                        argp->returnlen = 0;
                        nss_stat = NSS_NOTFOUND;
                }
        } else {
                /* GET_ALL */
                nss_stat = _exec_process_val(be, argp);
                _exec_cleanup(nss_stat, argp);
        }

        return (nss_stat);

}

static nss_status_t
getbynam(ldap_backend_ptr be, void *a)
{
        char            searchfilter[SEARCHFILTERLEN];
        char            userdata[SEARCHFILTERLEN];
        char            name[SEARCHFILTERLEN];
        int             ret;
        nss_status_t    nss_stat;
        nss_XbyY_args_t *argp = (nss_XbyY_args_t *)a;
        _priv_execattr  *_priv_exec = (_priv_execattr *)(argp->key.attrp);
        const char      *policy = _priv_exec->policy;
        const char      *type = _priv_exec->type;

        if (policy != NULL && strpbrk(policy, "*()\\") != NULL ||
            type != NULL && strpbrk(type, "*()\\") != NULL ||
            _ldap_filter_name(name, _priv_exec->name, sizeof (name)) != 0)
                return ((nss_status_t)NSS_NOTFOUND);
        ret = snprintf(searchfilter, sizeof (searchfilter),
            _EXEC_GETEXECNAME, name, ISWILD(policy), ISWILD(type));
        if (ret >= sizeof (searchfilter) || ret < 0)
                return ((nss_status_t)NSS_NOTFOUND);
        ret = snprintf(userdata, sizeof (userdata),
            _EXEC_GETEXECNAME_SSD, name, ISWILD(policy), ISWILD(type));
        if (ret >= sizeof (userdata) || ret < 0)
                return ((nss_status_t)NSS_NOTFOUND);

        nss_stat = _nss_ldap_nocb_lookup(be, argp, _EXECATTR,
            searchfilter, NULL, _merge_SSD_filter, userdata);

        if (nss_stat ==  NSS_SUCCESS)
                nss_stat = exec_attr_process_val(be, argp);

        return (nss_stat);
}

static nss_status_t
getbyid(ldap_backend_ptr be, void *a)
{
        nss_status_t    nss_stat = NSS_SUCCESS;
        nss_XbyY_args_t *argp = (nss_XbyY_args_t *)a;

        nss_stat = get_wild(be, argp, NSS_DBOP_EXECATTR_BYID);

        if (nss_stat ==  NSS_SUCCESS)
                nss_stat = exec_attr_process_val(be, argp);

        return (nss_stat);
}


static nss_status_t
getbynameid(ldap_backend_ptr be, void *a)
{
        nss_status_t    nss_stat;
        nss_XbyY_args_t *argp = (nss_XbyY_args_t *)a;

        nss_stat = get_wild(be, argp, NSS_DBOP_EXECATTR_BYNAMEID);

        if (nss_stat ==  NSS_SUCCESS)
                nss_stat = exec_attr_process_val(be, argp);

        return (nss_stat);
}


static ldap_backend_op_t execattr_ops[] = {
        _nss_ldap_destr,
        _nss_ldap_endent,
        _nss_ldap_setent,
        _nss_ldap_getent,
        getbynam,
        getbyid,
        getbynameid
};


/*ARGSUSED0*/
nss_backend_t *
_nss_ldap_exec_attr_constr(const char *dummy1,
    const char *dummy2,
    const char *dummy3,
    const char *dummy4,
    const char *dummy5,
    const char *dummy6,
    const char *dummy7)
{
#ifdef  DEBUG
        (void) fprintf(stdout,
            "\n[getexecattr.c: _nss_ldap_exec_attr_constr]\n");
#endif
        return ((nss_backend_t *)_nss_ldap_constr(execattr_ops,
            sizeof (execattr_ops)/sizeof (execattr_ops[0]), _EXECATTR,
            exec_attrs, _nss_ldap_exec2str));
}