Extract common code, add host IP to --debug and --enable-vm-service options

dart/runtime/bin/main.cc

 // Copyright (c) 2012, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 #include <stdlib.h>
 #include <string.h>
 #include <stdio.h>
 
 #include "include/dart_api.h"
 #include "include/dart_debugger_api.h"
@@ skipping 27 matching lines @@
 
 // Global state that indicates whether there is a debug breakpoint.
 // This pointer points into an argv buffer and does not need to be
 // free'd.
 static const char* breakpoint_at = NULL;
 
 
 // Global state that indicates whether we should open a connection
 // and listen for a debugger to connect.
 static bool start_debugger = false;
+static const char* debug_ip = NULL;
+static int debug_port = -1;
+static const char* DEFAULT_DEBUG_IP = "127.0.0.1";
 static const int DEFAULT_DEBUG_PORT = 5858;
-static const char* DEFAULT_DEBUG_IP = "127.0.0.1";
-static const char* debug_ip = DEFAULT_DEBUG_IP;
-static int debug_port = -1;
 
 // Value of the --package-root flag.
 // (This pointer points into an argv buffer and does not need to be
 // free'd.)
 static const char* package_root = NULL;
 
 
 // Global flag that is used to indicate that we want to compile all the
 // dart functions and not run anything.
 static bool has_compile_all = false;
 
 // Global flag that is used to indicate that we want to print the source code
 // for script that is being run.
 static bool has_print_script = false;
 
 
 // VM Service options.
 static bool start_vm_service = false;
+static const char *vm_service_server_ip = NULL;
 static int vm_service_server_port = -1;
+static const char *DEFAULT_VM_SERVICE_SERVER_IP = "127.0.0.1";
 static const int DEFAULT_VM_SERVICE_SERVER_PORT = 8181;
 
 // The environment provided through the command line using -D options.
 static dart::HashMap* environment = NULL;
 
 static bool IsValidFlag(const char* name,
                         const char* prefix,
                         intptr_t prefix_length) {
   intptr_t name_length = strlen(name);
   return ((name_length > prefix_length) &&
@@ skipping 48 matching lines @@
   }
   package_root = arg;
   return true;
 }
 
 
 static void* GetHashmapKeyFromString(char* key) {
   return reinterpret_cast<void*>(key);
 }
 
+
+static bool ExtractPortAndIP(const char *option_value,
+                             int *out_port,
+                             const char **out_ip,
+                             int default_port,
+                             const char *default_ip) {
+  // [option_value] has to be one of the following formats:
+  //   - ""
+  //   - ":8181"
+  //   - "=8181"
+  //   - ":8181/192.168.0.1"
+  //   - "=8181/192.168.0.1"
+
+  if (*option_value== '\0') {
+    *out_ip = default_ip;
+    *out_port = default_port;
+    return true;
+  }
+
+  if ((*option_value != '=') && (*option_value != ':')) {
+    return false;
+  }
+
+  int port = atoi(option_value + 1);
+  const char *slash = strstr(option_value, "/");
+  if (slash == NULL) {
+    *out_ip = default_ip;
+    *out_port = port;
+    return true;
+  }
+
+  int _, n;
+  if (sscanf(option_value + 1, "%d/%d.%d.%d.%d%n",  // NOLINT(runtime/printf)
+             &_, &_, &_, &_, &_, &n)) {
+    if (option_value[1 + n] == '\0') {
+      *out_ip = slash + 1;
+      *out_port = port;
+      return true;
+    }
+  }
+  return false;
+}
+
+
 static bool ProcessEnvironmentOption(const char* arg) {
   ASSERT(arg != NULL);
   if (*arg == '\0') {
     // Ignore empty -D option.
     Log::PrintErr("No arguments given to -D option\n");
     return true;
   }
   if (environment == NULL) {
     environment = new HashMap(&HashMap::SameStringValue, 4);
   }
@@ skipping 27 matching lines @@
 
 static bool ProcessCompileAllOption(const char* arg) {
   ASSERT(arg != NULL);
   if (*arg != '\0') {
     return false;
   }
   has_compile_all = true;
   return true;
 }
 
-
-static bool ProcessDebugOption(const char* port) {
-  // TODO(hausner): Add support for specifying an IP address on which
-  // the debugger should listen.
-  ASSERT(port != NULL);
-  debug_port = -1;
-  if (*port == '\0') {
-    debug_port = DEFAULT_DEBUG_PORT;
-  } else {
-    if ((*port == '=') || (*port == ':')) {
-      debug_port = atoi(port + 1);
-    }
-  }
-  if (debug_port < 0) {
+static bool ProcessDebugOption(const char* option_value) {
+  ASSERT(option_value != NULL);
+  if (!ExtractPortAndIP(option_value, &debug_port, &debug_ip,
+                        DEFAULT_DEBUG_PORT, DEFAULT_DEBUG_IP)) {
     Log::PrintErr("unrecognized --debug option syntax. "
-                    "Use --debug[:<port number>]\n");
+                  "Use --debug[:<port number>[/<IPv4 address>]]\n");
     return false;
   }
+
   breakpoint_at = "main";
   start_debugger = true;
   return true;
 }
 
 
 static bool ProcessPrintScriptOption(const char* arg) {
   ASSERT(arg != NULL);
   if (*arg != '\0') {
     return false;
@@ skipping 17 matching lines @@
                     filename);
       return false;
     }
     generate_script_snapshot = true;
     return true;
   }
   return false;
 }
 
 
-static bool ProcessEnableVmServiceOption(const char* port) {
-  ASSERT(port != NULL);
-  vm_service_server_port = -1;
-  if (*port == '\0') {
-    vm_service_server_port = DEFAULT_VM_SERVICE_SERVER_PORT;
-  } else {
-    if ((*port == '=') || (*port == ':')) {
-      vm_service_server_port = atoi(port + 1);
-    }
-  }
-  if (vm_service_server_port < 0) {
+static bool ProcessEnableVmServiceOption(const char* option_value) {
+  ASSERT(option_value != NULL);
+
+  if (!ExtractPortAndIP(option_value,
+                        &vm_service_server_port,
+                        &vm_service_server_ip,
+                        DEFAULT_VM_SERVICE_SERVER_PORT,
+                        DEFAULT_VM_SERVICE_SERVER_IP)) {
     Log::PrintErr("unrecognized --enable-vm-service option syntax. "
-                    "Use --enable-vm-service[:<port number>]\n");
+                  "Use --enable-vm-service[:<port number>[/<IPv4 address>]]\n");
     return false;
   }
+
   start_vm_service = true;
   return true;
 }
 
 bool trace_debug_protocol = false;
 static bool ProcessTraceDebugProtocolOption(const char* arg) {
   if (*arg != '\0') {
     return false;
   }
   trace_debug_protocol = true;
@@ skipping 671 matching lines @@
       Log::Print("Debugger listening on port %d\n", debug_port);
     }
   }
 
   // Start event handler.
   EventHandler::Start();
 
   ASSERT(Dart_CurrentIsolate() == NULL);
   // Start the VM service isolate, if necessary.
   if (start_vm_service) {
-    ASSERT(vm_service_server_port >= 0);
-    bool r = VmService::Start(vm_service_server_port);
+    ASSERT(vm_service_server_ip != NULL && vm_service_server_port >= 0);
+    bool r = VmService::Start(vm_service_server_ip , vm_service_server_port);
     if (!r) {
       Log::PrintErr("Could not start VM Service isolate %s\n",
                     VmService::GetErrorMessage());
     }
     Dart_RegisterIsolateServiceRequestCallback(
         "io", &ServiceRequestHandler, NULL);
   }
   ASSERT(Dart_CurrentIsolate() == NULL);
 
   // Call CreateIsolateAndSetup which creates an isolate and loads up
@@ skipping 138 matching lines @@
   exit(Process::GlobalExitCode());
 }
 
 }  // namespace bin
 }  // namespace dart
 
 int main(int argc, char** argv) {
   dart::bin::main(argc, argv);
   UNREACHABLE();
 }