tools/testing/selftests/bpf/progs/test_spin_lock.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 // Copyright (c) 2019 Facebook
   3 #include <linux/bpf.h>
   4 #include <linux/version.h>
   5 #include <bpf/bpf_helpers.h>
   6 #include "bpf_misc.h"
   7
   8 struct hmap_elem {
   9         volatile int cnt;
  10         struct bpf_spin_lock lock;
  11         int test_padding;
  12 };
  13
  14 struct {
  15         __uint(type, BPF_MAP_TYPE_HASH);
  16         __uint(max_entries, 1);
  17         __type(key, int);
  18         __type(value, struct hmap_elem);
  19 } hmap SEC(".maps");
  20
  21 struct cls_elem {
  22         struct bpf_spin_lock lock;
  23         volatile int cnt;
  24 };
  25
  26 struct {
  27         __uint(type, BPF_MAP_TYPE_CGROUP_STORAGE);
  28         __type(key, struct bpf_cgroup_storage_key);
  29         __type(value, struct cls_elem);
  30 } cls_map SEC(".maps");
  31
  32 struct bpf_vqueue {
  33         struct bpf_spin_lock lock;
  34         /* 4 byte hole */
  35         unsigned long long lasttime;
  36         int credit;
  37         unsigned int rate;
  38 };
  39
  40 struct {
  41         __uint(type, BPF_MAP_TYPE_ARRAY);
  42         __uint(max_entries, 1);
  43         __type(key, int);
  44         __type(value, struct bpf_vqueue);
  45 } vqueue SEC(".maps");
  46
  47 #define CREDIT_PER_NS(delta, rate) (((delta) * rate) >> 20)
  48
  49 SEC("cgroup_skb/ingress")
  50 int bpf_spin_lock_test(struct __sk_buff *skb)
  51 {
  52         volatile int credit = 0, max_credit = 100, pkt_len = 64;
  53         struct hmap_elem zero = {}, *val;
  54         unsigned long long curtime;
  55         struct bpf_vqueue *q;
  56         struct cls_elem *cls;
  57         int key = 0;
  58         int err = 0;
  59
  60         val = bpf_map_lookup_elem(&hmap, &key);
  61         if (!val) {
  62                 bpf_map_update_elem(&hmap, &key, &zero, 0);
  63                 val = bpf_map_lookup_elem(&hmap, &key);
  64                 if (!val) {
  65                         err = 1;
  66                         goto err;
  67                 }
  68         }
  69         /* spin_lock in hash map run time test */
  70         bpf_spin_lock(&val->lock);
  71         if (val->cnt)
  72                 val->cnt--;
  73         else
  74                 val->cnt++;
  75         if (val->cnt != 0 && val->cnt != 1)
  76                 err = 1;
  77         bpf_spin_unlock(&val->lock);
  78
  79         /* spin_lock in array. virtual queue demo */
  80         q = bpf_map_lookup_elem(&vqueue, &key);
  81         if (!q)
  82                 goto err;
  83         curtime = bpf_ktime_get_ns();
  84         bpf_spin_lock(&q->lock);
  85         q->credit += CREDIT_PER_NS(curtime - q->lasttime, q->rate);
  86         q->lasttime = curtime;
  87         if (q->credit > max_credit)
  88                 q->credit = max_credit;
  89         q->credit -= pkt_len;
  90         credit = q->credit;
  91         bpf_spin_unlock(&q->lock);
  92
  93         __sink(credit);
  94
  95         /* spin_lock in cgroup local storage */
  96         cls = bpf_get_local_storage(&cls_map, 0);
  97         bpf_spin_lock(&cls->lock);
  98         cls->cnt++;
  99         bpf_spin_unlock(&cls->lock);
 100
 101 err:
 102         return err;
 103 }
 104 char _license[] SEC("license") = "GPL";