1 /* 2 * Copyright (c) 2022 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 16 #ifndef HDC_DAEMON_UPDATER_H 17 #define HDC_DAEMON_UPDATER_H 18 19 #include <unordered_map> 20 21 #include "commander_factory.h" 22 #include "transfer.h" 23 24 namespace Hdc { 25 class DaemonUpdater : public HdcTransferBase { 26 public: 27 DISALLOW_COPY_MOVE(DaemonUpdater); 28 explicit DaemonUpdater(HTaskInfo hTaskInfo); 29 virtual ~DaemonUpdater(); 30 bool CommandDispatch(const uint16_t command, uint8_t *payload, const int payloadSize) override; 31 32 private: 33 void Init(); 34 void CheckCommand(const uint8_t *payload, int payloadSize); 35 void DataCommand(const uint8_t *payload, int payloadSize) const; 36 void EraseCommand(const uint8_t *payload, int payloadSize); 37 void FormatCommand(const uint8_t *payload, int payloadSize); 38 bool SendToHost(Flashd::CmdType type, Flashd::UpdaterState state, const std::string &msg); 39 std::unique_ptr<Flashd::Commander> CreateCommander(const std::string &cmd); 40 bool IsDeviceLocked() const; 41 42 bool isInit_ = false; 43 std::unique_ptr<Flashd::Commander> commander_ = nullptr; 44 std::unordered_map<uint16_t, std::function<void(uint8_t *payload, int payloadSize)>> cmdFunc_; 45 static std::atomic<bool> isRunning_; 46 }; 47 } // namespace Hdc 48 #endif