/*
 * Copyright (c) 2016-2019, Arm Limited and affiliates.
 * SPDX-License-Identifier: Apache-2.0
 *
 * Licensed 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 "nsconfig.h"
#include <string.h>
#include "ns_types.h"
#include "ns_list.h"
#include "ns_trace.h"
#include "nsdynmemLIB.h"
#include "NWK_INTERFACE/Include/protocol.h"
#include "Common_Protocols/ipv6_constants.h"
#include "socket_api.h"
#include "6LoWPAN/ws/ws_config.h"
#include "Security/protocols/sec_prot_certs.h"
#include "Security/protocols/sec_prot_keys.h"

#ifdef HAVE_WS

#define TRACE_GROUP "spce"

int8_t sec_prot_certs_init(sec_prot_certs_t *certs)
{
    if (!certs) {
        return -1;
    }

    sec_prot_certs_chain_entry_init(&certs->own_cert_chain);
    ns_list_init(&certs->trusted_cert_chain_list);
    ns_list_init(&certs->cert_revocat_lists);

    return 0;
}

void sec_prot_certs_delete(sec_prot_certs_t *certs)
{
    if (!certs) {
        return;
    }

    sec_prot_certs_chain_entry_init(&certs->own_cert_chain);
    sec_prot_certs_chain_list_delete(&certs->trusted_cert_chain_list);
    sec_prot_certs_revocat_lists_delete(&certs->cert_revocat_lists);
}

cert_chain_entry_t *sec_prot_certs_chain_entry_create(void)
{
    cert_chain_entry_t *entry = ns_dyn_mem_alloc(sizeof(cert_chain_entry_t));
    if (!entry) {
        return NULL;
    }
    sec_prot_certs_chain_entry_init(entry);
    return entry;
}

void sec_prot_certs_chain_entry_init(cert_chain_entry_t *entry)
{
    memset(entry, 0, sizeof(cert_chain_entry_t));
}

int8_t sec_prot_certs_cert_set(cert_chain_entry_t *entry, uint8_t index, uint8_t *cert, uint16_t cert_len)
{
    if (!entry || index >= SEC_PROT_CERT_CHAIN_DEPTH) {
        return -1;
    }

    entry->cert[index] = cert;
    entry->cert_len[index] = cert_len;

    return 0;
}

uint8_t *sec_prot_certs_cert_get(const cert_chain_entry_t *entry, uint8_t index, uint16_t *cert_len)
{
    if (!entry || index >= SEC_PROT_CERT_CHAIN_DEPTH || !entry->cert[index]) {
        return NULL;
    }

    *cert_len = entry->cert_len[index];
    return entry->cert[index];
}

int8_t sec_prot_certs_priv_key_set(cert_chain_entry_t *entry, uint8_t *key, uint8_t key_len)
{
    if (!entry) {
        return -1;
    }

    entry->key = key;
    entry->key_len = key_len;

    return 0;
}

uint8_t *sec_prot_certs_priv_key_get(const cert_chain_entry_t *entry, uint8_t *key_len)
{
    if (!entry) {
        return NULL;
    }
    *key_len = entry->key_len;
    return entry->key;
}

void sec_prot_certs_chain_list_add(cert_chain_list_t *cert_chain_list, cert_chain_entry_t *entry)
{
    ns_list_add_to_end(cert_chain_list, entry);
}

void sec_prot_certs_chain_list_delete(cert_chain_list_t *chain_list)
{
    ns_list_foreach_safe(cert_chain_entry_t, entry, chain_list) {
        ns_list_remove(chain_list, entry);
        ns_dyn_mem_free(entry);
    }
}

cert_revocat_list_entry_t *sec_prot_certs_revocat_list_entry_create(void)
{
    cert_revocat_list_entry_t *entry = ns_dyn_mem_alloc(sizeof(cert_revocat_list_entry_t));
    if (!entry) {
        return NULL;
    }
    sec_prot_certs_revocat_list_entry_init(entry);
    return entry;
}

void sec_prot_certs_revocat_list_entry_init(cert_revocat_list_entry_t *entry)
{
    memset(entry, 0, sizeof(cert_revocat_list_entry_t));
}

int8_t sec_prot_certs_revocat_list_set(cert_revocat_list_entry_t *entry, uint8_t *crl, uint16_t crl_len)
{
    if (!entry) {
        return -1;
    }

    entry->crl = crl;
    entry->crl_len = crl_len;

    return 0;
}

uint8_t *sec_prot_certs_revocat_list_get(const cert_revocat_list_entry_t *entry, uint16_t *crl_len)
{
    if (!entry) {
        return NULL;
    }
    *crl_len = entry->crl_len;
    return entry->crl;
}

void sec_prot_certs_revocat_lists_add(cert_revocat_lists_t *cert_revocat_lists, cert_revocat_list_entry_t *entry)
{
    ns_list_add_to_end(cert_revocat_lists, entry);
}

void sec_prot_certs_revocat_lists_delete(cert_revocat_lists_t *cert_revocat_lists)
{
    ns_list_foreach_safe(cert_revocat_list_entry_t, entry, cert_revocat_lists) {
        ns_list_remove(cert_revocat_lists, entry);
        ns_dyn_mem_free(entry);
    }
}

#endif /* HAVE_WS */