/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *  http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

#include <bootutil/sign_key.h>

/*
 * Even though this is in principle a Zephyr-specific file, the
 * simulator builds it and uses it as well. Because of that, we can't
 * use Kconfig symbols for key types, and have to rely on the MCUBoot
 * symbols (which Zephyr provides via this header, and the simulator
 * provides via the compiler command line).
 */
#include <mcuboot_config/mcuboot_config.h>
#include <zephyr/sys/util.h>

#if !defined(MCUBOOT_HW_KEY)
#if defined(MCUBOOT_SIGN_RSA) || defined(MCUBOOT_SIGN_EC256) || defined(MCUBOOT_SIGN_ED25519)
#define HAVE_KEYS

#ifndef MCUBOOT_SIGN_KEY_COUNT
#error "MCUBOOT_SIGN_KEY_COUNT must be defined by the build system"
#endif

#define _BOOT_KEY_CAT(a, b) a##b
#define BOOT_KEY_CAT(a, b) _BOOT_KEY_CAT(a, b)

#if defined(MCUBOOT_SIGN_RSA)
#  define BOOT_KEY_PRIMARY rsa_pub_key
#elif defined(MCUBOOT_SIGN_EC256)
#  define BOOT_KEY_PRIMARY ecdsa_pub_key
#elif defined(MCUBOOT_SIGN_ED25519)
#  define BOOT_KEY_PRIMARY ed25519_pub_key
#endif

#define BOOT_KEY_NAME(N) BOOT_KEY_CAT(BOOT_KEY_PRIMARY, BOOT_KEY_CAT(_, N))

#define BOOT_KEY_DECL_AT(i, _) \
    extern const unsigned char BOOT_KEY_NAME(UTIL_INC(i))[]; \
    extern unsigned int BOOT_KEY_CAT(BOOT_KEY_NAME(UTIL_INC(i)), _len);

#define BOOT_KEY_ENTRY_AT(i, _) \
    { .key = BOOT_KEY_NAME(UTIL_INC(i)), \
      .len = &BOOT_KEY_CAT(BOOT_KEY_NAME(UTIL_INC(i)), _len) },

extern const unsigned char BOOT_KEY_PRIMARY[];
extern unsigned int BOOT_KEY_CAT(BOOT_KEY_PRIMARY, _len);
LISTIFY(UTIL_DEC(MCUBOOT_SIGN_KEY_COUNT), BOOT_KEY_DECL_AT, ())
#endif

/*
 * NOTE: *_pub_key and *_pub_key_len are autogenerated based on the provided
 *       key file. If no key file was configured, the array and length must be
 *       provided and added to the build manually.
 */
#if defined(HAVE_KEYS)
const struct bootutil_key bootutil_keys[] = {
    {
        .key = BOOT_KEY_PRIMARY,
        .len = &BOOT_KEY_CAT(BOOT_KEY_PRIMARY, _len),
    },
    LISTIFY(UTIL_DEC(MCUBOOT_SIGN_KEY_COUNT), BOOT_KEY_ENTRY_AT, ())
};
const int bootutil_key_cnt = sizeof(bootutil_keys) / sizeof(bootutil_keys[0]);
#endif /* HAVE_KEYS */
#else
unsigned int pub_key_len;
struct bootutil_key bootutil_keys[1] = {
    {
        .key = 0,
        .len = &pub_key_len,
    }
};
const int bootutil_key_cnt = 1;
#endif /* !MCUBOOT_HW_KEY */

#if defined(MCUBOOT_ENCRYPT_RSA) || defined(MCUBOOT_ENCRYPT_X25519) || defined(MCUBOOT_ENCRYPT_EC256)
extern const unsigned char enc_priv_key[];
extern unsigned int enc_priv_key_len;
const struct bootutil_key bootutil_enc_key = {
    .key = enc_priv_key,
    .len = &enc_priv_key_len,
};
#elif defined(MCUBOOT_ENCRYPT_KW)
#error "Encrypted images with AES-KW is not implemented yet."
#endif