ipv4mutate.py

import ipaddress
from random import randint
from typing import List

class IPv4Mutate:
	"""Mutates IPv4 addresses provided in dotted-decimal notation."""

	def __init__(self, iparg: str) -> None:
		# Validate and store the provided IP as an IPv4Address object
		self._ip_obj = ipaddress.IPv4Address(iparg)
		self._ip_str = iparg

	@property
	def mutate_binary(self) -> str:
		"""Each octet as its 8-bit binary representation."""
		return ".".join(f"{int(x):08b}" for x in self._ip_str.split("."))

	@property
	def mutate_dotted_hex(self) -> str:
		"""Each octet as its hexadecimal representation."""
		return ".".join(hex(int(x)) for x in self._ip_str.split("."))

	@property
	def mutate_hex(self) -> str:
		"""The full IP as a single hexadecimal string."""
		return f"0x{int(self._ip_obj):08x}"

	@property
	def mutate_octal(self) -> str:
		"""Each octet as its octal representation (leading 0o removed)."""
		return ".".join(oct(int(x)).replace("0o", "0") for x in self._ip_str.split("."))

	@property
	def mutate_octal_padded(self) -> str:
		"""Each octet as octal with random leading zero padding."""
		octets = [oct(int(x)).replace("0o", "0") for x in self._ip_str.split(".")]
		return ".".join(f"{'0' * randint(1, 20)}{o}" for o in octets)

	@property
	def mutate_zero_padded(self) -> str:
		"""Dotted-decimal with octets left-padded with zeros, to 3 digits."""
		return ".".join(x.zfill(3) for x in self._ip_str.split("."))

	@property
	def mutate_decimal(self) -> int:
		"""The integer representation of the IP."""
		return int(self._ip_obj)

	@property
	def mutate_zero_stripped(self) -> str:
		"""Strips middle octets if they are zero (0). This may change the provided IP,
		such as:
		- provided 127.0.5.7
		- this creates 127.5.7
		- expands back to 127.5.0.7, NOT the same IP as provided"""
		parts = self._ip_str.split(".")
		return ".".join([x for x in parts if x != "0"])

	@property
	def mutate_urlencoded(self) -> str:
		"""URL-encoded representation of the IP string."""
		return "".join(f"%{ord(c):02x}" for c in self._ip_str)

	def __repr__(self) -> str:
		return self._ip_str

	def __str__(self) -> str:
		return self._ip_str