SIB/SecureRest
0
1import tensorflow as tf2import gradio as gr3from joblib import load4import numpy as np 5 6 7model_multi= tf.keras.models.load_model('bestmodel_attack.h5')8model_bin = load("neural.joblib")9# protocol icmp tcp udp10 11# service IRC X11 Z3950 aol auth bgp courier csnet_ns ctf daytime discard domain domain_u echo eco_i ecr_i12# efs exec finger ftp ftp_data gopher harvest hostnames http http_2784 http_443 http_8001 imap413# iso_tsap klogin kshell ldap link login mtp name netbios_dgm netbios_ns netbios_ssn 14# netstat nnsp nntp ntp_u other pm_dump pop_2 pop_3 printer private red_i remote_job rje 15# shell smtp sql_net ssh sunrpc supdup systat telnet tftp_u tim_i time urh_i urp_i uucp16# uucp_path vmnet whois 17 18 19# flag 20# OTH RSTO RSTOS0 RSTR S0 S1 S2 S3 SF SH 21 22# duration23 24# src_bytes25 26# dst_bytes27 28# urgent29 30# root_shell 31 32# su_attempted33 34# num_file_creations35 36# num_shells37 38# num_access_files39 40# is_guest_login41 42data_mining = {43 "protocol": ["icmp", "tcp", "udp"],44 "service":['IRC', 'X11', 'Z39_50', 'aol','auth', 'bgp', 'courier', 'csnet_ns','ctf', 'daytime', 45 'discard', 'domain','domain_u', 'echo', 'eco_i', 'ecr_i','efs', 'exec', 'finger', 'ftp',46 'ftp_data', 'gopher', 'harvest','hostnames', 'http', 'http_2784','http_443', 'http_8001', 47 'imap4','iso_tsap', 'klogin', 'kshell', 'ldap','link', 'login', 'mtp', 'name','netbios_dgm', 48 'netbios_ns', 'netbios_ssn','netstat', 'nnsp', 'nntp', 'ntp_u','other', 'pm_dump', 49 'pop_2', 'pop_3','printer', 'private', 'red_i','remote_job', 'rje', 'shell', 'smtp',50 'sql_net', 'ssh', 'sunrpc', 'supdup','systat', 'telnet', 'tftp_u', 'tim_i','time', 51 'urh_i', 'urp_i', 'uucp','uucp_path', 'vmnet', 'whois'],52 'flag':['OTH', 'REJ', 'RSTO', 'RSTOS0', 'RSTR','S0', 'S1', 'S2', 'S3', 'SF', 'SH']53}54def greet(choix,protocol,services, flag,duration,src_bytes,dst_bytes,land,wrong_fragment,urgent,hot, # ok55num_failed_logins,logged_in,num_compromised,root_shell,su_attempted,num_root,num_file_creations,num_shells # ok56,num_access_files,num_outbound_cmds,is_host_login,is_guest_login,count,srv_count,serror_rate,srv_serror_rate #ok57,rerror_rate,srv_rerror_rate,same_srv_rate,diff_srv_rate,srv_diff_host_rate,dst_host_count,dst_host_srv_count #ok58,dst_host_same_srv_rate,dst_host_diff_srv_rate,dst_host_same_src_port_rate,dst_host_srv_diff_host_rate,dst_host_serror_rate59,dst_host_srv_serror_rate,dst_host_rerror_rate,dst_host_srv_rerror_rate):60 liste_head = [0]* len(data_mining["protocol"])61 liste_head[data_mining["protocol"].index(protocol)] = 162 liste_head2 = [0]* len(data_mining["service"])63 liste_head2[data_mining["service"].index(services)] = 164 liste_head3 = [0]* len(data_mining["flag"])65 liste_head3[data_mining["flag"].index(flag)] = 166 list1 = list()67 pred = None68 name = str()69 70 list1 = liste_head + liste_head2 + liste_head3 + [duration,src_bytes,dst_bytes,land,wrong_fragment,urgent,hot,71num_failed_logins,logged_in,num_compromised,root_shell,su_attempted,num_root,num_file_creations,num_shells72,num_access_files,num_outbound_cmds,is_host_login,is_guest_login,count,srv_count,serror_rate,srv_serror_rate73,rerror_rate,srv_rerror_rate,same_srv_rate,diff_srv_rate,srv_diff_host_rate,dst_host_count,dst_host_srv_count74,dst_host_same_srv_rate,dst_host_diff_srv_rate,dst_host_same_src_port_rate,dst_host_srv_diff_host_rate,dst_host_serror_rate75,dst_host_srv_serror_rate,dst_host_rerror_rate,dst_host_srv_rerror_rate]76 arr = np.array(list1)77 if choix == "Binaire":78 pred = int(model_bin.predict(arr.reshape(1,arr.shape[0])).argmax())79 if pred == 0:80 name = "Aucune Attaque détectée"81 else:82 name= "Attaque détectée"83 else:84 pred = int(model_multi.predict(arr.reshape(1,arr.shape[0])).argmax())85 if pred == 0:86 name = "Aucune Attaque détectée"87 elif pred == 1:88 name = "Attaque DOS détecté"89 elif pred == 2:90 name = "Attaque Probe détecté"91 elif pred == 3:92 name = "Attaque de Privilège détecté"93 else:94 name = "Attaque d'accès détecter"95 return f"La prédiction est : {name} "96 97 98 99iface = gr.Interface(100 fn=greet, 101 title="API de Test de pénétration",102 description="Cette API est utilisé pour deux actions binaires et multidetection assisté d'un environnement test continuel",103 inputs=[104 gr.inputs.Radio(["Binaire", "Catégorisation"],label="Choix de Prediction"), # choix105 gr.inputs.Radio(["icmp", "tcp", "udp"],label="protocol de communication"), # protocol106 gr.inputs.Radio(['IRC', 'X11', 'Z39_50', 'aol','auth', 'bgp', 'courier', 'csnet_ns','ctf', 'daytime', 107 'discard', 'domain','domain_u', 'echo', 'eco_i', 'ecr_i','efs', 'exec', 'finger', 'ftp',108 'ftp_data', 'gopher', 'harvest','hostnames', 'http', 'http_2784','http_443', 'http_8001', 109 'imap4','iso_tsap', 'klogin', 'kshell', 'ldap','link', 'login', 'mtp', 'name','netbios_dgm', 110 'netbios_ns', 'netbios_ssn','netstat', 'nnsp', 'nntp', 'ntp_u','other', 'pm_dump', 111 'pop_2', 'pop_3','printer', 'private', 'red_i','remote_job', 'rje', 'shell', 'smtp',112 'sql_net', 'ssh', 'sunrpc', 'supdup','systat', 'telnet', 'tftp_u', 'tim_i','time', 113 'urh_i', 'urp_i', 'uucp','uucp_path', 'vmnet', 'whois']114 ,label="services de communication"), # service115 gr.inputs.Radio(['OTH', 'REJ', 'RSTO', 'RSTOS0', 'RSTR','S0', 'S1', 'S2', 'S3', 'SF', 'SH'], # 10 116 label="drapeau de communication"), # flag117 gr.inputs.Slider(minimum=0, maximum=360,label="durée de l'execution"), # duration118 gr.inputs.Slider(minimum=0, maximum=360,label="nombre de bits sources"), # src_bytes119 gr.inputs.Slider(minimum=0, maximum=360,label="nombre de bits destination"), # dst_bytes120 gr.inputs.Slider(minimum=0, maximum=360,label="land"), # land 121 gr.inputs.Slider(minimum=0, maximum=360,label="wrong_fragment"), # wrong_fragment122 gr.inputs.Slider(minimum=0, maximum=150,step=1,label="l'urgence de l'execution"), # urgent123 gr.inputs.Slider(minimum=0, maximum=45,step=1,label="hot"), # hot124 gr.inputs.Slider(minimum=0, maximum=100,step=1,label="num_failed_logins"), # num_failed_logins125 gr.inputs.Slider(minimum=0, maximum=3,step=1,label="logged_in"), # logged_in126 gr.inputs.Slider(minimum=0, maximum=350,step=1,label="num_compromised"), # num_compromised127 gr.inputs.Slider(minimum=0, maximum=35,step=1,label="ouverture du shell en super utilisateur"), # root_shell128 gr.inputs.Slider(minimum=0, maximum=50,step=1,label="utilisation du mode super utilisateur"), # su_attempted129 gr.inputs.Slider(minimum=0, maximum=315,step=1,label="num_root"), # num_root130 gr.inputs.Slider(minimum=0, maximum=200,step=1,label="nombre de fichier en création"), # num_file_creations131 gr.inputs.Slider(minimum=0, maximum=200,step=1,label="nombre de shell"), # num_shells132 gr.inputs.Slider(minimum=0, maximum=200,step=1,label="nombre d'accès de fichier"), # num_access_files133 gr.inputs.Slider(minimum=0, maximum=1,step=1,label="num_outbound_cmds"), # num_outbound_cmds134 gr.inputs.Slider(minimum=0, maximum=400,step=1,label="is_host_login"), # is_host_login135 gr.inputs.Slider(minimum=0, maximum=15,step=1,label="is_guest_login"), # is_guest_login136 gr.inputs.Slider(minimum=0, maximum=5,step=1,label="count"), # count137 gr.inputs.Slider(minimum=0, maximum=7,step=1,label="srv_count"), # srv_count138 gr.inputs.Slider(minimum=0, maximum=1.6,step=.1,label="serror_rate"), # serror_rate139 gr.inputs.Slider(minimum=0, maximum=2,step=.1,label="srv_serror_rate"), # srv_rerror_rate140 gr.inputs.Slider(minimum=0, maximum=3,step=.1,label="rerror_rate"), # rerror_rate141 gr.inputs.Slider(minimum=0, maximum=4,step=.1,label="srv_rerror_rate"), # srv_rerror_rate142 gr.inputs.Slider(minimum=0, maximum=1,step=.1,label="same_srv_rate"), # same_srv_rate143 gr.inputs.Slider(minimum=0, maximum=6,step=1,label="diff_srv_rate"), # diff_srv_rate144 gr.inputs.Slider(minimum=0, maximum=6,step=1,label="srv_diff_host_rate"), # srv_diff_host_rate145 gr.inputs.Slider(minimum=0, maximum=10,step=1,label="dst_host_count"), # dst_host_count146 gr.inputs.Slider(minimum=0, maximum=10,step=1,label="dst_host_srv_count"), # dst_host_srv_count147 gr.inputs.Slider(minimum=0, maximum=10,step=1,label="d,st_host_same_srv_rate"), # dst_host_same_srv_rate148 gr.inputs.Slider(minimum=0, maximum=10,step=1,label="dst_host_same_srv_rate"), # dst_host_diff_srv_rate149 gr.inputs.Slider(minimum=0, maximum=10,step=1,label="dst_host_same_src_port_rate"), # dst_host_same_src_port_rate150 gr.inputs.Slider(minimum=0, maximum=10,step=1,label="dst_host_srv_diff_host_rate"), # dst_host_srv_diff_host_rate151 gr.inputs.Slider(minimum=0, maximum=10,step=1,label="dst_host_serror_rate"), # dst_host_serror_rate152 gr.inputs.Slider(minimum=0, maximum=10,step=1,label="dst_host_srv_serror_rate"), # dst_host_srv_serror_rate153 gr.inputs.Slider(minimum=0, maximum=10,step=1,label="dst_host_rerror_rate"), # dst_host_rerror_rate154 gr.inputs.Slider(minimum=0, maximum=10,step=1,label="dst_host_srv_rerror_rate") #, # dst_host_srv_rerror_rate155 # gr.inputs.Number(label="sources bytes de la chaine d'execution"),156 # gr.inputs.Number(label="destination bytes de la chaine d'execution"),157 # gr.inputs.Slider(minimum=0, maximum=80,step=1,label="land"), # land158 # gr.inputs.Number(label="wrong fragment"),159 # gr.inputs.Slider(minimum=0, maximum=150,step=1,label="l'urgence de l'execution"),160 # gr.inputs.Slider(minimum=0, maximum=45,step=1,label="hot"),161 # gr.inputs.Slider(minimum=0, maximum=100,step=1,label="num_failed_logins"),162 # gr.inputs.Slider(minimum=0, maximum=3,step=1,label="logged_in"),163 # gr.inputs.Slider(minimum=0, maximum=350,step=1,label="num_compromised"),164 # gr.inputs.Slider(minimum=0, maximum=35,step=1,label="ouverture du shell en super utilisateur"),165 # gr.inputs.Slider(minimum=0, maximum=50,step=1,label="utilisation du mode super utilisateur"),166 # gr.inputs.Slider(minimum=0, maximum=315,step=1,label=",num_root"),167 # gr.inputs.Slider(minimum=0, maximum=200,step=1,label=",nombre de fichier en création"),168 # gr.inputs.Slider(minimum=0, maximum=200,step=1,label=",nombre de shell"),169 # gr.inputs.Slider(minimum=0, maximum=200,step=1,label=",nombre d'accès de fichier"),170 # gr.inputs.Slider(minimum=0, maximum=1,step=1,label="num_outbound_cmds"),171 # gr.inputs.Slider(minimum=0, maximum=400,step=1,label=",is_host_login"),172 # gr.inputs.Slider(minimum=0, maximum=15,step=1,label="i,s_guest_login"),173 # gr.inputs.Slider(minimum=0, maximum=5,step=1,label="co,unt"),174 # gr.inputs.Slider(minimum=0, maximum=7,step=1,label="sr,v_count"),175 # gr.inputs.Slider(minimum=0, maximum=1.6,step=.1,label=,"serror_rate"),176 # gr.inputs.Slider(minimum=0, maximum=2,step=.1,label="s,rv_serror_rate"),177 # gr.inputs.Slider(minimum=0, maximum=3,step=.1,label="r,error_rate"), 178 # gr.inputs.Slider(minimum=0, maximum=4,step=.1,label="s,rv_rerror_rate"), 179 # gr.inputs.Slider(minimum=0, maximum=1,step=.1,label="s,ame_srv_rate"), 180 # gr.inputs.Slider(minimum=0, maximum=6,step=1,label="di,ff_srv_rate"),181 # gr.inputs.Slider(minimum=0, maximum=6,step=1,label="sr,v_diff_host_rate"),182 # gr.inputs.Slider(minimum=0, maximum=10,step=1,label="d,st_host_count"),183 # gr.inputs.Slider(minimum=0, maximum=10,step=1,label="d,st_host_srv_count"),184 # gr.inputs.Slider(minimum=0, maximum=10,step=1,label="d,st_host_same_srv_rate"),185 # gr.inputs.Slider(minimum=0, maximum=10,step=1,label="d,st_host_same_src_port_rate"),186 # gr.inputs.Slider(minimum=0, maximum=10,step=1,label="d,st_host_srv_diff_host_rate"),187 # gr.inputs.Slider(minimum=0, maximum=10,step=1,label="d,st_host_serror_rate"),188 # gr.inputs.Slider(minimum=0, maximum=10,step=1,label="d,st_host_srv_serror_rate"),189 # gr.inputs.Slider(minimum=0, maximum=10,step=1,label="d,st_host_srv_rerror_rate")#,190 # gr.inputs.Slider(minimum=0, maximum=15,step=1,label=,"nombre de shell crée"),191 # gr.inputs.Slider(minimum=0, maximum=15,step=1,label=,"accès aux fichiers systèmes"),192 # gr.inputs.Slider(minimum=0, maximum=1,step=1,label="est en login")193 ], # ok 194 195 outputs="text",196 examples=[197 ["Binaire", "icmp", 'IRC','OTH',4000,45,35,2,0,1,20,5,7,0],198 ["Catégorisation", "tcp", 'netstat','REJ',3500,25,15,0,1,0,17,10,2,0],199 ["Catégorisation", "udp", 'login','S0',150,30,5,0,1,20,2,15,1,1]200 ],201 flagging_options=["this", "or", "that"],202 theme="dark-seafoam",203 thumbnail='images.jpg'204 )205iface.launch()# share= True,auth=("admin", "pass1234"))206