def getPlotData(self, plottype, plotfilename, doubleaxis, converty, converty2, normalized, desvest): exdata = ExtractData(plottype, plotfilename, doubleaxis, desvest) #xdata, ydata, xdata2, ydata2 plotdata = exdata.get_data() #print("convert; ", converty, " ", converty2) #TODO: smoothing data if converty: for i in range(len(plotdata[1])): plotdata[1][i] = np.array( plotdata[1][i]).astype(float) * float(converty) if converty2: for i in range(len(plotdata[3])): plotdata[3][i] = np.array( plotdata[3][i]).astype(float) * float(converty2) if normalized == "Normalized by max": for i in range(len(plotdata[1])): maxvalue = max(plotdata[1][i]) for j in range(len(plotdata[1][i])): plotdata[1][i][j] = plotdata[1][i][j] / maxvalue elif normalized == "Normalized by first": for i in range(len(plotdata[1])): firstvalue = plotdata[1][i][0] for j in range(len(plotdata[1][i])): plotdata[1][i][j] = plotdata[1][i][j] / firstvalue return plotdata
def main(): print("Please enter Company Number") companyNumberInput = input() searchCompanyByNumber = SearchCompanyByNumber() companyNumberSearchResult = searchCompanyByNumber.searchByCompanyNumber(companyNumberInput) dataExtractedFromCompanyNumberSearchResult = ExtractData() extractedData = dataExtractedFromCompanyNumberSearchResult.extractInformationFromCompanyNumberSearch(companyNumberSearchResult) print(extractedData)
def callback(header, data): src_ip = None dst_ip = None src_port = None dst_port = None layer4Type = "" output = "" payloadSize = 0 extract = ExtractData() # Parse packet decoder = ImpactDecoder.EthDecoder() packet = decoder.decode(data) # Parse IP packet inside ethernet one iphdr = packet.child() if isinstance(iphdr, IP): # Parse TCP packet inside IP one hdr = iphdr.child() if isinstance(hdr, TCP) or isinstance(hdr, UDP): if isinstance(hdr, TCP): layer4Type = "TCP" # Only look at SYN packets, not ACK ones if hdr.get_SYN() and not hdr.get_ACK(): # Get src and dest IPs src_ip = iphdr.get_ip_src() dst_ip = iphdr.get_ip_dst() src_port = hdr.get_th_dport() dst_port = hdr.get_th_sport() payloadSize = hdr.get_size() - hdr.get_header_size() elif isinstance(hdr, UDP): layer4Type = "UDP" # Get src and dest IPs src_ip = iphdr.get_ip_src() dst_ip = iphdr.get_ip_dst() src_port = hdr.get_th_dport() dst_port = hdr.get_th_sport() payloadSize = hdr.get_size() - hdr.get_header_size() # Results are printed output = "(%s) Connection attempted from: %s:%s to: %s:%s\n" % ( layer4Type, src_ip, src_port, dst_ip, dst_port, ) if payloadSize != 0: output += "\nPayload size: %d\n----%s----\n----\n" % (payloadSize, hdr.get_data_as_string()) print output if src_ip and dst_ip: extract.writeToFile("packetOutput.txt", output, "a") else: print "\nIP header doesn't exist.\n" iphdr = None
def callback(header, data): src_ip = None dst_ip = None src_port = None dst_port = None layer4Type = "" output = "" payloadSize = 0 extract = ExtractData() #Parse packet decoder = ImpactDecoder.EthDecoder() packet = decoder.decode(data) #Parse IP packet inside ethernet one iphdr = packet.child() if isinstance(iphdr, IP): #Parse TCP packet inside IP one hdr = iphdr.child() if isinstance(hdr, TCP) or isinstance(hdr, UDP): if isinstance(hdr, TCP): layer4Type = "TCP" #Only look at SYN packets, not ACK ones if hdr.get_SYN() and not hdr.get_ACK(): #Get src and dest IPs src_ip = iphdr.get_ip_src() dst_ip = iphdr.get_ip_dst() src_port = hdr.get_th_dport() dst_port = hdr.get_th_sport() payloadSize = hdr.get_size() - hdr.get_header_size() elif isinstance(hdr, UDP): layer4Type = "UDP" #Get src and dest IPs src_ip = iphdr.get_ip_src() dst_ip = iphdr.get_ip_dst() src_port = hdr.get_th_dport() dst_port = hdr.get_th_sport() payloadSize = hdr.get_size() - hdr.get_header_size() #Results are printed output = "(%s) Connection attempted from: %s:%s to: %s:%s\n" % ( layer4Type, src_ip, src_port, dst_ip, dst_port) if (payloadSize != 0): output += "\nPayload size: %d\n----%s----\n----\n" % ( payloadSize, hdr.get_data_as_string()) print output if (src_ip and dst_ip): extract.writeToFile("packetOutput.txt", output, "a") else: print "\nIP header doesn't exist.\n" iphdr = None
from extractdata import ExtractData extract = ExtractData() extract.runAttack()