Skip to content

Commit

Permalink
Rearrange OS folder
Browse files Browse the repository at this point in the history
  • Loading branch information
PixelyIon committed Aug 21, 2019
1 parent 7dd04af commit da55a1d
Show file tree
Hide file tree
Showing 14 changed files with 62 additions and 102 deletions.
1 change: 0 additions & 1 deletion app/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,6 @@ add_library(lightswitch SHARED
${source_DIR}/lightswitch.cpp
${source_DIR}/switch/os/os.cpp
${source_DIR}/switch/os/ipc.cpp
${source_DIR}/switch/os/kernel.cpp
${source_DIR}/switch/os/svc.cpp
${source_DIR}/switch/hw/cpu.cpp
${source_DIR}/switch/hw/memory.cpp
Expand Down
11 changes: 7 additions & 4 deletions app/src/main/cpp/lightswitch.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
#include "switch/common.h"

std::thread *emu_thread;
bool halt = false;

void thread_main(std::string rom_path, std::string pref_path, std::string log_path) {
auto log = std::make_shared<lightSwitch::Logger>(log_path);
Expand All @@ -27,16 +28,18 @@ void thread_main(std::string rom_path, std::string pref_path, std::string log_pa

extern "C"
JNIEXPORT void JNICALL
Java_emu_lightswitch_MainActivity_loadFile(JNIEnv *env, jobject instance, jstring rom_path_,
jstring pref_path_, jstring log_path_) {
Java_emu_lightswitch_MainActivity_loadFile(JNIEnv *env, jobject instance, jstring rom_path_, jstring pref_path_, jstring log_path_) {
const char *rom_path = env->GetStringUTFChars(rom_path_, 0);
const char *pref_path = env->GetStringUTFChars(pref_path_, 0);
const char *log_path = env->GetStringUTFChars(log_path_, 0);

if (emu_thread) pthread_kill(emu_thread->native_handle(), SIGABRT);

if (emu_thread) {
halt=true; // This'll cause execution to stop after the next breakpoint
emu_thread->join();
}
// Running on UI thread is not a good idea, any crashes and such will be propagated
emu_thread = new std::thread(thread_main, std::string(rom_path, strlen(rom_path)), std::string(pref_path, strlen(pref_path)), std::string(log_path, strlen(log_path)));

env->ReleaseStringUTFChars(rom_path_, rom_path);
env->ReleaseStringUTFChars(pref_path_, pref_path);
env->ReleaseStringUTFChars(log_path_, log_path);
Expand Down
4 changes: 1 addition & 3 deletions app/src/main/cpp/switch/common.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -66,10 +66,8 @@ namespace lightSwitch {
}

void Logger::write(Logger::LogLevel level, std::string str) {
#ifdef NDEBUG
if (level == DEBUG)
if (level == DEBUG && debug_build)
return;
#endif
syslog(level_syslog[level], "%s", str.c_str());
log_file << "1|" << level_str[level] << "|" << str << "\n";
log_file.flush();
Expand Down
1 change: 0 additions & 1 deletion app/src/main/cpp/switch/common.h
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@
#include <fmt/format.h>
#include "hw/cpu.h"
#include "hw/memory.h"
#include "constant.h"

namespace lightSwitch {
class Settings {
Expand Down
6 changes: 6 additions & 0 deletions app/src/main/cpp/switch/constant.h
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,12 @@
namespace lightSwitch {
typedef std::runtime_error exception;

#ifdef NDEBUG
constexpr bool debug_build = 0;
#else
constexpr bool debug_build = 1;
#endif

namespace constant {
constexpr uint64_t base_addr = 0x80000000;
constexpr uint64_t stack_addr = 0x3000000;
Expand Down
2 changes: 1 addition & 1 deletion app/src/main/cpp/switch/device.h
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ namespace lightSwitch {
if (rom_ext == "nro") loader::NroLoader loader(rom_file, state);
else throw exception("Unsupported ROM extension.");

cpu->Execute(hw::Memory::text, memory, os.SvcHandler, &state);
cpu->Execute(hw::Memory::text, memory, std::bind(&os::OS::SvcHandler, std::ref(os), std::placeholders::_1, std::placeholders::_2), &state);
}
};
};
61 changes: 23 additions & 38 deletions app/src/main/cpp/switch/hw/cpu.cpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
#include "cpu.h"
#include "../constant.h"
extern bool halt;

namespace lightSwitch::hw {
Cpu::~Cpu() {
Expand All @@ -8,30 +8,30 @@ namespace lightSwitch::hw {

long *Cpu::ReadMemory(uint64_t address) { // Return a single word (32-bit)
status = ptrace(PTRACE_PEEKDATA, child, address, NULL);
if (status == -1) throw std::runtime_error("Cannot read memory!");
if (status == -1) throw std::runtime_error("Cannot read memory");
return &status;
}

void Cpu::WriteMemory(uint64_t address) { // Write a single word (32-bit)
status = ptrace(PTRACE_GETREGSET, child, NT_PRSTATUS, &iov);
if (status == -1) throw std::runtime_error("Cannot write memory!");
if (status == -1) throw std::runtime_error("Cannot write memory");
}

void Cpu::ReadRegisters() { // Read all registers into 'regs'
iov = {&regs, sizeof(regs)};
status = ptrace(PTRACE_GETREGSET, child, NT_PRSTATUS, &iov);
if (status == -1) throw std::runtime_error("Cannot read registers!");
if (status == -1) throw std::runtime_error("Cannot read registers");
}

void Cpu::WriteRegisters() { // Write all registers from 'regs'
iov = {&regs, sizeof(regs)};
status = ptrace(PTRACE_SETREGSET, child, NT_PRSTATUS, &iov);
if (status == -1) throw std::runtime_error("Cannot write registers!");
if (status == -1) throw std::runtime_error("Cannot write registers");
}

void Cpu::ResumeProcess() { // Resumes a process stopped due to a signal
status = ptrace(PTRACE_CONT, child, NULL, NULL);
if (status == -1) throw std::runtime_error("Cannot resume process!");
if (status == -1) throw std::runtime_error("Cannot resume process");
}

void Cpu::WriteBreakpoint(uint64_t address_, uint64_t size) {
Expand All @@ -41,68 +41,54 @@ namespace lightSwitch::hw {
auto instr_mrs = reinterpret_cast<instr::mrs *>(address + iter);

if (instr_svc->verify()) {
#ifdef NDEBUG
syslog(LOG_WARNING, "Found SVC call: 0x%X, At location 0x%X", instr_svc->value, ((uint64_t)address)+iter);
#endif

if(debug_build)
syslog(LOG_WARNING, "Found SVC call: 0x%X, At location 0x%X", instr_svc->value, address+iter);
instr::brk brk(static_cast<uint16_t>(instr_svc->value));
address[iter] = *reinterpret_cast<uint32_t *>(&brk);
} else if (instr_mrs->verify() && instr_mrs->src_reg == constant::tpidrro_el0) {
#ifdef NDEBUG
syslog(LOG_WARNING, "Found MRS call: 0x%X, At location 0x%X", instr_mrs->dst_reg, ((uint64_t)address)+iter);
#endif

if(debug_build)
syslog(LOG_WARNING, "Found MRS call: 0x%X, At location 0x%X", instr_mrs->dst_reg, address+iter);
instr::brk brk(static_cast<uint16_t>(constant::svc_last + 1 + instr_mrs->dst_reg));
address[iter] = *reinterpret_cast<uint32_t *>(&brk);
}
}
}

void Cpu::Execute(Memory::Region region, std::shared_ptr<Memory> memory, std::function<void(uint16_t, void *)> svc_handler, void *device) {
tls = memory->region_map.at(hw::Memory::tls).address;

hw::Memory::RegionData exec = memory->region_map.at(hw::Memory::text);
WriteBreakpoint(exec.address, exec.size);

hw::Memory::RegionData exec = memory->region_map.at(region);
WriteBreakpoint(exec.address, exec.size); // We write the BRK instructions to replace SVC & MRS so we receive a breakpoint
child = ExecuteChild(exec.address);

int stat = 0;
while (waitpid(child, &stat, 0)) {
if (WIFSTOPPED(stat)) {
while (waitpid(child, &pid_status, 0)) {
if (WIFSTOPPED(pid_status)) {
ReadRegisters();

#ifdef NDEBUG
syslog(LOG_INFO, "PC is at 0x%X", regs.pc);
#endif

if(debug_build)
syslog(LOG_INFO, "PC is at 0x%X", regs.pc);
if (!regs.pc || regs.pc == 0xBADC0DE) break;

// We store the instruction value as the immediate value. 0x0 to 0x7F are SVC, 0x80 to 0x9E is MRS for TPIDRRO_EL0.
auto instr = reinterpret_cast<instr::brk *>(ReadMemory(regs.pc));
if (instr->verify()) {
if (instr->value <= constant::svc_last) {
svc_handler(static_cast<uint16_t>(instr->value), device);
syslog(LOG_ERR, "SVC has been called 0x%X", instr->value);

if (debug_build)
syslog(LOG_ERR, "SVC has been called 0x%X", instr->value);
if (halt) break;
} else if (instr->value > constant::svc_last && instr->value <= constant::svc_last + constant::num_regs) {
// Catch MRS that reads the value of TPIDRRO_EL0 (TLS)
// https://switchbrew.org/wiki/Thread_Local_Storage
SetRegister(xreg(instr->value - (constant::svc_last + 1)), tls);
syslog(LOG_ERR, "MRS has been called 0x%X", instr->value - (constant::svc_last + 1));
if (debug_build)
syslog(LOG_ERR, "MRS has been called 0x%X", instr->value - (constant::svc_last + 1));
} else syslog(LOG_ERR, "Received unhandled BRK 0x%X", instr->value);
}

regs.pc += 4; // Increment program counter by a single instruction (32 bits)
WriteRegisters();
} else if (WIFEXITED(stat)) break;

} else if (WIFEXITED(pid_status)) break;
ResumeProcess();
}

kill(child, SIGABRT);
child = 0;
halt = false; // TODO: Global variable
pid_status = 0;
halt = false;
}

pid_t Cpu::ExecuteChild(uint64_t address) {
Expand All @@ -111,7 +97,6 @@ namespace lightSwitch::hw {
ptrace(PTRACE_TRACEME, 0, NULL, NULL);
asm volatile("br %0" :: "r"(address));
}

return pid;
}

Expand Down
6 changes: 3 additions & 3 deletions app/src/main/cpp/switch/hw/cpu.h
Original file line number Diff line number Diff line change
Expand Up @@ -7,18 +7,18 @@
#include <sys/wait.h>
#include <linux/uio.h>
#include <linux/elf.h>
#include <switch/constant.h>
#include "../constant.h"
#include "memory.h"

namespace lightSwitch::hw {
class Cpu {
private:
bool halt = false;
long status = 0;
int pid_status = 0;
pid_t child;
iovec iov;
user_pt_regs regs;
uint64_t tls;
uint64_t tls = constant::tls_addr;

static pid_t ExecuteChild(uint64_t address);

Expand Down
1 change: 0 additions & 1 deletion app/src/main/cpp/switch/hw/memory.cpp
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
#include <sys/mman.h>
#include <cerrno>
#include "memory.h"
#include "../constant.h"

namespace lightSwitch::hw {
Memory::Memory() {
Expand Down
1 change: 0 additions & 1 deletion app/src/main/cpp/switch/os/ipc.h
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
#pragma once

#include <cstdint>
#include "switch/common.h"

namespace lightSwitch::os::ipc {
Expand Down
11 changes: 0 additions & 11 deletions app/src/main/cpp/switch/os/kernel.cpp

This file was deleted.

30 changes: 0 additions & 30 deletions app/src/main/cpp/switch/os/kernel.h

This file was deleted.

6 changes: 4 additions & 2 deletions app/src/main/cpp/switch/os/os.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,9 @@ namespace lightSwitch::os {
}
}

void OS::HandleSvc(uint16_t svc) {
SvcHandler(svc, &state);
uint32_t OS::NewHandle(KObjectPtr obj) {
handles.insert({handle_index, obj});
state.logger->write(Logger::DEBUG, "Creating new handle 0x{0:x}", handle_index);
return handle_index++;
}
}
23 changes: 17 additions & 6 deletions app/src/main/cpp/switch/os/os.h
Original file line number Diff line number Diff line change
@@ -1,21 +1,32 @@
#pragma once

#include <cstdint>
#include <err.h>
#include "switch/common.h"
#include <unordered_map>
#include "../common.h"
#include "ipc.h"
#include "kernel.h"
#include "svc.h"

namespace lightSwitch::os {
class KObject {
private:
uint32_t handle;
public:
KObject(uint32_t handle) : handle(handle) {}

uint32_t Handle() { return handle; }
};

typedef std::shared_ptr<KObject> KObjectPtr;

class OS {
private:
device_state state;
uint32_t handle_index = constant::base_handle_index;
std::unordered_map<uint32_t, KObjectPtr> handles;
public:
OS(device_state state_);

static void SvcHandler(uint16_t svc, void *vstate);
void SvcHandler(uint16_t svc, void *vstate);

void HandleSvc(uint16_t svc);
uint32_t NewHandle(KObjectPtr obj);
};
}

0 comments on commit da55a1d

Please sign in to comment.