From eef09ed26f470fd5d528a803425adb3995c64bcc Mon Sep 17 00:00:00 2001
From: fangJinliang1 <fangjinliang1@huawei.com>
Date: Mon, 21 Mar 2022 22:41:54 +0800
Subject: [PATCH] fixed 51f2c2f from
 https://gitee.com/fangJinliang1/notification_ans_standard/pulls/360 code
 safety modify

Signed-off-by: fangJinliang1 <fangjinliang1@huawei.com>
Change-Id: I435a84a129b5d3e83fbd80ac6ea80dffba986f56
---
 frameworks/ans/core/src/ans_subscriber_stub.cpp | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/frameworks/ans/core/src/ans_subscriber_stub.cpp b/frameworks/ans/core/src/ans_subscriber_stub.cpp
index 7340a0c0c..745708d3c 100644
--- a/frameworks/ans/core/src/ans_subscriber_stub.cpp
+++ b/frameworks/ans/core/src/ans_subscriber_stub.cpp
@@ -188,6 +188,10 @@ ErrCode AnsSubscriberStub::HandleOnUpdated(MessageParcel &data, MessageParcel &r
 ErrCode AnsSubscriberStub::HandleOnDoNotDisturbDateChange(MessageParcel &data, MessageParcel &reply)
 {
     sptr<NotificationDoNotDisturbDate> date = data.ReadParcelable<NotificationDoNotDisturbDate>();
+    if (!date) {
+        ANS_LOGW("[HandleOnDoNotDisturbDateChange] fail: date ReadParcelable failed");
+        return ERR_ANS_PARCELABLE_FAILED;
+    }
     OnDoNotDisturbDateChange(date);
     return ERR_OK;
 }
@@ -195,6 +199,10 @@ ErrCode AnsSubscriberStub::HandleOnDoNotDisturbDateChange(MessageParcel &data, M
 ErrCode AnsSubscriberStub::HandleOnEnabledNotificationChanged(MessageParcel &data, MessageParcel &reply)
 {
     sptr<EnabledNotificationCallbackData> callbackData = data.ReadParcelable<EnabledNotificationCallbackData>();
+    if (!callbackData) {
+        ANS_LOGW("[HandleOnEnabledNotificationChanged] fail: callbackData ReadParcelable failed");
+        return ERR_ANS_PARCELABLE_FAILED;
+    }
     OnEnabledNotificationChanged(callbackData);
     return ERR_OK;
 }
-- 
Gitee