init: workspace opencode sync
This commit is contained in:
+142
@@ -0,0 +1,142 @@
|
||||
import socket, hashlib, time
|
||||
from datetime import datetime, timedelta
|
||||
|
||||
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(10)
|
||||
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')
|
||||
|
||||
# Get router time
|
||||
r = api.cmd('/system/clock/print')[0]
|
||||
print(f"Router time: {r['date']} {r['time']}")
|
||||
|
||||
# Parse router time
|
||||
months = {'jan':'jan','feb':'feb','mar':'mar','apr':'apr','may':'may','jun':'jun',
|
||||
'jul':'jul','aug':'aug','sep':'sep','oct':'oct','nov':'nov','dec':'dec'}
|
||||
parts = r['date'].split('/')
|
||||
router_date = f"{parts[0]}/{parts[1]}/{parts[2]}"
|
||||
router_time = r['time']
|
||||
|
||||
# Create comment: 2 minutes in the past
|
||||
past_comment = f"{router_date} 00:00:00"
|
||||
print(f"Setting user comment (past): {past_comment}")
|
||||
|
||||
# Find next scheduler run
|
||||
for item in api.cmd('/system/scheduler/print'):
|
||||
name = item['name']
|
||||
if 'Bulanan2' in name:
|
||||
next_run = item.get('next-run', '?')
|
||||
run_before = int(item.get('run-count', '0'))
|
||||
print(f"Bulanan2: next-run={next_run} run-count={run_before}")
|
||||
|
||||
# Create test user
|
||||
test_name = "tes1menit"
|
||||
r = api.cmd('/ip/hotspot/user/add', **{
|
||||
'name': test_name,
|
||||
'password': 'x',
|
||||
'profile': 'Bulanan2',
|
||||
'comment': past_comment
|
||||
})
|
||||
print(f"Created user: {test_name}")
|
||||
|
||||
# Wait for scheduler
|
||||
wait_sec = 180
|
||||
print(f"\nWaiting {wait_sec}s for scheduler to run...")
|
||||
for i in range(wait_sec):
|
||||
time.sleep(1)
|
||||
if i > 0 and i % 30 == 0:
|
||||
r = api.cmd('/ip/hotspot/user/print', **{'?name': test_name})
|
||||
if r:
|
||||
print(f" [{i}s] User exists. limit-uptime={r[0].get('limit-uptime','')}")
|
||||
else:
|
||||
print(f" [{i}s] User REMOVED!")
|
||||
break
|
||||
|
||||
# Final check
|
||||
print()
|
||||
r = api.cmd('/ip/hotspot/user/print', **{'?name': test_name})
|
||||
if r:
|
||||
u = r[0]
|
||||
print(f"User still exists. limit-uptime={u.get('limit-uptime')} comment={u.get('comment')}")
|
||||
if u.get('limit-uptime') == '1s':
|
||||
print("=> EXPIRED (limit-uptime=1s) ✅")
|
||||
else:
|
||||
print("=> NOT EXPIRED yet")
|
||||
# Cleanup
|
||||
api.cmd('/ip/hotspot/user/remove', **{'.id': u['.id']})
|
||||
print("Cleaned up.")
|
||||
else:
|
||||
print("User was REMOVED by scheduler. EXPIRY WORKING ✅")
|
||||
|
||||
api.close()
|
||||
Reference in New Issue
Block a user