def sendServerWorkspace(vw, wsname, wsserver):
try:
events = vw.exportWorkspace()
server = viv_server.connectToServer(wsserver)
server.addNewWorkspace(wsname, events)
except Exception, e:
vw.vprint('Workspace Server Error: %s' % e)
return
def sendServerWorkspace(vw, wsname, wsserver):
try:
events = vw.exportWorkspace()
server = viv_server.connectToServer(wsserver)
server.addNewWorkspace(wsname, events)
except Exception, e:
vw.vprint('Workspace Server Error: %s' % e)
return
def connServerAndWorkspace(vw, host, parent=None):
# NOTE: do *not* touch parent (or qt) in here!
try:
server = viv_server.connectToServer(host)
wslist = server.listWorkspaces()
selectServerWorkspace(vw, server, wslist, parent=parent)
except Exception, e:
vw.vprint('Server Error: %s' % e)
return
def connServerAndWorkspace(vw, host,parent=None):
# NOTE: do *not* touch parent (or qt) in here!
try:
server = viv_server.connectToServer(host)
wslist = server.listWorkspaces()
selectServerWorkspace(vw, server, wslist, parent=parent)
except Exception, e:
vw.vprint('Server Error: %s' % e)
return
def test_basic(self):
testfile = helpers.getTestPath('windows', 'amd64', 'firefox.exe')
good = vivisect.VivWorkspace()
good.loadFromFile(testfile)
host = '0.0.0.0'
port = 0x4097
with tempfile.TemporaryDirectory() as tmpd:
with tempfile.NamedTemporaryFile(dir=tmpd) as tmpf:
proc = mp.Process(target=runServer, args=(
tmpf.name,
port,
))
proc.daemon = True
proc.start()
# give the other process time to spin up
time.sleep(0.5)
# So...yea. The server could not be up yet, but I'm not waiting a mmillion years to
# wait for it.
retry = 0
conn = False
while retry < 5:
try:
server = v_r_server.connectToServer(host, port)
conn = True
break
except:
retry += 1
time.sleep(0.2)
if not conn:
self.fail('Could not connect to %s:%s' % (host, port))
wslist = server.listWorkspaces()
self.assertEqual(len(wslist), 1)
self.assertEqual(server.getServerVersion(), 20130820)
othr = v_r_server.getServerWorkspace(server, wslist[0])
# So the consumption of events from the server is *also* threaded, so I've got to do some blocking
# to get us to wait on things
retry = 0
while retry < 5:
locs = othr.getLocations()
if len(locs) != 1380:
retry += 1
time.sleep(0.2)
else:
break
self.assertEqual(len(othr.getLocations()), 1380)
self.assertEqual(set(othr.getLocations()),
set(good.getLocations()))
self.assertEqual(set(othr.getXrefs()), set(good.getXrefs()))
try:
proc.terminate()
proc.close()
except:
pass
def test_basic(self):
testfile = helpers.getTestPath('windows', 'amd64', 'firefox.exe')
good = vivisect.VivWorkspace()
good.loadFromFile(testfile)
host = 'localhost'
port = 0x4097
with tempfile.TemporaryDirectory() as tmpd:
tmpf = tempfile.NamedTemporaryFile(dir=tmpd, delete=False)
try:
proc = mp.Process(target=runServer, args=(tmpf.name, port,))
proc.daemon = True
proc.start()
# give the other process time to spin up
time.sleep(0.5)
# So...yea. The server could not be up yet, but I'm not waiting a million years to
# wait for it.
retry = 0
conn = False
while retry < 5:
try:
server = v_r_server.connectToServer(host, port)
conn = True
break
except:
retry += 1
time.sleep(0.2)
if not conn:
self.fail('Could not connect to %s:%s' % (host, port))
wslist = server.listWorkspaces()
self.assertEqual(len(wslist), 1)
self.assertEqual(server.getServerVersion(), 20130820)
othr = v_r_server.getServerWorkspace(server, wslist[0])
# So the consumption of events from the server is *also* threaded, so I've got to do some blocking
# to get us to wait on things
retry = 0
while retry < 5:
locs = othr.getLocations()
if len(locs) != 1389:
retry += 1
time.sleep(0.2)
else:
break
self.assertEqual(len(othr.getLocations()), 1389)
self.assertEqual(set(othr.getLocations()), set(good.getLocations()))
self.assertEqual(set(othr.getXrefs()), set(good.getXrefs()))
try:
othr.server = None
q = othr.chan_lookup.get(othr.rchan)
if q:
# So it's not reeeealy auto analysis fini, but it's a good enough stand-in to get
# the server thread to shutdown cleaner
q.puts((v_const.VWE_AUTOANALFIN, None))
proc.terminate()
proc.close()
except:
pass
finally:
tmpf.close()
os.unlink(tmpf.name)
