Commit 78a08c26 by fwkz

Adding IPv4 validator.

parent 070c247b
class RoutersploitException(Exception): class RoutersploitException(Exception):
pass pass
class OptionValidationError(RoutersploitException):
pass
\ No newline at end of file
import socket
from .exceptions import OptionValidationError
def url(address): def url(address):
"""Sanitize url. """Sanitize url.
...@@ -7,3 +12,18 @@ def url(address): ...@@ -7,3 +12,18 @@ def url(address):
return address return address
else: else:
return "http://{}".format(address) return "http://{}".format(address)
def ipv4(address):
try:
socket.inet_pton(socket.AF_INET, address)
except AttributeError:
try:
socket.inet_aton(address)
except socket.error:
raise OptionValidationError("Option have to be valid IP address.")
return address.count('.') == 3
except socket.error:
raise OptionValidationError("Option have to be valid IP address.")
return address
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment