diff --git a/services/flow_update/update_bin/component_processor.h b/services/flow_update/update_bin/component_processor.h index 72599a14181ae4cfc63a27d27ebcc80a005e57cc..f33642558dcb7dac17043600670308523ed572b3 100644 --- a/services/flow_update/update_bin/component_processor.h +++ b/services/flow_update/update_bin/component_processor.h @@ -133,9 +133,10 @@ public: int32_t PreProcess(Uscript::UScriptEnv &env) override; int32_t DoProcess(Uscript::UScriptEnv &env) override; int32_t PostProcess(Uscript::UScriptEnv &env) override; -private: - int GetWritePathAndOffset(const std::string &partitionName, std::string &writePath, uint64_t &offset, +protected: + virtual int GetWritePathAndOffset(const std::string &partitionName, std::string &writePath, uint64_t &offset, uint64_t &partitionSize); +private: virtual int RawImageWriteProcessor(const Hpackage::PkgBuffer &buffer, size_t size, size_t start, bool isFinish, const void* context); std::unique_ptr writer_ = nullptr;