class Ctrl():
def __init__(self):
self.conf = Conf()
self.listOfComp = self.conf.getComp()
self.listOfSub = self.conf.getSub()
self.tmpOpt1 = []
self.tmpOpt2 = []
@staticmethod
def openfile(e=None):
from gui import Gui
filename = Gui.exploreFile()
if e!=None:
Ctrl.setText(e, filename)
@staticmethod
def emptyText(e):
e.delete(0,tk.END)
@staticmethod
def setText(e, text):
Ctrl.emptyText(e)
e.insert(0, text)
@staticmethod
def checkRun(e, p, w):
from gui import Gui
if not os.path.exists(e.get()):
Gui.alertErr("Error", "Configuration file not found", w)
return
try:
conf = e.get()
ce = CompilationEngine(conf)
m = Manipulator(conf)
p['value'] = 20
w.update_idletasks()
m.generateSource()
p['value'] = 40
w.update_idletasks()
ce.createExes()
p['value'] = 60
w.update_idletasks()
complist = ce.getReport()
vt = VirusTotal()
p['value'] = 80
w.update_idletasks()
report = vt.getScore(complist)
p['value'] = 100
w.update_idletasks()
str = ""
i = 0
for x,y in complist.items():
str += "<{} : {}>\n".format(x,report[i])
i+=1
Gui.alertInfo("Result", str, w)
except Exception as err:
p['value'] = 0
w.update_idletasks()
Gui.alertErr("Error", err, w)
@staticmethod
def checkClear(e, p, w):
from gui import Gui
if not os.path.exists(e.get()):
Gui.alertErr("Error", "Configuration file not found", w)
return
try:
conf = e.get()
ce = CompilationEngine(conf)
ce.clear()
p['value'] = 100
w.update_idletasks()
Gui.alertInfo("Info", "Cleaning completed", w)
except Exception as err:
p['value'] = 0
w.update_idletasks()
Gui.alertErr("Error", err, w)
def remFromListComp(self,l):
from gui import Gui
if not l.curselection():
Gui.alertInfo("Info", "Select an element to delete it")
return
del(self.listOfComp[l.curselection()[0]])
Gui.remFromListSelected(l)
def remFromListSub(self,l):
from gui import Gui
if not l.curselection():
Gui.alertInfo("Info", "Select an element to delete it")
return
del(self.listOfSub[l.curselection()[0]])
Gui.remFromListSelected(l)
def remFromListOpt(self,l, index, root):
from gui import Gui
if not l.curselection():
Gui.alertInfo("Info", "Select an element to delete it", root)
return
if index == 0 :
del(self.tmpOpt1[l.curselection()[0]])
else:
del(self.tmpOpt2[l.curselection()[0]])
Gui.remFromListSelected(l)
def fillCompField(self, e, l, list):
from gui import Gui
inex = self.listOfComp[list.curselection()[0]]
Ctrl.setText(e[0], inex.getName())
Ctrl.setText(e[1], inex.getPath())
if len(inex.getOpt1()) > 0:
for el in inex.getOpt1():
Gui.addElToList(l[0], el)
self.tmpOpt1.append(el)
if len(inex.getOpt2()) > 0:
for el in inex.getOpt2():
Gui.addElToList(l[1], el)
self.tmpOpt2.append(el)
def checkSub(self, top, e, list):
from gui import Gui
if len(e[0].get()) != 0 and len(e[1].get()) != 0:
self.conf.addToSub((e[0].get(),e[1].get()))
Gui.addElToList(list, (e[0].get(),e[1].get()))
Gui.destroyTop(top)
else:
Gui.alertErr("Error", "Both fields are obligatory", top)
def checkOption(self, e, l, index, root):
from gui import Gui
if len(e[0].get()) > 0:
t = None
if len(e[1].get()) > 0:
if len(e[2].get()) > 0:
t = (e[0].get(), e[1].get().split(e[2].get()))
Gui.addElToList(l, t)
else:
t = (e[0].get(), e[1].get().split(None))
Gui.addElToList(l, t)
else:
t = (e[0].get(),[])
Gui.addElToList(l, t)
if index == 0:
self.tmpOpt1.append(t)
else:
self.tmpOpt2.append(t)
else:
Gui.alertErr("Error", "Option name is obligatory", root)
Ctrl.emptyText(e[0])
Ctrl.emptyText(e[1])
Ctrl.emptyText(e[2])
def checkComp(self, top, e, list):
from gui import Gui
comp = Comp()
if len(e[0].get()) != 0 and len(e[1].get()) != 0:
comp.setName(e[0].get())
comp.setPath(e[1].get())
for t in self.tmpOpt1:
comp.addOpt1(t)
for t in self.tmpOpt2:
comp.addOpt2(t)
self.tmpOpt1 = []
self.tmpOpt2 = []
self.listOfComp.append(comp)
Gui.addElToList(list, str(comp))
Gui.destroyTop(top)
else:
Gui.alertErr("Error", "Fill obligatory fields", top)
def checkConf(self, e, l):
from gui import Gui
if len(e[0].get()) == 0 or len(e[1].get()) == 0 or len(e[2].get()) == 0 or len(e[3].get()) == 0:
Gui.alertErr("Error", "Fill obligatory fields")
return
if len(self.listOfComp) == 0:
Gui.alertErr("Error", "Insert at least a compilation test")
return
if len(e[6].get()) == 0:
Gui.alertErr("Error", "Insert configuration file name")
return
self.conf.setTemplatePath(e[0].get())
self.conf.setPayloadPath(e[1].get())
self.conf.setSpecialChar(e[2].get())
self.conf.setPlaceholderPayload(e[3].get())
if len(e[4].get()) != 0:
try:
rate = float(e[4].get())
if not(rate >= 0 and rate < 1):
Gui.alertErr("Error", "The rate must be a decimal between (0,1]")
return
self.conf.setFreq(rate)
except ValueError:
Gui.alertErr("Error", "The rate must be a decimal between (0,1]")
return
if len(e[5].get()) != 0:
self.conf.setOut(e[5].get())
#vettore sub gestito a runtime
print(str(self.conf))
with open(e[6].get(), "w") as f:
tmp = str(self.conf)
tmp = tmp.replace("\'", "\"")
f.write(tmp)
Gui.alertErr("Info", "File saved : {}".format(e[6].get()))
def updateComp(self, top, e, list, index):
comp = Comp()
if len(e[0].get()) != 0 and len(e[1].get()) != 0:
from gui import Gui
comp.setName(e[0].get())
comp.setPath(e[1].get())
for t in self.tmpOpt1:
comp.addOpt1(t)
for t in self.tmpOpt2:
comp.addOpt2(t)
self.tmpOpt1 = []
self.tmpOpt2 = []
self.listOfComp[index] = comp
Gui.remFromList(list, index)
Gui.addElToList(list, str(comp), index)
Gui.destroyTop(top)
class ConfigCreator():
def __init__(self):
self.conf = Conf()
self.listOfComp = self.conf.getComp()
self.listOfSub = self.conf.getSub()
self.tmpOpt1 = []
self.tmpOpt2 = []
def create(self, payload_file, template_file, special_char,
payload_placeholder, rate, outputname, subs1, string1,
compilers, option1_gcc, option2_gcc, option1_gplusplus,
option2_gplusplus, value1_gcc, value2_gcc, value1_gplusplus,
value2_gplusplus, hexed_filename):
ops1_gcc = []
ops2_gcc = []
ops1_gplusplus = []
ops2_gplusplus = []
ops1_gcc.extend([option1_gcc, value1_gcc])
ops1_gplusplus.extend([option1_gplusplus, value1_gplusplus])
ops2_gcc.extend([option2_gcc, value2_gcc])
ops2_gplusplus.extend([option2_gplusplus, value2_gplusplus])
config_filename = hexed_filename + ".conf.json"
path = os.path.join(app.config['UPLOAD_FOLDER'], hexed_filename,
"config", config_filename)
self.conf.setTemplatePath(template_file)
self.conf.setPayloadPath(payload_file)
self.conf.setSpecialChar(special_char)
self.conf.setPlaceholderPayload(payload_placeholder)
rate = float(rate)
self.conf.setFreq(rate)
self.conf.setOut(outputname)
for subs, string in zip(subs1, string1):
self.conf.addToSub((subs, string))
comps = []
for compiler in compilers:
comp = Comp()
comp.setName(compiler)
comp.setPath("edit")
if compiler == "gcc":
self.tmpOpt1.append(ops1_gcc)
self.tmpOpt2.append(ops2_gcc)
elif compiler == "g++":
self.tmpOpt1 = []
self.tmpOpt2 = []
self.tmpOpt1.append(ops1_gplusplus)
self.tmpOpt2.append(ops2_gplusplus)
for t in self.tmpOpt1:
comp.addOpt1(t)
for t in self.tmpOpt2:
comp.addOpt2(t)
comps.append(comp)
self.tmpOpt1 = []
self.tmpOpt2 = []
for comp in comps:
self.listOfComp.append(comp)
with open(path, "w") as f:
tmp = str(self.conf)
tmp = tmp.replace("\'", "\"")
f.write(tmp)
return config_filename
