// Copyright 2014 Citra Emulator Project
// Licensed under GPLv2 or any later version
// Refer to the license.txt file included.

#include <random>
#include "common/common_types.h"
#include "core/hle/service/ssl_c.h"

////////////////////////////////////////////////////////////////////////////////////////////////////
// Namespace SSL_C

namespace SSL_C {

// TODO: Implement a proper CSPRNG in the future when actual security is needed
static std::mt19937 rand_gen;

static void Initialize(Service::Interface* self) {
    u32* cmd_buff = Kernel::GetCommandBuffer();

    // Seed random number generator when the SSL service is initialized
    std::random_device rand_device;
    rand_gen.seed(rand_device());

    // Stub, return success
    cmd_buff[1] = RESULT_SUCCESS.raw;
}

static void GenerateRandomData(Service::Interface* self) {
    u32* cmd_buff = Kernel::GetCommandBuffer();

    u32 size = cmd_buff[1];
    VAddr address = cmd_buff[3];

    // Fill the output buffer with random data.
    u32 data = 0;
    u32 i = 0;
    while (i < size) {
        if ((i % 4) == 0) {
            // The random number generator returns 4 bytes worth of data, so generate new random
            // data when i == 0 and when i is divisible by 4
            data = rand_gen();
        }

        if (size > 4) {
            // Use up the entire 4 bytes of the random data for as long as possible
            Memory::Write32(address + i, data);
            i += 4;
        } else if (size == 2) {
            Memory::Write16(address + i, static_cast<u16>(data & 0xffff));
            i += 2;
        } else {
            Memory::Write8(address + i, static_cast<u8>(data & 0xff));
            i++;
        }
    }

    // Stub, return success
    cmd_buff[1] = RESULT_SUCCESS.raw;
}

const Interface::FunctionInfo FunctionTable[] = {
    {0x00010002, Initialize, "Initialize"},
    {0x000200C2, nullptr, "CreateContext"},
    {0x00030000, nullptr, "CreateRootCertChain"},
    {0x00040040, nullptr, "DestroyRootCertChain"},
    {0x00050082, nullptr, "AddTrustedRootCA"},
    {0x00060080, nullptr, "RootCertChainAddDefaultCert"},
    {0x00070080, nullptr, "RootCertChainRemoveCert"},
    {0x000E0040, nullptr, "OpenDefaultClientCertContext"},
    {0x000F0040, nullptr, "CloseClientCertContext"},
    {0x00110042, GenerateRandomData, "GenerateRandomData"},
    {0x00120042, nullptr, "InitializeConnectionSession"},
    {0x00130040, nullptr, "StartConnection"},
    {0x00140040, nullptr, "StartConnectionGetOut"},
    {0x00150082, nullptr, "Read"},
    {0x00170082, nullptr, "Write"},
    {0x00180080, nullptr, "ContextSetRootCertChain"},
    {0x00190080, nullptr, "ContextSetClientCert"},
    {0x001B0080, nullptr, "ContextClearOpt"},
    {0x001E0040, nullptr, "DestroyContext"},
    {0x001F0082, nullptr, "ContextInitSharedmem"},
};

////////////////////////////////////////////////////////////////////////////////////////////////////
// Interface class

Interface::Interface() {
    Register(FunctionTable);
}

} // namespace
