blob: f13c543b7b3673013ef5ee126f2fc03fcbfac01b [file] [log] [blame]
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001/*
2 * Linux Socket Filter - Kernel level socket filtering
3 *
4 * Based on the design of the Berkeley Packet Filter. The new
5 * internal format has been designed by PLUMgrid:
6 *
7 * Copyright (c) 2011 - 2014 PLUMgrid, http://plumgrid.com
8 *
9 * Authors:
10 *
11 * Jay Schulist <jschlst@samba.org>
12 * Alexei Starovoitov <ast@plumgrid.com>
13 * Daniel Borkmann <dborkman@redhat.com>
14 *
15 * This program is free software; you can redistribute it and/or
16 * modify it under the terms of the GNU General Public License
17 * as published by the Free Software Foundation; either version
18 * 2 of the License, or (at your option) any later version.
19 *
20 * Andi Kleen - Fix a few bad bugs and races.
Alexei Starovoitov4df95ff2014-07-30 20:34:14 -070021 * Kris Katterjohn - Added many additional checks in bpf_check_classic()
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -070022 */
Daniel Borkmann738cbe72014-09-08 08:04:47 +020023
Yonghong Song838e9692018-11-19 15:29:11 -080024#include <uapi/linux/btf.h>
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -070025#include <linux/filter.h>
26#include <linux/skbuff.h>
Daniel Borkmann60a3b222014-09-02 22:53:44 +020027#include <linux/vmalloc.h>
Daniel Borkmann738cbe72014-09-08 08:04:47 +020028#include <linux/random.h>
29#include <linux/moduleloader.h>
Alexei Starovoitov09756af2014-09-26 00:17:00 -070030#include <linux/bpf.h>
Yonghong Song838e9692018-11-19 15:29:11 -080031#include <linux/btf.h>
Josh Poimboeuf39853cc2016-02-28 22:22:37 -060032#include <linux/frame.h>
Daniel Borkmann74451e662017-02-16 22:24:50 +010033#include <linux/rbtree_latch.h>
34#include <linux/kallsyms.h>
35#include <linux/rcupdate.h>
Yonghong Songc195651e2018-04-28 22:28:08 -070036#include <linux/perf_event.h>
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -070037
Daniel Borkmann3324b582015-05-29 23:23:07 +020038#include <asm/unaligned.h>
39
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -070040/* Registers */
41#define BPF_R0 regs[BPF_REG_0]
42#define BPF_R1 regs[BPF_REG_1]
43#define BPF_R2 regs[BPF_REG_2]
44#define BPF_R3 regs[BPF_REG_3]
45#define BPF_R4 regs[BPF_REG_4]
46#define BPF_R5 regs[BPF_REG_5]
47#define BPF_R6 regs[BPF_REG_6]
48#define BPF_R7 regs[BPF_REG_7]
49#define BPF_R8 regs[BPF_REG_8]
50#define BPF_R9 regs[BPF_REG_9]
51#define BPF_R10 regs[BPF_REG_10]
52
53/* Named registers */
54#define DST regs[insn->dst_reg]
55#define SRC regs[insn->src_reg]
56#define FP regs[BPF_REG_FP]
Daniel Borkmann144cd912019-01-03 00:58:28 +010057#define AX regs[BPF_REG_AX]
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -070058#define ARG1 regs[BPF_REG_ARG1]
59#define CTX regs[BPF_REG_CTX]
60#define IMM insn->imm
61
62/* No hurry in this branch
63 *
64 * Exported for the bpf jit load helper.
65 */
66void *bpf_internal_load_pointer_neg_helper(const struct sk_buff *skb, int k, unsigned int size)
67{
68 u8 *ptr = NULL;
69
70 if (k >= SKF_NET_OFF)
71 ptr = skb_network_header(skb) + k - SKF_NET_OFF;
72 else if (k >= SKF_LL_OFF)
73 ptr = skb_mac_header(skb) + k - SKF_LL_OFF;
Daniel Borkmann3324b582015-05-29 23:23:07 +020074
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -070075 if (ptr >= skb->head && ptr + size <= skb_tail_pointer(skb))
76 return ptr;
77
78 return NULL;
79}
80
Daniel Borkmann60a3b222014-09-02 22:53:44 +020081struct bpf_prog *bpf_prog_alloc(unsigned int size, gfp_t gfp_extra_flags)
82{
Michal Hocko19809c22017-05-08 15:57:44 -070083 gfp_t gfp_flags = GFP_KERNEL | __GFP_ZERO | gfp_extra_flags;
Alexei Starovoitov09756af2014-09-26 00:17:00 -070084 struct bpf_prog_aux *aux;
Daniel Borkmann60a3b222014-09-02 22:53:44 +020085 struct bpf_prog *fp;
86
87 size = round_up(size, PAGE_SIZE);
88 fp = __vmalloc(size, gfp_flags, PAGE_KERNEL);
89 if (fp == NULL)
90 return NULL;
91
Alexei Starovoitov09756af2014-09-26 00:17:00 -070092 aux = kzalloc(sizeof(*aux), GFP_KERNEL | gfp_extra_flags);
93 if (aux == NULL) {
Daniel Borkmann60a3b222014-09-02 22:53:44 +020094 vfree(fp);
95 return NULL;
96 }
97
98 fp->pages = size / PAGE_SIZE;
Alexei Starovoitov09756af2014-09-26 00:17:00 -070099 fp->aux = aux;
Daniel Borkmanne9d8afa2015-10-29 14:58:08 +0100100 fp->aux->prog = fp;
Alexei Starovoitov60b58afc2017-12-14 17:55:14 -0800101 fp->jit_requested = ebpf_jit_enabled();
Daniel Borkmann60a3b222014-09-02 22:53:44 +0200102
Daniel Borkmann74451e662017-02-16 22:24:50 +0100103 INIT_LIST_HEAD_RCU(&fp->aux->ksym_lnode);
104
Daniel Borkmann60a3b222014-09-02 22:53:44 +0200105 return fp;
106}
107EXPORT_SYMBOL_GPL(bpf_prog_alloc);
108
Martin KaFai Lauc454a462018-12-07 16:42:25 -0800109int bpf_prog_alloc_jited_linfo(struct bpf_prog *prog)
110{
111 if (!prog->aux->nr_linfo || !prog->jit_requested)
112 return 0;
113
114 prog->aux->jited_linfo = kcalloc(prog->aux->nr_linfo,
115 sizeof(*prog->aux->jited_linfo),
116 GFP_KERNEL | __GFP_NOWARN);
117 if (!prog->aux->jited_linfo)
118 return -ENOMEM;
119
120 return 0;
121}
122
123void bpf_prog_free_jited_linfo(struct bpf_prog *prog)
124{
125 kfree(prog->aux->jited_linfo);
126 prog->aux->jited_linfo = NULL;
127}
128
129void bpf_prog_free_unused_jited_linfo(struct bpf_prog *prog)
130{
131 if (prog->aux->jited_linfo && !prog->aux->jited_linfo[0])
132 bpf_prog_free_jited_linfo(prog);
133}
134
135/* The jit engine is responsible to provide an array
136 * for insn_off to the jited_off mapping (insn_to_jit_off).
137 *
138 * The idx to this array is the insn_off. Hence, the insn_off
139 * here is relative to the prog itself instead of the main prog.
140 * This array has one entry for each xlated bpf insn.
141 *
142 * jited_off is the byte off to the last byte of the jited insn.
143 *
144 * Hence, with
145 * insn_start:
146 * The first bpf insn off of the prog. The insn off
147 * here is relative to the main prog.
148 * e.g. if prog is a subprog, insn_start > 0
149 * linfo_idx:
150 * The prog's idx to prog->aux->linfo and jited_linfo
151 *
152 * jited_linfo[linfo_idx] = prog->bpf_func
153 *
154 * For i > linfo_idx,
155 *
156 * jited_linfo[i] = prog->bpf_func +
157 * insn_to_jit_off[linfo[i].insn_off - insn_start - 1]
158 */
159void bpf_prog_fill_jited_linfo(struct bpf_prog *prog,
160 const u32 *insn_to_jit_off)
161{
162 u32 linfo_idx, insn_start, insn_end, nr_linfo, i;
163 const struct bpf_line_info *linfo;
164 void **jited_linfo;
165
166 if (!prog->aux->jited_linfo)
167 /* Userspace did not provide linfo */
168 return;
169
170 linfo_idx = prog->aux->linfo_idx;
171 linfo = &prog->aux->linfo[linfo_idx];
172 insn_start = linfo[0].insn_off;
173 insn_end = insn_start + prog->len;
174
175 jited_linfo = &prog->aux->jited_linfo[linfo_idx];
176 jited_linfo[0] = prog->bpf_func;
177
178 nr_linfo = prog->aux->nr_linfo - linfo_idx;
179
180 for (i = 1; i < nr_linfo && linfo[i].insn_off < insn_end; i++)
181 /* The verifier ensures that linfo[i].insn_off is
182 * strictly increasing
183 */
184 jited_linfo[i] = prog->bpf_func +
185 insn_to_jit_off[linfo[i].insn_off - insn_start - 1];
186}
187
188void bpf_prog_free_linfo(struct bpf_prog *prog)
189{
190 bpf_prog_free_jited_linfo(prog);
191 kvfree(prog->aux->linfo);
192}
193
Daniel Borkmann60a3b222014-09-02 22:53:44 +0200194struct bpf_prog *bpf_prog_realloc(struct bpf_prog *fp_old, unsigned int size,
195 gfp_t gfp_extra_flags)
196{
Michal Hocko19809c22017-05-08 15:57:44 -0700197 gfp_t gfp_flags = GFP_KERNEL | __GFP_ZERO | gfp_extra_flags;
Daniel Borkmann60a3b222014-09-02 22:53:44 +0200198 struct bpf_prog *fp;
Daniel Borkmann5ccb0712016-12-18 01:52:58 +0100199 u32 pages, delta;
200 int ret;
Daniel Borkmann60a3b222014-09-02 22:53:44 +0200201
202 BUG_ON(fp_old == NULL);
203
204 size = round_up(size, PAGE_SIZE);
Daniel Borkmann5ccb0712016-12-18 01:52:58 +0100205 pages = size / PAGE_SIZE;
206 if (pages <= fp_old->pages)
Daniel Borkmann60a3b222014-09-02 22:53:44 +0200207 return fp_old;
208
Daniel Borkmann5ccb0712016-12-18 01:52:58 +0100209 delta = pages - fp_old->pages;
210 ret = __bpf_prog_charge(fp_old->aux->user, delta);
211 if (ret)
212 return NULL;
213
Daniel Borkmann60a3b222014-09-02 22:53:44 +0200214 fp = __vmalloc(size, gfp_flags, PAGE_KERNEL);
Daniel Borkmann5ccb0712016-12-18 01:52:58 +0100215 if (fp == NULL) {
216 __bpf_prog_uncharge(fp_old->aux->user, delta);
217 } else {
Daniel Borkmann60a3b222014-09-02 22:53:44 +0200218 memcpy(fp, fp_old, fp_old->pages * PAGE_SIZE);
Daniel Borkmann5ccb0712016-12-18 01:52:58 +0100219 fp->pages = pages;
Daniel Borkmanne9d8afa2015-10-29 14:58:08 +0100220 fp->aux->prog = fp;
Daniel Borkmann60a3b222014-09-02 22:53:44 +0200221
Alexei Starovoitov09756af2014-09-26 00:17:00 -0700222 /* We keep fp->aux from fp_old around in the new
Daniel Borkmann60a3b222014-09-02 22:53:44 +0200223 * reallocated structure.
224 */
Alexei Starovoitov09756af2014-09-26 00:17:00 -0700225 fp_old->aux = NULL;
Daniel Borkmann60a3b222014-09-02 22:53:44 +0200226 __bpf_prog_free(fp_old);
227 }
228
229 return fp;
230}
Daniel Borkmann60a3b222014-09-02 22:53:44 +0200231
232void __bpf_prog_free(struct bpf_prog *fp)
233{
Alexei Starovoitov09756af2014-09-26 00:17:00 -0700234 kfree(fp->aux);
Daniel Borkmann60a3b222014-09-02 22:53:44 +0200235 vfree(fp);
236}
Daniel Borkmann60a3b222014-09-02 22:53:44 +0200237
Daniel Borkmannf1f77142017-01-13 23:38:15 +0100238int bpf_prog_calc_tag(struct bpf_prog *fp)
Daniel Borkmann7bd509e2016-12-04 23:19:41 +0100239{
240 const u32 bits_offset = SHA_MESSAGE_BYTES - sizeof(__be64);
Daniel Borkmannf1f77142017-01-13 23:38:15 +0100241 u32 raw_size = bpf_prog_tag_scratch_size(fp);
242 u32 digest[SHA_DIGEST_WORDS];
Daniel Borkmannaafe6ae2016-12-18 01:52:57 +0100243 u32 ws[SHA_WORKSPACE_WORDS];
Daniel Borkmann7bd509e2016-12-04 23:19:41 +0100244 u32 i, bsize, psize, blocks;
Daniel Borkmannaafe6ae2016-12-18 01:52:57 +0100245 struct bpf_insn *dst;
Daniel Borkmann7bd509e2016-12-04 23:19:41 +0100246 bool was_ld_map;
Daniel Borkmannaafe6ae2016-12-18 01:52:57 +0100247 u8 *raw, *todo;
Daniel Borkmann7bd509e2016-12-04 23:19:41 +0100248 __be32 *result;
249 __be64 *bits;
250
Daniel Borkmannaafe6ae2016-12-18 01:52:57 +0100251 raw = vmalloc(raw_size);
252 if (!raw)
253 return -ENOMEM;
254
Daniel Borkmannf1f77142017-01-13 23:38:15 +0100255 sha_init(digest);
Daniel Borkmann7bd509e2016-12-04 23:19:41 +0100256 memset(ws, 0, sizeof(ws));
257
258 /* We need to take out the map fd for the digest calculation
259 * since they are unstable from user space side.
260 */
Daniel Borkmannaafe6ae2016-12-18 01:52:57 +0100261 dst = (void *)raw;
Daniel Borkmann7bd509e2016-12-04 23:19:41 +0100262 for (i = 0, was_ld_map = false; i < fp->len; i++) {
263 dst[i] = fp->insnsi[i];
264 if (!was_ld_map &&
265 dst[i].code == (BPF_LD | BPF_IMM | BPF_DW) &&
266 dst[i].src_reg == BPF_PSEUDO_MAP_FD) {
267 was_ld_map = true;
268 dst[i].imm = 0;
269 } else if (was_ld_map &&
270 dst[i].code == 0 &&
271 dst[i].dst_reg == 0 &&
272 dst[i].src_reg == 0 &&
273 dst[i].off == 0) {
274 was_ld_map = false;
275 dst[i].imm = 0;
276 } else {
277 was_ld_map = false;
278 }
279 }
280
Daniel Borkmannaafe6ae2016-12-18 01:52:57 +0100281 psize = bpf_prog_insn_size(fp);
282 memset(&raw[psize], 0, raw_size - psize);
Daniel Borkmann7bd509e2016-12-04 23:19:41 +0100283 raw[psize++] = 0x80;
284
285 bsize = round_up(psize, SHA_MESSAGE_BYTES);
286 blocks = bsize / SHA_MESSAGE_BYTES;
Daniel Borkmannaafe6ae2016-12-18 01:52:57 +0100287 todo = raw;
Daniel Borkmann7bd509e2016-12-04 23:19:41 +0100288 if (bsize - psize >= sizeof(__be64)) {
289 bits = (__be64 *)(todo + bsize - sizeof(__be64));
290 } else {
291 bits = (__be64 *)(todo + bsize + bits_offset);
292 blocks++;
293 }
294 *bits = cpu_to_be64((psize - 1) << 3);
295
296 while (blocks--) {
Daniel Borkmannf1f77142017-01-13 23:38:15 +0100297 sha_transform(digest, todo, ws);
Daniel Borkmann7bd509e2016-12-04 23:19:41 +0100298 todo += SHA_MESSAGE_BYTES;
299 }
300
Daniel Borkmannf1f77142017-01-13 23:38:15 +0100301 result = (__force __be32 *)digest;
Daniel Borkmann7bd509e2016-12-04 23:19:41 +0100302 for (i = 0; i < SHA_DIGEST_WORDS; i++)
Daniel Borkmannf1f77142017-01-13 23:38:15 +0100303 result[i] = cpu_to_be32(digest[i]);
304 memcpy(fp->tag, result, sizeof(fp->tag));
Daniel Borkmannaafe6ae2016-12-18 01:52:57 +0100305
306 vfree(raw);
307 return 0;
Daniel Borkmann7bd509e2016-12-04 23:19:41 +0100308}
309
Jakub Kicinski2cbd95a2019-01-22 22:45:18 -0800310static int bpf_adj_delta_to_imm(struct bpf_insn *insn, u32 pos, s32 end_old,
311 s32 end_new, u32 curr, const bool probe_pass)
Daniel Borkmannc237ee52016-05-13 19:08:30 +0200312{
Daniel Borkmann050fad72018-05-17 01:44:11 +0200313 const s64 imm_min = S32_MIN, imm_max = S32_MAX;
Jakub Kicinski2cbd95a2019-01-22 22:45:18 -0800314 s32 delta = end_new - end_old;
Daniel Borkmann050fad72018-05-17 01:44:11 +0200315 s64 imm = insn->imm;
316
Jakub Kicinski2cbd95a2019-01-22 22:45:18 -0800317 if (curr < pos && curr + imm + 1 >= end_old)
Daniel Borkmann050fad72018-05-17 01:44:11 +0200318 imm += delta;
Jakub Kicinski2cbd95a2019-01-22 22:45:18 -0800319 else if (curr >= end_new && curr + imm + 1 < end_new)
Daniel Borkmann050fad72018-05-17 01:44:11 +0200320 imm -= delta;
321 if (imm < imm_min || imm > imm_max)
322 return -ERANGE;
323 if (!probe_pass)
324 insn->imm = imm;
325 return 0;
326}
327
Jakub Kicinski2cbd95a2019-01-22 22:45:18 -0800328static int bpf_adj_delta_to_off(struct bpf_insn *insn, u32 pos, s32 end_old,
329 s32 end_new, u32 curr, const bool probe_pass)
Daniel Borkmann050fad72018-05-17 01:44:11 +0200330{
331 const s32 off_min = S16_MIN, off_max = S16_MAX;
Jakub Kicinski2cbd95a2019-01-22 22:45:18 -0800332 s32 delta = end_new - end_old;
Daniel Borkmann050fad72018-05-17 01:44:11 +0200333 s32 off = insn->off;
334
Jakub Kicinski2cbd95a2019-01-22 22:45:18 -0800335 if (curr < pos && curr + off + 1 >= end_old)
Daniel Borkmann050fad72018-05-17 01:44:11 +0200336 off += delta;
Jakub Kicinski2cbd95a2019-01-22 22:45:18 -0800337 else if (curr >= end_new && curr + off + 1 < end_new)
Daniel Borkmann050fad72018-05-17 01:44:11 +0200338 off -= delta;
339 if (off < off_min || off > off_max)
340 return -ERANGE;
341 if (!probe_pass)
342 insn->off = off;
343 return 0;
344}
345
Jakub Kicinski2cbd95a2019-01-22 22:45:18 -0800346static int bpf_adj_branches(struct bpf_prog *prog, u32 pos, s32 end_old,
347 s32 end_new, const bool probe_pass)
Daniel Borkmann050fad72018-05-17 01:44:11 +0200348{
Jakub Kicinski2cbd95a2019-01-22 22:45:18 -0800349 u32 i, insn_cnt = prog->len + (probe_pass ? end_new - end_old : 0);
Daniel Borkmannc237ee52016-05-13 19:08:30 +0200350 struct bpf_insn *insn = prog->insnsi;
Daniel Borkmann050fad72018-05-17 01:44:11 +0200351 int ret = 0;
Daniel Borkmannc237ee52016-05-13 19:08:30 +0200352
353 for (i = 0; i < insn_cnt; i++, insn++) {
Daniel Borkmann050fad72018-05-17 01:44:11 +0200354 u8 code;
355
356 /* In the probing pass we still operate on the original,
357 * unpatched image in order to check overflows before we
358 * do any other adjustments. Therefore skip the patchlet.
359 */
360 if (probe_pass && i == pos) {
Jakub Kicinski2cbd95a2019-01-22 22:45:18 -0800361 i = end_new;
362 insn = prog->insnsi + end_old;
Alexei Starovoitov1ea47e02017-12-14 17:55:13 -0800363 }
Daniel Borkmann050fad72018-05-17 01:44:11 +0200364 code = insn->code;
Jiong Wang092ed092019-01-26 12:26:01 -0500365 if ((BPF_CLASS(code) != BPF_JMP &&
366 BPF_CLASS(code) != BPF_JMP32) ||
Daniel Borkmann050fad72018-05-17 01:44:11 +0200367 BPF_OP(code) == BPF_EXIT)
368 continue;
369 /* Adjust offset of jmps if we cross patch boundaries. */
370 if (BPF_OP(code) == BPF_CALL) {
371 if (insn->src_reg != BPF_PSEUDO_CALL)
372 continue;
Jakub Kicinski2cbd95a2019-01-22 22:45:18 -0800373 ret = bpf_adj_delta_to_imm(insn, pos, end_old,
374 end_new, i, probe_pass);
Daniel Borkmann050fad72018-05-17 01:44:11 +0200375 } else {
Jakub Kicinski2cbd95a2019-01-22 22:45:18 -0800376 ret = bpf_adj_delta_to_off(insn, pos, end_old,
377 end_new, i, probe_pass);
Daniel Borkmann050fad72018-05-17 01:44:11 +0200378 }
379 if (ret)
380 break;
Daniel Borkmannc237ee52016-05-13 19:08:30 +0200381 }
Daniel Borkmann050fad72018-05-17 01:44:11 +0200382
383 return ret;
Daniel Borkmannc237ee52016-05-13 19:08:30 +0200384}
385
Martin KaFai Lauc454a462018-12-07 16:42:25 -0800386static void bpf_adj_linfo(struct bpf_prog *prog, u32 off, u32 delta)
387{
388 struct bpf_line_info *linfo;
389 u32 i, nr_linfo;
390
391 nr_linfo = prog->aux->nr_linfo;
392 if (!nr_linfo || !delta)
393 return;
394
395 linfo = prog->aux->linfo;
396
397 for (i = 0; i < nr_linfo; i++)
398 if (off < linfo[i].insn_off)
399 break;
400
401 /* Push all off < linfo[i].insn_off by delta */
402 for (; i < nr_linfo; i++)
403 linfo[i].insn_off += delta;
404}
405
Daniel Borkmannc237ee52016-05-13 19:08:30 +0200406struct bpf_prog *bpf_patch_insn_single(struct bpf_prog *prog, u32 off,
407 const struct bpf_insn *patch, u32 len)
408{
409 u32 insn_adj_cnt, insn_rest, insn_delta = len - 1;
Daniel Borkmann050fad72018-05-17 01:44:11 +0200410 const u32 cnt_max = S16_MAX;
Daniel Borkmannc237ee52016-05-13 19:08:30 +0200411 struct bpf_prog *prog_adj;
412
413 /* Since our patchlet doesn't expand the image, we're done. */
414 if (insn_delta == 0) {
415 memcpy(prog->insnsi + off, patch, sizeof(*patch));
416 return prog;
417 }
418
419 insn_adj_cnt = prog->len + insn_delta;
420
Daniel Borkmann050fad72018-05-17 01:44:11 +0200421 /* Reject anything that would potentially let the insn->off
422 * target overflow when we have excessive program expansions.
423 * We need to probe here before we do any reallocation where
424 * we afterwards may not fail anymore.
425 */
426 if (insn_adj_cnt > cnt_max &&
Jakub Kicinski2cbd95a2019-01-22 22:45:18 -0800427 bpf_adj_branches(prog, off, off + 1, off + len, true))
Daniel Borkmann050fad72018-05-17 01:44:11 +0200428 return NULL;
429
Daniel Borkmannc237ee52016-05-13 19:08:30 +0200430 /* Several new instructions need to be inserted. Make room
431 * for them. Likely, there's no need for a new allocation as
432 * last page could have large enough tailroom.
433 */
434 prog_adj = bpf_prog_realloc(prog, bpf_prog_size(insn_adj_cnt),
435 GFP_USER);
436 if (!prog_adj)
437 return NULL;
438
439 prog_adj->len = insn_adj_cnt;
440
441 /* Patching happens in 3 steps:
442 *
443 * 1) Move over tail of insnsi from next instruction onwards,
444 * so we can patch the single target insn with one or more
445 * new ones (patching is always from 1 to n insns, n > 0).
446 * 2) Inject new instructions at the target location.
447 * 3) Adjust branch offsets if necessary.
448 */
449 insn_rest = insn_adj_cnt - off - len;
450
451 memmove(prog_adj->insnsi + off + len, prog_adj->insnsi + off + 1,
452 sizeof(*patch) * insn_rest);
453 memcpy(prog_adj->insnsi + off, patch, sizeof(*patch) * len);
454
Daniel Borkmann050fad72018-05-17 01:44:11 +0200455 /* We are guaranteed to not fail at this point, otherwise
456 * the ship has sailed to reverse to the original state. An
457 * overflow cannot happen at this point.
458 */
Jakub Kicinski2cbd95a2019-01-22 22:45:18 -0800459 BUG_ON(bpf_adj_branches(prog_adj, off, off + 1, off + len, false));
Daniel Borkmannc237ee52016-05-13 19:08:30 +0200460
Martin KaFai Lauc454a462018-12-07 16:42:25 -0800461 bpf_adj_linfo(prog_adj, off, insn_delta);
462
Daniel Borkmannc237ee52016-05-13 19:08:30 +0200463 return prog_adj;
464}
465
Jakub Kicinski52875a02019-01-22 22:45:20 -0800466int bpf_remove_insns(struct bpf_prog *prog, u32 off, u32 cnt)
467{
468 /* Branch offsets can't overflow when program is shrinking, no need
469 * to call bpf_adj_branches(..., true) here
470 */
471 memmove(prog->insnsi + off, prog->insnsi + off + cnt,
472 sizeof(struct bpf_insn) * (prog->len - off - cnt));
473 prog->len -= cnt;
474
475 return WARN_ON_ONCE(bpf_adj_branches(prog, off, off + cnt, off, false));
476}
477
Daniel Borkmann7d1982b2018-06-15 02:30:47 +0200478void bpf_prog_kallsyms_del_subprogs(struct bpf_prog *fp)
479{
480 int i;
481
482 for (i = 0; i < fp->aux->func_cnt; i++)
483 bpf_prog_kallsyms_del(fp->aux->func[i]);
484}
485
486void bpf_prog_kallsyms_del_all(struct bpf_prog *fp)
487{
488 bpf_prog_kallsyms_del_subprogs(fp);
489 bpf_prog_kallsyms_del(fp);
490}
491
Daniel Borkmannb954d832014-09-10 15:01:02 +0200492#ifdef CONFIG_BPF_JIT
Daniel Borkmannfa9dd592018-01-20 01:24:33 +0100493/* All BPF JIT sysctl knobs here. */
494int bpf_jit_enable __read_mostly = IS_BUILTIN(CONFIG_BPF_JIT_ALWAYS_ON);
495int bpf_jit_harden __read_mostly;
496int bpf_jit_kallsyms __read_mostly;
Daniel Borkmannfdadd042018-12-11 12:14:12 +0100497long bpf_jit_limit __read_mostly;
Daniel Borkmannfa9dd592018-01-20 01:24:33 +0100498
Daniel Borkmann74451e662017-02-16 22:24:50 +0100499static __always_inline void
500bpf_get_prog_addr_region(const struct bpf_prog *prog,
501 unsigned long *symbol_start,
502 unsigned long *symbol_end)
503{
504 const struct bpf_binary_header *hdr = bpf_jit_binary_hdr(prog);
505 unsigned long addr = (unsigned long)hdr;
506
507 WARN_ON_ONCE(!bpf_prog_ebpf_jited(prog));
508
509 *symbol_start = addr;
510 *symbol_end = addr + hdr->pages * PAGE_SIZE;
511}
512
513static void bpf_get_prog_name(const struct bpf_prog *prog, char *sym)
514{
Martin KaFai Lau368211f2017-10-05 21:52:13 -0700515 const char *end = sym + KSYM_NAME_LEN;
Yonghong Song838e9692018-11-19 15:29:11 -0800516 const struct btf_type *type;
517 const char *func_name;
Martin KaFai Lau368211f2017-10-05 21:52:13 -0700518
Daniel Borkmann74451e662017-02-16 22:24:50 +0100519 BUILD_BUG_ON(sizeof("bpf_prog_") +
Martin KaFai Lau368211f2017-10-05 21:52:13 -0700520 sizeof(prog->tag) * 2 +
521 /* name has been null terminated.
522 * We should need +1 for the '_' preceding
523 * the name. However, the null character
524 * is double counted between the name and the
525 * sizeof("bpf_prog_") above, so we omit
526 * the +1 here.
527 */
528 sizeof(prog->aux->name) > KSYM_NAME_LEN);
Daniel Borkmann74451e662017-02-16 22:24:50 +0100529
530 sym += snprintf(sym, KSYM_NAME_LEN, "bpf_prog_");
531 sym = bin2hex(sym, prog->tag, sizeof(prog->tag));
Yonghong Song838e9692018-11-19 15:29:11 -0800532
533 /* prog->aux->name will be ignored if full btf name is available */
Martin KaFai Lau73372242018-12-05 17:35:43 -0800534 if (prog->aux->func_info_cnt) {
Yonghong Songba64e7d2018-11-24 23:20:44 -0800535 type = btf_type_by_id(prog->aux->btf,
536 prog->aux->func_info[prog->aux->func_idx].type_id);
Yonghong Song838e9692018-11-19 15:29:11 -0800537 func_name = btf_name_by_offset(prog->aux->btf, type->name_off);
538 snprintf(sym, (size_t)(end - sym), "_%s", func_name);
539 return;
540 }
541
Martin KaFai Lau368211f2017-10-05 21:52:13 -0700542 if (prog->aux->name[0])
543 snprintf(sym, (size_t)(end - sym), "_%s", prog->aux->name);
544 else
545 *sym = 0;
Daniel Borkmann74451e662017-02-16 22:24:50 +0100546}
547
548static __always_inline unsigned long
549bpf_get_prog_addr_start(struct latch_tree_node *n)
550{
551 unsigned long symbol_start, symbol_end;
552 const struct bpf_prog_aux *aux;
553
554 aux = container_of(n, struct bpf_prog_aux, ksym_tnode);
555 bpf_get_prog_addr_region(aux->prog, &symbol_start, &symbol_end);
556
557 return symbol_start;
558}
559
560static __always_inline bool bpf_tree_less(struct latch_tree_node *a,
561 struct latch_tree_node *b)
562{
563 return bpf_get_prog_addr_start(a) < bpf_get_prog_addr_start(b);
564}
565
566static __always_inline int bpf_tree_comp(void *key, struct latch_tree_node *n)
567{
568 unsigned long val = (unsigned long)key;
569 unsigned long symbol_start, symbol_end;
570 const struct bpf_prog_aux *aux;
571
572 aux = container_of(n, struct bpf_prog_aux, ksym_tnode);
573 bpf_get_prog_addr_region(aux->prog, &symbol_start, &symbol_end);
574
575 if (val < symbol_start)
576 return -1;
577 if (val >= symbol_end)
578 return 1;
579
580 return 0;
581}
582
583static const struct latch_tree_ops bpf_tree_ops = {
584 .less = bpf_tree_less,
585 .comp = bpf_tree_comp,
586};
587
588static DEFINE_SPINLOCK(bpf_lock);
589static LIST_HEAD(bpf_kallsyms);
590static struct latch_tree_root bpf_tree __cacheline_aligned;
591
Daniel Borkmann74451e662017-02-16 22:24:50 +0100592static void bpf_prog_ksym_node_add(struct bpf_prog_aux *aux)
593{
594 WARN_ON_ONCE(!list_empty(&aux->ksym_lnode));
595 list_add_tail_rcu(&aux->ksym_lnode, &bpf_kallsyms);
596 latch_tree_insert(&aux->ksym_tnode, &bpf_tree, &bpf_tree_ops);
597}
598
599static void bpf_prog_ksym_node_del(struct bpf_prog_aux *aux)
600{
601 if (list_empty(&aux->ksym_lnode))
602 return;
603
604 latch_tree_erase(&aux->ksym_tnode, &bpf_tree, &bpf_tree_ops);
605 list_del_rcu(&aux->ksym_lnode);
606}
607
608static bool bpf_prog_kallsyms_candidate(const struct bpf_prog *fp)
609{
610 return fp->jited && !bpf_prog_was_classic(fp);
611}
612
613static bool bpf_prog_kallsyms_verify_off(const struct bpf_prog *fp)
614{
615 return list_empty(&fp->aux->ksym_lnode) ||
616 fp->aux->ksym_lnode.prev == LIST_POISON2;
617}
618
619void bpf_prog_kallsyms_add(struct bpf_prog *fp)
620{
Daniel Borkmann74451e662017-02-16 22:24:50 +0100621 if (!bpf_prog_kallsyms_candidate(fp) ||
622 !capable(CAP_SYS_ADMIN))
623 return;
624
Hannes Frederic Sowad24f7c72017-04-27 01:39:33 +0200625 spin_lock_bh(&bpf_lock);
Daniel Borkmann74451e662017-02-16 22:24:50 +0100626 bpf_prog_ksym_node_add(fp->aux);
Hannes Frederic Sowad24f7c72017-04-27 01:39:33 +0200627 spin_unlock_bh(&bpf_lock);
Daniel Borkmann74451e662017-02-16 22:24:50 +0100628}
629
630void bpf_prog_kallsyms_del(struct bpf_prog *fp)
631{
Daniel Borkmann74451e662017-02-16 22:24:50 +0100632 if (!bpf_prog_kallsyms_candidate(fp))
633 return;
634
Hannes Frederic Sowad24f7c72017-04-27 01:39:33 +0200635 spin_lock_bh(&bpf_lock);
Daniel Borkmann74451e662017-02-16 22:24:50 +0100636 bpf_prog_ksym_node_del(fp->aux);
Hannes Frederic Sowad24f7c72017-04-27 01:39:33 +0200637 spin_unlock_bh(&bpf_lock);
Daniel Borkmann74451e662017-02-16 22:24:50 +0100638}
639
640static struct bpf_prog *bpf_prog_kallsyms_find(unsigned long addr)
641{
642 struct latch_tree_node *n;
643
644 if (!bpf_jit_kallsyms_enabled())
645 return NULL;
646
647 n = latch_tree_find((void *)addr, &bpf_tree, &bpf_tree_ops);
648 return n ?
649 container_of(n, struct bpf_prog_aux, ksym_tnode)->prog :
650 NULL;
651}
652
653const char *__bpf_address_lookup(unsigned long addr, unsigned long *size,
654 unsigned long *off, char *sym)
655{
656 unsigned long symbol_start, symbol_end;
657 struct bpf_prog *prog;
658 char *ret = NULL;
659
660 rcu_read_lock();
661 prog = bpf_prog_kallsyms_find(addr);
662 if (prog) {
663 bpf_get_prog_addr_region(prog, &symbol_start, &symbol_end);
664 bpf_get_prog_name(prog, sym);
665
666 ret = sym;
667 if (size)
668 *size = symbol_end - symbol_start;
669 if (off)
670 *off = addr - symbol_start;
671 }
672 rcu_read_unlock();
673
674 return ret;
675}
676
677bool is_bpf_text_address(unsigned long addr)
678{
679 bool ret;
680
681 rcu_read_lock();
682 ret = bpf_prog_kallsyms_find(addr) != NULL;
683 rcu_read_unlock();
684
685 return ret;
686}
687
688int bpf_get_kallsym(unsigned int symnum, unsigned long *value, char *type,
689 char *sym)
690{
Daniel Borkmann74451e662017-02-16 22:24:50 +0100691 struct bpf_prog_aux *aux;
692 unsigned int it = 0;
693 int ret = -ERANGE;
694
695 if (!bpf_jit_kallsyms_enabled())
696 return ret;
697
698 rcu_read_lock();
699 list_for_each_entry_rcu(aux, &bpf_kallsyms, ksym_lnode) {
700 if (it++ != symnum)
701 continue;
702
Daniel Borkmann74451e662017-02-16 22:24:50 +0100703 bpf_get_prog_name(aux->prog, sym);
704
Song Liudf073472018-11-02 10:16:15 -0700705 *value = (unsigned long)aux->prog->bpf_func;
Daniel Borkmann74451e662017-02-16 22:24:50 +0100706 *type = BPF_SYM_ELF_TYPE;
707
708 ret = 0;
709 break;
710 }
711 rcu_read_unlock();
712
713 return ret;
714}
715
Daniel Borkmannede95a632018-10-23 01:11:04 +0200716static atomic_long_t bpf_jit_current;
717
Daniel Borkmannfdadd042018-12-11 12:14:12 +0100718/* Can be overridden by an arch's JIT compiler if it has a custom,
719 * dedicated BPF backend memory area, or if neither of the two
720 * below apply.
721 */
722u64 __weak bpf_jit_alloc_exec_limit(void)
723{
Daniel Borkmannede95a632018-10-23 01:11:04 +0200724#if defined(MODULES_VADDR)
Daniel Borkmannfdadd042018-12-11 12:14:12 +0100725 return MODULES_END - MODULES_VADDR;
726#else
727 return VMALLOC_END - VMALLOC_START;
728#endif
729}
730
Daniel Borkmannede95a632018-10-23 01:11:04 +0200731static int __init bpf_jit_charge_init(void)
732{
733 /* Only used as heuristic here to derive limit. */
Daniel Borkmannfdadd042018-12-11 12:14:12 +0100734 bpf_jit_limit = min_t(u64, round_up(bpf_jit_alloc_exec_limit() >> 2,
735 PAGE_SIZE), LONG_MAX);
Daniel Borkmannede95a632018-10-23 01:11:04 +0200736 return 0;
737}
738pure_initcall(bpf_jit_charge_init);
Daniel Borkmannede95a632018-10-23 01:11:04 +0200739
740static int bpf_jit_charge_modmem(u32 pages)
741{
742 if (atomic_long_add_return(pages, &bpf_jit_current) >
743 (bpf_jit_limit >> PAGE_SHIFT)) {
744 if (!capable(CAP_SYS_ADMIN)) {
745 atomic_long_sub(pages, &bpf_jit_current);
746 return -EPERM;
747 }
748 }
749
750 return 0;
751}
752
753static void bpf_jit_uncharge_modmem(u32 pages)
754{
755 atomic_long_sub(pages, &bpf_jit_current);
756}
757
Ard Biesheuveldc002bb2018-11-23 23:18:03 +0100758void *__weak bpf_jit_alloc_exec(unsigned long size)
759{
760 return module_alloc(size);
761}
762
763void __weak bpf_jit_free_exec(void *addr)
764{
765 module_memfree(addr);
766}
767
Daniel Borkmann738cbe72014-09-08 08:04:47 +0200768struct bpf_binary_header *
769bpf_jit_binary_alloc(unsigned int proglen, u8 **image_ptr,
770 unsigned int alignment,
771 bpf_jit_fill_hole_t bpf_fill_ill_insns)
772{
773 struct bpf_binary_header *hdr;
Daniel Borkmannede95a632018-10-23 01:11:04 +0200774 u32 size, hole, start, pages;
Daniel Borkmann738cbe72014-09-08 08:04:47 +0200775
776 /* Most of BPF filters are really small, but if some of them
777 * fill a page, allow at least 128 extra bytes to insert a
778 * random section of illegal instructions.
779 */
780 size = round_up(proglen + sizeof(*hdr) + 128, PAGE_SIZE);
Daniel Borkmannede95a632018-10-23 01:11:04 +0200781 pages = size / PAGE_SIZE;
782
783 if (bpf_jit_charge_modmem(pages))
Daniel Borkmann738cbe72014-09-08 08:04:47 +0200784 return NULL;
Ard Biesheuveldc002bb2018-11-23 23:18:03 +0100785 hdr = bpf_jit_alloc_exec(size);
Daniel Borkmannede95a632018-10-23 01:11:04 +0200786 if (!hdr) {
787 bpf_jit_uncharge_modmem(pages);
788 return NULL;
789 }
Daniel Borkmann738cbe72014-09-08 08:04:47 +0200790
791 /* Fill space with illegal/arch-dep instructions. */
792 bpf_fill_ill_insns(hdr, size);
793
Daniel Borkmannede95a632018-10-23 01:11:04 +0200794 hdr->pages = pages;
Daniel Borkmann738cbe72014-09-08 08:04:47 +0200795 hole = min_t(unsigned int, size - (proglen + sizeof(*hdr)),
796 PAGE_SIZE - sizeof(*hdr));
Daniel Borkmannb7552e1b2016-05-18 14:14:28 +0200797 start = (get_random_int() % hole) & ~(alignment - 1);
Daniel Borkmann738cbe72014-09-08 08:04:47 +0200798
799 /* Leave a random number of instructions before BPF code. */
800 *image_ptr = &hdr->image[start];
801
802 return hdr;
803}
804
805void bpf_jit_binary_free(struct bpf_binary_header *hdr)
806{
Daniel Borkmannede95a632018-10-23 01:11:04 +0200807 u32 pages = hdr->pages;
808
Ard Biesheuveldc002bb2018-11-23 23:18:03 +0100809 bpf_jit_free_exec(hdr);
Daniel Borkmannede95a632018-10-23 01:11:04 +0200810 bpf_jit_uncharge_modmem(pages);
Daniel Borkmann738cbe72014-09-08 08:04:47 +0200811}
Daniel Borkmann4f3446b2016-05-13 19:08:32 +0200812
Daniel Borkmann74451e662017-02-16 22:24:50 +0100813/* This symbol is only overridden by archs that have different
814 * requirements than the usual eBPF JITs, f.e. when they only
815 * implement cBPF JIT, do not set images read-only, etc.
816 */
817void __weak bpf_jit_free(struct bpf_prog *fp)
818{
819 if (fp->jited) {
820 struct bpf_binary_header *hdr = bpf_jit_binary_hdr(fp);
821
822 bpf_jit_binary_unlock_ro(hdr);
823 bpf_jit_binary_free(hdr);
824
825 WARN_ON_ONCE(!bpf_prog_kallsyms_verify_off(fp));
826 }
827
828 bpf_prog_unlock_free(fp);
829}
830
Daniel Borkmanne2c95a62018-11-26 14:05:38 +0100831int bpf_jit_get_func_addr(const struct bpf_prog *prog,
832 const struct bpf_insn *insn, bool extra_pass,
833 u64 *func_addr, bool *func_addr_fixed)
834{
835 s16 off = insn->off;
836 s32 imm = insn->imm;
837 u8 *addr;
838
839 *func_addr_fixed = insn->src_reg != BPF_PSEUDO_CALL;
840 if (!*func_addr_fixed) {
841 /* Place-holder address till the last pass has collected
842 * all addresses for JITed subprograms in which case we
843 * can pick them up from prog->aux.
844 */
845 if (!extra_pass)
846 addr = NULL;
847 else if (prog->aux->func &&
848 off >= 0 && off < prog->aux->func_cnt)
849 addr = (u8 *)prog->aux->func[off]->bpf_func;
850 else
851 return -EINVAL;
852 } else {
853 /* Address of a BPF helper call. Since part of the core
854 * kernel, it's always at a fixed location. __bpf_call_base
855 * and the helper with imm relative to it are both in core
856 * kernel.
857 */
858 addr = (u8 *)__bpf_call_base + imm;
859 }
860
861 *func_addr = (unsigned long)addr;
862 return 0;
863}
864
Daniel Borkmann4f3446b2016-05-13 19:08:32 +0200865static int bpf_jit_blind_insn(const struct bpf_insn *from,
866 const struct bpf_insn *aux,
867 struct bpf_insn *to_buff)
868{
869 struct bpf_insn *to = to_buff;
Daniel Borkmannb7552e1b2016-05-18 14:14:28 +0200870 u32 imm_rnd = get_random_int();
Daniel Borkmann4f3446b2016-05-13 19:08:32 +0200871 s16 off;
872
873 BUILD_BUG_ON(BPF_REG_AX + 1 != MAX_BPF_JIT_REG);
874 BUILD_BUG_ON(MAX_BPF_REG + 1 != MAX_BPF_JIT_REG);
875
Daniel Borkmann9b73bfd2019-01-03 00:58:29 +0100876 /* Constraints on AX register:
877 *
878 * AX register is inaccessible from user space. It is mapped in
879 * all JITs, and used here for constant blinding rewrites. It is
880 * typically "stateless" meaning its contents are only valid within
881 * the executed instruction, but not across several instructions.
882 * There are a few exceptions however which are further detailed
883 * below.
884 *
885 * Constant blinding is only used by JITs, not in the interpreter.
886 * The interpreter uses AX in some occasions as a local temporary
887 * register e.g. in DIV or MOD instructions.
888 *
889 * In restricted circumstances, the verifier can also use the AX
890 * register for rewrites as long as they do not interfere with
891 * the above cases!
892 */
893 if (from->dst_reg == BPF_REG_AX || from->src_reg == BPF_REG_AX)
894 goto out;
895
Daniel Borkmann4f3446b2016-05-13 19:08:32 +0200896 if (from->imm == 0 &&
897 (from->code == (BPF_ALU | BPF_MOV | BPF_K) ||
898 from->code == (BPF_ALU64 | BPF_MOV | BPF_K))) {
899 *to++ = BPF_ALU64_REG(BPF_XOR, from->dst_reg, from->dst_reg);
900 goto out;
901 }
902
903 switch (from->code) {
904 case BPF_ALU | BPF_ADD | BPF_K:
905 case BPF_ALU | BPF_SUB | BPF_K:
906 case BPF_ALU | BPF_AND | BPF_K:
907 case BPF_ALU | BPF_OR | BPF_K:
908 case BPF_ALU | BPF_XOR | BPF_K:
909 case BPF_ALU | BPF_MUL | BPF_K:
910 case BPF_ALU | BPF_MOV | BPF_K:
911 case BPF_ALU | BPF_DIV | BPF_K:
912 case BPF_ALU | BPF_MOD | BPF_K:
913 *to++ = BPF_ALU32_IMM(BPF_MOV, BPF_REG_AX, imm_rnd ^ from->imm);
914 *to++ = BPF_ALU32_IMM(BPF_XOR, BPF_REG_AX, imm_rnd);
915 *to++ = BPF_ALU32_REG(from->code, from->dst_reg, BPF_REG_AX);
916 break;
917
918 case BPF_ALU64 | BPF_ADD | BPF_K:
919 case BPF_ALU64 | BPF_SUB | BPF_K:
920 case BPF_ALU64 | BPF_AND | BPF_K:
921 case BPF_ALU64 | BPF_OR | BPF_K:
922 case BPF_ALU64 | BPF_XOR | BPF_K:
923 case BPF_ALU64 | BPF_MUL | BPF_K:
924 case BPF_ALU64 | BPF_MOV | BPF_K:
925 case BPF_ALU64 | BPF_DIV | BPF_K:
926 case BPF_ALU64 | BPF_MOD | BPF_K:
927 *to++ = BPF_ALU64_IMM(BPF_MOV, BPF_REG_AX, imm_rnd ^ from->imm);
928 *to++ = BPF_ALU64_IMM(BPF_XOR, BPF_REG_AX, imm_rnd);
929 *to++ = BPF_ALU64_REG(from->code, from->dst_reg, BPF_REG_AX);
930 break;
931
932 case BPF_JMP | BPF_JEQ | BPF_K:
933 case BPF_JMP | BPF_JNE | BPF_K:
934 case BPF_JMP | BPF_JGT | BPF_K:
Daniel Borkmann92b31a92017-08-10 01:39:55 +0200935 case BPF_JMP | BPF_JLT | BPF_K:
Daniel Borkmann4f3446b2016-05-13 19:08:32 +0200936 case BPF_JMP | BPF_JGE | BPF_K:
Daniel Borkmann92b31a92017-08-10 01:39:55 +0200937 case BPF_JMP | BPF_JLE | BPF_K:
Daniel Borkmann4f3446b2016-05-13 19:08:32 +0200938 case BPF_JMP | BPF_JSGT | BPF_K:
Daniel Borkmann92b31a92017-08-10 01:39:55 +0200939 case BPF_JMP | BPF_JSLT | BPF_K:
Daniel Borkmann4f3446b2016-05-13 19:08:32 +0200940 case BPF_JMP | BPF_JSGE | BPF_K:
Daniel Borkmann92b31a92017-08-10 01:39:55 +0200941 case BPF_JMP | BPF_JSLE | BPF_K:
Daniel Borkmann4f3446b2016-05-13 19:08:32 +0200942 case BPF_JMP | BPF_JSET | BPF_K:
943 /* Accommodate for extra offset in case of a backjump. */
944 off = from->off;
945 if (off < 0)
946 off -= 2;
947 *to++ = BPF_ALU64_IMM(BPF_MOV, BPF_REG_AX, imm_rnd ^ from->imm);
948 *to++ = BPF_ALU64_IMM(BPF_XOR, BPF_REG_AX, imm_rnd);
949 *to++ = BPF_JMP_REG(from->code, from->dst_reg, BPF_REG_AX, off);
950 break;
951
Jiong Wanga7b76c82019-01-26 12:26:05 -0500952 case BPF_JMP32 | BPF_JEQ | BPF_K:
953 case BPF_JMP32 | BPF_JNE | BPF_K:
954 case BPF_JMP32 | BPF_JGT | BPF_K:
955 case BPF_JMP32 | BPF_JLT | BPF_K:
956 case BPF_JMP32 | BPF_JGE | BPF_K:
957 case BPF_JMP32 | BPF_JLE | BPF_K:
958 case BPF_JMP32 | BPF_JSGT | BPF_K:
959 case BPF_JMP32 | BPF_JSLT | BPF_K:
960 case BPF_JMP32 | BPF_JSGE | BPF_K:
961 case BPF_JMP32 | BPF_JSLE | BPF_K:
962 case BPF_JMP32 | BPF_JSET | BPF_K:
963 /* Accommodate for extra offset in case of a backjump. */
964 off = from->off;
965 if (off < 0)
966 off -= 2;
967 *to++ = BPF_ALU32_IMM(BPF_MOV, BPF_REG_AX, imm_rnd ^ from->imm);
968 *to++ = BPF_ALU32_IMM(BPF_XOR, BPF_REG_AX, imm_rnd);
969 *to++ = BPF_JMP32_REG(from->code, from->dst_reg, BPF_REG_AX,
970 off);
971 break;
972
Daniel Borkmann4f3446b2016-05-13 19:08:32 +0200973 case BPF_LD | BPF_IMM | BPF_DW:
974 *to++ = BPF_ALU64_IMM(BPF_MOV, BPF_REG_AX, imm_rnd ^ aux[1].imm);
975 *to++ = BPF_ALU64_IMM(BPF_XOR, BPF_REG_AX, imm_rnd);
976 *to++ = BPF_ALU64_IMM(BPF_LSH, BPF_REG_AX, 32);
977 *to++ = BPF_ALU64_REG(BPF_MOV, aux[0].dst_reg, BPF_REG_AX);
978 break;
979 case 0: /* Part 2 of BPF_LD | BPF_IMM | BPF_DW. */
980 *to++ = BPF_ALU32_IMM(BPF_MOV, BPF_REG_AX, imm_rnd ^ aux[0].imm);
981 *to++ = BPF_ALU32_IMM(BPF_XOR, BPF_REG_AX, imm_rnd);
982 *to++ = BPF_ALU64_REG(BPF_OR, aux[0].dst_reg, BPF_REG_AX);
983 break;
984
985 case BPF_ST | BPF_MEM | BPF_DW:
986 case BPF_ST | BPF_MEM | BPF_W:
987 case BPF_ST | BPF_MEM | BPF_H:
988 case BPF_ST | BPF_MEM | BPF_B:
989 *to++ = BPF_ALU64_IMM(BPF_MOV, BPF_REG_AX, imm_rnd ^ from->imm);
990 *to++ = BPF_ALU64_IMM(BPF_XOR, BPF_REG_AX, imm_rnd);
991 *to++ = BPF_STX_MEM(from->code, from->dst_reg, BPF_REG_AX, from->off);
992 break;
993 }
994out:
995 return to - to_buff;
996}
997
998static struct bpf_prog *bpf_prog_clone_create(struct bpf_prog *fp_other,
999 gfp_t gfp_extra_flags)
1000{
Michal Hocko19809c22017-05-08 15:57:44 -07001001 gfp_t gfp_flags = GFP_KERNEL | __GFP_ZERO | gfp_extra_flags;
Daniel Borkmann4f3446b2016-05-13 19:08:32 +02001002 struct bpf_prog *fp;
1003
1004 fp = __vmalloc(fp_other->pages * PAGE_SIZE, gfp_flags, PAGE_KERNEL);
1005 if (fp != NULL) {
Daniel Borkmann4f3446b2016-05-13 19:08:32 +02001006 /* aux->prog still points to the fp_other one, so
1007 * when promoting the clone to the real program,
1008 * this still needs to be adapted.
1009 */
1010 memcpy(fp, fp_other, fp_other->pages * PAGE_SIZE);
1011 }
1012
1013 return fp;
1014}
1015
1016static void bpf_prog_clone_free(struct bpf_prog *fp)
1017{
1018 /* aux was stolen by the other clone, so we cannot free
1019 * it from this path! It will be freed eventually by the
1020 * other program on release.
1021 *
1022 * At this point, we don't need a deferred release since
1023 * clone is guaranteed to not be locked.
1024 */
1025 fp->aux = NULL;
1026 __bpf_prog_free(fp);
1027}
1028
1029void bpf_jit_prog_release_other(struct bpf_prog *fp, struct bpf_prog *fp_other)
1030{
1031 /* We have to repoint aux->prog to self, as we don't
1032 * know whether fp here is the clone or the original.
1033 */
1034 fp->aux->prog = fp;
1035 bpf_prog_clone_free(fp_other);
1036}
1037
1038struct bpf_prog *bpf_jit_blind_constants(struct bpf_prog *prog)
1039{
1040 struct bpf_insn insn_buff[16], aux[2];
1041 struct bpf_prog *clone, *tmp;
1042 int insn_delta, insn_cnt;
1043 struct bpf_insn *insn;
1044 int i, rewritten;
1045
Alexei Starovoitov1c2a0882017-12-14 17:55:15 -08001046 if (!bpf_jit_blinding_enabled(prog) || prog->blinded)
Daniel Borkmann4f3446b2016-05-13 19:08:32 +02001047 return prog;
1048
1049 clone = bpf_prog_clone_create(prog, GFP_USER);
1050 if (!clone)
1051 return ERR_PTR(-ENOMEM);
1052
1053 insn_cnt = clone->len;
1054 insn = clone->insnsi;
1055
1056 for (i = 0; i < insn_cnt; i++, insn++) {
1057 /* We temporarily need to hold the original ld64 insn
1058 * so that we can still access the first part in the
1059 * second blinding run.
1060 */
1061 if (insn[0].code == (BPF_LD | BPF_IMM | BPF_DW) &&
1062 insn[1].code == 0)
1063 memcpy(aux, insn, sizeof(aux));
1064
1065 rewritten = bpf_jit_blind_insn(insn, aux, insn_buff);
1066 if (!rewritten)
1067 continue;
1068
1069 tmp = bpf_patch_insn_single(clone, i, insn_buff, rewritten);
1070 if (!tmp) {
1071 /* Patching may have repointed aux->prog during
1072 * realloc from the original one, so we need to
1073 * fix it up here on error.
1074 */
1075 bpf_jit_prog_release_other(prog, clone);
1076 return ERR_PTR(-ENOMEM);
1077 }
1078
1079 clone = tmp;
1080 insn_delta = rewritten - 1;
1081
1082 /* Walk new program and skip insns we just inserted. */
1083 insn = clone->insnsi + i + insn_delta;
1084 insn_cnt += insn_delta;
1085 i += insn_delta;
1086 }
1087
Alexei Starovoitov1c2a0882017-12-14 17:55:15 -08001088 clone->blinded = 1;
Daniel Borkmann4f3446b2016-05-13 19:08:32 +02001089 return clone;
1090}
Daniel Borkmannb954d832014-09-10 15:01:02 +02001091#endif /* CONFIG_BPF_JIT */
Daniel Borkmann738cbe72014-09-08 08:04:47 +02001092
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001093/* Base function for offset calculation. Needs to go into .text section,
1094 * therefore keeping it non-static as well; will also be used by JITs
Daniel Borkmann7105e822017-12-20 13:42:57 +01001095 * anyway later on, so do not let the compiler omit it. This also needs
1096 * to go into kallsyms for correlation from e.g. bpftool, so naming
1097 * must not change.
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001098 */
1099noinline u64 __bpf_call_base(u64 r1, u64 r2, u64 r3, u64 r4, u64 r5)
1100{
1101 return 0;
1102}
Alexei Starovoitov4d9c5c52015-07-20 20:34:19 -07001103EXPORT_SYMBOL_GPL(__bpf_call_base);
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001104
Daniel Borkmann5e581da2018-01-26 23:33:38 +01001105/* All UAPI available opcodes. */
1106#define BPF_INSN_MAP(INSN_2, INSN_3) \
1107 /* 32 bit ALU operations. */ \
1108 /* Register based. */ \
Jiong Wang2dc6b102018-12-05 13:52:34 -05001109 INSN_3(ALU, ADD, X), \
1110 INSN_3(ALU, SUB, X), \
1111 INSN_3(ALU, AND, X), \
1112 INSN_3(ALU, OR, X), \
1113 INSN_3(ALU, LSH, X), \
1114 INSN_3(ALU, RSH, X), \
1115 INSN_3(ALU, XOR, X), \
1116 INSN_3(ALU, MUL, X), \
1117 INSN_3(ALU, MOV, X), \
1118 INSN_3(ALU, ARSH, X), \
1119 INSN_3(ALU, DIV, X), \
1120 INSN_3(ALU, MOD, X), \
Daniel Borkmann5e581da2018-01-26 23:33:38 +01001121 INSN_2(ALU, NEG), \
1122 INSN_3(ALU, END, TO_BE), \
1123 INSN_3(ALU, END, TO_LE), \
1124 /* Immediate based. */ \
Jiong Wang2dc6b102018-12-05 13:52:34 -05001125 INSN_3(ALU, ADD, K), \
1126 INSN_3(ALU, SUB, K), \
1127 INSN_3(ALU, AND, K), \
1128 INSN_3(ALU, OR, K), \
1129 INSN_3(ALU, LSH, K), \
1130 INSN_3(ALU, RSH, K), \
1131 INSN_3(ALU, XOR, K), \
1132 INSN_3(ALU, MUL, K), \
1133 INSN_3(ALU, MOV, K), \
1134 INSN_3(ALU, ARSH, K), \
1135 INSN_3(ALU, DIV, K), \
1136 INSN_3(ALU, MOD, K), \
Daniel Borkmann5e581da2018-01-26 23:33:38 +01001137 /* 64 bit ALU operations. */ \
1138 /* Register based. */ \
1139 INSN_3(ALU64, ADD, X), \
1140 INSN_3(ALU64, SUB, X), \
1141 INSN_3(ALU64, AND, X), \
1142 INSN_3(ALU64, OR, X), \
1143 INSN_3(ALU64, LSH, X), \
1144 INSN_3(ALU64, RSH, X), \
1145 INSN_3(ALU64, XOR, X), \
1146 INSN_3(ALU64, MUL, X), \
1147 INSN_3(ALU64, MOV, X), \
1148 INSN_3(ALU64, ARSH, X), \
1149 INSN_3(ALU64, DIV, X), \
1150 INSN_3(ALU64, MOD, X), \
1151 INSN_2(ALU64, NEG), \
1152 /* Immediate based. */ \
1153 INSN_3(ALU64, ADD, K), \
1154 INSN_3(ALU64, SUB, K), \
1155 INSN_3(ALU64, AND, K), \
1156 INSN_3(ALU64, OR, K), \
1157 INSN_3(ALU64, LSH, K), \
1158 INSN_3(ALU64, RSH, K), \
1159 INSN_3(ALU64, XOR, K), \
1160 INSN_3(ALU64, MUL, K), \
1161 INSN_3(ALU64, MOV, K), \
1162 INSN_3(ALU64, ARSH, K), \
1163 INSN_3(ALU64, DIV, K), \
1164 INSN_3(ALU64, MOD, K), \
1165 /* Call instruction. */ \
1166 INSN_2(JMP, CALL), \
1167 /* Exit instruction. */ \
1168 INSN_2(JMP, EXIT), \
Jiong Wang503a8862019-01-26 12:26:04 -05001169 /* 32-bit Jump instructions. */ \
1170 /* Register based. */ \
1171 INSN_3(JMP32, JEQ, X), \
1172 INSN_3(JMP32, JNE, X), \
1173 INSN_3(JMP32, JGT, X), \
1174 INSN_3(JMP32, JLT, X), \
1175 INSN_3(JMP32, JGE, X), \
1176 INSN_3(JMP32, JLE, X), \
1177 INSN_3(JMP32, JSGT, X), \
1178 INSN_3(JMP32, JSLT, X), \
1179 INSN_3(JMP32, JSGE, X), \
1180 INSN_3(JMP32, JSLE, X), \
1181 INSN_3(JMP32, JSET, X), \
1182 /* Immediate based. */ \
1183 INSN_3(JMP32, JEQ, K), \
1184 INSN_3(JMP32, JNE, K), \
1185 INSN_3(JMP32, JGT, K), \
1186 INSN_3(JMP32, JLT, K), \
1187 INSN_3(JMP32, JGE, K), \
1188 INSN_3(JMP32, JLE, K), \
1189 INSN_3(JMP32, JSGT, K), \
1190 INSN_3(JMP32, JSLT, K), \
1191 INSN_3(JMP32, JSGE, K), \
1192 INSN_3(JMP32, JSLE, K), \
1193 INSN_3(JMP32, JSET, K), \
Daniel Borkmann5e581da2018-01-26 23:33:38 +01001194 /* Jump instructions. */ \
1195 /* Register based. */ \
1196 INSN_3(JMP, JEQ, X), \
1197 INSN_3(JMP, JNE, X), \
1198 INSN_3(JMP, JGT, X), \
1199 INSN_3(JMP, JLT, X), \
1200 INSN_3(JMP, JGE, X), \
1201 INSN_3(JMP, JLE, X), \
1202 INSN_3(JMP, JSGT, X), \
1203 INSN_3(JMP, JSLT, X), \
1204 INSN_3(JMP, JSGE, X), \
1205 INSN_3(JMP, JSLE, X), \
1206 INSN_3(JMP, JSET, X), \
1207 /* Immediate based. */ \
1208 INSN_3(JMP, JEQ, K), \
1209 INSN_3(JMP, JNE, K), \
1210 INSN_3(JMP, JGT, K), \
1211 INSN_3(JMP, JLT, K), \
1212 INSN_3(JMP, JGE, K), \
1213 INSN_3(JMP, JLE, K), \
1214 INSN_3(JMP, JSGT, K), \
1215 INSN_3(JMP, JSLT, K), \
1216 INSN_3(JMP, JSGE, K), \
1217 INSN_3(JMP, JSLE, K), \
1218 INSN_3(JMP, JSET, K), \
1219 INSN_2(JMP, JA), \
1220 /* Store instructions. */ \
1221 /* Register based. */ \
1222 INSN_3(STX, MEM, B), \
1223 INSN_3(STX, MEM, H), \
1224 INSN_3(STX, MEM, W), \
1225 INSN_3(STX, MEM, DW), \
1226 INSN_3(STX, XADD, W), \
1227 INSN_3(STX, XADD, DW), \
1228 /* Immediate based. */ \
1229 INSN_3(ST, MEM, B), \
1230 INSN_3(ST, MEM, H), \
1231 INSN_3(ST, MEM, W), \
1232 INSN_3(ST, MEM, DW), \
1233 /* Load instructions. */ \
1234 /* Register based. */ \
1235 INSN_3(LDX, MEM, B), \
1236 INSN_3(LDX, MEM, H), \
1237 INSN_3(LDX, MEM, W), \
1238 INSN_3(LDX, MEM, DW), \
1239 /* Immediate based. */ \
Daniel Borkmanne0cea7c2018-05-04 01:08:14 +02001240 INSN_3(LD, IMM, DW)
Daniel Borkmann5e581da2018-01-26 23:33:38 +01001241
1242bool bpf_opcode_in_insntable(u8 code)
1243{
1244#define BPF_INSN_2_TBL(x, y) [BPF_##x | BPF_##y] = true
1245#define BPF_INSN_3_TBL(x, y, z) [BPF_##x | BPF_##y | BPF_##z] = true
1246 static const bool public_insntable[256] = {
1247 [0 ... 255] = false,
1248 /* Now overwrite non-defaults ... */
1249 BPF_INSN_MAP(BPF_INSN_2_TBL, BPF_INSN_3_TBL),
Daniel Borkmanne0cea7c2018-05-04 01:08:14 +02001250 /* UAPI exposed, but rewritten opcodes. cBPF carry-over. */
1251 [BPF_LD | BPF_ABS | BPF_B] = true,
1252 [BPF_LD | BPF_ABS | BPF_H] = true,
1253 [BPF_LD | BPF_ABS | BPF_W] = true,
1254 [BPF_LD | BPF_IND | BPF_B] = true,
1255 [BPF_LD | BPF_IND | BPF_H] = true,
1256 [BPF_LD | BPF_IND | BPF_W] = true,
Daniel Borkmann5e581da2018-01-26 23:33:38 +01001257 };
1258#undef BPF_INSN_3_TBL
1259#undef BPF_INSN_2_TBL
1260 return public_insntable[code];
1261}
1262
Alexei Starovoitov290af862018-01-09 10:04:29 -08001263#ifndef CONFIG_BPF_JIT_ALWAYS_ON
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001264/**
Alexei Starovoitov7ae457c2014-07-30 20:34:16 -07001265 * __bpf_prog_run - run eBPF program on a given context
Valdis Kletnieksde1da682019-01-28 23:04:46 -05001266 * @regs: is the array of MAX_BPF_EXT_REG eBPF pseudo-registers
Alexei Starovoitov7ae457c2014-07-30 20:34:16 -07001267 * @insn: is the array of eBPF instructions
Valdis Kletnieksde1da682019-01-28 23:04:46 -05001268 * @stack: is the eBPF storage stack
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001269 *
Alexei Starovoitov7ae457c2014-07-30 20:34:16 -07001270 * Decode and execute eBPF instructions.
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001271 */
Alexei Starovoitov1ea47e02017-12-14 17:55:13 -08001272static u64 ___bpf_prog_run(u64 *regs, const struct bpf_insn *insn, u64 *stack)
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001273{
Daniel Borkmann5e581da2018-01-26 23:33:38 +01001274#define BPF_INSN_2_LBL(x, y) [BPF_##x | BPF_##y] = &&x##_##y
1275#define BPF_INSN_3_LBL(x, y, z) [BPF_##x | BPF_##y | BPF_##z] = &&x##_##y##_##z
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001276 static const void *jumptable[256] = {
1277 [0 ... 255] = &&default_label,
1278 /* Now overwrite non-defaults ... */
Daniel Borkmann5e581da2018-01-26 23:33:38 +01001279 BPF_INSN_MAP(BPF_INSN_2_LBL, BPF_INSN_3_LBL),
1280 /* Non-UAPI available opcodes. */
Alexei Starovoitov1ea47e02017-12-14 17:55:13 -08001281 [BPF_JMP | BPF_CALL_ARGS] = &&JMP_CALL_ARGS,
Alexei Starovoitov71189fa2017-05-30 13:31:27 -07001282 [BPF_JMP | BPF_TAIL_CALL] = &&JMP_TAIL_CALL,
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001283 };
Daniel Borkmann5e581da2018-01-26 23:33:38 +01001284#undef BPF_INSN_3_LBL
1285#undef BPF_INSN_2_LBL
Alexei Starovoitov04fd61ab2015-05-19 16:59:03 -07001286 u32 tail_call_cnt = 0;
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001287
1288#define CONT ({ insn++; goto select_insn; })
1289#define CONT_JMP ({ insn++; goto select_insn; })
1290
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001291select_insn:
1292 goto *jumptable[insn->code];
1293
1294 /* ALU */
1295#define ALU(OPCODE, OP) \
1296 ALU64_##OPCODE##_X: \
1297 DST = DST OP SRC; \
1298 CONT; \
1299 ALU_##OPCODE##_X: \
1300 DST = (u32) DST OP (u32) SRC; \
1301 CONT; \
1302 ALU64_##OPCODE##_K: \
1303 DST = DST OP IMM; \
1304 CONT; \
1305 ALU_##OPCODE##_K: \
1306 DST = (u32) DST OP (u32) IMM; \
1307 CONT;
1308
1309 ALU(ADD, +)
1310 ALU(SUB, -)
1311 ALU(AND, &)
1312 ALU(OR, |)
1313 ALU(LSH, <<)
1314 ALU(RSH, >>)
1315 ALU(XOR, ^)
1316 ALU(MUL, *)
1317#undef ALU
1318 ALU_NEG:
1319 DST = (u32) -DST;
1320 CONT;
1321 ALU64_NEG:
1322 DST = -DST;
1323 CONT;
1324 ALU_MOV_X:
1325 DST = (u32) SRC;
1326 CONT;
1327 ALU_MOV_K:
1328 DST = (u32) IMM;
1329 CONT;
1330 ALU64_MOV_X:
1331 DST = SRC;
1332 CONT;
1333 ALU64_MOV_K:
1334 DST = IMM;
1335 CONT;
Alexei Starovoitov02ab6952014-09-04 22:17:17 -07001336 LD_IMM_DW:
1337 DST = (u64) (u32) insn[0].imm | ((u64) (u32) insn[1].imm) << 32;
1338 insn++;
1339 CONT;
Jiong Wang2dc6b102018-12-05 13:52:34 -05001340 ALU_ARSH_X:
1341 DST = (u64) (u32) ((*(s32 *) &DST) >> SRC);
1342 CONT;
1343 ALU_ARSH_K:
1344 DST = (u64) (u32) ((*(s32 *) &DST) >> IMM);
1345 CONT;
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001346 ALU64_ARSH_X:
1347 (*(s64 *) &DST) >>= SRC;
1348 CONT;
1349 ALU64_ARSH_K:
1350 (*(s64 *) &DST) >>= IMM;
1351 CONT;
1352 ALU64_MOD_X:
Daniel Borkmann144cd912019-01-03 00:58:28 +01001353 div64_u64_rem(DST, SRC, &AX);
1354 DST = AX;
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001355 CONT;
1356 ALU_MOD_X:
Daniel Borkmann144cd912019-01-03 00:58:28 +01001357 AX = (u32) DST;
1358 DST = do_div(AX, (u32) SRC);
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001359 CONT;
1360 ALU64_MOD_K:
Daniel Borkmann144cd912019-01-03 00:58:28 +01001361 div64_u64_rem(DST, IMM, &AX);
1362 DST = AX;
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001363 CONT;
1364 ALU_MOD_K:
Daniel Borkmann144cd912019-01-03 00:58:28 +01001365 AX = (u32) DST;
1366 DST = do_div(AX, (u32) IMM);
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001367 CONT;
1368 ALU64_DIV_X:
Alexei Starovoitov876a7ae2015-04-27 14:40:37 -07001369 DST = div64_u64(DST, SRC);
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001370 CONT;
1371 ALU_DIV_X:
Daniel Borkmann144cd912019-01-03 00:58:28 +01001372 AX = (u32) DST;
1373 do_div(AX, (u32) SRC);
1374 DST = (u32) AX;
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001375 CONT;
1376 ALU64_DIV_K:
Alexei Starovoitov876a7ae2015-04-27 14:40:37 -07001377 DST = div64_u64(DST, IMM);
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001378 CONT;
1379 ALU_DIV_K:
Daniel Borkmann144cd912019-01-03 00:58:28 +01001380 AX = (u32) DST;
1381 do_div(AX, (u32) IMM);
1382 DST = (u32) AX;
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001383 CONT;
1384 ALU_END_TO_BE:
1385 switch (IMM) {
1386 case 16:
1387 DST = (__force u16) cpu_to_be16(DST);
1388 break;
1389 case 32:
1390 DST = (__force u32) cpu_to_be32(DST);
1391 break;
1392 case 64:
1393 DST = (__force u64) cpu_to_be64(DST);
1394 break;
1395 }
1396 CONT;
1397 ALU_END_TO_LE:
1398 switch (IMM) {
1399 case 16:
1400 DST = (__force u16) cpu_to_le16(DST);
1401 break;
1402 case 32:
1403 DST = (__force u32) cpu_to_le32(DST);
1404 break;
1405 case 64:
1406 DST = (__force u64) cpu_to_le64(DST);
1407 break;
1408 }
1409 CONT;
1410
1411 /* CALL */
1412 JMP_CALL:
1413 /* Function call scratches BPF_R1-BPF_R5 registers,
1414 * preserves BPF_R6-BPF_R9, and stores return value
1415 * into BPF_R0.
1416 */
1417 BPF_R0 = (__bpf_call_base + insn->imm)(BPF_R1, BPF_R2, BPF_R3,
1418 BPF_R4, BPF_R5);
1419 CONT;
1420
Alexei Starovoitov1ea47e02017-12-14 17:55:13 -08001421 JMP_CALL_ARGS:
1422 BPF_R0 = (__bpf_call_base_args + insn->imm)(BPF_R1, BPF_R2,
1423 BPF_R3, BPF_R4,
1424 BPF_R5,
1425 insn + insn->off + 1);
1426 CONT;
1427
Alexei Starovoitov04fd61ab2015-05-19 16:59:03 -07001428 JMP_TAIL_CALL: {
1429 struct bpf_map *map = (struct bpf_map *) (unsigned long) BPF_R2;
1430 struct bpf_array *array = container_of(map, struct bpf_array, map);
1431 struct bpf_prog *prog;
Alexei Starovoitov90caccd2017-10-03 15:37:20 -07001432 u32 index = BPF_R3;
Alexei Starovoitov04fd61ab2015-05-19 16:59:03 -07001433
1434 if (unlikely(index >= array->map.max_entries))
1435 goto out;
Alexei Starovoitov04fd61ab2015-05-19 16:59:03 -07001436 if (unlikely(tail_call_cnt > MAX_TAIL_CALL_CNT))
1437 goto out;
1438
1439 tail_call_cnt++;
1440
Wang Nan2a36f0b2015-08-06 07:02:33 +00001441 prog = READ_ONCE(array->ptrs[index]);
Daniel Borkmann1ca1cc92016-06-28 12:18:23 +02001442 if (!prog)
Alexei Starovoitov04fd61ab2015-05-19 16:59:03 -07001443 goto out;
1444
Daniel Borkmannc4675f92015-07-13 20:49:32 +02001445 /* ARG1 at this point is guaranteed to point to CTX from
1446 * the verifier side due to the fact that the tail call is
1447 * handeled like a helper, that is, bpf_tail_call_proto,
1448 * where arg1_type is ARG_PTR_TO_CTX.
1449 */
Alexei Starovoitov04fd61ab2015-05-19 16:59:03 -07001450 insn = prog->insnsi;
1451 goto select_insn;
1452out:
1453 CONT;
1454 }
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001455 JMP_JA:
1456 insn += insn->off;
1457 CONT;
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001458 JMP_EXIT:
1459 return BPF_R0;
Jiong Wang503a8862019-01-26 12:26:04 -05001460 /* JMP */
1461#define COND_JMP(SIGN, OPCODE, CMP_OP) \
1462 JMP_##OPCODE##_X: \
1463 if ((SIGN##64) DST CMP_OP (SIGN##64) SRC) { \
1464 insn += insn->off; \
1465 CONT_JMP; \
1466 } \
1467 CONT; \
1468 JMP32_##OPCODE##_X: \
1469 if ((SIGN##32) DST CMP_OP (SIGN##32) SRC) { \
1470 insn += insn->off; \
1471 CONT_JMP; \
1472 } \
1473 CONT; \
1474 JMP_##OPCODE##_K: \
1475 if ((SIGN##64) DST CMP_OP (SIGN##64) IMM) { \
1476 insn += insn->off; \
1477 CONT_JMP; \
1478 } \
1479 CONT; \
1480 JMP32_##OPCODE##_K: \
1481 if ((SIGN##32) DST CMP_OP (SIGN##32) IMM) { \
1482 insn += insn->off; \
1483 CONT_JMP; \
1484 } \
1485 CONT;
1486 COND_JMP(u, JEQ, ==)
1487 COND_JMP(u, JNE, !=)
1488 COND_JMP(u, JGT, >)
1489 COND_JMP(u, JLT, <)
1490 COND_JMP(u, JGE, >=)
1491 COND_JMP(u, JLE, <=)
1492 COND_JMP(u, JSET, &)
1493 COND_JMP(s, JSGT, >)
1494 COND_JMP(s, JSLT, <)
1495 COND_JMP(s, JSGE, >=)
1496 COND_JMP(s, JSLE, <=)
1497#undef COND_JMP
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001498 /* STX and ST and LDX*/
1499#define LDST(SIZEOP, SIZE) \
1500 STX_MEM_##SIZEOP: \
1501 *(SIZE *)(unsigned long) (DST + insn->off) = SRC; \
1502 CONT; \
1503 ST_MEM_##SIZEOP: \
1504 *(SIZE *)(unsigned long) (DST + insn->off) = IMM; \
1505 CONT; \
1506 LDX_MEM_##SIZEOP: \
1507 DST = *(SIZE *)(unsigned long) (SRC + insn->off); \
1508 CONT;
1509
1510 LDST(B, u8)
1511 LDST(H, u16)
1512 LDST(W, u32)
1513 LDST(DW, u64)
1514#undef LDST
1515 STX_XADD_W: /* lock xadd *(u32 *)(dst_reg + off16) += src_reg */
1516 atomic_add((u32) SRC, (atomic_t *)(unsigned long)
1517 (DST + insn->off));
1518 CONT;
1519 STX_XADD_DW: /* lock xadd *(u64 *)(dst_reg + off16) += src_reg */
1520 atomic64_add((u64) SRC, (atomic64_t *)(unsigned long)
1521 (DST + insn->off));
1522 CONT;
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001523
1524 default_label:
Daniel Borkmann5e581da2018-01-26 23:33:38 +01001525 /* If we ever reach this, we have a bug somewhere. Die hard here
1526 * instead of just returning 0; we could be somewhere in a subprog,
1527 * so execution could continue otherwise which we do /not/ want.
1528 *
1529 * Note, verifier whitelists all opcodes in bpf_opcode_in_insntable().
1530 */
1531 pr_warn("BPF interpreter: unknown opcode %02x\n", insn->code);
1532 BUG_ON(1);
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001533 return 0;
1534}
Alexei Starovoitovf696b8f2017-05-30 13:31:28 -07001535STACK_FRAME_NON_STANDARD(___bpf_prog_run); /* jump table */
1536
Alexei Starovoitovb870aa92017-05-30 13:31:33 -07001537#define PROG_NAME(stack_size) __bpf_prog_run##stack_size
1538#define DEFINE_BPF_PROG_RUN(stack_size) \
1539static unsigned int PROG_NAME(stack_size)(const void *ctx, const struct bpf_insn *insn) \
1540{ \
1541 u64 stack[stack_size / sizeof(u64)]; \
Daniel Borkmann144cd912019-01-03 00:58:28 +01001542 u64 regs[MAX_BPF_EXT_REG]; \
Alexei Starovoitovb870aa92017-05-30 13:31:33 -07001543\
1544 FP = (u64) (unsigned long) &stack[ARRAY_SIZE(stack)]; \
1545 ARG1 = (u64) (unsigned long) ctx; \
1546 return ___bpf_prog_run(regs, insn, stack); \
Alexei Starovoitovf696b8f2017-05-30 13:31:28 -07001547}
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001548
Alexei Starovoitov1ea47e02017-12-14 17:55:13 -08001549#define PROG_NAME_ARGS(stack_size) __bpf_prog_run_args##stack_size
1550#define DEFINE_BPF_PROG_RUN_ARGS(stack_size) \
1551static u64 PROG_NAME_ARGS(stack_size)(u64 r1, u64 r2, u64 r3, u64 r4, u64 r5, \
1552 const struct bpf_insn *insn) \
1553{ \
1554 u64 stack[stack_size / sizeof(u64)]; \
Daniel Borkmann144cd912019-01-03 00:58:28 +01001555 u64 regs[MAX_BPF_EXT_REG]; \
Alexei Starovoitov1ea47e02017-12-14 17:55:13 -08001556\
1557 FP = (u64) (unsigned long) &stack[ARRAY_SIZE(stack)]; \
1558 BPF_R1 = r1; \
1559 BPF_R2 = r2; \
1560 BPF_R3 = r3; \
1561 BPF_R4 = r4; \
1562 BPF_R5 = r5; \
1563 return ___bpf_prog_run(regs, insn, stack); \
1564}
1565
Alexei Starovoitovb870aa92017-05-30 13:31:33 -07001566#define EVAL1(FN, X) FN(X)
1567#define EVAL2(FN, X, Y...) FN(X) EVAL1(FN, Y)
1568#define EVAL3(FN, X, Y...) FN(X) EVAL2(FN, Y)
1569#define EVAL4(FN, X, Y...) FN(X) EVAL3(FN, Y)
1570#define EVAL5(FN, X, Y...) FN(X) EVAL4(FN, Y)
1571#define EVAL6(FN, X, Y...) FN(X) EVAL5(FN, Y)
1572
1573EVAL6(DEFINE_BPF_PROG_RUN, 32, 64, 96, 128, 160, 192);
1574EVAL6(DEFINE_BPF_PROG_RUN, 224, 256, 288, 320, 352, 384);
1575EVAL4(DEFINE_BPF_PROG_RUN, 416, 448, 480, 512);
1576
Alexei Starovoitov1ea47e02017-12-14 17:55:13 -08001577EVAL6(DEFINE_BPF_PROG_RUN_ARGS, 32, 64, 96, 128, 160, 192);
1578EVAL6(DEFINE_BPF_PROG_RUN_ARGS, 224, 256, 288, 320, 352, 384);
1579EVAL4(DEFINE_BPF_PROG_RUN_ARGS, 416, 448, 480, 512);
1580
Alexei Starovoitovb870aa92017-05-30 13:31:33 -07001581#define PROG_NAME_LIST(stack_size) PROG_NAME(stack_size),
1582
1583static unsigned int (*interpreters[])(const void *ctx,
1584 const struct bpf_insn *insn) = {
1585EVAL6(PROG_NAME_LIST, 32, 64, 96, 128, 160, 192)
1586EVAL6(PROG_NAME_LIST, 224, 256, 288, 320, 352, 384)
1587EVAL4(PROG_NAME_LIST, 416, 448, 480, 512)
1588};
Alexei Starovoitov1ea47e02017-12-14 17:55:13 -08001589#undef PROG_NAME_LIST
1590#define PROG_NAME_LIST(stack_size) PROG_NAME_ARGS(stack_size),
1591static u64 (*interpreters_args[])(u64 r1, u64 r2, u64 r3, u64 r4, u64 r5,
1592 const struct bpf_insn *insn) = {
1593EVAL6(PROG_NAME_LIST, 32, 64, 96, 128, 160, 192)
1594EVAL6(PROG_NAME_LIST, 224, 256, 288, 320, 352, 384)
1595EVAL4(PROG_NAME_LIST, 416, 448, 480, 512)
1596};
1597#undef PROG_NAME_LIST
1598
1599void bpf_patch_call_args(struct bpf_insn *insn, u32 stack_depth)
1600{
1601 stack_depth = max_t(u32, stack_depth, 1);
1602 insn->off = (s16) insn->imm;
1603 insn->imm = interpreters_args[(round_up(stack_depth, 32) / 32) - 1] -
1604 __bpf_call_base_args;
1605 insn->code = BPF_JMP | BPF_CALL_ARGS;
1606}
Alexei Starovoitovb870aa92017-05-30 13:31:33 -07001607
Alexei Starovoitov290af862018-01-09 10:04:29 -08001608#else
Daniel Borkmannfa9dd592018-01-20 01:24:33 +01001609static unsigned int __bpf_prog_ret0_warn(const void *ctx,
1610 const struct bpf_insn *insn)
Alexei Starovoitov290af862018-01-09 10:04:29 -08001611{
Daniel Borkmannfa9dd592018-01-20 01:24:33 +01001612 /* If this handler ever gets executed, then BPF_JIT_ALWAYS_ON
1613 * is not working properly, so warn about it!
1614 */
1615 WARN_ON_ONCE(1);
Alexei Starovoitov290af862018-01-09 10:04:29 -08001616 return 0;
1617}
1618#endif
1619
Daniel Borkmann3324b582015-05-29 23:23:07 +02001620bool bpf_prog_array_compatible(struct bpf_array *array,
1621 const struct bpf_prog *fp)
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001622{
Josef Bacik9802d862017-12-11 11:36:48 -05001623 if (fp->kprobe_override)
1624 return false;
1625
Daniel Borkmann3324b582015-05-29 23:23:07 +02001626 if (!array->owner_prog_type) {
1627 /* There's no owner yet where we could check for
1628 * compatibility.
1629 */
Alexei Starovoitov04fd61ab2015-05-19 16:59:03 -07001630 array->owner_prog_type = fp->type;
1631 array->owner_jited = fp->jited;
Daniel Borkmann3324b582015-05-29 23:23:07 +02001632
1633 return true;
Alexei Starovoitov04fd61ab2015-05-19 16:59:03 -07001634 }
Daniel Borkmann3324b582015-05-29 23:23:07 +02001635
1636 return array->owner_prog_type == fp->type &&
1637 array->owner_jited == fp->jited;
Alexei Starovoitov04fd61ab2015-05-19 16:59:03 -07001638}
1639
Daniel Borkmann3324b582015-05-29 23:23:07 +02001640static int bpf_check_tail_call(const struct bpf_prog *fp)
Alexei Starovoitov04fd61ab2015-05-19 16:59:03 -07001641{
1642 struct bpf_prog_aux *aux = fp->aux;
1643 int i;
1644
1645 for (i = 0; i < aux->used_map_cnt; i++) {
Daniel Borkmann3324b582015-05-29 23:23:07 +02001646 struct bpf_map *map = aux->used_maps[i];
Alexei Starovoitov04fd61ab2015-05-19 16:59:03 -07001647 struct bpf_array *array;
Alexei Starovoitov04fd61ab2015-05-19 16:59:03 -07001648
Alexei Starovoitov04fd61ab2015-05-19 16:59:03 -07001649 if (map->map_type != BPF_MAP_TYPE_PROG_ARRAY)
1650 continue;
Daniel Borkmann3324b582015-05-29 23:23:07 +02001651
Alexei Starovoitov04fd61ab2015-05-19 16:59:03 -07001652 array = container_of(map, struct bpf_array, map);
1653 if (!bpf_prog_array_compatible(array, fp))
1654 return -EINVAL;
1655 }
1656
1657 return 0;
1658}
1659
Daniel Borkmann9facc332018-06-15 02:30:48 +02001660static void bpf_prog_select_func(struct bpf_prog *fp)
1661{
1662#ifndef CONFIG_BPF_JIT_ALWAYS_ON
1663 u32 stack_depth = max_t(u32, fp->aux->stack_depth, 1);
1664
1665 fp->bpf_func = interpreters[(round_up(stack_depth, 32) / 32) - 1];
1666#else
1667 fp->bpf_func = __bpf_prog_ret0_warn;
1668#endif
1669}
1670
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001671/**
Daniel Borkmann3324b582015-05-29 23:23:07 +02001672 * bpf_prog_select_runtime - select exec runtime for BPF program
Alexei Starovoitov7ae457c2014-07-30 20:34:16 -07001673 * @fp: bpf_prog populated with internal BPF program
Daniel Borkmannd1c55ab2016-05-13 19:08:31 +02001674 * @err: pointer to error variable
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001675 *
Daniel Borkmann3324b582015-05-29 23:23:07 +02001676 * Try to JIT eBPF program, if JIT is not available, use interpreter.
1677 * The BPF program will be executed via BPF_PROG_RUN() macro.
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001678 */
Daniel Borkmannd1c55ab2016-05-13 19:08:31 +02001679struct bpf_prog *bpf_prog_select_runtime(struct bpf_prog *fp, int *err)
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001680{
Daniel Borkmann9facc332018-06-15 02:30:48 +02001681 /* In case of BPF to BPF calls, verifier did all the prep
1682 * work with regards to JITing, etc.
1683 */
1684 if (fp->bpf_func)
1685 goto finalize;
Martin KaFai Lau8007e402017-06-28 10:41:24 -07001686
Daniel Borkmann9facc332018-06-15 02:30:48 +02001687 bpf_prog_select_func(fp);
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001688
Daniel Borkmannd1c55ab2016-05-13 19:08:31 +02001689 /* eBPF JITs can rewrite the program in case constant
1690 * blinding is active. However, in case of error during
1691 * blinding, bpf_int_jit_compile() must always return a
1692 * valid program, which in this case would simply not
1693 * be JITed, but falls back to the interpreter.
1694 */
Jakub Kicinskiab3f0062017-11-03 13:56:17 -07001695 if (!bpf_prog_is_dev_bound(fp->aux)) {
Martin KaFai Lauc454a462018-12-07 16:42:25 -08001696 *err = bpf_prog_alloc_jited_linfo(fp);
1697 if (*err)
1698 return fp;
1699
Jakub Kicinskiab3f0062017-11-03 13:56:17 -07001700 fp = bpf_int_jit_compile(fp);
Alexei Starovoitov290af862018-01-09 10:04:29 -08001701 if (!fp->jited) {
Martin KaFai Lauc454a462018-12-07 16:42:25 -08001702 bpf_prog_free_jited_linfo(fp);
1703#ifdef CONFIG_BPF_JIT_ALWAYS_ON
Alexei Starovoitov290af862018-01-09 10:04:29 -08001704 *err = -ENOTSUPP;
1705 return fp;
Alexei Starovoitov290af862018-01-09 10:04:29 -08001706#endif
Martin KaFai Lauc454a462018-12-07 16:42:25 -08001707 } else {
1708 bpf_prog_free_unused_jited_linfo(fp);
1709 }
Jakub Kicinskiab3f0062017-11-03 13:56:17 -07001710 } else {
1711 *err = bpf_prog_offload_compile(fp);
1712 if (*err)
1713 return fp;
1714 }
Daniel Borkmann9facc332018-06-15 02:30:48 +02001715
1716finalize:
Daniel Borkmann60a3b222014-09-02 22:53:44 +02001717 bpf_prog_lock_ro(fp);
Alexei Starovoitov04fd61ab2015-05-19 16:59:03 -07001718
Daniel Borkmann3324b582015-05-29 23:23:07 +02001719 /* The tail call compatibility check can only be done at
1720 * this late stage as we need to determine, if we deal
1721 * with JITed or non JITed program concatenations and not
1722 * all eBPF JITs might immediately support all features.
1723 */
Daniel Borkmannd1c55ab2016-05-13 19:08:31 +02001724 *err = bpf_check_tail_call(fp);
1725
1726 return fp;
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001727}
Alexei Starovoitov7ae457c2014-07-30 20:34:16 -07001728EXPORT_SYMBOL_GPL(bpf_prog_select_runtime);
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001729
Yonghong Songe87c6bc382017-10-23 23:53:08 -07001730static unsigned int __bpf_prog_ret1(const void *ctx,
1731 const struct bpf_insn *insn)
1732{
1733 return 1;
1734}
1735
1736static struct bpf_prog_dummy {
1737 struct bpf_prog prog;
1738} dummy_bpf_prog = {
1739 .prog = {
1740 .bpf_func = __bpf_prog_ret1,
1741 },
1742};
1743
Alexei Starovoitov324bda9e62017-10-02 22:50:21 -07001744/* to avoid allocating empty bpf_prog_array for cgroups that
1745 * don't have bpf program attached use one global 'empty_prog_array'
1746 * It will not be modified the caller of bpf_prog_array_alloc()
1747 * (since caller requested prog_cnt == 0)
1748 * that pointer should be 'freed' by bpf_prog_array_free()
1749 */
1750static struct {
1751 struct bpf_prog_array hdr;
1752 struct bpf_prog *null_prog;
1753} empty_prog_array = {
1754 .null_prog = NULL,
1755};
1756
Roman Gushchind29ab6e2018-07-13 12:41:10 -07001757struct bpf_prog_array *bpf_prog_array_alloc(u32 prog_cnt, gfp_t flags)
Alexei Starovoitov324bda9e62017-10-02 22:50:21 -07001758{
1759 if (prog_cnt)
1760 return kzalloc(sizeof(struct bpf_prog_array) +
Roman Gushchin394e40a2018-08-02 14:27:21 -07001761 sizeof(struct bpf_prog_array_item) *
1762 (prog_cnt + 1),
Alexei Starovoitov324bda9e62017-10-02 22:50:21 -07001763 flags);
1764
1765 return &empty_prog_array.hdr;
1766}
1767
1768void bpf_prog_array_free(struct bpf_prog_array __rcu *progs)
1769{
1770 if (!progs ||
1771 progs == (struct bpf_prog_array __rcu *)&empty_prog_array.hdr)
1772 return;
1773 kfree_rcu(progs, rcu);
1774}
1775
Roman Gushchin394e40a2018-08-02 14:27:21 -07001776int bpf_prog_array_length(struct bpf_prog_array __rcu *array)
Alexei Starovoitov468e2f62017-10-02 22:50:22 -07001777{
Roman Gushchin394e40a2018-08-02 14:27:21 -07001778 struct bpf_prog_array_item *item;
Alexei Starovoitov468e2f62017-10-02 22:50:22 -07001779 u32 cnt = 0;
1780
1781 rcu_read_lock();
Roman Gushchin394e40a2018-08-02 14:27:21 -07001782 item = rcu_dereference(array)->items;
1783 for (; item->prog; item++)
1784 if (item->prog != &dummy_bpf_prog.prog)
Yonghong Songc8c088b2017-11-30 13:47:54 -08001785 cnt++;
Alexei Starovoitov468e2f62017-10-02 22:50:22 -07001786 rcu_read_unlock();
1787 return cnt;
1788}
1789
Roman Gushchin394e40a2018-08-02 14:27:21 -07001790
1791static bool bpf_prog_array_copy_core(struct bpf_prog_array __rcu *array,
Yonghong Song3a38bb92018-04-10 09:37:32 -07001792 u32 *prog_ids,
1793 u32 request_cnt)
1794{
Roman Gushchin394e40a2018-08-02 14:27:21 -07001795 struct bpf_prog_array_item *item;
Yonghong Song3a38bb92018-04-10 09:37:32 -07001796 int i = 0;
1797
Yonghong Song965931e2018-08-14 11:01:12 -07001798 item = rcu_dereference_check(array, 1)->items;
Roman Gushchin394e40a2018-08-02 14:27:21 -07001799 for (; item->prog; item++) {
1800 if (item->prog == &dummy_bpf_prog.prog)
Yonghong Song3a38bb92018-04-10 09:37:32 -07001801 continue;
Roman Gushchin394e40a2018-08-02 14:27:21 -07001802 prog_ids[i] = item->prog->aux->id;
Yonghong Song3a38bb92018-04-10 09:37:32 -07001803 if (++i == request_cnt) {
Roman Gushchin394e40a2018-08-02 14:27:21 -07001804 item++;
Yonghong Song3a38bb92018-04-10 09:37:32 -07001805 break;
1806 }
1807 }
1808
Roman Gushchin394e40a2018-08-02 14:27:21 -07001809 return !!(item->prog);
Yonghong Song3a38bb92018-04-10 09:37:32 -07001810}
1811
Roman Gushchin394e40a2018-08-02 14:27:21 -07001812int bpf_prog_array_copy_to_user(struct bpf_prog_array __rcu *array,
Alexei Starovoitov468e2f62017-10-02 22:50:22 -07001813 __u32 __user *prog_ids, u32 cnt)
1814{
Alexei Starovoitov09112872018-02-02 15:14:05 -08001815 unsigned long err = 0;
Alexei Starovoitov09112872018-02-02 15:14:05 -08001816 bool nospc;
Yonghong Song3a38bb92018-04-10 09:37:32 -07001817 u32 *ids;
Alexei Starovoitov468e2f62017-10-02 22:50:22 -07001818
Alexei Starovoitov09112872018-02-02 15:14:05 -08001819 /* users of this function are doing:
1820 * cnt = bpf_prog_array_length();
1821 * if (cnt > 0)
1822 * bpf_prog_array_copy_to_user(..., cnt);
1823 * so below kcalloc doesn't need extra cnt > 0 check, but
1824 * bpf_prog_array_length() releases rcu lock and
1825 * prog array could have been swapped with empty or larger array,
1826 * so always copy 'cnt' prog_ids to the user.
1827 * In a rare race the user will see zero prog_ids
1828 */
Daniel Borkmann9c481b92018-02-14 15:31:00 +01001829 ids = kcalloc(cnt, sizeof(u32), GFP_USER | __GFP_NOWARN);
Alexei Starovoitov09112872018-02-02 15:14:05 -08001830 if (!ids)
1831 return -ENOMEM;
Alexei Starovoitov468e2f62017-10-02 22:50:22 -07001832 rcu_read_lock();
Roman Gushchin394e40a2018-08-02 14:27:21 -07001833 nospc = bpf_prog_array_copy_core(array, ids, cnt);
Alexei Starovoitov468e2f62017-10-02 22:50:22 -07001834 rcu_read_unlock();
Alexei Starovoitov09112872018-02-02 15:14:05 -08001835 err = copy_to_user(prog_ids, ids, cnt * sizeof(u32));
1836 kfree(ids);
1837 if (err)
1838 return -EFAULT;
1839 if (nospc)
Alexei Starovoitov468e2f62017-10-02 22:50:22 -07001840 return -ENOSPC;
1841 return 0;
1842}
1843
Roman Gushchin394e40a2018-08-02 14:27:21 -07001844void bpf_prog_array_delete_safe(struct bpf_prog_array __rcu *array,
Yonghong Songe87c6bc382017-10-23 23:53:08 -07001845 struct bpf_prog *old_prog)
1846{
Roman Gushchin394e40a2018-08-02 14:27:21 -07001847 struct bpf_prog_array_item *item = array->items;
Yonghong Songe87c6bc382017-10-23 23:53:08 -07001848
Roman Gushchin394e40a2018-08-02 14:27:21 -07001849 for (; item->prog; item++)
1850 if (item->prog == old_prog) {
1851 WRITE_ONCE(item->prog, &dummy_bpf_prog.prog);
Yonghong Songe87c6bc382017-10-23 23:53:08 -07001852 break;
1853 }
1854}
1855
1856int bpf_prog_array_copy(struct bpf_prog_array __rcu *old_array,
1857 struct bpf_prog *exclude_prog,
1858 struct bpf_prog *include_prog,
1859 struct bpf_prog_array **new_array)
1860{
1861 int new_prog_cnt, carry_prog_cnt = 0;
Roman Gushchin394e40a2018-08-02 14:27:21 -07001862 struct bpf_prog_array_item *existing;
Yonghong Songe87c6bc382017-10-23 23:53:08 -07001863 struct bpf_prog_array *array;
Sean Young170a7e32018-05-27 12:24:08 +01001864 bool found_exclude = false;
Yonghong Songe87c6bc382017-10-23 23:53:08 -07001865 int new_prog_idx = 0;
1866
1867 /* Figure out how many existing progs we need to carry over to
1868 * the new array.
1869 */
1870 if (old_array) {
Roman Gushchin394e40a2018-08-02 14:27:21 -07001871 existing = old_array->items;
1872 for (; existing->prog; existing++) {
1873 if (existing->prog == exclude_prog) {
Sean Young170a7e32018-05-27 12:24:08 +01001874 found_exclude = true;
1875 continue;
1876 }
Roman Gushchin394e40a2018-08-02 14:27:21 -07001877 if (existing->prog != &dummy_bpf_prog.prog)
Yonghong Songe87c6bc382017-10-23 23:53:08 -07001878 carry_prog_cnt++;
Roman Gushchin394e40a2018-08-02 14:27:21 -07001879 if (existing->prog == include_prog)
Yonghong Songe87c6bc382017-10-23 23:53:08 -07001880 return -EEXIST;
1881 }
1882 }
1883
Sean Young170a7e32018-05-27 12:24:08 +01001884 if (exclude_prog && !found_exclude)
1885 return -ENOENT;
1886
Yonghong Songe87c6bc382017-10-23 23:53:08 -07001887 /* How many progs (not NULL) will be in the new array? */
1888 new_prog_cnt = carry_prog_cnt;
1889 if (include_prog)
1890 new_prog_cnt += 1;
1891
1892 /* Do we have any prog (not NULL) in the new array? */
1893 if (!new_prog_cnt) {
1894 *new_array = NULL;
1895 return 0;
1896 }
1897
1898 /* +1 as the end of prog_array is marked with NULL */
1899 array = bpf_prog_array_alloc(new_prog_cnt + 1, GFP_KERNEL);
1900 if (!array)
1901 return -ENOMEM;
1902
1903 /* Fill in the new prog array */
1904 if (carry_prog_cnt) {
Roman Gushchin394e40a2018-08-02 14:27:21 -07001905 existing = old_array->items;
1906 for (; existing->prog; existing++)
1907 if (existing->prog != exclude_prog &&
1908 existing->prog != &dummy_bpf_prog.prog) {
1909 array->items[new_prog_idx++].prog =
1910 existing->prog;
1911 }
Yonghong Songe87c6bc382017-10-23 23:53:08 -07001912 }
1913 if (include_prog)
Roman Gushchin394e40a2018-08-02 14:27:21 -07001914 array->items[new_prog_idx++].prog = include_prog;
1915 array->items[new_prog_idx].prog = NULL;
Yonghong Songe87c6bc382017-10-23 23:53:08 -07001916 *new_array = array;
1917 return 0;
1918}
1919
Yonghong Songf371b302017-12-11 11:39:02 -08001920int bpf_prog_array_copy_info(struct bpf_prog_array __rcu *array,
Yonghong Song3a38bb92018-04-10 09:37:32 -07001921 u32 *prog_ids, u32 request_cnt,
1922 u32 *prog_cnt)
Yonghong Songf371b302017-12-11 11:39:02 -08001923{
1924 u32 cnt = 0;
1925
1926 if (array)
1927 cnt = bpf_prog_array_length(array);
1928
Yonghong Song3a38bb92018-04-10 09:37:32 -07001929 *prog_cnt = cnt;
Yonghong Songf371b302017-12-11 11:39:02 -08001930
1931 /* return early if user requested only program count or nothing to copy */
1932 if (!request_cnt || !cnt)
1933 return 0;
1934
Yonghong Song3a38bb92018-04-10 09:37:32 -07001935 /* this function is called under trace/bpf_trace.c: bpf_event_mutex */
Roman Gushchin394e40a2018-08-02 14:27:21 -07001936 return bpf_prog_array_copy_core(array, prog_ids, request_cnt) ? -ENOSPC
Yonghong Song3a38bb92018-04-10 09:37:32 -07001937 : 0;
Yonghong Songf371b302017-12-11 11:39:02 -08001938}
1939
Daniel Borkmann60a3b222014-09-02 22:53:44 +02001940static void bpf_prog_free_deferred(struct work_struct *work)
1941{
Alexei Starovoitov09756af2014-09-26 00:17:00 -07001942 struct bpf_prog_aux *aux;
Alexei Starovoitov1c2a0882017-12-14 17:55:15 -08001943 int i;
Daniel Borkmann60a3b222014-09-02 22:53:44 +02001944
Alexei Starovoitov09756af2014-09-26 00:17:00 -07001945 aux = container_of(work, struct bpf_prog_aux, work);
Jakub Kicinskiab3f0062017-11-03 13:56:17 -07001946 if (bpf_prog_is_dev_bound(aux))
1947 bpf_prog_offload_destroy(aux->prog);
Yonghong Songc195651e2018-04-28 22:28:08 -07001948#ifdef CONFIG_PERF_EVENTS
1949 if (aux->prog->has_callchain_buf)
1950 put_callchain_buffers();
1951#endif
Alexei Starovoitov1c2a0882017-12-14 17:55:15 -08001952 for (i = 0; i < aux->func_cnt; i++)
1953 bpf_jit_free(aux->func[i]);
1954 if (aux->func_cnt) {
1955 kfree(aux->func);
1956 bpf_prog_unlock_free(aux->prog);
1957 } else {
1958 bpf_jit_free(aux->prog);
1959 }
Daniel Borkmann60a3b222014-09-02 22:53:44 +02001960}
1961
1962/* Free internal BPF program */
Alexei Starovoitov7ae457c2014-07-30 20:34:16 -07001963void bpf_prog_free(struct bpf_prog *fp)
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001964{
Alexei Starovoitov09756af2014-09-26 00:17:00 -07001965 struct bpf_prog_aux *aux = fp->aux;
Daniel Borkmann60a3b222014-09-02 22:53:44 +02001966
Alexei Starovoitov09756af2014-09-26 00:17:00 -07001967 INIT_WORK(&aux->work, bpf_prog_free_deferred);
Alexei Starovoitov09756af2014-09-26 00:17:00 -07001968 schedule_work(&aux->work);
Alexei Starovoitovf5bffec2014-07-22 23:01:58 -07001969}
Alexei Starovoitov7ae457c2014-07-30 20:34:16 -07001970EXPORT_SYMBOL_GPL(bpf_prog_free);
Alexei Starovoitovf89b7752014-10-23 18:41:08 -07001971
Daniel Borkmann3ad00402015-10-08 01:20:39 +02001972/* RNG for unpriviledged user space with separated state from prandom_u32(). */
1973static DEFINE_PER_CPU(struct rnd_state, bpf_user_rnd_state);
1974
1975void bpf_user_rnd_init_once(void)
1976{
1977 prandom_init_once(&bpf_user_rnd_state);
1978}
1979
Daniel Borkmannf3694e02016-09-09 02:45:31 +02001980BPF_CALL_0(bpf_user_rnd_u32)
Daniel Borkmann3ad00402015-10-08 01:20:39 +02001981{
1982 /* Should someone ever have the rather unwise idea to use some
1983 * of the registers passed into this function, then note that
1984 * this function is called from native eBPF and classic-to-eBPF
1985 * transformations. Register assignments from both sides are
1986 * different, f.e. classic always sets fn(ctx, A, X) here.
1987 */
1988 struct rnd_state *state;
1989 u32 res;
1990
1991 state = &get_cpu_var(bpf_user_rnd_state);
1992 res = prandom_u32_state(state);
Shaohua Lib761fe22016-09-27 08:42:41 -07001993 put_cpu_var(bpf_user_rnd_state);
Daniel Borkmann3ad00402015-10-08 01:20:39 +02001994
1995 return res;
1996}
1997
Daniel Borkmann3ba67da2015-03-05 23:27:51 +01001998/* Weak definitions of helper functions in case we don't have bpf syscall. */
1999const struct bpf_func_proto bpf_map_lookup_elem_proto __weak;
2000const struct bpf_func_proto bpf_map_update_elem_proto __weak;
2001const struct bpf_func_proto bpf_map_delete_elem_proto __weak;
Mauricio Vasquez Bf1a2e442018-10-18 15:16:25 +02002002const struct bpf_func_proto bpf_map_push_elem_proto __weak;
2003const struct bpf_func_proto bpf_map_pop_elem_proto __weak;
2004const struct bpf_func_proto bpf_map_peek_elem_proto __weak;
Daniel Borkmann3ba67da2015-03-05 23:27:51 +01002005
Daniel Borkmann03e69b52015-03-14 02:27:16 +01002006const struct bpf_func_proto bpf_get_prandom_u32_proto __weak;
Daniel Borkmannc04167c2015-03-14 02:27:17 +01002007const struct bpf_func_proto bpf_get_smp_processor_id_proto __weak;
Daniel Borkmann2d0e30c2016-10-21 12:46:33 +02002008const struct bpf_func_proto bpf_get_numa_node_id_proto __weak;
Daniel Borkmann17ca8cb2015-05-29 23:23:06 +02002009const struct bpf_func_proto bpf_ktime_get_ns_proto __weak;
Daniel Borkmannbd570ff2016-04-18 21:01:24 +02002010
Alexei Starovoitovffeedaf2015-06-12 19:39:12 -07002011const struct bpf_func_proto bpf_get_current_pid_tgid_proto __weak;
2012const struct bpf_func_proto bpf_get_current_uid_gid_proto __weak;
2013const struct bpf_func_proto bpf_get_current_comm_proto __weak;
Yonghong Songbf6fa2c82018-06-03 15:59:41 -07002014const struct bpf_func_proto bpf_get_current_cgroup_id_proto __weak;
Roman Gushchincd339432018-08-02 14:27:24 -07002015const struct bpf_func_proto bpf_get_local_storage_proto __weak;
Daniel Borkmannbd570ff2016-04-18 21:01:24 +02002016
Alexei Starovoitov0756ea32015-06-12 19:39:13 -07002017const struct bpf_func_proto * __weak bpf_get_trace_printk_proto(void)
2018{
2019 return NULL;
2020}
Daniel Borkmann03e69b52015-03-14 02:27:16 +01002021
Daniel Borkmann555c8a82016-07-14 18:08:05 +02002022u64 __weak
2023bpf_event_output(struct bpf_map *map, u64 flags, void *meta, u64 meta_size,
2024 void *ctx, u64 ctx_size, bpf_ctx_copy_t ctx_copy)
Daniel Borkmannbd570ff2016-04-18 21:01:24 +02002025{
Daniel Borkmann555c8a82016-07-14 18:08:05 +02002026 return -ENOTSUPP;
Daniel Borkmannbd570ff2016-04-18 21:01:24 +02002027}
Jakub Kicinski6cb5fb32018-05-03 18:37:10 -07002028EXPORT_SYMBOL_GPL(bpf_event_output);
Daniel Borkmannbd570ff2016-04-18 21:01:24 +02002029
Daniel Borkmann3324b582015-05-29 23:23:07 +02002030/* Always built-in helper functions. */
2031const struct bpf_func_proto bpf_tail_call_proto = {
2032 .func = NULL,
2033 .gpl_only = false,
2034 .ret_type = RET_VOID,
2035 .arg1_type = ARG_PTR_TO_CTX,
2036 .arg2_type = ARG_CONST_MAP_PTR,
2037 .arg3_type = ARG_ANYTHING,
2038};
2039
Daniel Borkmann93831912017-02-16 22:24:49 +01002040/* Stub for JITs that only support cBPF. eBPF programs are interpreted.
2041 * It is encouraged to implement bpf_int_jit_compile() instead, so that
2042 * eBPF and implicitly also cBPF can get JITed!
2043 */
Daniel Borkmannd1c55ab2016-05-13 19:08:31 +02002044struct bpf_prog * __weak bpf_int_jit_compile(struct bpf_prog *prog)
Daniel Borkmann3324b582015-05-29 23:23:07 +02002045{
Daniel Borkmannd1c55ab2016-05-13 19:08:31 +02002046 return prog;
Daniel Borkmann3324b582015-05-29 23:23:07 +02002047}
2048
Daniel Borkmann93831912017-02-16 22:24:49 +01002049/* Stub for JITs that support eBPF. All cBPF code gets transformed into
2050 * eBPF by the kernel and is later compiled by bpf_int_jit_compile().
2051 */
2052void __weak bpf_jit_compile(struct bpf_prog *prog)
2053{
2054}
2055
Martin KaFai Lau17bedab2016-12-07 15:53:11 -08002056bool __weak bpf_helper_changes_pkt_data(void *func)
Alexei Starovoitov969bf052016-05-05 19:49:10 -07002057{
2058 return false;
2059}
2060
Alexei Starovoitovf89b7752014-10-23 18:41:08 -07002061/* To execute LD_ABS/LD_IND instructions __bpf_prog_run() may call
2062 * skb_copy_bits(), so provide a weak definition of it for NET-less config.
2063 */
2064int __weak skb_copy_bits(const struct sk_buff *skb, int offset, void *to,
2065 int len)
2066{
2067 return -EFAULT;
2068}
Daniel Borkmanna67edbf2017-01-25 02:28:18 +01002069
2070/* All definitions of tracepoints related to BPF. */
2071#define CREATE_TRACE_POINTS
2072#include <linux/bpf_trace.h>
2073
2074EXPORT_TRACEPOINT_SYMBOL_GPL(xdp_exception);