xref: /xnu-11215.81.4/tools/lldbmacros/plugins/iosspeedtracer.py (revision d4514f0bc1d3f944c22d92e68b646ac3fb40d452)
1*d4514f0bSApple OSS Distributions# Feed user stacks to ios/speedtracer
2*d4514f0bSApple OSS Distributions
3*d4514f0bSApple OSS Distributionsdef plugin_init(kernel_target, config, lldb_obj, isConnected):
4*d4514f0bSApple OSS Distributions    """ initialize the common data as required by plugin """
5*d4514f0bSApple OSS Distributions    return None
6*d4514f0bSApple OSS Distributions
7*d4514f0bSApple OSS Distributionsdef plugin_execute(command_name, result_output):
8*d4514f0bSApple OSS Distributions    """ The xnu framework will call this function with output of a command.
9*d4514f0bSApple OSS Distributions        The options for returning are as follows
10*d4514f0bSApple OSS Distributions        returns:  (status, outstr, further_cmds)
11*d4514f0bSApple OSS Distributions           status: Boolean - specifying whether plugin execution succeeded(True) or failed. If failed then xnu will stop doing any further work with this command.
12*d4514f0bSApple OSS Distributions           outstr: str - string output for user to be printed at the prompt
13*d4514f0bSApple OSS Distributions           further_cmds: [] of str - this holds set of commands to execute at the lldb prompt. Empty array if nothing is required.
14*d4514f0bSApple OSS Distributions    """
15*d4514f0bSApple OSS Distributions    import subprocess,os
16*d4514f0bSApple OSS Distributions    status = True
17*d4514f0bSApple OSS Distributions    outstr = ''
18*d4514f0bSApple OSS Distributions    further_cmds = []
19*d4514f0bSApple OSS Distributions
20*d4514f0bSApple OSS Distributions    if command_name != 'showtaskuserstacks' :
21*d4514f0bSApple OSS Distributions        status = False
22*d4514f0bSApple OSS Distributions    else:
23*d4514f0bSApple OSS Distributions        ios_process = subprocess.Popen([os.path.join(os.path.dirname(os.path.abspath(__file__)), "iosspeedtracer.sh")], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
24*d4514f0bSApple OSS Distributions
25*d4514f0bSApple OSS Distributions        outstr += ios_process.communicate(input=result_output)[0]
26*d4514f0bSApple OSS Distributions
27*d4514f0bSApple OSS Distributions    return (status, outstr, further_cmds)
28*d4514f0bSApple OSS Distributions
29*d4514f0bSApple OSS Distributionsdef plugin_cleanup():
30*d4514f0bSApple OSS Distributions    """ A cleanup call from xnu which is a signal to wrap up any open file descriptors etc. """
31*d4514f0bSApple OSS Distributions    return None
32*d4514f0bSApple OSS Distributions
33*d4514f0bSApple OSS Distributions
34