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 #include "payload_data/get_device_info_data.h"
16 #include "media_log.h"
17 #include "media_mtp_utils.h"
18 #include "mtp_constants.h"
19 #include "mtp_operation_utils.h"
20 #include "mtp_packet_tools.h"
21 #include "mtp_manager.h"
22 #include "playback_formats.h"
23
24 using namespace std;
25
26 namespace OHOS {
27 namespace Media {
28 static const char *EXTENSION_DESC = "microsoft.com: 1.0; openharmony: 1.0;";
29
30 static const uint16_t CaptureFormats[] = {};
31
32 static const uint16_t DeviceProperties[] = {
33 MTP_DEVICE_PROPERTY_SYNCHRONIZATION_PARTNER_CODE,
34 MTP_DEVICE_PROPERTY_DEVICE_FRIENDLY_NAME_CODE,
35 MTP_DEVICE_PROPERTY_IMAGE_SIZE_CODE,
36 MTP_DEVICE_PROPERTY_BATTERY_LEVEL_CODE,
37 MTP_DEVICE_PROPERTY_PERCEIVED_DEVICE_TYPE_CODE,
38 MTP_DEVICE_PROPERTY_SESSION_INITIATOR_VERSION_INFO_CODE,
39 };
40
41 static const uint16_t Events[] = {
42 MTP_EVENT_OBJECT_ADDED_CODE,
43 MTP_EVENT_OBJECT_REMOVED_CODE,
44 MTP_EVENT_STORE_ADDED_CODE,
45 MTP_EVENT_STORE_REMOVED_CODE,
46 MTP_EVENT_DEVICE_PROP_CHANGED_CODE,
47 MTP_EVENT_OBJECT_INFO_CHANGED_CODE,
48 };
49
50 static const uint16_t Operations[] = {
51 MTP_OPERATION_GET_DEVICE_INFO_CODE,
52 MTP_OPERATION_OPEN_SESSION_CODE,
53 MTP_OPERATION_CLOSE_SESSION_CODE,
54 MTP_OPERATION_GET_STORAGE_IDS_CODE,
55 MTP_OPERATION_GET_STORAGE_INFO_CODE,
56 MTP_OPERATION_GET_NUM_OBJECTS_CODE,
57 MTP_OPERATION_GET_OBJECT_HANDLES_CODE,
58 MTP_OPERATION_GET_OBJECT_INFO_CODE,
59 MTP_OPERATION_GET_OBJECT_CODE,
60 MTP_OPERATION_GET_THUMB_CODE,
61 MTP_OPERATION_DELETE_OBJECT_CODE,
62 MTP_OPERATION_SEND_OBJECT_INFO_CODE,
63 MTP_OPERATION_SEND_OBJECT_CODE,
64 MTP_OPERATION_RESET_DEVICE_CODE,
65 MTP_OPERATION_GET_DEVICE_PROP_DESC_CODE,
66 MTP_OPERATION_GET_DEVICE_PROP_VALUE_CODE,
67 MTP_OPERATION_SET_DEVICE_PROP_VALUE_CODE,
68 MTP_OPERATION_RESET_DEVICE_PROP_VALUE_CODE,
69 MTP_OPERATION_MOVE_OBJECT_CODE,
70 MTP_OPERATION_COPY_OBJECT_CODE,
71 MTP_OPERATION_GET_PARTIAL_OBJECT_CODE,
72 MTP_OPERATION_GET_OBJECT_PROPS_SUPPORTED_CODE,
73 MTP_OPERATION_GET_OBJECT_PROP_DESC_CODE,
74 MTP_OPERATION_GET_OBJECT_PROP_VALUE_CODE,
75 MTP_OPERATION_SET_OBJECT_PROP_VALUE_CODE,
76 MTP_OPERATION_GET_OBJECT_PROP_LIST_CODE,
77 };
78
79 static const uint16_t OperationsPtp[] = {
80 MTP_OPERATION_GET_DEVICE_INFO_CODE,
81 MTP_OPERATION_OPEN_SESSION_CODE,
82 MTP_OPERATION_CLOSE_SESSION_CODE,
83 MTP_OPERATION_GET_STORAGE_IDS_CODE,
84 MTP_OPERATION_GET_STORAGE_INFO_CODE,
85 MTP_OPERATION_GET_NUM_OBJECTS_CODE,
86 MTP_OPERATION_GET_OBJECT_HANDLES_CODE,
87 MTP_OPERATION_GET_OBJECT_INFO_CODE,
88 MTP_OPERATION_GET_OBJECT_CODE,
89 MTP_OPERATION_GET_THUMB_CODE,
90 MTP_OPERATION_SEND_OBJECT_INFO_CODE,
91 MTP_OPERATION_SEND_OBJECT_CODE,
92 MTP_OPERATION_RESET_DEVICE_CODE,
93 MTP_OPERATION_GET_DEVICE_PROP_DESC_CODE,
94 MTP_OPERATION_GET_DEVICE_PROP_VALUE_CODE,
95 MTP_OPERATION_SET_DEVICE_PROP_VALUE_CODE,
96 MTP_OPERATION_RESET_DEVICE_PROP_VALUE_CODE,
97 MTP_OPERATION_COPY_OBJECT_CODE,
98 MTP_OPERATION_GET_PARTIAL_OBJECT_CODE,
99 MTP_OPERATION_GET_OBJECT_PROPS_SUPPORTED_CODE,
100 MTP_OPERATION_GET_OBJECT_PROP_DESC_CODE,
101 MTP_OPERATION_GET_OBJECT_PROP_VALUE_CODE,
102 MTP_OPERATION_SET_OBJECT_PROP_VALUE_CODE,
103 MTP_OPERATION_GET_OBJECT_PROP_LIST_CODE,
104 };
105
106 const int EXTENSION_ID = 6;
GetDeviceInfoData(std::shared_ptr<MtpOperationContext> & context)107 GetDeviceInfoData::GetDeviceInfoData(std::shared_ptr<MtpOperationContext> &context)
108 : PayloadData(context), functionalMode_(0), serialNum_("0")
109 {
110 standardVersion_ = MTP_STANDARD_VERSION;
111 vendorExtensionID_ = EXTENSION_ID;
112 vendorExtensionVersion_ = MTP_STANDARD_VERSION;
113 vendorExtensionDesc_ = EXTENSION_DESC;
114 manufacturer_ = DEFAULT_PRODUCT_MANUFACTURER;
115 model_ = DEFAULT_PRODUCT_MODEL;
116 version_ = DEFAULT_PRODUCT_SOFTWARE_VERSION;
117 Maker(mOutBuffer);
118 }
119
GetDeviceInfoData()120 GetDeviceInfoData::GetDeviceInfoData()
121 {
122 standardVersion_ = MTP_STANDARD_VERSION;
123 vendorExtensionID_ = EXTENSION_ID;
124 vendorExtensionVersion_ = MTP_STANDARD_VERSION;
125 vendorExtensionDesc_ = EXTENSION_DESC;
126 functionalMode_ = 0;
127 manufacturer_ = DEFAULT_PRODUCT_MANUFACTURER;
128 model_ = DEFAULT_PRODUCT_MODEL;
129 version_ = DEFAULT_PRODUCT_SOFTWARE_VERSION;
130 serialNum_ = "0";
131 Maker(mOutBuffer);
132 }
133
~GetDeviceInfoData()134 GetDeviceInfoData::~GetDeviceInfoData()
135 {
136 }
137
Parser(const std::vector<uint8_t> & buffer,int32_t readSize)138 int GetDeviceInfoData::Parser(const std::vector<uint8_t> &buffer, int32_t readSize)
139 {
140 return MTP_SUCCESS;
141 }
142
Maker(std::vector<uint8_t> & outBuffer)143 int GetDeviceInfoData::Maker(std::vector<uint8_t> &outBuffer)
144 {
145 MtpPacketTool::PutUInt16(outBuffer, standardVersion_);
146 MtpPacketTool::PutUInt32(outBuffer, vendorExtensionID_);
147 MtpPacketTool::PutUInt16(outBuffer, vendorExtensionVersion_);
148 MtpPacketTool::PutString(outBuffer, vendorExtensionDesc_);
149 MtpPacketTool::PutUInt16(outBuffer, functionalMode_);
150 if (MtpManager::GetInstance().IsMtpMode()) {
151 MtpPacketTool::PutAUInt16(outBuffer, Operations, sizeof(Operations) / sizeof(uint16_t));
152 } else {
153 MtpPacketTool::PutAUInt16(outBuffer, OperationsPtp, sizeof(OperationsPtp) / sizeof(uint16_t));
154 }
155 MtpPacketTool::PutAUInt16(outBuffer, Events, sizeof(Events) / sizeof(uint16_t));
156 MtpPacketTool::PutAUInt16(outBuffer, DeviceProperties, sizeof(DeviceProperties) / sizeof(uint16_t));
157 MtpPacketTool::PutAUInt16(outBuffer, CaptureFormats, sizeof(CaptureFormats) / sizeof(uint16_t));
158 MtpPacketTool::PutAUInt16(outBuffer, PLAYBACK_FORMATS, sizeof(PLAYBACK_FORMATS) / sizeof(uint16_t));
159 MtpPacketTool::PutString(outBuffer, manufacturer_);
160 MtpPacketTool::PutString(outBuffer, model_);
161 MtpPacketTool::PutString(outBuffer, version_);
162 MtpPacketTool::PutString(outBuffer, serialNum_);
163
164 return MTP_SUCCESS;
165 }
166
CalculateSize()167 uint32_t GetDeviceInfoData::CalculateSize()
168 {
169 std::vector<uint8_t> tmpVar;
170
171 int res = Maker(tmpVar);
172 if (res != MTP_SUCCESS) {
173 return res;
174 }
175
176 return tmpVar.size();
177 }
178
SetManufacturer(const std::string & manufacturer)179 void GetDeviceInfoData::SetManufacturer(const std::string &manufacturer)
180 {
181 manufacturer_ = manufacturer;
182 }
183
SetModel(const std::string & model)184 void GetDeviceInfoData::SetModel(const std::string &model)
185 {
186 model_ = model;
187 }
188
SetVersion(const std::string & version)189 void GetDeviceInfoData::SetVersion(const std::string &version)
190 {
191 version_ = version;
192 }
193
SetSerialNum(const std::string & serialNum)194 void GetDeviceInfoData::SetSerialNum(const std::string &serialNum)
195 {
196 serialNum_ = serialNum;
197 }
198 } // namespace Media
199 } // namespace OHOS