The iterator does not return any task_struct from the thread_group list because the first condition in the 'if not t or ...' will only be the first time None. Instead of keeping track of the state ourself in the next() function, we fall back using Python's generator. Signed-off-by: Daniel Wagner <daniel.wagner@bmw-carit.de> Signed-off-by: Jan Kiszka <jan.kiszka@siemens.com> Cc: Thomas Gleixner <tglx@linutronix.de> Cc: Jason Wessel <jason.wessel@windriver.com> Cc: Andi Kleen <andi@firstfloor.org> Cc: Ben Widawsky <ben@bwidawsk.net> Cc: Borislav Petkov <bp@suse.de> Signed-off-by: Andrew Morton <akpm@linux-foundation.org> Signed-off-by: Linus Torvalds <torvalds@linux-foundation.org>
		
			
				
	
	
		
			100 lines
		
	
	
	
		
			2.5 KiB
			
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			100 lines
		
	
	
	
		
			2.5 KiB
			
		
	
	
	
		
			Python
		
	
	
	
	
	
#
 | 
						|
# gdb helper commands and functions for Linux kernel debugging
 | 
						|
#
 | 
						|
#  task & thread tools
 | 
						|
#
 | 
						|
# Copyright (c) Siemens AG, 2011-2013
 | 
						|
#
 | 
						|
# Authors:
 | 
						|
#  Jan Kiszka <jan.kiszka@siemens.com>
 | 
						|
#
 | 
						|
# This work is licensed under the terms of the GNU GPL version 2.
 | 
						|
#
 | 
						|
 | 
						|
import gdb
 | 
						|
 | 
						|
from linux import utils
 | 
						|
 | 
						|
 | 
						|
task_type = utils.CachedType("struct task_struct")
 | 
						|
 | 
						|
def task_lists():
 | 
						|
    global task_type
 | 
						|
    task_ptr_type = task_type.get_type().pointer()
 | 
						|
    init_task = gdb.parse_and_eval("init_task").address
 | 
						|
    t = g = init_task
 | 
						|
 | 
						|
    while True:
 | 
						|
        while True:
 | 
						|
            yield t
 | 
						|
 | 
						|
            t = utils.container_of(t['thread_group']['next'],
 | 
						|
                                   task_ptr_type, "thread_group")
 | 
						|
            if t == g:
 | 
						|
                break
 | 
						|
 | 
						|
        t = g = utils.container_of(g['tasks']['next'],
 | 
						|
                                   task_ptr_type, "tasks")
 | 
						|
        if t == init_task:
 | 
						|
            return
 | 
						|
 | 
						|
def get_task_by_pid(pid):
 | 
						|
    for task in task_lists():
 | 
						|
        if int(task['pid']) == pid:
 | 
						|
            return task
 | 
						|
    return None
 | 
						|
 | 
						|
 | 
						|
class LxTaskByPidFunc(gdb.Function):
 | 
						|
    """Find Linux task by PID and return the task_struct variable.
 | 
						|
 | 
						|
$lx_task_by_pid(PID): Given PID, iterate over all tasks of the target and
 | 
						|
return that task_struct variable which PID matches."""
 | 
						|
 | 
						|
    def __init__(self):
 | 
						|
        super(LxTaskByPidFunc, self).__init__("lx_task_by_pid")
 | 
						|
 | 
						|
    def invoke(self, pid):
 | 
						|
        task = get_task_by_pid(pid)
 | 
						|
        if task:
 | 
						|
            return task.dereference()
 | 
						|
        else:
 | 
						|
            raise gdb.GdbError("No task of PID " + str(pid))
 | 
						|
 | 
						|
 | 
						|
LxTaskByPidFunc()
 | 
						|
 | 
						|
 | 
						|
thread_info_type = utils.CachedType("struct thread_info")
 | 
						|
 | 
						|
ia64_task_size = None
 | 
						|
 | 
						|
 | 
						|
def get_thread_info(task):
 | 
						|
    global thread_info_type
 | 
						|
    thread_info_ptr_type = thread_info_type.get_type().pointer()
 | 
						|
    if utils.is_target_arch("ia64"):
 | 
						|
        global ia64_task_size
 | 
						|
        if ia64_task_size is None:
 | 
						|
            ia64_task_size = gdb.parse_and_eval("sizeof(struct task_struct)")
 | 
						|
        thread_info_addr = task.address + ia64_task_size
 | 
						|
        thread_info = thread_info_addr.cast(thread_info_ptr_type)
 | 
						|
    else:
 | 
						|
        thread_info = task['stack'].cast(thread_info_ptr_type)
 | 
						|
    return thread_info.dereference()
 | 
						|
 | 
						|
 | 
						|
class LxThreadInfoFunc (gdb.Function):
 | 
						|
    """Calculate Linux thread_info from task variable.
 | 
						|
 | 
						|
$lx_thread_info(TASK): Given TASK, return the corresponding thread_info
 | 
						|
variable."""
 | 
						|
 | 
						|
    def __init__(self):
 | 
						|
        super(LxThreadInfoFunc, self).__init__("lx_thread_info")
 | 
						|
 | 
						|
    def invoke(self, task):
 | 
						|
        return get_thread_info(task)
 | 
						|
 | 
						|
 | 
						|
LxThreadInfoFunc()
 |