1 /*
2  * Copyright (C) 2018 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #pragma once
18 
19 #include <memory>
20 #include <string>
21 #include <unordered_map>
22 #include <utility>
23 #include <vector>
24 
25 #include <BootControlClient.h>
26 #include <aidl/android/hardware/fastboot/IFastboot.h>
27 #include <aidl/android/hardware/health/IHealth.h>
28 
29 #include "commands.h"
30 #include "transport.h"
31 #include "variables.h"
32 
33 class FastbootDevice {
34   public:
35     using BootControlClient = android::hal::BootControlClient;
36     FastbootDevice();
37     ~FastbootDevice();
38 
39     void CloseDevice();
40     void ExecuteCommands();
41     bool WriteStatus(FastbootResult result, const std::string& message);
42     bool HandleData(bool read, std::vector<char>* data);
43     bool HandleData(bool read, char* data, uint64_t size);
44     std::string GetCurrentSlot();
45 
46     // Shortcuts for writing status results.
47     bool WriteOkay(const std::string& message);
48     bool WriteFail(const std::string& message);
49     bool WriteInfo(const std::string& message);
50 
download_data()51     std::vector<char>& download_data() { return download_data_; }
get_transport()52     Transport* get_transport() { return transport_.get(); }
boot_control_hal()53     BootControlClient* boot_control_hal() const { return boot_control_hal_.get(); }
54     BootControlClient* boot1_1() const;
fastboot_hal()55     std::shared_ptr<aidl::android::hardware::fastboot::IFastboot> fastboot_hal() {
56         return fastboot_hal_;
57     }
health_hal()58     std::shared_ptr<aidl::android::hardware::health::IHealth> health_hal() { return health_hal_; }
59 
set_active_slot(const std::string & active_slot)60     void set_active_slot(const std::string& active_slot) { active_slot_ = active_slot; }
61 
62   private:
63     const std::unordered_map<std::string, CommandHandler> kCommandMap;
64 
65     std::unique_ptr<Transport> transport_;
66     std::unique_ptr<BootControlClient> boot_control_hal_;
67     std::shared_ptr<aidl::android::hardware::health::IHealth> health_hal_;
68     std::shared_ptr<aidl::android::hardware::fastboot::IFastboot> fastboot_hal_;
69     std::vector<char> download_data_;
70     std::string active_slot_;
71 };
72