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_listener_stub.h"
17 #include "media_log.h"
18 #include "media_errors.h"
19 
20 namespace {
21 constexpr OHOS::HiviewDFX::HiLogLabel LABEL = {LOG_CORE, LOG_DOMAIN_SCREENCAPTURE, "ScreenCaptureMonitorListenerStub"};
22 }
23 
24 namespace OHOS {
25 namespace Media {
ScreenCaptureMonitorListenerStub()26 ScreenCaptureMonitorListenerStub::ScreenCaptureMonitorListenerStub()
27 {
28     MEDIA_LOGD("0x%{public}06" PRIXPTR " Instances create", FAKE_POINTER(this));
29 }
30 
~ScreenCaptureMonitorListenerStub()31 ScreenCaptureMonitorListenerStub::~ScreenCaptureMonitorListenerStub()
32 {
33     MEDIA_LOGD("0x%{public}06" PRIXPTR " Instances destroy", FAKE_POINTER(this));
34 }
35 
OnRemoteRequest(uint32_t code,MessageParcel & data,MessageParcel & reply,MessageOption & option)36 int ScreenCaptureMonitorListenerStub::OnRemoteRequest(uint32_t code, MessageParcel &data, MessageParcel &reply,
37     MessageOption &option)
38 {
39     auto remoteDescriptor = data.ReadInterfaceToken();
40     if (ScreenCaptureMonitorListenerStub::GetDescriptor() != remoteDescriptor) {
41         MEDIA_LOGE("Invalid descriptor");
42         return MSERR_INVALID_OPERATION;
43     }
44     switch (code) {
45         case ScreenCaptureMonitorListenerMsg::ON_SCREEN_CAPTURE_STARTED: {
46             int pid = data.ReadInt32();
47             OnScreenCaptureStarted(pid);
48             return MSERR_OK;
49         }
50         case ScreenCaptureMonitorListenerMsg::ON_SCREEN_CAPTURE_FINISHED: {
51             int pid = data.ReadInt32();
52             OnScreenCaptureFinished(pid);
53             return MSERR_OK;
54         }
55         default: {
56             MEDIA_LOGE("default case, need check ScreenCaptureMonitorListenerStub");
57             return IPCObjectStub::OnRemoteRequest(code, data, reply, option);
58         }
59     }
60 }
61 
OnScreenCaptureStarted(int32_t pid)62 void ScreenCaptureMonitorListenerStub::OnScreenCaptureStarted(int32_t pid)
63 {
64     MEDIA_LOGI("ScreenCaptureMonitorListenerStub:0x%{public}06" PRIXPTR " OnScreenCaptureStarted",
65         FAKE_POINTER(this));
66     for (const auto& value : screenCaptureMonitorCallbacks_) {
67         if (value != nullptr) {
68             value->OnScreenCaptureStarted(pid);
69         }
70     }
71 }
72 
OnScreenCaptureFinished(int32_t pid)73 void ScreenCaptureMonitorListenerStub::OnScreenCaptureFinished(int32_t pid)
74 {
75     MEDIA_LOGI("ScreenCaptureMonitorListenerStub:0x%{public}06" PRIXPTR " OnScreenCaptureFinished",
76         FAKE_POINTER(this));
77     for (const auto& value : screenCaptureMonitorCallbacks_) {
78         if (value != nullptr) {
79             value->OnScreenCaptureFinished(pid);
80         }
81     }
82 }
83 
RegisterScreenCaptureMonitorListener(sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> listener)84 void ScreenCaptureMonitorListenerStub::RegisterScreenCaptureMonitorListener(
85     sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> listener)
86 {
87     MEDIA_LOGI("ScreenCaptureMonitorListenerStub:0x%{public}06" PRIXPTR " RegisterScreenCaptureMonitorListener",
88         FAKE_POINTER(this));
89     screenCaptureMonitorCallbacks_.insert(listener);
90 }
UnregisterScreenCaptureMonitorListener(sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> listener)91 void ScreenCaptureMonitorListenerStub::UnregisterScreenCaptureMonitorListener(
92     sptr<ScreenCaptureMonitor::ScreenCaptureMonitorListener> listener)
93 {
94     MEDIA_LOGI("ScreenCaptureMonitorListenerStub:0x%{public}06" PRIXPTR " UnregisterScreenCaptureMonitorListener",
95         FAKE_POINTER(this));
96     auto it = screenCaptureMonitorCallbacks_.find(listener);
97     if (it != screenCaptureMonitorCallbacks_.end()) {
98         screenCaptureMonitorCallbacks_.erase(it);
99     }
100 }
101 } // namespace Media
102 } // namespace OHOS