1// Copyright 2013 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#ifndef CHROME_BROWSER_EXTENSIONS_INSTALL_VERIFIER_H_
6#define CHROME_BROWSER_EXTENSIONS_INSTALL_VERIFIER_H_
7
8#include <queue>
9#include <set>
10#include <string>
11
12#include "base/basictypes.h"
13#include "base/callback.h"
14#include "base/memory/linked_ptr.h"
15#include "base/memory/scoped_ptr.h"
16#include "extensions/browser/management_policy.h"
17#include "extensions/common/extension.h"
18
19namespace net {
20class URLRequestContextGetter;
21}
22
23namespace extensions {
24
25class ExtensionPrefs;
26class InstallSigner;
27struct InstallSignature;
28
29// This class implements verification that a set of extensions are either from
30// the webstore or are whitelisted by enterprise policy.  The webstore
31// verification process works by sending a request to a backend server to get a
32// signature proving that a set of extensions are verified. This signature is
33// written into the extension preferences and is checked for validity when
34// being read back again.
35//
36// This class should be kept notified of runtime changes to the set of
37// extensions installed from the webstore.
38class InstallVerifier : public ManagementPolicy::Provider {
39 public:
40  InstallVerifier(ExtensionPrefs* prefs,
41                  net::URLRequestContextGetter* context_getter);
42  virtual ~InstallVerifier();
43
44  // Returns whether |extension| is of a type that needs verification.
45  static bool NeedsVerification(const Extension& extension);
46
47  // Initializes this object for use, including reading preferences and
48  // validating the stored signature.
49  void Init();
50
51  // Do we need to be bootstrapped? (i.e. do we have a signature already). If
52  // this is true, then consumers of this class should use Add/AddMany to get
53  // an initial one so that MustRemainDisabled can actually check against it.
54  bool NeedsBootstrap();
55
56  // Returns the timestamp of our InstallSignature, if we have one.
57  base::Time SignatureTimestamp();
58
59  // A callback for indicating success/failure of adding new ids.
60  typedef base::Callback<void(bool)> AddResultCallback;
61
62  // Try adding a new |id| (or set of ids) to the list of verified ids. When
63  // this process is finished |callback| will be run with success/failure of
64  // the signature request (not necessarily whether the ids were verified).
65  void Add(const std::string& id, const AddResultCallback& callback);
66  void AddMany(const ExtensionIdSet& ids,
67               const AddResultCallback& callback);
68
69  // Call this to add a set of ids that will immediately be considered allowed,
70  // and kick off an aysnchronous request to Add.
71  void AddProvisional(const ExtensionIdSet& ids);
72
73  // Removes an id or set of ids from the verified list.
74  void Remove(const std::string& id);
75  void RemoveMany(const ExtensionIdSet& ids);
76
77  // ManagementPolicy::Provider interface.
78  virtual std::string GetDebugPolicyProviderName() const OVERRIDE;
79  virtual bool MustRemainDisabled(const Extension* extension,
80                                  Extension::DisableReason* reason,
81                                  base::string16* error) const OVERRIDE;
82
83 private:
84  // We keep a list of operations to the current set of extensions - either
85  // additions or removals.
86  enum OperationType {
87    ADD,
88    REMOVE
89  };
90
91  // This is an operation we want to apply to the current set of verified ids.
92  struct PendingOperation {
93    OperationType type;
94
95    // This is the set of ids being either added or removed.
96    ExtensionIdSet ids;
97
98    AddResultCallback callback;
99
100    explicit PendingOperation();
101    ~PendingOperation();
102  };
103
104  // Removes any no-longer-installed ids, requesting a new signature if needed.
105  void GarbageCollect();
106
107  // Returns whether an extension id is allowed by policy.
108  bool AllowedByEnterprisePolicy(const std::string& id) const;
109
110  // Returns whether the given |id| is included in our verified signature.
111  bool IsVerified(const std::string& id) const;
112
113  // Returns true if the extension with |id| was installed later than the
114  // timestamp of our signature.
115  bool WasInstalledAfterSignature(const std::string& id) const;
116
117  // Begins the process of fetching a new signature, based on applying the
118  // operation at the head of the queue to the current set of ids in
119  // |signature_| (if any) and then sending a request to sign that.
120  void BeginFetch();
121
122  // Saves the current value of |signature_| to the prefs;
123  void SaveToPrefs();
124
125  // Called with the result of a signature request, or NULL on failure.
126  void SignatureCallback(scoped_ptr<InstallSignature> signature);
127
128  ExtensionPrefs* prefs_;
129  net::URLRequestContextGetter* context_getter_;
130
131  // This is the most up-to-date signature, read out of |prefs_| during
132  // initialization and updated anytime we get new id's added.
133  scoped_ptr<InstallSignature> signature_;
134
135  // The current InstallSigner, if we have a signature request running.
136  scoped_ptr<InstallSigner> signer_;
137
138  // A queue of operations to apply to the current set of allowed ids.
139  std::queue<linked_ptr<PendingOperation> > operation_queue_;
140
141  // A set of ids that have been provisionally added, which we're willing to
142  // consider allowed until we hear back from the server signature request.
143  ExtensionIdSet provisional_;
144
145  DISALLOW_COPY_AND_ASSIGN(InstallVerifier);
146};
147
148}  // namespace extensions
149
150#endif  // CHROME_BROWSER_EXTENSIONS_INSTALL_VERIFIER_H_
151