nstool/lib/nx/SystemCallHandler.cpp

#include "SystemCallHandler.h"
#include "SystemCallEntry.h"


nx::SystemCallHandler::SystemCallHandler() :
	mIsSet(false),
	mSystemCalls()
{}

bool nx::SystemCallHandler::operator==(const SystemCallHandler & other) const
{
	return isEqual(other);
}

bool nx::SystemCallHandler::operator!=(const SystemCallHandler & other) const
{
	return !isEqual(other);
}

void nx::SystemCallHandler::operator=(const SystemCallHandler & other)
{
	copyFrom(other);
}

void nx::SystemCallHandler::importKernelCapabilityList(const fnd::List<KernelCapability>& caps)
{
	SystemCallEntry entry;

	u8 syscallUpper, syscall;
	for (size_t i = 0; i < caps.getSize(); i++)
	{
		entry.setKernelCapability(caps[i]);

		syscallUpper = 24 * entry.getSystemCallUpperBits();
		for (u8 j = 0; j < 24; j++)
		{
			syscall = syscallUpper + j;

			if (((entry.getSystemCallLowerBits() >> j) & 1) == 1)
			{
				mSystemCalls.hasElement(syscall) == false ? mSystemCalls.addElement(syscall) : throw fnd::Exception(kModuleName, "SystemCall already added");
			}
		}
	}


	mIsSet = true;
}

void nx::SystemCallHandler::exportKernelCapabilityList(fnd::List<KernelCapability>& caps) const
{
	if (isSet() == false)
		return;

	fnd::List<SystemCallEntry> entries;
	for (size_t i = 0; i < kSyscallTotalEntryNum; i++)
	{
		entries[i].setSystemCallUpperBits(i);
		entries[i].setSystemCallLowerBits(0);
	}

	for (size_t i = 0; i < mSystemCalls.getSize(); i++)
	{
		if (mSystemCalls[i] > kMaxSystemCall)
		{
			throw fnd::Exception(kModuleName, "Illegal SystemCall. (range: 0x00-0xBF inclusive)");
		}

		entries[mSystemCalls[i] / 24].setSystemCallLowerBits(entries[mSystemCalls[i] / 24].getSystemCallLowerBits() | BIT(mSystemCalls[i] % 24));
	}

	for (size_t i = 0; i < entries.getSize(); i++)
	{
		if (entries[i].getSystemCallLowerBits() != 0)
		{
			caps.addElement(entries[i].getKernelCapability());
		}
	}
}

void nx::SystemCallHandler::clear()
{
	mIsSet = false;
	mSystemCalls.clear();
}

bool nx::SystemCallHandler::isSet() const
{
	return mIsSet;
}

const fnd::List<u8>& nx::SystemCallHandler::getSystemCalls() const
{
	return mSystemCalls;
}

void nx::SystemCallHandler::setSystemCalls(const fnd::List<u8>& calls)
{
	mSystemCalls.clear();
	for (size_t i = 0; i < calls.getSize(); i++)
	{
		if (mSystemCalls[i] > kMaxSystemCall)
		{
			throw fnd::Exception(kModuleName, "Illegal SystemCall. (range: 0x00-0xBF inclusive)");
		}

		mSystemCalls.hasElement(calls[i]) == false ? mSystemCalls.addElement(calls[i]) : throw fnd::Exception(kModuleName, "SystemCall already added");
	}

	mIsSet = true;
}

void nx::SystemCallHandler::copyFrom(const SystemCallHandler & other)
{
	mIsSet = other.mIsSet;
	mSystemCalls = other.mSystemCalls;
}

bool nx::SystemCallHandler::isEqual(const SystemCallHandler & other) const
{
	return (mIsSet == other.mIsSet) \
		&& (mSystemCalls == other.mSystemCalls);
}