From 50ef39af82413ef463609f24173b22af13fad268 Mon Sep 17 00:00:00 2001
From: Nathan Binkert <nate@binkert.org>
Date: Sun, 3 Aug 2008 18:19:55 -0700
Subject: sockets: Add a function to disable all listening sockets. When
 invoking several copies of m5 on the same machine at the same time, there can
 be a race for TCP ports for the terminal connections or remote gdb.  Expose a
 function to disable those ports, and have the regression scripts disable
 them.  There are some SimObjects that have no other function than to be used
 with ports (NativeTrace and EtherTap), so they will panic if the ports are
 disabled.

---
 src/base/remote_gdb.cc |  5 +++++
 src/base/socket.cc     | 18 ++++++++++++++++++
 src/base/socket.hh     |  8 ++++++++
 3 files changed, 31 insertions(+)

(limited to 'src/base')

diff --git a/src/base/remote_gdb.cc b/src/base/remote_gdb.cc
index 51293e118..06e04e19e 100644
--- a/src/base/remote_gdb.cc
+++ b/src/base/remote_gdb.cc
@@ -195,6 +195,11 @@ GDBListener::name()
 void
 GDBListener::listen()
 {
+    if (ListenSocket::allDisabled()) {
+        warn_once("Sockets disabled, not accepting gdb connections");
+        return;
+    }
+
     while (!listener.listen(port, true)) {
         DPRINTF(GDBMisc, "Can't bind port %d\n", port);
         port++;
diff --git a/src/base/socket.cc b/src/base/socket.cc
index adcc48735..bcc5236b0 100644
--- a/src/base/socket.cc
+++ b/src/base/socket.cc
@@ -43,6 +43,23 @@
 
 using namespace std;
 
+bool ListenSocket::listeningDisabled = false;
+bool ListenSocket::anyListening = false;
+
+void
+ListenSocket::disableAll()
+{
+    if (anyListening)
+        panic("Too late to disable all listeners, already have a listener");
+    listeningDisabled = true;
+}
+
+bool
+ListenSocket::allDisabled()
+{
+    return listeningDisabled;
+}
+
 ////////////////////////////////////////////////////////////////////////
 //
 //
@@ -92,6 +109,7 @@ ListenSocket::listen(int port, bool reuse)
 
     listening = true;
 
+    anyListening = true;
     return true;
 }
 
diff --git a/src/base/socket.hh b/src/base/socket.hh
index 8e55eae72..942318e45 100644
--- a/src/base/socket.hh
+++ b/src/base/socket.hh
@@ -33,6 +33,14 @@
 
 class ListenSocket
 {
+  protected:
+    static bool listeningDisabled;
+    static bool anyListening;
+
+  public:
+    static void disableAll();
+    static bool allDisabled();
+
   protected:
     bool listening;
     int fd;
-- 
cgit v1.2.3