Skip to content

Commit

Permalink
Merge pull request #66410 from al45tair/eng/PR-110262673-5.9
Browse files Browse the repository at this point in the history
[Backtracing][Linux] Add Linux support to swift-backtrace.
  • Loading branch information
al45tair committed Jun 7, 2023
2 parents 827b4fd + f61e72c commit 45339b6
Show file tree
Hide file tree
Showing 5 changed files with 276 additions and 25 deletions.
20 changes: 10 additions & 10 deletions stdlib/public/Backtracing/BacktraceFormatter.swift
Expand Up @@ -523,7 +523,7 @@ public struct BacktraceFormatter {
///
/// @result An array of strings, one per column.
public func formatColumns(frame: Backtrace.Frame,
addressWidth: Int = 64,
addressWidth: Int,
index: Int? = nil) -> [String] {
let pc: String
var attrs: [String] = []
Expand Down Expand Up @@ -563,7 +563,7 @@ public struct BacktraceFormatter {
///
/// @result An array of table rows.
public func formatRows(frame: Backtrace.Frame,
addressWidth: Int = 64,
addressWidth: Int,
index: Int? = nil) -> [TableRow] {
return [.columns(formatColumns(frame: frame,
addressWidth: addressWidth,
Expand All @@ -578,7 +578,7 @@ public struct BacktraceFormatter {
///
/// @result A `String` containing the formatted data.
public func format(frame: Backtrace.Frame,
addressWidth: Int = 64,
addressWidth: Int,
index: Int? = nil) -> String {
let rows = formatRows(frame: frame,
addressWidth: addressWidth,
Expand All @@ -593,7 +593,7 @@ public struct BacktraceFormatter {
///
/// @result A `String` containing the formatted data.
public func format(frames: some Sequence<Backtrace.Frame>,
addressWidth: Int = 64) -> String {
addressWidth: Int) -> String {
var rows: [TableRow] = []

var n = 0
Expand Down Expand Up @@ -734,7 +734,7 @@ public struct BacktraceFormatter {
///
/// @result An array of strings, one per column.
public func formatColumns(frame: SymbolicatedBacktrace.Frame,
addressWidth: Int = 64,
addressWidth: Int,
index: Int? = nil) -> [String] {
let pc: String
var attrs: [String] = []
Expand Down Expand Up @@ -851,7 +851,7 @@ public struct BacktraceFormatter {
///
/// @result An array of table rows.
public func formatRows(frame: SymbolicatedBacktrace.Frame,
addressWidth: Int = 64,
addressWidth: Int,
index: Int? = nil,
showSource: Bool = true) -> [TableRow] {
let columns = formatColumns(frame: frame,
Expand Down Expand Up @@ -880,7 +880,7 @@ public struct BacktraceFormatter {
///
/// @result A `String` containing the formatted data.
public func format(frame: SymbolicatedBacktrace.Frame,
addressWidth: Int = 64,
addressWidth: Int,
index: Int? = nil,
showSource: Bool = true) -> String {
let rows = formatRows(frame: frame, addressWidth: addressWidth,
Expand All @@ -902,7 +902,7 @@ public struct BacktraceFormatter {
///
/// @result A `String` containing the formatted data.
public func format(frames: some Sequence<SymbolicatedBacktrace.Frame>,
addressWidth: Int = 64) -> String {
addressWidth: Int) -> String {
var rows: [TableRow] = []
var sourceLocationsShown = Set<SymbolicatedBacktrace.SourceLocation>()

Expand Down Expand Up @@ -982,7 +982,7 @@ public struct BacktraceFormatter {
///
/// @result An array of strings, one per column.
public func formatColumns(image: Backtrace.Image,
addressWidth: Int = 64) -> [String] {
addressWidth: Int) -> [String] {
let addressRange = "\(hex(image.baseAddress, width: addressWidth))\(hex(image.endOfText, width: addressWidth))"
let buildID: String
if let bytes = image.buildID {
Expand Down Expand Up @@ -1011,7 +1011,7 @@ public struct BacktraceFormatter {
///
/// @result A string containing the formatted data.
public func format(images: some Sequence<Backtrace.Image>,
addressWidth: Int = 64) -> String {
addressWidth: Int) -> String {
let rows = images.map{
TableRow.columns(
formatColumns(image: $0,
Expand Down
3 changes: 2 additions & 1 deletion stdlib/public/libexec/swift-backtrace/CMakeLists.txt
Expand Up @@ -30,6 +30,7 @@ add_swift_target_executable(swift-backtrace BUILD_WITH_LIBEXEC
main.swift
AnsiColor.swift
TargetMacOS.swift
TargetLinux.swift
Themes.swift
Utils.swift

Expand All @@ -44,5 +45,5 @@ add_swift_target_executable(swift-backtrace BUILD_WITH_LIBEXEC
${BACKTRACING_COMPILE_FLAGS}
-parse-as-library

TARGET_SDKS OSX)
TARGET_SDKS OSX LINUX)

232 changes: 232 additions & 0 deletions stdlib/public/libexec/swift-backtrace/TargetLinux.swift
@@ -0,0 +1,232 @@
//===--- TargetLinux.swift - Represents a process we are inspecting -------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2022 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
//
// Defines `Target`, which represents the process we are inspecting.
// This is the Linux version.
//
//===----------------------------------------------------------------------===//

#if os(Linux)

import Glibc

import _Backtracing
@_spi(Internal) import _Backtracing
@_spi(Contexts) import _Backtracing
@_spi(MemoryReaders) import _Backtracing
@_spi(Utils) import _Backtracing

@_implementationOnly import Runtime

struct TargetThread {
typealias ThreadID = pid_t

var id: ThreadID
var context: HostContext?
var name: String
var backtrace: SymbolicatedBacktrace
}

class Target {
typealias Address = UInt64

var pid: pid_t
var name: String
var signal: UInt64
var faultAddress: Address
var crashingThread: TargetThread.ThreadID

var images: [Backtrace.Image] = []

var threads: [TargetThread] = []
var crashingThreadNdx: Int = -1

var signalName: String {
switch signal {
case UInt64(SIGQUIT): return "SIGQUIT"
case UInt64(SIGABRT): return "SIGABRT"
case UInt64(SIGBUS): return "SIGBUS"
case UInt64(SIGFPE): return "SIGFPE"
case UInt64(SIGILL): return "SIGILL"
case UInt64(SIGSEGV): return "SIGSEGV"
case UInt64(SIGTRAP): return "SIGTRAP"
default: return "\(signal)"
}
}

var signalDescription: String {
switch signal {
case UInt64(SIGQUIT): return "Terminated"
case UInt64(SIGABRT): return "Aborted"
case UInt64(SIGBUS): return "Bus error"
case UInt64(SIGFPE): return "Floating point exception"
case UInt64(SIGILL): return "Illegal instruction"
case UInt64(SIGSEGV): return "Bad pointer dereference"
case UInt64(SIGTRAP): return "System trap"
default:
return "Signal \(signal)"
}
}

var reader: MemserverMemoryReader

// Get the name of a process
private static func getProcessName(pid: pid_t) -> String {
let path = "/proc/\(pid)/comm"
guard let name = readString(from: path) else {
return ""
}
return String(stripWhitespace(name))
}

/// Get the name of a thread
private func getThreadName(tid: Int64) -> String {
let path = "/proc/\(pid)/task/\(tid)/comm"
guard let name = readString(from: path) else {
return ""
}
let trimmed = String(stripWhitespace(name))

// Allow the main thread to use the process' name, but other
// threads will have an empty name unless they've set the name
// explicitly
if trimmed == self.name && pid != tid {
return ""
}
return trimmed
}

init(crashInfoAddr: UInt64, limit: Int?, top: Int, cache: Bool) {
// fd #4 is reserved for the memory server
let memserverFd: CInt = 4

pid = getppid()
reader = MemserverMemoryReader(fd: memserverFd)
name = Self.getProcessName(pid: pid)

let crashInfo: CrashInfo
do {
crashInfo = try reader.fetch(from: crashInfoAddr, as: CrashInfo.self)
} catch {
print("swift-backtrace: unable to fetch crash info.")
exit(1)
}

crashingThread = TargetThread.ThreadID(crashInfo.crashing_thread)
signal = crashInfo.signal
faultAddress = crashInfo.fault_address

images = Backtrace.captureImages(using: reader,
forProcess: Int(pid))

do {
try fetchThreads(threadListHead: Address(crashInfo.thread_list),
limit: limit, top: top, cache: cache)
} catch {
print("swift-backtrace: failed to fetch thread information: \(error)")
exit(1)
}
}

/// Fetch information about all of the process's threads; the crash_info
/// structure contains a linked list of thread ucontexts, which may not
/// include every thread. In particular, if a thread was stuck in an
/// uninterruptible wait, we won't have a ucontext for it.
func fetchThreads(
threadListHead: Address,
limit: Int?, top: Int, cache: Bool
) throws {
var next = threadListHead

while next != 0 {
let t = try reader.fetch(from: next, as: thread.self)

next = t.next

guard let ucontext
= try? reader.fetch(from: t.uctx, as: ucontext_t.self) else {
// This can happen if a thread is in an uninterruptible wait
continue
}

let context = HostContext.fromHostMContext(ucontext.uc_mcontext)
let backtrace = try Backtrace.capture(from: context,
using: reader,
images: images,
limit: limit,
top: top)
guard let symbolicated
= backtrace.symbolicated(with: images,
sharedCacheInfo: nil,
useSymbolCache: cache) else {
print("unable to symbolicate backtrace for thread \(t.tid)")
exit(1)
}

threads.append(TargetThread(id: TargetThread.ThreadID(t.tid),
context: context,
name: getThreadName(tid: t.tid),
backtrace: symbolicated))
}

// Sort the threads by thread ID; the main thread always sorts
// lower than any other.
threads.sort {
return $0.id == pid || ($1.id != pid && $0.id < $1.id)
}

// Find the crashing thread index
if let ndx = threads.firstIndex(where: { $0.id == crashingThread }) {
crashingThreadNdx = ndx
} else {
print("unable to find the crashing thread")
exit(1)
}
}

func redoBacktraces(limit: Int?, top: Int, cache: Bool) {
for (ndx, thread) in threads.enumerated() {
guard let context = thread.context else {
continue
}

guard let backtrace = try? Backtrace.capture(from: context,
using: reader,
images: images,
limit: limit,
top: top) else {
print("unable to capture backtrace from context for thread \(ndx)")
continue
}

guard let symbolicated = backtrace.symbolicated(with: images,
sharedCacheInfo: nil,
useSymbolCache: cache) else {
print("unable to symbolicate backtrace from context for thread \(ndx)")
continue
}

threads[ndx].backtrace = symbolicated
}
}

func withDebugger(_ body: () -> ()) throws {
print("""
From another shell, please run
lldb --attach-pid \(pid) -o c
""")
body()
}
}

#endif // os(Linux)
6 changes: 3 additions & 3 deletions stdlib/public/libexec/swift-backtrace/Utils.swift
Expand Up @@ -51,7 +51,7 @@ internal func parseUInt64<S: StringProtocol>(_ s: S) -> UInt64? {
}
}

#if os(macOS)
#if os(macOS) || os(Linux)

struct PosixError: Error {
var errno: Int32
Expand Down Expand Up @@ -139,6 +139,8 @@ internal func spawn(_ path: String, args: [String]) throws {
}
}

#endif // os(macOS)

struct CFileStream: TextOutputStream {
var fp: UnsafeMutablePointer<FILE>

Expand All @@ -153,5 +155,3 @@ struct CFileStream: TextOutputStream {

var standardOutput = CFileStream(fp: stdout)
var standardError = CFileStream(fp: stderr)

#endif // os(macOS)

0 comments on commit 45339b6

Please sign in to comment.