File: client_storage.h

package info (click to toggle)
chromium 139.0.7258.138-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 6,120,676 kB
  • sloc: cpp: 35,100,869; ansic: 7,163,530; javascript: 4,103,002; python: 1,436,920; asm: 946,517; xml: 746,709; pascal: 187,653; perl: 88,691; sh: 88,436; objc: 79,953; sql: 51,488; cs: 44,583; fortran: 24,137; makefile: 22,147; tcl: 15,277; php: 13,980; yacc: 8,984; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (85 lines) | stat: -rw-r--r-- 2,857 bytes parent folder | download | duplicates (6)
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
81
82
83
84
85
// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef COMPONENTS_POLICY_TEST_SUPPORT_CLIENT_STORAGE_H_
#define COMPONENTS_POLICY_TEST_SUPPORT_CLIENT_STORAGE_H_

#include <stdint.h>

#include <map>
#include <optional>
#include <set>
#include <string>
#include <vector>

namespace policy {

// Stores information about registered clients.
class ClientStorage {
 public:
  struct ClientInfo {
    ClientInfo();
    ClientInfo(const ClientInfo& client_info);
    ClientInfo& operator=(const ClientInfo& client_info);
    ClientInfo(ClientInfo&& client_info);
    ClientInfo& operator=(ClientInfo&& client_info);
    ~ClientInfo();

    std::string device_id;
    std::string device_token;
    std::string machine_name;
    std::optional<std::string> username;
    std::vector<std::string> state_keys;
    std::set<std::string> allowed_policy_types;
  };

  ClientStorage();
  ClientStorage(ClientStorage&& client_storage);
  ClientStorage& operator=(ClientStorage&& client_storage);
  ~ClientStorage();

  // Register client so the server returns policy without the client having to
  // make a registration call. This could be called at any time (before or after
  // starting the server).
  void RegisterClient(const ClientInfo& client_info);

  // Returns true if there is a client associated with |device_id|.
  bool HasClient(const std::string& device_id) const;

  // Returns the client info associated with |device_id|. Fails if there is no
  // such a client.
  const ClientInfo& GetClient(const std::string& device_id) const;

  // Returns the client info associated with |device_id| or nullptr if there is
  // no such a client.
  const ClientInfo* GetClientOrNull(const std::string& device_id) const;

  // Returns the client info associated with |state_key| or nullptr if there is
  // no such a client.
  const ClientInfo* LookupByStateKey(const std::string& state_key) const;

  // Returns true if deletion of client with token |device_token| succeeded.
  bool DeleteClient(const std::string& device_token);

  // Returns the number of clients registered.
  size_t GetNumberOfRegisteredClients() const;

  // Returns hashes for all state keys registered with the server, which, when
  // divied by |modulus|, result in the specified |remainder|.
  std::vector<std::string> GetMatchingStateKeyHashes(uint64_t modulus,
                                                     uint64_t remainder) const;

  // Returns all the clients in the storage.
  std::vector<ClientInfo> GetAllClients();

 private:
  // Key: device ids.
  std::map<std::string, ClientInfo> clients_;
  // Maps device tokens to device IDs.
  std::map<std::string, std::string> registered_tokens_;
};

}  // namespace policy

#endif  // COMPONENTS_POLICY_TEST_SUPPORT_CLIENT_STORAGE_H_