Files
mikhmon-fix/tg-test-simple.py
2026-08-15 00:28:08 +07:00

124 lines
4.2 KiB
Python

import socket, hashlib, time
class R:
def __init__(self,h,p=8728): self.h=h; self.p=p; self.s=None
def c(self,u,p):
self.s=socket.socket(socket.AF_INET,socket.SOCK_STREAM); self.s.settimeout(15)
self.s.connect((self.h,self.p))
self._ws('/login','=name='+u,'=password='+p)
r=self._rr()
if any(isinstance(x,tuple) and x[0]=='!trap' for x in r):
for x in r:
if isinstance(x,dict) and 'ret' in x: ch=x['ret']
if ch and len(ch)==32:
h=hashlib.md5(b'\x00'+p.encode()+bytes.fromhex(ch)).hexdigest()
self._ws('/login','=name='+u,'=response=00'+h)
self._rr()
return self
def _el(self,l):
if l<0x80: return bytes([l])
elif l<0x4000: l|=0x8000; return bytes([(l>>8)&0xFF,l&0xFF])
elif l<0x200000: l|=0xC00000; return bytes([(l>>16)&0xFF,(l>>8)&0xFF,l&0xFF])
elif l<0x10000000: l|=0xE0000000; return bytes([(l>>24)&0xFF,(l>>16)&0xFF,(l>>8)&0xFF,l&0xFF])
else: return bytes([0xF0,(l>>24)&0xFF,(l>>16)&0xFF,(l>>8)&0xFF,l&0xFF])
def _ww(self,w): self.s.sendall(self._el(len(w))+w.encode('utf-8'))
def _ws(self,c,*ws): self._ww(c); [self._ww(w) for w in ws]; self.s.sendall(b'\x00')
def _rb(self): return self.s.recv(1)
def _rw(self):
f=self._rb()
if f==b'\x00': return None
lb=f[0]
if lb&0x80:
if (lb&0xC0)==0x80: l=((lb&0x3F)<<8)+self._rb()[0]
elif (lb&0xE0)==0xC0: l=((lb&0x1F)<<16)+(self._rb()[0]<<8)+self._rb()[0]
elif (lb&0xF0)==0xE0: l=((lb&0x0F)<<24)+(self._rb()[0]<<16)+(self._rb()[0]<<8)+self._rb()[0]
else: l=(self._rb()[0]<<24)+(self._rb()[0]<<16)+(self._rb()[0]<<8)+self._rb()[0]
else: l=lb
d=b''
while len(d)<l:
c=self.s.recv(l-len(d))
if not c: break
d+=c
return d.decode('utf-8',errors='replace')
def _rss(self):
ws=[]
while True:
w=self._rw()
if w is None: break
ws.append(w)
return ws
def _rr(self):
rs=[]
while True:
s=self._rss()
if not s: break
r=s[0]; t=s[1:]
if r=='!done': break
elif r=='!re':
d={}
for w in t:
if '=' in w:
p=w.split('=',2)
d[p[1]]=p[2] if len(p)>=3 else ''
else: d[w]=''
rs.append(d)
elif r=='!trap': rs.append(('!trap',t))
return rs
def cmd(self,c,**a):
ws=[c]
for k,v in a.items():
if k.startswith('?'): ws.append(f'{k}={v}')
else: ws.append(f'={k}={v}')
self._ws(*ws)
return self._rr()
def close(self):
if self.s: self.s.close(); self.s=None
api=R('192.168.100.2').c('mi4','m14')
# Cek waktu & Family scheduler
rt=api.cmd('/system/clock/print')[0]
print(f"Waktu router: {rt['date']} {rt['time']}")
for si in api.cmd('/system/scheduler/print'):
if si['name']=='Family':
print(f"Family: run={si.get('run-count')} next={si.get('next-run')}")
break
# Hapus user test sebelumnya
for u in api.cmd('/ip/hotspot/user/print',**{'?name':'tg_test_family'}):
api.cmd('/ip/hotspot/user/remove',**{'.id':u['.id']})
# Buat user Family expired
api.cmd('/ip/hotspot/user/add',**{
'name':'tg_test_family',
'password':'x',
'profile':'Family',
'comment':'jul/06/2026 00:00:00'
})
print("User tg_test_family dibuat (sudah expired)")
# Tunggu 3 menit (Family interval ~2m37s)
print("Tunggu 180 detik...")
api.close()
time.sleep(180)
# Buka koneksi baru untuk cek
api2=R('192.168.100.2').c('mi4','m14')
for u in api2.cmd('/ip/hotspot/user/print',**{'?name':'tg_test_family'}):
print(f"User ditemukan: limit-uptime={u.get('limit-uptime')}")
if u.get('limit-uptime')=='1s':
print("=> EXPIRED. Notif Telegram terkirim ke chat -765474293")
api2.cmd('/ip/hotspot/user/remove',**{'.id':u['.id']})
break
else:
print("User SUDAH DIHAPUS oleh scheduler. Notif Telegram terkirim!")
for si in api2.cmd('/system/scheduler/print'):
if si['name']=='Family':
print(f"Family run count now: {si.get('run-count')}")
api2.close()