#coding:utf-8
import csv
import networkx as nx
import matplotlib.pyplot as plt
import numpy as np
def csv_dataset(addr,ip):
    slow_body1=csv.reader(open('D:\\学习\\mesa\\DDoS图\\数据集1\\%s.csv' %addr,'r'))
    slow_body1_DoS={}
    slow_body1_normal={}
    #norm=0
    #DDoS=0
    m=0
    counter=0
    counter1=0
    for row in slow_body1:
        if(row[5]=='Length'):
            continue
        tmp_one_packet=[]
        if(row[1]==ip):
            if((row[3]) not in slow_body1_DoS):
                #slow_body1_DoS[(row[3]+":"+row[7])]=[]
                counter=0
                slow_body1_DoS[(row[3])]=[]
            if((counter==0) and (row[4]!='TCP')):
                tmp_one_packet.append(120)#包大小
                #int(row[5])
                tmp_one_packet.append(float(row[2]))#时间
                tmp_one_packet.append(row[4])#协议
                #slow_body1_DoS[row[3]+':'+row[7]].append(tmp_one_packet)
                slow_body1_DoS[row[3]].append(tmp_one_packet)
                counter=counter+1
            elif(row[4]=='TCP'):
                tmp_one_packet.append(int(row[5]))#包大小
                tmp_one_packet.append(float(row[2]))#时间
                tmp_one_packet.append(row[4])#协议
                #slow_body1_DoS[row[3]+':'+row[7]].append(tmp_one_packet)
                slow_body1_DoS[row[3]].append(tmp_one_packet)
                counter=0
            #DDoS=DDoS+1
            #if((row[3]+":"+row[7]) not in slow_body1_DoS):

        elif(row[3]==ip):
            #DDoS=DDoS+1
            #if((row[1]+":"+row[6]) not in slow_body1_DoS):
            counter=0
            if((row[1]) not in slow_body1_DoS):
                slow_body1_DoS[(row[1])]=[]
                #low_body1_DoS[(row[1]+":"+row[6])]=[]
            tmp_one_packet.append(int(row[5])*(-1))#包大小
            tmp_one_packet.append(float(row[2]))#时间
            tmp_one_packet.append(row[4])#协议
            #slow_body1_DoS[row[1]+':'+row[6]].append(tmp_one_packet)
            slow_body1_DoS[row[1]].append(tmp_one_packet)
        elif(m<50000):
            m+=1
            #norm=norm+1
            '''if(row[1]<row[3]):
                #tmp=row[1]+':'+row[6]+' '+row[3]+':'+row[7]#小的定义为client端，大的定义为server端
                tmp=row[1]+' '+row[3]'''

            if(((row[1]+' '+row[3]) not in slow_body1_normal) and ((row[3]+' '+row[1]) not in slow_body1_normal)):
                counter=0
                slow_body1_normal[row[1]+' '+row[3]]=[]
            a=row[1]+' '+row[3]
            b=row[3]+' '+row[1]
            if(a in slow_body1_normal.keys()):
                tmp_one_packet.append(int(row[5])*(-1))#包大小
                tmp_one_packet.append(float(row[2]))#时间
                tmp_one_packet.append(row[4])#协议
                slow_body1_normal[a].append(tmp_one_packet)
                counter1=0
            elif(b in slow_body1_normal.keys()):
                if((counter1==0) and ('TCP'!=row[4])):
                    tmp_one_packet.append(120)#包大小
                    #int(row[5])
                    tmp_one_packet.append(float(row[2]))#时间
                    tmp_one_packet.append(row[4])#协议
                    #slow_body1_DoS[row[3]+':'+row[7]].append(tmp_one_packet)
                    slow_body1_normal[b].append(tmp_one_packet)
                    counter1+=1
                    #tmp=row[3]+':'+row[7]+' '+row[1]+':'+row[6]
                elif(row[4]=='TCP'):
                    tmp_one_packet.append(int(row[5]))#包大小
                    tmp_one_packet.append(float(row[2]))#时间
                    tmp_one_packet.append(row[4])#协议
                    slow_body1_normal[b].append(tmp_one_packet)
                    counter1=0
            else:
                print('error %s' %row)
    #print('%s——normal:%d' %(addr,norm))
    #print('%s——DDoS:%d' %(addr,DDoS))

    return slow_body1_DoS,slow_body1_normal#{ip:[size,time,pro],[],ip:[[],[],[]}

#print(slow_body1_normal)
#画图
def main():
    lable_dict={}
    all_csv_dict={}
    addr=['1','2','3','4','5(1)','5(2)','6','7','8','9','10','11','12','13','14',
              '15','16','17','18','19','20','21','22','23','24','25','26']
    tag=[0,2,4,1,1,3,2,1,2,1,3,1,2,2,2,2,4,3,1,1,3,3,3,3,1,1,2]
    #tag=[0,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1]
    t=['slowbody','slowread','ddossim','goldeneye','slowheaders','rudy','hulk','slowloris']
    tag=['normal',t[0],t[1],t[2],t[3],t[4],t[5],t[2],t[5],t[6],t[4],t[3],t[0],t[0],t[0],t[5],t[1],t[4],
    t[6],t[6],t[7],t[4],t[7],t[4],t[3],t[3],t[6],t[5]]


    ip=['75.127.97.72','75.127.97.72','75.127.97.72','75.127.97.72','75.127.97.72',
            '74.63.40.21','75.127.97.72','97.74.144.108','208.113.162.153','69.84.133.138',
            '67.220.214.50','97.74.144.108','69.192.24.88','97.74.144.108','203.73.24.75',
            '97.74.144.108','74.55.1.4','97.74.104.201','74.55.1.4',
            '69.192.24.88','97.74.144.108','97.74.144.108',
            '75.127.97.72','75.127.97.72','69.192.24.88','75.127.97.72','74.55.1.4']
    num_node=40
        #for i in range(27):
    all_csv_dict[0]=[]
    i=2

    for i in range(27):
        if tag[i] not in all_csv_dict.keys():
            all_csv_dict[tag[i]]=[]
        DoS,normal=csv_dataset(addr[i],ip[i])
        if(DoS!={}):
            all_csv_dict[tag[i]].append(DoS)
            #print(len(all_csv_dict[1]))
        if(normal!={}):
            all_csv_dict["normal"].append(normal)

    for tags in all_csv_dict.keys():
        print(tags)
        all_flow=all_csv_dict[tags]
        for num in range(len(all_flow)):

            slow_body1_DoS=all_flow[num]
            for x in slow_body1_DoS.keys():
                all_traffic=slow_body1_DoS[x]
                num_graph=int(len(all_traffic)/num_node)
                #print(num_graph)
                last_graph=len(all_traffic)-num_node*num_graph
                for y in range(num_graph):
                    g = nx.Graph()
                    node_first=0
                    node_last=0
                    node_tags = []
                    node_tags1 = []
                    node_features = []
                    x_axis=1
                    y_axis=[]
                    y_pos=1
                    a=0

                    for z in range(num_node):
                        traffic=num_node*y+z
                        g.add_node(z)
                        if(all_traffic[traffic][2] not in lable_dict ):
                            lable_dict[all_traffic[traffic][2]]=len(lable_dict)
                        node_tags.append(lable_dict[all_traffic[traffic][2]])
                        node_tags1.append(all_traffic[traffic][2])
                        node_features.append(all_traffic[traffic][0])
                        if(z>0):#构造图的边
                            if(all_traffic[traffic][0]*all_traffic[traffic-1][0]>0):
                                y_pos+=1
                                #print(1)
                                g.add_edge(z-1, z,weight=all_traffic[traffic][1]-all_traffic[traffic-1][1])
                                #nx.draw(g,with_labels=True,pos=nx.circular_layout(g))
                                #plt.show()
                            else:
                                y_axis.append(y_pos)
                                y_pos=1
                                x_axis+=1
                                #print(2)
                                g.add_edge(z,node_first,weight=all_traffic[traffic][1]-all_traffic[traffic-node_first][1])

                                #nx.draw(g,with_labels=True,edge_color='r',pos=nx.circular_layout(g))
                                a=a+1
                                #plt.show()

                                if(a>=2):
                                    g.add_edge(node_last,z-1,weight=all_traffic[traffic-1][1]-all_traffic[traffic-node_last][1])
                                    #nx.draw(g,with_labels=True,edge_color='b',pos=nx.circular_layout(g))
                                node_first=z
                                node_last=z-1
                    g.add_edge(z,node_last,weight=all_traffic[traffic][1]-all_traffic[traffic-node_last][1])
                    y_axis.append(y_pos)
                    #print(x_axis,y_axis)
                    y_all=0.5*max(y_axis)
                    m=0
                    pos={}
                    labels={}
                    colors=[]
                    for pos_x in range(x_axis):
                        for pos_y in range(y_axis[pos_x]):
                            labels[m]=str(node_tags1[m])+"\n"+str(node_features[m])
                            if(node_features[m]>0):
                                colors.append('#c56822')
                            else:
                                colors.append('#1e76a6')
                            pos[m]=(pos_x,-y_all/(y_axis[pos_x]+1)*(pos_y+1))
                            m=m+1
                    if(tags=='slowread'):
                        #plt.title('normal')
                        #else:
                        plt.figure(figsize=(50,45))
                        plt.title(tags)
                        plt.xlim=(0,x_axis)
                        plt.ylim=(-y_all,0)

                        '''if node_features != []:
                            node_features = np.stack(node_features)
                            node_feature_flag = True
                        else:
                            node_features = None
                            node_feature_flag = False
    
    
                        g_list.append(S2VGraph(g, tag, node_tags))'''
                        nx.draw(g,
                                labels=labels, #这个选项让节点有名称
                                edge_color='black', # b stands for blue!
                                pos=pos, # 这个是选项选择点的排列方式，具体可以用 help(nx.drawing.layout) 查看
                                # 主要有spring_layout  (default), random_layout, circle_layout, shell_layout
                                # 这里是环形排布，还有随机排列等其他方式
                                node_color=colors, # r = red
                                node_size=1500, # 节点大小
                                width=1, # 边的宽度
                                font_color='white'

                                )
                        plt.show()

if __name__ == '__main__':
    main()