From mboxrd@z Thu Jan 1 00:00:00 1970 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 bombadil.infradead.org (bombadil.infradead.org [198.137.202.133]) (using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits)) (No client certificate requested) by smtp.lore.kernel.org (Postfix) with ESMTPS id 4F797FED3F1 for ; Fri, 24 Apr 2026 16:50:51 +0000 (UTC) DKIM-Signature: v=1; a=rsa-sha256; q=dns/txt; c=relaxed/relaxed; d=lists.infradead.org; s=bombadil.20210309; h=Sender:List-Subscribe:List-Help :List-Post:List-Archive:List-Unsubscribe:List-Id:Content-Type:Cc:To:From: Subject:Message-ID:References:Mime-Version:In-Reply-To:Date:Reply-To: Content-Transfer-Encoding:Content-ID:Content-Description:Resent-Date: Resent-From:Resent-Sender:Resent-To:Resent-Cc:Resent-Message-ID:List-Owner; bh=rMY4VEqc6R5YuPLO69SDBPYcLChwUILoRIT5/wHkEUI=; b=1CLzO9QT3cmeGDFZVt7e1U/gK3 H1rwYwzGc2V4fxFIbq82qXXeAz+/zKTRHJIim0hvNunYgjlCkE21vP5pjqaKokEQi5MgcGltbv0sV IzR/lWs/eCjd20rPunsh7f8Jf+FWo9YQqX8mVjndmGIzspIathDJ/N1IaRvUa/POKBUd2MsrvKLYU XMe55AmFZ/2n/1+p8dejq3S9NtMhGb/OCkMwQpkKRmkuCHJwfz6zPFgHxpzpc/OjUBOjuNmm2vAyq czrkIWI5MCgzc5E3gHNG/V5PFUgZ7IxPevVGdWRF/IK5rGB0PzVF1oBTH2jCjJczVitp5+mz1SHQl 8skgk34g==; Received: from localhost ([::1] helo=bombadil.infradead.org) by bombadil.infradead.org with esmtp (Exim 4.98.2 #2 (Red Hat Linux)) id 1wGJjf-0000000DVyt-3hF5; Fri, 24 Apr 2026 16:50:35 +0000 Received: from mail-dy1-x1349.google.com ([2607:f8b0:4864:20::1349]) by bombadil.infradead.org with esmtps (Exim 4.98.2 #2 (Red Hat Linux)) id 1wGJiU-0000000DUfb-1d0p for linux-arm-kernel@lists.infradead.org; Fri, 24 Apr 2026 16:49:31 +0000 Received: by mail-dy1-x1349.google.com with SMTP id 5a478bee46e88-2ddd8ef5343so7786130eec.1 for ; Fri, 24 Apr 2026 09:49:21 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=20251104; t=1777049360; x=1777654160; darn=lists.infradead.org; h=cc:to:from:subject:message-id:references:mime-version:in-reply-to :date:from:to:cc:subject:date:message-id:reply-to; bh=rMY4VEqc6R5YuPLO69SDBPYcLChwUILoRIT5/wHkEUI=; b=ukuRRs9fY6goWMQh6eSE8BIl1jPovxpHNDWnqpNQ1fb6VxpE6sIa2fGVZYuTF2e2j3 K7QR6krHhmVZlk5fFwHmDsb5srczf97NE1DTUNsgeRh6VUECylZmoV6aq6JPefx18ZP4 YO/amRY3z2A0u0IP3Ur5bjHYSD0ghFeCOVAgEXYXMElUqBY5tEHTOQyMipwF4nAHzzSX dEREXkGztqQLhPBSyix7L9age+AhyiQ8RQkbb/4VUwp1vUtXacUEr4iNlkA+UMJemHOB O+F+iN4lipJrAaqO7QaIHl1pn2j66nxb2Bd1I8UYINh/LELWCo3b4OxYBbBLXKkC1NN4 Fc6g== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20251104; t=1777049360; x=1777654160; h=cc:to:from:subject:message-id:references:mime-version:in-reply-to :date:x-gm-message-state:from:to:cc:subject:date:message-id:reply-to; bh=rMY4VEqc6R5YuPLO69SDBPYcLChwUILoRIT5/wHkEUI=; b=hmpwqLcLobq/kQOOBF8ZtaJx5JmXYeZmT/ZdZA6kHoFZ8Hj6yXLpwoJZI9JnGox6xZ YTXWcw+ZVzUPM+jcMBbCRvaLR0i0zK1vxTXVH3FPYtUWeURzgKj/6EHqK2c4B9HveiYn 0GNO5ngurEF/vEP71wW5C+2X5ImT4imTzrZ3x7aFzXRPzzXIK+Dx9Rii/yFiY8+emyFS AI3l979ab9blJcjTQGqWftw3M5C/tHIjFrFlMPOmxZXRHfo0Jsk3GM8ajVeDsIb2sCgh ZVvHJ+hcxfFs6l2Jv/PRv8hKlDW0gm+D3Ylii1uYOPcqqHBh57zM+ffq/deu2/YZ5htZ KEdA== X-Forwarded-Encrypted: i=1; AFNElJ9tkivGAF1HUT+tpsC18rUBqiVuyNwNmvJVgpC5CghDBZubOj8pZx7GBcisdH6edVk5972j/NZuUskusDfeHLYn@lists.infradead.org X-Gm-Message-State: AOJu0YwruRljifGlkz1E6pVyL1CAKicR85f6HqPe3D8stAD9svMXVfMR MQdLM/O1FH8CAZlrzRNQnAlLEb3NBUs+jHJdYBHPiD7BqWYNCig3moOakgVeGHzMAXxZwx3H1+Z FV5hIi2IlpQ== X-Received: from dybgt36.prod.google.com ([2002:a05:7301:2224:b0:2d1:6536:ac8a]) (user=irogers job=prod-delivery.src-stubby-dispatcher) by 2002:a05:693c:4011:b0:2e6:e916:70ff with SMTP id 5a478bee46e88-2e6e916770cmr10884574eec.28.1777049359938; Fri, 24 Apr 2026 09:49:19 -0700 (PDT) Date: Fri, 24 Apr 2026 09:47:07 -0700 In-Reply-To: <20260424164721.2229025-1-irogers@google.com> Mime-Version: 1.0 References: <20260423163406.1779809-1-irogers@google.com> <20260424164721.2229025-1-irogers@google.com> X-Mailer: git-send-email 2.54.0.545.g6539524ca2-goog Message-ID: <20260424164721.2229025-46-irogers@google.com> Subject: [PATCH v5 45/58] perf stackcollapse: Port stackcollapse to use python module From: Ian Rogers To: acme@kernel.org, adrian.hunter@intel.com, james.clark@linaro.org, leo.yan@linux.dev, namhyung@kernel.org, tmricht@linux.ibm.com Cc: alice.mei.rogers@gmail.com, dapeng1.mi@linux.intel.com, linux-arm-kernel@lists.infradead.org, linux-kernel@vger.kernel.org, linux-perf-users@vger.kernel.org, mingo@redhat.com, peterz@infradead.org, Ian Rogers Content-Type: text/plain; charset="UTF-8" X-CRM114-Version: 20100106-BlameMichelson ( TRE 0.8.0 (BSD) ) MR-646709E3 X-CRM114-CacheID: sfid-20260424_094922_748542_B24DF36E X-CRM114-Status: GOOD ( 16.38 ) X-BeenThere: linux-arm-kernel@lists.infradead.org X-Mailman-Version: 2.1.34 Precedence: list List-Id: List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Sender: "linux-arm-kernel" Errors-To: linux-arm-kernel-bounces+linux-arm-kernel=archiver.kernel.org@lists.infradead.org Modernize the legacy stackcollapse.py trace script by refactoring it into a class-based architecture (StackCollapseAnalyzer). The script uses perf.session for event processing and aggregates call stacks to produce output suitable for flame graphs. Assisted-by: Gemini:gemini-3.1-pro-preview Signed-off-by: Ian Rogers --- v2: - Fixed Callchain Check: Replaced hasattr(sample, "callchain") with getattr(sample, "callchain", None) and checked if it is not None . This avoids attempting to iterate over None when a sample lacks a callchain, which would raise a TypeError. - Fixed Comm Resolution: The code already used self.session.process(sample.sample_pid).comm() to resolve the command name using the session object (if available), avoiding the missing comm attribute on perf.sample_event. - Code Cleanup: Broke a long line in process_event to satisfy pylint. --- tools/perf/python/stackcollapse.py | 126 +++++++++++++++++++++++++++++ 1 file changed, 126 insertions(+) create mode 100755 tools/perf/python/stackcollapse.py diff --git a/tools/perf/python/stackcollapse.py b/tools/perf/python/stackcollapse.py new file mode 100755 index 000000000000..fae0f0f503a3 --- /dev/null +++ b/tools/perf/python/stackcollapse.py @@ -0,0 +1,126 @@ +#!/usr/bin/env python3 +# SPDX-License-Identifier: GPL-2.0 +""" +stackcollapse.py - format perf samples with one line per distinct call stack + +This script's output has two space-separated fields. The first is a semicolon +separated stack including the program name (from the "comm" field) and the +function names from the call stack. The second is a count: + + swapper;start_kernel;rest_init;cpu_idle;default_idle;native_safe_halt 2 + +The file is sorted according to the first field. + +Ported from tools/perf/scripts/python/stackcollapse.py +""" + +import argparse +from collections import defaultdict +import sys +import perf + + +class StackCollapseAnalyzer: + """Accumulates call stacks and prints them collapsed.""" + + def __init__(self, args: argparse.Namespace) -> None: + self.args = args + self.lines: dict[str, int] = defaultdict(int) + + def tidy_function_name(self, sym: str, dso: str) -> str: + """Beautify function names based on options.""" + if sym is None: + sym = "[unknown]" + + sym = sym.replace(";", ":") + if self.args.tidy_java: + # Beautify Java signatures + sym = sym.replace("<", "") + sym = sym.replace(">", "") + if sym.startswith("L") and "/" in sym: + sym = sym[1:] + try: + sym = sym[:sym.index("(")] + except ValueError: + pass + + if self.args.annotate_kernel and dso == "[kernel.kallsyms]": + return sym + "_[k]" + return sym + + def process_event(self, sample: perf.sample_event) -> None: + """Collect call stack for each sample.""" + stack = [] + callchain = getattr(sample, "callchain", None) + if callchain is not None: + for node in callchain: + stack.append(self.tidy_function_name(node.symbol, node.dso)) + else: + # Fallback if no callchain + sym = getattr(sample, "symbol", "[unknown]") + dso = getattr(sample, "dso", "[unknown]") + stack.append(self.tidy_function_name(sym, dso)) + + if self.args.include_comm: + if hasattr(self, 'session') and self.session: + comm = self.session.process(sample.sample_pid).comm() + else: + comm = "Unknown" + comm = comm.replace(" ", "_") + sep = "-" + if self.args.include_pid: + comm = f"{comm}{sep}{getattr(sample, 'sample_pid', 0)}" + sep = "/" + if self.args.include_tid: + comm = f"{comm}{sep}{getattr(sample, 'sample_tid', 0)}" + stack.append(comm) + + stack_string = ";".join(reversed(stack)) + self.lines[stack_string] += 1 + + def print_totals(self) -> None: + """Print sorted collapsed stacks.""" + for stack in sorted(self.lines): + print(f"{stack} {self.lines[stack]}") + + +def main(): + """Main function.""" + ap = argparse.ArgumentParser( + description="Format perf samples with one line per distinct call stack" + ) + ap.add_argument("-i", "--input", default="perf.data", help="Input file name") + ap.add_argument("--include-tid", action="store_true", help="include thread id in stack") + ap.add_argument("--include-pid", action="store_true", help="include process id in stack") + ap.add_argument("--no-comm", dest="include_comm", action="store_false", default=True, + help="do not separate stacks according to comm") + ap.add_argument("--tidy-java", action="store_true", help="beautify Java signatures") + ap.add_argument("--kernel", dest="annotate_kernel", action="store_true", + help="annotate kernel functions with _[k]") + + args = ap.parse_args() + + if args.include_tid and not args.include_comm: + print("requesting tid but not comm is invalid", file=sys.stderr) + sys.exit(1) + if args.include_pid and not args.include_comm: + print("requesting pid but not comm is invalid", file=sys.stderr) + sys.exit(1) + + analyzer = StackCollapseAnalyzer(args) + + try: + session = perf.session(perf.data(args.input), sample=analyzer.process_event) + analyzer.session = session + session.process_events() + except IOError as e: + print(f"Error: {e}", file=sys.stderr) + sys.exit(1) + except KeyboardInterrupt: + pass + + analyzer.print_totals() + + +if __name__ == "__main__": + main() -- 2.54.0.545.g6539524ca2-goog