X-Git-Url: https://git.saurik.com/apple/xnu.git/blobdiff_plain/3903760236c30e3b5ace7a4eefac3a269d68957c..5ba3f43ea354af8ad55bea84372a2bc834d8757c:/tools/lldbmacros/ktrace.py

diff --git a/tools/lldbmacros/ktrace.py b/tools/lldbmacros/ktrace.py
old mode 100644
new mode 100755
index 1f07e5d35..ac6ddbe34
--- a/tools/lldbmacros/ktrace.py
+++ b/tools/lldbmacros/ktrace.py
@@ -72,7 +72,7 @@ def GetKdebugTypefilter(typefilter):
             if print_class:
                 subclasses[j] = element
 
-        # if any of the bits were set in a class, print the entire class
+        ## if any of the bits were set in a class, print the entire class
         if print_class:
             out_str += '{:<20s}'.format(GetKdebugClassName(i))
             for element in subclasses:
@@ -93,14 +93,14 @@ def ShowKdebugTypefilter(cmd_args=None):
     """
 
     if cmd_args:
-        print GetKdebugTypefilter.header
-        print '-' * len(GetKdebugTypefilter.header)
-
         typefilter = kern.GetValueFromAddress(cmd_args[0], 'typefilter_t')
         if unsigned(typefilter) == 0:
             raise ArgumentError('argument provided is NULL')
 
-        print GetKdebugTypefilter()
+        print GetKdebugTypefilter.header
+        print '-' * len(GetKdebugTypefilter.header)
+
+        print GetKdebugTypefilter(typefilter)
         return
 
     typefilter = kern.globals.kdbg_typefilter
@@ -110,3 +110,142 @@ def ShowKdebugTypefilter(cmd_args=None):
     print GetKdebugTypefilter.header
     print '-' * len(GetKdebugTypefilter.header)
     print GetKdebugTypefilter(typefilter)
+
+def GetKdebugStatus():
+    """ Get a string summary of the kdebug subsystem.
+    """
+    out = ''
+
+    kdebug_flags = kern.globals.kd_ctrl_page.kdebug_flags
+    out += 'kdebug flags: {}\n'.format(xnudefines.GetStateString(xnudefines.kdebug_flags_strings, kdebug_flags))
+    events = kern.globals.nkdbufs
+    buf_mb = events * (64 if kern.arch == 'x86_64' or kern.arch.startswith('arm64') else 32) / 1000000
+    out += 'events allocated: {:<d} ({:<d} MB)\n'.format(events, buf_mb)
+    out += 'enabled: {}\n'.format('yes' if kern.globals.kdebug_enable != 0 else 'no')
+    if kdebug_flags & xnudefines.kdebug_typefilter_check:
+        out += 'typefilter:\n'
+        out += GetKdebugTypefilter.header + '\n'
+        out += '-' * len(GetKdebugTypefilter.header) + '\n'
+        typefilter = kern.globals.kdbg_typefilter
+        if unsigned(typefilter) != 0:
+            out += GetKdebugTypefilter(typefilter)
+
+    return out
+
+@lldb_command('showkdebug')
+def ShowKdebug(cmd_args=None):
+    """ Show the current kdebug state.
+
+        usage: showkdebug
+    """
+
+    print GetKdebugStatus()
+
+@lldb_type_summary(['kperf_timer'])
+@header('{:<10s} {:<7s} {:<20s}'.format('period-ns', 'action', 'pending'))
+def GetKperfTimerSummary(timer):
+    """ Get a string summary of a kperf timer.
+
+        params:
+            timer: the kperf_timer object to get a summary of
+    """
+    return '{:<10d} {:<7d} {:<20x}\n'.format(
+        kern.GetNanotimeFromAbstime(timer.period), timer.actionid, timer.pending_cpus)
+
+@lldb_type_summary(['action'])
+@header('{:<10s} {:<20s} {:<20s}'.format('pid-filter', 'user-data', 'samplers'))
+def GetKperfActionSummary(action):
+    """ Get a string summary of a kperf action.
+
+        params:
+            action: the action object to get a summary of
+    """
+    samplers = xnudefines.GetStateString(xnudefines.kperf_samplers_strings, action.sample)
+    return '{:<10s} {:<20x} {:<20s}\n'.format(
+        '-' if action.pid_filter < 0 else str(action.pid_filter), action.userdata, samplers)
+
+def GetKperfStatus():
+    """ Get a string summary of the kperf subsystem.
+    """
+    out = ''
+
+    kperf_status = kern.globals.sampling_status
+    out += 'sampling: '
+    if kperf_status == 0:
+        out += 'off\n'
+    elif kperf_status == 1:
+        out += 'on\n'
+    elif kperf_status == 2:
+        out += 'shutting down\n'
+    else:
+        out += 'unknown\n'
+
+    pet = 0# kern.globals.pet_running
+    pet_timer_id = kern.globals.pet_timer_id
+    if pet != 0:
+        pet_idle_rate = kern.globals.pet_idle_rate
+        out += 'legacy PET is active (timer = {:<d}, idle rate = {:<d})\n'.format(pet_timer_id, pet_idle_rate)
+    else:
+        out += 'legacy PET is off\n'
+
+    lw_pet = kern.globals.kperf_lightweight_pet_active
+    if lw_pet != 0:
+        lw_pet_gen = kern.globals.kperf_pet_gen
+        out += 'lightweight PET is active (timer = {:<d}, generation count = {:<d})\n'.format(pet_timer_id, lw_pet_gen)
+    else:
+        out += 'lightweight PET is off\n'
+
+    actions = kern.globals.actionc
+    actions_arr = kern.globals.actionv
+
+    out += 'actions:\n'
+    out += '{:<5s} '.format('id') + GetKperfActionSummary.header + '\n'
+    for i in xrange(0, actions):
+        out += '{:<5d} '.format(i) + GetKperfActionSummary(actions_arr[i])
+
+    timers = kern.globals.kperf_timerc
+    timers_arr = kern.globals.kperf_timerv
+
+    out += 'timers:\n'
+    out += '{:<5s} '.format('id') + GetKperfTimerSummary.header + '\n'
+    for i in xrange(0, timers):
+        out += '{:<5d} '.format(i) + GetKperfTimerSummary(timers_arr[i])
+
+    return out
+
+def GetKtraceStatus():
+    """ Get a string summary of the ktrace subsystem.
+    """
+    out = ''
+
+    state = kern.globals.ktrace_state
+    if state == GetEnumValue('ktrace_state::KTRACE_STATE_OFF'):
+        out += 'ktrace is off\n'
+    else:
+        out += 'ktrace is active ('
+        if state == GetEnumValue('ktrace_state::KTRACE_STATE_FG'):
+            out += 'foreground)'
+        else:
+            out += 'background)'
+        out += '\n'
+        owner = kern.globals.ktrace_last_owner_execname
+        out += 'owned by: {0: <s}\n'.format(owner)
+        active_mask = kern.globals.ktrace_active_mask
+        out += 'active systems: {:<#x}\n'.format(active_mask)
+
+    return out
+
+@lldb_command('showktrace')
+def ShowKtrace(cmd_args=None):
+    """ Show the current ktrace state, including subsystems.
+
+        usage: showktrace
+    """
+
+    print GetKtraceStatus()
+    print ' '
+    print 'kdebug:'
+    print GetKdebugStatus()
+    print ' '
+    print 'kperf:'
+    print GetKperfStatus()