aboutsummaryrefslogtreecommitdiffstats
path: root/kernel/bpf
diff options
context:
space:
mode:
authorJiri Kosina <jkosina@suse.cz>2014-11-20 08:42:02 -0500
committerJiri Kosina <jkosina@suse.cz>2014-11-20 08:42:02 -0500
commita02001086bbfb4da35d1228bebc2f1b442db455f (patch)
tree62ab47936cef06fd08657ca5b6cd1df98c19be57 /kernel/bpf
parenteff264efeeb0898408e8c9df72d8a32621035bed (diff)
parentfc14f9c1272f62c3e8d01300f52467c0d9af50f9 (diff)
Merge Linus' tree to be be to apply submitted patches to newer code than
current trivial.git base
Diffstat (limited to 'kernel/bpf')
-rw-r--r--kernel/bpf/Makefile4
-rw-r--r--kernel/bpf/core.c136
-rw-r--r--kernel/bpf/syscall.c606
-rw-r--r--kernel/bpf/test_stub.c116
-rw-r--r--kernel/bpf/verifier.c1924
5 files changed, 2784 insertions, 2 deletions
diff --git a/kernel/bpf/Makefile b/kernel/bpf/Makefile
index 6a71145e2769..0daf7f6ae7df 100644
--- a/kernel/bpf/Makefile
+++ b/kernel/bpf/Makefile
@@ -1 +1,5 @@
1obj-y := core.o 1obj-y := core.o
2obj-$(CONFIG_BPF_SYSCALL) += syscall.o verifier.o
3ifdef CONFIG_TEST_BPF
4obj-$(CONFIG_BPF_SYSCALL) += test_stub.o
5endif
diff --git a/kernel/bpf/core.c b/kernel/bpf/core.c
index 7f0dbcbb34af..d6594e457a25 100644
--- a/kernel/bpf/core.c
+++ b/kernel/bpf/core.c
@@ -20,9 +20,14 @@
20 * Andi Kleen - Fix a few bad bugs and races. 20 * Andi Kleen - Fix a few bad bugs and races.
21 * Kris Katterjohn - Added many additional checks in bpf_check_classic() 21 * Kris Katterjohn - Added many additional checks in bpf_check_classic()
22 */ 22 */
23
23#include <linux/filter.h> 24#include <linux/filter.h>
24#include <linux/skbuff.h> 25#include <linux/skbuff.h>
26#include <linux/vmalloc.h>
27#include <linux/random.h>
28#include <linux/moduleloader.h>
25#include <asm/unaligned.h> 29#include <asm/unaligned.h>
30#include <linux/bpf.h>
26 31
27/* Registers */ 32/* Registers */
28#define BPF_R0 regs[BPF_REG_0] 33#define BPF_R0 regs[BPF_REG_0]
@@ -63,6 +68,105 @@ void *bpf_internal_load_pointer_neg_helper(const struct sk_buff *skb, int k, uns
63 return NULL; 68 return NULL;
64} 69}
65 70
71struct bpf_prog *bpf_prog_alloc(unsigned int size, gfp_t gfp_extra_flags)
72{
73 gfp_t gfp_flags = GFP_KERNEL | __GFP_HIGHMEM | __GFP_ZERO |
74 gfp_extra_flags;
75 struct bpf_prog_aux *aux;
76 struct bpf_prog *fp;
77
78 size = round_up(size, PAGE_SIZE);
79 fp = __vmalloc(size, gfp_flags, PAGE_KERNEL);
80 if (fp == NULL)
81 return NULL;
82
83 aux = kzalloc(sizeof(*aux), GFP_KERNEL | gfp_extra_flags);
84 if (aux == NULL) {
85 vfree(fp);
86 return NULL;
87 }
88
89 fp->pages = size / PAGE_SIZE;
90 fp->aux = aux;
91
92 return fp;
93}
94EXPORT_SYMBOL_GPL(bpf_prog_alloc);
95
96struct bpf_prog *bpf_prog_realloc(struct bpf_prog *fp_old, unsigned int size,
97 gfp_t gfp_extra_flags)
98{
99 gfp_t gfp_flags = GFP_KERNEL | __GFP_HIGHMEM | __GFP_ZERO |
100 gfp_extra_flags;
101 struct bpf_prog *fp;
102
103 BUG_ON(fp_old == NULL);
104
105 size = round_up(size, PAGE_SIZE);
106 if (size <= fp_old->pages * PAGE_SIZE)
107 return fp_old;
108
109 fp = __vmalloc(size, gfp_flags, PAGE_KERNEL);
110 if (fp != NULL) {
111 memcpy(fp, fp_old, fp_old->pages * PAGE_SIZE);
112 fp->pages = size / PAGE_SIZE;
113
114 /* We keep fp->aux from fp_old around in the new
115 * reallocated structure.
116 */
117 fp_old->aux = NULL;
118 __bpf_prog_free(fp_old);
119 }
120
121 return fp;
122}
123EXPORT_SYMBOL_GPL(bpf_prog_realloc);
124
125void __bpf_prog_free(struct bpf_prog *fp)
126{
127 kfree(fp->aux);
128 vfree(fp);
129}
130EXPORT_SYMBOL_GPL(__bpf_prog_free);
131
132#ifdef CONFIG_BPF_JIT
133struct bpf_binary_header *
134bpf_jit_binary_alloc(unsigned int proglen, u8 **image_ptr,
135 unsigned int alignment,
136 bpf_jit_fill_hole_t bpf_fill_ill_insns)
137{
138 struct bpf_binary_header *hdr;
139 unsigned int size, hole, start;
140
141 /* Most of BPF filters are really small, but if some of them
142 * fill a page, allow at least 128 extra bytes to insert a
143 * random section of illegal instructions.
144 */
145 size = round_up(proglen + sizeof(*hdr) + 128, PAGE_SIZE);
146 hdr = module_alloc(size);
147 if (hdr == NULL)
148 return NULL;
149
150 /* Fill space with illegal/arch-dep instructions. */
151 bpf_fill_ill_insns(hdr, size);
152
153 hdr->pages = size / PAGE_SIZE;
154 hole = min_t(unsigned int, size - (proglen + sizeof(*hdr)),
155 PAGE_SIZE - sizeof(*hdr));
156 start = (prandom_u32() % hole) & ~(alignment - 1);
157
158 /* Leave a random number of instructions before BPF code. */
159 *image_ptr = &hdr->image[start];
160
161 return hdr;
162}
163
164void bpf_jit_binary_free(struct bpf_binary_header *hdr)
165{
166 module_free(NULL, hdr);
167}
168#endif /* CONFIG_BPF_JIT */
169
66/* Base function for offset calculation. Needs to go into .text section, 170/* Base function for offset calculation. Needs to go into .text section,
67 * therefore keeping it non-static as well; will also be used by JITs 171 * therefore keeping it non-static as well; will also be used by JITs
68 * anyway later on, so do not let the compiler omit it. 172 * anyway later on, so do not let the compiler omit it.
@@ -180,6 +284,7 @@ static unsigned int __bpf_prog_run(void *ctx, const struct bpf_insn *insn)
180 [BPF_LD | BPF_IND | BPF_W] = &&LD_IND_W, 284 [BPF_LD | BPF_IND | BPF_W] = &&LD_IND_W,
181 [BPF_LD | BPF_IND | BPF_H] = &&LD_IND_H, 285 [BPF_LD | BPF_IND | BPF_H] = &&LD_IND_H,
182 [BPF_LD | BPF_IND | BPF_B] = &&LD_IND_B, 286 [BPF_LD | BPF_IND | BPF_B] = &&LD_IND_B,
287 [BPF_LD | BPF_IMM | BPF_DW] = &&LD_IMM_DW,
183 }; 288 };
184 void *ptr; 289 void *ptr;
185 int off; 290 int off;
@@ -239,6 +344,10 @@ select_insn:
239 ALU64_MOV_K: 344 ALU64_MOV_K:
240 DST = IMM; 345 DST = IMM;
241 CONT; 346 CONT;
347 LD_IMM_DW:
348 DST = (u64) (u32) insn[0].imm | ((u64) (u32) insn[1].imm) << 32;
349 insn++;
350 CONT;
242 ALU64_ARSH_X: 351 ALU64_ARSH_X:
243 (*(s64 *) &DST) >>= SRC; 352 (*(s64 *) &DST) >>= SRC;
244 CONT; 353 CONT;
@@ -523,12 +632,35 @@ void bpf_prog_select_runtime(struct bpf_prog *fp)
523 632
524 /* Probe if internal BPF can be JITed */ 633 /* Probe if internal BPF can be JITed */
525 bpf_int_jit_compile(fp); 634 bpf_int_jit_compile(fp);
635 /* Lock whole bpf_prog as read-only */
636 bpf_prog_lock_ro(fp);
526} 637}
527EXPORT_SYMBOL_GPL(bpf_prog_select_runtime); 638EXPORT_SYMBOL_GPL(bpf_prog_select_runtime);
528 639
529/* free internal BPF program */ 640static void bpf_prog_free_deferred(struct work_struct *work)
641{
642 struct bpf_prog_aux *aux;
643
644 aux = container_of(work, struct bpf_prog_aux, work);
645 bpf_jit_free(aux->prog);
646}
647
648/* Free internal BPF program */
530void bpf_prog_free(struct bpf_prog *fp) 649void bpf_prog_free(struct bpf_prog *fp)
531{ 650{
532 bpf_jit_free(fp); 651 struct bpf_prog_aux *aux = fp->aux;
652
653 INIT_WORK(&aux->work, bpf_prog_free_deferred);
654 aux->prog = fp;
655 schedule_work(&aux->work);
533} 656}
534EXPORT_SYMBOL_GPL(bpf_prog_free); 657EXPORT_SYMBOL_GPL(bpf_prog_free);
658
659/* To execute LD_ABS/LD_IND instructions __bpf_prog_run() may call
660 * skb_copy_bits(), so provide a weak definition of it for NET-less config.
661 */
662int __weak skb_copy_bits(const struct sk_buff *skb, int offset, void *to,
663 int len)
664{
665 return -EFAULT;
666}
diff --git a/kernel/bpf/syscall.c b/kernel/bpf/syscall.c
new file mode 100644
index 000000000000..ba61c8c16032
--- /dev/null
+++ b/kernel/bpf/syscall.c
@@ -0,0 +1,606 @@
1/* Copyright (c) 2011-2014 PLUMgrid, http://plumgrid.com
2 *
3 * This program is free software; you can redistribute it and/or
4 * modify it under the terms of version 2 of the GNU General Public
5 * License as published by the Free Software Foundation.
6 *
7 * This program is distributed in the hope that it will be useful, but
8 * WITHOUT ANY WARRANTY; without even the implied warranty of
9 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
10 * General Public License for more details.
11 */
12#include <linux/bpf.h>
13#include <linux/syscalls.h>
14#include <linux/slab.h>
15#include <linux/anon_inodes.h>
16#include <linux/file.h>
17#include <linux/license.h>
18#include <linux/filter.h>
19
20static LIST_HEAD(bpf_map_types);
21
22static struct bpf_map *find_and_alloc_map(union bpf_attr *attr)
23{
24 struct bpf_map_type_list *tl;
25 struct bpf_map *map;
26
27 list_for_each_entry(tl, &bpf_map_types, list_node) {
28 if (tl->type == attr->map_type) {
29 map = tl->ops->map_alloc(attr);
30 if (IS_ERR(map))
31 return map;
32 map->ops = tl->ops;
33 map->map_type = attr->map_type;
34 return map;
35 }
36 }
37 return ERR_PTR(-EINVAL);
38}
39
40/* boot time registration of different map implementations */
41void bpf_register_map_type(struct bpf_map_type_list *tl)
42{
43 list_add(&tl->list_node, &bpf_map_types);
44}
45
46/* called from workqueue */
47static void bpf_map_free_deferred(struct work_struct *work)
48{
49 struct bpf_map *map = container_of(work, struct bpf_map, work);
50
51 /* implementation dependent freeing */
52 map->ops->map_free(map);
53}
54
55/* decrement map refcnt and schedule it for freeing via workqueue
56 * (unrelying map implementation ops->map_free() might sleep)
57 */
58void bpf_map_put(struct bpf_map *map)
59{
60 if (atomic_dec_and_test(&map->refcnt)) {
61 INIT_WORK(&map->work, bpf_map_free_deferred);
62 schedule_work(&map->work);
63 }
64}
65
66static int bpf_map_release(struct inode *inode, struct file *filp)
67{
68 struct bpf_map *map = filp->private_data;
69
70 bpf_map_put(map);
71 return 0;
72}
73
74static const struct file_operations bpf_map_fops = {
75 .release = bpf_map_release,
76};
77
78/* helper macro to check that unused fields 'union bpf_attr' are zero */
79#define CHECK_ATTR(CMD) \
80 memchr_inv((void *) &attr->CMD##_LAST_FIELD + \
81 sizeof(attr->CMD##_LAST_FIELD), 0, \
82 sizeof(*attr) - \
83 offsetof(union bpf_attr, CMD##_LAST_FIELD) - \
84 sizeof(attr->CMD##_LAST_FIELD)) != NULL
85
86#define BPF_MAP_CREATE_LAST_FIELD max_entries
87/* called via syscall */
88static int map_create(union bpf_attr *attr)
89{
90 struct bpf_map *map;
91 int err;
92
93 err = CHECK_ATTR(BPF_MAP_CREATE);
94 if (err)
95 return -EINVAL;
96
97 /* find map type and init map: hashtable vs rbtree vs bloom vs ... */
98 map = find_and_alloc_map(attr);
99 if (IS_ERR(map))
100 return PTR_ERR(map);
101
102 atomic_set(&map->refcnt, 1);
103
104 err = anon_inode_getfd("bpf-map", &bpf_map_fops, map, O_RDWR | O_CLOEXEC);
105
106 if (err < 0)
107 /* failed to allocate fd */
108 goto free_map;
109
110 return err;
111
112free_map:
113 map->ops->map_free(map);
114 return err;
115}
116
117/* if error is returned, fd is released.
118 * On success caller should complete fd access with matching fdput()
119 */
120struct bpf_map *bpf_map_get(struct fd f)
121{
122 struct bpf_map *map;
123
124 if (!f.file)
125 return ERR_PTR(-EBADF);
126
127 if (f.file->f_op != &bpf_map_fops) {
128 fdput(f);
129 return ERR_PTR(-EINVAL);
130 }
131
132 map = f.file->private_data;
133
134 return map;
135}
136
137/* helper to convert user pointers passed inside __aligned_u64 fields */
138static void __user *u64_to_ptr(__u64 val)
139{
140 return (void __user *) (unsigned long) val;
141}
142
143/* last field in 'union bpf_attr' used by this command */
144#define BPF_MAP_LOOKUP_ELEM_LAST_FIELD value
145
146static int map_lookup_elem(union bpf_attr *attr)
147{
148 void __user *ukey = u64_to_ptr(attr->key);
149 void __user *uvalue = u64_to_ptr(attr->value);
150 int ufd = attr->map_fd;
151 struct fd f = fdget(ufd);
152 struct bpf_map *map;
153 void *key, *value;
154 int err;
155
156 if (CHECK_ATTR(BPF_MAP_LOOKUP_ELEM))
157 return -EINVAL;
158
159 map = bpf_map_get(f);
160 if (IS_ERR(map))
161 return PTR_ERR(map);
162
163 err = -ENOMEM;
164 key = kmalloc(map->key_size, GFP_USER);
165 if (!key)
166 goto err_put;
167
168 err = -EFAULT;
169 if (copy_from_user(key, ukey, map->key_size) != 0)
170 goto free_key;
171
172 err = -ESRCH;
173 rcu_read_lock();
174 value = map->ops->map_lookup_elem(map, key);
175 if (!value)
176 goto err_unlock;
177
178 err = -EFAULT;
179 if (copy_to_user(uvalue, value, map->value_size) != 0)
180 goto err_unlock;
181
182 err = 0;
183
184err_unlock:
185 rcu_read_unlock();
186free_key:
187 kfree(key);
188err_put:
189 fdput(f);
190 return err;
191}
192
193#define BPF_MAP_UPDATE_ELEM_LAST_FIELD value
194
195static int map_update_elem(union bpf_attr *attr)
196{
197 void __user *ukey = u64_to_ptr(attr->key);
198 void __user *uvalue = u64_to_ptr(attr->value);
199 int ufd = attr->map_fd;
200 struct fd f = fdget(ufd);
201 struct bpf_map *map;
202 void *key, *value;
203 int err;
204
205 if (CHECK_ATTR(BPF_MAP_UPDATE_ELEM))
206 return -EINVAL;
207
208 map = bpf_map_get(f);
209 if (IS_ERR(map))
210 return PTR_ERR(map);
211
212 err = -ENOMEM;
213 key = kmalloc(map->key_size, GFP_USER);
214 if (!key)
215 goto err_put;
216
217 err = -EFAULT;
218 if (copy_from_user(key, ukey, map->key_size) != 0)
219 goto free_key;
220
221 err = -ENOMEM;
222 value = kmalloc(map->value_size, GFP_USER);
223 if (!value)
224 goto free_key;
225
226 err = -EFAULT;
227 if (copy_from_user(value, uvalue, map->value_size) != 0)
228 goto free_value;
229
230 /* eBPF program that use maps are running under rcu_read_lock(),
231 * therefore all map accessors rely on this fact, so do the same here
232 */
233 rcu_read_lock();
234 err = map->ops->map_update_elem(map, key, value);
235 rcu_read_unlock();
236
237free_value:
238 kfree(value);
239free_key:
240 kfree(key);
241err_put:
242 fdput(f);
243 return err;
244}
245
246#define BPF_MAP_DELETE_ELEM_LAST_FIELD key
247
248static int map_delete_elem(union bpf_attr *attr)
249{
250 void __user *ukey = u64_to_ptr(attr->key);
251 int ufd = attr->map_fd;
252 struct fd f = fdget(ufd);
253 struct bpf_map *map;
254 void *key;
255 int err;
256
257 if (CHECK_ATTR(BPF_MAP_DELETE_ELEM))
258 return -EINVAL;
259
260 map = bpf_map_get(f);
261 if (IS_ERR(map))
262 return PTR_ERR(map);
263
264 err = -ENOMEM;
265 key = kmalloc(map->key_size, GFP_USER);
266 if (!key)
267 goto err_put;
268
269 err = -EFAULT;
270 if (copy_from_user(key, ukey, map->key_size) != 0)
271 goto free_key;
272
273 rcu_read_lock();
274 err = map->ops->map_delete_elem(map, key);
275 rcu_read_unlock();
276
277free_key:
278 kfree(key);
279err_put:
280 fdput(f);
281 return err;
282}
283
284/* last field in 'union bpf_attr' used by this command */
285#define BPF_MAP_GET_NEXT_KEY_LAST_FIELD next_key
286
287static int map_get_next_key(union bpf_attr *attr)
288{
289 void __user *ukey = u64_to_ptr(attr->key);
290 void __user *unext_key = u64_to_ptr(attr->next_key);
291 int ufd = attr->map_fd;
292 struct fd f = fdget(ufd);
293 struct bpf_map *map;
294 void *key, *next_key;
295 int err;
296
297 if (CHECK_ATTR(BPF_MAP_GET_NEXT_KEY))
298 return -EINVAL;
299
300 map = bpf_map_get(f);
301 if (IS_ERR(map))
302 return PTR_ERR(map);
303
304 err = -ENOMEM;
305 key = kmalloc(map->key_size, GFP_USER);
306 if (!key)
307 goto err_put;
308
309 err = -EFAULT;
310 if (copy_from_user(key, ukey, map->key_size) != 0)
311 goto free_key;
312
313 err = -ENOMEM;
314 next_key = kmalloc(map->key_size, GFP_USER);
315 if (!next_key)
316 goto free_key;
317
318 rcu_read_lock();
319 err = map->ops->map_get_next_key(map, key, next_key);
320 rcu_read_unlock();
321 if (err)
322 goto free_next_key;
323
324 err = -EFAULT;
325 if (copy_to_user(unext_key, next_key, map->key_size) != 0)
326 goto free_next_key;
327
328 err = 0;
329
330free_next_key:
331 kfree(next_key);
332free_key:
333 kfree(key);
334err_put:
335 fdput(f);
336 return err;
337}
338
339static LIST_HEAD(bpf_prog_types);
340
341static int find_prog_type(enum bpf_prog_type type, struct bpf_prog *prog)
342{
343 struct bpf_prog_type_list *tl;
344
345 list_for_each_entry(tl, &bpf_prog_types, list_node) {
346 if (tl->type == type) {
347 prog->aux->ops = tl->ops;
348 prog->aux->prog_type = type;
349 return 0;
350 }
351 }
352 return -EINVAL;
353}
354
355void bpf_register_prog_type(struct bpf_prog_type_list *tl)
356{
357 list_add(&tl->list_node, &bpf_prog_types);
358}
359
360/* fixup insn->imm field of bpf_call instructions:
361 * if (insn->imm == BPF_FUNC_map_lookup_elem)
362 * insn->imm = bpf_map_lookup_elem - __bpf_call_base;
363 * else if (insn->imm == BPF_FUNC_map_update_elem)
364 * insn->imm = bpf_map_update_elem - __bpf_call_base;
365 * else ...
366 *
367 * this function is called after eBPF program passed verification
368 */
369static void fixup_bpf_calls(struct bpf_prog *prog)
370{
371 const struct bpf_func_proto *fn;
372 int i;
373
374 for (i = 0; i < prog->len; i++) {
375 struct bpf_insn *insn = &prog->insnsi[i];
376
377 if (insn->code == (BPF_JMP | BPF_CALL)) {
378 /* we reach here when program has bpf_call instructions
379 * and it passed bpf_check(), means that
380 * ops->get_func_proto must have been supplied, check it
381 */
382 BUG_ON(!prog->aux->ops->get_func_proto);
383
384 fn = prog->aux->ops->get_func_proto(insn->imm);
385 /* all functions that have prototype and verifier allowed
386 * programs to call them, must be real in-kernel functions
387 */
388 BUG_ON(!fn->func);
389 insn->imm = fn->func - __bpf_call_base;
390 }
391 }
392}
393
394/* drop refcnt on maps used by eBPF program and free auxilary data */
395static void free_used_maps(struct bpf_prog_aux *aux)
396{
397 int i;
398
399 for (i = 0; i < aux->used_map_cnt; i++)
400 bpf_map_put(aux->used_maps[i]);
401
402 kfree(aux->used_maps);
403}
404
405void bpf_prog_put(struct bpf_prog *prog)
406{
407 if (atomic_dec_and_test(&prog->aux->refcnt)) {
408 free_used_maps(prog->aux);
409 bpf_prog_free(prog);
410 }
411}
412
413static int bpf_prog_release(struct inode *inode, struct file *filp)
414{
415 struct bpf_prog *prog = filp->private_data;
416
417 bpf_prog_put(prog);
418 return 0;
419}
420
421static const struct file_operations bpf_prog_fops = {
422 .release = bpf_prog_release,
423};
424
425static struct bpf_prog *get_prog(struct fd f)
426{
427 struct bpf_prog *prog;
428
429 if (!f.file)
430 return ERR_PTR(-EBADF);
431
432 if (f.file->f_op != &bpf_prog_fops) {
433 fdput(f);
434 return ERR_PTR(-EINVAL);
435 }
436
437 prog = f.file->private_data;
438
439 return prog;
440}
441
442/* called by sockets/tracing/seccomp before attaching program to an event
443 * pairs with bpf_prog_put()
444 */
445struct bpf_prog *bpf_prog_get(u32 ufd)
446{
447 struct fd f = fdget(ufd);
448 struct bpf_prog *prog;
449
450 prog = get_prog(f);
451
452 if (IS_ERR(prog))
453 return prog;
454
455 atomic_inc(&prog->aux->refcnt);
456 fdput(f);
457 return prog;
458}
459
460/* last field in 'union bpf_attr' used by this command */
461#define BPF_PROG_LOAD_LAST_FIELD log_buf
462
463static int bpf_prog_load(union bpf_attr *attr)
464{
465 enum bpf_prog_type type = attr->prog_type;
466 struct bpf_prog *prog;
467 int err;
468 char license[128];
469 bool is_gpl;
470
471 if (CHECK_ATTR(BPF_PROG_LOAD))
472 return -EINVAL;
473
474 /* copy eBPF program license from user space */
475 if (strncpy_from_user(license, u64_to_ptr(attr->license),
476 sizeof(license) - 1) < 0)
477 return -EFAULT;
478 license[sizeof(license) - 1] = 0;
479
480 /* eBPF programs must be GPL compatible to use GPL-ed functions */
481 is_gpl = license_is_gpl_compatible(license);
482
483 if (attr->insn_cnt >= BPF_MAXINSNS)
484 return -EINVAL;
485
486 /* plain bpf_prog allocation */
487 prog = bpf_prog_alloc(bpf_prog_size(attr->insn_cnt), GFP_USER);
488 if (!prog)
489 return -ENOMEM;
490
491 prog->len = attr->insn_cnt;
492
493 err = -EFAULT;
494 if (copy_from_user(prog->insns, u64_to_ptr(attr->insns),
495 prog->len * sizeof(struct bpf_insn)) != 0)
496 goto free_prog;
497
498 prog->orig_prog = NULL;
499 prog->jited = false;
500
501 atomic_set(&prog->aux->refcnt, 1);
502 prog->aux->is_gpl_compatible = is_gpl;
503
504 /* find program type: socket_filter vs tracing_filter */
505 err = find_prog_type(type, prog);
506 if (err < 0)
507 goto free_prog;
508
509 /* run eBPF verifier */
510 err = bpf_check(prog, attr);
511
512 if (err < 0)
513 goto free_used_maps;
514
515 /* fixup BPF_CALL->imm field */
516 fixup_bpf_calls(prog);
517
518 /* eBPF program is ready to be JITed */
519 bpf_prog_select_runtime(prog);
520
521 err = anon_inode_getfd("bpf-prog", &bpf_prog_fops, prog, O_RDWR | O_CLOEXEC);
522
523 if (err < 0)
524 /* failed to allocate fd */
525 goto free_used_maps;
526
527 return err;
528
529free_used_maps:
530 free_used_maps(prog->aux);
531free_prog:
532 bpf_prog_free(prog);
533 return err;
534}
535
536SYSCALL_DEFINE3(bpf, int, cmd, union bpf_attr __user *, uattr, unsigned int, size)
537{
538 union bpf_attr attr = {};
539 int err;
540
541 /* the syscall is limited to root temporarily. This restriction will be
542 * lifted when security audit is clean. Note that eBPF+tracing must have
543 * this restriction, since it may pass kernel data to user space
544 */
545 if (!capable(CAP_SYS_ADMIN))
546 return -EPERM;
547
548 if (!access_ok(VERIFY_READ, uattr, 1))
549 return -EFAULT;
550
551 if (size > PAGE_SIZE) /* silly large */
552 return -E2BIG;
553
554 /* If we're handed a bigger struct than we know of,
555 * ensure all the unknown bits are 0 - i.e. new
556 * user-space does not rely on any kernel feature
557 * extensions we dont know about yet.
558 */
559 if (size > sizeof(attr)) {
560 unsigned char __user *addr;
561 unsigned char __user *end;
562 unsigned char val;
563
564 addr = (void __user *)uattr + sizeof(attr);
565 end = (void __user *)uattr + size;
566
567 for (; addr < end; addr++) {
568 err = get_user(val, addr);
569 if (err)
570 return err;
571 if (val)
572 return -E2BIG;
573 }
574 size = sizeof(attr);
575 }
576
577 /* copy attributes from user space, may be less than sizeof(bpf_attr) */
578 if (copy_from_user(&attr, uattr, size) != 0)
579 return -EFAULT;
580
581 switch (cmd) {
582 case BPF_MAP_CREATE:
583 err = map_create(&attr);
584 break;
585 case BPF_MAP_LOOKUP_ELEM:
586 err = map_lookup_elem(&attr);
587 break;
588 case BPF_MAP_UPDATE_ELEM:
589 err = map_update_elem(&attr);
590 break;
591 case BPF_MAP_DELETE_ELEM:
592 err = map_delete_elem(&attr);
593 break;
594 case BPF_MAP_GET_NEXT_KEY:
595 err = map_get_next_key(&attr);
596 break;
597 case BPF_PROG_LOAD:
598 err = bpf_prog_load(&attr);
599 break;
600 default:
601 err = -EINVAL;
602 break;
603 }
604
605 return err;
606}
diff --git a/kernel/bpf/test_stub.c b/kernel/bpf/test_stub.c
new file mode 100644
index 000000000000..fcaddff4003e
--- /dev/null
+++ b/kernel/bpf/test_stub.c
@@ -0,0 +1,116 @@
1/* Copyright (c) 2011-2014 PLUMgrid, http://plumgrid.com
2 *
3 * This program is free software; you can redistribute it and/or
4 * modify it under the terms of version 2 of the GNU General Public
5 * License as published by the Free Software Foundation.
6 */
7#include <linux/kernel.h>
8#include <linux/types.h>
9#include <linux/slab.h>
10#include <linux/err.h>
11#include <linux/bpf.h>
12
13/* test stubs for BPF_MAP_TYPE_UNSPEC and for BPF_PROG_TYPE_UNSPEC
14 * to be used by user space verifier testsuite
15 */
16struct bpf_context {
17 u64 arg1;
18 u64 arg2;
19};
20
21static u64 test_func(u64 r1, u64 r2, u64 r3, u64 r4, u64 r5)
22{
23 return 0;
24}
25
26static struct bpf_func_proto test_funcs[] = {
27 [BPF_FUNC_unspec] = {
28 .func = test_func,
29 .gpl_only = true,
30 .ret_type = RET_PTR_TO_MAP_VALUE_OR_NULL,
31 .arg1_type = ARG_CONST_MAP_PTR,
32 .arg2_type = ARG_PTR_TO_MAP_KEY,
33 },
34};
35
36static const struct bpf_func_proto *test_func_proto(enum bpf_func_id func_id)
37{
38 if (func_id < 0 || func_id >= ARRAY_SIZE(test_funcs))
39 return NULL;
40 return &test_funcs[func_id];
41}
42
43static const struct bpf_context_access {
44 int size;
45 enum bpf_access_type type;
46} test_ctx_access[] = {
47 [offsetof(struct bpf_context, arg1)] = {
48 FIELD_SIZEOF(struct bpf_context, arg1),
49 BPF_READ
50 },
51 [offsetof(struct bpf_context, arg2)] = {
52 FIELD_SIZEOF(struct bpf_context, arg2),
53 BPF_READ
54 },
55};
56
57static bool test_is_valid_access(int off, int size, enum bpf_access_type type)
58{
59 const struct bpf_context_access *access;
60
61 if (off < 0 || off >= ARRAY_SIZE(test_ctx_access))
62 return false;
63
64 access = &test_ctx_access[off];
65 if (access->size == size && (access->type & type))
66 return true;
67
68 return false;
69}
70
71static struct bpf_verifier_ops test_ops = {
72 .get_func_proto = test_func_proto,
73 .is_valid_access = test_is_valid_access,
74};
75
76static struct bpf_prog_type_list tl_prog = {
77 .ops = &test_ops,
78 .type = BPF_PROG_TYPE_UNSPEC,
79};
80
81static struct bpf_map *test_map_alloc(union bpf_attr *attr)
82{
83 struct bpf_map *map;
84
85 map = kzalloc(sizeof(*map), GFP_USER);
86 if (!map)
87 return ERR_PTR(-ENOMEM);
88
89 map->key_size = attr->key_size;
90 map->value_size = attr->value_size;
91 map->max_entries = attr->max_entries;
92 return map;
93}
94
95static void test_map_free(struct bpf_map *map)
96{
97 kfree(map);
98}
99
100static struct bpf_map_ops test_map_ops = {
101 .map_alloc = test_map_alloc,
102 .map_free = test_map_free,
103};
104
105static struct bpf_map_type_list tl_map = {
106 .ops = &test_map_ops,
107 .type = BPF_MAP_TYPE_UNSPEC,
108};
109
110static int __init register_test_ops(void)
111{
112 bpf_register_map_type(&tl_map);
113 bpf_register_prog_type(&tl_prog);
114 return 0;
115}
116late_initcall(register_test_ops);
diff --git a/kernel/bpf/verifier.c b/kernel/bpf/verifier.c
new file mode 100644
index 000000000000..9f81818f2941
--- /dev/null
+++ b/kernel/bpf/verifier.c
@@ -0,0 +1,1924 @@
1/* Copyright (c) 2011-2014 PLUMgrid, http://plumgrid.com
2 *
3 * This program is free software; you can redistribute it and/or
4 * modify it under the terms of version 2 of the GNU General Public
5 * License as published by the Free Software Foundation.
6 *
7 * This program is distributed in the hope that it will be useful, but
8 * WITHOUT ANY WARRANTY; without even the implied warranty of
9 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
10 * General Public License for more details.
11 */
12#include <linux/kernel.h>
13#include <linux/types.h>
14#include <linux/slab.h>
15#include <linux/bpf.h>
16#include <linux/filter.h>
17#include <net/netlink.h>
18#include <linux/file.h>
19#include <linux/vmalloc.h>
20
21/* bpf_check() is a static code analyzer that walks eBPF program
22 * instruction by instruction and updates register/stack state.
23 * All paths of conditional branches are analyzed until 'bpf_exit' insn.
24 *
25 * The first pass is depth-first-search to check that the program is a DAG.
26 * It rejects the following programs:
27 * - larger than BPF_MAXINSNS insns
28 * - if loop is present (detected via back-edge)
29 * - unreachable insns exist (shouldn't be a forest. program = one function)
30 * - out of bounds or malformed jumps
31 * The second pass is all possible path descent from the 1st insn.
32 * Since it's analyzing all pathes through the program, the length of the
33 * analysis is limited to 32k insn, which may be hit even if total number of
34 * insn is less then 4K, but there are too many branches that change stack/regs.
35 * Number of 'branches to be analyzed' is limited to 1k
36 *
37 * On entry to each instruction, each register has a type, and the instruction
38 * changes the types of the registers depending on instruction semantics.
39 * If instruction is BPF_MOV64_REG(BPF_REG_1, BPF_REG_5), then type of R5 is
40 * copied to R1.
41 *
42 * All registers are 64-bit.
43 * R0 - return register
44 * R1-R5 argument passing registers
45 * R6-R9 callee saved registers
46 * R10 - frame pointer read-only
47 *
48 * At the start of BPF program the register R1 contains a pointer to bpf_context
49 * and has type PTR_TO_CTX.
50 *
51 * Verifier tracks arithmetic operations on pointers in case:
52 * BPF_MOV64_REG(BPF_REG_1, BPF_REG_10),
53 * BPF_ALU64_IMM(BPF_ADD, BPF_REG_1, -20),
54 * 1st insn copies R10 (which has FRAME_PTR) type into R1
55 * and 2nd arithmetic instruction is pattern matched to recognize
56 * that it wants to construct a pointer to some element within stack.
57 * So after 2nd insn, the register R1 has type PTR_TO_STACK
58 * (and -20 constant is saved for further stack bounds checking).
59 * Meaning that this reg is a pointer to stack plus known immediate constant.
60 *
61 * Most of the time the registers have UNKNOWN_VALUE type, which
62 * means the register has some value, but it's not a valid pointer.
63 * (like pointer plus pointer becomes UNKNOWN_VALUE type)
64 *
65 * When verifier sees load or store instructions the type of base register
66 * can be: PTR_TO_MAP_VALUE, PTR_TO_CTX, FRAME_PTR. These are three pointer
67 * types recognized by check_mem_access() function.
68 *
69 * PTR_TO_MAP_VALUE means that this register is pointing to 'map element value'
70 * and the range of [ptr, ptr + map's value_size) is accessible.
71 *
72 * registers used to pass values to function calls are checked against
73 * function argument constraints.
74 *
75 * ARG_PTR_TO_MAP_KEY is one of such argument constraints.
76 * It means that the register type passed to this function must be
77 * PTR_TO_STACK and it will be used inside the function as
78 * 'pointer to map element key'
79 *
80 * For example the argument constraints for bpf_map_lookup_elem():
81 * .ret_type = RET_PTR_TO_MAP_VALUE_OR_NULL,
82 * .arg1_type = ARG_CONST_MAP_PTR,
83 * .arg2_type = ARG_PTR_TO_MAP_KEY,
84 *
85 * ret_type says that this function returns 'pointer to map elem value or null'
86 * function expects 1st argument to be a const pointer to 'struct bpf_map' and
87 * 2nd argument should be a pointer to stack, which will be used inside
88 * the helper function as a pointer to map element key.
89 *
90 * On the kernel side the helper function looks like:
91 * u64 bpf_map_lookup_elem(u64 r1, u64 r2, u64 r3, u64 r4, u64 r5)
92 * {
93 * struct bpf_map *map = (struct bpf_map *) (unsigned long) r1;
94 * void *key = (void *) (unsigned long) r2;
95 * void *value;
96 *
97 * here kernel can access 'key' and 'map' pointers safely, knowing that
98 * [key, key + map->key_size) bytes are valid and were initialized on
99 * the stack of eBPF program.
100 * }
101 *
102 * Corresponding eBPF program may look like:
103 * BPF_MOV64_REG(BPF_REG_2, BPF_REG_10), // after this insn R2 type is FRAME_PTR
104 * BPF_ALU64_IMM(BPF_ADD, BPF_REG_2, -4), // after this insn R2 type is PTR_TO_STACK
105 * BPF_LD_MAP_FD(BPF_REG_1, map_fd), // after this insn R1 type is CONST_PTR_TO_MAP
106 * BPF_RAW_INSN(BPF_JMP | BPF_CALL, 0, 0, 0, BPF_FUNC_map_lookup_elem),
107 * here verifier looks at prototype of map_lookup_elem() and sees:
108 * .arg1_type == ARG_CONST_MAP_PTR and R1->type == CONST_PTR_TO_MAP, which is ok,
109 * Now verifier knows that this map has key of R1->map_ptr->key_size bytes
110 *
111 * Then .arg2_type == ARG_PTR_TO_MAP_KEY and R2->type == PTR_TO_STACK, ok so far,
112 * Now verifier checks that [R2, R2 + map's key_size) are within stack limits
113 * and were initialized prior to this call.
114 * If it's ok, then verifier allows this BPF_CALL insn and looks at
115 * .ret_type which is RET_PTR_TO_MAP_VALUE_OR_NULL, so it sets
116 * R0->type = PTR_TO_MAP_VALUE_OR_NULL which means bpf_map_lookup_elem() function
117 * returns ether pointer to map value or NULL.
118 *
119 * When type PTR_TO_MAP_VALUE_OR_NULL passes through 'if (reg != 0) goto +off'
120 * insn, the register holding that pointer in the true branch changes state to
121 * PTR_TO_MAP_VALUE and the same register changes state to CONST_IMM in the false
122 * branch. See check_cond_jmp_op().
123 *
124 * After the call R0 is set to return type of the function and registers R1-R5
125 * are set to NOT_INIT to indicate that they are no longer readable.
126 */
127
128/* types of values stored in eBPF registers */
129enum bpf_reg_type {
130 NOT_INIT = 0, /* nothing was written into register */
131 UNKNOWN_VALUE, /* reg doesn't contain a valid pointer */
132 PTR_TO_CTX, /* reg points to bpf_context */
133 CONST_PTR_TO_MAP, /* reg points to struct bpf_map */
134 PTR_TO_MAP_VALUE, /* reg points to map element value */
135 PTR_TO_MAP_VALUE_OR_NULL,/* points to map elem value or NULL */
136 FRAME_PTR, /* reg == frame_pointer */
137 PTR_TO_STACK, /* reg == frame_pointer + imm */
138 CONST_IMM, /* constant integer value */
139};
140
141struct reg_state {
142 enum bpf_reg_type type;
143 union {
144 /* valid when type == CONST_IMM | PTR_TO_STACK */
145 int imm;
146
147 /* valid when type == CONST_PTR_TO_MAP | PTR_TO_MAP_VALUE |
148 * PTR_TO_MAP_VALUE_OR_NULL
149 */
150 struct bpf_map *map_ptr;
151 };
152};
153
154enum bpf_stack_slot_type {
155 STACK_INVALID, /* nothing was stored in this stack slot */
156 STACK_SPILL, /* 1st byte of register spilled into stack */
157 STACK_SPILL_PART, /* other 7 bytes of register spill */
158 STACK_MISC /* BPF program wrote some data into this slot */
159};
160
161struct bpf_stack_slot {
162 enum bpf_stack_slot_type stype;
163 struct reg_state reg_st;
164};
165
166/* state of the program:
167 * type of all registers and stack info
168 */
169struct verifier_state {
170 struct reg_state regs[MAX_BPF_REG];
171 struct bpf_stack_slot stack[MAX_BPF_STACK];
172};
173
174/* linked list of verifier states used to prune search */
175struct verifier_state_list {
176 struct verifier_state state;
177 struct verifier_state_list *next;
178};
179
180/* verifier_state + insn_idx are pushed to stack when branch is encountered */
181struct verifier_stack_elem {
182 /* verifer state is 'st'
183 * before processing instruction 'insn_idx'
184 * and after processing instruction 'prev_insn_idx'
185 */
186 struct verifier_state st;
187 int insn_idx;
188 int prev_insn_idx;
189 struct verifier_stack_elem *next;
190};
191
192#define MAX_USED_MAPS 64 /* max number of maps accessed by one eBPF program */
193
194/* single container for all structs
195 * one verifier_env per bpf_check() call
196 */
197struct verifier_env {
198 struct bpf_prog *prog; /* eBPF program being verified */
199 struct verifier_stack_elem *head; /* stack of verifier states to be processed */
200 int stack_size; /* number of states to be processed */
201 struct verifier_state cur_state; /* current verifier state */
202 struct verifier_state_list **explored_states; /* search pruning optimization */
203 struct bpf_map *used_maps[MAX_USED_MAPS]; /* array of map's used by eBPF program */
204 u32 used_map_cnt; /* number of used maps */
205};
206
207/* verbose verifier prints what it's seeing
208 * bpf_check() is called under lock, so no race to access these global vars
209 */
210static u32 log_level, log_size, log_len;
211static char *log_buf;
212
213static DEFINE_MUTEX(bpf_verifier_lock);
214
215/* log_level controls verbosity level of eBPF verifier.
216 * verbose() is used to dump the verification trace to the log, so the user
217 * can figure out what's wrong with the program
218 */
219static void verbose(const char *fmt, ...)
220{
221 va_list args;
222
223 if (log_level == 0 || log_len >= log_size - 1)
224 return;
225
226 va_start(args, fmt);
227 log_len += vscnprintf(log_buf + log_len, log_size - log_len, fmt, args);
228 va_end(args);
229}
230
231/* string representation of 'enum bpf_reg_type' */
232static const char * const reg_type_str[] = {
233 [NOT_INIT] = "?",
234 [UNKNOWN_VALUE] = "inv",
235 [PTR_TO_CTX] = "ctx",
236 [CONST_PTR_TO_MAP] = "map_ptr",
237 [PTR_TO_MAP_VALUE] = "map_value",
238 [PTR_TO_MAP_VALUE_OR_NULL] = "map_value_or_null",
239 [FRAME_PTR] = "fp",
240 [PTR_TO_STACK] = "fp",
241 [CONST_IMM] = "imm",
242};
243
244static void print_verifier_state(struct verifier_env *env)
245{
246 enum bpf_reg_type t;
247 int i;
248
249 for (i = 0; i < MAX_BPF_REG; i++) {
250 t = env->cur_state.regs[i].type;
251 if (t == NOT_INIT)
252 continue;
253 verbose(" R%d=%s", i, reg_type_str[t]);
254 if (t == CONST_IMM || t == PTR_TO_STACK)
255 verbose("%d", env->cur_state.regs[i].imm);
256 else if (t == CONST_PTR_TO_MAP || t == PTR_TO_MAP_VALUE ||
257 t == PTR_TO_MAP_VALUE_OR_NULL)
258 verbose("(ks=%d,vs=%d)",
259 env->cur_state.regs[i].map_ptr->key_size,
260 env->cur_state.regs[i].map_ptr->value_size);
261 }
262 for (i = 0; i < MAX_BPF_STACK; i++) {
263 if (env->cur_state.stack[i].stype == STACK_SPILL)
264 verbose(" fp%d=%s", -MAX_BPF_STACK + i,
265 reg_type_str[env->cur_state.stack[i].reg_st.type]);
266 }
267 verbose("\n");
268}
269
270static const char *const bpf_class_string[] = {
271 [BPF_LD] = "ld",
272 [BPF_LDX] = "ldx",
273 [BPF_ST] = "st",
274 [BPF_STX] = "stx",
275 [BPF_ALU] = "alu",
276 [BPF_JMP] = "jmp",
277 [BPF_RET] = "BUG",
278 [BPF_ALU64] = "alu64",
279};
280
281static const char *const bpf_alu_string[] = {
282 [BPF_ADD >> 4] = "+=",
283 [BPF_SUB >> 4] = "-=",
284 [BPF_MUL >> 4] = "*=",
285 [BPF_DIV >> 4] = "/=",
286 [BPF_OR >> 4] = "|=",
287 [BPF_AND >> 4] = "&=",
288 [BPF_LSH >> 4] = "<<=",
289 [BPF_RSH >> 4] = ">>=",
290 [BPF_NEG >> 4] = "neg",
291 [BPF_MOD >> 4] = "%=",
292 [BPF_XOR >> 4] = "^=",
293 [BPF_MOV >> 4] = "=",
294 [BPF_ARSH >> 4] = "s>>=",
295 [BPF_END >> 4] = "endian",
296};
297
298static const char *const bpf_ldst_string[] = {
299 [BPF_W >> 3] = "u32",
300 [BPF_H >> 3] = "u16",
301 [BPF_B >> 3] = "u8",
302 [BPF_DW >> 3] = "u64",
303};
304
305static const char *const bpf_jmp_string[] = {
306 [BPF_JA >> 4] = "jmp",
307 [BPF_JEQ >> 4] = "==",
308 [BPF_JGT >> 4] = ">",
309 [BPF_JGE >> 4] = ">=",
310 [BPF_JSET >> 4] = "&",
311 [BPF_JNE >> 4] = "!=",
312 [BPF_JSGT >> 4] = "s>",
313 [BPF_JSGE >> 4] = "s>=",
314 [BPF_CALL >> 4] = "call",
315 [BPF_EXIT >> 4] = "exit",
316};
317
318static void print_bpf_insn(struct bpf_insn *insn)
319{
320 u8 class = BPF_CLASS(insn->code);
321
322 if (class == BPF_ALU || class == BPF_ALU64) {
323 if (BPF_SRC(insn->code) == BPF_X)
324 verbose("(%02x) %sr%d %s %sr%d\n",
325 insn->code, class == BPF_ALU ? "(u32) " : "",
326 insn->dst_reg,
327 bpf_alu_string[BPF_OP(insn->code) >> 4],
328 class == BPF_ALU ? "(u32) " : "",
329 insn->src_reg);
330 else
331 verbose("(%02x) %sr%d %s %s%d\n",
332 insn->code, class == BPF_ALU ? "(u32) " : "",
333 insn->dst_reg,
334 bpf_alu_string[BPF_OP(insn->code) >> 4],
335 class == BPF_ALU ? "(u32) " : "",
336 insn->imm);
337 } else if (class == BPF_STX) {
338 if (BPF_MODE(insn->code) == BPF_MEM)
339 verbose("(%02x) *(%s *)(r%d %+d) = r%d\n",
340 insn->code,
341 bpf_ldst_string[BPF_SIZE(insn->code) >> 3],
342 insn->dst_reg,
343 insn->off, insn->src_reg);
344 else if (BPF_MODE(insn->code) == BPF_XADD)
345 verbose("(%02x) lock *(%s *)(r%d %+d) += r%d\n",
346 insn->code,
347 bpf_ldst_string[BPF_SIZE(insn->code) >> 3],
348 insn->dst_reg, insn->off,
349 insn->src_reg);
350 else
351 verbose("BUG_%02x\n", insn->code);
352 } else if (class == BPF_ST) {
353 if (BPF_MODE(insn->code) != BPF_MEM) {
354 verbose("BUG_st_%02x\n", insn->code);
355 return;
356 }
357 verbose("(%02x) *(%s *)(r%d %+d) = %d\n",
358 insn->code,
359 bpf_ldst_string[BPF_SIZE(insn->code) >> 3],
360 insn->dst_reg,
361 insn->off, insn->imm);
362 } else if (class == BPF_LDX) {
363 if (BPF_MODE(insn->code) != BPF_MEM) {
364 verbose("BUG_ldx_%02x\n", insn->code);
365 return;
366 }
367 verbose("(%02x) r%d = *(%s *)(r%d %+d)\n",
368 insn->code, insn->dst_reg,
369 bpf_ldst_string[BPF_SIZE(insn->code) >> 3],
370 insn->src_reg, insn->off);
371 } else if (class == BPF_LD) {
372 if (BPF_MODE(insn->code) == BPF_ABS) {
373 verbose("(%02x) r0 = *(%s *)skb[%d]\n",
374 insn->code,
375 bpf_ldst_string[BPF_SIZE(insn->code) >> 3],
376 insn->imm);
377 } else if (BPF_MODE(insn->code) == BPF_IND) {
378 verbose("(%02x) r0 = *(%s *)skb[r%d + %d]\n",
379 insn->code,
380 bpf_ldst_string[BPF_SIZE(insn->code) >> 3],
381 insn->src_reg, insn->imm);
382 } else if (BPF_MODE(insn->code) == BPF_IMM) {
383 verbose("(%02x) r%d = 0x%x\n",
384 insn->code, insn->dst_reg, insn->imm);
385 } else {
386 verbose("BUG_ld_%02x\n", insn->code);
387 return;
388 }
389 } else if (class == BPF_JMP) {
390 u8 opcode = BPF_OP(insn->code);
391
392 if (opcode == BPF_CALL) {
393 verbose("(%02x) call %d\n", insn->code, insn->imm);
394 } else if (insn->code == (BPF_JMP | BPF_JA)) {
395 verbose("(%02x) goto pc%+d\n",
396 insn->code, insn->off);
397 } else if (insn->code == (BPF_JMP | BPF_EXIT)) {
398 verbose("(%02x) exit\n", insn->code);
399 } else if (BPF_SRC(insn->code) == BPF_X) {
400 verbose("(%02x) if r%d %s r%d goto pc%+d\n",
401 insn->code, insn->dst_reg,
402 bpf_jmp_string[BPF_OP(insn->code) >> 4],
403 insn->src_reg, insn->off);
404 } else {
405 verbose("(%02x) if r%d %s 0x%x goto pc%+d\n",
406 insn->code, insn->dst_reg,
407 bpf_jmp_string[BPF_OP(insn->code) >> 4],
408 insn->imm, insn->off);
409 }
410 } else {
411 verbose("(%02x) %s\n", insn->code, bpf_class_string[class]);
412 }
413}
414
415static int pop_stack(struct verifier_env *env, int *prev_insn_idx)
416{
417 struct verifier_stack_elem *elem;
418 int insn_idx;
419
420 if (env->head == NULL)
421 return -1;
422
423 memcpy(&env->cur_state, &env->head->st, sizeof(env->cur_state));
424 insn_idx = env->head->insn_idx;
425 if (prev_insn_idx)
426 *prev_insn_idx = env->head->prev_insn_idx;
427 elem = env->head->next;
428 kfree(env->head);
429 env->head = elem;
430 env->stack_size--;
431 return insn_idx;
432}
433
434static struct verifier_state *push_stack(struct verifier_env *env, int insn_idx,
435 int prev_insn_idx)
436{
437 struct verifier_stack_elem *elem;
438
439 elem = kmalloc(sizeof(struct verifier_stack_elem), GFP_KERNEL);
440 if (!elem)
441 goto err;
442
443 memcpy(&elem->st, &env->cur_state, sizeof(env->cur_state));
444 elem->insn_idx = insn_idx;
445 elem->prev_insn_idx = prev_insn_idx;
446 elem->next = env->head;
447 env->head = elem;
448 env->stack_size++;
449 if (env->stack_size > 1024) {
450 verbose("BPF program is too complex\n");
451 goto err;
452 }
453 return &elem->st;
454err:
455 /* pop all elements and return */
456 while (pop_stack(env, NULL) >= 0);
457 return NULL;
458}
459
460#define CALLER_SAVED_REGS 6
461static const int caller_saved[CALLER_SAVED_REGS] = {
462 BPF_REG_0, BPF_REG_1, BPF_REG_2, BPF_REG_3, BPF_REG_4, BPF_REG_5
463};
464
465static void init_reg_state(struct reg_state *regs)
466{
467 int i;
468
469 for (i = 0; i < MAX_BPF_REG; i++) {
470 regs[i].type = NOT_INIT;
471 regs[i].imm = 0;
472 regs[i].map_ptr = NULL;
473 }
474
475 /* frame pointer */
476 regs[BPF_REG_FP].type = FRAME_PTR;
477
478 /* 1st arg to a function */
479 regs[BPF_REG_1].type = PTR_TO_CTX;
480}
481
482static void mark_reg_unknown_value(struct reg_state *regs, u32 regno)
483{
484 BUG_ON(regno >= MAX_BPF_REG);
485 regs[regno].type = UNKNOWN_VALUE;
486 regs[regno].imm = 0;
487 regs[regno].map_ptr = NULL;
488}
489
490enum reg_arg_type {
491 SRC_OP, /* register is used as source operand */
492 DST_OP, /* register is used as destination operand */
493 DST_OP_NO_MARK /* same as above, check only, don't mark */
494};
495
496static int check_reg_arg(struct reg_state *regs, u32 regno,
497 enum reg_arg_type t)
498{
499 if (regno >= MAX_BPF_REG) {
500 verbose("R%d is invalid\n", regno);
501 return -EINVAL;
502 }
503
504 if (t == SRC_OP) {
505 /* check whether register used as source operand can be read */
506 if (regs[regno].type == NOT_INIT) {
507 verbose("R%d !read_ok\n", regno);
508 return -EACCES;
509 }
510 } else {
511 /* check whether register used as dest operand can be written to */
512 if (regno == BPF_REG_FP) {
513 verbose("frame pointer is read only\n");
514 return -EACCES;
515 }
516 if (t == DST_OP)
517 mark_reg_unknown_value(regs, regno);
518 }
519 return 0;
520}
521
522static int bpf_size_to_bytes(int bpf_size)
523{
524 if (bpf_size == BPF_W)
525 return 4;
526 else if (bpf_size == BPF_H)
527 return 2;
528 else if (bpf_size == BPF_B)
529 return 1;
530 else if (bpf_size == BPF_DW)
531 return 8;
532 else
533 return -EINVAL;
534}
535
536/* check_stack_read/write functions track spill/fill of registers,
537 * stack boundary and alignment are checked in check_mem_access()
538 */
539static int check_stack_write(struct verifier_state *state, int off, int size,
540 int value_regno)
541{
542 struct bpf_stack_slot *slot;
543 int i;
544
545 if (value_regno >= 0 &&
546 (state->regs[value_regno].type == PTR_TO_MAP_VALUE ||
547 state->regs[value_regno].type == PTR_TO_STACK ||
548 state->regs[value_regno].type == PTR_TO_CTX)) {
549
550 /* register containing pointer is being spilled into stack */
551 if (size != 8) {
552 verbose("invalid size of register spill\n");
553 return -EACCES;
554 }
555
556 slot = &state->stack[MAX_BPF_STACK + off];
557 slot->stype = STACK_SPILL;
558 /* save register state */
559 slot->reg_st = state->regs[value_regno];
560 for (i = 1; i < 8; i++) {
561 slot = &state->stack[MAX_BPF_STACK + off + i];
562 slot->stype = STACK_SPILL_PART;
563 slot->reg_st.type = UNKNOWN_VALUE;
564 slot->reg_st.map_ptr = NULL;
565 }
566 } else {
567
568 /* regular write of data into stack */
569 for (i = 0; i < size; i++) {
570 slot = &state->stack[MAX_BPF_STACK + off + i];
571 slot->stype = STACK_MISC;
572 slot->reg_st.type = UNKNOWN_VALUE;
573 slot->reg_st.map_ptr = NULL;
574 }
575 }
576 return 0;
577}
578
579static int check_stack_read(struct verifier_state *state, int off, int size,
580 int value_regno)
581{
582 int i;
583 struct bpf_stack_slot *slot;
584
585 slot = &state->stack[MAX_BPF_STACK + off];
586
587 if (slot->stype == STACK_SPILL) {
588 if (size != 8) {
589 verbose("invalid size of register spill\n");
590 return -EACCES;
591 }
592 for (i = 1; i < 8; i++) {
593 if (state->stack[MAX_BPF_STACK + off + i].stype !=
594 STACK_SPILL_PART) {
595 verbose("corrupted spill memory\n");
596 return -EACCES;
597 }
598 }
599
600 if (value_regno >= 0)
601 /* restore register state from stack */
602 state->regs[value_regno] = slot->reg_st;
603 return 0;
604 } else {
605 for (i = 0; i < size; i++) {
606 if (state->stack[MAX_BPF_STACK + off + i].stype !=
607 STACK_MISC) {
608 verbose("invalid read from stack off %d+%d size %d\n",
609 off, i, size);
610 return -EACCES;
611 }
612 }
613 if (value_regno >= 0)
614 /* have read misc data from the stack */
615 mark_reg_unknown_value(state->regs, value_regno);
616 return 0;
617 }
618}
619
620/* check read/write into map element returned by bpf_map_lookup_elem() */
621static int check_map_access(struct verifier_env *env, u32 regno, int off,
622 int size)
623{
624 struct bpf_map *map = env->cur_state.regs[regno].map_ptr;
625
626 if (off < 0 || off + size > map->value_size) {
627 verbose("invalid access to map value, value_size=%d off=%d size=%d\n",
628 map->value_size, off, size);
629 return -EACCES;
630 }
631 return 0;
632}
633
634/* check access to 'struct bpf_context' fields */
635static int check_ctx_access(struct verifier_env *env, int off, int size,
636 enum bpf_access_type t)
637{
638 if (env->prog->aux->ops->is_valid_access &&
639 env->prog->aux->ops->is_valid_access(off, size, t))
640 return 0;
641
642 verbose("invalid bpf_context access off=%d size=%d\n", off, size);
643 return -EACCES;
644}
645
646/* check whether memory at (regno + off) is accessible for t = (read | write)
647 * if t==write, value_regno is a register which value is stored into memory
648 * if t==read, value_regno is a register which will receive the value from memory
649 * if t==write && value_regno==-1, some unknown value is stored into memory
650 * if t==read && value_regno==-1, don't care what we read from memory
651 */
652static int check_mem_access(struct verifier_env *env, u32 regno, int off,
653 int bpf_size, enum bpf_access_type t,
654 int value_regno)
655{
656 struct verifier_state *state = &env->cur_state;
657 int size, err = 0;
658
659 size = bpf_size_to_bytes(bpf_size);
660 if (size < 0)
661 return size;
662
663 if (off % size != 0) {
664 verbose("misaligned access off %d size %d\n", off, size);
665 return -EACCES;
666 }
667
668 if (state->regs[regno].type == PTR_TO_MAP_VALUE) {
669 err = check_map_access(env, regno, off, size);
670 if (!err && t == BPF_READ && value_regno >= 0)
671 mark_reg_unknown_value(state->regs, value_regno);
672
673 } else if (state->regs[regno].type == PTR_TO_CTX) {
674 err = check_ctx_access(env, off, size, t);
675 if (!err && t == BPF_READ && value_regno >= 0)
676 mark_reg_unknown_value(state->regs, value_regno);
677
678 } else if (state->regs[regno].type == FRAME_PTR) {
679 if (off >= 0 || off < -MAX_BPF_STACK) {
680 verbose("invalid stack off=%d size=%d\n", off, size);
681 return -EACCES;
682 }
683 if (t == BPF_WRITE)
684 err = check_stack_write(state, off, size, value_regno);
685 else
686 err = check_stack_read(state, off, size, value_regno);
687 } else {
688 verbose("R%d invalid mem access '%s'\n",
689 regno, reg_type_str[state->regs[regno].type]);
690 return -EACCES;
691 }
692 return err;
693}
694
695static int check_xadd(struct verifier_env *env, struct bpf_insn *insn)
696{
697 struct reg_state *regs = env->cur_state.regs;
698 int err;
699
700 if ((BPF_SIZE(insn->code) != BPF_W && BPF_SIZE(insn->code) != BPF_DW) ||
701 insn->imm != 0) {
702 verbose("BPF_XADD uses reserved fields\n");
703 return -EINVAL;
704 }
705
706 /* check src1 operand */
707 err = check_reg_arg(regs, insn->src_reg, SRC_OP);
708 if (err)
709 return err;
710
711 /* check src2 operand */
712 err = check_reg_arg(regs, insn->dst_reg, SRC_OP);
713 if (err)
714 return err;
715
716 /* check whether atomic_add can read the memory */
717 err = check_mem_access(env, insn->dst_reg, insn->off,
718 BPF_SIZE(insn->code), BPF_READ, -1);
719 if (err)
720 return err;
721
722 /* check whether atomic_add can write into the same memory */
723 return check_mem_access(env, insn->dst_reg, insn->off,
724 BPF_SIZE(insn->code), BPF_WRITE, -1);
725}
726
727/* when register 'regno' is passed into function that will read 'access_size'
728 * bytes from that pointer, make sure that it's within stack boundary
729 * and all elements of stack are initialized
730 */
731static int check_stack_boundary(struct verifier_env *env,
732 int regno, int access_size)
733{
734 struct verifier_state *state = &env->cur_state;
735 struct reg_state *regs = state->regs;
736 int off, i;
737
738 if (regs[regno].type != PTR_TO_STACK)
739 return -EACCES;
740
741 off = regs[regno].imm;
742 if (off >= 0 || off < -MAX_BPF_STACK || off + access_size > 0 ||
743 access_size <= 0) {
744 verbose("invalid stack type R%d off=%d access_size=%d\n",
745 regno, off, access_size);
746 return -EACCES;
747 }
748
749 for (i = 0; i < access_size; i++) {
750 if (state->stack[MAX_BPF_STACK + off + i].stype != STACK_MISC) {
751 verbose("invalid indirect read from stack off %d+%d size %d\n",
752 off, i, access_size);
753 return -EACCES;
754 }
755 }
756 return 0;
757}
758
759static int check_func_arg(struct verifier_env *env, u32 regno,
760 enum bpf_arg_type arg_type, struct bpf_map **mapp)
761{
762 struct reg_state *reg = env->cur_state.regs + regno;
763 enum bpf_reg_type expected_type;
764 int err = 0;
765
766 if (arg_type == ARG_ANYTHING)
767 return 0;
768
769 if (reg->type == NOT_INIT) {
770 verbose("R%d !read_ok\n", regno);
771 return -EACCES;
772 }
773
774 if (arg_type == ARG_PTR_TO_STACK || arg_type == ARG_PTR_TO_MAP_KEY ||
775 arg_type == ARG_PTR_TO_MAP_VALUE) {
776 expected_type = PTR_TO_STACK;
777 } else if (arg_type == ARG_CONST_STACK_SIZE) {
778 expected_type = CONST_IMM;
779 } else if (arg_type == ARG_CONST_MAP_PTR) {
780 expected_type = CONST_PTR_TO_MAP;
781 } else {
782 verbose("unsupported arg_type %d\n", arg_type);
783 return -EFAULT;
784 }
785
786 if (reg->type != expected_type) {
787 verbose("R%d type=%s expected=%s\n", regno,
788 reg_type_str[reg->type], reg_type_str[expected_type]);
789 return -EACCES;
790 }
791
792 if (arg_type == ARG_CONST_MAP_PTR) {
793 /* bpf_map_xxx(map_ptr) call: remember that map_ptr */
794 *mapp = reg->map_ptr;
795
796 } else if (arg_type == ARG_PTR_TO_MAP_KEY) {
797 /* bpf_map_xxx(..., map_ptr, ..., key) call:
798 * check that [key, key + map->key_size) are within
799 * stack limits and initialized
800 */
801 if (!*mapp) {
802 /* in function declaration map_ptr must come before
803 * map_key, so that it's verified and known before
804 * we have to check map_key here. Otherwise it means
805 * that kernel subsystem misconfigured verifier
806 */
807 verbose("invalid map_ptr to access map->key\n");
808 return -EACCES;
809 }
810 err = check_stack_boundary(env, regno, (*mapp)->key_size);
811
812 } else if (arg_type == ARG_PTR_TO_MAP_VALUE) {
813 /* bpf_map_xxx(..., map_ptr, ..., value) call:
814 * check [value, value + map->value_size) validity
815 */
816 if (!*mapp) {
817 /* kernel subsystem misconfigured verifier */
818 verbose("invalid map_ptr to access map->value\n");
819 return -EACCES;
820 }
821 err = check_stack_boundary(env, regno, (*mapp)->value_size);
822
823 } else if (arg_type == ARG_CONST_STACK_SIZE) {
824 /* bpf_xxx(..., buf, len) call will access 'len' bytes
825 * from stack pointer 'buf'. Check it
826 * note: regno == len, regno - 1 == buf
827 */
828 if (regno == 0) {
829 /* kernel subsystem misconfigured verifier */
830 verbose("ARG_CONST_STACK_SIZE cannot be first argument\n");
831 return -EACCES;
832 }
833 err = check_stack_boundary(env, regno - 1, reg->imm);
834 }
835
836 return err;
837}
838
839static int check_call(struct verifier_env *env, int func_id)
840{
841 struct verifier_state *state = &env->cur_state;
842 const struct bpf_func_proto *fn = NULL;
843 struct reg_state *regs = state->regs;
844 struct bpf_map *map = NULL;
845 struct reg_state *reg;
846 int i, err;
847
848 /* find function prototype */
849 if (func_id < 0 || func_id >= __BPF_FUNC_MAX_ID) {
850 verbose("invalid func %d\n", func_id);
851 return -EINVAL;
852 }
853
854 if (env->prog->aux->ops->get_func_proto)
855 fn = env->prog->aux->ops->get_func_proto(func_id);
856
857 if (!fn) {
858 verbose("unknown func %d\n", func_id);
859 return -EINVAL;
860 }
861
862 /* eBPF programs must be GPL compatible to use GPL-ed functions */
863 if (!env->prog->aux->is_gpl_compatible && fn->gpl_only) {
864 verbose("cannot call GPL only function from proprietary program\n");
865 return -EINVAL;
866 }
867
868 /* check args */
869 err = check_func_arg(env, BPF_REG_1, fn->arg1_type, &map);
870 if (err)
871 return err;
872 err = check_func_arg(env, BPF_REG_2, fn->arg2_type, &map);
873 if (err)
874 return err;
875 err = check_func_arg(env, BPF_REG_3, fn->arg3_type, &map);
876 if (err)
877 return err;
878 err = check_func_arg(env, BPF_REG_4, fn->arg4_type, &map);
879 if (err)
880 return err;
881 err = check_func_arg(env, BPF_REG_5, fn->arg5_type, &map);
882 if (err)
883 return err;
884
885 /* reset caller saved regs */
886 for (i = 0; i < CALLER_SAVED_REGS; i++) {
887 reg = regs + caller_saved[i];
888 reg->type = NOT_INIT;
889 reg->imm = 0;
890 }
891
892 /* update return register */
893 if (fn->ret_type == RET_INTEGER) {
894 regs[BPF_REG_0].type = UNKNOWN_VALUE;
895 } else if (fn->ret_type == RET_VOID) {
896 regs[BPF_REG_0].type = NOT_INIT;
897 } else if (fn->ret_type == RET_PTR_TO_MAP_VALUE_OR_NULL) {
898 regs[BPF_REG_0].type = PTR_TO_MAP_VALUE_OR_NULL;
899 /* remember map_ptr, so that check_map_access()
900 * can check 'value_size' boundary of memory access
901 * to map element returned from bpf_map_lookup_elem()
902 */
903 if (map == NULL) {
904 verbose("kernel subsystem misconfigured verifier\n");
905 return -EINVAL;
906 }
907 regs[BPF_REG_0].map_ptr = map;
908 } else {
909 verbose("unknown return type %d of func %d\n",
910 fn->ret_type, func_id);
911 return -EINVAL;
912 }
913 return 0;
914}
915
916/* check validity of 32-bit and 64-bit arithmetic operations */
917static int check_alu_op(struct reg_state *regs, struct bpf_insn *insn)
918{
919 u8 opcode = BPF_OP(insn->code);
920 int err;
921
922 if (opcode == BPF_END || opcode == BPF_NEG) {
923 if (opcode == BPF_NEG) {
924 if (BPF_SRC(insn->code) != 0 ||
925 insn->src_reg != BPF_REG_0 ||
926 insn->off != 0 || insn->imm != 0) {
927 verbose("BPF_NEG uses reserved fields\n");
928 return -EINVAL;
929 }
930 } else {
931 if (insn->src_reg != BPF_REG_0 || insn->off != 0 ||
932 (insn->imm != 16 && insn->imm != 32 && insn->imm != 64)) {
933 verbose("BPF_END uses reserved fields\n");
934 return -EINVAL;
935 }
936 }
937
938 /* check src operand */
939 err = check_reg_arg(regs, insn->dst_reg, SRC_OP);
940 if (err)
941 return err;
942
943 /* check dest operand */
944 err = check_reg_arg(regs, insn->dst_reg, DST_OP);
945 if (err)
946 return err;
947
948 } else if (opcode == BPF_MOV) {
949
950 if (BPF_SRC(insn->code) == BPF_X) {
951 if (insn->imm != 0 || insn->off != 0) {
952 verbose("BPF_MOV uses reserved fields\n");
953 return -EINVAL;
954 }
955
956 /* check src operand */
957 err = check_reg_arg(regs, insn->src_reg, SRC_OP);
958 if (err)
959 return err;
960 } else {
961 if (insn->src_reg != BPF_REG_0 || insn->off != 0) {
962 verbose("BPF_MOV uses reserved fields\n");
963 return -EINVAL;
964 }
965 }
966
967 /* check dest operand */
968 err = check_reg_arg(regs, insn->dst_reg, DST_OP);
969 if (err)
970 return err;
971
972 if (BPF_SRC(insn->code) == BPF_X) {
973 if (BPF_CLASS(insn->code) == BPF_ALU64) {
974 /* case: R1 = R2
975 * copy register state to dest reg
976 */
977 regs[insn->dst_reg] = regs[insn->src_reg];
978 } else {
979 regs[insn->dst_reg].type = UNKNOWN_VALUE;
980 regs[insn->dst_reg].map_ptr = NULL;
981 }
982 } else {
983 /* case: R = imm
984 * remember the value we stored into this reg
985 */
986 regs[insn->dst_reg].type = CONST_IMM;
987 regs[insn->dst_reg].imm = insn->imm;
988 }
989
990 } else if (opcode > BPF_END) {
991 verbose("invalid BPF_ALU opcode %x\n", opcode);
992 return -EINVAL;
993
994 } else { /* all other ALU ops: and, sub, xor, add, ... */
995
996 bool stack_relative = false;
997
998 if (BPF_SRC(insn->code) == BPF_X) {
999 if (insn->imm != 0 || insn->off != 0) {
1000 verbose("BPF_ALU uses reserved fields\n");
1001 return -EINVAL;
1002 }
1003 /* check src1 operand */
1004 err = check_reg_arg(regs, insn->src_reg, SRC_OP);
1005 if (err)
1006 return err;
1007 } else {
1008 if (insn->src_reg != BPF_REG_0 || insn->off != 0) {
1009 verbose("BPF_ALU uses reserved fields\n");
1010 return -EINVAL;
1011 }
1012 }
1013
1014 /* check src2 operand */
1015 err = check_reg_arg(regs, insn->dst_reg, SRC_OP);
1016 if (err)
1017 return err;
1018
1019 if ((opcode == BPF_MOD || opcode == BPF_DIV) &&
1020 BPF_SRC(insn->code) == BPF_K && insn->imm == 0) {
1021 verbose("div by zero\n");
1022 return -EINVAL;
1023 }
1024
1025 /* pattern match 'bpf_add Rx, imm' instruction */
1026 if (opcode == BPF_ADD && BPF_CLASS(insn->code) == BPF_ALU64 &&
1027 regs[insn->dst_reg].type == FRAME_PTR &&
1028 BPF_SRC(insn->code) == BPF_K)
1029 stack_relative = true;
1030
1031 /* check dest operand */
1032 err = check_reg_arg(regs, insn->dst_reg, DST_OP);
1033 if (err)
1034 return err;
1035
1036 if (stack_relative) {
1037 regs[insn->dst_reg].type = PTR_TO_STACK;
1038 regs[insn->dst_reg].imm = insn->imm;
1039 }
1040 }
1041
1042 return 0;
1043}
1044
1045static int check_cond_jmp_op(struct verifier_env *env,
1046 struct bpf_insn *insn, int *insn_idx)
1047{
1048 struct reg_state *regs = env->cur_state.regs;
1049 struct verifier_state *other_branch;
1050 u8 opcode = BPF_OP(insn->code);
1051 int err;
1052
1053 if (opcode > BPF_EXIT) {
1054 verbose("invalid BPF_JMP opcode %x\n", opcode);
1055 return -EINVAL;
1056 }
1057
1058 if (BPF_SRC(insn->code) == BPF_X) {
1059 if (insn->imm != 0) {
1060 verbose("BPF_JMP uses reserved fields\n");
1061 return -EINVAL;
1062 }
1063
1064 /* check src1 operand */
1065 err = check_reg_arg(regs, insn->src_reg, SRC_OP);
1066 if (err)
1067 return err;
1068 } else {
1069 if (insn->src_reg != BPF_REG_0) {
1070 verbose("BPF_JMP uses reserved fields\n");
1071 return -EINVAL;
1072 }
1073 }
1074
1075 /* check src2 operand */
1076 err = check_reg_arg(regs, insn->dst_reg, SRC_OP);
1077 if (err)
1078 return err;
1079
1080 /* detect if R == 0 where R was initialized to zero earlier */
1081 if (BPF_SRC(insn->code) == BPF_K &&
1082 (opcode == BPF_JEQ || opcode == BPF_JNE) &&
1083 regs[insn->dst_reg].type == CONST_IMM &&
1084 regs[insn->dst_reg].imm == insn->imm) {
1085 if (opcode == BPF_JEQ) {
1086 /* if (imm == imm) goto pc+off;
1087 * only follow the goto, ignore fall-through
1088 */
1089 *insn_idx += insn->off;
1090 return 0;
1091 } else {
1092 /* if (imm != imm) goto pc+off;
1093 * only follow fall-through branch, since
1094 * that's where the program will go
1095 */
1096 return 0;
1097 }
1098 }
1099
1100 other_branch = push_stack(env, *insn_idx + insn->off + 1, *insn_idx);
1101 if (!other_branch)
1102 return -EFAULT;
1103
1104 /* detect if R == 0 where R is returned value from bpf_map_lookup_elem() */
1105 if (BPF_SRC(insn->code) == BPF_K &&
1106 insn->imm == 0 && (opcode == BPF_JEQ ||
1107 opcode == BPF_JNE) &&
1108 regs[insn->dst_reg].type == PTR_TO_MAP_VALUE_OR_NULL) {
1109 if (opcode == BPF_JEQ) {
1110 /* next fallthrough insn can access memory via
1111 * this register
1112 */
1113 regs[insn->dst_reg].type = PTR_TO_MAP_VALUE;
1114 /* branch targer cannot access it, since reg == 0 */
1115 other_branch->regs[insn->dst_reg].type = CONST_IMM;
1116 other_branch->regs[insn->dst_reg].imm = 0;
1117 } else {
1118 other_branch->regs[insn->dst_reg].type = PTR_TO_MAP_VALUE;
1119 regs[insn->dst_reg].type = CONST_IMM;
1120 regs[insn->dst_reg].imm = 0;
1121 }
1122 } else if (BPF_SRC(insn->code) == BPF_K &&
1123 (opcode == BPF_JEQ || opcode == BPF_JNE)) {
1124
1125 if (opcode == BPF_JEQ) {
1126 /* detect if (R == imm) goto
1127 * and in the target state recognize that R = imm
1128 */
1129 other_branch->regs[insn->dst_reg].type = CONST_IMM;
1130 other_branch->regs[insn->dst_reg].imm = insn->imm;
1131 } else {
1132 /* detect if (R != imm) goto
1133 * and in the fall-through state recognize that R = imm
1134 */
1135 regs[insn->dst_reg].type = CONST_IMM;
1136 regs[insn->dst_reg].imm = insn->imm;
1137 }
1138 }
1139 if (log_level)
1140 print_verifier_state(env);
1141 return 0;
1142}
1143
1144/* return the map pointer stored inside BPF_LD_IMM64 instruction */
1145static struct bpf_map *ld_imm64_to_map_ptr(struct bpf_insn *insn)
1146{
1147 u64 imm64 = ((u64) (u32) insn[0].imm) | ((u64) (u32) insn[1].imm) << 32;
1148
1149 return (struct bpf_map *) (unsigned long) imm64;
1150}
1151
1152/* verify BPF_LD_IMM64 instruction */
1153static int check_ld_imm(struct verifier_env *env, struct bpf_insn *insn)
1154{
1155 struct reg_state *regs = env->cur_state.regs;
1156 int err;
1157
1158 if (BPF_SIZE(insn->code) != BPF_DW) {
1159 verbose("invalid BPF_LD_IMM insn\n");
1160 return -EINVAL;
1161 }
1162 if (insn->off != 0) {
1163 verbose("BPF_LD_IMM64 uses reserved fields\n");
1164 return -EINVAL;
1165 }
1166
1167 err = check_reg_arg(regs, insn->dst_reg, DST_OP);
1168 if (err)
1169 return err;
1170
1171 if (insn->src_reg == 0)
1172 /* generic move 64-bit immediate into a register */
1173 return 0;
1174
1175 /* replace_map_fd_with_map_ptr() should have caught bad ld_imm64 */
1176 BUG_ON(insn->src_reg != BPF_PSEUDO_MAP_FD);
1177
1178 regs[insn->dst_reg].type = CONST_PTR_TO_MAP;
1179 regs[insn->dst_reg].map_ptr = ld_imm64_to_map_ptr(insn);
1180 return 0;
1181}
1182
1183/* non-recursive DFS pseudo code
1184 * 1 procedure DFS-iterative(G,v):
1185 * 2 label v as discovered
1186 * 3 let S be a stack
1187 * 4 S.push(v)
1188 * 5 while S is not empty
1189 * 6 t <- S.pop()
1190 * 7 if t is what we're looking for:
1191 * 8 return t
1192 * 9 for all edges e in G.adjacentEdges(t) do
1193 * 10 if edge e is already labelled
1194 * 11 continue with the next edge
1195 * 12 w <- G.adjacentVertex(t,e)
1196 * 13 if vertex w is not discovered and not explored
1197 * 14 label e as tree-edge
1198 * 15 label w as discovered
1199 * 16 S.push(w)
1200 * 17 continue at 5
1201 * 18 else if vertex w is discovered
1202 * 19 label e as back-edge
1203 * 20 else
1204 * 21 // vertex w is explored
1205 * 22 label e as forward- or cross-edge
1206 * 23 label t as explored
1207 * 24 S.pop()
1208 *
1209 * convention:
1210 * 0x10 - discovered
1211 * 0x11 - discovered and fall-through edge labelled
1212 * 0x12 - discovered and fall-through and branch edges labelled
1213 * 0x20 - explored
1214 */
1215
1216enum {
1217 DISCOVERED = 0x10,
1218 EXPLORED = 0x20,
1219 FALLTHROUGH = 1,
1220 BRANCH = 2,
1221};
1222
1223#define STATE_LIST_MARK ((struct verifier_state_list *) -1L)
1224
1225static int *insn_stack; /* stack of insns to process */
1226static int cur_stack; /* current stack index */
1227static int *insn_state;
1228
1229/* t, w, e - match pseudo-code above:
1230 * t - index of current instruction
1231 * w - next instruction
1232 * e - edge
1233 */
1234static int push_insn(int t, int w, int e, struct verifier_env *env)
1235{
1236 if (e == FALLTHROUGH && insn_state[t] >= (DISCOVERED | FALLTHROUGH))
1237 return 0;
1238
1239 if (e == BRANCH && insn_state[t] >= (DISCOVERED | BRANCH))
1240 return 0;
1241
1242 if (w < 0 || w >= env->prog->len) {
1243 verbose("jump out of range from insn %d to %d\n", t, w);
1244 return -EINVAL;
1245 }
1246
1247 if (e == BRANCH)
1248 /* mark branch target for state pruning */
1249 env->explored_states[w] = STATE_LIST_MARK;
1250
1251 if (insn_state[w] == 0) {
1252 /* tree-edge */
1253 insn_state[t] = DISCOVERED | e;
1254 insn_state[w] = DISCOVERED;
1255 if (cur_stack >= env->prog->len)
1256 return -E2BIG;
1257 insn_stack[cur_stack++] = w;
1258 return 1;
1259 } else if ((insn_state[w] & 0xF0) == DISCOVERED) {
1260 verbose("back-edge from insn %d to %d\n", t, w);
1261 return -EINVAL;
1262 } else if (insn_state[w] == EXPLORED) {
1263 /* forward- or cross-edge */
1264 insn_state[t] = DISCOVERED | e;
1265 } else {
1266 verbose("insn state internal bug\n");
1267 return -EFAULT;
1268 }
1269 return 0;
1270}
1271
1272/* non-recursive depth-first-search to detect loops in BPF program
1273 * loop == back-edge in directed graph
1274 */
1275static int check_cfg(struct verifier_env *env)
1276{
1277 struct bpf_insn *insns = env->prog->insnsi;
1278 int insn_cnt = env->prog->len;
1279 int ret = 0;
1280 int i, t;
1281
1282 insn_state = kcalloc(insn_cnt, sizeof(int), GFP_KERNEL);
1283 if (!insn_state)
1284 return -ENOMEM;
1285
1286 insn_stack = kcalloc(insn_cnt, sizeof(int), GFP_KERNEL);
1287 if (!insn_stack) {
1288 kfree(insn_state);
1289 return -ENOMEM;
1290 }
1291
1292 insn_state[0] = DISCOVERED; /* mark 1st insn as discovered */
1293 insn_stack[0] = 0; /* 0 is the first instruction */
1294 cur_stack = 1;
1295
1296peek_stack:
1297 if (cur_stack == 0)
1298 goto check_state;
1299 t = insn_stack[cur_stack - 1];
1300
1301 if (BPF_CLASS(insns[t].code) == BPF_JMP) {
1302 u8 opcode = BPF_OP(insns[t].code);
1303
1304 if (opcode == BPF_EXIT) {
1305 goto mark_explored;
1306 } else if (opcode == BPF_CALL) {
1307 ret = push_insn(t, t + 1, FALLTHROUGH, env);
1308 if (ret == 1)
1309 goto peek_stack;
1310 else if (ret < 0)
1311 goto err_free;
1312 } else if (opcode == BPF_JA) {
1313 if (BPF_SRC(insns[t].code) != BPF_K) {
1314 ret = -EINVAL;
1315 goto err_free;
1316 }
1317 /* unconditional jump with single edge */
1318 ret = push_insn(t, t + insns[t].off + 1,
1319 FALLTHROUGH, env);
1320 if (ret == 1)
1321 goto peek_stack;
1322 else if (ret < 0)
1323 goto err_free;
1324 /* tell verifier to check for equivalent states
1325 * after every call and jump
1326 */
1327 env->explored_states[t + 1] = STATE_LIST_MARK;
1328 } else {
1329 /* conditional jump with two edges */
1330 ret = push_insn(t, t + 1, FALLTHROUGH, env);
1331 if (ret == 1)
1332 goto peek_stack;
1333 else if (ret < 0)
1334 goto err_free;
1335
1336 ret = push_insn(t, t + insns[t].off + 1, BRANCH, env);
1337 if (ret == 1)
1338 goto peek_stack;
1339 else if (ret < 0)
1340 goto err_free;
1341 }
1342 } else {
1343 /* all other non-branch instructions with single
1344 * fall-through edge
1345 */
1346 ret = push_insn(t, t + 1, FALLTHROUGH, env);
1347 if (ret == 1)
1348 goto peek_stack;
1349 else if (ret < 0)
1350 goto err_free;
1351 }
1352
1353mark_explored:
1354 insn_state[t] = EXPLORED;
1355 if (cur_stack-- <= 0) {
1356 verbose("pop stack internal bug\n");
1357 ret = -EFAULT;
1358 goto err_free;
1359 }
1360 goto peek_stack;
1361
1362check_state:
1363 for (i = 0; i < insn_cnt; i++) {
1364 if (insn_state[i] != EXPLORED) {
1365 verbose("unreachable insn %d\n", i);
1366 ret = -EINVAL;
1367 goto err_free;
1368 }
1369 }
1370 ret = 0; /* cfg looks good */
1371
1372err_free:
1373 kfree(insn_state);
1374 kfree(insn_stack);
1375 return ret;
1376}
1377
1378/* compare two verifier states
1379 *
1380 * all states stored in state_list are known to be valid, since
1381 * verifier reached 'bpf_exit' instruction through them
1382 *
1383 * this function is called when verifier exploring different branches of
1384 * execution popped from the state stack. If it sees an old state that has
1385 * more strict register state and more strict stack state then this execution
1386 * branch doesn't need to be explored further, since verifier already
1387 * concluded that more strict state leads to valid finish.
1388 *
1389 * Therefore two states are equivalent if register state is more conservative
1390 * and explored stack state is more conservative than the current one.
1391 * Example:
1392 * explored current
1393 * (slot1=INV slot2=MISC) == (slot1=MISC slot2=MISC)
1394 * (slot1=MISC slot2=MISC) != (slot1=INV slot2=MISC)
1395 *
1396 * In other words if current stack state (one being explored) has more
1397 * valid slots than old one that already passed validation, it means
1398 * the verifier can stop exploring and conclude that current state is valid too
1399 *
1400 * Similarly with registers. If explored state has register type as invalid
1401 * whereas register type in current state is meaningful, it means that
1402 * the current state will reach 'bpf_exit' instruction safely
1403 */
1404static bool states_equal(struct verifier_state *old, struct verifier_state *cur)
1405{
1406 int i;
1407
1408 for (i = 0; i < MAX_BPF_REG; i++) {
1409 if (memcmp(&old->regs[i], &cur->regs[i],
1410 sizeof(old->regs[0])) != 0) {
1411 if (old->regs[i].type == NOT_INIT ||
1412 (old->regs[i].type == UNKNOWN_VALUE &&
1413 cur->regs[i].type != NOT_INIT))
1414 continue;
1415 return false;
1416 }
1417 }
1418
1419 for (i = 0; i < MAX_BPF_STACK; i++) {
1420 if (memcmp(&old->stack[i], &cur->stack[i],
1421 sizeof(old->stack[0])) != 0) {
1422 if (old->stack[i].stype == STACK_INVALID)
1423 continue;
1424 return false;
1425 }
1426 }
1427 return true;
1428}
1429
1430static int is_state_visited(struct verifier_env *env, int insn_idx)
1431{
1432 struct verifier_state_list *new_sl;
1433 struct verifier_state_list *sl;
1434
1435 sl = env->explored_states[insn_idx];
1436 if (!sl)
1437 /* this 'insn_idx' instruction wasn't marked, so we will not
1438 * be doing state search here
1439 */
1440 return 0;
1441
1442 while (sl != STATE_LIST_MARK) {
1443 if (states_equal(&sl->state, &env->cur_state))
1444 /* reached equivalent register/stack state,
1445 * prune the search
1446 */
1447 return 1;
1448 sl = sl->next;
1449 }
1450
1451 /* there were no equivalent states, remember current one.
1452 * technically the current state is not proven to be safe yet,
1453 * but it will either reach bpf_exit (which means it's safe) or
1454 * it will be rejected. Since there are no loops, we won't be
1455 * seeing this 'insn_idx' instruction again on the way to bpf_exit
1456 */
1457 new_sl = kmalloc(sizeof(struct verifier_state_list), GFP_USER);
1458 if (!new_sl)
1459 return -ENOMEM;
1460
1461 /* add new state to the head of linked list */
1462 memcpy(&new_sl->state, &env->cur_state, sizeof(env->cur_state));
1463 new_sl->next = env->explored_states[insn_idx];
1464 env->explored_states[insn_idx] = new_sl;
1465 return 0;
1466}
1467
1468static int do_check(struct verifier_env *env)
1469{
1470 struct verifier_state *state = &env->cur_state;
1471 struct bpf_insn *insns = env->prog->insnsi;
1472 struct reg_state *regs = state->regs;
1473 int insn_cnt = env->prog->len;
1474 int insn_idx, prev_insn_idx = 0;
1475 int insn_processed = 0;
1476 bool do_print_state = false;
1477
1478 init_reg_state(regs);
1479 insn_idx = 0;
1480 for (;;) {
1481 struct bpf_insn *insn;
1482 u8 class;
1483 int err;
1484
1485 if (insn_idx >= insn_cnt) {
1486 verbose("invalid insn idx %d insn_cnt %d\n",
1487 insn_idx, insn_cnt);
1488 return -EFAULT;
1489 }
1490
1491 insn = &insns[insn_idx];
1492 class = BPF_CLASS(insn->code);
1493
1494 if (++insn_processed > 32768) {
1495 verbose("BPF program is too large. Proccessed %d insn\n",
1496 insn_processed);
1497 return -E2BIG;
1498 }
1499
1500 err = is_state_visited(env, insn_idx);
1501 if (err < 0)
1502 return err;
1503 if (err == 1) {
1504 /* found equivalent state, can prune the search */
1505 if (log_level) {
1506 if (do_print_state)
1507 verbose("\nfrom %d to %d: safe\n",
1508 prev_insn_idx, insn_idx);
1509 else
1510 verbose("%d: safe\n", insn_idx);
1511 }
1512 goto process_bpf_exit;
1513 }
1514
1515 if (log_level && do_print_state) {
1516 verbose("\nfrom %d to %d:", prev_insn_idx, insn_idx);
1517 print_verifier_state(env);
1518 do_print_state = false;
1519 }
1520
1521 if (log_level) {
1522 verbose("%d: ", insn_idx);
1523 print_bpf_insn(insn);
1524 }
1525
1526 if (class == BPF_ALU || class == BPF_ALU64) {
1527 err = check_alu_op(regs, insn);
1528 if (err)
1529 return err;
1530
1531 } else if (class == BPF_LDX) {
1532 if (BPF_MODE(insn->code) != BPF_MEM ||
1533 insn->imm != 0) {
1534 verbose("BPF_LDX uses reserved fields\n");
1535 return -EINVAL;
1536 }
1537 /* check src operand */
1538 err = check_reg_arg(regs, insn->src_reg, SRC_OP);
1539 if (err)
1540 return err;
1541
1542 err = check_reg_arg(regs, insn->dst_reg, DST_OP_NO_MARK);
1543 if (err)
1544 return err;
1545
1546 /* check that memory (src_reg + off) is readable,
1547 * the state of dst_reg will be updated by this func
1548 */
1549 err = check_mem_access(env, insn->src_reg, insn->off,
1550 BPF_SIZE(insn->code), BPF_READ,
1551 insn->dst_reg);
1552 if (err)
1553 return err;
1554
1555 } else if (class == BPF_STX) {
1556 if (BPF_MODE(insn->code) == BPF_XADD) {
1557 err = check_xadd(env, insn);
1558 if (err)
1559 return err;
1560 insn_idx++;
1561 continue;
1562 }
1563
1564 if (BPF_MODE(insn->code) != BPF_MEM ||
1565 insn->imm != 0) {
1566 verbose("BPF_STX uses reserved fields\n");
1567 return -EINVAL;
1568 }
1569 /* check src1 operand */
1570 err = check_reg_arg(regs, insn->src_reg, SRC_OP);
1571 if (err)
1572 return err;
1573 /* check src2 operand */
1574 err = check_reg_arg(regs, insn->dst_reg, SRC_OP);
1575 if (err)
1576 return err;
1577
1578 /* check that memory (dst_reg + off) is writeable */
1579 err = check_mem_access(env, insn->dst_reg, insn->off,
1580 BPF_SIZE(insn->code), BPF_WRITE,
1581 insn->src_reg);
1582 if (err)
1583 return err;
1584
1585 } else if (class == BPF_ST) {
1586 if (BPF_MODE(insn->code) != BPF_MEM ||
1587 insn->src_reg != BPF_REG_0) {
1588 verbose("BPF_ST uses reserved fields\n");
1589 return -EINVAL;
1590 }
1591 /* check src operand */
1592 err = check_reg_arg(regs, insn->dst_reg, SRC_OP);
1593 if (err)
1594 return err;
1595
1596 /* check that memory (dst_reg + off) is writeable */
1597 err = check_mem_access(env, insn->dst_reg, insn->off,
1598 BPF_SIZE(insn->code), BPF_WRITE,
1599 -1);
1600 if (err)
1601 return err;
1602
1603 } else if (class == BPF_JMP) {
1604 u8 opcode = BPF_OP(insn->code);
1605
1606 if (opcode == BPF_CALL) {
1607 if (BPF_SRC(insn->code) != BPF_K ||
1608 insn->off != 0 ||
1609 insn->src_reg != BPF_REG_0 ||
1610 insn->dst_reg != BPF_REG_0) {
1611 verbose("BPF_CALL uses reserved fields\n");
1612 return -EINVAL;
1613 }
1614
1615 err = check_call(env, insn->imm);
1616 if (err)
1617 return err;
1618
1619 } else if (opcode == BPF_JA) {
1620 if (BPF_SRC(insn->code) != BPF_K ||
1621 insn->imm != 0 ||
1622 insn->src_reg != BPF_REG_0 ||
1623 insn->dst_reg != BPF_REG_0) {
1624 verbose("BPF_JA uses reserved fields\n");
1625 return -EINVAL;
1626 }
1627
1628 insn_idx += insn->off + 1;
1629 continue;
1630
1631 } else if (opcode == BPF_EXIT) {
1632 if (BPF_SRC(insn->code) != BPF_K ||
1633 insn->imm != 0 ||
1634 insn->src_reg != BPF_REG_0 ||
1635 insn->dst_reg != BPF_REG_0) {
1636 verbose("BPF_EXIT uses reserved fields\n");
1637 return -EINVAL;
1638 }
1639
1640 /* eBPF calling convetion is such that R0 is used
1641 * to return the value from eBPF program.
1642 * Make sure that it's readable at this time
1643 * of bpf_exit, which means that program wrote
1644 * something into it earlier
1645 */
1646 err = check_reg_arg(regs, BPF_REG_0, SRC_OP);
1647 if (err)
1648 return err;
1649
1650process_bpf_exit:
1651 insn_idx = pop_stack(env, &prev_insn_idx);
1652 if (insn_idx < 0) {
1653 break;
1654 } else {
1655 do_print_state = true;
1656 continue;
1657 }
1658 } else {
1659 err = check_cond_jmp_op(env, insn, &insn_idx);
1660 if (err)
1661 return err;
1662 }
1663 } else if (class == BPF_LD) {
1664 u8 mode = BPF_MODE(insn->code);
1665
1666 if (mode == BPF_ABS || mode == BPF_IND) {
1667 verbose("LD_ABS is not supported yet\n");
1668 return -EINVAL;
1669 } else if (mode == BPF_IMM) {
1670 err = check_ld_imm(env, insn);
1671 if (err)
1672 return err;
1673
1674 insn_idx++;
1675 } else {
1676 verbose("invalid BPF_LD mode\n");
1677 return -EINVAL;
1678 }
1679 } else {
1680 verbose("unknown insn class %d\n", class);
1681 return -EINVAL;
1682 }
1683
1684 insn_idx++;
1685 }
1686
1687 return 0;
1688}
1689
1690/* look for pseudo eBPF instructions that access map FDs and
1691 * replace them with actual map pointers
1692 */
1693static int replace_map_fd_with_map_ptr(struct verifier_env *env)
1694{
1695 struct bpf_insn *insn = env->prog->insnsi;
1696 int insn_cnt = env->prog->len;
1697 int i, j;
1698
1699 for (i = 0; i < insn_cnt; i++, insn++) {
1700 if (insn[0].code == (BPF_LD | BPF_IMM | BPF_DW)) {
1701 struct bpf_map *map;
1702 struct fd f;
1703
1704 if (i == insn_cnt - 1 || insn[1].code != 0 ||
1705 insn[1].dst_reg != 0 || insn[1].src_reg != 0 ||
1706 insn[1].off != 0) {
1707 verbose("invalid bpf_ld_imm64 insn\n");
1708 return -EINVAL;
1709 }
1710
1711 if (insn->src_reg == 0)
1712 /* valid generic load 64-bit imm */
1713 goto next_insn;
1714
1715 if (insn->src_reg != BPF_PSEUDO_MAP_FD) {
1716 verbose("unrecognized bpf_ld_imm64 insn\n");
1717 return -EINVAL;
1718 }
1719
1720 f = fdget(insn->imm);
1721
1722 map = bpf_map_get(f);
1723 if (IS_ERR(map)) {
1724 verbose("fd %d is not pointing to valid bpf_map\n",
1725 insn->imm);
1726 fdput(f);
1727 return PTR_ERR(map);
1728 }
1729
1730 /* store map pointer inside BPF_LD_IMM64 instruction */
1731 insn[0].imm = (u32) (unsigned long) map;
1732 insn[1].imm = ((u64) (unsigned long) map) >> 32;
1733
1734 /* check whether we recorded this map already */
1735 for (j = 0; j < env->used_map_cnt; j++)
1736 if (env->used_maps[j] == map) {
1737 fdput(f);
1738 goto next_insn;
1739 }
1740
1741 if (env->used_map_cnt >= MAX_USED_MAPS) {
1742 fdput(f);
1743 return -E2BIG;
1744 }
1745
1746 /* remember this map */
1747 env->used_maps[env->used_map_cnt++] = map;
1748
1749 /* hold the map. If the program is rejected by verifier,
1750 * the map will be released by release_maps() or it
1751 * will be used by the valid program until it's unloaded
1752 * and all maps are released in free_bpf_prog_info()
1753 */
1754 atomic_inc(&map->refcnt);
1755
1756 fdput(f);
1757next_insn:
1758 insn++;
1759 i++;
1760 }
1761 }
1762
1763 /* now all pseudo BPF_LD_IMM64 instructions load valid
1764 * 'struct bpf_map *' into a register instead of user map_fd.
1765 * These pointers will be used later by verifier to validate map access.
1766 */
1767 return 0;
1768}
1769
1770/* drop refcnt of maps used by the rejected program */
1771static void release_maps(struct verifier_env *env)
1772{
1773 int i;
1774
1775 for (i = 0; i < env->used_map_cnt; i++)
1776 bpf_map_put(env->used_maps[i]);
1777}
1778
1779/* convert pseudo BPF_LD_IMM64 into generic BPF_LD_IMM64 */
1780static void convert_pseudo_ld_imm64(struct verifier_env *env)
1781{
1782 struct bpf_insn *insn = env->prog->insnsi;
1783 int insn_cnt = env->prog->len;
1784 int i;
1785
1786 for (i = 0; i < insn_cnt; i++, insn++)
1787 if (insn->code == (BPF_LD | BPF_IMM | BPF_DW))
1788 insn->src_reg = 0;
1789}
1790
1791static void free_states(struct verifier_env *env)
1792{
1793 struct verifier_state_list *sl, *sln;
1794 int i;
1795
1796 if (!env->explored_states)
1797 return;
1798
1799 for (i = 0; i < env->prog->len; i++) {
1800 sl = env->explored_states[i];
1801
1802 if (sl)
1803 while (sl != STATE_LIST_MARK) {
1804 sln = sl->next;
1805 kfree(sl);
1806 sl = sln;
1807 }
1808 }
1809
1810 kfree(env->explored_states);
1811}
1812
1813int bpf_check(struct bpf_prog *prog, union bpf_attr *attr)
1814{
1815 char __user *log_ubuf = NULL;
1816 struct verifier_env *env;
1817 int ret = -EINVAL;
1818
1819 if (prog->len <= 0 || prog->len > BPF_MAXINSNS)
1820 return -E2BIG;
1821
1822 /* 'struct verifier_env' can be global, but since it's not small,
1823 * allocate/free it every time bpf_check() is called
1824 */
1825 env = kzalloc(sizeof(struct verifier_env), GFP_KERNEL);
1826 if (!env)
1827 return -ENOMEM;
1828
1829 env->prog = prog;
1830
1831 /* grab the mutex to protect few globals used by verifier */
1832 mutex_lock(&bpf_verifier_lock);
1833
1834 if (attr->log_level || attr->log_buf || attr->log_size) {
1835 /* user requested verbose verifier output
1836 * and supplied buffer to store the verification trace
1837 */
1838 log_level = attr->log_level;
1839 log_ubuf = (char __user *) (unsigned long) attr->log_buf;
1840 log_size = attr->log_size;
1841 log_len = 0;
1842
1843 ret = -EINVAL;
1844 /* log_* values have to be sane */
1845 if (log_size < 128 || log_size > UINT_MAX >> 8 ||
1846 log_level == 0 || log_ubuf == NULL)
1847 goto free_env;
1848
1849 ret = -ENOMEM;
1850 log_buf = vmalloc(log_size);
1851 if (!log_buf)
1852 goto free_env;
1853 } else {
1854 log_level = 0;
1855 }
1856
1857 ret = replace_map_fd_with_map_ptr(env);
1858 if (ret < 0)
1859 goto skip_full_check;
1860
1861 env->explored_states = kcalloc(prog->len,
1862 sizeof(struct verifier_state_list *),
1863 GFP_USER);
1864 ret = -ENOMEM;
1865 if (!env->explored_states)
1866 goto skip_full_check;
1867
1868 ret = check_cfg(env);
1869 if (ret < 0)
1870 goto skip_full_check;
1871
1872 ret = do_check(env);
1873
1874skip_full_check:
1875 while (pop_stack(env, NULL) >= 0);
1876 free_states(env);
1877
1878 if (log_level && log_len >= log_size - 1) {
1879 BUG_ON(log_len >= log_size);
1880 /* verifier log exceeded user supplied buffer */
1881 ret = -ENOSPC;
1882 /* fall through to return what was recorded */
1883 }
1884
1885 /* copy verifier log back to user space including trailing zero */
1886 if (log_level && copy_to_user(log_ubuf, log_buf, log_len + 1) != 0) {
1887 ret = -EFAULT;
1888 goto free_log_buf;
1889 }
1890
1891 if (ret == 0 && env->used_map_cnt) {
1892 /* if program passed verifier, update used_maps in bpf_prog_info */
1893 prog->aux->used_maps = kmalloc_array(env->used_map_cnt,
1894 sizeof(env->used_maps[0]),
1895 GFP_KERNEL);
1896
1897 if (!prog->aux->used_maps) {
1898 ret = -ENOMEM;
1899 goto free_log_buf;
1900 }
1901
1902 memcpy(prog->aux->used_maps, env->used_maps,
1903 sizeof(env->used_maps[0]) * env->used_map_cnt);
1904 prog->aux->used_map_cnt = env->used_map_cnt;
1905
1906 /* program is valid. Convert pseudo bpf_ld_imm64 into generic
1907 * bpf_ld_imm64 instructions
1908 */
1909 convert_pseudo_ld_imm64(env);
1910 }
1911
1912free_log_buf:
1913 if (log_level)
1914 vfree(log_buf);
1915free_env:
1916 if (!prog->aux->used_maps)
1917 /* if we didn't copy map pointers into bpf_prog_info, release
1918 * them now. Otherwise free_bpf_prog_info() will release them.
1919 */
1920 release_maps(env);
1921 kfree(env);
1922 mutex_unlock(&bpf_verifier_lock);
1923 return ret;
1924}