How can implement a function like __sync_fetch_and_add of C in Aya?

[tim-chow]

Suppose I have a C eBPF program like below:

//go:build ignore

#include "common.h"

char __license[] SEC("license") = "Dual MIT/GPL";

struct bpf_map_def SEC("maps") kprobe_map = {
	.type        = BPF_MAP_TYPE_ARRAY,
	.key_size    = sizeof(u32),
	.value_size  = sizeof(u64),
	.max_entries = 1,
};

SEC("kprobe/sys_execve")
int kprobe_execve() {
	u32 key     = 0;
	u64 initval = 1, *valp;

	valp = bpf_map_lookup_elem(&kprobe_map, &key);
	if (!valp) {
		bpf_map_update_elem(&kprobe_map, &key, &initval, BPF_ANY);
		return 0;
	}
	__sync_fetch_and_add(valp, 1);

	return 0;
}

If I rewrite the program using Aya, How can I implement the atomic function __sync_fetch_and_add?

[tim-chow]

AtomicUxx not works~

[alessandrod]

atomic_xadd_* https://doc.rust-lang.org/stable/std/?search=atomic