/*

 * Copyright 1999-2025 The OpenSSL Project Authors. All Rights Reserved.

 *

 * Licensed under the Apache License 2.0 (the "License").  You may not use

 * this file except in compliance with the License.  You can obtain a copy

 * in the file LICENSE in the source distribution or at

 * https://www.openssl.org/source/license.html

 */

/* X509 v3 extension utilities */

#include <stdio.h>

#include "internal/cryptlib.h"

#include <openssl/conf.h>

#include <openssl/x509v3.h>

#include "ext_dat.h"

#include "x509_local.h"

static STACK_OF(X509V3_EXT_METHOD) *ext_list = NULL;

static int ext_cmp(const X509V3_EXT_METHOD *const *a,

    const X509V3_EXT_METHOD *const *b);

static void ext_list_free(X509V3_EXT_METHOD *ext);

int X509V3_EXT_add(X509V3_EXT_METHOD *ext)

{

    if (ext_list == NULL

        && (ext_list = sk_X509V3_EXT_METHOD_new(ext_cmp)) == NULL) {

        ERR_raise(ERR_LIB_X509V3, ERR_R_CRYPTO_LIB);

        return 0;

    }

    if (!sk_X509V3_EXT_METHOD_push(ext_list, ext)) {

        ERR_raise(ERR_LIB_X509V3, ERR_R_CRYPTO_LIB);

        return 0;

    }

    return 1;

}

static int ext_cmp(const X509V3_EXT_METHOD *const *a,

    const X509V3_EXT_METHOD *const *b)

{

    return ((*a)->ext_nid - (*b)->ext_nid);

}

DECLARE_OBJ_BSEARCH_CMP_FN(const X509V3_EXT_METHOD *,

    const X509V3_EXT_METHOD *, ext);

IMPLEMENT_OBJ_BSEARCH_CMP_FN(const X509V3_EXT_METHOD *,

    const X509V3_EXT_METHOD *, ext);

#include "standard_exts.h"

const X509V3_EXT_METHOD *X509V3_EXT_get_nid(int nid)

{

    X509V3_EXT_METHOD tmp;

    const X509V3_EXT_METHOD *t = &tmp, *const * ret;

    int idx;

    if (nid < 0)

        return NULL;

    tmp.ext_nid = nid;

    ret = OBJ_bsearch_ext(&t, standard_exts, STANDARD_EXTENSION_COUNT);

    if (ret)

        return *ret;

    if (!ext_list)

        return NULL;

    /* Ideally, this would be done under a lock */

    sk_X509V3_EXT_METHOD_sort(ext_list);

    idx = sk_X509V3_EXT_METHOD_find(ext_list, &tmp);

    /* A failure to locate the item is handled by the value method */

    return sk_X509V3_EXT_METHOD_value(ext_list, idx);

}

const X509V3_EXT_METHOD *X509V3_EXT_get(const X509_EXTENSION *ext)

{

    int nid;

    if ((nid = OBJ_obj2nid(X509_EXTENSION_get_object(ext))) == NID_undef)

        return NULL;

    return X509V3_EXT_get_nid(nid);

}

int X509V3_EXT_add_list(X509V3_EXT_METHOD *extlist)

{

    for (; extlist->ext_nid != -1; extlist++)

        if (!X509V3_EXT_add(extlist))

            return 0;

    return 1;

}

int X509V3_EXT_add_alias(int nid_to, int nid_from)

{

    const X509V3_EXT_METHOD *ext;

    X509V3_EXT_METHOD *tmpext;

    if ((ext = X509V3_EXT_get_nid(nid_from)) == NULL) {

        ERR_raise(ERR_LIB_X509V3, X509V3_R_EXTENSION_NOT_FOUND);

        return 0;

    }

    if ((tmpext = OPENSSL_malloc(sizeof(*tmpext))) == NULL)

        return 0;

    *tmpext = *ext;

    tmpext->ext_nid = nid_to;

    tmpext->ext_flags |= X509V3_EXT_DYNAMIC;

    if (!X509V3_EXT_add(tmpext)) {

        OPENSSL_free(tmpext);

        return 0;

    }

    return 1;

}

void X509V3_EXT_cleanup(void)

{

    sk_X509V3_EXT_METHOD_pop_free(ext_list, ext_list_free);

    ext_list = NULL;

}

static void ext_list_free(X509V3_EXT_METHOD *ext)

{

    if (ext->ext_flags & X509V3_EXT_DYNAMIC)

        OPENSSL_free(ext);

}

/*

 * Legacy function: we don't need to add standard extensions any more because

 * they are now kept in ext_dat.h.

 */

int X509V3_add_standard_extensions(void)

{

    return 1;

}

int ossl_ignored_x509_extension(const X509_EXTENSION *ex, int flags)

{

    /*

     * Empty OCTET STRINGs and empty SEQUENCEs encode to just two bytes of tag

     * (0x04 or 0x30) and length (0x00).  We use this fact to suppress empty

     * AKID and SKID extensions that may be briefly generated when processing

     * the "= none" value or only ":nonss"-qualified AKIDs when the subject is

     * self-signed.

     *

     * The resulting extension is empty, and must not be retained, but does

     * serve to drop any previous value of the same extension, when called

     * via

     * - X509v3_add_extensions(), or

     * - either of X509V3_add1_i2d() or X509V3_EXT_add_nconf_sk(),

     *   with a flags (or ctx->flags) value that allows replacement.

     */

    if (ex->value.length == 2

        && (ex->value.data[0] == 0x30 || ex->value.data[0] == 0x04)) {

        ASN1_OBJECT *obj = ex->object;

        ASN1_OBJECT *skid = OBJ_nid2obj(NID_subject_key_identifier);

        ASN1_OBJECT *akid = OBJ_nid2obj(NID_authority_key_identifier);

        if (OBJ_cmp(obj, skid) == 0 || OBJ_cmp(obj, akid) == 0) {

            if ((flags & X509V3_ADD_SILENT) == 0)

                ERR_raise_data(ERR_LIB_X509, X509_R_INVALID_EXTENSION,

                    "Invalid empty X.509 %s extension", obj->sn);

            return 1;

        }

    }

    return 0;

}

/* Return an extension internal structure */

void *X509V3_EXT_d2i(const X509_EXTENSION *ext)

{

    const X509V3_EXT_METHOD *method;

    const unsigned char *p;

    const ASN1_STRING *extvalue;

    int extlen;

    if ((method = X509V3_EXT_get(ext)) == NULL)

        return NULL;

    extvalue = X509_EXTENSION_get_data(ext);

    p = ASN1_STRING_get0_data(extvalue);

    extlen = ASN1_STRING_length(extvalue);

    if (method->it)

        return ASN1_item_d2i(NULL, &p, extlen, ASN1_ITEM_ptr(method->it));

    return method->d2i(NULL, &p, extlen);

}

/*-

 * Get critical flag and decoded version of extension from a NID.

 * The "idx" variable returns the last found extension and can

 * be used to retrieve multiple extensions of the same NID.

 * However multiple extensions with the same NID is usually

 * due to a badly encoded certificate so if idx is NULL we

 * choke if multiple extensions exist.

 * The "crit" variable is set to the critical value.

 * The return value is the decoded extension or NULL on

 * error. The actual error can have several different causes,

 * the value of *crit reflects the cause:

 * >= 0, extension found but not decoded (reflects critical value).

 * -1 extension not found.

 * -2 extension occurs more than once.

 */

void *X509V3_get_d2i(const STACK_OF(X509_EXTENSION) *x, int nid, int *crit,

    int *idx)

{

    int lastpos, i;

    X509_EXTENSION *ex, *found_ex = NULL;

    if (!x) {

        if (idx)

            *idx = -1;

        if (crit)

            *crit = -1;

        return NULL;

    }

    if (idx)

        lastpos = *idx + 1;

    else

        lastpos = 0;

    if (lastpos < 0)

        lastpos = 0;

    for (i = lastpos; i < sk_X509_EXTENSION_num(x); i++) {

        ex = sk_X509_EXTENSION_value(x, i);

        if (OBJ_obj2nid(X509_EXTENSION_get_object(ex)) == nid) {

            if (idx) {

                *idx = i;

                found_ex = ex;

                break;

            } else if (found_ex) {

                /* Found more than one */

                if (crit)

                    *crit = -2;

                return NULL;

            }

            found_ex = ex;

        }

    }

    if (found_ex) {

        /* Found it */

        if (crit)

            *crit = X509_EXTENSION_get_critical(found_ex);

        return X509V3_EXT_d2i(found_ex);

    }

    /* Extension not found */

    if (idx)

        *idx = -1;

    if (crit)

        *crit = -1;

    return NULL;

}

/*

 * This function is a general extension append, replace and delete utility.

 * The precise operation is governed by the 'flags' value. The 'crit' and

 * 'value' arguments (if relevant) are the extensions internal structure.

 */

int X509V3_add1_i2d(STACK_OF(X509_EXTENSION) **x, int nid, void *value,

    int crit, unsigned long flags)

{

    int errcode, extidx = -1;

    X509_EXTENSION *ext = NULL, *extmp;

    STACK_OF(X509_EXTENSION) *ret = NULL;

    unsigned long ext_op = flags & X509V3_ADD_OP_MASK;

    /*

     * If appending we don't care if it exists, otherwise look for existing

     * extension.

     */

    if (ext_op != X509V3_ADD_APPEND)

        extidx = X509v3_get_ext_by_NID(*x, nid, -1);

    /* See if extension exists */

    if (extidx >= 0) {

        /* If keep existing, nothing to do */

        if (ext_op == X509V3_ADD_KEEP_EXISTING)

            return 1;

        /* If default then its an error */

        if (ext_op == X509V3_ADD_DEFAULT) {

            errcode = X509V3_R_EXTENSION_EXISTS;

            goto err;

        }

        /* If delete, just delete it */

        if (ext_op == X509V3_ADD_DELETE) {

            extmp = sk_X509_EXTENSION_delete(*x, extidx);

            if (extmp == NULL)

                return -1;

            X509_EXTENSION_free(extmp);

            return 1;

        }

    } else {

        /*

         * If replace existing or delete, error since extension must exist

         */

        if ((ext_op == X509V3_ADD_REPLACE_EXISTING) || (ext_op == X509V3_ADD_DELETE)) {

            errcode = X509V3_R_EXTENSION_NOT_FOUND;

            goto err;

        }

    }

    /*

     * If we get this far then we have to create an extension: could have

     * some flags for alternative encoding schemes...

     */

    ext = X509V3_EXT_i2d(nid, crit, value);

    if (!ext) {

        ERR_raise(ERR_LIB_X509V3, X509V3_R_ERROR_CREATING_EXTENSION);

        return 0;

    }

    /* If extension exists replace it.. */

    if (extidx >= 0) {

        extmp = sk_X509_EXTENSION_value(*x, extidx);

        if (ossl_ignored_x509_extension(ext, X509V3_ADD_SILENT)) {

            if (!sk_X509_EXTENSION_delete(*x, extidx))

                return -1;

        } else if (!sk_X509_EXTENSION_set(*x, extidx, ext)) {

            return -1;

        }

        X509_EXTENSION_free(extmp);

        return 1;

    }

    ret = *x;

    if (*x == NULL

        && (ret = sk_X509_EXTENSION_new_null()) == NULL)

        goto m_fail;

    if (!sk_X509_EXTENSION_push(ret, ext))

        goto m_fail;

    *x = ret;

    return 1;

m_fail:

    /* ERR_raise(ERR_LIB_X509V3, ERR_R_CRYPTO_LIB); */

    if (ret != *x)

        sk_X509_EXTENSION_free(ret);

    X509_EXTENSION_free(ext);

    return -1;

err:

    if (!(flags & X509V3_ADD_SILENT))

        ERR_raise(ERR_LIB_X509V3, errcode);

    return 0;

}