98 lines
2.6 KiB
C
98 lines
2.6 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
/* Bounds checks for PTR_TO_FLOW_KEYS pointer arithmetic. */
|
|
|
|
#include "vmlinux.h"
|
|
#include <bpf/bpf_helpers.h>
|
|
#include "bpf_misc.h"
|
|
|
|
/* sizeof(struct bpf_flow_keys) is well under 4096, so +0x1000 is OOB. */
|
|
|
|
SEC("flow_dissector")
|
|
__description("flow_keys: in-bounds constant pointer arithmetic accepted")
|
|
__success
|
|
__naked void flow_keys_const_inbounds(void)
|
|
{
|
|
asm volatile (" \
|
|
r1 = *(u64 *)(r1 + %[flow_keys]); \
|
|
r1 += 8; \
|
|
r0 = *(u64 *)(r1 + 0); \
|
|
r0 = 0; \
|
|
exit; \
|
|
" :
|
|
: __imm_const(flow_keys, offsetof(struct __sk_buff, flow_keys))
|
|
: __clobber_all);
|
|
}
|
|
|
|
SEC("flow_dissector")
|
|
__description("flow_keys: OOB via constant pointer arithmetic rejected")
|
|
__failure __msg("invalid access to flow keys off=4096 size=8")
|
|
__naked void flow_keys_const_oob_read(void)
|
|
{
|
|
asm volatile (" \
|
|
r1 = *(u64 *)(r1 + %[flow_keys]); \
|
|
r1 += 4096; \
|
|
r0 = *(u64 *)(r1 + 0); \
|
|
r0 = 0; \
|
|
exit; \
|
|
" :
|
|
: __imm_const(flow_keys, offsetof(struct __sk_buff, flow_keys))
|
|
: __clobber_all);
|
|
}
|
|
|
|
SEC("flow_dissector")
|
|
__description("flow_keys: OOB write via constant pointer arithmetic rejected")
|
|
__failure __msg("invalid access to flow keys off=4096 size=8")
|
|
__naked void flow_keys_const_oob_write(void)
|
|
{
|
|
asm volatile (" \
|
|
r1 = *(u64 *)(r1 + %[flow_keys]); \
|
|
r1 += 4096; \
|
|
r2 = 0; \
|
|
*(u64 *)(r1 + 0) = r2; \
|
|
r0 = 0; \
|
|
exit; \
|
|
" :
|
|
: __imm_const(flow_keys, offsetof(struct __sk_buff, flow_keys))
|
|
: __clobber_all);
|
|
}
|
|
|
|
/* Equivalent OOB expressed directly in insn->off; this form was always
|
|
* rejected and is kept to show both forms now share one diagnostic.
|
|
*/
|
|
SEC("flow_dissector")
|
|
__description("flow_keys: OOB via insn->off rejected")
|
|
__failure __msg("invalid access to flow keys off=4096 size=8")
|
|
__naked void flow_keys_insn_off_oob(void)
|
|
{
|
|
asm volatile (" \
|
|
r1 = *(u64 *)(r1 + %[flow_keys]); \
|
|
r0 = *(u64 *)(r1 + 4096); \
|
|
r0 = 0; \
|
|
exit; \
|
|
" :
|
|
: __imm_const(flow_keys, offsetof(struct __sk_buff, flow_keys))
|
|
: __clobber_all);
|
|
}
|
|
|
|
SEC("flow_dissector")
|
|
__description("flow_keys: variable pointer arithmetic rejected")
|
|
__failure __msg("R1 pointer arithmetic on flow_keys prohibited")
|
|
__naked void flow_keys_var_read(void)
|
|
{
|
|
asm volatile (" \
|
|
r6 = r1; \
|
|
call %[bpf_get_prandom_u32]; \
|
|
r0 &= 0xFFFF; \
|
|
r1 = *(u64 *)(r6 + %[flow_keys]); \
|
|
r1 += r0; \
|
|
r0 = *(u64 *)(r1 + 0); \
|
|
r0 = 0; \
|
|
exit; \
|
|
" :
|
|
: __imm_const(flow_keys, offsetof(struct __sk_buff, flow_keys)),
|
|
__imm(bpf_get_prandom_u32)
|
|
: __clobber_all);
|
|
}
|
|
|
|
char _license[] SEC("license") = "GPL";
|