Atmosphere/stratosphere/ams_mitm/source/sysupdater/sysupdater_thread_allocator.cpp

/*
 * Copyright (c) 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 <stratosphere.hpp>
#include "sysupdater_thread_allocator.hpp"

namespace ams::mitm::sysupdater {

    Result ThreadAllocator::Allocate(ThreadInfo *out) {
        std::scoped_lock lk(m_mutex);

        for (int i = 0; i < m_thread_count; ++i) {
            const u64 mask = (static_cast<u64>(1) << i);
            if ((m_bitmap & mask) == 0) {
                *out = {
                    .thread     = m_thread_list + i,
                    .priority   = m_thread_priority,
                    .stack      = m_stack_heap + (m_stack_size * i),
                    .stack_size = m_stack_size,
                };
                m_bitmap |= mask;
                R_SUCCEED();
            }
        }

        R_THROW(ns::ResultOutOfMaxRunningTask());
    }

    void ThreadAllocator::Free(const ThreadInfo &info) {
        std::scoped_lock lk(m_mutex);

        for (int i = 0; i < m_thread_count; ++i) {
            if (info.thread == std::addressof(m_thread_list[i])) {
                const u64 mask = (static_cast<u64>(1) << i);
                m_bitmap &= ~mask;
                return;
            }
        }

        AMS_ABORT("Invalid thread passed to ThreadAllocator::Free");
    }

}