diff options
Diffstat (limited to 'lldb/scripts/swig_bot_lib/client.py')
| -rw-r--r-- | lldb/scripts/swig_bot_lib/client.py | 93 |
1 files changed, 73 insertions, 20 deletions
diff --git a/lldb/scripts/swig_bot_lib/client.py b/lldb/scripts/swig_bot_lib/client.py index 9fff5bfb411..8dd1a8f5ae4 100644 --- a/lldb/scripts/swig_bot_lib/client.py +++ b/lldb/scripts/swig_bot_lib/client.py @@ -1,5 +1,10 @@ #!/usr/bin/env python +""" +SWIG generation client. Supports both local and remote generation of SWIG +bindings for multiple languages. +""" + # Future imports from __future__ import absolute_import from __future__ import print_function @@ -8,11 +13,20 @@ from __future__ import print_function import argparse import logging import os +import socket +import struct import sys # LLDB modules import use_lldb_suite from lldbsuite.support import fs +from lldbsuite.support import sockutil + +# package imports +from . import local + +default_ip = "127.0.0.1" +default_port = 8537 def process_args(args): """Returns options processed from the provided command line. @@ -20,9 +34,14 @@ def process_args(args): @param args the command line to process. """ + # A custom action used by the --local command line option. It can be + # used with either 0 or 1 argument. If used with 0 arguments, it + # searches for a copy of swig located on the physical machine. If + # used with 1 argument, the argument is the path to a swig executable. class FindLocalSwigAction(argparse.Action): def __init__(self, option_strings, dest, **kwargs): - super(FindLocalSwigAction, self).__init__(option_strings, dest, nargs='?', **kwargs) + super(FindLocalSwigAction, self).__init__( + option_strings, dest, nargs='?', **kwargs) def __call__(self, parser, namespace, values, option_string=None): swig_exe = None if values is None: @@ -31,17 +50,33 @@ def process_args(args): swig_exe = values setattr(namespace, self.dest, os.path.normpath(swig_exe)) + # A custom action used by the --remote command line option. It can be + # used with either 0 or 1 arguments. If used with 0 arguments it chooses + # a default connection string. If used with one argument it is a string + # of the form `ip_address[:port]`. If the port is unspecified, the + # default port is used. + class RemoteIpAction(argparse.Action): + def __init__(self, option_strings, dest, **kwargs): + super(RemoteIpAction, self).__init__( + option_strings, dest, nargs='?', **kwargs) + def __call__(self, parser, namespace, values, option_string=None): + ip_port = None + if values is None: + ip_port = (default_ip, default_port) + else: + result = values.split(':') + if len(result)==1: + ip_port = (result[0], default_port) + elif len(result)==2: + ip_port = (result[0], int(result[1])) + else: + raise ValueError("Invalid connection string") + setattr(namespace, self.dest, ip_port) + # Setup the parser arguments that are accepted. parser = argparse.ArgumentParser( description='Generate SWIG bindings.') - # Arguments to control logging verbosity. - parser.add_argument( - "--verbose", "-v", - action="store_true", - default=False, - help="Increase logging verbosity level.") - parser.add_argument( "--local", action=FindLocalSwigAction, @@ -52,7 +87,7 @@ def process_args(args): parser.add_argument( "--remote", - action="store", + action=RemoteIpAction, help=( "Use the given connection string to connect to a remote " "generation service")) @@ -85,24 +120,42 @@ def process_args(args): logging.error("Must specify either --local or --remote") sys.exit(-3) - # Set logging level based on verbosity count. - if options.verbose: - log_level = logging.DEBUG - else: - log_level = logging.NOTSET - logging.basicConfig(level=log_level) - logging.info("logging is using level: %d", log_level) - return options +def establish_remote_connection(ip_port): + logging.debug("Creating socket...") + s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) + logging.info("Connecting to server {} on port {}" + .format(ip_port[0], ip_port[1])) + s.connect(ip_port) + logging.info("Connection established...") + return s + +def transmit_data(connection, packed_input): + logging.info("Sending {} bytes of compressed data." + .format(len(packed_input))) + connection.sendall(struct.pack("!I", len(packed_input))) + connection.sendall(packed_input) + logging.info("Awaiting response.") + response_len = struct.unpack("!I", sockutil.recvall(connection, 4))[0] + logging.debug("Expecting {} byte response".format(response_len)) + response = sockutil.recvall(connection, response_len) + return response def run(args): options = process_args(args) if options.remote is None: + logging.info("swig bot client using local swig installation at '{}'" + .format(options.swig_executable)) if not os.path.isfile(options.swig_executable): - logging.error("Swig executable '%s' does not exist." % options.swig_executable) - from . import local + logging.error("Swig executable '{}' does not exist." + .format(options.swig_executable)) local.generate(options) else: - logging.error("Remote path is not yet implemented!") + logging.info("swig bot client using remote generation with server '{}'" + .format(options.remote)) + packed_input = local.pack_input(options) + connection = establish_remote_connection(options.remote) + response = transmit_data(connection, packed_input) + logging.debug("Received {} byte response.".format(len(response))) |
