added the ability to scan a range of IPs as part of the miner network by passing a string formatted as {ip_range_1_start}-{ip_range_1_end}, {ip_range_2_start}-{ip_range_2_end} to the miner network
This commit is contained in:
@@ -3,8 +3,8 @@ from settings import DEBUG
|
|||||||
|
|
||||||
|
|
||||||
logging.basicConfig(
|
logging.basicConfig(
|
||||||
filename="logfile.txt",
|
# filename="logfile.txt",
|
||||||
filemode="a",
|
# filemode="a",
|
||||||
format='[%(levelname)s][%(asctime)s](%(name)s) - %(message)s',
|
format='[%(levelname)s][%(asctime)s](%(name)s) - %(message)s',
|
||||||
datefmt='%x %X'
|
datefmt='%x %X'
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -1,5 +1,7 @@
|
|||||||
import ipaddress
|
import ipaddress
|
||||||
import asyncio
|
import asyncio
|
||||||
|
|
||||||
|
from network.net_range import MinerNetworkRange
|
||||||
from miners.miner_factory import MinerFactory
|
from miners.miner_factory import MinerFactory
|
||||||
from settings import NETWORK_PING_RETRIES as PING_RETRIES, NETWORK_PING_TIMEOUT as PING_TIMEOUT, \
|
from settings import NETWORK_PING_RETRIES as PING_RETRIES, NETWORK_PING_TIMEOUT as PING_TIMEOUT, \
|
||||||
NETWORK_SCAN_THREADS as SCAN_THREADS
|
NETWORK_SCAN_THREADS as SCAN_THREADS
|
||||||
@@ -18,29 +20,32 @@ class MinerNetwork:
|
|||||||
def __repr__(self):
|
def __repr__(self):
|
||||||
return str(self.network)
|
return str(self.network)
|
||||||
|
|
||||||
|
|
||||||
def get_network(self) -> ipaddress.ip_network:
|
def get_network(self) -> ipaddress.ip_network:
|
||||||
"""Get the network using the information passed to the MinerNetwork or from cache."""
|
"""Get the network using the information passed to the MinerNetwork or from cache."""
|
||||||
# if we have a network cached already, use that
|
# if we have a network cached already, use that
|
||||||
if self.network:
|
if self.network:
|
||||||
return self.network
|
return self.network
|
||||||
|
|
||||||
# if there is no IP address passed, default to 192.168.1.0
|
if "-" in self.ip_addr:
|
||||||
if not self.ip_addr:
|
print("getting network")
|
||||||
default_gateway = "192.168.1.0"
|
self.network = MinerNetworkRange(self.ip_addr)
|
||||||
# if we do have an IP address passed, use that
|
|
||||||
else:
|
else:
|
||||||
default_gateway = self.ip_addr
|
# if there is no IP address passed, default to 192.168.1.0
|
||||||
|
if not self.ip_addr:
|
||||||
|
default_gateway = "192.168.1.0"
|
||||||
|
# if we do have an IP address passed, use that
|
||||||
|
else:
|
||||||
|
default_gateway = self.ip_addr
|
||||||
|
|
||||||
# if there is no subnet mask passed, default to /24
|
# if there is no subnet mask passed, default to /24
|
||||||
if not self.mask:
|
if not self.mask:
|
||||||
subnet_mask = "24"
|
subnet_mask = "24"
|
||||||
# if we do have a mask passed, use that
|
# if we do have a mask passed, use that
|
||||||
else:
|
else:
|
||||||
subnet_mask = str(self.mask)
|
subnet_mask = str(self.mask)
|
||||||
|
|
||||||
# save the network and return it
|
# save the network and return it
|
||||||
self.network = ipaddress.ip_network(f"{default_gateway}/{subnet_mask}", strict=False)
|
self.network = ipaddress.ip_network(f"{default_gateway}/{subnet_mask}", strict=False)
|
||||||
return self.network
|
return self.network
|
||||||
|
|
||||||
async def scan_network_for_miners(self) -> None or list:
|
async def scan_network_for_miners(self) -> None or list:
|
||||||
|
|||||||
26
network/net_range.py
Normal file
26
network/net_range.py
Normal file
@@ -0,0 +1,26 @@
|
|||||||
|
import ipaddress
|
||||||
|
|
||||||
|
|
||||||
|
class MinerNetworkRange:
|
||||||
|
"""A MinerNetwork that takes a range of IP addresses.
|
||||||
|
|
||||||
|
:param ip_range: A range of IP addresses to put in the network.
|
||||||
|
Takes a string formatted as
|
||||||
|
{ip_range_1_start}-{ip_range_1_end}, {ip_range_2_start}-{ip_range_2_end}
|
||||||
|
|
||||||
|
"""
|
||||||
|
def __init__(self, ip_range: str):
|
||||||
|
ip_ranges = ip_range.replace(" ", "").split(",")
|
||||||
|
self.host_ips = []
|
||||||
|
for item in ip_ranges:
|
||||||
|
start, end = item.split("-")
|
||||||
|
start_ip = ipaddress.ip_address(start)
|
||||||
|
end_ip = ipaddress.ip_address(end)
|
||||||
|
networks = ipaddress.summarize_address_range(start_ip, end_ip)
|
||||||
|
for network in networks:
|
||||||
|
for host in network.hosts():
|
||||||
|
self.host_ips.append(host)
|
||||||
|
|
||||||
|
def hosts(self):
|
||||||
|
for x in self.host_ips:
|
||||||
|
yield x
|
||||||
@@ -16,5 +16,5 @@ whatsminer_pwd = "admin"
|
|||||||
### DEBUG MODE ###
|
### DEBUG MODE ###
|
||||||
# change this to debug = true
|
# change this to debug = true
|
||||||
# to enable debug mode.
|
# to enable debug mode.
|
||||||
debug = false
|
# debug = false
|
||||||
# debug = true
|
debug = true
|
||||||
|
|||||||
Reference in New Issue
Block a user