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