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 #ifndef FRAMEWORKS_SERVICES_MEDIA_MTP_INCLUDE_MTP_OPERATION_UTILS_H_
16 #define FRAMEWORKS_SERVICES_MEDIA_MTP_INCLUDE_MTP_OPERATION_UTILS_H_
17 #include <memory>
18 #include <vector>
19 #include "mtp_operation_context.h"
20 #include "mtp_media_library.h"
21 #include "payload_data.h"
22 namespace OHOS {
23 namespace Media {
24 class MtpMedialibraryManager;
25 class MtpOperationUtils {
26 public:
27     explicit MtpOperationUtils(const std::shared_ptr<MtpOperationContext> &context);
28     ~MtpOperationUtils();
29 
30     uint16_t GetRespCommonData(std::shared_ptr<PayloadData> &data, int errorCode);
31     uint16_t GetDeviceInfo(std::shared_ptr<PayloadData> &data, uint16_t containerType, int &errorCode);
32     uint16_t GetNumObjects(std::shared_ptr<PayloadData> &data);
33     uint16_t GetObjectHandles(std::shared_ptr<PayloadData> &data, uint16_t containerType, int &errorCode);
34     uint16_t GetObjectInfo(std::shared_ptr<PayloadData> &data, uint16_t containerType, int &errorCode);
35     uint16_t GetObjectPropDesc(std::shared_ptr<PayloadData> &data, uint16_t containerType, int &errorCode);
36     uint16_t GetObjectPropValue(std::shared_ptr<PayloadData> &data, uint16_t containerType, int &errorCode);
37     void DoSetObjectPropValue(int &errorCode);
38     uint16_t GetObjectPropList(std::shared_ptr<PayloadData> &data, uint16_t containerType, int &errorCode);
39     uint16_t GetObjectReferences(std::shared_ptr<PayloadData> &data, uint16_t containerType, int &errorCode);
40     uint16_t SetObjectReferences(std::shared_ptr<PayloadData> &data);
41     uint16_t GetObjectDataDeal();
42     uint16_t GetObject(std::shared_ptr<PayloadData> &data, int errorCode);
43     int32_t DoRecevieSendObject();
44     uint16_t GetThumb(std::shared_ptr<PayloadData> &data, uint16_t containerType, int &errorCode);
45     uint16_t SendObjectInfo(std::shared_ptr<PayloadData> &data, int &errorCode);
46     uint16_t GetPartialObject(std::shared_ptr<PayloadData> &data);
47     uint16_t GetObjectPropsSupported(std::shared_ptr<PayloadData> &data);
48     uint16_t DeleteObject(std::shared_ptr<PayloadData> &data, int &errorCode);
49     uint16_t MoveObject(std::shared_ptr<PayloadData> &data, int &errorCode);
50     uint16_t CopyObject(std::shared_ptr<PayloadData> &data, int &errorCode);
51     uint16_t GetStorageIDs(std::shared_ptr<PayloadData> &data, uint16_t containerType, int &errorCode);
52     uint16_t GetStorageInfo(std::shared_ptr<PayloadData> &data, uint16_t containerType, int &errorCode);
53     uint16_t GetOpenSession(std::shared_ptr<PayloadData> &data, int errorCode);
54     uint16_t GetCloseSession(std::shared_ptr<PayloadData> &data);
55     uint16_t GetPropDesc(std::shared_ptr<PayloadData> &data, uint16_t containerType, int &errorCode);
56     uint16_t GetPropValue(std::shared_ptr<PayloadData> &data, uint16_t containerType, int &errorCode);
57     uint16_t SetDevicePropValueResp(std::shared_ptr<PayloadData> &data);
58     uint16_t ResetDevicePropResp(std::shared_ptr<PayloadData> &data);
59     uint16_t ObjectEvent(std::shared_ptr<PayloadData> &data, const int32_t payload);
60     uint16_t GetPathByHandle(const uint32_t &handle, std::string &path, std::string &realPath);
61     int32_t GetHandleByPaths(std::string path, uint32_t &handle);
62     bool TryAddExternalStorage(const std::string &fsUuid, uint32_t &storageId);
63     bool TryRemoveExternalStorage(const std::string &fsUuid, uint32_t &storageId);
64     static int32_t GetBatteryLevel();
65     static std::string GetPropertyInner(const std::string &property, const std::string &defValue);
66     static bool SetPropertyInner(const std::string &property, const std::string &value);
67 
68 private:
69     uint16_t CheckErrorCode(int errorCode);
70     void PreDealFd(const bool deal, const int fd);
71     void SendEventPacket(uint32_t objectHandle, uint16_t eventCode);
72     uint16_t HasStorage(int &errorCode);
73     int32_t RecevieSendObject(MtpFileRange &object, int fd);
74 
75     std::shared_ptr<MtpOperationContext> context_;
76     std::shared_ptr<MtpMedialibraryManager> mtpMedialibraryManager_;
77     std::shared_ptr<MtpMediaLibrary> mtpMediaLibrary_;
78 };
79 } // namespace Media
80 } // namespace OHOS
81 #endif  // FRAMEWORKS_SERVICES_MEDIA_MTP_INCLUDE_MTP_OPERATION_UTILS_H_
82