summaryrefslogtreecommitdiff
path: root/arch/x86/net/bpf_timed_may_goto.S
blob: 20de4a14dc4cf24f3d548753e289e47de5c01b05 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
// SPDX-License-Identifier: GPL-2.0
/* Copyright (c) 2025 Meta Platforms, Inc. and affiliates. */

#include <linux/export.h>
#include <linux/linkage.h>
#include <asm/nospec-branch.h>

	.code64
	.section .text, "ax"

SYM_FUNC_START(arch_bpf_timed_may_goto)
	ANNOTATE_NOENDBR

	/* Save r0-r5. */
	pushq %rax
	pushq %rdi
	pushq %rsi
	pushq %rdx
	pushq %rcx
	pushq %r8

	/*
	 * r10 passes us stack depth, load the pointer to count and timestamp as
	 * first argument to the call below.
	 */
	leaq (%rbp, %r10, 1), %rdi

	/* Emit call depth accounting for call below. */
	CALL_DEPTH_ACCOUNT
	call bpf_check_timed_may_goto

	/* BPF_REG_AX=r10 will be stored into count, so move return value to it. */
	movq %rax, %r10

	/* Restore r5-r0. */
	popq %r8
	popq %rcx
	popq %rdx
	popq %rsi
	popq %rdi
	popq %rax

	RET
SYM_FUNC_END(arch_bpf_timed_may_goto)