summaryrefslogtreecommitdiffstats
path: root/toolkit/components/reputationservice/LoginReputation.h
blob: a95a9d1d8a8c8928984f592554e98221c1eb076e (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
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* 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 LoginReputation_h__
#define LoginReputation_h__

#include "mozilla/Logging.h"
#include "mozilla/MozPromise.h"
#include "nsILoginReputation.h"
#include "nsIObserver.h"
#include "nsISupportsImpl.h"

class LoginWhitelist;

namespace mozilla {

typedef uint32_t VerdictType;
typedef MozPromise<VerdictType, nsresult, false> ReputationPromise;

class LoginReputationService final : public nsILoginReputationService,
                                     public nsIObserver {
 public:
  NS_DECL_ISUPPORTS
  NS_DECL_NSILOGINREPUTATIONSERVICE
  NS_DECL_NSIOBSERVER

 public:
  static already_AddRefed<LoginReputationService> GetSingleton();

  static already_AddRefed<nsILoginReputationQuery> ConstructQueryParam(
      nsIURI* aURI);

  static nsCString VerdictTypeToString(VerdictType aVerdict);

 private:
  struct QueryRequest {
    QueryRequest(nsILoginReputationQuery* aParam,
                 nsILoginReputationQueryCallback* aCallback)
        : mParam(aParam), mCallback(aCallback) {}

    nsCOMPtr<nsILoginReputationQuery> mParam;
    nsCOMPtr<nsILoginReputationQueryCallback> mCallback;
  };

  /**
   * Global singleton object for holding this factory service.
   */
  static LoginReputationService* gLoginReputationService;

  LoginReputationService();
  ~LoginReputationService();

  nsresult Enable();

  nsresult Disable();

  nsresult QueryLoginWhitelist(QueryRequest* aRequest);

  // Called when a query request is finished.
  nsresult Finish(const QueryRequest* aRequest, nsresult aStatus,
                  VerdictType aVerdict);

  // Clear data and join the worker threads.
  nsresult Shutdown();

  RefPtr<LoginWhitelist> mLoginWhitelist;

  // Array that holds ongoing query requests which are added when
  // ::QueryReputation is called.
  nsTArray<UniquePtr<QueryRequest>> mQueryRequests;
};

}  // namespace mozilla

#endif  // LoginReputation_h__