From nobody Sun Feb 8 04:57:33 2026 Return-Path: X-Spam-Checker-Version: SpamAssassin 3.4.0 (2014-02-07) on aws-us-west-2-korg-lkml-1.web.codeaurora.org Received: from vger.kernel.org (vger.kernel.org [23.128.96.18]) by smtp.lore.kernel.org (Postfix) with ESMTP id 7969BC04A94 for ; Sun, 30 Jul 2023 08:27:39 +0000 (UTC) Received: (majordomo@vger.kernel.org) by vger.kernel.org via listexpand id S229809AbjG3I1h (ORCPT ); Sun, 30 Jul 2023 04:27:37 -0400 Received: from lindbergh.monkeyblade.net ([23.128.96.19]:37564 "EHLO lindbergh.monkeyblade.net" rhost-flags-OK-OK-OK-OK) by vger.kernel.org with ESMTP id S229791AbjG3I13 (ORCPT ); Sun, 30 Jul 2023 04:27:29 -0400 Received: from mail-ed1-x52e.google.com (mail-ed1-x52e.google.com [IPv6:2a00:1450:4864:20::52e]) by lindbergh.monkeyblade.net (Postfix) with ESMTPS id 9A19910FF for ; Sun, 30 Jul 2023 01:27:26 -0700 (PDT) Received: by mail-ed1-x52e.google.com with SMTP id 4fb4d7f45d1cf-52229f084beso5456370a12.2 for ; Sun, 30 Jul 2023 01:27:26 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20221208; t=1690705645; x=1691310445; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:from:to:cc:subject:date :message-id:reply-to; bh=Yd7KEXZAW7eX+hNRe4Tt2lcKQC5uzSvrHXquUEQC3kY=; b=X3jcZ16BOTMFY1dukAFIkwTfr2KyBV8n5ZDI4kbF3CoGafX/TJUBMJiQkBUoVTHwCC b8IbdVy80EB1kPNjI9N0wV79yQtYDCQJiwk6VBT1WAUI/EfqoShTTTiS5O7gaNRS+4/K DjwbRtSC4LU0oCIgJOF1+4Bq/UyRGGkCHLyqjVHbl7DP+2QjNhH5pODah/+gDw+XHE+o JyDkdCuHrggD0nya2sRBTFG8DJowgGuDrJx61PppyEhkpjC97zmWAscuUlBWG5Ab6aRK W3Em69f9YrzcZO2DDzgI/BftShxtZqIpBMxxN3E5shez7j3lMQHglhwUQPTKAydC76R8 r9kg== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20221208; t=1690705645; x=1691310445; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:x-gm-message-state:from:to:cc :subject:date:message-id:reply-to; bh=Yd7KEXZAW7eX+hNRe4Tt2lcKQC5uzSvrHXquUEQC3kY=; b=kxAw3ilMTZX7J8gbL3RjtRMN+FjgWh1EUhMMwQzpKhdsPdb3AvS4YeOVMMOor4ZpPP Gday7FVV1c8K28Dw307k1TomooXZOL/Yth7i2mrPPeweafODyPNhDQiCX9t20jldkj4r jCNSTLvN+tYtiXjl/SIU1EQjtAUIO/+JVIJtxjjOUF11aVHBs96YTDtSs80ygAQfSddm Whl76uPRd4pUzCBwqmJQqvsmy+00yvFxTf+0vi9wFigryKKB/u6wj04lNXbvTbN4AWNc DRY2kEGnGllrrGQxGdWlvZvRIsBKozzZc709kWeK4appNjmA5EFBQkZXT28pwzNdh3Pz ceOQ== X-Gm-Message-State: ABy/qLbFRV339S+ZCmsXnkJ+NV4AGY4o8j4+VQw1/3q3Ov2ff6vPQuN/ Zg1Mg5fJyqIRGTvKQTlMUCo= X-Google-Smtp-Source: APBJJlG4Fd0Z04vIN8O51EYm+AoZ1UaXmlDkenzktyyD+nL0+Zu6mVBfLEg9aEJRCFHUIoyhA4DvwQ== X-Received: by 2002:a05:6402:b0e:b0:521:7a5e:ab1e with SMTP id bm14-20020a0564020b0e00b005217a5eab1emr7141332edb.21.1690705644988; Sun, 30 Jul 2023 01:27:24 -0700 (PDT) Received: from nam-dell (ip-217-105-46-58.ip.prioritytelecom.net. [217.105.46.58]) by smtp.gmail.com with ESMTPSA id m17-20020aa7d351000000b005224c6b2183sm3790791edr.39.2023.07.30.01.27.24 (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Sun, 30 Jul 2023 01:27:24 -0700 (PDT) From: Nam Cao To: Paul Walmsley , Palmer Dabbelt , Albert Ou , linux-riscv@lists.infradead.org, linux-kernel@vger.kernel.org Cc: Nam Cao Subject: test code for kprobe Date: Sun, 30 Jul 2023 10:27:10 +0200 Message-Id: X-Mailer: git-send-email 2.34.1 In-Reply-To: References: MIME-Version: 1.0 Content-Transfer-Encoding: quoted-printable Precedence: bulk List-ID: X-Mailing-List: linux-kernel@vger.kernel.org Content-Type: text/plain; charset="utf-8" --- drivers/test_kprobe/Makefile | 3 + drivers/test_kprobe/test_kprobe.c | 265 ++++++++++++++++++++++++++++++ 2 files changed, 268 insertions(+) create mode 100644 drivers/test_kprobe/Makefile create mode 100644 drivers/test_kprobe/test_kprobe.c diff --git a/drivers/test_kprobe/Makefile b/drivers/test_kprobe/Makefile new file mode 100644 index 000000000000..c3c39bd0f8b5 --- /dev/null +++ b/drivers/test_kprobe/Makefile @@ -0,0 +1,3 @@ +# SPDX-License-Identifier: GPL-2.0 +# +obj-m +=3D test_kprobe.o diff --git a/drivers/test_kprobe/test_kprobe.c b/drivers/test_kprobe/test_k= probe.c new file mode 100644 index 000000000000..543108c5fc8a --- /dev/null +++ b/drivers/test_kprobe/test_kprobe.c @@ -0,0 +1,265 @@ +#include +#include +#include +#include +MODULE_LICENSE("Dual BSD/GPL"); + +__attribute__ ((naked)) +int sample_c_j(void) { + __asm__( + "c.j label_cj\n" + "li a0, 0\n ret\n" + "li a0, 1\n ret\n" + "li a0, 2\n ret\n" + "li a0, 3\n ret\n" + "label_cj:\n" + "li a0, 4\n ret\n" + "li a0, 5\n ret\n" + "li a0, 6\n ret\n" + ); +} + +__attribute__ ((naked)) +int sample_c_jr(void) { + __asm__( + "la a0, label_c_jr\n" + "c_jr_location:\n" + "c.jr a0\n" + "li a0, 0\n ret\n" + "li a0, 1\n ret\n" + "li a0, 2\n ret\n" + "li a0, 3\n ret\n" + "label_c_jr:\n" + "li a0, 4\n ret\n" + "li a0, 5\n ret\n" + "li a0, 6\n ret\n" + ); +} + +__attribute__ ((naked)) +int sample_c_jalr(void) { + __asm__( + "mv a1, x1\n" + "la a0, label_c_jalr\n" + "c_jalr_location:\n" + "c.jalr a0\n" + + "addi a0, a0, -5\n" + "jr a1\n" + + "label_c_jalr:\n" + "li a0, 9\n ret\n" + ); +} + +__attribute__ ((naked)) +int sample_c_beqz(int a0) { + __asm__( + "c.beqz a0, beqz_label\n" + "li a0, 10\n ret\n" + "beqz_label:\n" + "li a0, 4\n ret\n" + ); +} + +__attribute__ ((naked)) +int sample_c_bnez(int a0) { + __asm__( + "c.bnez a0, bnez_label\n" + "li a0, 10\n ret\n" + "bnez_label:\n" + "li a0, 4\n ret\n" + ); +} + +static int pre_handler(struct kprobe *p, struct pt_regs *regs) { + printk("pre_handler() called\n"); + + return 0; +} + +static int test_c_j(void) { + static struct kprobe kp; + + int ret; + + /* Test C.J */ + kp.symbol_name =3D "sample_c_j"; + kp.pre_handler =3D pre_handler; + + ret =3D register_kprobe(&kp); + if (ret) { + printk("Couldn't register kprobe, err=3D%d\n", ret); + return -1; + } + + ret =3D sample_c_j(); + if (ret !=3D 4) { + printk("ERROR: expect value 4, got %d\n", ret); + return -1; + } + else { + printk("Got value 4, all good!\n"); + return 0; + } +} + +static int test_c_jr(void) { + static struct kprobe kp; + int ret; + + /* Test C.JR */ + kp.symbol_name =3D "c_jr_location"; + kp.pre_handler =3D pre_handler; + + ret =3D register_kprobe(&kp); + if (ret) { + printk("Couldn't register kprobe, err=3D%d\n", ret); + return -1; + } + + ret =3D sample_c_jr(); + if (ret !=3D 4) { + printk("Expect value 4, got %d\n", ret); + return -1; + } + else { + printk("Got value 4, all good!\n"); + return 0; + } +} + +static int test_c_jalr(void) { + struct kprobe kp; + int ret; + + memset(&kp, 0, sizeof(kp)); + + /* Test C.JR */ + kp.symbol_name =3D "c_jalr_location"; + kp.pre_handler =3D pre_handler; + + ret =3D register_kprobe(&kp); + if (ret) { + printk("Couldn't register kprobe, err=3D%d\n", ret); + return -1; + } + + ret =3D sample_c_jalr(); + if (ret !=3D 4) { + printk("Expect value 4, got %d\n", ret); + return -1; + } + else { + printk("Got value 4, all good!\n"); + } + + unregister_kprobe(&kp); + + return 0; +} + +static int test_c_bnez(void) { + static struct kprobe kp; + + int ret; + + /* Test C.JR */ + kp.symbol_name =3D "sample_c_bnez"; + kp.pre_handler =3D pre_handler; + + ret =3D register_kprobe(&kp); + if (ret) { + printk("Couldn't register kprobe, err=3D%d\n", ret); + return -1; + } + + ret =3D sample_c_bnez(1); + if (ret !=3D 4) { + printk("Expect value 4, got %d\n", ret); + return -1; + } else { + printk("Got value 4, all good!\n"); + } + + ret =3D sample_c_bnez(0); + if (ret !=3D 10) { + printk("Expect value 10, got %d\n", ret); + return -1; + } else { + printk("Got value 4, all good!\n"); + } + + return 0; +} + +static int test_c_beqz(void) { + static struct kprobe kp; + + int ret; + + /* Test C.JR */ + kp.symbol_name =3D "sample_c_beqz"; + kp.pre_handler =3D pre_handler; + + ret =3D register_kprobe(&kp); + if (ret) { + printk("Couldn't register kprobe, err=3D%d\n", ret); + return -1; + } + + ret =3D sample_c_beqz(0); + if (ret !=3D 4) { + printk("Expect value 4, got %d\n", ret); + return -1; + } + else { + printk("Got value 4, all good!\n"); + return 0; + } + + ret =3D sample_c_beqz(1); + if (ret !=3D 10) { + printk("Expect value 10, got %d\n", ret); + return -1; + } + else { + printk("Got value 4, all good!\n"); + return 0; + } +} + +static int hello_init(void) +{ + printk("Hello\n"); + + printk("Testing C.J...\n"); + if (test_c_j()) + return -1; + + printk("Testing C.JR...\n"); + if (test_c_jr()) + return -1; + + printk("Testing C.JALR...\n"); + if (test_c_jalr()) + return -1; + + printk("Testing C.BNEZ...\n"); + if (test_c_bnez()) + return -1; + + printk("Testing C.BEQZ...\n"); + if (test_c_beqz()) + return -1; + + return 0; +} + +static void hello_exit(void) +{ + printk("Goodbye\n"); +} + +module_init(hello_init); +module_exit(hello_exit); --=20 2.34.1