sandbox/linux/bpf_dsl/test_trap_registry.h

   1 // Copyright 2015 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 SANDBOX_LINUX_BPF_DSL_TEST_TRAP_REGISTRY_H_
   6 #define SANDBOX_LINUX_BPF_DSL_TEST_TRAP_REGISTRY_H_
   7
   8 #include <stdint.h>
   9
  10 #include <map>
  11 #include <utility>
  12
  13 #include "sandbox/linux/bpf_dsl/trap_registry.h"
  14
  15 namespace sandbox {
  16 namespace bpf_dsl {
  17
  18 class TestTrapRegistry : public TrapRegistry {
  19  public:
  20   TestTrapRegistry();
  21   virtual ~TestTrapRegistry();
  22
  23   uint16_t Add(TrapFnc fnc, const void* aux, bool safe) override;
  24   bool EnableUnsafeTraps() override;
  25
  26  private:
  27   using Key = std::pair<TrapFnc, const void*>;
  28
  29   std::map<Key, uint16_t> map_;
  30
  31   DISALLOW_COPY_AND_ASSIGN(TestTrapRegistry);
  32 };
  33
  34 }  // namespace bpf_dsl
  35 }  // namespace sandbox
  36
  37 #endif  // SANDBOX_LINUX_BPF_DSL_TEST_TRAP_REGISTRY_H_