c23ffbdf7e
Privileged test case for BpfBitmap. Test: atest BpfBitmapTest Bug: 179733303 Change-Id: I9c32d8843176815a725eccfd1f0400ec129cd94d
54 lines
1.8 KiB
C
54 lines
1.8 KiB
C
/*
|
|
* Copyright (C) 2021 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 <linux/if_ether.h>
|
|
#include <linux/in.h>
|
|
#include <linux/ip.h>
|
|
|
|
// The resulting .o needs to load on the Android S bpfloader v0.2
|
|
#define BPFLOADER_MIN_VER 2u
|
|
|
|
#include "bpf_helpers.h"
|
|
#include "bpf_net_helpers.h"
|
|
#include "bpf_tethering.h"
|
|
|
|
// Used only by TetheringPrivilegedTests, not by production code.
|
|
DEFINE_BPF_MAP_GRW(tether_downstream6_map, HASH, TetherDownstream6Key, Tether6Value, 16,
|
|
AID_NETWORK_STACK)
|
|
// Used only by BpfBitmapTest, not by production code.
|
|
DEFINE_BPF_MAP_GRW(bitmap, ARRAY, int, uint64_t, 2,
|
|
AID_NETWORK_STACK)
|
|
|
|
DEFINE_BPF_PROG_KVER("xdp/drop_ipv4_udp_ether", AID_ROOT, AID_NETWORK_STACK,
|
|
xdp_test, KVER(5, 9, 0))
|
|
(struct xdp_md *ctx) {
|
|
void *data = (void *)(long)ctx->data;
|
|
void *data_end = (void *)(long)ctx->data_end;
|
|
|
|
struct ethhdr *eth = data;
|
|
int hsize = sizeof(*eth);
|
|
|
|
struct iphdr *ip = data + hsize;
|
|
hsize += sizeof(struct iphdr);
|
|
|
|
if (data + hsize > data_end) return XDP_PASS;
|
|
if (eth->h_proto != htons(ETH_P_IP)) return XDP_PASS;
|
|
if (ip->protocol == IPPROTO_UDP) return XDP_DROP;
|
|
return XDP_PASS;
|
|
}
|
|
|
|
LICENSE("Apache 2.0");
|