Kiran Bandla
65 lines
1.5 KiB
Python
Executable File
65 lines
1.5 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
"""
|
|
(c) Immunity, Inc. 2004-2007
|
|
|
|
|
|
U{Immunity Inc.<http://www.immunityinc.com>}
|
|
|
|
Immunity PyCommand Template
|
|
|
|
"""
|
|
|
|
__VERSION__ = '0.0'
|
|
|
|
import immlib
|
|
import getopt
|
|
|
|
DESC= "Immunity PyCommand Template" #description used by PyCommands GUI
|
|
|
|
def usage(imm):
|
|
""" All the options"""
|
|
imm.Log("!template example command")
|
|
imm.Log("!template [-a] [-b] [-c] ",focus=1) # focus the usage
|
|
|
|
|
|
def main(args):
|
|
imm = immlib.Debugger()
|
|
|
|
if not args:
|
|
imm.Log("### Immunity's PyCommand template ###")
|
|
return "Command ok - no args"
|
|
try:
|
|
opts, argo = getopt.getopt(args, "a:bc:")
|
|
except getopt.GetoptError: #get args, if error, show usage
|
|
usage(imm)
|
|
return "Bad argument %s" % args[0]
|
|
|
|
#parsing args
|
|
for o,a in opts:
|
|
if o == "-a":
|
|
#processing args
|
|
ret=processA(imm,a)
|
|
elif o == "-b":
|
|
ret=processB(imm,a)
|
|
elif o == "-c":
|
|
ret=processC(imm,a)
|
|
|
|
#ret is the string shown at status bar
|
|
return ret
|
|
|
|
|
|
|
|
def processA(imm,arg):
|
|
"""do whatever"""
|
|
imm.Log("Argument received: %s" % str(arg))
|
|
return "Command ok with: %s" %str(arg) #string, string, string!
|
|
|
|
def processB(imm,arg):
|
|
imm.Log("Argument received: %s" % str(arg))
|
|
return "Command ok with: %s" %str(arg)
|
|
|
|
def processC(imm,arg):
|
|
imm.Log("Argument received: %s" % str(arg))
|
|
return "Command ok with: %s" %str(arg)
|
|
|