1 // Copyright 2014 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_SECCOMP_BPF_BPF_TESTER_COMPATIBILITY_DELEGATE_H_ 6 #define SANDBOX_LINUX_SECCOMP_BPF_BPF_TESTER_COMPATIBILITY_DELEGATE_H_ 7 8 #include "base/macros.h" 9 #include "base/memory/scoped_ptr.h" 10 #include "sandbox/linux/seccomp-bpf/sandbox_bpf_test_runner.h" 11 12 namespace sandbox { 13 14 // This templated class allows building a BPFTesterDelegate from a 15 // deprecated-style BPF policy (that is a SyscallEvaluator function pointer, 16 // instead of a SandboxBPFPolicy class), specified in |policy_function| and a 17 // function pointer to a test in |test_function|. 18 // This allows both the policy and the test function to take a pointer to an 19 // object of type "Aux" as a parameter. This is used to implement the BPF_TEST 20 // macro and should generally not be used directly. 21 template <class Policy, class Aux> 22 class BPFTesterCompatibilityDelegate : public BPFTesterDelegate { 23 public: 24 typedef void (*TestFunction)(Aux*); 25 BPFTesterCompatibilityDelegate(TestFunction test_function)26 explicit BPFTesterCompatibilityDelegate(TestFunction test_function) 27 : aux_(), test_function_(test_function) {} 28 ~BPFTesterCompatibilityDelegate()29 ~BPFTesterCompatibilityDelegate() override {} 30 GetSandboxBPFPolicy()31 scoped_ptr<bpf_dsl::Policy> GetSandboxBPFPolicy() override { 32 // The current method is guaranteed to only run in the child process 33 // running the test. In this process, the current object is guaranteed 34 // to live forever. So it's ok to pass aux_pointer_for_policy_ to 35 // the policy, which could in turn pass it to the kernel via Trap(). 36 return scoped_ptr<bpf_dsl::Policy>(new Policy(&aux_)); 37 } 38 RunTestFunction()39 void RunTestFunction() override { 40 // Run the actual test. 41 // The current object is guaranteed to live forever in the child process 42 // where this will run. 43 test_function_(&aux_); 44 } 45 46 private: 47 Aux aux_; 48 TestFunction test_function_; 49 50 DISALLOW_COPY_AND_ASSIGN(BPFTesterCompatibilityDelegate); 51 }; 52 53 } // namespace sandbox 54 55 #endif // SANDBOX_LINUX_SECCOMP_BPF_BPF_TESTER_COMPATIBILITY_DELEGATE_H_ 56