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 std::list<int32_t> ScreenCaptureMonitorServer::IsScreenCaptureWorking()
70 {
71 MEDIA_LOGI("ScreenCaptureMonitorServer:0x%{public}06" PRIXPTR " IsScreenCaptureWorking S", FAKE_POINTER(this));
72 std::list<int32_t> pidList{};
73 OHOS::Media::ScreenCaptureServer::GetRunningScreenCaptureInstancePid(pidList);
74 for (auto pid: pidList) {
75 MEDIA_LOGD("ScreenCaptureMonitorServer::IsScreenCaptureWorking pid %{public}d", pid);
76 }
77 return pidList;
78 }
79
SetScreenCaptureMonitorCallback(sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> callback)80 void ScreenCaptureMonitorServer::SetScreenCaptureMonitorCallback(
81 sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> callback)
82 {
83 MediaTrace trace("ScreenCaptureMonitorServer::SetScreenCaptureCallback");
84 std::lock_guard<std::mutex> lock(mutex_);
85 CHECK_AND_RETURN_LOG(callback != nullptr, "SetScreenCaptureCallback failed, callback is nullptr");
86 screenCaptureMonitorCbSet_.insert(callback);
87 MEDIA_LOGI("ScreenCaptureMonitorServer: 0x%{public}06" PRIXPTR "SetScreenCaptureCallback OK.", FAKE_POINTER(this));
88 }
89
RemoveScreenCaptureMonitorCallback(sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> callback)90 void ScreenCaptureMonitorServer::RemoveScreenCaptureMonitorCallback(
91 sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> callback)
92 {
93 MediaTrace trace("ScreenCaptureMonitorServer::RemoveScreenCaptureMonitorCallback");
94 std::lock_guard<std::mutex> lock(mutex_);
95 CHECK_AND_RETURN_LOG(callback != nullptr, "RemoveScreenCaptureMonitorCallback failed, callback is nullptr");
96 screenCaptureMonitorCbSet_.erase(callback);
97 MEDIA_LOGI("ScreenCaptureMonitorServer: 0x%{public}06" PRIXPTR "RemoveScreenCaptureMonitorCallback OK.",
98 FAKE_POINTER(this));
99 }
100
RegisterScreenCaptureMonitorListener(sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> callback)101 void ScreenCaptureMonitorServer::RegisterScreenCaptureMonitorListener(
102 sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> callback)
103 {
104 MEDIA_LOGI("ScreenCaptureMonitorServer:0x%{public}06" PRIXPTR " RegisterScreenCaptureMonitorListener",
105 FAKE_POINTER(this));
106 }
107
UnregisterScreenCaptureMonitorListener(sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> listener)108 void ScreenCaptureMonitorServer::UnregisterScreenCaptureMonitorListener(
109 sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> listener)
110 {
111 MEDIA_LOGI("ScreenCaptureMonitorServer:0x%{public}06" PRIXPTR " UnregisterScreenCaptureMonitorListener",
112 FAKE_POINTER(this));
113 }
114
CallOnScreenCaptureStarted(int32_t pid)115 int32_t ScreenCaptureMonitorServer::CallOnScreenCaptureStarted(int32_t pid)
116 {
117 std::lock_guard<std::mutex> lockCb(mutexCb_);
118 MEDIA_LOGI("ScreenCaptureMonitorServer::CallOnScreenCaptureStarted S");
119 for (const auto& value : screenCaptureMonitorCbSet_) {
120 if (value != nullptr) {
121 value->OnScreenCaptureStarted(pid);
122 }
123 }
124 return MSERR_OK;
125 }
126
CallOnScreenCaptureFinished(int32_t pid)127 int32_t ScreenCaptureMonitorServer::CallOnScreenCaptureFinished(int32_t pid)
128 {
129 std::lock_guard<std::mutex> lockCb(mutexCb_);
130 MEDIA_LOGI("ScreenCaptureMonitorServer::CallOnScreenCaptureFinished S");
131 for (const auto& value : screenCaptureMonitorCbSet_) {
132 if (value != nullptr) {
133 value->OnScreenCaptureFinished(pid);
134 }
135 }
136 return MSERR_OK;
137 }
138
SetSystemScreenRecorderStatus(bool started)139 void ScreenCaptureMonitorServer::SetSystemScreenRecorderStatus(bool started)
140 {
141 MEDIA_LOGI("ScreenCaptureMonitorServer::SetSystemScreenRecorderStatus S, state: %{public}d", started);
142 isSystemScreenRecorderWorking_ = started;
143 }
144
IsSystemScreenRecorder(int32_t pid)145 bool ScreenCaptureMonitorServer::IsSystemScreenRecorder(int32_t pid)
146 {
147 MEDIA_LOGI("ScreenCaptureMonitorServer::IsSystemScreenRecorder S");
148 if (pid < 0) {
149 MEDIA_LOGW("ScreenCaptureMonitorServer::IsSystemScreenRecorder invalid pid: %{public}d", pid);
150 return false;
151 }
152 bool result = ScreenCaptureServer::CheckPidIsScreenRecorder(pid);
153 MEDIA_LOGI("ScreenCaptureMonitorServer::IsSystemScreenRecorder result: %{public}d", result);
154 return result;
155 }
156
IsSystemScreenRecorderWorking()157 bool ScreenCaptureMonitorServer::IsSystemScreenRecorderWorking()
158 {
159 MEDIA_LOGI("ScreenCaptureMonitorServer::IsSystemScreenRecorderWorking S");
160 return isSystemScreenRecorderWorking_;
161 }
162 } // namespace Media
163 } // namespace OHOS