/*
 * Copyright (C) 2020 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#define TLOG_TAG "receiver"

#include <assert.h>

#include <inttypes.h>
#include <lk/err_ptr.h>
#include <lk/macros.h>
#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/auxv.h>
#include <sys/mman.h>

#include <lib/tipc/tipc.h>
#include <lib/tipc/tipc_srv.h>
#include <lib/unittest/unittest.h>

#include <trusty/sys/mman.h>
#include <trusty_unittest.h>

/* Number of pages to expect from NS */
static const size_t num_pages = 10;

static struct tipc_port_acl receiver_port_acl = {
        .flags = IPC_PORT_ALLOW_NS_CONNECT,
        .uuid_num = 0,
        .uuids = NULL,
        .extra_data = NULL,
};

static struct tipc_port receiver_port = {
        .name = "com.android.trusty.memref.receiver",
        .msg_max_size = 1,
        .msg_queue_len = 1,
        .acl = &receiver_port_acl,
        .priv = NULL,
};

static int receiver_on_message(const struct tipc_port* port,
                               handle_t chan,
                               void* ctx) {
    assert(port == &receiver_port);
    assert(ctx == NULL);
    handle_t handle;
    int rc;

    struct ipc_msg msg = {
            .iov = NULL,
            .num_iov = 0,
            .handles = &handle,
            .num_handles = 1,
    };

    struct ipc_msg_info msg_inf;
    rc = get_msg(chan, &msg_inf);
    if (rc) {
        return rc;
    }

    if (msg_inf.num_handles != 1) {
        TLOGE("Message had no handles\n");
        return -1;
    }

    rc = read_msg(chan, msg_inf.id, 0, &msg);
    put_msg(chan, msg_inf.id);
    if (rc < 0) {
        TLOGE("Failed to read message\n");
        return rc;
    }

    size_t page_size = getauxval(AT_PAGESZ);

    char* out = mmap(0, page_size * num_pages, PROT_READ | PROT_WRITE, 0,
                     handle, 0);
    if (out == MAP_FAILED) {
        rc = (intptr_t)out;
        TLOGE("Failed to mmap handle\n");
        return rc;
    }

    for (size_t skip = 0; skip < num_pages; skip++) {
        strcpy(&out[skip * page_size], "Hello from Trusty!");
    }

    rc = munmap((void*)out, page_size * num_pages);
    if (rc != NO_ERROR) {
        TLOGE("munmap() failed: %d\n", rc);
        return rc;
    }

    close(handle);

    // Send a message for sync
    char c = 0;
    tipc_send1(chan, &c, 1);

    return rc;
}

static struct tipc_srv_ops receiver_ops = {
        .on_message = receiver_on_message,
};

int main(void) {
    struct tipc_hset* hset = tipc_hset_create();

    if (IS_ERR(hset)) {
        return PTR_ERR(hset);
    }

    int rc = tipc_add_service(hset, &receiver_port, 1, 1, &receiver_ops);
    if (rc < 0) {
        return rc;
    }

    rc = tipc_run_event_loop(hset);
    TLOGE("receiver going down: (%d)\n", rc);
    return rc;
}