/* Copyright (C) 2017 mod_auth_gssapi contributors - See COPYING for (C) terms */

#include "mod_auth_gssapi.h"

#define NSS_BUF_MIN 1024

#define NSS_BUF_MAX 1024*1024

static char *get_buf(char *cur, size_t *len)

{

    if (*len == 0) {

        *len = NSS_BUF_MIN;

    } else {

        *len *= 2;

    }

    if (*len > NSS_BUF_MAX) {

        *len = 0; /* will free the buf and return NULL */

    }

    return realloc(cur, *len);

}

int mag_get_user_uid(const char *name, uid_t *uid)

{

    struct passwd pwd, *user;

    size_t buflen = 0;

    char *buf = NULL;

    int ret;

    do {

        buf = get_buf(buf, &buflen);

        if (buf == NULL || buflen == 0) {

            ret = ENOMEM;

            break;

        }

        ret = getpwnam_r(name, &pwd, buf, buflen, &user);

    } while (ret == ERANGE);

    if (ret != 0 || user != &pwd) {

        ret = (ret == 0) ? EINVAL : ret;

    } else {

        *uid = user->pw_uid;

    }

    free(buf);

    return ret;

}

int mag_get_group_gid(const char *name, gid_t *gid)

{

    struct group grp, *group;

    size_t buflen = 0;

    char *buf = NULL;

    int ret;

    do {

        buf = get_buf(buf, &buflen);

        if (buf == NULL || buflen == 0) {

            ret = ENOMEM;

            break;

        }

        ret = getgrnam_r(name, &grp, buf, buflen, &group);

    } while (ret == ERANGE);

    if (ret != 0 || group != &grp) {

        ret = (ret == 0) ? EINVAL : ret;

    } else {

        *gid = group->gr_gid;

    }

    free(buf);

    return ret;

}

bool mag_strbuf_equal(const char *str, gss_buffer_t buf)

{

    if (strncmp(str, buf->value, buf->length) != 0) return false;

    return buf->length == strlen(str);

}