Magic-Mirror/llvm-capstone
1
0
Fork 0
mirror of https://github.com/capstone-engine/llvm-capstone.git synced 2025-01-10 10:01:42 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Make swig_bot server support Ctrl+C

Browse Source 
llvm-svn: 254024
This commit is contained in:
Zachary Turner 2015-11-24 21:35:40 +00:00
parent 0a2899ca82
commit cc3609362e
2 changed files with 20 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lldb/scripts/swig_bot.py
View File

@ -57,6 +57,9 @@ if __name__ == "__main__":
else:
logging.error("Unknown mode specified. Expected client or server.")
sys.exit(-1)
except KeyboardInterrupt as e:
logging.info("Ctrl+C received. Shutting down...")
sys.exit(-1)
except Exception as e:
error = traceback.format_exc()
logging.error("An error occurred running swig-bot.")

26
lldb/scripts/swig_bot_lib/server.py
View File

@ -13,6 +13,7 @@ from __future__ import print_function
import argparse
import logging
import os
import select
import socket
import struct
import sys
@ -53,16 +54,23 @@ def initialize_listening_socket(options):
def accept_once(sock, options):
logging.debug("Waiting for connection...")
client, addr = sock.accept()
logging.info("Received connection from {}".format(addr))
data_size = struct.unpack("!I", sockutil.recvall(client, 4))[0]
logging.debug("Expecting {} bytes of data from client".format(data_size))
data = sockutil.recvall(client, data_size)
logging.info("Received {} bytes of data from client".format(len(data)))
while True:
rlist, wlist, xlist = select.select([sock], [], [], 0.5)
if not rlist:
continue
logging.info("Sending {} byte response".format(len(data)))
client.sendall(struct.pack("!I", len(data)))
client.sendall(data)
client, addr = sock.accept()
logging.info("Received connection from {}".format(addr))
data_size = struct.unpack("!I", sockutil.recvall(client, 4))[0]
logging.debug("Expecting {} bytes of data from client"
.format(data_size))
data = sockutil.recvall(client, data_size)
logging.info("Received {} bytes of data from client"
.format(len(data)))
logging.info("Sending {} byte response".format(len(data)))
client.sendall(struct.pack("!I", len(data)))
client.sendall(data)
def accept_loop(sock, options):
while True: