Files
mgmt/misc/filter-golang-stack.py
James Shubin d30ff6cfae legal: Remove year
Instead of constantly making these updates, let's just remove the year
since things are stored in git anyways, and this is not an actual modern
legal risk anymore.
2025-01-26 16:24:51 -05:00

124 lines
3.3 KiB
Python
Executable File

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright (C) James Shubin and the project contributors
# Written by James Shubin <james@shubin.ca> and the project contributors
#
# 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 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will 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.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
# This program filters golang stack traces. Modify the filter by editing the
# filter_chunk function. This is a quick hack to help me get work done. It's
# written in python to encourage some golang enthusiast to rewrite it in golang
# and to expand the functionality.
import sys
if len(sys.argv) == 2 and sys.argv[1] != "-":
lines = open(sys.argv[1], "r").readlines()
else:
lines = sys.stdin.readlines()
print("read: %d lines" % len(lines))
# find program start
start = -1
for i in range(len(lines)):
line = lines[i]
if line.startswith("PC="):
start=i
break
if start == -1:
print("could not find program start, looking for PC=???", file=sys.stderr)
sys.exit(1)
print("starts at line: %d" % (start+1)) # +1 because we're zero based
def is_chunk(line):
if line == "main.main()" or line.startswith("goroutine"):
return True
return False
def is_end(line):
# register junk as seen at the end
rx = ["rax", "rbx", "rcx", "rdx", "rdi", "rsi", "rbp", "rsp", "r8", "r9", "r10", "r11", "r12", "r13", "r14", "r15", "rip", "rflags", "cs", "fs", "gs"]
for x in rx:
if line.startswith(x + " "):
return True
return False
def filter_chunk(chunk):
lines = chunk.split("\n")
if len(lines) < 2:
return False
package_line = lines[1]
if package_line.startswith("github.com/purpleidea/mgmt/vendor/"):
return False
if package_line.startswith("github.com/") and not package_line.startswith("github.com/purpleidea/mgmt/"):
return False
if package_line.startswith("internal/poll"):
return False
if package_line.startswith("context.propagateCancel"):
return False
if package_line.startswith("runtime.gopark"):
return False
if package_line.startswith("runtime.futex"):
return False
if package_line.startswith("os/signal.signal_recv"):
return False
return True
chunks = []
for i in range(start+1, len(lines)):
line = lines[i]
# found a chunk
if is_chunk(line):
chunk = line
for j in range(i+1, len(lines)):
jline = lines[j]
if is_chunk(jline) or is_end(jline):
chunks.append(chunk)
i = j
if is_end(jline):
end = j
break
# else
chunk = chunk + jline
print("found %d chunks" % len(chunks))
count = 0
for i in range(len(chunks)):
chunk = chunks[i]
if not filter_chunk(chunk):
continue
print(">>> chunk %d:" % i)
print(chunk)
print("")
count = count + 1
print("got %d filtered chunks" % count)
print("ends at line: %d" % (end+1)) # +1 because we're zero based
for i in range(end, len(lines)):
line = lines[i]
print("%s" % line, end='') # already comes with a newline