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 #include <utility>
16 #include "verify_http_task.h"
17 #include "app_domain_verify_hilog.h"
18
19 namespace OHOS {
20 namespace AppDomainVerify {
21 const static size_t MAX_RESPONSE_LEN = 20 * 1024; // 20K
22
VerifyHttpTask(std::string uri,std::shared_ptr<IVerifyTask> verifyTask)23 VerifyHttpTask::VerifyHttpTask(std::string uri, std::shared_ptr<IVerifyTask> verifyTask)
24 : uri_(std::move(uri)), verifyTask_(verifyTask)
25 {
26 APP_DOMAIN_VERIFY_HILOGD(APP_DOMAIN_VERIFY_AGENT_MODULE_SERVICE, "VerifyHttpTask.");
27 }
CreateHttpClientTask()28 std::shared_ptr<OHOS::NetStack::HttpClient::HttpClientTask> VerifyHttpTask::CreateHttpClientTask()
29 {
30 APP_DOMAIN_VERIFY_HILOGD(APP_DOMAIN_VERIFY_AGENT_MODULE_SERVICE, "CreateHttpClientTask.");
31 OHOS::NetStack::HttpClient::HttpClientRequest httpClientRequest;
32 if (verifyTask_) {
33 if (!verifyTask_->OnPreRequest(httpClientRequest, uri_)) {
34 APP_DOMAIN_VERIFY_HILOGE(APP_DOMAIN_VERIFY_AGENT_MODULE_SERVICE, "OnPreRequest failed.");
35 return nullptr;
36 }
37 auto httpClientTask = std::make_shared<OHOS::NetStack::HttpClient::HttpClientTask>(httpClientRequest);
38 return httpClientTask;
39 }
40 APP_DOMAIN_VERIFY_HILOGE(APP_DOMAIN_VERIFY_AGENT_MODULE_SERVICE, "CreateHttpClientTask failed.");
41 return nullptr;
42 }
OnSuccess(const HttpClientRequest & request,const HttpClientResponse & response)43 void VerifyHttpTask::OnSuccess(const HttpClientRequest& request, const HttpClientResponse& response)
44 {
45 APP_DOMAIN_VERIFY_HILOGI(APP_DOMAIN_VERIFY_AGENT_MODULE_SERVICE, "OnSuccess.");
46 if (verifyTask_) {
47 verifyTask_->OnPostVerify(uri_, response);
48 }
49 }
OnFail(const HttpClientRequest & request,const HttpClientResponse & response,const HttpClientError & error)50 void VerifyHttpTask::OnFail(
51 const HttpClientRequest& request, const HttpClientResponse& response, const HttpClientError& error)
52 {
53 APP_DOMAIN_VERIFY_HILOGE(APP_DOMAIN_VERIFY_AGENT_MODULE_SERVICE, "OnFail.");
54 if (verifyTask_) {
55 verifyTask_->OnPostVerify(uri_, response);
56 }
57 }
OnCancel(const HttpClientRequest & request,const HttpClientResponse & response)58 void VerifyHttpTask::OnCancel(const HttpClientRequest& request, const HttpClientResponse& response)
59 {
60 APP_DOMAIN_VERIFY_HILOGW(APP_DOMAIN_VERIFY_AGENT_MODULE_SERVICE, "OnCance.");
61 if (verifyTask_) {
62 verifyTask_->OnPostVerify(uri_, response);
63 }
64 }
OnDataReceive(std::shared_ptr<OHOS::NetStack::HttpClient::HttpClientTask> task,const HttpClientRequest & request,const uint8_t * data,size_t length)65 void VerifyHttpTask::OnDataReceive(std::shared_ptr<OHOS::NetStack::HttpClient::HttpClientTask> task,
66 const HttpClientRequest& request, const uint8_t* data, size_t length)
67 {
68 APP_DOMAIN_VERIFY_HILOGI(APP_DOMAIN_VERIFY_AGENT_MODULE_SERVICE, "OnDataReceive size:%{public}zu.", length);
69 if (length_ + length > MAX_RESPONSE_LEN) {
70 APP_DOMAIN_VERIFY_HILOGW(APP_DOMAIN_VERIFY_AGENT_MODULE_SERVICE, "recieve data exceeds, will cancel.");
71 task->Cancel();
72 } else {
73 length_ += length;
74 }
75 }
~VerifyHttpTask()76 VerifyHttpTask::~VerifyHttpTask()
77 {
78 APP_DOMAIN_VERIFY_HILOGD(APP_DOMAIN_VERIFY_AGENT_MODULE_SERVICE, "~VerifyHttpTask.");
79 }
80 }
81 }