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 #undef LOG_TAG
17 #define LOG_TAG "AVSessionDynamicLoader"
18 
19 #include <dlfcn.h>
20 #include <openssl/crypto.h>
21 #include "avsession_log.h"
22 #include "avsession_errors.h"
23 #include "avsession_dynamic_loader.h"
24 #include "directory_ex.h"
25 
26 namespace OHOS {
27 namespace AVSession {
28 using namespace std;
29 
AVSessionDynamicLoader()30 AVSessionDynamicLoader::AVSessionDynamicLoader()
31 {
32     SLOGI("AVSessionDynamicLoader ctor");
33 }
34 
~AVSessionDynamicLoader()35 AVSessionDynamicLoader::~AVSessionDynamicLoader()
36 {
37     SLOGI("AVSessionDynamicLoader dtor");
38     for (auto iterator = dynamicLibHandle_.begin(); iterator != dynamicLibHandle_.end(); ++iterator) {
39 #ifndef TEST_COVERAGE
40         if (iterator->second != nullptr) {
41             OPENSSL_thread_stop();
42         }
43         dlclose(iterator->second);
44 #endif
45         SLOGI("close library avsession_dynamic success: %{public}s", iterator->first.c_str());
46     }
47 }
48 
OpenDynamicHandle(std::string dynamicLibrary)49 void* AVSessionDynamicLoader::OpenDynamicHandle(std::string dynamicLibrary)
50 {
51     std::lock_guard loaderLock(libLock_);
52     // if not opened, then open directly
53     // do we need lock?
54     // further optimization:
55     // 1. split all dependencies to separate libraries
56     // 2. just close each library not all
57     char realCachePath[PATH_MAX] = { 0X00 };
58     char *realPathRes = realpath(dynamicLibrary.c_str(), realCachePath);
59     if (realPathRes == nullptr || dynamicLibrary.find(".so") == std::string::npos) {
60         SLOGD("OpenDynamicHandle get dynamicLibrary %{public}s", dynamicLibrary.c_str());
61     }
62     if (dynamicLibHandle_[dynamicLibrary] == nullptr) {
63         char sourceLibraryRealPath[PATH_MAX] = { 0x00 };
64         if (realpath(dynamicLibrary.c_str(), sourceLibraryRealPath) == nullptr) {
65             SLOGE("check avsession_dynamic path failed %{public}s", dynamicLibrary.c_str());
66             return nullptr;
67         }
68 
69 #ifndef TEST_COVERAGE
70         void* dynamicLibHandle = dlopen(sourceLibraryRealPath, RTLD_NOW);
71         if (dynamicLibHandle == nullptr) {
72             SLOGE("Failed to open library avsession_dynamic, reason: %{public}sn", dlerror());
73             return nullptr;
74         }
75         SLOGI("open library %{public}s success", dynamicLibrary.c_str());
76         dynamicLibHandle_[dynamicLibrary] = dynamicLibHandle;
77 #else
78         SLOGI("in test coverage state, dlclose/dlopen may conflict with llvm");
79 #endif
80     }
81     return dynamicLibHandle_[dynamicLibrary];
82 }
83 
GetFuntion(std::string dynamicLibrary, std::string function)84 void* AVSessionDynamicLoader::GetFuntion(std::string dynamicLibrary, std::string function)
85 {
86     std::lock_guard loaderLock(libLock_);
87     // if not opened, then open directly
88     if (dynamicLibHandle_[dynamicLibrary] == nullptr) {
89         OpenDynamicHandle(dynamicLibrary);
90     }
91 
92     void* handle = nullptr;
93     if (dynamicLibHandle_[dynamicLibrary] != nullptr) {
94         handle = dlsym(dynamicLibHandle_[dynamicLibrary], function.c_str());
95         if (handle == nullptr) {
96             SLOGE("Failed to load %{public}s, reason: %{public}sn", function.c_str(), dlerror());
97             return nullptr;
98         }
99         SLOGI("GetFuntion %{public}s success", function.c_str());
100     }
101     return handle;
102 }
103 
CloseDynamicHandle(std::string dynamicLibrary)104 void AVSessionDynamicLoader::CloseDynamicHandle(std::string dynamicLibrary)
105 {
106     std::lock_guard loaderLock(libLock_);
107     // if already opened, then close all
108     if (dynamicLibHandle_[dynamicLibrary] != nullptr) {
109 #ifndef TEST_COVERAGE
110         OPENSSL_thread_stop();
111         dlclose(dynamicLibHandle_[dynamicLibrary]);
112 #endif
113         dynamicLibHandle_[dynamicLibrary] = nullptr;
114         SLOGI("close library avsession_dynamic success: %{public}s", dynamicLibrary.c_str());
115     }
116 }
117 
118 }  // namespace AVSession
119 }  // namespace OHOS