#include "i386.h"
#include "Process.h"
#include "Syscall.h"
#include "Console.h"

extern "C" void syscall_entry(RegisterDump&);
extern "C" void syscall_ISR();
extern volatile RegisterDump* syscallRegDump;

asm(
    ".globl syscall_ISR \n"
    "syscall_ISR:\n"
    "    pusha\n"
    "    pushw %ds\n"
    "    pushw %es\n"
    "    pushw %fs\n"
    "    pushw %gs\n"
    "    pushw %ss\n"
    "    pushw %ss\n"
    "    pushw %ss\n"
    "    pushw %ss\n"
    "    pushw %ss\n"
    "    popw %ds\n"
    "    popw %es\n"
    "    popw %fs\n"
    "    popw %gs\n"
    "    mov %esp, %eax\n"
    "    call syscall_entry\n"
    "    popw %gs\n"
    "    popw %gs\n"
    "    popw %fs\n"
    "    popw %es\n"
    "    popw %ds\n"
    "    popa\n"
    "    iret\n"
);

namespace Syscall {

void initialize()
{
    registerUserCallableInterruptHandler(0x80, syscall_ISR);
    kprintf("syscall: int 0x80 handler installed\n");
}

static DWORD handle(RegisterDump& regs, DWORD function, DWORD arg1, DWORD arg2, DWORD arg3)
{
    ASSERT_INTERRUPTS_ENABLED();
    switch (function) {
    case Syscall::Yield:
        yield();
        break;
    case Syscall::PutCharacter:
        Console::the().putChar(arg1 & 0xff);
        break;
    case Syscall::Sleep:
        return current->sys$sleep(arg1);
    case Syscall::PosixGettimeofday:
        return current->sys$gettimeofday((timeval*)arg1);
    case Syscall::Spawn:
        return current->sys$spawn((const char*)arg1, (const char**)arg2, (const char**)arg3);
    case Syscall::GetDirEntries:
        return current->sys$get_dir_entries((int)arg1, (void*)arg2, (size_t)arg3);
    case Syscall::PosixLstat:
        return current->sys$lstat((const char*)arg1, (Unix::stat*)arg2);
    case Syscall::PosixStat:
        return current->sys$stat((const char*)arg1, (Unix::stat*)arg2);
    case Syscall::PosixGetcwd:
        return current->sys$getcwd((char*)arg1, (size_t)arg2);
    case Syscall::PosixOpen:
        return current->sys$open((const char*)arg1, (int)arg2);
    case Syscall::PosixWrite:
        return current->sys$write((int)arg1, (const void*)arg2, (size_t)arg3);
    case Syscall::PosixClose:
        //kprintf("syscall: close(%d)\n", arg1);
        return current->sys$close((int)arg1);
    case Syscall::PosixRead:
        //kprintf("syscall: read(%d, %p, %u)\n", arg1, arg2, arg3);
        return current->sys$read((int)arg1, (void*)arg2, (size_t)arg3);
    case Syscall::PosixLseek:
        return current->sys$lseek((int)arg1, (off_t)arg2, (int)arg3);
    case Syscall::PosixKill:
        return current->sys$kill((pid_t)arg1, (int)arg2);
    case Syscall::PosixGetuid:
        return current->sys$getuid();
    case Syscall::PosixGetgid:
        return current->sys$getgid();
    case Syscall::PosixGetpid:
        return current->sys$getpid();
    case Syscall::PosixWaitpid:
        return current->sys$waitpid((pid_t)arg1, (int*)arg2, (int)arg3);
    case Syscall::PosixMmap:
        return (dword)current->sys$mmap((void*)arg1, (size_t)arg2);
    case Syscall::PosixMunmap:
        return current->sys$munmap((void*)arg1, (size_t)arg2);
    case Syscall::PosixGethostname:
        return current->sys$gethostname((char*)arg1, (size_t)arg2);
    case Syscall::PosixExit:
        cli();
        current->sys$exit((int)arg1);
        ASSERT_NOT_REACHED();
        return 0;
    case Syscall::GetArguments:
        return current->sys$get_arguments((int*)arg1, (char***)arg2);
    case Syscall::GetEnvironment:
        return current->sys$get_environment((char***)arg1);
    case Syscall::PosixChdir:
        return current->sys$chdir((const char*)arg1);
    case Syscall::PosixUname:
        return current->sys$uname((utsname*)arg1);
    case Syscall::SetMmapName:
        return current->sys$set_mmap_name((void*)arg1, (size_t)arg2, (const char*)arg3);
    case Syscall::PosixReadlink:
        return current->sys$readlink((const char*)arg1, (char*)arg2, (size_t)arg3);
    case Syscall::PosixTtynameR:
        return current->sys$ttyname_r((int)arg1, (char*)arg2, (size_t)arg3);
    case Syscall::PosixSetsid:
        return current->sys$setsid();
    case Syscall::PosixGetsid:
        return current->sys$getsid((pid_t)arg1);
    case Syscall::PosixSetpgid:
        return current->sys$setpgid((pid_t)arg1, (pid_t)arg2);
    case Syscall::PosixGetpgid:
        return current->sys$getpgid((pid_t)arg1);
    case Syscall::PosixGetpgrp:
        return current->sys$getpgrp();
    case Syscall::PosixTcgetpgrp:
        return current->sys$tcgetpgrp((int)arg1);
    case Syscall::PosixTcsetpgrp:
        return current->sys$tcsetpgrp((int)arg1, (pid_t)arg2);
    case Syscall::PosixFork:
        return current->sys$fork(regs);
    case Syscall::PosixExecve:
        return current->sys$execve((const char*)arg1, (const char**)arg2, (const char**)arg3);
    case Syscall::PosixGeteuid:
        return current->sys$geteuid();
    case Syscall::PosixGetegid:
        return current->sys$getegid();
    case Syscall::PosixSignal:
        return (dword)current->sys$signal((int)arg1, (Unix::sighandler_t)arg2);
    case Syscall::PosixIsatty:
        return current->sys$isatty((int)arg1);
    case Syscall::Getdtablesize:
        return current->sys$getdtablesize();
    case Syscall::Dup:
        return current->sys$dup((int)arg1);
    case Syscall::Dup2:
        return current->sys$dup2((int)arg1, (int)arg2);
    case Syscall::Sigaction:
        return current->sys$sigaction((int)arg1, (const Unix::sigaction*)arg2, (Unix::sigaction*)arg3);
    default:
        kprintf("<%u> int0x80: Unknown function %x requested {%x, %x, %x}\n", current->pid(), function, arg1, arg2, arg3);
        break;
    }
    return 0;
}

}

void syscall_entry(RegisterDump& regs)
{
    DWORD function = regs.eax;
    DWORD arg1 = regs.edx;
    DWORD arg2 = regs.ecx;
    DWORD arg3 = regs.ebx;
    regs.eax = Syscall::handle(regs, function, arg1, arg2, arg3);
}