mirror of
https://github.com/openharmony/update_updateservice.git
synced 2026-07-01 21:03:59 -04:00
mamingshuai
109 lines
3.2 KiB
C++
Executable File
109 lines
3.2 KiB
C++
Executable File
/*
|
|
* Copyright (c) 2021 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.
|
|
*/
|
|
#ifndef UPDATER_THREAD_H
|
|
#define UPDATER_THREAD_H
|
|
|
|
#include <atomic>
|
|
#include <functional>
|
|
#include <iostream>
|
|
#include <thread>
|
|
#include "cJSON.h"
|
|
#include "curl/curl.h"
|
|
#include "curl/easy.h"
|
|
#include "if_system_ability_manager.h"
|
|
#include "ipc_skeleton.h"
|
|
#include "iremote_stub.h"
|
|
#include "iupdate_service.h"
|
|
#include "openssl/err.h"
|
|
#include "openssl/ssl.h"
|
|
#include "system_ability.h"
|
|
#include "update_service_stub.h"
|
|
|
|
#define ENGINE_CHECK_NO_LOG(retCode, exper) \
|
|
if (!(retCode)) { \
|
|
exper; \
|
|
}
|
|
|
|
namespace OHOS {
|
|
namespace update_engine {
|
|
constexpr uint32_t DOWNLOAD_FINISH_PERCENT = 100;
|
|
constexpr uint32_t DOWNLOAD_PERIOD_PERCENT = 5;
|
|
constexpr int32_t TIMEOUT_FOR_DOWNLOAD = 600;
|
|
#ifndef UPDATER_UT
|
|
constexpr int32_t TIMEOUT_FOR_CONNECT = 10;
|
|
#else
|
|
constexpr int32_t TIMEOUT_FOR_CONNECT = 1;
|
|
#endif
|
|
|
|
class ProgressThread {
|
|
public:
|
|
ProgressThread() = default;
|
|
virtual ~ProgressThread()
|
|
{
|
|
StopProgress();
|
|
}
|
|
|
|
protected:
|
|
int32_t StartProgress();
|
|
void StopProgress();
|
|
void ExecuteThreadFunc();
|
|
|
|
virtual bool ProcessThreadExecute() = 0;
|
|
virtual void ProcessThreadExit() = 0;
|
|
|
|
private:
|
|
std::thread *pDealThread_ { nullptr };
|
|
std::mutex mutex_;
|
|
std::condition_variable condition_;
|
|
bool isWake_ = false;
|
|
bool isExitThread_ = false;
|
|
};
|
|
|
|
class DownloadThread : public ProgressThread {
|
|
public:
|
|
using ProgressCallback = std::function<int (const std::string &fileName, const Progress &progress)>;
|
|
DownloadThread(ProgressCallback callback) : ProgressThread(), callback_(callback) {}
|
|
~DownloadThread() override {}
|
|
|
|
int32_t StartDownload(const std::string &fileName, const std::string &url);
|
|
void StopDownload();
|
|
|
|
static size_t GetLocalFileLength(const std::string &fileName);
|
|
static size_t WriteFunc(void *ptr, size_t size, size_t nmemb, const void *stream);
|
|
static int32_t DownloadProgress(const void *localData,
|
|
double dlTotal, double dlNow, double ulTotal, double ulNow);
|
|
|
|
double GetPackageSize()
|
|
{
|
|
return static_cast<double>(packageSize_);
|
|
};
|
|
protected:
|
|
bool ProcessThreadExecute() override;
|
|
void ProcessThreadExit() override;
|
|
int32_t DownloadCallback(uint32_t percent, UpgradeStatus status, const std::string &error);
|
|
|
|
private:
|
|
Progress downloadProgress_ {};
|
|
ProgressCallback callback_;
|
|
CURL *downloadHandle_ { nullptr };
|
|
FILE *downloadFile_ { nullptr };
|
|
std::string serverUrl_;
|
|
std::atomic<bool> exitDownload_ { false };
|
|
size_t packageSize_ { 1 };
|
|
std::string downloadFileName_;
|
|
};
|
|
}
|
|
} // namespace OHOS
|
|
#endif // UPDATER_THREAD_H
|