File: search_and_assistant_enabled_checker.cc

package info (click to toggle)
chromium 139.0.7258.127-1
  • links: PTS, VCS
  • area: main
  • in suites:
  • size: 6,122,068 kB
  • sloc: cpp: 35,100,771; ansic: 7,163,530; javascript: 4,103,002; python: 1,436,920; asm: 946,517; xml: 746,709; pascal: 187,653; perl: 88,691; sh: 88,436; objc: 79,953; sql: 51,488; cs: 44,583; fortran: 24,137; makefile: 22,147; tcl: 15,277; php: 13,980; yacc: 8,984; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (130 lines) | stat: -rw-r--r-- 4,973 bytes parent folder | download | duplicates (6)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/ui/ash/assistant/search_and_assistant_enabled_checker.h"

#include <string>
#include <utility>

#include "base/functional/bind.h"
#include "base/functional/callback_helpers.h"
#include "base/json/json_writer.h"
#include "base/strings/string_util.h"
#include "chromeos/ash/services/assistant/public/cpp/assistant_prefs.h"
#include "chromeos/services/assistant/public/shared/constants.h"
#include "net/base/url_util.h"
#include "services/data_decoder/public/cpp/data_decoder.h"
#include "services/network/public/cpp/resource_request.h"
#include "services/network/public/cpp/shared_url_loader_factory.h"
#include "services/network/public/cpp/simple_url_loader.h"
#include "services/network/public/mojom/url_response_head.mojom.h"
#include "url/gurl.h"

namespace {

// String to prepend to JSON responses to prevent XSSI. See http://go/xssi.
constexpr char kJsonSafetyPrefix[] = ")]}'\n";

constexpr int kMaxBodySize = 1024;

constexpr net::NetworkTrafficAnnotationTag
    kSearchAndAssistantEnabledCheckerNetworkTag =
        net::DefineNetworkTrafficAnnotation(
            "search_and_assistant_enabled_checker",
            R"(
        semantics {
          sender: "Search and Assistant Enabled Checker"
          description:
            "HTTP request used to check if Search and Assistant (SAA) "
            "service has been disabled by dasher admin. When service is "
            "disabled, SAA will provide offline experiences for user."
          trigger:
            "User logs in, switches active profile, enables SAA or "
            "updates account information so that assistant is allowed. "
            "Also triggered when dasher admin enables SAA service for "
            "user. "
          data: "User GAIA Credentials"
          destination: GOOGLE_OWNED_SERVICE
        }
        policy {
         cookies_allowed: YES
         cookies_store: "user"
         chrome_policy {
           AssistantVoiceMatchEnabledDuringOobe {
             AssistantVoiceMatchEnabledDuringOobe : false
           }
         }
         setting:
           "User can turn on/off SAA in ChromeOS Settings under the "
           "Search and Assistant menu. For managed devices, dasher "
           "admin can disable SAA from the Additional Services menu."
        })");

bool HasJsonSafetyPrefix(std::string& json_body) {
  return base::StartsWith(json_body, kJsonSafetyPrefix,
                          base::CompareCase::SENSITIVE);
}

}  // namespace

SearchAndAssistantEnabledChecker::SearchAndAssistantEnabledChecker(
    network::mojom::URLLoaderFactory* url_loader_factory,
    Delegate* delegate)
    : url_loader_factory_(url_loader_factory), delegate_(delegate) {}

SearchAndAssistantEnabledChecker::~SearchAndAssistantEnabledChecker() = default;

void SearchAndAssistantEnabledChecker::SyncSearchAndAssistantState() {
  auto resource_request = std::make_unique<network::ResourceRequest>();
  resource_request->url = net::AppendOrReplaceQueryParameter(
      GURL(chromeos::assistant::kServiceIdEndpoint),
      chromeos::assistant::kPayloadParamName,
      chromeos::assistant::kServiceIdRequestPayload);
  resource_request->site_for_cookies =
      net::SiteForCookies::FromUrl(resource_request->url);
  url_loader_ = network::SimpleURLLoader::Create(
      std::move(resource_request), kSearchAndAssistantEnabledCheckerNetworkTag);
  url_loader_->DownloadToString(
      url_loader_factory_,
      base::BindOnce(
          &SearchAndAssistantEnabledChecker::OnSimpleURLLoaderComplete,
          weak_factory_.GetWeakPtr()),
      kMaxBodySize);
}

void SearchAndAssistantEnabledChecker::OnSimpleURLLoaderComplete(
    std::unique_ptr<std::string> response_body) {
  if (!response_body || url_loader_->NetError() != net::OK ||
      !url_loader_->ResponseInfo() || !url_loader_->ResponseInfo()->headers) {
    LOG(ERROR) << "Network error. Failed to get response.";
    delegate_->OnError();
    return;
  }

  if (!HasJsonSafetyPrefix(*response_body)) {
    LOG(ERROR) << "Invalid response.";
    delegate_->OnError();
    return;
  }

  // Strip the JsonSafetyPrefix and parse the response.
  data_decoder::DataDecoder::ParseJsonIsolated(
      response_body->substr(strlen(kJsonSafetyPrefix)),
      base::BindOnce(&SearchAndAssistantEnabledChecker::OnJsonParsed,
                     weak_factory_.GetWeakPtr()));
}

void SearchAndAssistantEnabledChecker::OnJsonParsed(
    data_decoder::DataDecoder::ValueOrError response) {
  if (!response.has_value() || !response->is_dict()) {
    LOG(ERROR) << "JSON parsing failed: " << response.error();
    delegate_->OnError();
    return;
  }

  // |result| is true if the Search and Assistant bit is disabled.
  auto is_disabled = response->GetDict().FindBool("result");

  delegate_->OnSearchAndAssistantStateReceived(is_disabled.value());
}