1/*
2 * Copyright (c) 2021 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15#ifndef UPDATE_PROCESSOR_H
16#define UPDATE_PROCESSOR_H
17
18#include <cstdio>
19#include <string>
20#include <sys/wait.h>
21#include <unistd.h>
22#include "applypatch/data_writer.h"
23#include "pkg_manager.h"
24#include "script_instruction.h"
25#include "script_manager.h"
26
27using Uscript::UScriptEnv;
28using Uscript::UScriptInstructionFactory;
29using Uscript::UScriptInstructionFactoryPtr;
30using Uscript::UScriptInstructionPtr;
31
32namespace Updater {
33enum EXIT_CODES {
34    EXIT_INVALID_ARGS = EXIT_SUCCESS + 1,
35    EXIT_READ_PACKAGE_ERROR = 2,
36    EXIT_FOUND_SCRIPT_ERROR = 3,
37    EXIT_PARSE_SCRIPT_ERROR = 4,
38    EXIT_EXEC_SCRIPT_ERROR = 5,
39    EXIT_VERIFY_SCRIPT_ERROR = 6,
40};
41
42int ProcessUpdater(bool retry, int pipeFd, const std::string &packagePath, const std::string &keyPath);
43void GetPartitionPathFromName(const std::string &partitionName, std::string &partitionPath);
44
45#ifdef __cplusplus
46#if __cplusplus
47extern "C" {
48#endif
49#endif /* __cplusplus */
50int32_t GetFinalBinaryResult(int32_t result);
51#ifdef __cplusplus
52#if __cplusplus
53}
54#endif
55#endif /* __cplusplus */
56
57class UpdaterInstructionFactory : public UScriptInstructionFactory {
58public:
59    virtual int32_t CreateInstructionInstance(UScriptInstructionPtr& instr, const std::string& name);
60    UpdaterInstructionFactory() {}
61    virtual ~UpdaterInstructionFactory() {}
62};
63
64class UScriptInstructionRawImageWrite : public Uscript::UScriptInstruction {
65public:
66    UScriptInstructionRawImageWrite() {}
67    virtual ~UScriptInstructionRawImageWrite() {}
68    int32_t Execute(Uscript::UScriptEnv &env, Uscript::UScriptContext &context) override;
69
70private:
71    static int RawImageWriteProcessor(const Hpackage::PkgBuffer &buffer, size_t size, size_t start, bool isFinish,
72        const void* context);
73    int GetWritePathAndOffset(const std::string &partitionName, std::string &writePath, uint64_t &offset,
74        uint64_t &partitionSize);
75    bool WriteRawImage(const std::string &partitionName, const std::unique_ptr<DataWriter> &writer,
76        uint64_t partitionSize, Uscript::UScriptEnv &env);
77    static size_t totalSize_;
78    static size_t readSize_;
79};
80
81class UScriptInstructionPkgExtract : public Uscript::UScriptInstruction {
82public:
83    UScriptInstructionPkgExtract() {}
84    virtual ~UScriptInstructionPkgExtract() {}
85    int32_t Execute(Uscript::UScriptEnv &env, Uscript::UScriptContext &context) override;
86};
87
88class UScriptInstructionPkgExtractRetSuc : public UScriptInstructionPkgExtract {
89public:
90    UScriptInstructionPkgExtractRetSuc() {}
91    ~UScriptInstructionPkgExtractRetSuc() override {}
92    int32_t Execute(Uscript::UScriptEnv &env, Uscript::UScriptContext &context) override;
93};
94
95class UScriptInstructionUpdateFromBin : public Uscript::UScriptInstruction {
96public:
97    UScriptInstructionUpdateFromBin() {}
98    virtual ~UScriptInstructionUpdateFromBin() {}
99    int32_t Execute(Uscript::UScriptEnv &env, Uscript::UScriptContext &context) override;
100    constexpr static uint32_t STASH_BUFFER_SIZE = 1024 * 1024 * 4;
101    constexpr static uint32_t BUFFER_NUM = 16;
102    static int UnCompressDataProducer(const Hpackage::PkgBuffer &buffer, size_t size, size_t start, bool isFinish,
103        const void* context);
104private:
105    static size_t stashDataSize_;
106};
107} // Updater
108#endif /* UPDATE_PROCESSOR_H */
109