eden/src/core/hle/service/es/es.cpp

// SPDX-FileCopyrightText: Copyright 2018 yuzu Emulator Project
// SPDX-License-Identifier: GPL-2.0-or-later

#include "core/crypto/key_manager.h"
#include "core/hle/service/es/es.h"
#include "core/hle/service/ipc_helpers.h"
#include "core/hle/service/server_manager.h"
#include "core/hle/service/service.h"

namespace Service::ES {

constexpr Result ERROR_INVALID_ARGUMENT{ErrorModule::ETicket, 2};
constexpr Result ERROR_INVALID_RIGHTS_ID{ErrorModule::ETicket, 3};

class ETicket final : public ServiceFramework<ETicket> {
public:
    explicit ETicket(Core::System& system_) : ServiceFramework{system_, "es"} {
        // clang-format off
        static const FunctionInfo functions[] = {
            {1, &ETicket::ImportTicket, "ImportTicket"},
            {2, nullptr, "ImportTicketCertificateSet"},
            {3, nullptr, "DeleteTicket"},
            {4, nullptr, "DeletePersonalizedTicket"},
            {5, nullptr, "DeleteAllCommonTicket"},
            {6, nullptr, "DeleteAllPersonalizedTicket"},
            {7, nullptr, "DeleteAllPersonalizedTicketEx"},
            {8, &ETicket::GetTitleKey, "GetTitleKey"},
            {9, &ETicket::CountCommonTicket, "CountCommonTicket"},
            {10, &ETicket::CountPersonalizedTicket, "CountPersonalizedTicket"},
            {11, &ETicket::ListCommonTicketRightsIds, "ListCommonTicketRightsIds"},
            {12, &ETicket::ListPersonalizedTicketRightsIds, "ListPersonalizedTicketRightsIds"},
            {13, nullptr, "ListMissingPersonalizedTicket"},
            {14, &ETicket::GetCommonTicketSize, "GetCommonTicketSize"},
            {15, &ETicket::GetPersonalizedTicketSize, "GetPersonalizedTicketSize"},
            {16, &ETicket::GetCommonTicketData, "GetCommonTicketData"},
            {17, &ETicket::GetPersonalizedTicketData, "GetPersonalizedTicketData"},
            {18, nullptr, "OwnTicket"},
            {19, nullptr, "GetTicketInfo"},
            {20, nullptr, "ListLightTicketInfo"},
            {21, nullptr, "SignData"},
            {22, nullptr, "GetCommonTicketAndCertificateSize"},
            {23, nullptr, "GetCommonTicketAndCertificateData"},
            {24, nullptr, "ImportPrepurchaseRecord"},
            {25, nullptr, "DeletePrepurchaseRecord"},
            {26, nullptr, "DeleteAllPrepurchaseRecord"},
            {27, nullptr, "CountPrepurchaseRecord"},
            {28, nullptr, "ListPrepurchaseRecordRightsIds"},
            {29, nullptr, "ListPrepurchaseRecordInfo"},
            {30, nullptr, "CountTicket"},
            {31, nullptr, "ListTicketRightsIds"},
            {32, nullptr, "CountPrepurchaseRecordEx"},
            {33, nullptr, "ListPrepurchaseRecordRightsIdsEx"},
            {34, nullptr, "GetEncryptedTicketSize"},
            {35, nullptr, "GetEncryptedTicketData"},
            {36, nullptr, "DeleteAllInactiveELicenseRequiredPersonalizedTicket"},
            {37, nullptr, "OwnTicket2"},
            {38, nullptr, "OwnTicket3"},
            {39, nullptr, "DeleteAllInactivePersonalizedTicket"},
            {40, nullptr, "DeletePrepurchaseRecordByNintendoAccountId"},
            {101, nullptr, "Unknown101"}, // 18.0.0+
            {102, nullptr, "Unknown102"}, // 18.0.0+
            {103, nullptr, "Unknown103"}, // 18.0.0+
            {104, nullptr, "Unknown104"}, // 18.0.0+
            {105, nullptr, "Unknown105"}, // 20.0.0+
            {201, nullptr, "Unknown201"}, // 18.0.0+
            {202, nullptr, "Unknown202"}, // 18.0.0+
            {203, nullptr, "Unknown203"}, // 18.0.0+
            {204, nullptr, "Unknown204"}, // 18.0.0+
            {501, nullptr, "Unknown501"}, // 6.0.0+
            {502, nullptr, "Unknown502"}, // 6.0.0+
            {503, nullptr, "GetTitleKey"}, // 6.0.0+
            {504, nullptr, "Unknown504"}, // 6.0.0+
            {508, nullptr, "Unknown508"}, // 6.0.0+
            {509, nullptr, "Unknown509"}, // 6.0.0+
            {510, nullptr, "Unknown510"}, // 6.0.0+
            {511, nullptr, "Unknown511"}, // 9.0.0+
            {1001, nullptr, "Unknown1001"}, // 6.0.0+
            {1002, nullptr, "Unknown1002"}, // 6.0.0+
            {1003, nullptr, "GetIAsyncValue"}, // 6.0.0+
            {1004, nullptr, "Unknown1004"}, // 6.0.0+
            {1005, nullptr, "Unknown1005"}, // 6.0.0+
            {1006, nullptr, "Unknown1006"}, // 6.0.0+
            {1007, nullptr, "Unknown1007"}, // 6.0.0+
            {1009, nullptr, "Unknown1009"}, // 6.0.0+
            {1010, nullptr, "Unknown1010"}, // 6.0.0+
            {1011, nullptr, "Unknown1011"}, // 6.0.0+
            {1012, nullptr, "Unknown1012"}, // 6.0.0+
            {1013, nullptr, "Unknown1013"}, // 6.0.0+
            {1014, nullptr, "Unknown1014"}, // 6.0.0+
            {1015, nullptr, "Unknown1015"}, // 6.0.0+
            {1016, nullptr, "Unknown1016"}, // 6.0.0+
            {1017, nullptr, "Unknown1017"}, // 9.0.0+
            {1018, nullptr, "Unknown1018"}, // 9.0.0+
            {1019, nullptr, "Unknown1019"}, // 9.0.0+
            {1020, nullptr, "Unknown1020"}, // 9.0.0+
            {1021, nullptr, "Unknown1021"}, // 9.0.0+
            {1022, nullptr, "Unknown1022"}, // 15.0.0+
            {1023, nullptr, "Unknown1023"}, // 17.0.0+
            {1024, nullptr, "Unknown1024"}, // 17.0.0+
            {1025, nullptr, "Unknown1025"}, // 17.0.0+
            {1026, nullptr, "Unknown1026"}, // 17.0.0+
            {1027, nullptr, "Unknown1027"}, // 17.0.0+
            {1028, nullptr, "Unknown1028"}, // 18.0.0+
            {1029, nullptr, "Unknown1029"}, // 19.0.0+
            {1030, nullptr, "Unknown1030"}, // 20.0.0+
            {1031, nullptr, "Unknown1031"}, // 20.0.0+
            {1032, nullptr, "Unknown1032"}, // 20.0.0+
            {1033, nullptr, "Unknown1033"}, // 20.0.0+
            {1034, nullptr, "Unknown1034"}, // 20.0.0+
            {1035, nullptr, "Unknown1035"}, // 20.0.0+
            {1036, nullptr, "Unknown1036"}, // 20.0.0+
            {1037, nullptr, "Unknown1037"}, // 20.0.0+
            {1501, nullptr, "Unknown1501"}, // 6.0.0+
            {1502, nullptr, "Unknown1502"}, // 6.0.0+
            {1503, nullptr, "Unknown1503"}, // 6.0.0+
            {1504, nullptr, "Unknown1504"}, // 6.0.0+
            {1505, nullptr, "Unknown1505"}, // 6.0.0+
            {1506, nullptr, "Unknown1506"}, // 13.0.0+
            {1601, nullptr, "Unknown1601"}, // 20.0.0+
            {1602, nullptr, "Unknown1602"}, // 20.0.0+
            {1603, nullptr, "Unknown1603"}, // 20.0.0+
            {1604, nullptr, "Unknown1604"}, // 20.0.0+
            {1605, nullptr, "Unknown1605"}, // 20.0.0+
            {1606, nullptr, "Unknown1606"}, // 20.0.0+
            {2000, nullptr, "GetIActiveRightsContext"}, // 6.0.0+
            {2001, nullptr, "GetIActiveRightsContext2"}, // 9.0.0+
            {2002, nullptr, "Unknown2002"}, // 13.0.0-16.1.0
            {2003, nullptr, "Unknown2003"}, // 13.0.0-16.1.0
            {2100, nullptr, "Unknown2100"}, // 7.0.0+
            {2501, nullptr, "Unknown2501"}, // 6.0.0+
            {2502, nullptr, "Unknown2502"}, // 6.0.0+
            {2601, nullptr, "Unknown2601"}, // 13.0.0+
            {3001, nullptr, "Unknown3001"}, // 7.0.0-15.0.1
            {3002, nullptr, "Unknown3002"} // 7.0.0-15.0.1
        };
        // clang-format on
        RegisterHandlers(functions);

        keys.PopulateTickets();
        keys.SynthesizeTickets();
    }

private:
    bool CheckRightsId(HLERequestContext& ctx, const u128& rights_id) {
        if (rights_id == u128{}) {
            LOG_ERROR(Service_ETicket, "The rights ID was invalid!");
            IPC::ResponseBuilder rb{ctx, 2};
            rb.Push(ERROR_INVALID_RIGHTS_ID);
            return false;
        }

        return true;
    }

    void ImportTicket(HLERequestContext& ctx) {
        const auto raw_ticket = ctx.ReadBuffer();
        [[maybe_unused]] const auto cert = ctx.ReadBuffer(1);

        if (raw_ticket.size() < sizeof(Core::Crypto::Ticket)) {
            LOG_ERROR(Service_ETicket, "The input buffer is not large enough!");
            IPC::ResponseBuilder rb{ctx, 2};
            rb.Push(ERROR_INVALID_ARGUMENT);
            return;
        }

        Core::Crypto::Ticket ticket = Core::Crypto::Ticket::Read(raw_ticket);
        if (!keys.AddTicket(ticket)) {
            LOG_ERROR(Service_ETicket, "The ticket could not be imported!");
            IPC::ResponseBuilder rb{ctx, 2};
            rb.Push(ERROR_INVALID_ARGUMENT);
            return;
        }

        IPC::ResponseBuilder rb{ctx, 2};
        rb.Push(ResultSuccess);
    }

    void GetTitleKey(HLERequestContext& ctx) {
        IPC::RequestParser rp{ctx};
        const auto rights_id = rp.PopRaw<u128>();

        LOG_DEBUG(Service_ETicket, "called, rights_id={:016X}{:016X}", rights_id[1], rights_id[0]);

        if (!CheckRightsId(ctx, rights_id))
            return;

        const auto key =
            keys.GetKey(Core::Crypto::S128KeyType::Titlekey, rights_id[1], rights_id[0]);

        if (key == Core::Crypto::Key128{}) {
            LOG_ERROR(Service_ETicket,
                      "The titlekey doesn't exist in the KeyManager or the rights ID was invalid!");
            IPC::ResponseBuilder rb{ctx, 2};
            rb.Push(ERROR_INVALID_RIGHTS_ID);
            return;
        }

        ctx.WriteBuffer(key);

        IPC::ResponseBuilder rb{ctx, 2};
        rb.Push(ResultSuccess);
    }

    void CountCommonTicket(HLERequestContext& ctx) {
        LOG_DEBUG(Service_ETicket, "called");

        const u32 count = static_cast<u32>(keys.GetCommonTickets().size());

        IPC::ResponseBuilder rb{ctx, 3};
        rb.Push(ResultSuccess);
        rb.Push<u32>(count);
    }

    void CountPersonalizedTicket(HLERequestContext& ctx) {
        LOG_DEBUG(Service_ETicket, "called");

        const u32 count = static_cast<u32>(keys.GetPersonalizedTickets().size());

        IPC::ResponseBuilder rb{ctx, 3};
        rb.Push(ResultSuccess);
        rb.Push<u32>(count);
    }

    void ListCommonTicketRightsIds(HLERequestContext& ctx) {
        size_t out_entries = 0;
        if (!keys.GetCommonTickets().empty()) {
            out_entries = ctx.GetWriteBufferNumElements<u128>();
        }
        LOG_DEBUG(Service_ETicket, "called, entries={:016X}", out_entries);

        keys.PopulateTickets();
        const auto tickets = keys.GetCommonTickets();
        std::vector<u128> ids;
        std::transform(tickets.begin(), tickets.end(), std::back_inserter(ids),
                       [](const auto& pair) { return pair.first; });

        out_entries = std::min(ids.size(), out_entries);
        ctx.WriteBuffer(ids.data(), out_entries * sizeof(u128));

        IPC::ResponseBuilder rb{ctx, 3};
        rb.Push(ResultSuccess);
        rb.Push<u32>(static_cast<u32>(out_entries));
    }

    void ListPersonalizedTicketRightsIds(HLERequestContext& ctx) {
        size_t out_entries = 0;
        if (!keys.GetPersonalizedTickets().empty()) {
            out_entries = ctx.GetWriteBufferNumElements<u128>();
        }

        LOG_DEBUG(Service_ETicket, "called, entries={:016X}", out_entries);

        keys.PopulateTickets();
        const auto tickets = keys.GetPersonalizedTickets();
        std::vector<u128> ids;
        std::transform(tickets.begin(), tickets.end(), std::back_inserter(ids),
                       [](const auto& pair) { return pair.first; });

        out_entries = std::min(ids.size(), out_entries);
        ctx.WriteBuffer(ids.data(), out_entries * sizeof(u128));

        IPC::ResponseBuilder rb{ctx, 3};
        rb.Push(ResultSuccess);
        rb.Push<u32>(static_cast<u32>(out_entries));
    }

    void GetCommonTicketSize(HLERequestContext& ctx) {
        IPC::RequestParser rp{ctx};
        const auto rights_id = rp.PopRaw<u128>();

        LOG_DEBUG(Service_ETicket, "called, rights_id={:016X}{:016X}", rights_id[1], rights_id[0]);

        if (!CheckRightsId(ctx, rights_id))
            return;

        const auto ticket = keys.GetCommonTickets().at(rights_id);

        IPC::ResponseBuilder rb{ctx, 4};
        rb.Push(ResultSuccess);
        rb.Push<u64>(ticket.GetSize());
    }

    void GetPersonalizedTicketSize(HLERequestContext& ctx) {
        IPC::RequestParser rp{ctx};
        const auto rights_id = rp.PopRaw<u128>();

        LOG_DEBUG(Service_ETicket, "called, rights_id={:016X}{:016X}", rights_id[1], rights_id[0]);

        if (!CheckRightsId(ctx, rights_id))
            return;

        const auto ticket = keys.GetPersonalizedTickets().at(rights_id);

        IPC::ResponseBuilder rb{ctx, 4};
        rb.Push(ResultSuccess);
        rb.Push<u64>(ticket.GetSize());
    }

    void GetCommonTicketData(HLERequestContext& ctx) {
        IPC::RequestParser rp{ctx};
        const auto rights_id = rp.PopRaw<u128>();

        LOG_DEBUG(Service_ETicket, "called, rights_id={:016X}{:016X}", rights_id[1], rights_id[0]);

        if (!CheckRightsId(ctx, rights_id))
            return;

        const auto ticket = keys.GetCommonTickets().at(rights_id);

        const auto write_size = std::min<u64>(ticket.GetSize(), ctx.GetWriteBufferSize());
        ctx.WriteBuffer(&ticket, write_size);

        IPC::ResponseBuilder rb{ctx, 4};
        rb.Push(ResultSuccess);
        rb.Push<u64>(write_size);
    }

    void GetPersonalizedTicketData(HLERequestContext& ctx) {
        IPC::RequestParser rp{ctx};
        const auto rights_id = rp.PopRaw<u128>();

        LOG_DEBUG(Service_ETicket, "called, rights_id={:016X}{:016X}", rights_id[1], rights_id[0]);

        if (!CheckRightsId(ctx, rights_id))
            return;

        const auto ticket = keys.GetPersonalizedTickets().at(rights_id);

        const auto write_size = std::min<u64>(ticket.GetSize(), ctx.GetWriteBufferSize());
        ctx.WriteBuffer(&ticket, write_size);

        IPC::ResponseBuilder rb{ctx, 4};
        rb.Push(ResultSuccess);
        rb.Push<u64>(write_size);
    }

    Core::Crypto::KeyManager& keys = Core::Crypto::KeyManager::Instance();
};

void LoopProcess(Core::System& system) {
    auto server_manager = std::make_unique<ServerManager>(system);

    server_manager->RegisterNamedService("es", std::make_shared<ETicket>(system));
    ServerManager::RunServer(std::move(server_manager));
}

} // namespace Service::ES