mirror of
https://github.com/Atmosphere-NX/Atmosphere.git
synced 2024-12-23 18:56:03 +00:00
102 lines
2.9 KiB
C++
102 lines
2.9 KiB
C++
/*
|
|
* Copyright (c) 2018-2019 Atmosphère-NX
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms and conditions of the GNU General Public License,
|
|
* version 2, as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
|
* more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <switch.h>
|
|
#include <stratosphere.hpp>
|
|
#include <algorithm>
|
|
#include <array>
|
|
#include <cstdio>
|
|
#include "ldr_launch_queue.hpp"
|
|
#include "meta_tools.hpp"
|
|
|
|
static std::array<LaunchQueue::LaunchItem, LAUNCH_QUEUE_SIZE> g_launch_queue = {};
|
|
|
|
Result LaunchQueue::Add(u64 tid, const char *args, u64 arg_size) {
|
|
if (arg_size > LAUNCH_QUEUE_ARG_SIZE_MAX) {
|
|
return ResultLoaderTooLongArgument;
|
|
}
|
|
|
|
int idx = GetFreeIndex(tid);
|
|
if (idx == LAUNCH_QUEUE_FULL) {
|
|
return ResultLoaderTooManyArguments;
|
|
}
|
|
|
|
g_launch_queue[idx].tid = tid;
|
|
g_launch_queue[idx].arg_size = arg_size;
|
|
|
|
std::copy(args, args + arg_size, g_launch_queue[idx].args);
|
|
return ResultSuccess;
|
|
}
|
|
|
|
Result LaunchQueue::AddCopy(u64 tid_base, u64 tid) {
|
|
int idx = GetIndex(tid_base);
|
|
if (idx == LAUNCH_QUEUE_FULL) {
|
|
return ResultSuccess;
|
|
}
|
|
|
|
return Add(tid, g_launch_queue[idx].args, g_launch_queue[idx].arg_size);
|
|
}
|
|
|
|
|
|
Result LaunchQueue::AddItem(const LaunchItem *item) {
|
|
if (item->arg_size > LAUNCH_QUEUE_ARG_SIZE_MAX) {
|
|
return ResultLoaderTooLongArgument;
|
|
}
|
|
|
|
int idx = GetFreeIndex(item->tid);
|
|
if (idx == LAUNCH_QUEUE_FULL) {
|
|
return ResultLoaderTooManyArguments;
|
|
}
|
|
|
|
g_launch_queue[idx] = *item;
|
|
return ResultSuccess;
|
|
}
|
|
|
|
int LaunchQueue::GetIndex(u64 tid) {
|
|
auto it = std::find_if(g_launch_queue.begin(), g_launch_queue.end(), member_equals_fn(&LaunchQueue::LaunchItem::tid, tid));
|
|
if (it == g_launch_queue.end()) {
|
|
return LAUNCH_QUEUE_FULL;
|
|
}
|
|
return std::distance(g_launch_queue.begin(), it);
|
|
}
|
|
|
|
int LaunchQueue::GetFreeIndex(u64 tid) {
|
|
for (unsigned int i = 0; i < LAUNCH_QUEUE_SIZE; i++) {
|
|
if (g_launch_queue[i].tid == tid || g_launch_queue[i].tid == 0x0) {
|
|
return i;
|
|
}
|
|
}
|
|
return LAUNCH_QUEUE_FULL;
|
|
}
|
|
|
|
bool LaunchQueue::Contains(u64 tid) {
|
|
return GetIndex(tid) != LAUNCH_QUEUE_FULL;
|
|
}
|
|
|
|
void LaunchQueue::Clear() {
|
|
for (unsigned int i = 0; i < LAUNCH_QUEUE_SIZE; i++) {
|
|
g_launch_queue[i].tid = 0;
|
|
}
|
|
}
|
|
|
|
|
|
LaunchQueue::LaunchItem *LaunchQueue::GetItem(u64 tid) {
|
|
int idx = GetIndex(tid);
|
|
if (idx == LAUNCH_QUEUE_FULL) {
|
|
return NULL;
|
|
}
|
|
return &g_launch_queue[idx];
|
|
}
|