summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorPaul Jensen <pauljensen@google.com>2016-01-29 13:27:08 +0000
committerandroid-build-merger <android-build-merger@google.com>2016-01-29 13:27:08 +0000
commit5c0ab328284953ac3e3a3338b9b1393a4b6024c3 (patch)
tree114a4c9f502a5087c1dd5be87bdbfd41ccc85124
parent7a984ff54e870a623829c6570242354bba8b6d2e (diff)
parente1caf6d87d16f7dfbc83a5501d4ecfb85faa83f3 (diff)
downloadapf-5c0ab328284953ac3e3a3338b9b1393a4b6024c3.tar.gz
Android packet filter interpreter
am: e1caf6d87d * commit 'e1caf6d87d16f7dfbc83a5501d4ecfb85faa83f3': Android packet filter interpreter Initial empty repository
-rw-r--r--Android.mk17
-rw-r--r--LICENSE201
-rw-r--r--MODULE_LICENSE_APACHE20
-rw-r--r--apf_interpreter.c420
-rw-r--r--apf_interpreter.h53
5 files changed, 691 insertions, 0 deletions
diff --git a/Android.mk b/Android.mk
new file mode 100644
index 0000000..0faf418
--- /dev/null
+++ b/Android.mk
@@ -0,0 +1,17 @@
+# Copyright 2016 The Android Open Source Project
+
+LOCAL_PATH:= $(call my-dir)
+include $(CLEAR_VARS)
+
+LOCAL_INCLUDES += $(LOCAL_PATH)
+
+LOCAL_CFLAGS += \
+ -DAPF_FRAME_HEADER_SIZE=14 \
+ -std=c99 \
+ -Werror
+
+LOCAL_SRC_FILES += apf_interpreter.c
+
+LOCAL_MODULE:= libapf
+
+include $(BUILD_STATIC_LIBRARY)
diff --git a/LICENSE b/LICENSE
new file mode 100644
index 0000000..261eeb9
--- /dev/null
+++ b/LICENSE
@@ -0,0 +1,201 @@
+ Apache License
+ Version 2.0, January 2004
+ http://www.apache.org/licenses/
+
+ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+ 1. Definitions.
+
+ "License" shall mean the terms and conditions for use, reproduction,
+ and distribution as defined by Sections 1 through 9 of this document.
+
+ "Licensor" shall mean the copyright owner or entity authorized by
+ the copyright owner that is granting the License.
+
+ "Legal Entity" shall mean the union of the acting entity and all
+ other entities that control, are controlled by, or are under common
+ control with that entity. For the purposes of this definition,
+ "control" means (i) the power, direct or indirect, to cause the
+ direction or management of such entity, whether by contract or
+ otherwise, or (ii) ownership of fifty percent (50%) or more of the
+ outstanding shares, or (iii) beneficial ownership of such entity.
+
+ "You" (or "Your") shall mean an individual or Legal Entity
+ exercising permissions granted by this License.
+
+ "Source" form shall mean the preferred form for making modifications,
+ including but not limited to software source code, documentation
+ source, and configuration files.
+
+ "Object" form shall mean any form resulting from mechanical
+ transformation or translation of a Source form, including but
+ not limited to compiled object code, generated documentation,
+ and conversions to other media types.
+
+ "Work" shall mean the work of authorship, whether in Source or
+ Object form, made available under the License, as indicated by a
+ copyright notice that is included in or attached to the work
+ (an example is provided in the Appendix below).
+
+ "Derivative Works" shall mean any work, whether in Source or Object
+ form, that is based on (or derived from) the Work and for which the
+ editorial revisions, annotations, elaborations, or other modifications
+ represent, as a whole, an original work of authorship. For the purposes
+ of this License, Derivative Works shall not include works that remain
+ separable from, or merely link (or bind by name) to the interfaces of,
+ the Work and Derivative Works thereof.
+
+ "Contribution" shall mean any work of authorship, including
+ the original version of the Work and any modifications or additions
+ to that Work or Derivative Works thereof, that is intentionally
+ submitted to Licensor for inclusion in the Work by the copyright owner
+ or by an individual or Legal Entity authorized to submit on behalf of
+ the copyright owner. For the purposes of this definition, "submitted"
+ means any form of electronic, verbal, or written communication sent
+ to the Licensor or its representatives, including but not limited to
+ communication on electronic mailing lists, source code control systems,
+ and issue tracking systems that are managed by, or on behalf of, the
+ Licensor for the purpose of discussing and improving the Work, but
+ excluding communication that is conspicuously marked or otherwise
+ designated in writing by the copyright owner as "Not a Contribution."
+
+ "Contributor" shall mean Licensor and any individual or Legal Entity
+ on behalf of whom a Contribution has been received by Licensor and
+ subsequently incorporated within the Work.
+
+ 2. Grant of Copyright License. Subject to the terms and conditions of
+ this License, each Contributor hereby grants to You a perpetual,
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+ copyright license to reproduce, prepare Derivative Works of,
+ publicly display, publicly perform, sublicense, and distribute the
+ Work and such Derivative Works in Source or Object form.
+
+ 3. Grant of Patent License. Subject to the terms and conditions of
+ this License, each Contributor hereby grants to You a perpetual,
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+ (except as stated in this section) patent license to make, have made,
+ use, offer to sell, sell, import, and otherwise transfer the Work,
+ where such license applies only to those patent claims licensable
+ by such Contributor that are necessarily infringed by their
+ Contribution(s) alone or by combination of their Contribution(s)
+ with the Work to which such Contribution(s) was submitted. If You
+ institute patent litigation against any entity (including a
+ cross-claim or counterclaim in a lawsuit) alleging that the Work
+ or a Contribution incorporated within the Work constitutes direct
+ or contributory patent infringement, then any patent licenses
+ granted to You under this License for that Work shall terminate
+ as of the date such litigation is filed.
+
+ 4. Redistribution. You may reproduce and distribute copies of the
+ Work or Derivative Works thereof in any medium, with or without
+ modifications, and in Source or Object form, provided that You
+ meet the following conditions:
+
+ (a) You must give any other recipients of the Work or
+ Derivative Works a copy of this License; and
+
+ (b) You must cause any modified files to carry prominent notices
+ stating that You changed the files; and
+
+ (c) You must retain, in the Source form of any Derivative Works
+ that You distribute, all copyright, patent, trademark, and
+ attribution notices from the Source form of the Work,
+ excluding those notices that do not pertain to any part of
+ the Derivative Works; and
+
+ (d) If the Work includes a "NOTICE" text file as part of its
+ distribution, then any Derivative Works that You distribute must
+ include a readable copy of the attribution notices contained
+ within such NOTICE file, excluding those notices that do not
+ pertain to any part of the Derivative Works, in at least one
+ of the following places: within a NOTICE text file distributed
+ as part of the Derivative Works; within the Source form or
+ documentation, if provided along with the Derivative Works; or,
+ within a display generated by the Derivative Works, if and
+ wherever such third-party notices normally appear. The contents
+ of the NOTICE file are for informational purposes only and
+ do not modify the License. You may add Your own attribution
+ notices within Derivative Works that You distribute, alongside
+ or as an addendum to the NOTICE text from the Work, provided
+ that such additional attribution notices cannot be construed
+ as modifying the License.
+
+ You may add Your own copyright statement to Your modifications and
+ may provide additional or different license terms and conditions
+ for use, reproduction, or distribution of Your modifications, or
+ for any such Derivative Works as a whole, provided Your use,
+ reproduction, and distribution of the Work otherwise complies with
+ the conditions stated in this License.
+
+ 5. Submission of Contributions. Unless You explicitly state otherwise,
+ any Contribution intentionally submitted for inclusion in the Work
+ by You to the Licensor shall be under the terms and conditions of
+ this License, without any additional terms or conditions.
+ Notwithstanding the above, nothing herein shall supersede or modify
+ the terms of any separate license agreement you may have executed
+ with Licensor regarding such Contributions.
+
+ 6. Trademarks. This License does not grant permission to use the trade
+ names, trademarks, service marks, or product names of the Licensor,
+ except as required for reasonable and customary use in describing the
+ origin of the Work and reproducing the content of the NOTICE file.
+
+ 7. Disclaimer of Warranty. Unless required by applicable law or
+ agreed to in writing, Licensor provides the Work (and each
+ Contributor provides its Contributions) on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+ implied, including, without limitation, any warranties or conditions
+ of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+ PARTICULAR PURPOSE. You are solely responsible for determining the
+ appropriateness of using or redistributing the Work and assume any
+ risks associated with Your exercise of permissions under this License.
+
+ 8. Limitation of Liability. In no event and under no legal theory,
+ whether in tort (including negligence), contract, or otherwise,
+ unless required by applicable law (such as deliberate and grossly
+ negligent acts) or agreed to in writing, shall any Contributor be
+ liable to You for damages, including any direct, indirect, special,
+ incidental, or consequential damages of any character arising as a
+ result of this License or out of the use or inability to use the
+ Work (including but not limited to damages for loss of goodwill,
+ work stoppage, computer failure or malfunction, or any and all
+ other commercial damages or losses), even if such Contributor
+ has been advised of the possibility of such damages.
+
+ 9. Accepting Warranty or Additional Liability. While redistributing
+ the Work or Derivative Works thereof, You may choose to offer,
+ and charge a fee for, acceptance of support, warranty, indemnity,
+ or other liability obligations and/or rights consistent with this
+ License. However, in accepting such obligations, You may act only
+ on Your own behalf and on Your sole responsibility, not on behalf
+ of any other Contributor, and only if You agree to indemnify,
+ defend, and hold each Contributor harmless for any liability
+ incurred by, or claims asserted against, such Contributor by reason
+ of your accepting any such warranty or additional liability.
+
+ END OF TERMS AND CONDITIONS
+
+ APPENDIX: How to apply the Apache License to your work.
+
+ To apply the Apache License to your work, attach the following
+ boilerplate notice, with the fields enclosed by brackets "[]"
+ replaced with your own identifying information. (Don't include
+ the brackets!) The text should be enclosed in the appropriate
+ comment syntax for the file format. We also recommend that a
+ file or class name and description of purpose be included on the
+ same "printed page" as the copyright notice for easier
+ identification within third-party archives.
+
+ Copyright [yyyy] [name of copyright owner]
+
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+
+ http://www.apache.org/licenses/LICENSE-2.0
+
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
diff --git a/MODULE_LICENSE_APACHE2 b/MODULE_LICENSE_APACHE2
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/MODULE_LICENSE_APACHE2
diff --git a/apf_interpreter.c b/apf_interpreter.c
new file mode 100644
index 0000000..001acb9
--- /dev/null
+++ b/apf_interpreter.c
@@ -0,0 +1,420 @@
+/*
+ * Copyright 2016, The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "apf_interpreter.h"
+
+#include <string.h> // For memcmp
+
+// A brief overview of APF:
+//
+// APF machine is composed of:
+// 1. A read-only program consisting of bytecodes as described below.
+// 2. Two 32-bit registers, called R0 and R1.
+// 3. Sixteen 32-bit memory slots.
+// 4. A read-only packet.
+// The program is executed by the interpreter below and parses the packet
+// to determine if the application processor (AP) should be woken up to
+// handle the packet or if can be dropped.
+//
+// APF bytecode description:
+//
+// The APF interpreter uses big-endian byte order for loads from the packet
+// and for storing immediates in instructions.
+//
+// Each instruction starts with a byte composed of:
+// Top 5 bits form "opcode" field, see *_OPCODE defines below.
+// Next 2 bits form "size field", which indicate the length of an immediate
+// value which follows the first byte. Values in this field:
+// 0 => immediate value is 0 and no bytes follow.
+// 1 => immediate value is 1 byte big.
+// 2 => immediate value is 2 bytes big.
+// 3 => immediate value is 4 bytes big.
+// Bottom bit forms "register" field, which indicates which register this
+// instruction operates on.
+//
+// There are three main categories of instructions:
+// Load instructions
+// These instructions load byte(s) of the packet into a register.
+// They load either 1, 2 or 4 bytes, as determined by the "opcode" field.
+// They load into the register specified by the "register" field.
+// The immediate value that follows the first byte of the instruction is
+// the byte offset from the begining of the packet to load from.
+// There are "indexing" loads which add the value in R1 to the byte offset
+// to load from. The "opcode" field determines which loads are "indexing".
+// Arithmetic instructions
+// These instructions perform simple operations, like addition, on register
+// values. The result of these instructions is always written into R0. One
+// argument of the arithmetic operation is R0's value. The other argument
+// of the arithmetic operation is determined by the "register" field:
+// If the "register" field is 0 then the immediate value following
+// the first byte of the instruction is used as the other argument
+// to the arithmetic operation.
+// If the "register" field is 1 then R1's value is used as the other
+// argument to the arithmetic operation.
+// Conditional jump instructions
+// These instructions compare register R0's value with another value, and if
+// the comparison succeeds, jump (i.e. adjust the program counter). The
+// immediate value that follows the first byte of the instruction
+// represents the jump target offset, i.e. the value added to the program
+// counter if the comparison succeeds. The other value compared is
+// determined by the "register" field:
+// If the "register" field is 0 then another immediate value
+// follows the jump target offset. This immediate value is of the
+// same size as the jump target offset, and represents the value
+// to compare against.
+// If the "register" field is 1 then register R1's value is
+// compared against.
+// The type of comparison (e.g. equal to, greater than etc) is determined
+// by the "opcode" field. The comparison interprets both values being
+// compared as unsigned values.
+//
+// Miscellaneous details:
+//
+// Pre-filled memory slot values
+// When the APF program begins execution, three of the sixteen memory slots
+// are pre-filled by the interpreter with values that may be useful for
+// programs:
+// Slot #13 is filled with the IPv4 header length. This value is calculated
+// by loading the first byte of the IPv4 header and taking the
+// bottom 4 bits and multiplying their value by 4. This value is
+// set to zero if the first 4 bits after the link layer header are
+// not 4, indicating not IPv4.
+// Slot #14 is filled with size of the packet in bytes, including the
+// link-layer header if any.
+// Slot #15 is filled with the filter age in seconds. This is the number of
+// seconds since the AP send the program to the chipset. This may
+// be used by filters that should have a particular lifetime. For
+// example, it can be used to rate-limit particular packets to one
+// every N seconds.
+// Special jump targets:
+// When an APF program executes a jump to the byte immediately after the last
+// byte of the progam (i.e., one byte past the end of the program), this
+// signals the program has completed and determined the packet should be
+// passed to the AP.
+// When an APF program executes a jump two bytes past the end of the program,
+// this signals the program has completed and determined the packet should
+// be dropped.
+// Jump if byte sequence doesn't match:
+// This is a special instruction to facilitate matching long sequences of
+// bytes in the packet. Initially it is encoded like a conditional jump
+// instruction with two exceptions:
+// The first byte of the instruction is always followed by two immediate
+// fields: The first immediate field is the jump target offset like other
+// conditional jump instructions. The second immediate field specifies the
+// number of bytes to compare.
+// These two immediate fields are followed by a sequence of bytes. These
+// bytes are compared with the bytes in the packet starting from the
+// position specified by the value of the register specified by the
+// "register" field of the instruction.
+
+// Number of memory slots, see ldm/stm instructions.
+#define MEMORY_ITEMS 16
+// Upon program execution starting some memory slots are prefilled:
+#define MEMORY_OFFSET_IPV4_HEADER_SIZE 13 // 4*([APF_FRAME_HEADER_SIZE]&15)
+#define MEMORY_OFFSET_PACKET_SIZE 14 // Size of packet in bytes.
+#define MEMORY_OFFSET_FILTER_AGE 15 // Age since filter installed in seconds.
+
+// Leave 0 opcode unused as it's a good indicator of accidental incorrect execution (e.g. data).
+#define LDB_OPCODE 1 // Load 1 byte from immediate offset, e.g. "ldb R0, [5]"
+#define LDH_OPCODE 2 // Load 2 bytes from immediate offset, e.g. "ldh R0, [5]"
+#define LDW_OPCODE 3 // Load 4 bytes from immediate offset, e.g. "ldw R0, [5]"
+#define LDBX_OPCODE 4 // Load 1 byte from immediate offset plus register, e.g. "ldbx R0, [5]R0"
+#define LDHX_OPCODE 5 // Load 2 byte from immediate offset plus register, e.g. "ldhx R0, [5]R0"
+#define LDWX_OPCODE 6 // Load 4 byte from immediate offset plus register, e.g. "ldwx R0, [5]R0"
+#define ADD_OPCODE 7 // Add, e.g. "add R0,5"
+#define MUL_OPCODE 8 // Multiply, e.g. "mul R0,5"
+#define DIV_OPCODE 9 // Divide, e.g. "div R0,5"
+#define AND_OPCODE 10 // And, e.g. "and R0,5"
+#define OR_OPCODE 11 // Or, e.g. "or R0,5"
+#define SH_OPCODE 12 // Left shift, e.g, "sh R0, 5" or "sh R0, -5" (shifts right)
+#define LI_OPCODE 13 // Load immediate, e.g. "li R0,5" (immediate encoded as signed value)
+#define JMP_OPCODE 14 // Unconditional jump, e.g. "jmp label"
+#define JEQ_OPCODE 15 // Compare equal and branch, e.g. "jeq R0,5,label"
+#define JNE_OPCODE 16 // Compare not equal and branch, e.g. "jne R0,5,label"
+#define JGT_OPCODE 17 // Compare greater than and branch, e.g. "jgt R0,5,label"
+#define JLT_OPCODE 18 // Compare less than and branch, e.g. "jlt R0,5,label"
+#define JSET_OPCODE 19 // Compare any bits set and branch, e.g. "jset R0,5,label"
+#define JNEBS_OPCODE 20 // Compare not equal byte sequence, e.g. "jnebs R0,5,label,0x1122334455"
+#define EXT_OPCODE 21 // Immediate value is one of *_EXT_OPCODE
+// Extended opcodes. These all have an opcode of EXT_OPCODE
+// and specify the actual opcode in the immediate field.
+#define LDM_EXT_OPCODE 0 // Load from memory, e.g. "ldm R0,5"
+ // Values 0-15 represent loading the different memory slots.
+#define STM_EXT_OPCODE 16 // Store to memory, e.g. "stm R0,5"
+ // Values 16-31 represent storing to the different memory slots.
+#define NOT_EXT_OPCODE 32 // Not, e.g. "not R0"
+#define NEG_EXT_OPCODE 33 // Negate, e.g. "neg R0"
+#define SWAP_EXT_OPCODE 34 // Swap, e.g. "swap R0,R1"
+#define MOV_EXT_OPCODE 35 // Move, e.g. "move R0,R1"
+
+#define EXTRACT_OPCODE(i) (((i) >> 3) & 31)
+#define EXTRACT_REGISTER(i) ((i) & 1)
+#define EXTRACT_IMM_LENGTH(i) (((i) >> 1) & 3)
+
+// Return code indicating "packet" should accepted.
+#define PASS_PACKET 1
+// Return code indicating "packet" should be dropped.
+#define DROP_PACKET 0
+// Verify an internal condition and accept packet if it fails.
+#define ASSERT_RETURN(c) if (!(c)) return PASS_PACKET
+// If "c" is of an unsigned type, generate a compile warning that gets promoted to an error.
+// This makes bounds checking simpler because ">= 0" can be avoided. Otherwise adding
+// superfluous ">= 0" with unsigned expressions generates compile warnings.
+#define ENFORCE_UNSIGNED(c) ((c)==(uint32_t)(c))
+
+/**
+ * Runs a packet filtering program over a packet.
+ *
+ * @param program the program bytecode.
+ * @param program_len the length of {@code apf_program} in bytes.
+ * @param packet the packet bytes, starting from the 802.3 header and not
+ * including any CRC bytes at the end.
+ * @param packet_len the length of {@code packet} in bytes.
+ * @param filter_age the number of seconds since the filter was programmed.
+ *
+ * @return non-zero if packet should be passed to AP, zero if
+ * packet should be dropped.
+ */
+int accept_packet(const uint8_t* program, uint32_t program_len,
+ const uint8_t* packet, uint32_t packet_len,
+ uint32_t filter_age) {
+// Is offset within program bounds?
+#define IN_PROGRAM_BOUNDS(p) (ENFORCE_UNSIGNED(p), (p) < program_len)
+// Is offset within packet bounds?
+#define IN_PACKET_BOUNDS(p) (ENFORCE_UNSIGNED(p), (p) < packet_len)
+// Accept packet if not within program bounds
+#define ASSERT_IN_PROGRAM_BOUNDS(p) ASSERT_RETURN(IN_PROGRAM_BOUNDS(p))
+// Accept packet if not within packet bounds
+#define ASSERT_IN_PACKET_BOUNDS(p) ASSERT_RETURN(IN_PACKET_BOUNDS(p))
+ // Program counter.
+ uint32_t pc = 0;
+// Accept packet if not within program or not ahead of program counter
+#define ASSERT_FORWARD_IN_PROGRAM(p) ASSERT_RETURN(IN_PROGRAM_BOUNDS(p) && (p) >= pc)
+ // Memory slot values.
+ uint32_t memory[MEMORY_ITEMS] = {};
+ // Fill in pre-filled memory slot values.
+ memory[MEMORY_OFFSET_PACKET_SIZE] = packet_len;
+ memory[MEMORY_OFFSET_FILTER_AGE] = filter_age;
+ ASSERT_IN_PACKET_BOUNDS(APF_FRAME_HEADER_SIZE);
+ // Only populate if IP version is IPv4.
+ if ((packet[APF_FRAME_HEADER_SIZE] & 0xf0) == 0x40) {
+ memory[MEMORY_OFFSET_IPV4_HEADER_SIZE] = (packet[APF_FRAME_HEADER_SIZE] & 15) * 4;
+ }
+ // Register values.
+ uint32_t registers[2] = {};
+ // Count of instructions remaining to execute. This is done to ensure an
+ // upper bound on execution time. It should never be hit and is only for
+ // safety. Initialize to the number of bytes in the program which is an
+ // upper bound on the number of instructions in the program.
+ uint32_t instructions_remaining = program_len;
+
+ do {
+ if (pc == program_len) {
+ return PASS_PACKET;
+ } else if (pc == (program_len + 1)) {
+ return DROP_PACKET;
+ }
+ ASSERT_IN_PROGRAM_BOUNDS(pc);
+ const uint8_t bytecode = program[pc++];
+ const uint32_t opcode = EXTRACT_OPCODE(bytecode);
+ const uint32_t reg_num = EXTRACT_REGISTER(bytecode);
+#define REG (registers[reg_num])
+#define OTHER_REG (registers[reg_num ^ 1])
+ // All instructions have immediate fields, so load them now.
+ const uint32_t len_field = EXTRACT_IMM_LENGTH(bytecode);
+ uint32_t imm = 0;
+ int32_t signed_imm = 0;
+ if (len_field != 0) {
+ const uint32_t imm_len = 1 << (len_field - 1);
+ ASSERT_FORWARD_IN_PROGRAM(pc + imm_len - 1);
+ for (uint32_t i = 0; i < imm_len; i++)
+ imm = (imm << 8) | program[pc++];
+ // Sign extend imm into signed_imm.
+ signed_imm = imm << ((4 - imm_len) * 8);
+ signed_imm >>= (4 - imm_len) * 8;
+ }
+ switch (opcode) {
+ case LDB_OPCODE:
+ case LDH_OPCODE:
+ case LDW_OPCODE:
+ case LDBX_OPCODE:
+ case LDHX_OPCODE:
+ case LDWX_OPCODE: {
+ uint32_t offs = imm;
+ if (opcode >= LDBX_OPCODE) {
+ // Note: this can overflow and actually decrease offs.
+ offs += registers[1];
+ }
+ ASSERT_IN_PACKET_BOUNDS(offs);
+ uint32_t load_size;
+ switch (opcode) {
+ case LDB_OPCODE:
+ case LDBX_OPCODE:
+ load_size = 1;
+ break;
+ case LDH_OPCODE:
+ case LDHX_OPCODE:
+ load_size = 2;
+ break;
+ case LDW_OPCODE:
+ case LDWX_OPCODE:
+ load_size = 4;
+ break;
+ // Immediately enclosing switch statement guarantees
+ // opcode cannot be any other value.
+ }
+ const uint32_t end_offs = offs + (load_size - 1);
+ // Catch overflow/wrap-around.
+ ASSERT_RETURN(end_offs >= offs);
+ ASSERT_IN_PACKET_BOUNDS(end_offs);
+ uint32_t val = 0;
+ while (load_size--)
+ val = (val << 8) | packet[offs++];
+ REG = val;
+ break;
+ }
+ case JMP_OPCODE:
+ // This can jump backwards. Infinite looping prevented by instructions_remaining.
+ pc += imm;
+ break;
+ case JEQ_OPCODE:
+ case JNE_OPCODE:
+ case JGT_OPCODE:
+ case JLT_OPCODE:
+ case JSET_OPCODE:
+ case JNEBS_OPCODE: {
+ // Load second immediate field.
+ uint32_t cmp_imm = 0;
+ if (reg_num == 1) {
+ cmp_imm = registers[1];
+ } else if (len_field != 0) {
+ uint32_t cmp_imm_len = 1 << (len_field - 1);
+ ASSERT_FORWARD_IN_PROGRAM(pc + cmp_imm_len - 1);
+ for (uint32_t i = 0; i < cmp_imm_len; i++)
+ cmp_imm = (cmp_imm << 8) | program[pc++];
+ }
+ switch (opcode) {
+ case JEQ_OPCODE:
+ if (registers[0] == cmp_imm)
+ pc += imm;
+ break;
+ case JNE_OPCODE:
+ if (registers[0] != cmp_imm)
+ pc += imm;
+ break;
+ case JGT_OPCODE:
+ if (registers[0] > cmp_imm)
+ pc += imm;
+ break;
+ case JLT_OPCODE:
+ if (registers[0] < cmp_imm)
+ pc += imm;
+ break;
+ case JSET_OPCODE:
+ if (registers[0] & cmp_imm)
+ pc += imm;
+ break;
+ case JNEBS_OPCODE: {
+ // cmp_imm is size in bytes of data to compare.
+ // pc is offset of program bytes to compare.
+ // imm is jump target offset.
+ // REG is offset of packet bytes to compare.
+ ASSERT_FORWARD_IN_PROGRAM(pc + cmp_imm - 1);
+ ASSERT_IN_PACKET_BOUNDS(REG);
+ const uint32_t last_packet_offs = REG + cmp_imm - 1;
+ ASSERT_RETURN(last_packet_offs >= REG);
+ ASSERT_IN_PACKET_BOUNDS(last_packet_offs);
+ if (memcmp(program + pc, packet + REG, cmp_imm))
+ pc += imm;
+ // skip past comparison bytes
+ pc += cmp_imm;
+ break;
+ }
+ }
+ break;
+ }
+ case ADD_OPCODE:
+ registers[0] += reg_num ? registers[1] : imm;
+ break;
+ case MUL_OPCODE:
+ registers[0] *= reg_num ? registers[1] : imm;
+ break;
+ case DIV_OPCODE: {
+ const uint32_t div_operand = reg_num ? registers[1] : imm;
+ ASSERT_RETURN(div_operand);
+ registers[0] /= div_operand;
+ break;
+ }
+ case AND_OPCODE:
+ registers[0] &= reg_num ? registers[1] : imm;
+ break;
+ case OR_OPCODE:
+ registers[0] |= reg_num ? registers[1] : imm;
+ break;
+ case SH_OPCODE: {
+ const int32_t shift_val = reg_num ? (int32_t)registers[1] : signed_imm;
+ if (shift_val > 0)
+ registers[0] <<= shift_val;
+ else
+ registers[0] >>= -shift_val;
+ break;
+ }
+ case LI_OPCODE:
+ REG = signed_imm;
+ break;
+ case EXT_OPCODE:
+ if (
+// If LDM_EXT_OPCODE is 0 and imm is compared with it, a compiler error will result,
+// instead just enforce that imm is unsigned (so it's always greater or equal to 0).
+#if LDM_EXT_OPCODE == 0
+ ENFORCE_UNSIGNED(imm),
+#else
+ imm >= LDM_EXT_OPCODE &&
+#endif
+ imm < (LDM_EXT_OPCODE + MEMORY_ITEMS)) {
+ REG = memory[imm - LDM_EXT_OPCODE];
+ } else if (imm >= STM_EXT_OPCODE && imm < (STM_EXT_OPCODE + MEMORY_ITEMS)) {
+ memory[imm - STM_EXT_OPCODE] = REG;
+ } else switch (imm) {
+ case NOT_EXT_OPCODE:
+ REG = ~REG;
+ break;
+ case NEG_EXT_OPCODE:
+ REG = -REG;
+ break;
+ case SWAP_EXT_OPCODE: {
+ uint32_t tmp = REG;
+ REG = OTHER_REG;
+ OTHER_REG = tmp;
+ break;
+ }
+ case MOV_EXT_OPCODE:
+ REG = OTHER_REG;
+ break;
+ // Unknown extended opcode
+ default:
+ // Bail out
+ return PASS_PACKET;
+ }
+ break;
+ // Unknown opcode
+ default:
+ // Bail out
+ return PASS_PACKET;
+ }
+ } while (instructions_remaining--);
+ return PASS_PACKET;
+}
diff --git a/apf_interpreter.h b/apf_interpreter.h
new file mode 100644
index 0000000..2012c28
--- /dev/null
+++ b/apf_interpreter.h
@@ -0,0 +1,53 @@
+/*
+ * Copyright 2015, The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#ifndef APF_INTERPRETER_H_
+#define APF_INTERPRETER_H_
+
+#include <stdint.h>
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+/**
+ * Version of APF instruction set processed by accept_packet().
+ * Should be returned by wifi_get_packet_filter_info.
+ */
+#define APF_VERSION 2
+
+/**
+ * Runs a packet filtering program over a packet.
+ *
+ * @param program the program bytecode.
+ * @param program_len the length of {@code apf_program} in bytes.
+ * @param packet the packet bytes, starting from the 802.3 header and not
+ * including any CRC bytes at the end.
+ * @param packet_len the length of {@code packet} in bytes.
+ * @param filter_age the number of seconds since the filter was programmed.
+ *
+ * @return non-zero if packet should be passed to AP, zero if
+ * packet should be dropped.
+ */
+int accept_packet(const uint8_t* program, uint32_t program_len,
+ const uint8_t* packet, uint32_t packet_len,
+ uint32_t filter_age);
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif // APF_INTERPRETER_H_