File: beacon.cc

package info (click to toggle)
chromium 139.0.7258.127-1
  • links: PTS, VCS
  • area: main
  • in suites:
  • size: 6,122,068 kB
  • sloc: cpp: 35,100,771; 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 (99 lines) | stat: -rw-r--r-- 3,560 bytes parent folder | download | duplicates (7)
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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/chrome_elf/third_party_dlls/beacon.h"

#include "chrome/chrome_elf/blocklist_constants.h"
#include "chrome/chrome_elf/nt_registry/nt_registry.h"
#include "chrome/install_static/install_util.h"

namespace third_party_dlls {

bool LeaveSetupBeacon() {
  HANDLE key_handle = INVALID_HANDLE_VALUE;

  if (!nt::CreateRegKey(nt::HKCU,
                        install_static::GetRegistryPath()
                            .append(blocklist::kRegistryBeaconKeyName)
                            .c_str(),
                        KEY_QUERY_VALUE | KEY_SET_VALUE, &key_handle)) {
    return false;
  }

  DWORD blocking_state = blocklist::BLOCKLIST_STATE_MAX;
  if (!nt::QueryRegValueDWORD(key_handle, blocklist::kBeaconState,
                              &blocking_state) ||
      blocking_state == blocklist::BLOCKLIST_DISABLED) {
    nt::CloseRegKey(key_handle);
    return false;
  }

  // Handle attempt count.
  // Only return true if BL is enabled and succeeded on previous run.
  bool success = false;
  if (blocking_state == blocklist::BLOCKLIST_ENABLED) {
    // If the blocking was successfully initialized on the previous run, reset
    // the failure counter. Then update the beacon state.
    if (nt::SetRegValueDWORD(key_handle, blocklist::kBeaconAttemptCount,
                             static_cast<DWORD>(0))) {
      if (nt::SetRegValueDWORD(key_handle, blocklist::kBeaconState,
                               blocklist::BLOCKLIST_SETUP_RUNNING))
        success = true;
    }
  } else {
    // Some part of the blocking setup failed last time. If this has occurred
    // blocklist::kBeaconMaxAttempts times in a row, we switch the state to
    // failed and skip setting up the blocking.
    DWORD attempt_count = 0;

    nt::QueryRegValueDWORD(key_handle, blocklist::kBeaconAttemptCount,
                           &attempt_count);
    ++attempt_count;
    nt::SetRegValueDWORD(key_handle, blocklist::kBeaconAttemptCount,
                         attempt_count);

    if (attempt_count >= blocklist::kBeaconMaxAttempts) {
      blocking_state = blocklist::BLOCKLIST_SETUP_FAILED;
      nt::SetRegValueDWORD(key_handle, blocklist::kBeaconState, blocking_state);
    }
  }

  nt::CloseRegKey(key_handle);
  return success;
}

bool ResetBeacon() {
  HANDLE key_handle = INVALID_HANDLE_VALUE;

  if (!nt::CreateRegKey(nt::HKCU,
                        install_static::GetRegistryPath()
                            .append(blocklist::kRegistryBeaconKeyName)
                            .c_str(),
                        KEY_QUERY_VALUE | KEY_SET_VALUE, &key_handle)) {
    return false;
  }

  DWORD blocking_state = blocklist::BLOCKLIST_STATE_MAX;
  if (!nt::QueryRegValueDWORD(key_handle, blocklist::kBeaconState,
                              &blocking_state)) {
    nt::CloseRegKey(key_handle);
    return false;
  }

  // Reaching this point with the setup running state means the setup did not
  // crash, so we reset to enabled.  Any other state indicates that setup was
  // skipped; in that case we leave the state alone for later recording.
  if (blocking_state == blocklist::BLOCKLIST_SETUP_RUNNING) {
    if (!nt::SetRegValueDWORD(key_handle, blocklist::kBeaconState,
                              blocklist::BLOCKLIST_ENABLED)) {
      nt::CloseRegKey(key_handle);
      return false;
    }
  }

  nt::CloseRegKey(key_handle);
  return true;
}

}  // namespace third_party_dlls