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


content / browser / preloading / prefetch / prefetch_cookie_listener.cc [blame]

// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "content/browser/preloading/prefetch/prefetch_cookie_listener.h"

#include <memory>

#include "base/check.h"
#include "mojo/public/cpp/bindings/receiver.h"
#include "services/network/public/mojom/cookie_manager.mojom.h"
#include "url/gurl.h"

namespace content {

std::unique_ptr<PrefetchCookieListener> PrefetchCookieListener::MakeAndRegister(
    const GURL& url,
    network::mojom::CookieManager* cookie_manager) {
  DCHECK(cookie_manager);

  std::unique_ptr<PrefetchCookieListener> cookie_listener =
      std::make_unique<PrefetchCookieListener>(url);

  // |cookie_listener| will get updates whenever host cookies for |url| or
  // domain cookies that match |url| are changed.
  cookie_manager->AddCookieChangeListener(
      url, std::nullopt,
      cookie_listener->cookie_listener_receiver_.BindNewPipeAndPassRemote());

  return cookie_listener;
}

PrefetchCookieListener::PrefetchCookieListener(const GURL& url) : url_(url) {}

PrefetchCookieListener::~PrefetchCookieListener() = default;

void PrefetchCookieListener::StopListening() {
  cookie_listener_receiver_.reset();
}

void PrefetchCookieListener::OnCookieChange(
    const net::CookieChangeInfo& change) {
  DCHECK(url_.DomainIs(change.cookie.DomainWithoutDot()));
  have_cookies_changed_ = true;

  // Once we record one change to the cookies associated with |url_|, we don't
  // care about any subsequent changes.
  StopListening();
}

}  // namespace content