/*

 * Copyright 1999-2020 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

 */

#include <stdio.h>

#include "internal/cryptlib.h"

#include <openssl/asn1t.h>

#include <openssl/x509.h>

#include "crypto/x509.h"

/* Minor tweak to operation: zero private key data */

static int pkey_cb(int operation, ASN1_VALUE **pval, const ASN1_ITEM *it,

                   void *exarg)

{

    PKCS8_PRIV_KEY_INFO *key;

    int version;

    switch (operation) {

    case ASN1_OP_FREE_PRE:

        /* The structure is still valid during ASN1_OP_FREE_PRE */

        key = (PKCS8_PRIV_KEY_INFO *)*pval;

        if (key->pkey)

            OPENSSL_cleanse(key->pkey->data, key->pkey->length);

        break;

    case ASN1_OP_D2I_POST:

        /* Insist on a valid version now that the structure is decoded */

        key = (PKCS8_PRIV_KEY_INFO *)*pval;

        version = ASN1_INTEGER_get(key->version);

        if (version < 0 || version > 1)

            return 0;

        if (version == 0 && key->kpub != NULL)

            return 0;

        break;

    }

    return 1;

}

ASN1_SEQUENCE_cb(PKCS8_PRIV_KEY_INFO, pkey_cb) = {

        ASN1_SIMPLE(PKCS8_PRIV_KEY_INFO, version, ASN1_INTEGER),

        ASN1_SIMPLE(PKCS8_PRIV_KEY_INFO, pkeyalg, X509_ALGOR),

        ASN1_SIMPLE(PKCS8_PRIV_KEY_INFO, pkey, ASN1_OCTET_STRING),

        ASN1_IMP_SET_OF_OPT(PKCS8_PRIV_KEY_INFO, attributes, X509_ATTRIBUTE, 0),

        ASN1_IMP_OPT(PKCS8_PRIV_KEY_INFO, kpub, ASN1_BIT_STRING, 1)

} ASN1_SEQUENCE_END_cb(PKCS8_PRIV_KEY_INFO, PKCS8_PRIV_KEY_INFO)

IMPLEMENT_ASN1_FUNCTIONS(PKCS8_PRIV_KEY_INFO)

int PKCS8_pkey_set0(PKCS8_PRIV_KEY_INFO *priv, ASN1_OBJECT *aobj,

                    int version,

                    int ptype, void *pval, unsigned char *penc, int penclen)

{

    if (version >= 0) {

        /* We only support PKCS#8 v1 (0) and v2 (1). */

        if (version > 1)

            return 0;

        if (!ASN1_INTEGER_set(priv->version, version))

            return 0;

    }

    if (!X509_ALGOR_set0(priv->pkeyalg, aobj, ptype, pval))

        return 0;

    if (penc)

        ASN1_STRING_set0(priv->pkey, penc, penclen);

    return 1;

}

int PKCS8_pkey_get0(const ASN1_OBJECT **ppkalg,

                    const unsigned char **pk, int *ppklen,

                    const X509_ALGOR **pa, const PKCS8_PRIV_KEY_INFO *p8)

{

    if (ppkalg)

        *ppkalg = p8->pkeyalg->algorithm;

    if (pk) {

        *pk = ASN1_STRING_get0_data(p8->pkey);

        *ppklen = ASN1_STRING_length(p8->pkey);

    }

    if (pa)

        *pa = p8->pkeyalg;

    return 1;

}

const STACK_OF(X509_ATTRIBUTE) *

PKCS8_pkey_get0_attrs(const PKCS8_PRIV_KEY_INFO *p8)

{

    return p8->attributes;

}

int PKCS8_pkey_add1_attr_by_NID(PKCS8_PRIV_KEY_INFO *p8, int nid, int type,

                                const unsigned char *bytes, int len)

{

    if (X509at_add1_attr_by_NID(&p8->attributes, nid, type, bytes, len) != NULL)

        return 1;

    return 0;

}

int PKCS8_pkey_add1_attr_by_OBJ(PKCS8_PRIV_KEY_INFO *p8, const ASN1_OBJECT *obj, int type,

                                const unsigned char *bytes, int len)

{

    return (X509at_add1_attr_by_OBJ(&p8->attributes, obj, type, bytes, len) != NULL);

}

int PKCS8_pkey_add1_attr(PKCS8_PRIV_KEY_INFO *p8, X509_ATTRIBUTE *attr)

{

    return (X509at_add1_attr(&p8->attributes, attr) != NULL);

}