1# 2# gdb helper commands and functions for Linux kernel debugging 3# 4# task & thread tools 5# 6# Copyright (c) Siemens AG, 2011-2013 7# 8# Authors: 9# Jan Kiszka <jan.kiszka@siemens.com> 10# 11# This work is licensed under the terms of the GNU GPL version 2. 12# 13 14import gdb 15 16from linux import utils 17 18 19task_type = utils.CachedType("struct task_struct") 20 21 22def task_lists(): 23 task_ptr_type = task_type.get_type().pointer() 24 init_task = gdb.parse_and_eval("init_task").address 25 t = g = init_task 26 27 while True: 28 while True: 29 yield t 30 31 t = utils.container_of(t['thread_group']['next'], 32 task_ptr_type, "thread_group") 33 if t == g: 34 break 35 36 t = g = utils.container_of(g['tasks']['next'], 37 task_ptr_type, "tasks") 38 if t == init_task: 39 return 40 41 42def get_task_by_pid(pid): 43 for task in task_lists(): 44 if int(task['pid']) == pid: 45 return task 46 return None 47 48 49class LxTaskByPidFunc(gdb.Function): 50 """Find Linux task by PID and return the task_struct variable. 51 52$lx_task_by_pid(PID): Given PID, iterate over all tasks of the target and 53return that task_struct variable which PID matches.""" 54 55 def __init__(self): 56 super(LxTaskByPidFunc, self).__init__("lx_task_by_pid") 57 58 def invoke(self, pid): 59 task = get_task_by_pid(pid) 60 if task: 61 return task.dereference() 62 else: 63 raise gdb.GdbError("No task of PID " + str(pid)) 64 65 66LxTaskByPidFunc() 67 68 69class LxPs(gdb.Command): 70 """Dump Linux tasks.""" 71 72 def __init__(self): 73 super(LxPs, self).__init__("lx-ps", gdb.COMMAND_DATA) 74 75 def invoke(self, arg, from_tty): 76 for task in task_lists(): 77 gdb.write("{address} {pid} {comm}\n".format( 78 address=task, 79 pid=task["pid"], 80 comm=task["comm"].string())) 81 82 83LxPs() 84 85 86thread_info_type = utils.CachedType("struct thread_info") 87 88ia64_task_size = None 89 90 91def get_thread_info(task): 92 thread_info_ptr_type = thread_info_type.get_type().pointer() 93 if utils.is_target_arch("ia64"): 94 global ia64_task_size 95 if ia64_task_size is None: 96 ia64_task_size = gdb.parse_and_eval("sizeof(struct task_struct)") 97 thread_info_addr = task.address + ia64_task_size 98 thread_info = thread_info_addr.cast(thread_info_ptr_type) 99 else: 100 if task.type.fields()[0].type == thread_info_type.get_type(): 101 return task['thread_info'] 102 thread_info = task['stack'].cast(thread_info_ptr_type) 103 return thread_info.dereference() 104 105 106class LxThreadInfoFunc (gdb.Function): 107 """Calculate Linux thread_info from task variable. 108 109$lx_thread_info(TASK): Given TASK, return the corresponding thread_info 110variable.""" 111 112 def __init__(self): 113 super(LxThreadInfoFunc, self).__init__("lx_thread_info") 114 115 def invoke(self, task): 116 return get_thread_info(task) 117 118 119LxThreadInfoFunc() 120 121 122class LxThreadInfoByPidFunc (gdb.Function): 123 """Calculate Linux thread_info from task variable found by pid 124 125$lx_thread_info_by_pid(PID): Given PID, return the corresponding thread_info 126variable.""" 127 128 def __init__(self): 129 super(LxThreadInfoByPidFunc, self).__init__("lx_thread_info_by_pid") 130 131 def invoke(self, pid): 132 task = get_task_by_pid(pid) 133 if task: 134 return get_thread_info(task.dereference()) 135 else: 136 raise gdb.GdbError("No task of PID " + str(pid)) 137 138 139LxThreadInfoByPidFunc() 140