def saveglobals(self):
"""Save globals() into filename"""
ns = self.get_current_namespace()
from trex.utils.iofuncs import iofunctions
settings = read_packet(self.i_request)
filename = read_packet(self.i_request)
data = get_remote_data(ns, settings, mode='picklable').copy()
return iofunctions.save(data, filename)
def setglobal(self, name):
"""
Set global reference value
"""
ns = self.get_reference_namespace(name)
ns[name] = read_packet(self.i_request)
self.refresh_after_eval = True
def set_remote_view_settings(self):
"""
Set the namespace remote view settings
(see the namespace browser widget)
"""
self.remote_view_settings = read_packet(self.i_request)
self.enable_refresh_after_eval()
def test_bsdsockets():
"""Test write-read packet methods."""
# socket read/write testing - client and server in one thread
# (techtonik): the stuff below is placed into public domain
address = ("127.0.0.1", 9999)
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.setblocking(0)
server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
server.bind(address)
server.listen(2)
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client.connect(address)
client.send("data to be catched".encode('utf-8'))
# accepted server socket is the one we can read from
# note that it is different from server socket
accsock, addr = server.accept()
assert accsock.recv(4096) == b'data to be catched'
# Testing BSD socket write_packet/read_packet
write_packet(client, "a tiny piece of data")
read = read_packet(accsock)
assert read == "a tiny piece of data"
client.close()
server.close()
def loadglobals(self):
"""Load globals() from filename"""
glbs = self.mglobals()
from trex.utils.iofuncs import iofunctions
filename = read_packet(self.i_request)
ext = read_packet(self.i_request)
load_func = iofunctions.load_funcs[ext]
data, error_message = load_func(filename)
if error_message:
return error_message
for key in list(data.keys()):
new_key = fix_reference_name(key, blacklist=list(glbs.keys()))
if new_key != key:
data[new_key] = data.pop(key)
try:
glbs.update(data)
except Exception as error:
return str(error)
self.refresh_after_eval = True
def run(self):
"""Start notification thread"""
while True:
if self.notify_socket is None:
continue
output = None
try:
try:
cdict = read_packet(self.notify_socket)
except:
# This except statement is intended to handle a struct.error
# (but when writing 'except struct.error', it doesn't work)
# Note: struct.error is raised when the communication has
# been interrupted and the received data is not a string
# of length 8 as required by struct.unpack (see read_packet)
break
if cdict is None:
# Another notification thread has just terminated and
# then wrote 'None' in the notification socket
# (see the 'finally' statement below)
continue
if not isinstance(cdict, dict):
raise TypeError("Invalid data type: %r" % cdict)
command = cdict['command']
data = cdict.get('data')
if command == 'pdb_step':
fname, lineno = data
self.sig_pdb.emit(fname, lineno)
self.refresh_namespace_browser.emit()
elif command == 'refresh':
self.refresh_namespace_browser.emit()
elif command == 'remote_view':
self.sig_process_remote_view.emit(data)
elif command == 'open_file':
fname, lineno = data
self.open_file.emit(fname, lineno)
else:
raise RuntimeError('Unsupported command: %r' % command)
if DEBUG_INTROSPECTION:
logging.debug("received command: %r" % command)
except:
log_last_error(LOG_FILENAME, "notification thread")
finally:
try:
write_packet(self.notify_socket, output)
except:
# The only reason why it should fail is that TRex is
# closing while this thread is still alive
break
def run(self):
"""Start server"""
sock = socket.socket(socket.AF_INET)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
sock.bind(("127.0.0.1", self.port))
while True:
sock.listen(2)
try:
conn, _addr = sock.accept()
except socket.error as e:
# See Issue 1275 for details on why errno EINTR is
# silently ignored here.
eintr = errno.WSAEINTR if os.name == 'nt' else errno.EINTR
if e.args[0] == eintr:
continue
raise
shell_id = read_packet(conn)
if shell_id is not None:
self.send_socket(shell_id, conn)
def run(self):
while True:
output = pickle.dumps(None, PICKLE_HIGHEST_PROTOCOL)
glbs = self.mglobals()
try:
if DEBUG_MONITOR:
logging.debug("****** Introspection request /Begin ******")
command = PACKET_NOT_RECEIVED
try:
timeout = self.timeout if self.auto_refresh else None
command = read_packet(self.i_request, timeout=timeout)
if command is None:
continue
timed_out = False
except socket.timeout:
timed_out = True
except struct.error:
# This should mean that TRex GUI has crashed
if DEBUG_MONITOR:
logging.debug("struct.error -> quitting monitor")
break
if timed_out:
if DEBUG_MONITOR:
logging.debug(
"connection timed out -> updating remote view")
self.update_remote_view()
if DEBUG_MONITOR:
logging.debug(
"****** Introspection request /End ******")
continue
if DEBUG_MONITOR:
logging.debug("command: %r" % command)
lcls = self.mlocals()
result = eval(command, glbs, lcls)
if DEBUG_MONITOR:
logging.debug(" result: %r" % result)
if self.pdb_obj is None:
lcls["_"] = result
# old com implementation: (see solution (1) in Issue 434)
output = pickle.dumps(result, PICKLE_HIGHEST_PROTOCOL)
# # new com implementation: (see solution (2) in Issue 434)
# output = pickle.dumps((command, result),
# PICKLE_HIGHEST_PROTOCOL)
except SystemExit:
break
except:
if DEBUG_MONITOR:
logging.debug("error!")
log_last_error(LOG_FILENAME, command)
finally:
try:
if DEBUG_MONITOR:
logging.debug("updating remote view")
if self.refresh_after_eval:
self.update_remote_view()
self.refresh_after_eval = False
if DEBUG_MONITOR:
logging.debug("sending result")
logging.debug(
"****** Introspection request /End ******")
if command is not PACKET_NOT_RECEIVED:
if write_packet is None:
# This may happen during interpreter shutdown
break
else:
write_packet(self.i_request,
output,
already_pickled=True)
except AttributeError as error:
if "'NoneType' object has no attribute" in str(error):
# This may happen during interpreter shutdown
break
else:
raise
except TypeError as error:
if "'NoneType' object is not subscriptable" in str(error):
# This may happen during interpreter shutdown
break
else:
raise
self.i_request.close()
self.n_request.close()
def setenv(self):
"""Set os.environ"""
env = read_packet(self.i_request)
os.environ = env