extern spinlock_t btf_idr_lock;
 extern struct kobject *btf_kobj;
 
+typedef u64 (*bpf_callback_t)(u64, u64, u64, u64, u64);
 typedef int (*bpf_iter_init_seq_priv_t)(void *private_data,
                                        struct bpf_iter_aux_info *aux);
 typedef void (*bpf_iter_fini_seq_priv_t)(void *private_data);
        int (*map_set_for_each_callback_args)(struct bpf_verifier_env *env,
                                              struct bpf_func_state *caller,
                                              struct bpf_func_state *callee);
-       int (*map_for_each_callback)(struct bpf_map *map, void *callback_fn,
+       int (*map_for_each_callback)(struct bpf_map *map,
+                                    bpf_callback_t callback_fn,
                                     void *callback_ctx, u64 flags);
 
        /* BTF name and id of struct allocated by map_alloc */
 
                .off   = 0,                                     \
                .imm   = TGT })
 
-/* Function call */
-
-#define BPF_CAST_CALL(x)                                       \
-               ((u64 (*)(u64, u64, u64, u64, u64))(x))
-
 /* Convert function address to BPF immediate */
 
 #define BPF_CALL_IMM(x)        ((void *)(x) - (void *)__bpf_call_base)
 
        .seq_priv_size          = sizeof(struct bpf_iter_seq_array_map_info),
 };
 
-static int bpf_for_each_array_elem(struct bpf_map *map, void *callback_fn,
+static int bpf_for_each_array_elem(struct bpf_map *map, bpf_callback_t callback_fn,
                                   void *callback_ctx, u64 flags)
 {
        u32 i, key, num_elems = 0;
                        val = array->value + array->elem_size * i;
                num_elems++;
                key = i;
-               ret = BPF_CAST_CALL(callback_fn)((u64)(long)map,
-                                       (u64)(long)&key, (u64)(long)val,
-                                       (u64)(long)callback_ctx, 0);
+               ret = callback_fn((u64)(long)map, (u64)(long)&key,
+                                 (u64)(long)val, (u64)(long)callback_ctx, 0);
                /* return value: 0 - continue, 1 - stop and return */
                if (ret)
                        break;
 
        .seq_priv_size          = sizeof(struct bpf_iter_seq_hash_map_info),
 };
 
-static int bpf_for_each_hash_elem(struct bpf_map *map, void *callback_fn,
+static int bpf_for_each_hash_elem(struct bpf_map *map, bpf_callback_t callback_fn,
                                  void *callback_ctx, u64 flags)
 {
        struct bpf_htab *htab = container_of(map, struct bpf_htab, map);
                                val = elem->key + roundup_key_size;
                        }
                        num_elems++;
-                       ret = BPF_CAST_CALL(callback_fn)((u64)(long)map,
-                                       (u64)(long)key, (u64)(long)val,
-                                       (u64)(long)callback_ctx, 0);
+                       ret = callback_fn((u64)(long)map, (u64)(long)key,
+                                         (u64)(long)val, (u64)(long)callback_ctx, 0);
                        /* return value: 0 - continue, 1 - stop and return */
                        if (ret) {
                                rcu_read_unlock();
 
        struct bpf_hrtimer *t = container_of(hrtimer, struct bpf_hrtimer, timer);
        struct bpf_map *map = t->map;
        void *value = t->value;
-       void *callback_fn;
+       bpf_callback_t callback_fn;
        void *key;
        u32 idx;
 
                key = value - round_up(map->key_size, 8);
        }
 
-       BPF_CAST_CALL(callback_fn)((u64)(long)map, (u64)(long)key,
-                                  (u64)(long)value, 0, 0);
+       callback_fn((u64)(long)map, (u64)(long)key, (u64)(long)value, 0, 0);
        /* The verifier checked that return value is zero. */
 
        this_cpu_write(hrtimer_running, NULL);