36 lines
895 B
Python
36 lines
895 B
Python
import threading
|
|
import copy
|
|
import os
|
|
import subprocess
|
|
from mswp import Datapack
|
|
from forwarder import receive_queues, send_queue
|
|
from config import msw_queue
|
|
from config import dprint as print
|
|
receive_queue = receive_queues[__name__]
|
|
|
|
|
|
def main():
|
|
while True:
|
|
dp = receive_queue.get()
|
|
command = dp.body.decode()
|
|
try:
|
|
result = subprocess.check_output(command, shell=True)
|
|
except Exception as e:
|
|
result = 'Command %s error, %s: %s' % (command, type(e), str(e))
|
|
result = result.encode()
|
|
|
|
ndp = dp.reply()
|
|
ndp.body = try_decode_and_encode(result)
|
|
send_queue.put(ndp)
|
|
|
|
|
|
def try_decode_and_encode(data):
|
|
try:
|
|
return data.decode('gb2312').encode()
|
|
except:
|
|
return data.decode('utf-8').encode()
|
|
|
|
thread = threading.Thread(target=main, args=(), daemon=True)
|
|
thread.start()
|
|
|