Hridya Valsaraju | dea91b4 | 2018-07-17 11:14:01 -0700 | [diff] [blame] | 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 | |
Hridya Valsaraju | 31d2c26 | 2018-07-20 13:35:50 -0700 | [diff] [blame] | 25 | #include <android/hardware/boot/1.0/IBootControl.h> |
Hridya Valsaraju | bf9f8d1 | 2018-09-05 16:57:24 -0700 | [diff] [blame] | 26 | #include <android/hardware/fastboot/1.0/IFastboot.h> |
Hridya Valsaraju | 47658ca | 2018-09-28 11:41:22 -0700 | [diff] [blame] | 27 | #include <android/hardware/health/2.0/IHealth.h> |
Hridya Valsaraju | 31d2c26 | 2018-07-20 13:35:50 -0700 | [diff] [blame] | 28 | |
Hridya Valsaraju | dea91b4 | 2018-07-17 11:14:01 -0700 | [diff] [blame] | 29 | #include "commands.h" |
| 30 | #include "transport.h" |
Hridya Valsaraju | 31d2c26 | 2018-07-20 13:35:50 -0700 | [diff] [blame] | 31 | #include "variables.h" |
Hridya Valsaraju | dea91b4 | 2018-07-17 11:14:01 -0700 | [diff] [blame] | 32 | |
| 33 | class FastbootDevice { |
| 34 | public: |
| 35 | FastbootDevice(); |
| 36 | ~FastbootDevice(); |
| 37 | |
| 38 | void CloseDevice(); |
| 39 | void ExecuteCommands(); |
| 40 | bool WriteStatus(FastbootResult result, const std::string& message); |
| 41 | bool HandleData(bool read, std::vector<char>* data); |
Hridya Valsaraju | 31d2c26 | 2018-07-20 13:35:50 -0700 | [diff] [blame] | 42 | std::string GetCurrentSlot(); |
Hridya Valsaraju | dea91b4 | 2018-07-17 11:14:01 -0700 | [diff] [blame] | 43 | |
David Anderson | 0f62663 | 2018-08-31 16:44:25 -0700 | [diff] [blame] | 44 | // Shortcuts for writing status results. |
David Anderson | 856b7ec | 2018-08-08 17:58:56 -0700 | [diff] [blame] | 45 | bool WriteOkay(const std::string& message); |
| 46 | bool WriteFail(const std::string& message); |
David Anderson | 0f62663 | 2018-08-31 16:44:25 -0700 | [diff] [blame] | 47 | bool WriteInfo(const std::string& message); |
David Anderson | 856b7ec | 2018-08-08 17:58:56 -0700 | [diff] [blame] | 48 | |
David Anderson | 12211d1 | 2018-07-24 15:21:20 -0700 | [diff] [blame] | 49 | std::vector<char>& download_data() { return download_data_; } |
Hridya Valsaraju | dea91b4 | 2018-07-17 11:14:01 -0700 | [diff] [blame] | 50 | Transport* get_transport() { return transport_.get(); } |
Hridya Valsaraju | 31d2c26 | 2018-07-20 13:35:50 -0700 | [diff] [blame] | 51 | android::sp<android::hardware::boot::V1_0::IBootControl> boot_control_hal() { |
| 52 | return boot_control_hal_; |
| 53 | } |
Hridya Valsaraju | bf9f8d1 | 2018-09-05 16:57:24 -0700 | [diff] [blame] | 54 | android::sp<android::hardware::fastboot::V1_0::IFastboot> fastboot_hal() { |
| 55 | return fastboot_hal_; |
| 56 | } |
Hridya Valsaraju | 47658ca | 2018-09-28 11:41:22 -0700 | [diff] [blame] | 57 | android::sp<android::hardware::health::V2_0::IHealth> health_hal() { return health_hal_; } |
Hridya Valsaraju | dea91b4 | 2018-07-17 11:14:01 -0700 | [diff] [blame] | 58 | |
Hridya Valsaraju | 20bdf89 | 2018-10-10 11:02:19 -0700 | [diff] [blame] | 59 | void set_active_slot(const std::string& active_slot) { active_slot_ = active_slot; } |
| 60 | |
Hridya Valsaraju | dea91b4 | 2018-07-17 11:14:01 -0700 | [diff] [blame] | 61 | private: |
| 62 | const std::unordered_map<std::string, CommandHandler> kCommandMap; |
| 63 | |
| 64 | std::unique_ptr<Transport> transport_; |
Hridya Valsaraju | 31d2c26 | 2018-07-20 13:35:50 -0700 | [diff] [blame] | 65 | android::sp<android::hardware::boot::V1_0::IBootControl> boot_control_hal_; |
Hridya Valsaraju | 47658ca | 2018-09-28 11:41:22 -0700 | [diff] [blame] | 66 | android::sp<android::hardware::health::V2_0::IHealth> health_hal_; |
Hridya Valsaraju | bf9f8d1 | 2018-09-05 16:57:24 -0700 | [diff] [blame] | 67 | android::sp<android::hardware::fastboot::V1_0::IFastboot> fastboot_hal_; |
Hridya Valsaraju | dea91b4 | 2018-07-17 11:14:01 -0700 | [diff] [blame] | 68 | std::vector<char> download_data_; |
Hridya Valsaraju | 20bdf89 | 2018-10-10 11:02:19 -0700 | [diff] [blame] | 69 | std::string active_slot_; |
Hridya Valsaraju | dea91b4 | 2018-07-17 11:14:01 -0700 | [diff] [blame] | 70 | }; |