/* -*- mode:c; c-file-style:"k&r"; c-basic-offset: 4; tab-width:4; indent-tabs-mode:nil; mode:auto-fill; fill-column:78; -*- */
/* vim: set ts=4 sw=4 et tw=78 fo=cqt wm=0: */

/* Copyright (C) 2014 OSCAR lab, Stony Brook University
   This file is part of Graphene Library OS.

   Graphene Library OS is free software: you can redistribute it and/or
   modify it under the terms of the GNU General Public License
   as published by the Free Software Foundation, either version 3 of the
   License, or (at your option) any later version.

   Graphene Library OS is distributed in the hope that 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/>.  */

/*
 * db_process.c
 *
 * This source file contains functions to create a child process and terminate
 * the running process. Child does not inherit any objects or memory from its
 * parent pricess. A Parent process may not modify the execution of its
 * children. It can wait for a child to exit using its handle. Also, parent and
 * child may communicate through I/O streams provided by the parent to the child
 * at creation.
 */

#include "pal_defs.h"
#include "pal_linux_defs.h"
#include "pal.h"
#include "pal_internal.h"
#include "pal_linux.h"
#include "pal_debug.h"
#include "pal_error.h"
#include "pal_security.h"
#include "graphene.h"
#include "graphene-ipc.h"
#include "api.h"

#include <linux/sched.h>
#include <linux/types.h>
typedef __kernel_pid_t pid_t;
#include <asm/fcntl.h>
#include <sys/socket.h>
#include <asm-errno.h>

#ifndef SEEK_SET
# define SEEK_SET 0
#endif

int _DkProcessCreate (PAL_HANDLE * handle,
                      const char * uri, int flags, const char ** args)
{
    int ret, rete;

    const char * manifest_uri   = pal_config.manifest;
    PAL_HANDLE   manifest       = pal_config.manifest_handle;
    int          manifest_fd    = -1;
    const char * exec_uri       = NULL;
    PAL_HANDLE   exec           = NULL;
    int          exec_fd        = -1;
    bool         noexec         = false;

    if (uri) {
        exec_uri = uri;
        if ((ret = _DkStreamOpen(&exec, uri, PAL_ACCESS_RDONLY, 0, 0, 0)) < 0)
            return ret;

        if (check_elf_object(exec) < 0) {
            manifest = exec;
            manifest_uri = uri;
            exec = NULL;
            exec_uri = NULL;
        }

        exec_fd = exec->file.fd;
        INLINE_SYSCALL(fcntl, 3, exec_fd, F_SETFD, 0);
    } else {
        noexec = true;
    }

    if (manifest) {
        manifest_fd = manifest->file.fd;
        INLINE_SYSCALL(fcntl, 3, manifest_fd, F_SETFD, 0);
    }

    int fds[6] = { -1, -1, -1, -1, -1, -1 };

    if (IS_ERR((ret = INLINE_SYSCALL(pipe2, 2, &fds[0], 0))) ||
        IS_ERR((ret = INLINE_SYSCALL(pipe2, 2, &fds[2], 0))) ||
        IS_ERR((ret = INLINE_SYSCALL(socketpair, 4, AF_UNIX, SOCK_STREAM,
                                     0, &fds[4])))) {
        ret = -PAL_ERROR_DENIED;
        goto out;
    }

    int proc_fds[2][3] = {
        { fds[0], fds[3], fds[4] },
        { fds[2], fds[1], fds[5] },
    };

    int argc = 0;
    if (args) for (; args[argc] ; argc++);
    const char ** argv = __alloca(sizeof(const char *) * (argc + 2));
    argv[0] = PAL_LOADER;
    if (args) memcpy(&argv[1], args, sizeof(const char *) * argc);
    argv[argc + 1] = NULL;

    ret = ARCH_VFORK();

    if (IS_ERR(ret)) {
        ret = -PAL_ERROR_DENIED;
        goto out;
    }

    if (!ret) {
        for (int i = 0 ; i < 3 ; i++)
            INLINE_SYSCALL(close, 1, proc_fds[1][i]);

        INLINE_SYSCALL(close, 1, PROC_INIT_FD);
        rete = INLINE_SYSCALL(dup2, 2, proc_fds[0][0], PROC_INIT_FD);
        if (IS_ERR(rete))
            goto out_child;

        if (manifest_fd >= 0)
            INLINE_SYSCALL(fcntl, 3, manifest_fd, F_SETFD, 0);

        rete = INLINE_SYSCALL(execve, 3, PAL_LOADER, argv, NULL);

        /* shouldn't get to here */
        printf("unexpected failure of new process\n");
out_child:
        asm("hlt");
        return 0;
    }

    if (IS_ERR(rete)) {
        ret = -PAL_ERROR_DENIED;
        goto out;
    }

    for (int i = 0 ; i < 3 ; i++)
        INLINE_SYSCALL(close, 1, proc_fds[0][i]);

    int pipe_in = proc_fds[1][0], pipe_out = proc_fds[1][1];
    unsigned short data_size = 0;
    unsigned short exec_uri_offset = 0, manifest_uri_offset = 0;

    if (exec_uri) {
        int len = strlen(exec_uri);
        exec_uri_offset = data_size;
        data_size += len + 1;
    }

    if (manifest_fd >= 0) {
        int len = strlen(manifest_uri);
        manifest_uri_offset = data_size;
        data_size += len + 1;
    }

    struct pal_proc_args * proc_args = __alloca(sizeof(struct pal_proc_args) +
                                                data_size);
    void * data = ((void *) proc_args) + sizeof(struct pal_proc_args);
    memset(proc_args, 0, sizeof(struct pal_proc_args));
    memcpy(&proc_args->pal_sec_info, &pal_sec_info, sizeof(struct pal_sec_info));
    proc_args->pal_sec_info._dl_debug_state = NULL;
    proc_args->pal_sec_info._r_debug = NULL;
    proc_args->proc_fds[0] = proc_fds[0][0];
    proc_args->proc_fds[1] = proc_fds[0][1];
    proc_args->proc_fds[2] = proc_fds[0][2];
    proc_args->parent_pid  = pal_linux_config.pid;
    proc_args->exec_fd = (exec_fd == -1) ? PAL_IDX_POISON : exec_fd;
    proc_args->noexec  = noexec;
    proc_args->manifest_fd = (manifest_fd == -1) ? PAL_IDX_POISON : manifest_fd;

    if (exec_uri)
        memcpy(data + (proc_args->exec_uri_offset = exec_uri_offset),
               exec_uri, strlen(exec_uri) + 1);

    if (manifest_uri)
        memcpy(data + (proc_args->manifest_uri_offset = manifest_uri_offset),
               manifest_uri, strlen(manifest_uri) + 1);

    proc_args->data_size = data_size;

    ret = INLINE_SYSCALL(write, 3, pipe_out, proc_args,
                         sizeof(struct pal_proc_args) + data_size);

    if (IS_ERR(ret) || ret < sizeof(struct pal_proc_args) + data_size) {
        ret = -PAL_ERROR_DENIED;
        goto out;
    }

    ret = INLINE_SYSCALL(read, 3, pipe_in, &rete, sizeof(int));

    if (IS_ERR(ret) || ret < sizeof(int)) {
        ret = -PAL_ERROR_DENIED;
        goto out;
    }

    if (rete < 0) {
        ret = rete;
        goto out;
    }

    for (int i = 0 ; i < 3 ; i++)
        INLINE_SYSCALL(fcntl, 3, proc_fds[1][i], F_SETFD, FD_CLOEXEC);

    int pid = ret;
    PAL_HANDLE hdl = malloc(HANDLE_SIZE(process));
    SET_HANDLE_TYPE(hdl, process);
    hdl->__in.flags |= RFD(0)|WFD(1)|RFD(2)|WFD(2)|WRITEABLE(1)|WRITEABLE(2);
    hdl->process.stream_in  = proc_fds[1][0];
    hdl->process.stream_out = proc_fds[1][1];
    hdl->process.cargo      = proc_fds[1][2];
    hdl->process.pid = pid;
    hdl->process.nonblocking = PAL_FALSE;
    *handle = hdl;
    ret = 0;
out:
    if (ret < 0) {
        for (int i = 0 ; i < 6 ; i++)
            if (fds[i] >= 0)
                INLINE_SYSCALL(close, 1, fds[i]);
    }

    return ret;
}

int init_child_process (struct pal_proc_args * proc_args, void * proc_data)
{
    memcpy(&pal_sec_info, &proc_args->pal_sec_info, sizeof(pal_sec_info));

    PAL_HANDLE parent = malloc(HANDLE_SIZE(process));
    SET_HANDLE_TYPE(parent, process);
    parent->__in.flags |= RFD(0)|WFD(1)|RFD(2)|WFD(2)|WRITEABLE(1)|WRITEABLE(2);
    parent->process.stream_in  = proc_args->proc_fds[0];
    parent->process.stream_out = proc_args->proc_fds[1];
    parent->process.cargo      = proc_args->proc_fds[2];
    parent->process.pid        = proc_args->parent_pid;
    parent->process.nonblocking = PAL_FALSE;
    __pal_control.parent_process = parent;

    if (proc_args->exec_fd != PAL_IDX_POISON) {
        char * uri = (char *) proc_data + proc_args->exec_uri_offset;
        char * exec_uri = remalloc(uri, strlen(uri) + 1);
        INLINE_SYSCALL(lseek, 3, proc_args->exec_fd, 0, SEEK_SET);
        PAL_HANDLE exec = malloc(HANDLE_SIZE(file));
        SET_HANDLE_TYPE(exec, file);
        exec->__in.flags |= RFD(0);
        exec->file.fd = proc_args->exec_fd;
        exec->file.offset = 0;
        exec->file.append = PAL_FALSE;
        exec->file.pass   = PAL_FALSE;
        exec->file.realpath = remalloc(exec_uri + 5, strlen(exec_uri + 5) + 1);
        pal_config.exec = exec_uri;
        pal_config.exec_handle = exec;
    } else {
        pal_linux_config.noexec = proc_args->noexec;
    }

    if (proc_args->manifest_fd != PAL_IDX_POISON) {
        char * uri = (char *) proc_data + proc_args->manifest_uri_offset;
        char * manifest_uri = remalloc(uri, strlen(uri) + 1);
        INLINE_SYSCALL(lseek, 3, proc_args->manifest_fd, 0, SEEK_SET);
        PAL_HANDLE manifest = malloc(HANDLE_SIZE(file));
        SET_HANDLE_TYPE(manifest, file);
        manifest->__in.flags |= RFD(0);
        manifest->file.fd = proc_args->manifest_fd;
        manifest->file.offset = 0;
        manifest->file.append = PAL_FALSE;
        manifest->file.pass   = PAL_FALSE;
        manifest->file.realpath = remalloc(manifest_uri + 5,
                                           strlen(manifest_uri + 5) + 1);
        pal_config.manifest = manifest_uri;
        pal_config.manifest_handle = manifest;
    }

    int child_status = 0;
    int ret = INLINE_SYSCALL(write, 3, proc_args->proc_fds[1], &child_status,
                             sizeof(int));
    if (IS_ERR(ret))
        return -PAL_ERROR_DENIED;

    return 0;
}

void _DkProcessExit (int exitcode)
{
    if (__pal_control.parent_process)
        _DkObjectClose(__pal_control.parent_process);

    if (__pal_control.manifest_handle)
        _DkObjectClose(__pal_control.manifest_handle);

    INLINE_SYSCALL(exit_group, 1, exitcode);
}

int ioctl_set_graphene (struct config_store * config, int ndefault,
                        const struct graphene_user_policy * default_policies);

static int set_graphene_task (const char * uri, int flags)
{
    PAL_HANDLE handle = NULL;
    int ret;

    if ((ret = _DkStreamOpen(&handle, uri, PAL_ACCESS_RDONLY, 0, 0, 0)) < 0)
        return ret;

    PAL_STREAM_ATTR attr;

    if ((ret = _DkStreamAttributesQuerybyHandle(handle, &attr)) < 0)
        goto out;

    void * addr = NULL;
    size_t size = attr.size;

    if ((ret = _DkStreamMap(handle, &addr, PAL_PROT_READ, 0,
                            ALLOC_ALIGNUP(size))) < 0)
        goto out;

    struct config_store cfg;
    cfg.raw_data = addr;
    cfg.raw_size = size;
    cfg.malloc = malloc;
    cfg.free = free;

    if ((ret = read_config(&cfg, NULL, NULL)) < 0)
        goto out_mem;

    const char * manifest = uri;

    struct graphene_user_policy manifest_policy;
    if (!memcmp(manifest, "file:", 5)) {
        manifest_policy.type = GRAPHENE_FS_PATH | GRAPHENE_FS_READ;
        manifest_policy.value = manifest + 5;
    } else {
        manifest_policy.type = 0;
    }

    if (flags & PAL_SANDBOX_PIPE) {
        do {
            getrand(&pal_sec_info.mcast_port, sizeof(unsigned short));
        } while (pal_sec_info.mcast_port < 1024);
    }

    struct graphene_net_policy mcast_rules[2];
    memset(mcast_rules, 0, sizeof(struct graphene_net_policy) * 2);

    mcast_rules[0].family = AF_INET;
    mcast_rules[0].local.port_begin = pal_sec_info.mcast_port;
    mcast_rules[0].local.port_end = pal_sec_info.mcast_port;
    mcast_rules[0].peer.port_begin = 0;
    mcast_rules[0].peer.port_end = 65535;

    mcast_rules[1].family = AF_INET;
    mcast_rules[1].local.port_begin = 0;
    mcast_rules[1].local.port_end = 65535;
    inet_pton(AF_INET, MCAST_GROUP, &mcast_rules[1].peer.addr);
    mcast_rules[1].peer.port_begin = pal_sec_info.mcast_port;
    mcast_rules[1].peer.port_end = pal_sec_info.mcast_port;

    if (flags & PAL_SANDBOX_PIPE) {
        char pipe_root[sizeof(GRAPHENE_PIPEDIR) + 20];
        char pipe_prefix[9];
        int sandboxid;

        snprintf(pipe_root,
                 sizeof(GRAPHENE_PIPEDIR) + 20, GRAPHENE_PIPEDIR "/%08x",
                 pal_sec_info.domain_id);

        getrand(&sandboxid, sizeof(int));
        snprintf(pipe_prefix, 9, "%08x", sandboxid);

        struct graphene_user_policy default_policies[] = {
            { .type = GRAPHENE_UNIX_ROOT,    .value = pipe_root, },
            { .type = GRAPHENE_UNIX_PREFIX,  .value = pipe_prefix, },
            { .type = GRAPHENE_NET_RULE,     .value = &mcast_rules[0], },
            { .type = GRAPHENE_NET_RULE,     .value = &mcast_rules[1], },
            manifest_policy,
        };

        ret = ioctl_set_graphene(&cfg, manifest_policy.type ? 5 : 4,
                                 default_policies);
        if (ret < 0)
            goto out_mem;

        memcpy(&pal_sec_info.pipe_prefix, pipe_prefix, 9);
    } else {
        const struct graphene_user_policy default_policies[] = {
            { .type = GRAPHENE_NET_RULE,    .value = &mcast_rules[0], },
            { .type = GRAPHENE_NET_RULE,    .value = &mcast_rules[1], },
            manifest_policy,
        };

        ret = ioctl_set_graphene(&cfg, manifest_policy.type ? 3 : 2,
                                 default_policies);
        if (ret < 0)
            goto out_mem;
    }

    pal_config.manifest = manifest;
    _DkObjectClose(pal_config.manifest_handle);
    pal_config.manifest_handle = handle;

    free_config(&cfg);
out_mem:
    _DkStreamUnmap(cfg.raw_data, ALLOC_ALIGNUP(cfg.raw_size));
out:
    DkObjectClose(handle);
    return ret;
}

int _DkProcessSandboxCreate (const char * manifest, int flags)
{
    return set_graphene_task(manifest, flags);
}

static int proc_read (PAL_HANDLE handle, int offset, int count,
                          void * buffer)
{
    int bytes = INLINE_SYSCALL(read, 3, handle->process.stream_in, buffer,
                               count);

    if (IS_ERR(bytes))
        switch(ERRNO(bytes)) {
            case EWOULDBLOCK:
                return-PAL_ERROR_TRYAGAIN;
            case EINTR:
                return -PAL_ERROR_INTERRUPTED;
            default:
                return -PAL_ERROR_DENIED;
        }

    return bytes;
}

static int proc_write (PAL_HANDLE handle, int offset, int count,
                       const void * buffer)
{
    int bytes = INLINE_SYSCALL(write, 3, handle->process.stream_out, buffer,
                               count);

    if (IS_ERR(bytes))
        switch(ERRNO(bytes)) {
            case EWOULDBLOCK:
                handle->__in.flags &= ~WRITEABLE(1);
                return-PAL_ERROR_TRYAGAIN;
            case EINTR:
                return -PAL_ERROR_INTERRUPTED;
            default:
                return -PAL_ERROR_DENIED;
        }

    if (bytes == count)
        handle->__in.flags |= WRITEABLE(1);
    else
        handle->__in.flags &= ~WRITEABLE(1);

    return bytes;
}

static int proc_close (PAL_HANDLE handle)
{
    if (handle->process.stream_in != PAL_IDX_POISON) {
        INLINE_SYSCALL(close, 1, handle->process.stream_in);
        handle->process.stream_in = PAL_IDX_POISON;
    }

    if (handle->process.stream_out != PAL_IDX_POISON) {
        INLINE_SYSCALL(close, 1, handle->process.stream_out);
        handle->process.stream_out = PAL_IDX_POISON;
    }

    if (handle->process.cargo != PAL_IDX_POISON) {
        INLINE_SYSCALL(close, 1, handle->process.cargo);
        handle->process.cargo = PAL_IDX_POISON;
    }

    return 0;
}

static int proc_delete (PAL_HANDLE handle, int access)
{
    int shutdown;
    switch (access) {
        case 0:
            shutdown = SHUT_RDWR;
            break;
        case PAL_DELETE_RD:
            shutdown = SHUT_RD;
            break;
        case PAL_DELETE_WR:
            shutdown = SHUT_WR;
            break;
        default:
            return -PAL_ERROR_INVAL;
    }

    if (access != PAL_DELETE_WR &&
        handle->process.stream_in != PAL_IDX_POISON) {
        INLINE_SYSCALL(close, 1, handle->process.stream_in);
        handle->process.stream_in = PAL_IDX_POISON;
    }

    if (access != PAL_DELETE_RD &&
        handle->process.stream_out != PAL_IDX_POISON) {
        INLINE_SYSCALL(close, 1, handle->process.stream_out);
        handle->process.stream_out = PAL_IDX_POISON;
    }

    if (handle->process.cargo != PAL_IDX_POISON)
        INLINE_SYSCALL(shutdown, 2, handle->process.cargo, shutdown);

    return 0;
}

#ifndef FIONREAD
# define FIONREAD 0x541B
#endif

static int proc_attrquerybyhdl (PAL_HANDLE handle, PAL_STREAM_ATTR * attr)
{
    int ret, val;

    if (handle->process.stream_in == PAL_IDX_POISON)
        return -PAL_ERROR_BADHANDLE;

    memset(attr, 0, sizeof(PAL_STREAM_ATTR));

    ret = INLINE_SYSCALL(ioctl, 3, handle->process.stream_in, FIONREAD, &val);
    if (!IS_ERR(ret))
        attr->size = val;

    attr->disconnected = handle->__in.flags & (ERROR(0)|ERROR(1));
    attr->readable = (attr->size > 0);
    attr->writeable = handle->__in.flags & WRITEABLE(1);
    attr->nonblocking = handle->process.nonblocking;
    return 0;
}

static int proc_attrsetbyhdl (PAL_HANDLE handle, PAL_STREAM_ATTR * attr)
{
    if (handle->process.stream_in == PAL_IDX_POISON)
        return -PAL_ERROR_BADHANDLE;

    int ret;
    if (attr->nonblocking != handle->process.nonblocking) {
        ret = INLINE_SYSCALL(fcntl, 3, handle->process.stream_in, F_SETFL,
                             handle->process.nonblocking ? O_NONBLOCK : 0);

        if (IS_ERR(ret))
            return unix_to_pal_error(ERRNO(ret));

        handle->process.nonblocking = attr->nonblocking;
    }

    return 0;
}

struct handle_ops proc_ops = {
        .read           = &proc_read,
        .write          = &proc_write,
        .close          = &proc_close,
        .delete         = &proc_delete,
        .attrquerybyhdl = &proc_attrquerybyhdl,
        .attrsetbyhdl   = &proc_attrsetbyhdl,
    };