REDROOM

Navigation

Breadcrumbs

Create

Edit File Size: 1.38 KB

//usr/share/doc/bpfcc-tools/examples/lua/task_switch.lua

Text Base64

#!/usr/bin/env bcc-lua
--[[
Copyright 2016 GitHub, Inc

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
]]

local program = [[
#include <uapi/linux/ptrace.h>
#include <linux/sched.h>

struct key_t {
  u32 prev_pid;
  u32 curr_pid;
};
// map_type, key_type, leaf_type, table_name, num_entry
BPF_HASH(stats, struct key_t, u64, 1024);
int count_sched(struct pt_regs *ctx, struct task_struct *prev) {
  struct key_t key = {};
  u64 zero = 0, *val;

key.curr_pid = bpf_get_current_pid_tgid();
  key.prev_pid = prev->pid;

val = stats.lookup_or_try_init(&key, &zero);
  if (val) {
    (*val)++;
  }
  return 0;
}
]]

return function(BPF)
  local b = BPF:new{text=program, debug=0}
  b:attach_kprobe{event="finish_task_switch", fn_name="count_sched"}

print("Press any key...")
  io.read()

local t = b:get_table("stats")
  for k, v in t:items() do
    print("task_switch[%d -> %d] = %d" % {k.prev_pid, k.curr_pid, tonumber(v)})
  end
end

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Name	Size	Perms	Modified	Actions
bashreadline.c	476 B	lrw-r--r--	2023-12-08 15:36:51	Edit Download recursive recursive
bashreadline.lua	1.08 KB	lrwxr-xr-x	2023-12-08 15:36:51	Edit Download recursive recursive
kprobe-latency.lua	2.46 KB	lrwxr-xr-x	2023-12-08 15:36:51	Edit Download recursive recursive
kprobe-write.lua	1.10 KB	lrwxr-xr-x	2023-12-08 15:36:51	Edit Download recursive recursive
memleak.lua	6.05 KB	lrwxr-xr-x	2023-12-08 15:36:51	Edit Download recursive recursive
offcputime.lua	3.04 KB	lrwxr-xr-x	2023-12-08 15:36:51	Edit Download recursive recursive
sock-parse-dns.lua	1.65 KB	lrwxr-xr-x	2023-12-08 15:36:51	Edit Download recursive recursive
sock-parse-http.lua	1.98 KB	lrwxr-xr-x	2023-12-08 15:36:51	Edit Download recursive recursive
sock-proto.lua	1.40 KB	lrwxr-xr-x	2023-12-08 15:36:51	Edit Download recursive recursive
sock-protolen.lua	1.50 KB	lrwxr-xr-x	2023-12-08 15:36:51	Edit Download recursive recursive
strlen_count.lua	1.27 KB	lrwxr-xr-x	2023-12-08 15:36:51	Edit Download recursive recursive
task_switch.lua	1.38 KB	lrwxr-xr-x	2023-12-08 15:36:51	Edit Download recursive recursive
tracepoint-offcputime.lua	2.59 KB	lrwxr-xr-x	2023-12-08 15:36:51	Edit Download recursive recursive
uprobe-readline-perf.lua	1.88 KB	lrwxr-xr-x	2023-12-08 15:36:51	Edit Download recursive recursive
uprobe-readline.lua	1.52 KB	lrwxr-xr-x	2023-12-08 15:36:51	Edit Download recursive recursive
uprobe-tailkt.lua	2.50 KB	lrwxr-xr-x	2023-12-08 15:36:51	Edit Download recursive recursive
usdt_ruby.lua	1.19 KB	lrwxr-xr-x	2023-12-08 15:36:51	Edit Download recursive recursive

Directory Contents