1*19c3b8c2SApple OSS Distributionsimport json, urllib, urllib2 2*19c3b8c2SApple OSS Distributionsfrom urllib2 import Request, urlopen, HTTPError 3*19c3b8c2SApple OSS Distributions 4*19c3b8c2SApple OSS Distributionskern_version = None 5*19c3b8c2SApple OSS Distributionsdef plugin_init(kernel_target, config, lldb_obj, isConnected): 6*19c3b8c2SApple OSS Distributions """ initialize the common data as required by plugin """ 7*19c3b8c2SApple OSS Distributions global kern_version 8*19c3b8c2SApple OSS Distributions kern_version = str(kernel_target.version) 9*19c3b8c2SApple OSS Distributions 10*19c3b8c2SApple OSS Distributionsdef plugin_execute(command_name, result_output): 11*19c3b8c2SApple OSS Distributions """ The xnu framework will call this function with output of a command. 12*19c3b8c2SApple OSS Distributions The options for returning are as follows 13*19c3b8c2SApple OSS Distributions returns: (status, outstr, further_cmds) 14*19c3b8c2SApple 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. 15*19c3b8c2SApple OSS Distributions outstr: str - string output for user to be printed at the prompt 16*19c3b8c2SApple OSS Distributions further_cmds: [] of str - this holds set of commands to execute at the lldb prompt. Empty array if nothing is required. 17*19c3b8c2SApple OSS Distributions """ 18*19c3b8c2SApple OSS Distributions status = True 19*19c3b8c2SApple OSS Distributions outstr = '' 20*19c3b8c2SApple OSS Distributions further_cmds = [] 21*19c3b8c2SApple OSS Distributions submitvars = {} 22*19c3b8c2SApple OSS Distributions submitvars['log_content']=result_output 23*19c3b8c2SApple OSS Distributions 24*19c3b8c2SApple OSS Distributions submiturl = "https://speedtracer.apple.com/api/v2/trace" 25*19c3b8c2SApple OSS Distributions encoded_data = urllib.urlencode(submitvars) 26*19c3b8c2SApple OSS Distributions request = urllib2.Request(submiturl, encoded_data) 27*19c3b8c2SApple OSS Distributions request.add_header("Accept", "application/json") 28*19c3b8c2SApple OSS Distributions request.add_header("X-ST-GroupName", "core-os") 29*19c3b8c2SApple OSS Distributions try: 30*19c3b8c2SApple OSS Distributions response = urllib2.urlopen(request) 31*19c3b8c2SApple OSS Distributions response_str = response.read() 32*19c3b8c2SApple OSS Distributions j = json.loads(response_str) 33*19c3b8c2SApple OSS Distributions outstr += "\nspeedtracer output:\n\n" 34*19c3b8c2SApple OSS Distributions stacks = j.get("symbolicated_log") 35*19c3b8c2SApple OSS Distributions if stacks: 36*19c3b8c2SApple OSS Distributions outstr += stacks 37*19c3b8c2SApple OSS Distributions else: 38*19c3b8c2SApple OSS Distributions outstr += json.dumps(j) 39*19c3b8c2SApple OSS Distributions except HTTPError as e: 40*19c3b8c2SApple OSS Distributions outstr += "speedtracer replied with\n" + str(e.info()) 41*19c3b8c2SApple OSS Distributions status = False 42*19c3b8c2SApple OSS Distributions 43*19c3b8c2SApple OSS Distributions return (status, outstr, further_cmds) 44*19c3b8c2SApple OSS Distributions 45*19c3b8c2SApple OSS Distributionsdef plugin_cleanup(): 46*19c3b8c2SApple OSS Distributions """ A cleanup call from xnu which is a signal to wrap up any open file descriptors etc. """ 47*19c3b8c2SApple OSS Distributions return None 48*19c3b8c2SApple OSS Distributions 49*19c3b8c2SApple OSS Distributions 50