80 lines
2.1 KiB
C
80 lines
2.1 KiB
C
// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
|
|
/* Copyright (c) 2020 Facebook */
|
|
#include "vmlinux.h"
|
|
#include <bpf/bpf_helpers.h>
|
|
#include <bpf/bpf_tracing.h>
|
|
#include <bpf/bpf_core_read.h>
|
|
|
|
#include "replica.h"
|
|
|
|
char LICENSE[] SEC("license") = "Dual BSD/GPL";
|
|
|
|
struct {
|
|
__uint(type, BPF_MAP_TYPE_PERF_EVENT_ARRAY);
|
|
__uint(key_size, sizeof(int));
|
|
__uint(value_size, sizeof(int));
|
|
} events SEC(".maps");
|
|
|
|
/* __completed_cmd(const uint8_t *cmd, size_t len, unsigned long long seq); */
|
|
SEC("uprobe//home/lian/share/9.1-kvstore/kvstore:__completed_cmd")
|
|
int BPF_KPROBE(handle_completed_cmd,
|
|
const __u8 *cmd, size_t len, __u64 seq)
|
|
{
|
|
struct replica_event evt = {};
|
|
__u32 copy_len;
|
|
|
|
evt.type = EVENT_COMPLETED_CMD;
|
|
evt.complete.seq = seq;
|
|
|
|
copy_len = len;
|
|
if (copy_len > MAX_CMD_LEN)
|
|
copy_len = MAX_CMD_LEN;
|
|
|
|
evt.complete.len = copy_len;
|
|
|
|
if (cmd)
|
|
bpf_probe_read_user(evt.complete.cmd, copy_len, cmd);
|
|
|
|
bpf_perf_event_output(ctx, &events,
|
|
BPF_F_CURRENT_CPU,
|
|
&evt, sizeof(evt));
|
|
return 0;
|
|
}
|
|
|
|
/* __ssync(const uint8_t *ip, uint32_t ip_len, int port, unsigned long long seq); */
|
|
SEC("uprobe//home/lian/share/9.1-kvstore/kvstore:__ssync")
|
|
int BPF_KPROBE(handle_ssync,
|
|
const __u8 *ip, __u32 ip_len, int port, __u64 seq)
|
|
{
|
|
struct replica_event evt = {};
|
|
|
|
evt.type = EVENT_SSYNC;
|
|
evt.sync.seq = seq;
|
|
evt.sync.port = port;
|
|
|
|
__u32 copy_len = ip_len;
|
|
if (copy_len > sizeof(evt.sync.ip))
|
|
copy_len = sizeof(evt.sync.ip);
|
|
|
|
if (ip)
|
|
bpf_probe_read_user(evt.sync.ip, copy_len, ip);
|
|
|
|
bpf_perf_event_output(ctx, &events,
|
|
BPF_F_CURRENT_CPU,
|
|
&evt, sizeof(evt));
|
|
return 0;
|
|
}
|
|
|
|
/* __sready(void); */
|
|
SEC("uprobe//home/lian/share/9.1-kvstore/kvstore:__sready")
|
|
int BPF_KPROBE(handle_sready)
|
|
{
|
|
struct replica_event evt = {};
|
|
|
|
evt.type = EVENT_SREADY;
|
|
|
|
bpf_perf_event_output(ctx, &events,
|
|
BPF_F_CURRENT_CPU,
|
|
&evt, sizeof(evt));
|
|
return 0;
|
|
} |