// Copyright 2016 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <launchpad/launchpad.h>
#include <launchpad/vmo.h>

#include <zircon/process.h>
#include <zircon/processargs.h>
#include <zircon/syscalls.h>
#include <lib/fdio/io.h>
#include <lib/fdio/namespace.h>
#include <lib/fdio/fd.h>
#include <lib/fdio/fdio.h>
#include <lib/fdio/directory.h>

#include <stdlib.h>
#include <unistd.h>

static zx_status_t add_fdio(launchpad_t* lp,
                            zx_handle_t* handle,
                            uint32_t type,
                            zx_status_t status) {
    if (status == ZX_ERR_BAD_HANDLE)
        return ZX_OK;
    if (status == ZX_ERR_NOT_SUPPORTED)
        return ZX_OK;
    if (status == ZX_OK) {
        return launchpad_add_handles(lp, 1, handle, &type);
    } else {
        launchpad_abort(lp, status, "add_fdio: failed");
        return status;
    }
}

zx_status_t launchpad_clone(launchpad_t* lp, uint32_t what) {
    if (what & LP_CLONE_FDIO_NAMESPACE) {
        fdio_flat_namespace_t* flat;
        zx_status_t status = fdio_ns_export_root(&flat);
        if (status == ZX_OK) {
            launchpad_set_nametable(lp, flat->count, flat->path);
            launchpad_add_handles(lp, flat->count, flat->handle, flat->type);
            free(flat);
        } else if (status != ZX_ERR_NOT_FOUND) {
            launchpad_abort(lp, status, "clone: error cloning namespace");
            return status;
        }
    }
    if (what & LP_CLONE_FDIO_STDIO) {
        for (int fd = 0; fd < 3; fd++) {
            zx_handle_t handle = ZX_HANDLE_INVALID;
            add_fdio(lp, &handle, PA_HND(PA_FD, fd), fdio_fd_clone(fd, &handle));
        }
    }
    if (what & LP_CLONE_ENVIRON) {
        launchpad_set_environ(lp, (const char* const*)environ);
    }
    if (what & LP_CLONE_DEFAULT_JOB) {
        zx_handle_t job;
        if (zx_handle_duplicate(zx_job_default(), ZX_RIGHT_SAME_RIGHTS, &job) == ZX_OK) {
            launchpad_add_handle(lp, job, PA_HND(PA_JOB_DEFAULT, 0));
        }
    }
    return launchpad_get_status(lp);
}