Source Code for network and port scanner, TRW algorithm, and reduction method implementations.

trw_real.py 3.0KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. import ipaddress
  2. import json
  3. #Para usar Silk
  4. #SilkFile object (represents a channel for writing to or reading from Silk flow file)
  5. #FGlob object (allows retireval of filenames in a silk data store.)
  6. #threshold of the ratio
  7. p = 5
  8. flowHash={}
  9. otherHash= {}
  10. myNum = 0
  11. counter = 0
  12. PATH = '/Users/Sara/Documents/Univ Classes/Investigacion/Programas/netflows.txt'
  13. myFile = open(PATH, 'r')
  14. ip = myFile.read()
  15. flow = json.loads(ip)
  16. #Funcion que convierte y devuelve el int ipaddress en notacion punto-decimal
  17. def ipConversion(number, position):
  18. mystr = ''
  19. ipadd = (str(ipaddress.IPv4Address(number))).split(".") #Devuelve un arreglo
  20. for i in range(position+1):
  21. if i ==position:
  22. mystr = mystr + ipadd[i]
  23. else:
  24. mystr = mystr + ipadd[i] + '.'
  25. return mystr #devuelve los numeros en notacion string
  26. def AnalisisReduciendo(sipList, num):
  27. for i in flow["flows"]: #itera por cada elemento del diccionario de flows
  28. sip = ipConversion(i["sip"], num) #Devuelve el ip en notacion punto-decimal
  29. flags = i["tcpflags"].split(",") #array of all tcp flags that are set
  30. if num != 0 and ipConversion(i["sip"], num-1) not in sipList: #Si en el paso anterior se vio que no
  31. #tiene el length de puertos requerido, se ignora
  32. continue
  33. else:
  34. connection = [0] * 2 #Lista para contener los valores de conecciones fallidas y conecciones buenas
  35. if 'A' in flags: #if the acknowledge flag is set
  36. connection[1]=1 #good conections
  37. else:
  38. connection [0] =1 #failed conections
  39. if sip in ratioHash: #si sip esta en ratioHash => que posA esta en sampleHash
  40. #por lo tanto ya se puede sumar las conecciones al ratio del dip
  41. ratioHash[sip][0]+=connection[0]
  42. ratioHash[sip][1] += connection[1]]
  43. else: #si sip no esta en ratioHash tampoco
  44. ratioHash= {sip: [connection[0], connection[1]]}
  45. print (ratioHash)
  46. return ratioHash
  47. while myNum <4: #Se itera las cuatro veces de acuerdo con la notacion de ipv4
  48. sip_hash= AnalisisReduciendo(sipList, myNum)
  49. sipList = [] #Se borra el hash para agregar elementos nuevos con la nueva etapa de la busqueda
  50. for sip in reduced_sip: #se itera por todos los sip
  51. if (ratioHash[sip][0] / ratioHash[sip][1]) < 1: #si la cantidad de succesful
  52. #b #g #connections es mas que failed connections
  53. #not scanner, ignore
  54. elif (ratioHash[sip][0] / ratioHash[sip][1]) < p: #mas failed que succesful, pero no llega al threshold
  55. #not scanner, ignore #se debe tener en cuenta que es suspicious pero no taaanto
  56. else: #el ratio de failed a succesful llega al threshold pautado
  57. #scanner, oh oh
  58. sipList.append(sip)
  59. myNum += 1
  60. #print (flowHash)