keenetic-dpr-bypass/main.py

98 lines
2.8 KiB
Python
Raw Normal View History

2023-09-04 23:43:45 +03:00
import httplib
import hashlib
import json
2023-09-05 09:14:32 +03:00
import argparse
# import ipaddress
import socket
import urllib
2023-09-04 23:43:45 +03:00
class KeeneticAPI:
def __init__(self, destination="192.168.1.1"):
self._conn = httplib.HTTPConnection(destination)
self._cookie = ""
pass
def _get(self, url):
headers = {"Cookie": self._cookie}
self._conn.request("GET", url, "", headers)
res = self._conn.getresponse()
data = res.read()
return res, data
def _post(self, url, body):
headers = {"Content-type": "application/json", "Cookie": self._cookie}
self._conn.request("POST", url, body, headers)
res = self._conn.getresponse()
data = res.read()
return res, data
2023-09-05 09:14:32 +03:00
def _delete(self, url):
headers = {"Cookie": self._cookie}
self._conn.request("DELETE", url, "", headers)
res = self._conn.getresponse()
data = res.read()
return res, data
2023-09-04 23:43:45 +03:00
def auth(self, login, passw):
res, _ = self._get("/auth")
self._cookie = res.getheader("Set-Cookie")
if res.status == 401:
md5 = login + ":" + res.getheader("X-NDM-Realm") + ":" + passw
md5 = hashlib.md5(md5.encode('utf-8'))
sha = res.getheader("X-NDM-Challenge") + md5.hexdigest()
sha = hashlib.sha256(sha.encode('utf-8'))
self._post("/auth", json.dumps({"login": login, "password": sha.hexdigest()}))
def show_ip_route(self):
2023-09-05 09:14:32 +03:00
res, data = self._get("/rci/ip/route")
return json.loads(data)
2023-09-04 23:43:45 +03:00
2023-09-05 09:14:32 +03:00
def set_ip_route(self, route_data):
res, data = self._post("/rci/ip/route", json.dumps(route_data))
return res.status == 200
2023-09-04 23:43:45 +03:00
2023-09-05 09:14:32 +03:00
def remove_ip_route(self, route):
res, _ = self._delete("/rci/ip/route?" + urllib.urlencode(route))
return res.status == 200
def get_interfaces(self):
res, data = self._get("/rci/show/interface")
return json.loads(data, encoding='utf-8')
def interfaces(args):
2023-09-04 23:43:45 +03:00
api = KeeneticAPI()
api.auth("test", "test")
2023-09-05 09:14:32 +03:00
ifaces = api.get_interfaces()
print "{:>30} {:>30}".format("ID", "DESCRIPTION")
for iface in ifaces:
print "{:>30} {:>30}".format(iface, (ifaces[iface].get('description') or '-'))
def main():
parser = argparse.ArgumentParser(description="a script to do stuff")
subparsers = parser.add_subparsers()
parser_foo = subparsers.add_parser('interfaces')
parser_foo.set_defaults(func=interfaces)
args = parser.parse_args()
args.func(args)
# api = KeeneticAPI()
# api.auth("test", "test")
# routes = api.show_ip_route()
# for r in routes:
# print r['comment']
# hostname_data = socket.gethostbyname_ex('grafana.com')
2023-09-04 23:43:45 +03:00
2023-09-05 09:14:32 +03:00
# print hostname_data[2]
2023-09-04 23:43:45 +03:00
if __name__ == "__main__":
main()