| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "sandbox/linux/bpf_dsl/codegen.h" | 5 #include "sandbox/linux/bpf_dsl/codegen.h" |
| 6 | 6 |
| 7 #include <linux/filter.h> | |
| 8 | |
| 9 #include <limits> | 7 #include <limits> |
| 10 #include <utility> | 8 #include <utility> |
| 11 | 9 |
| 12 #include "base/logging.h" | 10 #include "base/logging.h" |
| 11 #include "sandbox/linux/system_headers/linux_filter.h" |
| 13 | 12 |
| 14 // This CodeGen implementation strives for simplicity while still | 13 // This CodeGen implementation strives for simplicity while still |
| 15 // generating acceptable BPF programs under typical usage patterns | 14 // generating acceptable BPF programs under typical usage patterns |
| 16 // (e.g., by PolicyCompiler). | 15 // (e.g., by PolicyCompiler). |
| 17 // | 16 // |
| 18 // The key to its simplicity is that BPF programs only support forward | 17 // The key to its simplicity is that BPF programs only support forward |
| 19 // jumps/branches, which allows constraining the DAG construction API | 18 // jumps/branches, which allows constraining the DAG construction API |
| 20 // to make instruction nodes immutable. Immutable nodes admits a | 19 // to make instruction nodes immutable. Immutable nodes admits a |
| 21 // simple greedy approach of emitting new instructions as needed and | 20 // simple greedy approach of emitting new instructions as needed and |
| 22 // then reusing existing ones that have already been emitted. This | 21 // then reusing existing ones that have already been emitted. This |
| (...skipping 102 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 125 CHECK_LE(jf, kBranchRange); | 124 CHECK_LE(jf, kBranchRange); |
| 126 } else { | 125 } else { |
| 127 CHECK_EQ(0U, jt); | 126 CHECK_EQ(0U, jt); |
| 128 CHECK_EQ(0U, jf); | 127 CHECK_EQ(0U, jf); |
| 129 } | 128 } |
| 130 | 129 |
| 131 CHECK_LT(program_.size(), static_cast<size_t>(BPF_MAXINSNS)); | 130 CHECK_LT(program_.size(), static_cast<size_t>(BPF_MAXINSNS)); |
| 132 CHECK_EQ(program_.size(), equivalent_.size()); | 131 CHECK_EQ(program_.size(), equivalent_.size()); |
| 133 | 132 |
| 134 Node res = program_.size(); | 133 Node res = program_.size(); |
| 135 program_.push_back(sock_filter{code, jt, jf, k}); | 134 program_.push_back(sock_filter{ |
| 135 code, static_cast<uint8_t>(jt), static_cast<uint8_t>(jf), k}); |
| 136 equivalent_.push_back(res); | 136 equivalent_.push_back(res); |
| 137 return res; | 137 return res; |
| 138 } | 138 } |
| 139 | 139 |
| 140 size_t CodeGen::Offset(Node target) const { | 140 size_t CodeGen::Offset(Node target) const { |
| 141 CHECK_LT(target, program_.size()) << "Bogus offset target node"; | 141 CHECK_LT(target, program_.size()) << "Bogus offset target node"; |
| 142 return (program_.size() - 1) - target; | 142 return (program_.size() - 1) - target; |
| 143 } | 143 } |
| 144 | 144 |
| 145 // TODO(mdempsky): Move into a general base::Tuple helper library. | 145 // TODO(mdempsky): Move into a general base::Tuple helper library. |
| 146 bool CodeGen::MemoKeyLess::operator()(const MemoKey& lhs, | 146 bool CodeGen::MemoKeyLess::operator()(const MemoKey& lhs, |
| 147 const MemoKey& rhs) const { | 147 const MemoKey& rhs) const { |
| 148 if (get<0>(lhs) != get<0>(rhs)) | 148 if (get<0>(lhs) != get<0>(rhs)) |
| 149 return get<0>(lhs) < get<0>(rhs); | 149 return get<0>(lhs) < get<0>(rhs); |
| 150 if (get<1>(lhs) != get<1>(rhs)) | 150 if (get<1>(lhs) != get<1>(rhs)) |
| 151 return get<1>(lhs) < get<1>(rhs); | 151 return get<1>(lhs) < get<1>(rhs); |
| 152 if (get<2>(lhs) != get<2>(rhs)) | 152 if (get<2>(lhs) != get<2>(rhs)) |
| 153 return get<2>(lhs) < get<2>(rhs); | 153 return get<2>(lhs) < get<2>(rhs); |
| 154 if (get<3>(lhs) != get<3>(rhs)) | 154 if (get<3>(lhs) != get<3>(rhs)) |
| 155 return get<3>(lhs) < get<3>(rhs); | 155 return get<3>(lhs) < get<3>(rhs); |
| 156 return false; | 156 return false; |
| 157 } | 157 } |
| 158 | 158 |
| 159 } // namespace sandbox | 159 } // namespace sandbox |
| OLD | NEW |