The beginnings of strace. At the moment prints only syscall number and
return value. git-svn-id: file:///srv/svn/repos/haiku/trunk/current@11319 a95241bf-73f2-0310-859d-f6bbb57e9c96
This commit is contained in:
Ingo Weinhold
@@ -0,0 +1,314 @@
|
|||||||
|
// strace.cpp
|
||||||
|
|
||||||
|
#include <ctype.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <string.h>
|
||||||
|
|
||||||
|
#include <debugger.h>
|
||||||
|
#include <image.h>
|
||||||
|
|
||||||
|
static const char *kDefaultCommandName = "strace";
|
||||||
|
|
||||||
|
// usage
|
||||||
|
static const char *kUsage =
|
||||||
|
"Usage: %s [ <options> ] [ <thread or team ID> | <executable with args> ]\n"
|
||||||
|
"\n"
|
||||||
|
"Traces the the syscalls of a thread or a team. If an executable with\n"
|
||||||
|
"arguments is supplied, it is loaded and it's main thread traced.\n"
|
||||||
|
"\n"
|
||||||
|
"Options:\n"
|
||||||
|
" -a - Don't print syscall arguments.\n"
|
||||||
|
" -c - Don't colorize output.\n"
|
||||||
|
" -f - Fast mode. Syscall arguments contents aren't retrieved.\n"
|
||||||
|
" -h, --help - Print this text.\n"
|
||||||
|
" -r - Don't print syscall return values.\n"
|
||||||
|
" -s - Also trace all threads spawned by the supplied thread,\n"
|
||||||
|
" respectively the loaded executable's main thread.\n"
|
||||||
|
" -T - Trace all threads of the supplied or loaded executable's\n"
|
||||||
|
" team. If an ID is supplied, it is interpreted as a team\n"
|
||||||
|
" ID."
|
||||||
|
;
|
||||||
|
|
||||||
|
// command line args
|
||||||
|
static int sArgc;
|
||||||
|
static const char *const *sArgv;
|
||||||
|
|
||||||
|
// print_usage
|
||||||
|
void
|
||||||
|
print_usage(bool error)
|
||||||
|
{
|
||||||
|
// get nice program name
|
||||||
|
const char *programName = (sArgc > 0 ? sArgv[0] : kDefaultCommandName);
|
||||||
|
if (const char *lastSlash = strrchr(programName, '/'))
|
||||||
|
programName = lastSlash + 1;
|
||||||
|
|
||||||
|
// print usage
|
||||||
|
fprintf((error ? stderr : stdout), kUsage, programName);
|
||||||
|
}
|
||||||
|
|
||||||
|
// print_usage_and_exit
|
||||||
|
static
|
||||||
|
void
|
||||||
|
print_usage_and_exit(bool error)
|
||||||
|
{
|
||||||
|
print_usage(error);
|
||||||
|
exit(error ? 1 : 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
// get_id
|
||||||
|
static
|
||||||
|
bool
|
||||||
|
get_id(const char *str, int32 &id)
|
||||||
|
{
|
||||||
|
int32 len = strlen(str);
|
||||||
|
for (int32 i = 0; i < len; i++) {
|
||||||
|
if (!isdigit(str[i]))
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
id = atol(str);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
// load_program
|
||||||
|
thread_id
|
||||||
|
load_program(const char *const *args, int32 argCount)
|
||||||
|
{
|
||||||
|
// TODO: We need to find the program in the PATH, if no absolute or relative
|
||||||
|
// path has been given (i.e. only a name).
|
||||||
|
return load_image(argCount, (const char**)args, (const char**)environ);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// set_team_debugging_flags
|
||||||
|
static
|
||||||
|
void
|
||||||
|
set_team_debugging_flags(port_id nubPort, int32 flags)
|
||||||
|
{
|
||||||
|
debug_nub_set_team_flags message;
|
||||||
|
message.flags = flags;
|
||||||
|
|
||||||
|
while (true) {
|
||||||
|
status_t error = write_port(nubPort, B_DEBUG_MESSAGE_SET_TEAM_FLAGS,
|
||||||
|
&message, sizeof(message));
|
||||||
|
if (error == B_OK)
|
||||||
|
return;
|
||||||
|
|
||||||
|
if (error != B_INTERRUPTED) {
|
||||||
|
fprintf(stderr, "Failed to set team debug flags: %s\n",
|
||||||
|
strerror(error));
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// set_thread_debugging_flags
|
||||||
|
static
|
||||||
|
void
|
||||||
|
set_thread_debugging_flags(port_id nubPort, thread_id thread, int32 flags)
|
||||||
|
{
|
||||||
|
debug_nub_set_thread_flags message;
|
||||||
|
message.thread = thread;
|
||||||
|
message.flags = flags;
|
||||||
|
|
||||||
|
while (true) {
|
||||||
|
status_t error = write_port(nubPort, B_DEBUG_MESSAGE_SET_THREAD_FLAGS,
|
||||||
|
&message, sizeof(message));
|
||||||
|
if (error == B_OK)
|
||||||
|
return;
|
||||||
|
|
||||||
|
if (error != B_INTERRUPTED) {
|
||||||
|
fprintf(stderr, "Failed to set thread debug flags: %s\n",
|
||||||
|
strerror(error));
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// run_thread
|
||||||
|
static
|
||||||
|
void
|
||||||
|
run_thread(port_id nubPort, thread_id thread)
|
||||||
|
{
|
||||||
|
debug_nub_run_thread message;
|
||||||
|
message.thread = thread;
|
||||||
|
|
||||||
|
while (true) {
|
||||||
|
status_t error = write_port(nubPort, B_DEBUG_MESSAGE_RUN_THREAD,
|
||||||
|
&message, sizeof(message));
|
||||||
|
if (error == B_OK)
|
||||||
|
return;
|
||||||
|
|
||||||
|
if (error != B_INTERRUPTED) {
|
||||||
|
fprintf(stderr, "Failed to run thread %ld: %s\n",
|
||||||
|
thread, strerror(error));
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// main
|
||||||
|
int
|
||||||
|
main(int argc, const char *const *argv)
|
||||||
|
{
|
||||||
|
sArgc = argc;
|
||||||
|
sArgv = argv;
|
||||||
|
|
||||||
|
// parameters
|
||||||
|
const char *const *programArgs = NULL;
|
||||||
|
int32 programArgCount = 0;
|
||||||
|
bool printArguments = true;
|
||||||
|
bool colorize = true;
|
||||||
|
bool fastMode = false;
|
||||||
|
bool printReturnValues = true;
|
||||||
|
bool traceChildThreads = false;
|
||||||
|
bool traceTeam = false;
|
||||||
|
|
||||||
|
// parse arguments
|
||||||
|
for (int argi = 1; argi < argc; argi++) {
|
||||||
|
const char *arg = argv[argi];
|
||||||
|
if (arg[0] == '-') {
|
||||||
|
if (strcmp(arg, "-h") == 0 || strcmp(arg, "--help") == 0) {
|
||||||
|
print_usage_and_exit(false);
|
||||||
|
} else if (strcmp(arg, "-a") == 0) {
|
||||||
|
printArguments = false;
|
||||||
|
} else if (strcmp(arg, "-c") == 0) {
|
||||||
|
colorize = false;
|
||||||
|
} else if (strcmp(arg, "-f") == 0) {
|
||||||
|
fastMode = true;
|
||||||
|
} else if (strcmp(arg, "-r") == 0) {
|
||||||
|
printReturnValues = false;
|
||||||
|
} else if (strcmp(arg, "-s") == 0) {
|
||||||
|
traceChildThreads = true;
|
||||||
|
} else if (strcmp(arg, "-T") == 0) {
|
||||||
|
traceTeam = true;
|
||||||
|
} else {
|
||||||
|
print_usage_and_exit(true);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
programArgs = argv + argi;
|
||||||
|
programArgCount = argc - argi;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// check parameters
|
||||||
|
if (!programArgs)
|
||||||
|
print_usage_and_exit(true);
|
||||||
|
|
||||||
|
thread_id thread = -1;
|
||||||
|
team_id team = -1;
|
||||||
|
if (programArgCount > 1
|
||||||
|
|| !get_id(*programArgs, (traceTeam ? team : thread))) {
|
||||||
|
// we've been given an executable and need to load it
|
||||||
|
thread = load_program(programArgs, programArgCount);
|
||||||
|
}
|
||||||
|
|
||||||
|
// get the team ID, if we have none yet
|
||||||
|
if (team < 0) {
|
||||||
|
thread_info threadInfo;
|
||||||
|
status_t error = get_thread_info(thread, &threadInfo);
|
||||||
|
if (error != B_OK) {
|
||||||
|
fprintf(stderr, "Failed to get info for thread %ld: %s\n", thread,
|
||||||
|
strerror(error));
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
team = threadInfo.team;
|
||||||
|
}
|
||||||
|
|
||||||
|
// create a debugger port
|
||||||
|
port_id debuggerPort = create_port(10, "debugger port");
|
||||||
|
if (debuggerPort < 0) {
|
||||||
|
fprintf(stderr, "Failed to create debugger port: %s\n",
|
||||||
|
strerror(debuggerPort));
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
// install ourselves as the team debugger
|
||||||
|
port_id nubPort = install_team_debugger(team, debuggerPort);
|
||||||
|
if (nubPort < 0) {
|
||||||
|
fprintf(stderr, "Failed to install team debugger: %s\n",
|
||||||
|
strerror(nubPort));
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
// set team debugging flags
|
||||||
|
int32 teamDebugFlags = (fastMode ? B_TEAM_DEBUG_SYSCALL_FAST_TRACE : 0)
|
||||||
|
| (traceTeam ? B_TEAM_DEBUG_POST_SYSCALL : 0);
|
||||||
|
set_team_debugging_flags(nubPort, teamDebugFlags);
|
||||||
|
|
||||||
|
// set thread debugging flags
|
||||||
|
if (thread >= 0) {
|
||||||
|
int32 threadDebugFlags = 0;
|
||||||
|
if (!traceTeam) {
|
||||||
|
threadDebugFlags = B_THREAD_DEBUG_POST_SYSCALL
|
||||||
|
| (traceChildThreads
|
||||||
|
? B_THREAD_DEBUG_SYSCALL_TRACE_CHILD_THREADS : 0);
|
||||||
|
}
|
||||||
|
set_thread_debugging_flags(nubPort, thread, threadDebugFlags);
|
||||||
|
|
||||||
|
// resume the target thread to be sure, it's running
|
||||||
|
resume_thread(thread);
|
||||||
|
}
|
||||||
|
|
||||||
|
// debug loop
|
||||||
|
while (true) {
|
||||||
|
int32 code;
|
||||||
|
debug_debugger_message_data message;
|
||||||
|
ssize_t messageSize = read_port(debuggerPort, &code, &message,
|
||||||
|
sizeof(message));
|
||||||
|
|
||||||
|
if (messageSize < 0) {
|
||||||
|
if (messageSize == B_INTERRUPTED)
|
||||||
|
continue;
|
||||||
|
// Our {read,write}_port() return B_BAD_PORT_ID when being interrupted!
|
||||||
|
//port_info debuggerPortInfo;
|
||||||
|
//if (messageSize == B_BAD_PORT_ID
|
||||||
|
// && get_port_info(debuggerPort, &debuggerPortInfo) == B_OK) {
|
||||||
|
// continue;
|
||||||
|
//}
|
||||||
|
|
||||||
|
fprintf(stderr, "Reading from debugger port failed: %s\n",
|
||||||
|
strerror(messageSize));
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
thread_id concernedThread = -1;
|
||||||
|
switch (code) {
|
||||||
|
case B_DEBUGGER_MESSAGE_THREAD_STOPPED:
|
||||||
|
concernedThread = message.thread_stopped.thread;
|
||||||
|
printf("B_DEBUGGER_MESSAGE_THREAD_STOPPED: thread: %ld\n", concernedThread);
|
||||||
|
break;
|
||||||
|
case B_DEBUGGER_MESSAGE_PRE_SYSCALL:
|
||||||
|
concernedThread = message.pre_syscall.thread;
|
||||||
|
printf("B_DEBUGGER_MESSAGE_PRE_SYSCALL: thread: %ld\n", concernedThread);
|
||||||
|
break;
|
||||||
|
case B_DEBUGGER_MESSAGE_SIGNAL_RECEIVED:
|
||||||
|
concernedThread = message.signal_received.thread;
|
||||||
|
printf("B_DEBUGGER_MESSAGE_SIGNAL_RECEIVED: thread: %ld\n", concernedThread);
|
||||||
|
break;
|
||||||
|
case B_DEBUGGER_MESSAGE_POST_SYSCALL:
|
||||||
|
{
|
||||||
|
concernedThread = message.post_syscall.thread;
|
||||||
|
printf("[%6ld] syscall: %lu, return value: %llx\n",
|
||||||
|
concernedThread, message.post_syscall.syscall,
|
||||||
|
message.post_syscall.return_value);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case B_DEBUGGER_MESSAGE_TEAM_DELETED:
|
||||||
|
{
|
||||||
|
printf("B_DEBUGGER_MESSAGE_TEAM_DELETED: team: %ld\n",
|
||||||
|
message.team_deleted.team);
|
||||||
|
exit(0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// tell the thread to continue
|
||||||
|
if (concernedThread >= 0) {
|
||||||
|
run_thread(nubPort, concernedThread);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user