mirror of
https://github.com/elseif/MikroTikPatch.git
synced 2025-01-23 13:35:09 +03:00
60 lines
2.4 KiB
Python
60 lines
2.4 KiB
Python
from mitmproxy.tools.dump import DumpMaster
|
|
from mitmproxy import options,http
|
|
import os
|
|
|
|
#https://upgrade.mikrotik.com/routeros/NEWESTa7.stable
|
|
#https://upgrade.mikrotik.com/routeros/7.15.1/CHANGELOG
|
|
|
|
|
|
class UpgradeAddon:
|
|
def __init__(self, upstream_server):
|
|
self.upstream_server = upstream_server
|
|
def request(self,flow: http.HTTPFlow) -> None:
|
|
flow.request.host = self.upstream_server
|
|
flow.request.scheme = "https"
|
|
flow.request.port = 443
|
|
print(flow.request.url)
|
|
if len(flow.request.path_components)==3 and flow.request.path_components[0] == 'routeros':
|
|
version = flow.request.path_components[1]
|
|
file = os.path.join(version,flow.request.path_components[2])
|
|
if flow.request.method == 'HEAD':
|
|
if os.path.exists(version) and os.path.isfile(file):
|
|
flow.response = http.Response.make(
|
|
status_code=200,
|
|
headers={
|
|
'Content-Type': 'application/octet-stream',
|
|
'Accept-Ranges':'bytes',
|
|
'Content-Length': str(os.stat(file).st_size),
|
|
}
|
|
)
|
|
else:
|
|
flow.response = http.Response.make(status_code=404)
|
|
elif flow.request.method == 'GET' and flow.request.path_components[2].endswith('.npk'):
|
|
if os.path.exists(version) and os.path.isfile(file):
|
|
flow.response = http.Response.make(
|
|
status_code=200,
|
|
content=open(file,'rb').read(),
|
|
headers={'Content-Type': 'application/octet-stream',},
|
|
)
|
|
else:
|
|
flow.response = http.Response.make(status_code=404)
|
|
|
|
async def start_listen(port):
|
|
opts = options.Options(listen_host='0.0.0.0',listen_port=port)
|
|
upstream_server = "upgrade.mikrotik.com"
|
|
master = DumpMaster(opts)
|
|
master.addons.add(UpgradeAddon(upstream_server))
|
|
try:
|
|
await master.run()
|
|
except KeyboardInterrupt:
|
|
master.shutdown()
|
|
if __name__ == "__main__":
|
|
import asyncio
|
|
from package import check_install_package
|
|
check_install_package(['mitmproxy'])
|
|
print(f'ip dns static add name=upgrade.mikrotik.com address=<your ip address>')
|
|
print(f'ip dns cache flush')
|
|
asyncio.run(start_listen(80))
|
|
|
|
|