1 /* 2 * Copyright (c) 2021-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 USBMGR_USB_SERVICE_H 17 #define USBMGR_USB_SERVICE_H 18 19 #include <map> 20 #include <vector> 21 #include "delayed_sp_singleton.h" 22 #include "iremote_object.h" 23 #include "iusb_srv.h" 24 #include "system_ability.h" 25 #include "system_ability_status_change_stub.h" 26 #include "usb_device_manager.h" 27 #include "usb_host_manager.h" 28 #include "usb_port_manager.h" 29 #include "usb_right_manager.h" 30 #include "usb_server_event_handler.h" 31 #include "usb_server_stub.h" 32 #include "usb_service_subscriber.h" 33 #include "usbd_type.h" 34 #include "v1_0/iusb_interface.h" 35 #include "v1_0/iusbd_bulk_callback.h" 36 #include "v1_0/iusbd_subscriber.h" 37 #include "v1_0/usb_types.h" 38 39 namespace OHOS { 40 namespace USB { 41 const std::string USB_HOST = "usb_host"; 42 const std::string USB_DEVICE = "usb_device"; 43 const std::string USB_PORT = "usb_port"; 44 const std::string USB_HELP = "-h"; 45 class UsbService : public SystemAbility, public UsbServerStub { 46 DECLARE_SYSTEM_ABILITY(UsbService) 47 DECLARE_DELAYED_SP_SINGLETON(UsbService); 48 49 public: 50 void OnStart() override; 51 void OnStop() override; 52 int Dump(int fd, const std::vector<std::u16string> &args) override; 53 IsServiceReady()54 bool IsServiceReady() const 55 { 56 return ready_; 57 } 58 GetHandler()59 std::shared_ptr<UsbServerEventHandler> GetHandler() const 60 { 61 return handler_; 62 } 63 64 int32_t OpenDevice(uint8_t busNum, uint8_t devAddr) override; 65 bool HasRight(std::string deviceName) override; 66 int32_t RequestRight(std::string deviceName) override; 67 int32_t RemoveRight(std::string deviceName) override; 68 int32_t GetDevices(std::vector<UsbDevice> &deviceList) override; 69 int32_t GetCurrentFunctions(int32_t &funcs) override; 70 int32_t SetCurrentFunctions(int32_t funcs) override; 71 int32_t UsbFunctionsFromString(std::string_view funcs) override; 72 std::string UsbFunctionsToString(int32_t funcs) override; 73 int32_t GetPorts(std::vector<UsbPort> &ports) override; 74 int32_t GetSupportedModes(int32_t portId, int32_t &result) override; 75 int32_t SetPortRole(int32_t portId, int32_t powerRole, int32_t dataRole) override; 76 77 int32_t ClaimInterface(uint8_t busNum, uint8_t devAddr, uint8_t interfaceid, uint8_t force) override; 78 int32_t ReleaseInterface(uint8_t busNum, uint8_t devAddr, uint8_t interfaceid) override; 79 int32_t BulkTransferRead(const HDI::Usb::V1_0::UsbDev &dev, const HDI::Usb::V1_0::UsbPipe &pipe, 80 std::vector<uint8_t> &bufferData, int32_t timeOut) override; 81 int32_t BulkTransferWrite(const HDI::Usb::V1_0::UsbDev &dev, const HDI::Usb::V1_0::UsbPipe &pipe, 82 const std::vector<uint8_t> &bufferData, int32_t timeOut) override; 83 int32_t ControlTransfer(const HDI::Usb::V1_0::UsbDev &dev, const HDI::Usb::V1_0::UsbCtrlTransfer &ctrl, 84 std::vector<uint8_t> &bufferData) override; 85 int32_t SetActiveConfig(uint8_t busNum, uint8_t devAddr, uint8_t configIndex) override; 86 int32_t GetActiveConfig(uint8_t busNum, uint8_t devAddr, uint8_t &configIndex) override; 87 int32_t SetInterface(uint8_t busNum, uint8_t devAddr, uint8_t interfaceid, uint8_t altIndex) override; 88 int32_t GetRawDescriptor(uint8_t busNum, uint8_t devAddr, std::vector<uint8_t> &bufferData) override; 89 int32_t GetFileDescriptor(uint8_t busNum, uint8_t devAddr, int32_t &fd) override; 90 int32_t RequestQueue(const HDI::Usb::V1_0::UsbDev &dev, const HDI::Usb::V1_0::UsbPipe &pipe, 91 const std::vector<uint8_t> &clientData, const std::vector<uint8_t> &bufferData) override; 92 int32_t RequestWait(const HDI::Usb::V1_0::UsbDev &dev, int32_t timeOut, std::vector<uint8_t> &clientData, 93 std::vector<uint8_t> &bufferData) override; 94 int32_t RequestCancel(uint8_t busNum, uint8_t devAddr, uint8_t interfaceid, uint8_t endpointId) override; 95 int32_t Close(uint8_t busNum, uint8_t devAddr) override; 96 bool AddDevice(uint8_t busNum, uint8_t devAddr); 97 bool DelDevice(uint8_t busNum, uint8_t devAddr); 98 void UpdateUsbPort(int32_t portId, int32_t powerRole, int32_t dataRole, int32_t mode); 99 void UpdateDeviceState(int32_t status); 100 int32_t GetDeviceInfo(uint8_t busNum, uint8_t devAddr, UsbDevice &dev); 101 int32_t GetDeviceInfoDescriptor(const HDI::Usb::V1_0::UsbDev &uDev, std::vector<uint8_t> &descriptor, 102 UsbDevice &dev); 103 int32_t GetConfigDescriptor(UsbDevice &dev, std::vector<uint8_t> &descriptor); 104 105 int32_t RegBulkCallback(const HDI::Usb::V1_0::UsbDev &devInfo, const HDI::Usb::V1_0::UsbPipe &pipe, 106 const sptr<IRemoteObject> &cb) override; 107 int32_t UnRegBulkCallback(const HDI::Usb::V1_0::UsbDev &devInfo, const HDI::Usb::V1_0::UsbPipe &pipe) override; 108 int32_t BulkRead(const HDI::Usb::V1_0::UsbDev &devInfo, const HDI::Usb::V1_0::UsbPipe &pipe, 109 sptr<Ashmem> &ashmem) override; 110 int32_t BulkWrite(const HDI::Usb::V1_0::UsbDev &devInfo, const HDI::Usb::V1_0::UsbPipe &pipe, 111 sptr<Ashmem> &ashmem) override; 112 int32_t BulkCancel(const HDI::Usb::V1_0::UsbDev &devInfo, const HDI::Usb::V1_0::UsbPipe &pipe) override; 113 int32_t AddRight(const std::string &bundleName, const std::string &deviceName) override; 114 115 private: 116 class SystemAbilityStatusChangeListener : public SystemAbilityStatusChangeStub { 117 public: 118 explicit SystemAbilityStatusChangeListener(sptr<UsbServiceSubscriber> usbdSubscriber); 119 ~SystemAbilityStatusChangeListener() = default; 120 void OnAddSystemAbility(int32_t systemAbilityId, const std::string &deviceId) override; 121 void OnRemoveSystemAbility(int32_t systemAbilityId, const std::string &deviceId) override; 122 private: 123 sptr<UsbServiceSubscriber> usbdSubscriber_; 124 }; 125 126 private: 127 bool Init(); 128 bool InitUsbd(); 129 bool IsCommonEventServiceAbilityExist(); 130 bool GetBundleName(std::string &bundleName); 131 int32_t FillDevStrings(UsbDevice &dev); 132 std::string GetDevStringValFromIdx(uint8_t busNum, uint8_t devAddr, uint8_t idx); 133 void DumpHelp(int32_t fd); 134 bool ready_ = false; 135 int32_t commEventRetryTimes_ = 0; 136 std::mutex mutex_; 137 std::shared_ptr<UsbHostManager> usbHostManger_; 138 std::shared_ptr<UsbRightManager> usbRightManager_; 139 std::shared_ptr<UsbPortManager> usbPortManager_; 140 std::shared_ptr<UsbDeviceManager> usbDeviceManager_; 141 std::shared_ptr<AppExecFwk::EventRunner> eventRunner_; 142 std::shared_ptr<UsbServerEventHandler> handler_; 143 sptr<UsbServiceSubscriber> usbdSubscriber_; 144 sptr<HDI::Usb::V1_0::IUsbdBulkCallback> hdiCb_ = nullptr; 145 sptr<HDI::Usb::V1_0::IUsbInterface> usbd_ = nullptr; 146 }; 147 } // namespace USB 148 } // namespace OHOS 149 #endif // USBMGR_USB_SERVICE_H 150