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 #include "screen_capture_monitor_server.h"
17 #include "map"
18 #include "media_log.h"
19 #include "media_errors.h"
20 #include "accesstoken_kit.h"
21 #include "ipc_skeleton.h"
22 #include "media_dfx.h"
23 #include "hitrace/tracechain.h"
24 #include "media_utils.h"
25 #include "screen_capture_server.h"
26
27 namespace {
28 constexpr OHOS::HiviewDFX::HiLogLabel LABEL = {LOG_CORE, LOG_DOMAIN_SCREENCAPTURE, "ScreenCaptureMonitorServer"};
29 }
30
31 namespace OHOS {
32 namespace Media {
33 std::shared_ptr<ScreenCaptureMonitorServer> screenCaptureMonitorServer = nullptr;
GetInstance()34 std::shared_ptr<ScreenCaptureMonitorServer> ScreenCaptureMonitorServer::GetInstance()
35 {
36 if (!screenCaptureMonitorServer) {
37 screenCaptureMonitorServer = std::make_shared<ScreenCaptureMonitorServer>();
38 int32_t ret = screenCaptureMonitorServer->Init();
39 CHECK_AND_RETURN_RET_LOG(ret == MSERR_OK, nullptr, "failed to init ScreenCaptureMonitorServer");
40 }
41 return screenCaptureMonitorServer;
42 }
43
ScreenCaptureMonitorServer()44 ScreenCaptureMonitorServer::ScreenCaptureMonitorServer()
45 {
46 MEDIA_LOGD("0x%{public}06" PRIXPTR " Instances create", FAKE_POINTER(this));
47 }
48
~ScreenCaptureMonitorServer()49 ScreenCaptureMonitorServer::~ScreenCaptureMonitorServer()
50 {
51 MEDIA_LOGD("0x%{public}06" PRIXPTR " Instances destroy", FAKE_POINTER(this));
52 Release();
53 }
54
Init()55 int32_t ScreenCaptureMonitorServer::Init()
56 {
57 MediaTrace trace("ScreenCaptureMonitorServer::Init");
58 return MSERR_OK;
59 }
60
Release()61 int32_t ScreenCaptureMonitorServer::Release()
62 {
63 MEDIA_LOGI("ScreenCaptureMonitorServer:0x%{public}06" PRIXPTR " Release S", FAKE_POINTER(this));
64 std::lock_guard<std::mutex> lockCb(mutexCb_);
65 screenCaptureMonitorCbSet_.clear();
66 return MSERR_OK;
67 }
68
IsScreenCaptureWorking()69 int32_t ScreenCaptureMonitorServer::IsScreenCaptureWorking()
70 {
71 MEDIA_LOGI("ScreenCaptureMonitorServer:0x%{public}06" PRIXPTR " IsScreenCaptureWorking S", FAKE_POINTER(this));
72 int32_t pid = -1;
73 OHOS::Media::ScreenCaptureServer::GetRunningScreenCaptureInstancePid(pid);
74 MEDIA_LOGI("IsScreenCaptureWorking pid %{public}d E", pid);
75 return pid;
76 }
77
SetScreenCaptureMonitorCallback(sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> callback)78 void ScreenCaptureMonitorServer::SetScreenCaptureMonitorCallback(
79 sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> callback)
80 {
81 MediaTrace trace("ScreenCaptureMonitorServer::SetScreenCaptureCallback");
82 std::lock_guard<std::mutex> lock(mutex_);
83 CHECK_AND_RETURN_LOG(callback != nullptr, "SetScreenCaptureCallback failed, callback is nullptr");
84 screenCaptureMonitorCbSet_.insert(callback);
85 MEDIA_LOGI("ScreenCaptureMonitorServer: 0x%{public}06" PRIXPTR "SetScreenCaptureCallback OK.", FAKE_POINTER(this));
86 }
87
RemoveScreenCaptureMonitorCallback(sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> callback)88 void ScreenCaptureMonitorServer::RemoveScreenCaptureMonitorCallback(
89 sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> callback)
90 {
91 MediaTrace trace("ScreenCaptureMonitorServer::RemoveScreenCaptureMonitorCallback");
92 std::lock_guard<std::mutex> lock(mutex_);
93 CHECK_AND_RETURN_LOG(callback != nullptr, "RemoveScreenCaptureMonitorCallback failed, callback is nullptr");
94 screenCaptureMonitorCbSet_.erase(callback);
95 MEDIA_LOGI("ScreenCaptureMonitorServer: 0x%{public}06" PRIXPTR "RemoveScreenCaptureMonitorCallback OK.",
96 FAKE_POINTER(this));
97 }
98
RegisterScreenCaptureMonitorListener(sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> callback)99 void ScreenCaptureMonitorServer::RegisterScreenCaptureMonitorListener(
100 sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> callback)
101 {
102 MEDIA_LOGI("ScreenCaptureMonitorServer:0x%{public}06" PRIXPTR " RegisterScreenCaptureMonitorListener",
103 FAKE_POINTER(this));
104 }
105
UnregisterScreenCaptureMonitorListener(sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> listener)106 void ScreenCaptureMonitorServer::UnregisterScreenCaptureMonitorListener(
107 sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> listener)
108 {
109 MEDIA_LOGI("ScreenCaptureMonitorServer:0x%{public}06" PRIXPTR " UnregisterScreenCaptureMonitorListener",
110 FAKE_POINTER(this));
111 }
112
CallOnScreenCaptureStarted(int32_t pid)113 int32_t ScreenCaptureMonitorServer::CallOnScreenCaptureStarted(int32_t pid)
114 {
115 std::lock_guard<std::mutex> lockCb(mutexCb_);
116 MEDIA_LOGI("ScreenCaptureMonitorServer::CallOnScreenCaptureStarted S");
117 for (const auto& value : screenCaptureMonitorCbSet_) {
118 if (value != nullptr) {
119 value->OnScreenCaptureStarted(pid);
120 }
121 }
122 return MSERR_OK;
123 }
124
CallOnScreenCaptureFinished(int32_t pid)125 int32_t ScreenCaptureMonitorServer::CallOnScreenCaptureFinished(int32_t pid)
126 {
127 std::lock_guard<std::mutex> lockCb(mutexCb_);
128 MEDIA_LOGI("ScreenCaptureMonitorServer::CallOnScreenCaptureFinished S");
129 for (const auto& value : screenCaptureMonitorCbSet_) {
130 if (value != nullptr) {
131 value->OnScreenCaptureFinished(pid);
132 }
133 }
134 return MSERR_OK;
135 }
136 } // namespace Media
137 } // namespace OHOS