finish conversion to python 3
This commit is contained in:
@@ -16,8 +16,7 @@ import shutil
|
|||||||
import zlib
|
import zlib
|
||||||
import subprocess
|
import subprocess
|
||||||
import syslog
|
import syslog
|
||||||
|
import codecs
|
||||||
import syslog
|
|
||||||
|
|
||||||
|
|
||||||
PORT = 50003
|
PORT = 50003
|
||||||
@@ -136,7 +135,7 @@ def dicttos(ID, d):
|
|||||||
|
|
||||||
def stodict(msg):
|
def stodict(msg):
|
||||||
d = {}
|
d = {}
|
||||||
if len(msg) > 0 and msg[0] == "!":
|
if len(msg) > 0 and chr(msg[0]) == "!":
|
||||||
pk = zlib.decompress(msg[5:]).decode()
|
pk = zlib.decompress(msg[5:]).decode()
|
||||||
d['ID'] = msg[1:4].decode()
|
d['ID'] = msg[1:4].decode()
|
||||||
else:
|
else:
|
||||||
@@ -151,8 +150,10 @@ def stodict(msg):
|
|||||||
d[k] = None
|
d[k] = None
|
||||||
else:
|
else:
|
||||||
v = vr[1].strip()
|
v = vr[1].strip()
|
||||||
if v[0].isdigit():
|
try:
|
||||||
v = eval(v)
|
v = eval(v)
|
||||||
|
except:
|
||||||
|
pass
|
||||||
d[k] = v
|
d[k] = v
|
||||||
if verbose: print("msg is %s" % d)
|
if verbose: print("msg is %s" % d)
|
||||||
return d
|
return d
|
||||||
@@ -241,7 +242,7 @@ def doexec(conn, data):
|
|||||||
def doupdate(conn, msgDict):
|
def doupdate(conn, msgDict):
|
||||||
fail = None
|
fail = None
|
||||||
try:
|
try:
|
||||||
code = msgDict['code'].decode('base64')
|
code = codecs.decode(msgDict['code'],'base64').decode()
|
||||||
csum = msgDict['csum']
|
csum = msgDict['csum']
|
||||||
except:
|
except:
|
||||||
fail = "csum/code missing"
|
fail = "csum/code missing"
|
||||||
@@ -258,8 +259,8 @@ def doupdate(conn, msgDict):
|
|||||||
|
|
||||||
def doupdateone(code, csum):
|
def doupdateone(code, csum):
|
||||||
|
|
||||||
m = md5.new()
|
m = md5()
|
||||||
m.update(code)
|
m.update(code.encode())
|
||||||
icsum = m.hexdigest()
|
icsum = m.hexdigest()
|
||||||
if icsum != csum:
|
if icsum != csum:
|
||||||
return "checksum error"
|
return "checksum error"
|
||||||
@@ -330,7 +331,7 @@ def process():
|
|||||||
for rfh in r[0]:
|
for rfh in r[0]:
|
||||||
conn = conIds[rfh]
|
conn = conIds[rfh]
|
||||||
data, addr = ifiles[rfh].recvfrom(MAXRECV)
|
data, addr = ifiles[rfh].recvfrom(MAXRECV)
|
||||||
# if verbose: print("sock.recvfrom: %s (%s) %s" % (addr, len(data), data[:4]))
|
if verbose: print("sock.recvfrom: %s (%s) %s" % (addr, len(data), data[:4]))
|
||||||
try:
|
try:
|
||||||
msgDict = stodict(data)
|
msgDict = stodict(data)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
|||||||
@@ -26,6 +26,7 @@ import queue
|
|||||||
from hashlib import md5
|
from hashlib import md5
|
||||||
import json
|
import json
|
||||||
import zlib
|
import zlib
|
||||||
|
import codecs
|
||||||
|
|
||||||
from subprocess import Popen, STDOUT, PIPE
|
from subprocess import Popen, STDOUT, PIPE
|
||||||
|
|
||||||
@@ -501,10 +502,11 @@ def updatecode(ucode, uname):
|
|||||||
except Exception as e:
|
except Exception as e:
|
||||||
fail = "cannot read new code: %s" % e
|
fail = "cannot read new code: %s" % e
|
||||||
if not fail:
|
if not fail:
|
||||||
m = md5.new()
|
m = md5()
|
||||||
m.update(new_code)
|
new_codeE = new_code.encode()
|
||||||
|
m.update(new_codeE)
|
||||||
icsum = m.hexdigest()
|
icsum = m.hexdigest()
|
||||||
rmsg = {'csum': icsum, 'code': new_code.encode('base64','strict') }
|
rmsg = {'csum': icsum, 'code': codecs.encode(new_codeE, 'base64') }
|
||||||
hbdclass.Host.hosts[uname].cmds.append(('UPD',rmsg))
|
hbdclass.Host.hosts[uname].cmds.append(('UPD',rmsg))
|
||||||
return fail
|
return fail
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user