OpenHarmony-v6.0-Release/s

/*
 * Copyright (c) 2023-2023 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.
 */
#define HST_LOG_TAG "HttpCurlClient"

#include "http_curl_client.h"
#include <algorithm>
#include <regex>
#include <vector>
#include "common/log.h"
#include "osal/task/autolock.h"
#include "securec.h"
#include "net_conn_client.h"
#include <fcntl.h>

namespace {
constexpr OHOS::HiviewDFX::HiLogLabel LABEL = { LOG_CORE, LOG_DOMAIN_STREAM_SOURCE, "HiStreamer" };
}

namespace OHOS {
namespace Media {
namespace Plugins {
namespace HttpPlugin {
const uint32_t MAX_STRING_LENGTH = 4096;
constexpr uint32_t DEFAULT_LOW_SPEED_LIMIT = 1L;
constexpr uint32_t DEFAULT_LOW_SPEED_TIME = 10L;
constexpr uint32_t MILLS_TO_SECOND = 1000;
constexpr uint32_t HTTP_ERROR_THRESHOLD = 400;

std::string ToString(const std::list<std::string> &lists, char tab)
{
    std::string str;
    for (auto it = lists.begin(); it != lists.end(); ++it) {
        if (it != lists.begin()) {
            str.append(1, tab);
        }
        str.append(*it);
    }
    return str;
}

std::string InsertCharBefore(std::string input, char from, char preChar, char nextChar)
{
    std::string output = input;
    char arr[] = {preChar, from};
    unsigned long strSize = sizeof(arr) / sizeof(arr[0]);
    std::string str(arr, strSize);
    std::size_t pos = output.find(from);
    std::size_t length = output.length();
    while (pos >= 0 && length >= 1 && pos <= length - 1 && pos != std::string::npos) {
        char nextCharTemp = pos >= length ? '\0' : output[pos + 1];
        if (nextChar == '\0' || nextCharTemp == '\0' || nextCharTemp != nextChar) {
            output.replace(pos, 1, str);
            length += (strSize - 1);
        }
        pos = output.find(from, pos + strSize);
    }
    return output;
}

std::string Trim(std::string str)
{
    if (str.empty()) {
        return str;
    }
    while (std::isspace(str[0])) {
        str.erase(0, 1);
    }
    if (str.empty()) {
        return str;
    }
    while (str.size() >= 1 && std::isspace(str[str.size() - 1])) {
            str.erase(str.size() - 1, 1);
    }
    return str;
}

std::string GetHostnameFromURL(const std::string &url)
{
    std::string delimiter = "://";
    std::string tempUrl = url;
    size_t posStart = tempUrl.find(delimiter);
    if (posStart != std::string::npos) {
        posStart += delimiter.length();
    } else {
        posStart = 0;
    }
    size_t posEnd = std::min({tempUrl.find(":", posStart), tempUrl.find("/", posStart), tempUrl.find("?", posStart)});
    if (posEnd != std::string::npos) {
        return tempUrl.substr(posStart, posEnd - posStart);
    }
    return tempUrl.substr(posStart);
}
bool IsRegexValid(const std::string &regex)
{
    if (Trim(regex).empty()) {
        return false;
    }
    return regex_match(regex, std::regex("^[a-zA-Z0-9\\-_\\.*]+$"));
}

std::string ReplaceCharacters(const std::string &input)
{
    std::string output = InsertCharBefore(input, '*', '.', '\0');
    output = InsertCharBefore(output, '.', '\\', '*');
    return output;
}

bool IsMatch(const std::string &str, const std::string &patternStr)
{
    if (patternStr.empty()) {
        return false;
    }
    if (patternStr == "*") {
        return true;
    }
    if (!IsRegexValid(patternStr)) {
        return patternStr == str;
    }
    std::regex pattern(ReplaceCharacters(patternStr));
    bool isMatch = patternStr != "" && std::regex_match(str, pattern);
    return isMatch;
}

bool IsExcluded(const std::string &str, const std::string &exclusions, const std::string &split)
{
    if (Trim(exclusions).empty()) {
        return false;
    }
    std::size_t start = 0;
    std::size_t end = exclusions.find(split);
    while (end != std::string::npos) {
        if (end - start > 0 && IsMatch(str, Trim(exclusions.substr(start, end - start)))) {
            return true;
        }
        start = end + 1;
        end = exclusions.find(split, start);
    }
    return IsMatch(str, Trim(exclusions.substr(start)));
}

bool IsHostNameExcluded(const std::string &url, const std::string &exclusions, const std::string &split)
{
    std::string hostName = GetHostnameFromURL(url);
    return IsExcluded(hostName, exclusions, split);
}

void GetHttpProxyInfo(std::string &host, int32_t &port, std::string &exclusions)
{
    using namespace NetManagerStandard;
    NetManagerStandard::HttpProxy httpProxy;
    NetConnClient::GetInstance().GetDefaultHttpProxy(httpProxy);
    host = httpProxy.GetHost();
    port = httpProxy.GetPort();
    exclusions = ToString(httpProxy.GetExclusionList());
}

std::shared_ptr<NetworkClient> NetworkClient::GetInstance(RxHeader headCallback, RxBody bodyCallback, void *userParam)
{
    return std::make_shared<HttpCurlClient>(headCallback, bodyCallback, userParam);
}

HttpCurlClient::HttpCurlClient(RxHeader headCallback, RxBody bodyCallback, void *userParam)
    : rxHeader_(headCallback), rxBody_(bodyCallback), userParam_(userParam)
{
    MEDIA_LOG_I("HttpCurlClient ctor");
}

HttpCurlClient::~HttpCurlClient()
{
    MEDIA_LOG_I("~HttpCurlClient dtor");
    Close(false);
}

Status HttpCurlClient::Init()
{
    FALSE_LOG(curl_global_init(CURL_GLOBAL_ALL) == CURLE_OK);
    return Status::OK;
}

std::string HttpCurlClient::ClearHeadTailSpace(std::string& str)
{
    if (str.empty()) {
        return str;
    }
    str.erase(0, str.find_first_not_of(" "));
    str.erase(str.find_last_not_of(" ") + 1);
    return str;
}

void HttpCurlClient::HttpHeaderParse(std::map<std::string, std::string> httpHeader)
{
    if (httpHeader.empty()) {
        MEDIA_LOG_D("Set http header fail, http header is empty.");
        return;
    }
    for (std::map<std::string, std::string>::iterator iter = httpHeader.begin(); iter != httpHeader.end(); iter++) {
        std::string setKey = iter->first;
        std::string setValue = iter->second;
        if (setKey.length() <= MAX_STRING_LENGTH && setValue.length() <= MAX_STRING_LENGTH) {
            ClearHeadTailSpace(setKey);
            std::string headerStr = setKey + ":" + setValue;
            const char* str = headerStr.c_str();
            headerList_ = curl_slist_append(headerList_, str);
        } else {
            MEDIA_LOG_E("Set httpHeader fail, the length of key or value is too long, more than 512.");
            MEDIA_LOG_E("key: " PUBLIC_LOG_S " value: " PUBLIC_LOG_S, setKey.c_str(), setValue.c_str());
        }
    }
}

Status HttpCurlClient::Open(const std::string& url, const std::map<std::string, std::string>& httpHeader,
                            int32_t timeoutMs)
{
    MEDIA_LOG_I("Open client in");
    if (easyHandle_ == nullptr) {
        MEDIA_LOG_E("EasyHandle is nullptr, init easyHandle.");
        easyHandle_ = curl_easy_init();
    }
    FALSE_RETURN_V(easyHandle_ != nullptr, Status::ERROR_NULL_POINTER);
    std::map<std::string, std::string> header = httpHeader;
    if (isFirstOpen_) {
        HttpHeaderParse(header);
        isFirstOpen_ = false;
    }
    InitCurlEnvironment(url, timeoutMs);
    MEDIA_LOG_I("Open client out");
    return Status::OK;
}

Status HttpCurlClient::Close(bool isAsync)
{
    MEDIA_LOG_I("Close client in");
    {
        AutoLock lock(mutex_);
        if (easyHandle_) {
            curl_easy_setopt(easyHandle_, CURLOPT_TIMEOUT_MS, 1);
        }
    }
    if (isAsync) {
        MEDIA_LOG_I("Close client Async out");
        return Status::OK;
    }
    AutoLock lock(mutex_);
    if (easyHandle_) {
        curl_easy_cleanup(easyHandle_);
        easyHandle_ = nullptr;
    }
    ipFlag_ = false;
    if (!ip_.empty()) {
        ip_.clear();
    }
    MEDIA_LOG_I("Close client out");
    return Status::OK;
}

Status HttpCurlClient::Deinit()
{
    MEDIA_LOG_I("Deinit in");
    AutoLock lock(mutex_);
    if (easyHandle_) {
        curl_easy_setopt(easyHandle_, CURLOPT_TIMEOUT_MS, 1);
    }
    if (easyHandle_) {
        curl_easy_cleanup(easyHandle_);
        easyHandle_ = nullptr;
    }
    ipFlag_ = false;
    if (!ip_.empty()) {
        ip_.clear();
    }
    curl_global_cleanup();
    MEDIA_LOG_I("Deinit out");
    return Status::OK;
}

Status HttpCurlClient::GetIp(std::string &ip)
{
    if (!ip_.empty()) {
        std::string obj(ip_);
        ip = obj;
    } else {
        MEDIA_LOG_E("Get ip failed, ip is null.");
    }
    return Status::OK;
}

void HttpCurlClient::InitCurProxy(const std::string& url)
{
    std::string host;
    std::string exclusions;
    int32_t port = 0;
    GetHttpProxyInfo(host, port, exclusions);
    if (!host.empty() && !IsHostNameExcluded(url, exclusions, ",")) {
        MEDIA_LOG_I("InitCurlEnvironment host: " PUBLIC_LOG_S ", port " PUBLIC_LOG_U32 ", exclusions " PUBLIC_LOG_S,
            host.c_str(), port, exclusions.c_str());
        curl_easy_setopt(easyHandle_, CURLOPT_PROXY, host.c_str());
        curl_easy_setopt(easyHandle_, CURLOPT_PROXYPORT, port);
        auto curlTunnelValue = (url.find("https://") != std::string::npos) ? 1L : 0L;
        curl_easy_setopt(easyHandle_, CURLOPT_HTTPPROXYTUNNEL, curlTunnelValue);
        auto proxyType = (host.find("https://") != std::string::npos) ? CURLPROXY_HTTPS : CURLPROXY_HTTP;
        curl_easy_setopt(easyHandle_, CURLOPT_PROXYTYPE, proxyType);
    } else {
        if (IsHostNameExcluded(url, exclusions, ",")) {
            MEDIA_LOG_I("InitCurlEnvironment host name is excluded.");
        }
    }
}

void HttpCurlClient::InitCurlEnvironment(const std::string& url, int32_t timeoutMs)
{
    curl_easy_setopt(easyHandle_, CURLOPT_URL, UrlParse(url).c_str());
    curl_easy_setopt(easyHandle_, CURLOPT_CONNECTTIMEOUT, 5); // 5
    curl_easy_setopt(easyHandle_, CURLOPT_SSL_VERIFYPEER, 0L);
    curl_easy_setopt(easyHandle_, CURLOPT_SSL_VERIFYHOST, 0L);
#ifndef CA_DIR
    curl_easy_setopt(easyHandle_, CURLOPT_CAINFO, "/etc/ssl/certs/" "cacert.pem");
#else
    curl_easy_setopt(easyHandle_, CURLOPT_CAINFO, CA_DIR "cacert.pem");
#endif
    curl_easy_setopt(easyHandle_, CURLOPT_HTTPGET, 1L);
    curl_easy_setopt(easyHandle_, CURLOPT_FORBID_REUSE, 0L);
    curl_easy_setopt(easyHandle_, CURLOPT_FOLLOWLOCATION, 1L);
    curl_easy_setopt(easyHandle_, CURLOPT_WRITEFUNCTION, rxBody_);
    curl_easy_setopt(easyHandle_, CURLOPT_WRITEDATA, userParam_);
    curl_easy_setopt(easyHandle_, CURLOPT_HEADERFUNCTION, rxHeader_);
    curl_easy_setopt(easyHandle_, CURLOPT_HEADERDATA, userParam_);
    curl_easy_setopt(easyHandle_, CURLOPT_TCP_KEEPALIVE, 1L);
    curl_easy_setopt(easyHandle_, CURLOPT_TCP_KEEPINTVL, 5L); // 5 心跳
    int32_t timeout = timeoutMs > 0 ? timeoutMs / MILLS_TO_SECOND : DEFAULT_LOW_SPEED_TIME;
    curl_easy_setopt(easyHandle_, CURLOPT_LOW_SPEED_LIMIT, DEFAULT_LOW_SPEED_LIMIT);
    curl_easy_setopt(easyHandle_, CURLOPT_LOW_SPEED_TIME, timeout);
    InitCurProxy(url);
}

std::string HttpCurlClient::UrlParse(const std::string& url) const
{
    std::string s;
    std::regex_replace(std::back_inserter(s), url.begin(), url.end(), std::regex(" "), "%20");
    return s;
}

void HttpCurlClient::CheckRequestRange(long startPos, int len)
{
    if (startPos >= 0) {
        char requestRange[128] = {0};
        if (len > 0) {
            snprintf_s(requestRange, sizeof(requestRange), sizeof(requestRange) - 1, "%ld-%ld",
                       startPos, startPos + len - 1);
        } else {
            snprintf_s(requestRange, sizeof(requestRange), sizeof(requestRange) - 1, "%ld-", startPos);
        }
        MEDIA_LOG_DD("RequestData: requestRange " PUBLIC_LOG_S, requestRange);
        std::string requestStr(requestRange);
        AutoLock lock(mutex_);
        if (easyHandle_) {
            curl_easy_setopt(easyHandle_, CURLOPT_RANGE, requestStr.c_str());
        }
    }
}

// RequestData run in HttpDownload thread,
// Open, Close, Deinit run in other thread.
// Should call Open before start HttpDownload thread.
// Should Pause HttpDownload thread then Close, Deinit.
Status HttpCurlClient::RequestData(long startPos, int len, const RequestInfo& requestInfo,
    HandleResponseCbFunc completedCb)
{
    FALSE_RETURN_V(easyHandle_ != nullptr, Status::ERROR_NULL_POINTER);
    CheckRequestRange(startPos, len);
    if (isFirstRequest_) {
        headerList_ = curl_slist_append(headerList_, "Accept: */*");
        headerList_ = curl_slist_append(headerList_, "Connection: Keep-alive");
        headerList_ = curl_slist_append(headerList_, "Keep-Alive: timeout=120");
        isFirstRequest_ = false;
    }
    int32_t clientCode = 0;
    int32_t serverCode = 0;
    Status ret = Status::OK;
    {
        AutoLock lock(mutex_);
        FALSE_RETURN_V(easyHandle_ != nullptr, Status::ERROR_NULL_POINTER);
        curl_easy_setopt(easyHandle_, CURLOPT_HTTPHEADER, headerList_);
        MEDIA_LOG_D("RequestData: startPos " PUBLIC_LOG_D32 ", len " PUBLIC_LOG_D32, static_cast<int>(startPos), len);
        CURLcode returnCode = curl_easy_perform(easyHandle_);
        if (returnCode != CURLE_OK) {
            MEDIA_LOG_E("Curl error " PUBLIC_LOG_D32, returnCode);
            clientCode = returnCode;
            ret = Status::ERROR_CLIENT;
        } else {
            int64_t httpCode = 0;
            curl_easy_getinfo(easyHandle_, CURLINFO_RESPONSE_CODE, &httpCode);
            if (httpCode >= HTTP_ERROR_THRESHOLD) {
                MEDIA_LOG_E("Http error " PUBLIC_LOG_D64, httpCode);
                serverCode = httpCode;
                ret = Status::ERROR_SERVER;
            }
            SetIp();
        }
    }
    completedCb(clientCode, serverCode, ret);
    return ret;
}

Status HttpCurlClient::SetIp()
{
    FALSE_RETURN_V(easyHandle_ != nullptr, Status::ERROR_NULL_POINTER);
    Status retSetIp = Status::OK;
    if (!ipFlag_) {
        char* ip = nullptr;
        if (!curl_easy_getinfo(easyHandle_, CURLINFO_PRIMARY_IP, &ip) && ip) {
            ip_ = ip;
            ipFlag_ = true;
        } else {
            ip_ = "";
            MEDIA_LOG_E("Set sever ip failed.");
            retSetIp = Status::ERROR_UNKNOWN;
        }
    }
    return retSetIp;
}

void HttpCurlClient::SetAppUid(int32_t appUid)
{
    appUid_ = appUid;
}
}
}
}
}