Spaces:
Running
Running
| # -*- coding: utf-8 -*- | |
| """ | |
| Created on Sun Dec 18 21:43:54 2022 | |
| @author: amaer | |
| """ | |
| import numpy as np | |
| from ModelTEC import * | |
| from ipywidgets import HTML | |
| class Cheakscenario: | |
| rateerror=0 | |
| PrintFuri=True | |
| def __init__(self,ob=None,sequensed=None): | |
| self.obTECSoft=ob | |
| self.lensoft=len(ob.obMP.Classes) | |
| self.inintlist() | |
| self.onsequens=sequensed | |
| def inintlist(self): | |
| self.listsoftware=np.zeros((self.lensoft,5)) | |
| self.Counter=0 | |
| def createtable(self,row,stat,index=0): | |
| st='<table border="1" style="" ><tr style="background-color:'+stat[0]+'; color:#fff"><th style="text-align:center" >Tactic Name</th><th style="text-align:center" >Technique Name</th></tr>' | |
| for ob in row: | |
| st+='<tr><td style="text-align:center" >'+ob[0]+'</td><td style="text-align:center" >'+ob[1]+'</td></tr>' | |
| st+='<tr style="background-color:'+stat[0]+'; color:#fff"><td colspan = "2" style="text-align:center" >attack scenario status (' | |
| st+=''+ stat[1] +' ) </td></tr>' | |
| return st+'</table>' | |
| def getScenario(self,itmes,k=0,index=0): | |
| stat=self.getAStatescenario(k,index) | |
| if Cheakscenario.PrintFuri==False and stat[1]=='Failure':return None | |
| rowd=[] | |
| for ob in itmes: | |
| if ob.strip() !='': | |
| itms=ob.split("*->") | |
| rowd.append([itms[0],itms[1]]) | |
| return rowd,stat,k | |
| def process(self,path,cinput,isclean=True): | |
| itmes=path.split("=> ") | |
| pathT=self.getpathTech(itmes) | |
| if isclean: | |
| da=self.obTECSoft.clean_dataT([pathT+' ' +cinput+" use "]) | |
| else: | |
| da=[cinput] | |
| # print(pathT,da) | |
| if(len(da)>0): | |
| try: | |
| feat=self.obTECSoft.obMP.obVec.transform(da) | |
| pyd=self.obTECSoft.Model.predict_proba(feat) | |
| self.listsoftware[:,4]+=np.array(pyd[0]) | |
| #for i in range(self.lensoft): | |
| # self.listsoftware[i,4]=pyd[0][i] | |
| index=np.argmax(pyd) | |
| k=pyd[0][index]*100+Cheakscenario.rateerror | |
| row=self.getScenario(itmes,k,index) | |
| if row is not None and len(row[0])>0: | |
| self.Counter+=1 | |
| if self.onsequens is not None: | |
| self.onsequens((row,pyd[0])) | |
| return row,pyd[0] | |
| except: pass | |
| return None | |
| def print_table(self,path,cinput): | |
| # self.obTECSoft.ChangeModel(ob='svmL') | |
| rows=self.process(path, cinput) | |
| if rows is not None : | |
| row,pa=rows | |
| td=self.createtable(row[0],row[1],self.Counter) | |
| print('---------------------------scenario (',self.Counter,')------------------------------') | |
| display(HTML(td)) | |
| return rows | |
| def getAStatescenario(self,k,index): | |
| if k>=70: | |
| self.listsoftware[index][0]+=1 | |
| return ('#04AA6D','Success') | |
| if k>=60: | |
| self.listsoftware[index][1]+=1 | |
| return ('#1ec7c3','close to success') | |
| if k>=30: | |
| self.listsoftware[index][2]+=1 | |
| return ('#fb4bd1','Initi') | |
| self.listsoftware[index][3]+=1 | |
| return ('#e91c07','Failure') | |
| def getpathTech(self,items): | |
| path=' use ' | |
| for ob in items: | |
| if ob.strip() !='': | |
| itms=ob.split("*->") | |
| path+=' '+itms[1] | |
| return path | |