def do_GsettingsTest(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("GsettingsTest")
e.OutputEquals("gsettings get " + command['<section>'] + " " + command['<value>'],
command['<expectation>'])
print e.r.retVal
def do_String(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("String")
#e.Log(args)
if command["split"]:
split = command['<input>'].split(command['<delimiter>'])
result = ""
for i in command['INDEX']:
result += split[int(i)] + command['<outputdelimiter>']
elif command["join"]: # should be command['use']
result = ""
for s in command['STRINGS']:
result += s + command['<outputdelimiter>']
elif command["replace"]:
result = ""
for s in command['STRINGS']:
result += s.replace(command['<old>'], command['<new>']) + " "
elif command["trim"]:
result = str(command['<input>']).strip()
elif command["clip"]:
start = int(command['<start>'])
end = int(command ['<end>'])
if end == 0:
end = - len(command['<input>'])
result = str(command['<input>'] [start:] [:-end])
print result
e.setExecCodes(command, result, "Pass")
def do_Dot1x(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Dot1x")
e.EvalRetVal("regset --temp Network-eth0 Enable802 yes")
e.EvalRetVal("regset Network-eth0 ca_cert " + command['<cacert>'])
if (command["tls"] == True):
e.Log("Configuring for tls")
e.EvalRetVal("regset Network-eth0 Authentication TLS")
e.EvalRetVal("regset Network-eth0 client_cert " + command['<clientcert>'])
e.EvalRetVal("regset Network-eth0 private_key " + command['<privkey>'])
e.EvalRetVal("regset Network-eth0 private_key_password " + command['<pkeypass>'])
e.EvalRetVal("regset Network-eth0 Authmode " + command['<authmode>'])
elif(command["peap"] == True):
e.Log("Configuring for PEAP")
e.EvalRetVal("regset Network-eth0 Authentication PEAP")
e.EvalRetVal("regset Network-eth0 Authmode " + command['<authmode>'])
elif(command["reset"] == True):
e.Log("Doing dot1x reset")
e.EvalRetVal("regdel Network-eth0 Authmode")
e.EvalRetVal("regdel Network-eth0 Authentication")
e.EvalRetVal("regdel Network-eth0 Enable802")
e.EvalRetVal("regdel Network-eth0 ca_cert")
e.EvalRetVal("regdel Network-eth0 client_cert")
e.EvalRetVal("regdel Network-eth0 private_key")
e.EvalRetVal("regdel Network-eth0 private_key_password")
def do_Sleep(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Sleep")
e.Log("Sleep executing")
time.sleep(int(command['<sec>']))
e.setExecCodes(args, "Lion sleeps 20 h per day", "Pass")
def do_RegTest(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("RegTest")
e.OutputEquals(
"/usr/bin/regget " + command['<section>'] + " " +
command['<value>'], command['<expectation>'])
def do_Sleep(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Sleep")
e.Log("Sleep executing")
time.sleep(int(command['<sec>']))
e.setExecCodes(args, "Lion sleeps 20 h per day", "Pass")
def do_IP(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("IP")
interface = command["<interface>"]
myIp = self.myIP(interface)
e.Log("Found client ip: " + myIp)
if command["equals"]:
if command["<address>"] == myIp:
e.Log("ip addresss equal")
e.setExecCodes("", "ip addresses equal", "Pass")
retVal = "Pass"
else:
e.Log("IP address not equal")
e.setExecCodes("", "ip addresses inequal", "Fail")
retVal = "Fail"
elif command["contains"]:
if myIp.startswith(command["<address>"]):
e.Log("ip addresses contains")
e.setExecCodes("", "ip addresses contains", "Pass")
retVal = "Pass"
else:
e.Log("ip addresses does not contain")
e.setExecCodes("", "ip addresses does not contain", "Fail")
retVal = "Fail"
print retVal
def do_IP(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("IP")
interface = command["<interface>"]
myIp = self.myIP(interface)
e.Log("Found client ip: " + myIp)
if command["equals"]:
if command["<address>"] == myIp:
e.Log("ip addresss equal")
e.setExecCodes("", "ip addresses equal", "Pass")
retVal = "Pass"
else:
e.Log("IP address not equal")
e.setExecCodes("", "ip addresses inequal", "Fail")
retVal = "Fail"
elif command["contains"]:
if myIp.startswith(command["<address>"]):
e.Log("ip addresses contains")
e.setExecCodes("", "ip addresses contains", "Pass")
retVal = "Pass"
else:
e.Log("ip addresses does not contain")
e.setExecCodes("", "ip addresses does not contain", "Fail")
retVal = "Fail"
print retVal
def do_Dot1x(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Dot1x")
e.EvalRetVal("regset --temp Network-eth0 Enable802 yes")
e.EvalRetVal("regset Network-eth0 ca_cert " + command['<cacert>'])
if (command["tls"] == True):
e.Log("Configuring for tls")
e.EvalRetVal("regset Network-eth0 Authentication TLS")
e.EvalRetVal("regset Network-eth0 client_cert " +
command['<clientcert>'])
e.EvalRetVal("regset Network-eth0 private_key " +
command['<privkey>'])
e.EvalRetVal("regset Network-eth0 private_key_password " +
command['<pkeypass>'])
e.EvalRetVal("regset Network-eth0 Authmode " +
command['<authmode>'])
elif (command["peap"] == True):
e.Log("Configuring for PEAP")
e.EvalRetVal("regset Network-eth0 Authentication PEAP")
e.EvalRetVal("regset Network-eth0 Authmode " +
command['<authmode>'])
elif (command["reset"] == True):
e.Log("Doing dot1x reset")
e.EvalRetVal("regdel Network-eth0 Authmode")
e.EvalRetVal("regdel Network-eth0 Authentication")
e.EvalRetVal("regdel Network-eth0 Enable802")
e.EvalRetVal("regdel Network-eth0 ca_cert")
e.EvalRetVal("regdel Network-eth0 client_cert")
e.EvalRetVal("regdel Network-eth0 private_key")
e.EvalRetVal("regdel Network-eth0 private_key_password")
def do_Skip(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Skip")
e.Log("Skip command executing")
outstring = ("Skip", command['<fromrank>'], command['<torank>'], "null")
e.setExecCodes(args, outstring, "Skip")
def do_Repeat(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Repeat")
outstring = ('Repeat', command['<fromrank>'], command['<torank>'], command['<times>'])
e.Log("Repeat command called")
e.setExecCodes(args, outstring, "Repeat")
def do_String(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("String")
#e.Log(args)
if command["split"]:
split = command['<input>'].split(command['<delimiter>'])
result = ""
for i in command['INDEX']:
result += split[int(i)] + command['<outputdelimiter>']
elif command["join"]: # should be command['use']
result = ""
for s in command['STRINGS']:
result += s + command['<outputdelimiter>']
elif command["replace"]:
result = ""
for s in command['STRINGS']:
result += s.replace(command['<old>'], command['<new>']) + " "
elif command["trim"]:
result = str(command['<input>']).strip()
elif command["clip"]:
start = int(command['<start>'])
end = int(command['<end>'])
if end == 0:
end = -len(command['<input>'])
result = str(command['<input>'][start:][:-end])
print result
e.setExecCodes(command, result, "Pass")
def do_Echo(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Echo")
cmd = "echo -n " + ' '.join(command['ARGS'])
e.Log(cmd)
e.RetOutput(cmd)
print e.r.output
def do_Skip(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Skip")
e.Log("Skip command executing")
outstring = ("Skip", command['<fromrank>'], command['<torank>'],
"null")
e.setExecCodes(args, outstring, "Skip")
def do_Echo(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Echo")
cmd = "echo -n " + ' '.join(command['ARGS'])
e.Log(cmd)
e.RetOutput(cmd)
print e.r.output
def do_Depends(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Depends")
depstring = (command['<ipaddr>'], command['<rank>'])
e.Log("Waiting for 30 secs for %s to finish test case whose rank is %s" % (command['<ipaddr>'], command['<rank>']))
print "Waiting for 30 secs for %s to finish test case whose rank is %s" % (command['<ipaddr>'], command['<rank>'])
time.sleep(30)
e.setExecCodes(args, depstring, "Wait")
def do_Browser(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Browser")
if command["clearHistory"]:
e.Log("Browser clearHistory command is not implemented, marking Pass")
e.setExecCodes(args, "Not implemented", "Pass")
elif command["kiosk"]:
e.Log("Browser kiosk command is not implemented, marking Pass")
e.setExecCodes("Browser kiosk","Not implemented","Pass")
def do_Mcast(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Mcast")
if command['server']:
cmd = "python /tmp/mcastfirmware/server.py %s" %(command['<filepath>'])
e.Spawn(cmd)
elif command['client']:
cmd = "python /tmp/mcastfirmware/client.py"
e.Execute(cmd)
e.Log(e.r.getOutput())
def do_Network(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Network")
if (command["on"] == True):
e.EvalRetVal("service network start")
elif (command["off"] == True):
e.EvalRetVal("service network stop")
elif (command["restart"] == True):
e.EvalRetVal("service network stop")
e.EvalRetVal("service network start")
e.Log(e.r.getOutput())
def do_Network(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Network")
if (command["on"] == True):
e.EvalRetVal("service network start")
elif(command["off"] == True):
e.EvalRetVal("service network stop")
elif(command["restart"] == True):
e.EvalRetVal("service network stop")
e.EvalRetVal("service network start")
e.Log(e.r.getOutput())
def do_Depends(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Depends")
depstring = (command['<ipaddr>'], command['<rank>'])
e.Log(
"Waiting for 30 secs for %s to finish test case whose rank is %s" %
(command['<ipaddr>'], command['<rank>']))
print "Waiting for 30 secs for %s to finish test case whose rank is %s" % (
command['<ipaddr>'], command['<rank>'])
time.sleep(30)
e.setExecCodes(args, depstring, "Wait")
def do_Mcast(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Mcast")
if command['server']:
cmd = "python /tmp/mcastfirmware/server.py %s" % (
command['<filepath>'])
e.Spawn(cmd)
elif command['client']:
cmd = "python /tmp/mcastfirmware/client.py"
e.Execute(cmd)
e.Log(e.r.getOutput())
def do_Browser(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Browser")
if command["clearHistory"]:
e.Log(
"Browser clearHistory command is not implemented, marking Pass"
)
e.setExecCodes(args, "Not implemented", "Pass")
elif command["kiosk"]:
e.Log("Browser kiosk command is not implemented, marking Pass")
e.setExecCodes("Browser kiosk", "Not implemented", "Pass")
def do_SimulateDomain(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("SimulateDomain")
if (command["Logon"] == True):
print "regset --temp System Username " + command['<user>']
print "regset --temp System Domain " + command['<domain>']
print "regset --temp --encrypt System Password " + command[
'<password>']
elif (command["Logoff"] == True):
print "regdel --temp System Username "
print "regdel --temp System Domain "
print "regdel --temp System Password "
def do_File(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("File")
path = command['<path>']
if command['wait']:
print args
wm = pyinotify.WatchManager()
if command['delete']:
mask = pyinotify.IN_DELETE
elif command['create']:
if os.path.exists(path):
e.Log("File already exists, do not have to wait.")
e.setExecCodes(args, "already exists", "Pass")
return
else:
global watchpath
watchpath = path
path = os.path.dirname(path)
mask = pyinotify.IN_CREATE
elif command['write']:
mask = pyinotify.IN_CLOSE_WRITE | pyinotify.IN_CLOSE_NOWRITE
elif command['open']:
mask = pyinotify.IN_OPEN
elif command['move']:
mask = pyinotify.IN_MOVED_FROM | pyinotify.IN_MOVED_TO | pyinotify.IN_MOVED_SELF
elif command['access']:
mask = pyinotify.IN_ACCESS
#mask = mask | pyinotify.IN_ONESHOT
e.Log("Starting inotify to watch file for event")
handler = EventHandler()
notifier = pyinotify.Notifier(wm, handler)
wdd = wm.add_watch(path, mask, rec=True)
notifier.loop()
elif command["contains"]:
e.OutputContains("/bin/cat " + path, command['<expectation>'])
elif command['exists']:
if os.path.exists(path):
e.Log("file exists")
e.setExecCodes("File exists", "exists", "Pass")
else:
e.Log("file does not exist")
e.setExecCodes("File exists", "Doesnt exist", "Fail")
elif command['delete']:
if os.path.exists(path):
os.unlink(path)
elif command['create'] or command['purge']:
f = open(path, "w")
f.close()
print e.r.retVal
def do_Reboot(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Reboot")
e.setExecCodes("Reboot","Unknown error", "Fail")
prepare = 'echo %s|sudo -S shutdown -k +1' %(command['<sudopassword>'])
e.Log("Checking our permissions")
e.Log(prepare)
if e.Execute(prepare) == 0:
cmd = 'shutdown -r +1 &'
e.Log(cmd)
e.Log("Alert: This will wipe out all the variables created before.")
p = e.Execute('echo %s|sudo -S %s' % (command['<sudopassword>'], cmd))
e.setExecCodes("Reboot","Going down in 1 minute", "Reboot")
def do_Math(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Math")
e.Log(args)
if command["add"]:
result = int(command['<one>']) + int(command['<two>'])
elif command["subtract"]:
result = int(command['<two>']) - int(command['<one>'])
elif command["multiply"]:
result = int(command['<one>']) * int(command['<two>'])
elif command["divide"]:
result = int(command['<one>']) / int(command['<two>'])
print result
e.setExecCodes(command, result, "Pass")
def do_Math(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Math")
e.Log(args)
if command["add"]:
result = int(command['<one>']) + int(command['<two>'])
elif command["subtract"]:
result = int(command['<two>']) - int(command['<one>'])
elif command["multiply"]:
result = int(command['<one>']) * int(command['<two>'])
elif command["divide"]:
result = int(command['<one>']) / int(command['<two>'])
print result
e.setExecCodes(command, result, "Pass")
def do_Xenapp(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Xenapp")
if command["application"] or command["desktop"]:
citrixuser = command['<username>']
citrixpass = command['<password>']
citrixurl = command['<url>']
citrixapp = " ".join(command['APPNAME'])
citrixappmod = citrixapp.replace(" ", "_0020")
#driver = webdriver.Firefox()
profpath = os.path.join(os.getenv('HOME'), ".mozilla", "firefox",
"wyse_default")
if os.path.exists(profpath):
e.Log("using existing profile")
profile = FirefoxProfile(profpath)
else:
e.Log("using a temporary profile")
profile = FirefoxProfile()
driver = webdriver.Firefox(profile)
driver.implicitly_wait(60)
driver.get(citrixurl)
#driver.find_element_by_id("skipWizardLink").click()
driver.find_element_by_id("user").clear()
driver.find_element_by_id("user").send_keys(citrixuser)
driver.find_element_by_id("password").clear()
driver.find_element_by_id("password").send_keys(citrixpass)
driver.find_element_by_css_selector("span.rightDoor").click()
if command["desktop"]:
e.Log("finding ")
driver.find_element_by_css_selector(
"#Desktops_Text > span").click()
try:
appid = "a[id*='%s']" % citrixappmod
element = driver.find_element_by_css_selector(appid)
element.click()
#e.setExecCodes(args, "Check app launch using Process alive wfica.orig","Pass")
except Exception as e:
print str(e)
e.Log(str(e))
print "Oops app not found!!", citrixapp
e.setExecCodes(args, "Could not locate app", "Fail")
#The app takes its own sweet time coming up. Should sleep time be a parameter?
print "Waiting for 90 secs before checking for window creation"
time.sleep(90)
chkcmd = "./agent/cli/bin/windowinfo.sh '%s'" % citrixapp
e.RetOutput(chkcmd)
e.Log(e.r.getOutput())
def do_Reboot(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Reboot")
e.setExecCodes("Reboot", "Unknown error", "Fail")
prepare = 'echo %s|sudo -S shutdown -k +1' % (
command['<sudopassword>'])
e.Log("Checking our permissions")
e.Log(prepare)
if e.Execute(prepare) == 0:
cmd = 'shutdown -r +1 &'
e.Log(cmd)
e.Log(
"Alert: This will wipe out all the variables created before.")
p = e.Execute('echo %s|sudo -S %s' %
(command['<sudopassword>'], cmd))
e.setExecCodes("Reboot", "Going down in 1 minute", "Reboot")
def do_Xenapp(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Xenapp")
if command["application"] or command["desktop"]:
citrixuser = command['<username>']
citrixpass = command['<password>']
citrixurl = command['<url>']
citrixapp = " ".join(command['APPNAME'])
citrixappmod = citrixapp.replace(" ", "_0020")
#driver = webdriver.Firefox()
profpath = os.path.join(os.getenv('HOME'),".mozilla", "firefox", "wyse_default")
if os.path.exists(profpath):
e.Log("using existing profile")
profile = FirefoxProfile(profpath)
else:
e.Log("using a temporary profile")
profile = FirefoxProfile()
driver = webdriver.Firefox(profile)
driver.implicitly_wait(60)
driver.get(citrixurl)
#driver.find_element_by_id("skipWizardLink").click()
driver.find_element_by_id("user").clear()
driver.find_element_by_id("user").send_keys(citrixuser)
driver.find_element_by_id("password").clear()
driver.find_element_by_id("password").send_keys(citrixpass)
driver.find_element_by_css_selector("span.rightDoor").click()
if command["desktop"]:
e.Log("finding ")
driver.find_element_by_css_selector("#Desktops_Text > span").click()
try:
appid = "a[id*='%s']" % citrixappmod
element = driver.find_element_by_css_selector(appid)
element.click()
#e.setExecCodes(args, "Check app launch using Process alive wfica.orig","Pass")
except Exception as e:
print str(e)
e.Log(str(e))
print "Oops app not found!!", citrixapp
e.setExecCodes(args, "Could not locate app","Fail")
#The app takes its own sweet time coming up. Should sleep time be a parameter?
print "Waiting for 90 secs before checking for window creation"
time.sleep(90)
chkcmd = "./agent/cli/bin/windowinfo.sh '%s'" % citrixapp
e.RetOutput(chkcmd)
e.Log(e.r.getOutput())
def do_Gdm(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Gdm")
cmd = "pidof X"
pid = e.RetOutput(cmd).strip()
cmd = "sh -c 'xargs -0 < /proc/" + pid + "/cmdline'"
e.Log(cmd)
xcmdline = e.RetOutput(cmd).split()
count = len(xcmdline)
i = 0
while i < count:
if xcmdline[i] == "-auth":
xauthority = xcmdline[i + 1]
i += 1
os.environ['XAUTHORITY'] = xauthority
if command['login']:
for word in command['PHRASE']:
word = word.strip()
if word == "{Tab}":
cmd = 'echo %s|sudo -S xdotool key Tab' % (
command['<sudopassword>'])
elif word == "{Return}":
cmd = 'echo %s|sudo -S xdotool key Return' % (
command['<sudopassword>'])
elif word == "{Clear}":
cmd = 'echo %s|sudo -S xdotool key ctrl+a Delete' % (
command['<sudopassword>'])
elif word == "{Space}":
cmd = 'echo %s|sudo -S xdotool key space' % (
command['<sudopassword>'])
else:
cmd = 'echo %s|sudo -S xdotool type %s' % (
command['<sudopassword>'], word)
e.Log(cmd)
e.Execute(cmd)
time.sleep(2)
e.Log(e.r.output)
else:
cmd = "xdotool key ctrl+alt+Delete"
e.Execute(cmd)
time.sleep(2)
e.Execute("sleep 30 && xdotool key Return &")
e.setExecCodes("Reboot", "Going out in 1/2 minute", "Reboot")
e.Log(e.r.output)
def do_Http(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Http")
try:
if command["download"]:
e.Log("Http downloading")
urllib.urlretrieve (command["<url>"], command["<localfile>"])
e.setExecCodes("Http", "File downloaded", "Pass")
else:
e.Log("Http uploading")
toUpload = UploadFile(command["<localfile>"], 'r')
theHeaders= {'Content-Type': 'text/xml'}
theRequest = urllib2.Request(command["<url>"], toUpload, theHeaders)
response = urllib2.urlopen(theRequest)
toUpload.close()
e.Log("Http command passed")
except Exception as e:
print str(e)
e.Log("Http failed " + str(e))
e.setExecCodes("", str(e), "Fail")
def do_Gdm(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Gdm")
cmd = "pidof X"
pid = e.RetOutput(cmd).strip()
cmd = "sh -c 'xargs -0 < /proc/" + pid + "/cmdline'"
e.Log(cmd)
xcmdline = e.RetOutput(cmd).split()
count = len(xcmdline)
i = 0
while i < count:
if xcmdline[i] == "-auth":
xauthority = xcmdline[i + 1]
i += 1
os.environ['XAUTHORITY'] = xauthority
if command['login']:
for word in command['PHRASE']:
word = word.strip()
if word == "{Tab}":
cmd = 'echo %s|sudo -S xdotool key Tab' %(command['<sudopassword>'])
elif word == "{Return}":
cmd = 'echo %s|sudo -S xdotool key Return' %(command['<sudopassword>'])
elif word == "{Clear}":
cmd = 'echo %s|sudo -S xdotool key ctrl+a Delete' %(command['<sudopassword>'])
elif word == "{Space}":
cmd = 'echo %s|sudo -S xdotool key space' %(command['<sudopassword>'])
else:
cmd = 'echo %s|sudo -S xdotool type %s' %(command['<sudopassword>'], word)
e.Log(cmd)
e.Execute(cmd)
time.sleep(2)
e.Log(e.r.output)
else:
cmd = "xdotool key ctrl+alt+Delete"
e.Execute(cmd)
time.sleep(2)
e.Execute("sleep 30 && xdotool key Return &")
e.setExecCodes("Reboot","Going out in 1/2 minute", "Reboot")
e.Log(e.r.output)
def do_HTTPCounter(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("HTTPCounter")
sudopassword = command['<sudopassword>']
if command["initialize"]:
e.Log("HTTPCounter initalizing")
self.initialize(sudopassword)
e.setExecCodes(args, "Warning: Does not verify execution", "Pass")
elif command["resetCounter"]:
cmd = "echo %s|sudo -S iptables -Z AF" % (sudopassword)
e.Log("HTTPCounter resetCounter")
e.Execute(cmd)
e.setExecCodes(args,"Warning: Does not verify execution","Pass")
elif command["getCount"]:
cmd = "sh -c 'echo %s|sudo -S iptables -vL AF'" % (sudopassword)
lines = e.RetOutput(cmd).split(os.linesep)
output = " GET " + lines[2].split()[0]
output += " POST " + lines[3].split()[0]
output += " PUT " + lines[4].split()[0]
e.r.setOutput(output)
print output
def do_HTTPCounter(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("HTTPCounter")
sudopassword = command['<sudopassword>']
if command["initialize"]:
e.Log("HTTPCounter initalizing")
self.initialize(sudopassword)
e.setExecCodes(args, "Warning: Does not verify execution", "Pass")
elif command["resetCounter"]:
cmd = "echo %s|sudo -S iptables -Z AF" % (sudopassword)
e.Log("HTTPCounter resetCounter")
e.Execute(cmd)
e.setExecCodes(args, "Warning: Does not verify execution", "Pass")
elif command["getCount"]:
cmd = "sh -c 'echo %s|sudo -S iptables -vL AF'" % (sudopassword)
lines = e.RetOutput(cmd).split(os.linesep)
output = " GET " + lines[2].split()[0]
output += " POST " + lines[3].split()[0]
output += " PUT " + lines[4].split()[0]
e.r.setOutput(output)
print output
def do_Http(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Http")
try:
if command["download"]:
e.Log("Http downloading")
urllib.urlretrieve(command["<url>"], command["<localfile>"])
e.setExecCodes("Http", "File downloaded", "Pass")
else:
e.Log("Http uploading")
toUpload = UploadFile(command["<localfile>"], 'r')
theHeaders = {'Content-Type': 'text/xml'}
theRequest = urllib2.Request(command["<url>"], toUpload,
theHeaders)
response = urllib2.urlopen(theRequest)
toUpload.close()
e.Log("Http command passed")
except Exception as e:
print str(e)
e.Log("Http failed " + str(e))
e.setExecCodes("", str(e), "Fail")
def do_Ftp(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Ftp")
try:
ftpconn = ftplib.FTP(command['<server-ip>'])
ftpconn.login(command['<user>'], command['<password>'])
ftpconn.cwd(os.path.dirname(command['<remotefile>']))
if command['upload']:
localfile = open(command['<localfile>'], 'r')
ftpconn.storlines('STOR ' + command['<remotefile>'], localfile)
localfile.close()
ftpconn.quit()
e.setExecCodes("", "File uploaded", "Pass")
elif command['download']:
ftpconn.retrbinary("RETR " + os.path.basename(command['<remotefile>']),
open(command['<localfile>'], 'wb').write)
e.setExecCodes("", "File downloaded", "Pass")
e.Log("Ftp command passed")
except Exception as e:
print str(e)
e.Log("Ftp failed " + str(e))
e.setExecCodes("", str(e), "Fail")
def do_Process(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Process")
if command["monitor"]:
psopts = "/bin/ps -C %s o pcpu,pmem --cumulative --no-heading" % (
command["<name>"])
e.Log("Executing monitor " + psopts)
counter = 0
cpu = float(0)
mem = float(0)
while (counter < int(command["<duration>"])):
pcpu, pmem = self.parse(e.RetOutput(psopts))
cpu = cpu + pcpu
mem = mem + pmem
time.sleep(1)
counter = counter + 1
output = "%s consumed %s%% cpu and %s %% memory" % (
command["<name>"], cpu / counter, mem / counter)
e.Log(output)
print output
if cpu > 0 or mem > 0:
e.setExecCodes("Process monitor", output, "Pass")
else:
e.setExecCodes("Process monitor", output, "Fail")
elif command["kill"]:
killopts = "/usr/bin/killall %s" % (command["<name>"])
e.Log(killopts)
e.EvalRetVal(killopts)
elif command["spawn"]:
spawnopts = ' '.join(command["PATH"])
e.Log(spawnopts)
e.Spawn(spawnopts)
elif command["exec"]:
execopts = ' '.join(command["PATH"])
e.Log(execopts)
execopts = "/bin/sh -c '%s' " % (execopts)
e.EvalRetVal(execopts)
elif command["forfeit"]:
execopts = ' '.join(command["PATH"])
e.Log(execopts)
execopts = "/bin/sh -c '%s' " % (execopts)
e.EvalRetVal(execopts)
e.r.setRetVal("Fail")
elif command["thrive"]:
execopts = ' '.join(command["PATH"])
e.Log(execopts)
execopts = "/bin/sh -c '%s' " % (execopts)
e.EvalRetVal(execopts)
e.r.setRetVal("Pass")
elif command["alive"]:
killopts = "/usr/bin/killall -s 0 %s" % (command["<name>"])
e.Log(killopts)
e.EvalRetVal(killopts)
elif command["dead"]:
killopts = "pidof %s" % (command["<name>"])
e.Log(killopts)
e.EvalRetVal(killopts, 1)
elif command["pid"]:
execopts = "pidof %s" % (command["<name>"])
e.Log(execopts)
e.EvalRetVal(execopts)
elif command["output"]:
# Process output contains 1360x78 xdpyinfo | grep dimensions
execopts = ' '.join(command["PATH"])
e.Log(execopts)
execopts = "/bin/sh -c '%s' " % (execopts)
output = e.RetOutput(execopts)
if command['<expectation>'] in output:
e.setExecCodes(execopts, output, "Pass")
else:
e.setExecCodes(execopts, output, "Fail")
e.Log(e.r.getOutput())
print e.r.getRetVal()
def do_Selenium(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Selenium")
if command["start"]:
e.Spawn("python ./agent/cli/bin/SeleniumWorker.py %s" %
(command['<which>']))
time.sleep(5)
retval, output = self.ReadWorkerResult()
elif command["stop"]:
self.SendWorkerCommand("stop")
retval, output = self.ReadWorkerResult()
elif command["clickcss"]:
self.SendWorkerCommand("clickcss " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["clickname"]:
self.SendWorkerCommand("clickname " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["clickid"]:
self.SendWorkerCommand("clickid " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["clickxpath"]:
self.SendWorkerCommand("clickxpath " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["clickmenuitem"]:
self.SendWorkerCommand("clickmenuitem " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["clearcss"]:
self.SendWorkerCommand("clearcss " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["clearname"]:
self.SendWorkerCommand("clearname " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["clearid"]:
self.SendWorkerCommand("clearid " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["clearxpath"]:
self.SendWorkerCommand("clearxpath " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["open"]:
self.SendWorkerCommand("open " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["fillcss"]:
self.SendWorkerCommand("fillcss " + command['<which>'] + " " +
' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
elif command["fillname"]:
self.SendWorkerCommand("fillname " + command['<which>'] + " " +
' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
elif command["fillid"]:
self.SendWorkerCommand("fillid " + command['<which>'] + " " +
' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
elif command["fillxpath"]:
self.SendWorkerCommand("fillxpath " + command['<which>'] + " " +
' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
elif command["counttags"]:
self.SendWorkerCommand("counttags " + command['<which>'] + " " +
' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
elif command["namecontains"]:
self.SendWorkerCommand("namecontains " + command['<which>'] + " " +
' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
elif command["idcontains"]:
self.SendWorkerCommand("idcontains " + command['<which>'] + " " +
' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
elif command["csscontains"]:
self.SendWorkerCommand("csscontains " + command['<which>'] + " " +
' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
elif command["xpathcontains"]:
self.SendWorkerCommand("xpathcontains " + command['<which>'] +
" " + ' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
elif command["pagecontains"]:
self.SendWorkerCommand("pagecontains " + "null" + " " +
' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
else:
retval, output = "Fail", "Invalid command"
e.Log(output)
e.setExecCodes(args, output, retval)
def do_Desktop(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Desktop")
if command["logout"]:
e.EvalRetVal("/usr/bin/gnome-session-quit --no-prompt")
elif command["resolution"]:
cmd = "xrandr -d :0 --output LVDS --mode " + command["<resolution>"]
e.Log("Desktop Resolution: " + cmd)
e.EvalRetVal(cmd)
elif command["screenshot"]:
scrot = MSSLinux()
for filename in scrot.save(output="scrot.png", screen=-1):
shutil.move("scrot.png", command['<filename>'])
e.Log('File: "{}" created.'.format(filename))
elif command["legacyScreenshot"]:
os.system("gnome-screenshot -b -f " + command['<filename>'])
elif command["windowimg"]:
e.Log("Executing commands for windowimg")
cmd = 'xte "mousemove 0 0"'
e.Log(cmd)
e.Execute(cmd)
cmd = "./agent/cli/bin/windowimg.sh %s %s" % (
command['<filename>'], command['<windowid>'])
e.Log(cmd)
e.EvalRetVal(cmd)
elif command["windowid"]:
e.Log("Executing commands for windowid")
cmd = "./agent/cli/bin/windowid.sh %s" % (command['<process>'])
e.Log(cmd)
e.EvalRetVal(cmd)
elif command["windowfocus"]:
e.Log("Executing commands for windowfocus")
cmd = 'xdotool windowactivate %s' % (command['<windowid>'])
e.Log(cmd)
e.EvalRetVal(cmd)
elif command["windowclose"]:
cmd = 'wmctrl -i -c %s' % (command['<windowid>'])
e.Log("Executing windowclose: " + cmd)
e.EvalRetVal(cmd)
elif command["keydown"]:
cmd = "xte keydown %s" % (command['<key>'])
e.Log(cmd)
e.EvalRetVal(cmd)
elif command["keyup"]:
cmd = "xte keyup %s" % (command['<key>'])
e.Log(cmd)
e.EvalRetVal(cmd)
elif command["keypress"]:
cmd = 'xte "sleep 1" "key %s"' % (command['<key>'])
e.Log(cmd)
e.EvalRetVal(cmd)
elif command["type"]:
e.Log("Desktop Typing keys")
special = ["Alt", "Control", "Shift", "Super"]
press = []
release = []
l = 0
for k in command['KEYS']:
if k in special:
press.append('"keydown ' + k + '_R"')
release.append('"keyup ' + k + '_R"')
elif len(release) > 0:
press.append('"key ' + k + '"')
release.reverse()
press.extend(release)
release = []
else:
for char in k:
press.append('"usleep 30000" "str ' + char + '"')
#press.append('"str ' + k + '"')
cmd = "xte " + ' '.join(press)
e.EvalRetVal(cmd)
elif command["mouseclick"]:
e.Log("Desktop mouseclick")
cmd = "./agent/cli/bin/mouseclick.sh %s %s %s" % (
command['<windowid>'], command['<windowimg>'], ' '.join(
command['PATTERN']))
e.Log(cmd)
e.EvalRetVal(cmd)
#move the mouse out before it pollutes next screenshot
e.Execute('xte "sleep 1" "mousemove 0 0"')
def do_Selenium(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Selenium")
if command["start"]:
e.Spawn("python ./agent/cli/bin/SeleniumWorker.py %s" % (command['<which>']) )
time.sleep(5)
retval, output = self.ReadWorkerResult()
elif command["stop"]:
self.SendWorkerCommand("stop")
retval, output = self.ReadWorkerResult()
elif command["clickcss"]:
self.SendWorkerCommand("clickcss " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["clickname"]:
self.SendWorkerCommand("clickname " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["clickid"]:
self.SendWorkerCommand("clickid " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["clickxpath"]:
self.SendWorkerCommand("clickxpath " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["clickmenuitem"]:
self.SendWorkerCommand("clickmenuitem " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["clearcss"]:
self.SendWorkerCommand("clearcss " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["clearname"]:
self.SendWorkerCommand("clearname " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["clearid"]:
self.SendWorkerCommand("clearid " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["clearxpath"]:
self.SendWorkerCommand("clearxpath " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["open"]:
self.SendWorkerCommand("open " + command['<which>'] + "")
retval, output = self.ReadWorkerResult()
elif command["fillcss"]:
self.SendWorkerCommand("fillcss " + command['<which>'] + " " + ' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
elif command["fillname"]:
self.SendWorkerCommand("fillname " + command['<which>'] + " " + ' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
elif command["fillid"]:
self.SendWorkerCommand("fillid " + command['<which>'] + " " + ' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
elif command["fillxpath"]:
self.SendWorkerCommand("fillxpath " + command['<which>'] + " " + ' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
elif command["counttags"]:
self.SendWorkerCommand("counttags " + command['<which>'] + " " + ' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
elif command["namecontains"]:
self.SendWorkerCommand("namecontains " + command['<which>'] + " " + ' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
elif command["idcontains"]:
self.SendWorkerCommand("idcontains " + command['<which>'] + " " + ' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
elif command["csscontains"]:
self.SendWorkerCommand("csscontains " + command['<which>'] + " " + ' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
elif command["xpathcontains"]:
self.SendWorkerCommand("xpathcontains " + command['<which>'] + " " + ' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
elif command["pagecontains"]:
self.SendWorkerCommand("pagecontains " + "null" + " " + ' '.join(command['WHAT']) + "")
retval, output = self.ReadWorkerResult()
else:
retval, output = "Fail", "Invalid command"
e.Log(output)
e.setExecCodes(args, output, retval)
def do_Ping(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Ping")
e.EvalRetVal("ping -c 1 " + command['<host>'])
e.Log(e.r.getOutput())
def do_Process(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("Process")
if command["monitor"]:
psopts = "/bin/ps -C %s o pcpu,pmem --cumulative --no-heading" %(command["<name>"])
e.Log("Executing monitor " + psopts)
counter = 0
cpu = float(0)
mem = float(0)
while (counter < int(command["<duration>"])):
pcpu,pmem = self.parse(e.RetOutput(psopts))
cpu = cpu + pcpu
mem = mem + pmem
time.sleep(1)
counter = counter + 1
output = "%s consumed %s%% cpu and %s %% memory" % (command["<name>"],cpu/counter,mem/counter)
e.Log(output)
print output
if cpu > 0 or mem > 0:
e.setExecCodes("Process monitor",output,"Pass")
else:
e.setExecCodes("Process monitor",output,"Fail")
elif command["kill"]:
killopts = "/usr/bin/killall %s" %(command["<name>"])
e.Log(killopts)
e.EvalRetVal(killopts)
elif command["spawn"]:
spawnopts = ' '.join(command["PATH"])
e.Log(spawnopts)
e.Spawn(spawnopts)
elif command["exec"]:
execopts = ' '.join(command["PATH"])
e.Log(execopts)
execopts = "/bin/sh -c '%s' " %(execopts)
e.EvalRetVal(execopts)
elif command["forfeit"]:
execopts = ' '.join(command["PATH"])
e.Log(execopts)
execopts = "/bin/sh -c '%s' " %(execopts)
e.EvalRetVal(execopts)
e.r.setRetVal("Fail")
elif command["thrive"]:
execopts = ' '.join(command["PATH"])
e.Log(execopts)
execopts = "/bin/sh -c '%s' " %(execopts)
e.EvalRetVal(execopts)
e.r.setRetVal("Pass")
elif command["alive"]:
killopts = "/usr/bin/killall -s 0 %s" %(command["<name>"])
e.Log(killopts)
e.EvalRetVal(killopts)
elif command["dead"]:
killopts = "pidof %s" %(command["<name>"])
e.Log(killopts)
e.EvalRetVal(killopts, 1)
elif command["pid"]:
execopts = "pidof %s" %(command["<name>"])
e.Log(execopts)
e.EvalRetVal(execopts)
elif command["output"]:
# Process output contains 1360x78 xdpyinfo | grep dimensions
execopts = ' '.join(command["PATH"])
e.Log(execopts)
execopts = "/bin/sh -c '%s' " %(execopts)
output = e.RetOutput(execopts)
if command['<expectation>'] in output:
e.setExecCodes(execopts, output, "Pass")
else:
e.setExecCodes(execopts, output, "Fail")
e.Log(e.r.getOutput())
print e.r.getRetVal()
def do_RegTest(self, args):
command = docopt(str(self.doc), args)
e = ExecHelper()
e.reset("RegTest")
e.OutputEquals("/usr/bin/regget " + command['<section>'] + " " + command['<value>'],
command['<expectation>'])