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


content / browser / worker_host / dedicated_worker_service_impl.h [blame]

// 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.

#ifndef CONTENT_BROWSER_WORKER_HOST_DEDICATED_WORKER_SERVICE_IMPL_H_
#define CONTENT_BROWSER_WORKER_HOST_DEDICATED_WORKER_SERVICE_IMPL_H_

#include "base/containers/flat_map.h"
#include "base/memory/raw_ptr.h"
#include "base/observer_list.h"
#include "content/common/content_export.h"
#include "content/public/browser/dedicated_worker_service.h"
#include "url/gurl.h"

namespace content {

class DedicatedWorkerHost;

class CONTENT_EXPORT DedicatedWorkerServiceImpl
    : public DedicatedWorkerService {
 public:
  DedicatedWorkerServiceImpl();
  ~DedicatedWorkerServiceImpl() override;

  DedicatedWorkerServiceImpl(const DedicatedWorkerServiceImpl& other) = delete;

  // DedicatedWorkerService:
  void AddObserver(Observer* observer) override;
  void RemoveObserver(Observer* observer) override;
  void EnumerateDedicatedWorkers(Observer* observer) override;

  // Notifies all observers about a new worker.
  void NotifyWorkerCreated(DedicatedWorkerHost* host);

  // Notifies all observers about a worker being destroyed.
  void NotifyBeforeWorkerDestroyed(
      const blink::DedicatedWorkerToken& worker_token,
      DedicatedWorkerCreator creator);

  // Notifies all observers that a worker's final response URL was determined.
  void NotifyWorkerFinalResponseURLDetermined(
      const blink::DedicatedWorkerToken& worker_token,
      const GURL& url);

  // Returns true if a worker with the given token has already been registered
  // with the service. This allows for malformed messages with duplicated
  // tokens to be detected, and the offending renderer to be shutdown.
  bool HasToken(const blink::DedicatedWorkerToken& worker_token) const;

  // Returns the DedicatedWorkerHost associated with this token. Clients should
  // not hold on to the pointer, as it may become invalid when the worker exits.
  DedicatedWorkerHost* GetDedicatedWorkerHostFromToken(
      const blink::DedicatedWorkerToken& worker_token) const;

 private:
  base::ObserverList<Observer> observers_;
  base::flat_map<blink::DedicatedWorkerToken,
                 raw_ptr<DedicatedWorkerHost, CtnExperimental>>
      dedicated_worker_hosts_;
};

}  // namespace content

#endif  // CONTENT_BROWSER_WORKER_HOST_DEDICATED_WORKER_SERVICE_IMPL_H_