summaryrefslogtreecommitdiffstats
path: root/netwerk/protocol/http/EarlyHintRegistrar.h
blob: 1accf27840b21d14be3e21136e76ed70fc1991c6 (plain)
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
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef mozilla_net_EarlyHintRegistrar_h__
#define mozilla_net_EarlyHintRegistrar_h__

#include "mozilla/RefCounted.h"
#include "nsRefPtrHashtable.h"
#include "mozilla/AlreadyAddRefed.h"

class nsIParentChannel;

namespace mozilla::net {

class EarlyHintPreloader;

/*
 * Registrar for pairing EarlyHintPreloader and HttpChannelParent via
 * earlyHintPreloaderId. EarlyHintPreloader has to be registered first.
 * EarlyHintPreloader::OnParentReady will be invoked to notify the
 * EarlyHintpreloader about the existence of the associated HttpChannelParent.
 */
class EarlyHintRegistrar final : public RefCounted<EarlyHintRegistrar> {
  using EarlyHintHashtable =
      nsRefPtrHashtable<nsUint64HashKey, EarlyHintPreloader>;

 public:
  MOZ_DECLARE_REFCOUNTED_TYPENAME(EarlyHintRegistrar)

  EarlyHintRegistrar();
  ~EarlyHintRegistrar();

  /*
   * Store EarlyHintPreloader for the HttpChannelParent to connect back to.
   *
   * @param aEarlyHintPreloaderId identifying the EarlyHintPreloader
   * @param aEhp the EarlyHintPreloader to store
   */
  void RegisterEarlyHint(uint64_t aEarlyHintPreloaderId,
                         EarlyHintPreloader* aEhp);

  /*
   * Link the provided nsIParentChannel with the associated EarlyHintPreloader.
   *
   * @param aEarlyHintPreloaderId identifying the EarlyHintPreloader
   * @param aParent the nsIParentChannel to be paired
   * @param aChannelId the id of the nsIChannel connecting to the
   *        EarlyHintPreloader.
   */
  bool LinkParentChannel(uint64_t aEarlyHintPreloaderId,
                         nsIParentChannel* aParent, uint64_t aChannelId);

  /*
   * Delete previous stored EarlyHintPreloader
   *
   * @param aEarlyHintPreloaderId identifying the EarlyHintPreloader
   */
  void DeleteEntry(uint64_t aEarlyHintPreloaderId);

  /*
   * This is called when "profile-change-net-teardown" is observed. We use this
   * to cancel the ongoing preload and clear the linked EarlyHintPreloader
   * objects.
   */
  static void CleanUp();

  // Singleton accessor
  static already_AddRefed<EarlyHintRegistrar> GetOrCreate();

 private:
  // Store unlinked EarlyHintPreloader objects.
  EarlyHintHashtable mEarlyHint;
};

}  // namespace mozilla::net

#endif  // mozilla_net_EarlyHintRegistrar_h__