/*
 * Copyright (c) 2024 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 "dfx_kernel_stack.h"

#include <dlfcn.h>
#include <fcntl.h>
#include <regex>
#include <string_ex.h>
#include <sys/ioctl.h>
#include <unistd.h>

#include "dfx_log.h"

#define LOGGER_GET_STACK    _IO(0xAB, 9)
namespace OHOS {
namespace HiviewDFX {
namespace {
// keep sync with defines in kernel/hicollie
const char* const BBOX_PATH = "/dev/bbox";
const int BUFF_STACK_SIZE = 20 * 1024;
const uint32_t MAGIC_NUM = 0x9517;
typedef struct HstackVal {
    uint32_t magic {0};
    pid_t pid {0};
    char hstackLogBuff[BUFF_STACK_SIZE] {0};
} HstackVal;
}
int DfxGetKernelStack(int32_t pid, std::string& kernelStack)
{
    auto kstackBuf = std::make_shared<HstackVal>();
    if (kstackBuf == nullptr) {
        DFXLOG_WARN("Failed create HstackVal, errno:%d", errno);
        return -1;
    }
    kstackBuf->pid = pid;
    kstackBuf->magic = MAGIC_NUM;
    int fd = open(BBOX_PATH, O_WRONLY | O_CLOEXEC);
    if (fd < 0) {
        DFXLOG_WARN("Failed to open bbox, errno:%d", errno);
        return -1;
    }

    int ret = ioctl(fd, LOGGER_GET_STACK, kstackBuf.get());
    if (ret != 0) {
        DFXLOG_WARN("Failed to get pid(%d) kernel stack, errno:%d", pid, errno);
    } else {
        kernelStack = std::string(kstackBuf->hstackLogBuff);
    }
    close(fd);
    return ret;
}

bool FormatThreadKernelStack(const std::string& kernelStack, DfxThreadStack& threadStack)
{
#ifdef __aarch64__
    std::regex headerPattern(R"(name=(.{1,20}), tid=(\d{1,10}), ([\w\=\.]{1,256}, ){3}pid=(\d{1,10}))");
    std::smatch result;
    if (!regex_search(kernelStack, result, headerPattern)) {
        DFXLOG_INFO("%s", "search thread name failed");
        return false;
    }
    threadStack.threadName = result[1].str();
    int base {10};
    threadStack.tid = strtol(result[2].str().c_str(), nullptr, base); // 2 : second of searched element
    auto pos = kernelStack.rfind("pid=" + result[result.size() - 1].str());
    if (pos == std::string::npos) {
        return false;
    }
    size_t index = 0;
    std::regex framePattern(R"(\[(\w{16})\]\<[\w\?+/]{1,1024}\> \(([\w\-./]{1,1024})\))");
    for (std::sregex_iterator it = std::sregex_iterator(kernelStack.begin() + pos, kernelStack.end(), framePattern);
        it != std::sregex_iterator(); ++it) {
        if ((*it)[2].str().rfind(".elf") != std::string::npos) { // 2 : second of searched element is map name
            continue;
        }
        DfxFrame frame;
        frame.index = index++;
        base = 16; // 16 : Hexadecimal
        frame.relPc = strtoull((*it)[1].str().c_str(), nullptr, base);
        frame.mapName = (*it)[2].str(); // 2 : second of searched element is map name
        threadStack.frames.emplace_back(frame);
    }
    return true;
#else
    return false;
#endif
}

bool FormatProcessKernelStack(const std::string& kernelStack, std::vector<DfxThreadStack>& processStack)
{
#ifdef __aarch64__
    std::vector<std::string> threadKernelStackVec;
    OHOS::SplitStr(kernelStack, "Thread info:", threadKernelStackVec);
    if (threadKernelStackVec.size() == 1) {
        return false;
    }
    for (const std::string& threadKernelStack : threadKernelStackVec) {
        DfxThreadStack threadStack;
        if (FormatThreadKernelStack(threadKernelStack, threadStack)) {
            processStack.emplace_back(threadStack);
        }
    }
    return true;
#else
    return false;
#endif
}
}
}