refactor!: Remove unused functions and imports for cleaner code

Author: weilycoder

Jiyu_udp_attack/sender.py

@@ -2,125 +2,18 @@
 This module is used to send or broadcast UDP packets with spoofed IP addresses.
 """
 
-import struct
-import random
 import socket
 from typing import List, Optional, Tuple
 
 from scapy.all import send as scapy_send
-from scapy.all import IP, UDP, Raw, RandShort  # type: ignore
+from scapy.all import IP, UDP, Raw, RandShort  # pylint: disable=no-name-in-module  # type: ignore
 
 try:
     from Jiyu_udp_attack.ip_analyze import ip_analyze
 except ImportError:
     from ip_analyze import ip_analyze
 
 
-def calculate_checksum(data: bytes) -> int:
-    """
-    Calculates the checksum for the given data.
-
-    Args:
-        data (bytes): The data for which to calculate the checksum.
-
-    Returns:
-        int: The calculated checksum.
-    """
-    data = data + b"\x00" * (len(data) % 2)  # Ensure even length
-
-    total = 0
-    for word in struct.unpack("!" + "H" * (len(data) // 2), data):
-        total += word
-        if total > 0xFFFF:
-            total = (total & 0xFFFF) + (total >> 16)
-
-    return ~total & 0xFFFF
-
-
-def create_raw_udp_packet(
-    src_ip: str,
-    src_port: Optional[int],
-    dst_ip: str,
-    dst_port: int,
-    payload: bytes,
-    *,
-    ip_id: Optional[int] = None,
-) -> bytes:
-    """
-    Creates a raw UDP packet with a spoofed source IP address.
-
-    Args:
-        src_ip (str): The source IP address to spoof.
-        dst_ip (str): The destination IP address.
-        dst_port (int): The destination port number.
-        payload (bytes): The data payload to include in the packet.
-
-    Returns:
-        bytes: The constructed raw UDP packet.
-    """
-    # 1. Set IP header parameters
-    ip_ver = 4
-    ip_ihl = 5  # 5 * 4 = 20 bytes header
-    ip_tos = 0
-    ip_total_len = 20 + 8 + len(payload)  # IP header + UDP header + data
-    ip_id = random.randint(0, 65535) if ip_id is None else ip_id
-    ip_frag_off = 0
-    ip_ttl = 64
-    ip_proto = socket.IPPROTO_UDP
-    ip_check = 0  # Initial value is 0, will be calculated later
-
-    # 2. Build IP header (initial checksum is 0)
-    ip_header = struct.pack(
-        "!BBHHHBBH4s4s",
-        (ip_ver << 4) | ip_ihl,
-        ip_tos,
-        ip_total_len,
-        ip_id,
-        ip_frag_off,
-        ip_ttl,
-        ip_proto,
-        ip_check,
-        socket.inet_aton(src_ip),
-        socket.inet_aton(dst_ip),
-    )
-
-    # 3. Calculate IP header checksum
-    ip_check = calculate_checksum(ip_header)
-
-    # 4. Rebuild IP header with correct checksum
-    ip_header = struct.pack(
-        "!BBHHHBBH4s4s",
-        (ip_ver << 4) | ip_ihl,
-        ip_tos,
-        ip_total_len,
-        ip_id,
-        ip_frag_off,
-        ip_ttl,
-        ip_proto,
-        ip_check,
-        socket.inet_aton(src_ip),
-        socket.inet_aton(dst_ip),
-    )
-
-    # 5. Build UDP header (initial checksum is 0)
-    if src_port is None:
-        src_port = random.randint(1024, 65535)  # Random source port
-    udp_length = 8 + len(payload)
-    udp_header = struct.pack("!HHHH", src_port, dst_port, udp_length, 0)  # Initial checksum is 0
-
-    # 6. Create pseudo header for UDP checksum calculation
-    pseudo_header = struct.pack("!4s4sBBH", socket.inet_aton(src_ip), socket.inet_aton(dst_ip), 0, ip_proto, udp_length)
-
-    # 7. Calculate UDP checksum (including pseudo header)
-    udp_check = calculate_checksum(pseudo_header + udp_header + payload)
-
-    # 8. Rebuild UDP header with correct checksum
-    udp_header = struct.pack("!HHHH", src_port, dst_port, udp_length, udp_check)
-
-    # 9. Combine complete packet
-    return ip_header + udp_header + payload
-
-
 def send_packet(
     src_ip: Optional[str],
     src_port: Optional[int],

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "Jiyu-udp-attack"
-version = "1.2.1"
+version = "2.0.0"
 description = "Send packets to student machines disguised as Jiyu teacher machine"
 authors = ["weilycoder <[email protected]>"]
 license = "MIT"