/*
 * Copyright (c) 2021-2022 Huawei Device Co., Ltd.
 * 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.
 */
#include <signal.h>
#include <stdbool.h>
#include "appspawn_adapter.h"
#include "appspawn_msg.h"
#include "appspawn_server.h"
#include "appspawn_service.h"
#include "securec.h"
#include "init_param.h"
#include "syspara/parameter.h"
#include "nwebspawn_lancher.h"
#define APPSPAWN_PRELOAD "libappspawn_helper.z.so"

static void CheckPreload(char *const argv[])
{
    char *preload = getenv("LD_PRELOAD");
    if (preload && strstr(preload, APPSPAWN_PRELOAD)) {
        return;
    }
    char buf[128] = APPSPAWN_PRELOAD; // 128 is enough in most cases
    if (preload && preload[0]) {
        int len = sprintf_s(buf, sizeof(buf), "%s:" APPSPAWN_PRELOAD, preload);
        APPSPAWN_CHECK(len > 0, return, "preload too long: %{public}s", preload);
    }
    int ret = setenv("LD_PRELOAD", buf, true);
    APPSPAWN_CHECK(ret == 0, return, "setenv fail: %{public}s", buf);
    ssize_t nread = readlink("/proc/self/exe", buf, sizeof(buf) - 1);
    APPSPAWN_CHECK(nread != -1, return, "readlink fail: /proc/self/exe: %{public}d", errno);
    buf[nread] = 0;
    ret = execv(buf, argv);
    APPSPAWN_LOGE("execv fail: %{public}s: %{public}d: %{public}d", buf, errno, ret);
}

int main(int argc, char *const argv[])
{
    if (argc <= 0) {
        return 0;
    }

    CheckPreload(argv);
    (void)signal(SIGPIPE, SIG_IGN);
    uint32_t argvSize = 0;
    int mode = 0;
    pid_t pid = 1;
    if ((argc > PARAM_INDEX) && (strcmp(argv[START_INDEX], "cold-start") == 0)) {
        argvSize = APP_LEN_PROC_NAME;
        mode = 1;
    } else {
        // calculate child process long name size
        uintptr_t start = (uintptr_t)argv[0];
        uintptr_t end = (uintptr_t)strchr(argv[argc - 1], 0);
        if (end == 0) {
            return -1;
        }
        argvSize = end - start;
        if (argvSize > APP_MSG_MAX_SIZE) {
            return -1;
        }
        pid = NwebSpawnLanch();
        int isRet = memset_s(argv[0], argvSize, 0, (size_t)argvSize) != EOK;
        APPSPAWN_CHECK(!isRet, return -EINVAL, "Failed to memset argv[0]");
        if (pid == 0) {
            isRet = strncpy_s(argv[0], argvSize, NWEBSPAWN_SERVER_NAME, strlen(NWEBSPAWN_SERVER_NAME)) != EOK;
        } else {
            isRet = strncpy_s(argv[0], argvSize, APPSPAWN_SERVER_NAME, strlen(APPSPAWN_SERVER_NAME)) != EOK;
        }
        APPSPAWN_CHECK(!isRet, return -EINVAL, "strncpy_s appspawn server name error: %{public}d", errno);
    }

    if (pid == 0) {
        APPSPAWN_LOGI("NwebSpawnCreateContent argc %{public}d mode %{public}d %{public}u", argc, mode, argvSize);
        AppSpawnContent *content = AppSpawnCreateContent(NWEBSPAWN_SOCKET_NAME, argv[0], argvSize, mode);
        APPSPAWN_CHECK(content != NULL, return -1, "Invalid content for nwebspawn");
        APPSPAWN_CHECK(content->initAppSpawn != NULL, return -1, "Invalid content for nwebspawn");
        APPSPAWN_CHECK(content->runAppSpawn != NULL, return -1, "Invalid content for nwebspawn");

        // set common operation
        content->loadExtendLib = LoadExtendLibNweb;
        content->runChildProcessor = RunChildProcessorNweb;
        content->initAppSpawn(content);
        content->runAppSpawn(content, argc, argv);
    } else {
        APPSPAWN_LOGI("AppSpawnCreateContent argc %{public}d mode %{public}d %{public}u", argc, mode, argvSize);
        AppSpawnContent *content = AppSpawnCreateContent(APPSPAWN_SOCKET_NAME, argv[0], argvSize, mode);
        APPSPAWN_CHECK(content != NULL, return -1, "Invalid content for appspawn");
        APPSPAWN_CHECK(content->initAppSpawn != NULL, return -1, "Invalid content for appspawn");
        APPSPAWN_CHECK(content->runAppSpawn != NULL, return -1, "Invalid content for appspawn");

        // set common operation
        content->loadExtendLib = LoadExtendLib;
        content->runChildProcessor = RunChildProcessor;
        content->initAppSpawn(content);
        if (mode == 0) {
            AddNwebInfo(pid, NWEBSPAWN_SERVER_NAME);
            SystemSetParameter("bootevent.appspawn.started", "true");
        }
        content->runAppSpawn(content, argc, argv);
    }

    return 0;
}