finish conversion to python 3

This commit is contained in:
2020-01-07 15:43:49 -05:00
parent aab4793fe8
commit 7786b84285
2 changed files with 14 additions and 11 deletions
+9 -8
View File
@@ -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:
+5 -3
View File
@@ -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