forked from kunsi/dotfiles
102 lines
3.2 KiB
Python
102 lines
3.2 KiB
Python
|
#!/usr/bin/python3
|
||
|
|
||
|
import psutil
|
||
|
import time
|
||
|
import sys
|
||
|
import logging
|
||
|
import subprocess
|
||
|
import requests
|
||
|
import socket
|
||
|
import netifaces
|
||
|
from netaddr import IPNetwork, IPAddress
|
||
|
|
||
|
# Threshold in KB
|
||
|
threshold = 100*1000
|
||
|
|
||
|
# Interface
|
||
|
interface = 'wlp4s0'
|
||
|
|
||
|
|
||
|
|
||
|
logging.basicConfig(
|
||
|
level=logging.INFO,
|
||
|
format="%(asctime)s [%(threadName)-12.12s] [%(levelname)-5.5s] %(message)s",
|
||
|
handlers=[
|
||
|
logging.StreamHandler()
|
||
|
])
|
||
|
log = logging.getLogger(__name__)
|
||
|
|
||
|
last_traffic = 0
|
||
|
|
||
|
if interface not in netifaces.interfaces():
|
||
|
log.error(interface + ' was not found')
|
||
|
sys.exit(1)
|
||
|
|
||
|
while True:
|
||
|
traffic = psutil.net_io_counters(pernic=True)[interface]
|
||
|
|
||
|
traffic_wifi = int((traffic.bytes_sent + traffic.bytes_recv)/1024)
|
||
|
|
||
|
log.info('Traffic on ' + interface + ': ' + str(traffic_wifi) + ' KB')
|
||
|
|
||
|
if (traffic_wifi - last_traffic) > threshold or traffic_wifi < last_traffic:
|
||
|
log.warning('Traffic since last mac change above threshold (or traffic counter changed strangely), changing MAC')
|
||
|
|
||
|
last_traffic = traffic_wifi
|
||
|
|
||
|
subprocess.run(['ip', 'link', 'set', interface, 'down'])
|
||
|
subprocess.run(['macchanger', '-r', interface])
|
||
|
subprocess.run(['ip', 'link', 'set', interface, 'up'])
|
||
|
|
||
|
timeout = time.time()+20
|
||
|
ip = "0.0.0.0"
|
||
|
|
||
|
while True:
|
||
|
try:
|
||
|
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||
|
s.connect(('10.255.255.255', 1))
|
||
|
|
||
|
ip = s.getsockname()[0]
|
||
|
|
||
|
log.info('Got IPv4 ' + ip)
|
||
|
|
||
|
s.close()
|
||
|
|
||
|
break
|
||
|
except:
|
||
|
time.sleep(1)
|
||
|
|
||
|
if time.time() >= timeout:
|
||
|
log.error('Timeout while waiting for IP address')
|
||
|
break
|
||
|
|
||
|
if IPAddress(ip) in IPNetwork("172.18.0.0/16"):
|
||
|
headers = {
|
||
|
'Host': 'www.wifionice.de',
|
||
|
'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64; irv:58.0) Gecko/20100101 Firefox/58.0',
|
||
|
'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
|
||
|
'Accept-Language': 'en-US,en;q=0.5',
|
||
|
'Referer': 'http://www.wifionice.de/',
|
||
|
'Content-Type': 'application/x-www-form-urlencoded',
|
||
|
'Cookie': 'csrf=42',
|
||
|
'Connection': 'keep-alive',
|
||
|
'Upgrade-Insecure-Requests': '1'
|
||
|
}
|
||
|
try:
|
||
|
log.info('Trying login to WIFIonICE in 5 seconds ...')
|
||
|
|
||
|
time.sleep(5)
|
||
|
|
||
|
r = requests.post('http://www.wifionice.de/de/', data={'login': 'true', 'CSRFToken': '42', 'connect': ''}, headers=headers, timeout=15)
|
||
|
|
||
|
if r.status_code == 200:
|
||
|
log.info('Logged in to WIFIonICE')
|
||
|
else:
|
||
|
log.warning('Logging in failed, got status code ' + str(r,status_code))
|
||
|
except requests.exceptions.ConnectTimeout:
|
||
|
log.error('Timeout while trying to login, visit http://www.wifionice.de/ to check')
|
||
|
except:
|
||
|
log.error('Error while trying to log in')
|
||
|
|
||
|
time.sleep(10)
|