cregit-Linux how code gets into the kernel

Release 4.15 kernel/test_kprobes.c

Directory: kernel
/*
 * test_kprobes.c - simple sanity test for *probes
 *
 * Copyright IBM Corp. 2008
 *
 * This program is free software;  you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it would be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
 * the GNU General Public License for more details.
 */


#define pr_fmt(fmt) "Kprobe smoke test: " fmt

#include <linux/kernel.h>
#include <linux/kprobes.h>
#include <linux/random.h>


#define div_factor 3




static u32 rand1, preh_val, posth_val;



static int errors, handler_errors, num_tests;

static u32 (*target)(u32 value);

static u32 (*target2)(u32 value);


static noinline u32 kprobe_target(u32 value) { return (value / div_factor); }

Contributors

PersonTokensPropCommitsCommitProp
Ananth N. Mavinakayanahalli17100.00%1100.00%
Total17100.00%1100.00%


static int kp_pre_handler(struct kprobe *p, struct pt_regs *regs) { if (preemptible()) { handler_errors++; pr_err("pre-handler is preemptible\n"); } preh_val = (rand1 / div_factor); return 0; }

Contributors

PersonTokensPropCommitsCommitProp
Ananth N. Mavinakayanahalli2764.29%150.00%
Masami Hiramatsu1535.71%150.00%
Total42100.00%2100.00%


static void kp_post_handler(struct kprobe *p, struct pt_regs *regs, unsigned long flags) { if (preemptible()) { handler_errors++; pr_err("post-handler is preemptible\n"); } if (preh_val != (rand1 / div_factor)) { handler_errors++; pr_err("incorrect value in post_handler\n"); } posth_val = preh_val + div_factor; }

Contributors

PersonTokensPropCommitsCommitProp
Ananth N. Mavinakayanahalli4573.77%133.33%
Masami Hiramatsu1524.59%133.33%
Fabian Frederick11.64%133.33%
Total61100.00%3100.00%

static struct kprobe kp = { .symbol_name = "kprobe_target", .pre_handler = kp_pre_handler, .post_handler = kp_post_handler };
static int test_kprobe(void) { int ret; ret = register_kprobe(&kp); if (ret < 0) { pr_err("register_kprobe returned %d\n", ret); return ret; } ret = target(rand1); unregister_kprobe(&kp); if (preh_val == 0) { pr_err("kprobe pre_handler not called\n"); handler_errors++; } if (posth_val == 0) { pr_err("kprobe post_handler not called\n"); handler_errors++; } return 0; }

Contributors

PersonTokensPropCommitsCommitProp
Ananth N. Mavinakayanahalli5767.06%125.00%
Masami Hiramatsu2529.41%250.00%
Fabian Frederick33.53%125.00%
Total85100.00%4100.00%


static noinline u32 kprobe_target2(u32 value) { return (value / div_factor) + 1; }

Contributors

PersonTokensPropCommitsCommitProp
Masami Hiramatsu19100.00%1100.00%
Total19100.00%1100.00%


static int kp_pre_handler2(struct kprobe *p, struct pt_regs *regs) { preh_val = (rand1 / div_factor) + 1; return 0; }

Contributors

PersonTokensPropCommitsCommitProp
Masami Hiramatsu29100.00%1100.00%
Total29100.00%1100.00%


static void kp_post_handler2(struct kprobe *p, struct pt_regs *regs, unsigned long flags) { if (preh_val != (rand1 / div_factor) + 1) { handler_errors++; pr_err("incorrect value in post_handler2\n"); } posth_val = preh_val + div_factor; }

Contributors

PersonTokensPropCommitsCommitProp
Masami Hiramatsu4797.92%150.00%
Fabian Frederick12.08%150.00%
Total48100.00%2100.00%

static struct kprobe kp2 = { .symbol_name = "kprobe_target2", .pre_handler = kp_pre_handler2, .post_handler = kp_post_handler2 };
static int test_kprobes(void) { int ret; struct kprobe *kps[2] = {&kp, &kp2}; /* addr and flags should be cleard for reusing kprobe. */ kp.addr = NULL; kp.flags = 0; ret = register_kprobes(kps, 2); if (ret < 0) { pr_err("register_kprobes returned %d\n", ret); return ret; } preh_val = 0; posth_val = 0; ret = target(rand1); if (preh_val == 0) { pr_err("kprobe pre_handler not called\n"); handler_errors++; } if (posth_val == 0) { pr_err("kprobe post_handler not called\n"); handler_errors++; } preh_val = 0; posth_val = 0; ret = target2(rand1); if (preh_val == 0) { pr_err("kprobe pre_handler2 not called\n"); handler_errors++; } if (posth_val == 0) { pr_err("kprobe post_handler2 not called\n"); handler_errors++; } unregister_kprobes(kps, 2); return 0; }

Contributors

PersonTokensPropCommitsCommitProp
Masami Hiramatsu16697.08%266.67%
Fabian Frederick52.92%133.33%
Total171100.00%3100.00%

#if 0 static u32 jph_val; static u32 j_kprobe_target(u32 value) { if (preemptible()) { handler_errors++; pr_err("jprobe-handler is preemptible\n"); } if (value != rand1) { handler_errors++; pr_err("incorrect value in jprobe handler\n"); } jph_val = rand1; jprobe_return(); return 0; } static struct jprobe jp = { .entry = j_kprobe_target, .kp.symbol_name = "kprobe_target" }; static int test_jprobe(void) { int ret; ret = register_jprobe(&jp); if (ret < 0) { pr_err("register_jprobe returned %d\n", ret); return ret; } ret = target(rand1); unregister_jprobe(&jp); if (jph_val == 0) { pr_err("jprobe handler not called\n"); handler_errors++; } return 0; } static struct jprobe jp2 = { .entry = j_kprobe_target, .kp.symbol_name = "kprobe_target2" }; static int test_jprobes(void) { int ret; struct jprobe *jps[2] = {&jp, &jp2}; /* addr and flags should be cleard for reusing kprobe. */ jp.kp.addr = NULL; jp.kp.flags = 0; ret = register_jprobes(jps, 2); if (ret < 0) { pr_err("register_jprobes returned %d\n", ret); return ret; } jph_val = 0; ret = target(rand1); if (jph_val == 0) { pr_err("jprobe handler not called\n"); handler_errors++; } jph_val = 0; ret = target2(rand1); if (jph_val == 0) { pr_err("jprobe handler2 not called\n"); handler_errors++; } unregister_jprobes(jps, 2); return 0; } #else #define test_jprobe() (0) #define test_jprobes() (0) #endif #ifdef CONFIG_KRETPROBES static u32 krph_val;
static int entry_handler(struct kretprobe_instance *ri, struct pt_regs *regs) { if (preemptible()) { handler_errors++; pr_err("kretprobe entry handler is preemptible\n"); } krph_val = (rand1 / div_factor); return 0; }

Contributors

PersonTokensPropCommitsCommitProp
Masami Hiramatsu42100.00%2100.00%
Total42100.00%2100.00%


static int return_handler(struct kretprobe_instance *ri, struct pt_regs *regs) { unsigned long ret = regs_return_value(regs); if (preemptible()) { handler_errors++; pr_err("kretprobe return handler is preemptible\n"); } if (ret != (rand1 / div_factor)) { handler_errors++; pr_err("incorrect value in kretprobe handler\n"); } if (krph_val == 0) { handler_errors++; pr_err("call to kretprobe entry handler failed\n"); } krph_val = rand1; return 0; }

Contributors

PersonTokensPropCommitsCommitProp
Masami Hiramatsu5566.27%250.00%
Ananth N. Mavinakayanahalli2631.33%125.00%
Fabian Frederick22.41%125.00%
Total83100.00%4100.00%

static struct kretprobe rp = { .handler = return_handler, .entry_handler = entry_handler, .kp.symbol_name = "kprobe_target" };
static int test_kretprobe(void) { int ret; ret = register_kretprobe(&rp); if (ret < 0) { pr_err("register_kretprobe returned %d\n", ret); return ret; } ret = target(rand1); unregister_kretprobe(&rp); if (krph_val != rand1) { pr_err("kretprobe handler not called\n"); handler_errors++; } return 0; }

Contributors

PersonTokensPropCommitsCommitProp
Ananth N. Mavinakayanahalli5681.16%125.00%
Masami Hiramatsu1115.94%250.00%
Fabian Frederick22.90%125.00%
Total69100.00%4100.00%


static int return_handler2(struct kretprobe_instance *ri, struct pt_regs *regs) { unsigned long ret = regs_return_value(regs); if (ret != (rand1 / div_factor) + 1) { handler_errors++; pr_err("incorrect value in kretprobe handler2\n"); } if (krph_val == 0) { handler_errors++; pr_err("call to kretprobe entry handler failed\n"); } krph_val = rand1; return 0; }

Contributors

PersonTokensPropCommitsCommitProp
Ananth N. Mavinakayanahalli4970.00%125.00%
Abhishek Sagar1521.43%125.00%
Masami Hiramatsu45.71%125.00%
Fabian Frederick22.86%125.00%
Total70100.00%4100.00%

static struct kretprobe rp2 = { .handler = return_handler2, .entry_handler = entry_handler, .kp.symbol_name = "kprobe_target2" };
static int test_kretprobes(void) { int ret; struct kretprobe *rps[2] = {&rp, &rp2}; /* addr and flags should be cleard for reusing kprobe. */ rp.kp.addr = NULL; rp.kp.flags = 0; ret = register_kretprobes(rps, 2); if (ret < 0) { pr_err("register_kretprobe returned %d\n", ret); return ret; } krph_val = 0; ret = target(rand1); if (krph_val != rand1) { pr_err("kretprobe handler not called\n"); handler_errors++; } krph_val = 0; ret = target2(rand1); if (krph_val != rand1) { pr_err("kretprobe handler2 not called\n"); handler_errors++; } unregister_kretprobes(rps, 2); return 0; }

Contributors

PersonTokensPropCommitsCommitProp
Masami Hiramatsu7454.81%350.00%
Ananth N. Mavinakayanahalli5641.48%116.67%
Fabian Frederick32.22%116.67%
Abhishek Sagar21.48%116.67%
Total135100.00%6100.00%

#endif /* CONFIG_KRETPROBES */
int init_test_probes(void) { int ret; target = kprobe_target; target2 = kprobe_target2; do { rand1 = prandom_u32(); } while (rand1 <= div_factor); pr_info("started\n"); num_tests++; ret = test_kprobe(); if (ret < 0) errors++; num_tests++; ret = test_kprobes(); if (ret < 0) errors++; num_tests++; ret = test_jprobe(); if (ret < 0) errors++; num_tests++; ret = test_jprobes(); if (ret < 0) errors++; #ifdef CONFIG_KRETPROBES num_tests++; ret = test_kretprobe(); if (ret < 0) errors++; num_tests++; ret = test_kretprobes(); if (ret < 0) errors++; #endif /* CONFIG_KRETPROBES */ if (errors) pr_err("BUG: %d out of %d tests failed\n", errors, num_tests); else if (handler_errors) pr_err("BUG: %d error(s) running handlers\n", handler_errors); else pr_info("passed successfully\n"); return 0; }

Contributors

PersonTokensPropCommitsCommitProp
Ananth N. Mavinakayanahalli11262.22%120.00%
Masami Hiramatsu5932.78%240.00%
Fabian Frederick84.44%120.00%
Akinobu Mita10.56%120.00%
Total180100.00%5100.00%


Overall Contributors

PersonTokensPropCommitsCommitProp
Masami Hiramatsu64452.27%555.56%
Ananth N. Mavinakayanahalli53143.10%111.11%
Fabian Frederick342.76%111.11%
Abhishek Sagar221.79%111.11%
Akinobu Mita10.08%111.11%
Total1232100.00%9100.00%
Directory: kernel
Information contained on this website is for historical information purposes only and does not indicate or represent copyright ownership.
Created with cregit.