1 /*
2  * Copyright (C) 2024 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 #define MLOG_TAG "EnhancementDynamicLoader"
17 
18 #include "dynamic_loader.h"
19 
20 #include <dlfcn.h>
21 #include "media_log.h"
22 #include "medialibrary_tracer.h"
23 
24 namespace OHOS {
25 namespace Media {
26 using namespace std;
27 static const char *K_LIBRARY_SUFFIX = ".so";
DynamicLoader()28 DynamicLoader::DynamicLoader()
29 {
30     MEDIA_INFO_LOG("EnhancementDynamicLoader ctor");
31 }
32 
~DynamicLoader()33 DynamicLoader::~DynamicLoader()
34 {
35     MEDIA_INFO_LOG("EnhancementDynamicLoader dtor");
36     for (auto iterator = dynamicLibHandle_.begin(); iterator != dynamicLibHandle_.end(); ++iterator) {
37         dlclose(iterator->second);
38         MEDIA_INFO_LOG("close library media_cloud_enhancement_dynamic success: %{public}s", iterator->first.c_str());
39     }
40 }
41 
OpenDynamicHandle(std::string dynamicLibrary)42 void* DynamicLoader::OpenDynamicHandle(std::string dynamicLibrary)
43 {
44     MediaLibraryTracer tracer;
45     tracer.Start("EnhancementDynamicLoader::OpenDynamicHandle");
46     std::lock_guard loaderLock(libLock_);
47     if (!EndsWith(dynamicLibrary, K_LIBRARY_SUFFIX)) {
48         MEDIA_ERR_LOG("CloseDynamicHandle with error name!");
49         return nullptr;
50     }
51     if (dynamicLibHandle_[dynamicLibrary] == nullptr) {
52         void* dynamicLibHandle = dlopen(dynamicLibrary.c_str(), RTLD_NOW);
53         if (dynamicLibHandle == nullptr) {
54             MEDIA_ERR_LOG("Failed to open %{public}s, reason: %{public}sn", dynamicLibrary.c_str(), dlerror());
55             return nullptr;
56         }
57         MEDIA_INFO_LOG("open library %{public}s success", dynamicLibrary.c_str());
58         dynamicLibHandle_[dynamicLibrary] = dynamicLibHandle;
59     }
60     return dynamicLibHandle_[dynamicLibrary];
61 }
62 
GetFunction(const string dynamicLibrary,const string function)63 void* DynamicLoader::GetFunction(const string dynamicLibrary, const string function)
64 {
65     MediaLibraryTracer tracer;
66     tracer.Start("EnhancementDynamicLoader::GetFunction");
67     std::lock_guard loaderLock(libLock_);
68     // if not opened, then open directly
69     if (dynamicLibHandle_[dynamicLibrary] == nullptr) {
70         OpenDynamicHandle(dynamicLibrary);
71     }
72 
73     void* handle = nullptr;
74     if (dynamicLibHandle_[dynamicLibrary] != nullptr) {
75         handle = dlsym(dynamicLibHandle_[dynamicLibrary], function.c_str());
76         if (handle == nullptr) {
77             MEDIA_ERR_LOG("Failed to load %{public}s, reason: %{public}sn", function.c_str(), dlerror());
78             return nullptr;
79         }
80         MEDIA_INFO_LOG("GetFunction %{public}s success", function.c_str());
81     }
82     return handle;
83 }
84 
CloseDynamicHandle(std::string dynamicLibrary)85 void DynamicLoader::CloseDynamicHandle(std::string dynamicLibrary)
86 {
87     MediaLibraryTracer tracer;
88     tracer.Start("EnhancementDynamicLoader::CloseDynamicHandle");
89     std::lock_guard loaderLock(libLock_);
90     if (!EndsWith(dynamicLibrary, K_LIBRARY_SUFFIX)) {
91         MEDIA_ERR_LOG("CloseDynamicHandle with error name!");
92         return;
93     }
94     if (dynamicLibHandle_[dynamicLibrary] != nullptr) {
95         dlclose(dynamicLibHandle_[dynamicLibrary]);
96         dynamicLibHandle_[dynamicLibrary] = nullptr;
97         MEDIA_INFO_LOG("close library media_cloud_enhancement_dynamic success: %{public}s",
98             dynamicLibrary.c_str());
99     }
100 }
101 }  // namespace Camera
102 }  // namespace OHOS