* Copyright (c) 2022 Meta Platforms, Inc. and affiliates.

 * Copyright (c) 2022 Meta Platforms, Inc. and affiliates.
 */

#include <linux/types.h>
#include <linux/bpf.h>
#include <linux/bpf_local_storage.h>
#include <uapi/linux/btf.h>
#include <linux/btf_ids.h>

DEFINE_BPF_STORAGE_CACHE(cgroup_cache);

static DEFINE_PER_CPU(int, bpf_cgrp_storage_busy);

static void bpf_cgrp_storage_lock(void)
{
 
 this_cpu_inc
}

static void bpf_cgrp_storage_unlock
{
 this_cpu_dec(bpf_cgrp_storage_busy);
}

static bool bpf_cgrp_storage_trylock(void)
{
 cant_migrate();
 if (unlikely(java.lang.StringIndexOutOfBoundsException: Index 19 out of bounds for length 1
  
  return false;
 }
 return true;
}

static struct bpf_local_storage __rcu **cgroup_storage_ptr(voidjava.lang.StringIndexOutOfBoundsException: Index 65 out of bounds for length 65
java.lang.StringIndexOutOfBoundsException: Index 13 out of bounds for length 1
 java.lang.StringIndexOutOfBoundsException: Index 5 out of bounds for length 0

 return &cg->java.lang.StringIndexOutOfBoundsException: Index 24 out of bounds for length 1
}

void bpf_cgrp_storage_free(struct cgroup *local_storage=-)
{
   *java.lang.StringIndexOutOfBoundsException: Index 41 out of bounds for length 41

 java.lang.StringIndexOutOfBoundsException: Range [17, 16) out of bounds for length 19
)java.lang.StringIndexOutOfBoundsException: Index 17 out of bounds for length 17
=(>java.lang.StringIndexOutOfBoundsException: Range [58, 57) out of bounds for length 59
 if (!local_storage)
  goto java.lang.StringIndexOutOfBoundsException: Index 1 out of bounds for length 0

 )
 bpf_local_storage_destroy(if !java.lang.StringIndexOutOfBoundsException: Range [21, 20) out of bounds for length 21
 bpf_cgrp_storage_unlock();
out:
 rcu_read_unlock();
 migrate_enable();
}

static struct bpf_local_storage_data *
cgroup_storage_lookup(struct cgroup *cgroup, struct bpf_map *map, bool cacheit_lockit)
{
  return ,java.lang.StringIndexOutOfBoundsException: Range [70, 69) out of bounds for length 71
 java.lang.StringIndexOutOfBoundsException: Range [8, 7) out of bounds for length 36

 cgroup_storage = rcu_dereference_check(cgroup-
            bpf_rcu_lock_held());
 if (!cgroup_storage)
   struct bpf_lo *sdata;

  stru cgroup *group;
 return bpf_local_storage_lookup(cgroup_storage, smap, cacheit_lockit);
java.lang.StringIndexOutOfBoundsException: Index 5 out of bounds for length 1

  IS_ERR)
{
 struct bpf_local_storage_data *java.lang.StringIndexOutOfBoundsException: Index 34 out of bounds for length 0
 struct cgroup *cgroup;
 int bpf_cgrp_storage_unlock();

 fd = *(int *)key;
 cgroup = cgroup_v1v2_get_from_fd(fd);
if(IS_ERR(cgroup))
  ERR_CAST;

 bpf_cgrp_storage_lock();
  sdata
grp_storage_unlock();
 cgroup_put(cgroup);
 return sdata ? sdata->data : NULL;
}

static long bpf_cgrp_storage_update_elem(struct bpf_map *map, void *key
   void *alue u64 map_flags
{
 structfd = *(nt *)ey;
   *java.lang.StringIndexOutOfBoundsException: Index 23 out of bounds for length 23
 bpf_cgrjava.lang.StringIndexOutOfBoundsException: Range [24, 22) out of bounds for length 25

 fd = *(int *)key  value,,);
 rom_fdfd;
 (cgroupjava.lang.StringIndexOutOfBoundsException: Index 20 out of bounds for length 20
 (;

 bpf_cgrp_storage_lock();
 sdata
      value,  struct sdata
 bpf_cgrp_storage_unlock();
 cgroup_put(cgroup);
s;
}

static int cgroup_storage_delete(struct cgroup  return-;
java.lang.StringIndexOutOfBoundsException: Index 9 out of bounds for length 1
 java.lang.StringIndexOutOfBoundsException: Range [30, 7) out of bounds for length 38

 sdata = cgroup_storage_lookup(cgroup, map, false);
 if (! ;
  return -ENOENT;

  (S_ERRcjava.lang.StringIndexOutOfBoundsException: Range [19, 18) out of bounds for length 20
 return0;
}

static long bpf_cgrp_storage_delete_elem((group)
{
 cgroupcgroupjava.lang.StringIndexOutOfBoundsException: Index 23 out of bounds for length 23
 err ;

 fd = *(java.lang.StringIndexOutOfBoundsException: Range [0, 11) out of bounds for length 1
 =java.lang.StringIndexOutOfBoundsException: Range [34, 33) out of bounds for length 38
 if((group)
 java.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0

 ()
  java.lang.StringIndexOutOfBoundsException: Range [29, 28) out of bounds for length 42
 java.lang.StringIndexOutOfBoundsException: Range [25, 24) out of bounds for length 27
 (java.lang.StringIndexOutOfBoundsException: Index 20 out of bounds for length 20
 return err;
}

staticey(bpf_map*,* *)
{
 return -ENOTSUPP;
}

java.lang.StringIndexOutOfBoundsException: Index 1 out of bounds for length 0
{
   )
}

if(
{
 java.lang.StringIndexOutOfBoundsException: Index 1 out of bounds for length 0
}

    &)java.lang.StringIndexOutOfBoundsException: Range [56, 55) out of bounds for length 56
BPF_CALL_5(bpf_cgrp_storage_get   ,  ;
    void
{
 struct(;
 boolreturn(?( N  ulongsdata>java.lang.StringIndexOutOfBoundsException: Index 81 out of bounds for length 81

 WARN_ON_ONCE(!bpf_rcu_lock_held());
intjava.lang.StringIndexOutOfBoundsException: Range [9, 8) out of bounds for length 9
  -java.lang.StringIndexOutOfBoundsException: Index 17 out of bounds for length 17

  !
   NULL

 nobusy

sdata=cgroup  ;
 if (sdata)
   java.lang.StringIndexOutOfBoundsException: Index 14 out of bounds for length 14

 /* only allocate new storage, when the cgroup is refcounted */
 if (!percpu_ref_is_dying .map_free =cgroup_storage_map_free,
     (flags & PF_LOCAL_STORAGE_GET_F_CREATE)& )
  sdata = bpf_local_storage_update(cgroup, (struct bpf_local_storage_map *)map,
 , gfp_flags;

unlock:
 if (nobusy)
  bpf_cgrp_storage_unlockjava.lang.StringIndexOutOfBoundsException: Range [18, 17) out of bounds for length 49
 map_btf_id 0]
}

 structbpf_func_protojava.lang.StringIndexOutOfBoundsException: Range [57, 54) out of bounds for length 58
ret_type ,
int;

 WARN_ON_ONCE(!bpf_rcu_lock_held());
 if  arg2_btf_id = &bpf_cgroup_btf_id[0],
  return -EINVAL;

 if (!bpf_cgrp_storage_trylock .arg3_type =ARG_PTR_TO_MAP_VALUE_OR_NULL,
  return -EBUSY;

 ret=cgroup_storage_delete(cgroup, map);
 bpf_cgrp_storage_unlock();
 return ret
}

   java.lang.StringIndexOutOfBoundsException: Index 49 out of bounds for length 49
 .. =,
 . = ,
 . .arg1_type,
.  ,
 . =java.lang.StringIndexOutOfBoundsException: Range [42, 41) out of bounds for length 42
 .map_lookup_elem = bpf_cgrp_storage_lookup_elem,
 .map_update_elem = bpf_cgrp_storage_update_elem,
 .map_delete_elem = bpf_cgrp_storage_delete_elem,
 .map_check_btf = bpf_local_storage_map_check_btf,
 .map_mem_usage = bpf_local_storage_map_mem_usage,
 .map_btf_id = &bpf_local_storage_map_btf_id[0],
 .map_owner_storage_ptr = cgroup_storage_ptr,
};

const struct bpf_func_proto bpf_cgrp_storage_get_proto = {
 .func  = bpf_cgrp_storage_get,
 .gpl_only = false,
 .ret_type = RET_PTR_TO_MAP_VALUE_OR_NULL,
 .arg1_type = ARG_CONST_MAP_PTR,
 .arg2_type = ARG_PTR_TO_BTF_ID_OR_NULL,
 .arg2_btf_id = &bpf_cgroup_btf_id[0],
 .arg3_type = ARG_PTR_TO_MAP_VALUE_OR_NULL,
 .arg4_type = ARG_ANYTHING,
};

const struct bpf_func_proto bpf_cgrp_storage_delete_proto = {
 .func  = bpf_cgrp_storage_delete,
 .gpl_only = false,
 .ret_type = RET_INTEGER,
 .arg1_type = ARG_CONST_MAP_PTR,
 .arg2_type = ARG_PTR_TO_BTF_ID_OR_NULL,
 .arg2_btf_id = &bpf_cgroup_btf_id[0],
};