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 MTP_OPERATION_GET_OBJECT_REFERENCES_CODE,
78 MTP_OPERATION_SET_OBJECT_REFERENCES_CODE,
79 };
80
81 static const uint16_t OperationsPtp[] = {
82 MTP_OPERATION_GET_DEVICE_INFO_CODE,
83 MTP_OPERATION_OPEN_SESSION_CODE,
84 MTP_OPERATION_CLOSE_SESSION_CODE,
85 MTP_OPERATION_GET_STORAGE_IDS_CODE,
86 MTP_OPERATION_GET_STORAGE_INFO_CODE,
87 MTP_OPERATION_GET_NUM_OBJECTS_CODE,
88 MTP_OPERATION_GET_OBJECT_HANDLES_CODE,
89 MTP_OPERATION_GET_OBJECT_INFO_CODE,
90 MTP_OPERATION_GET_OBJECT_CODE,
91 MTP_OPERATION_GET_THUMB_CODE,
92 MTP_OPERATION_SEND_OBJECT_INFO_CODE,
93 MTP_OPERATION_SEND_OBJECT_CODE,
94 MTP_OPERATION_RESET_DEVICE_CODE,
95 MTP_OPERATION_GET_DEVICE_PROP_DESC_CODE,
96 MTP_OPERATION_GET_DEVICE_PROP_VALUE_CODE,
97 MTP_OPERATION_SET_DEVICE_PROP_VALUE_CODE,
98 MTP_OPERATION_RESET_DEVICE_PROP_VALUE_CODE,
99 MTP_OPERATION_COPY_OBJECT_CODE,
100 MTP_OPERATION_GET_PARTIAL_OBJECT_CODE,
101 MTP_OPERATION_GET_OBJECT_PROPS_SUPPORTED_CODE,
102 MTP_OPERATION_GET_OBJECT_PROP_DESC_CODE,
103 MTP_OPERATION_GET_OBJECT_PROP_VALUE_CODE,
104 MTP_OPERATION_SET_OBJECT_PROP_VALUE_CODE,
105 MTP_OPERATION_GET_OBJECT_PROP_LIST_CODE,
106 MTP_OPERATION_GET_OBJECT_REFERENCES_CODE,
107 MTP_OPERATION_SET_OBJECT_REFERENCES_CODE,
108 };
109
110 const int EXTENSION_ID = 6;
GetDeviceInfoData(std::shared_ptr<MtpOperationContext> & context)111 GetDeviceInfoData::GetDeviceInfoData(std::shared_ptr<MtpOperationContext> &context)
112 : PayloadData(context), functionalMode_(0), serialNum_("0")
113 {
114 standardVersion_ = MTP_STANDARD_VERSION;
115 vendorExtensionID_ = EXTENSION_ID;
116 vendorExtensionVersion_ = MTP_STANDARD_VERSION;
117 vendorExtensionDesc_ = EXTENSION_DESC;
118 manufacturer_ = DEFAULT_PRODUCT_MANUFACTURER;
119 model_ = DEFAULT_PRODUCT_MODEL;
120 version_ = DEFAULT_PRODUCT_SOFTWARE_VERSION;
121 Maker(mOutBuffer);
122 }
123
GetDeviceInfoData()124 GetDeviceInfoData::GetDeviceInfoData()
125 {
126 standardVersion_ = MTP_STANDARD_VERSION;
127 vendorExtensionID_ = EXTENSION_ID;
128 vendorExtensionVersion_ = MTP_STANDARD_VERSION;
129 vendorExtensionDesc_ = EXTENSION_DESC;
130 functionalMode_ = 0;
131 manufacturer_ = DEFAULT_PRODUCT_MANUFACTURER;
132 model_ = DEFAULT_PRODUCT_MODEL;
133 version_ = DEFAULT_PRODUCT_SOFTWARE_VERSION;
134 serialNum_ = "0";
135 Maker(mOutBuffer);
136 }
137
~GetDeviceInfoData()138 GetDeviceInfoData::~GetDeviceInfoData()
139 {
140 }
141
Parser(const std::vector<uint8_t> & buffer,int32_t readSize)142 int GetDeviceInfoData::Parser(const std::vector<uint8_t> &buffer, int32_t readSize)
143 {
144 return MTP_SUCCESS;
145 }
146
Maker(std::vector<uint8_t> & outBuffer)147 int GetDeviceInfoData::Maker(std::vector<uint8_t> &outBuffer)
148 {
149 MtpPacketTool::PutUInt16(outBuffer, standardVersion_);
150 MtpPacketTool::PutUInt32(outBuffer, vendorExtensionID_);
151 MtpPacketTool::PutUInt16(outBuffer, vendorExtensionVersion_);
152 MtpPacketTool::PutString(outBuffer, vendorExtensionDesc_);
153 MtpPacketTool::PutUInt16(outBuffer, functionalMode_);
154 if (MtpManager::GetInstance().IsMtpMode()) {
155 MtpPacketTool::PutAUInt16(outBuffer, Operations, sizeof(Operations) / sizeof(uint16_t));
156 } else {
157 MtpPacketTool::PutAUInt16(outBuffer, OperationsPtp, sizeof(OperationsPtp) / sizeof(uint16_t));
158 }
159 MtpPacketTool::PutAUInt16(outBuffer, Events, sizeof(Events) / sizeof(uint16_t));
160 MtpPacketTool::PutAUInt16(outBuffer, DeviceProperties, sizeof(DeviceProperties) / sizeof(uint16_t));
161 MtpPacketTool::PutAUInt16(outBuffer, CaptureFormats, sizeof(CaptureFormats) / sizeof(uint16_t));
162 MtpPacketTool::PutAUInt16(outBuffer, PLAYBACK_FORMATS, sizeof(PLAYBACK_FORMATS) / sizeof(uint16_t));
163 MtpPacketTool::PutString(outBuffer, manufacturer_);
164 MtpPacketTool::PutString(outBuffer, model_);
165 MtpPacketTool::PutString(outBuffer, version_);
166 MtpPacketTool::PutString(outBuffer, serialNum_);
167
168 return MTP_SUCCESS;
169 }
170
CalculateSize()171 uint32_t GetDeviceInfoData::CalculateSize()
172 {
173 std::vector<uint8_t> tmpVar;
174
175 int res = Maker(tmpVar);
176 if (res != MTP_SUCCESS) {
177 return res;
178 }
179
180 return tmpVar.size();
181 }
182
SetManufacturer(const std::string & manufacturer)183 void GetDeviceInfoData::SetManufacturer(const std::string &manufacturer)
184 {
185 manufacturer_ = manufacturer;
186 }
187
SetModel(const std::string & model)188 void GetDeviceInfoData::SetModel(const std::string &model)
189 {
190 model_ = model;
191 }
192
SetVersion(const std::string & version)193 void GetDeviceInfoData::SetVersion(const std::string &version)
194 {
195 version_ = version;
196 }
197
SetSerialNum(const std::string & serialNum)198 void GetDeviceInfoData::SetSerialNum(const std::string &serialNum)
199 {
200 serialNum_ = serialNum;
201 }
202 } // namespace Media
203 } // namespace OHOS