File: search_and_assistant_enabled_checker.cc

package info (click to toggle)
chromium 120.0.6099.224-1~deb11u1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 6,112,112 kB
  • sloc: cpp: 32,907,025; ansic: 8,148,123; javascript: 3,679,536; python: 2,031,248; asm: 959,718; java: 804,675; xml: 617,256; sh: 111,417; objc: 100,835; perl: 88,443; cs: 53,032; makefile: 29,579; fortran: 24,137; php: 21,162; tcl: 21,147; sql: 20,809; ruby: 17,735; pascal: 12,864; yacc: 8,045; lisp: 3,388; lex: 1,323; ada: 727; awk: 329; jsp: 267; csh: 117; exp: 43; sed: 37
file content (129 lines) | stat: -rw-r--r-- 4,885 bytes parent folder | download
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
// 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() {}

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);
  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());
}