1 files changed, 164 insertions, 0 deletions
diff --git a/netwerk/test/TestServ.js b/netwerk/test/TestServ.js
new file mode 100644
index 000000000..7716ac504
--- /dev/null
+++ b/netwerk/test/TestServ.js
@@ -0,0 +1,164 @@
+/* vim:set ts=2 sw=2 et: */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+/*
+ * To use try out this JS server socket implementation, just copy this file
+ * into the "components" directory of a Mozilla build.  Then load the URL
+ * http://localhost:4444/ in the browser.  You should see a page get loaded
+ * that was served up by this component :-)
+ *
+ * This code requires Mozilla 1.6 or better.
+ */
+
+const kTESTSERV_CONTRACTID = "@mozilla.org/network/test-serv;1";
+const kTESTSERV_CID = Components.ID("{a741fcd5-9695-42e8-a7f7-14f9a29f8200}");
+const nsISupports = Components.interfaces.nsISupports;
+const nsIObserver = Components.interfaces.nsIObserver;
+const nsIServerSocket = Components.interfaces.nsIServerSocket;
+const nsIServerSocketListener = Components.interfaces.nsIServerSocketListener;
+const nsITransport = Components.interfaces.nsITransport;
+const nsIScriptableInputStream = Components.interfaces.nsIScriptableInputStream;
+
+/** we'll listen on this port for HTTP requests **/
+const kPORT = 4444;
+
+function nsTestServ() { dump(">>> creating nsTestServ instance\n"); };
+
+nsTestServ.prototype =
+{
+  QueryInterface: function(iid)
+  {
+    if (iid.equals(nsIObserver) ||
+        iid.equals(nsIServerSocketListener) ||
+        iid.equals(nsISupports))
+      return this;
+
+    throw Components.results.NS_ERROR_NO_INTERFACE;
+  },
+
+  observe: function(subject, topic, data)
+  {
+    dump(">>> observe [" + topic + "]\n");
+    this.startListening();
+  },
+
+  /* this function is called when we receive a new connection */
+  onSocketAccepted: function(serverSocket, clientSocket)
+  {
+    dump(">>> accepted connection on "+clientSocket.host+":"+clientSocket.port+"\n");
+
+    var input = clientSocket.openInputStream(nsITransport.OPEN_BLOCKING, 0, 0);
+    var output = clientSocket.openOutputStream(nsITransport.OPEN_BLOCKING, 0, 0);
+
+    this.consumeInput(input);
+
+    const fixedResponse =
+      "HTTP/1.0 200 OK\r\nContent-Type: text/plain\r\n\r\nFooooopy!!\r\n";
+    var response = fixedResponse + "\r\n" + new Date().toString() + "\r\n";
+    var n = output.write(response, response.length);
+    dump(">>> wrote "+n+" bytes\n");
+
+    input.close();
+    output.close();
+  },
+
+  onStopListening: function(serverSocket, status)
+  {
+    dump(">>> shutting down server socket\n");
+  },
+
+  startListening: function()
+  {
+    const SERVERSOCKET_CONTRACTID = "@mozilla.org/network/server-socket;1";
+    var socket = Components.classes[SERVERSOCKET_CONTRACTID].createInstance(nsIServerSocket);
+    socket.init(kPORT, true /* loopback only */, 5);
+    dump(">>> listening on port "+socket.port+"\n");
+    socket.asyncListen(this);
+  },
+
+  consumeInput: function(input)
+  {
+    /* use nsIScriptableInputStream to consume all of the data on the stream */
+
+    var sin = Components.classes["@mozilla.org/scriptableinputstream;1"]
+                        .createInstance(nsIScriptableInputStream);
+    sin.init(input);
+
+    /* discard all data */
+    while (sin.available() > 0)
+      sin.read(512);
+  }
+}
+
+/**
+ * JS XPCOM component registration goop:
+ *
+ * We set ourselves up to observe the xpcom-startup category.  This provides
+ * us with a starting point.
+ */
+
+var servModule = new Object();
+
+servModule.registerSelf =
+function (compMgr, fileSpec, location, type)
+{
+  compMgr = compMgr.QueryInterface(Components.interfaces.nsIComponentRegistrar);
+  compMgr.registerFactoryLocation(kTESTSERV_CID,
+                                  "nsTestServ",
+                                  kTESTSERV_CONTRACTID,
+                                  fileSpec,
+                                  location,
+                                  type);
+
+  const CATMAN_CONTRACTID = "@mozilla.org/categorymanager;1";
+  const nsICategoryManager = Components.interfaces.nsICategoryManager;
+  var catman = Components.classes[CATMAN_CONTRACTID].getService(nsICategoryManager);
+  catman.addCategoryEntry("xpcom-startup",
+                          "TestServ",
+                          kTESTSERV_CONTRACTID,
+                          true,
+                          true);
+}
+
+servModule.getClassObject =
+function (compMgr, cid, iid)
+{
+  if (!cid.equals(kTESTSERV_CID))
+    throw Components.results.NS_ERROR_NO_INTERFACE;
+
+  if (!iid.equals(Components.interfaces.nsIFactory))
+    throw Components.results.NS_ERROR_NOT_IMPLEMENTED;
+
+  return servFactory;
+}
+
+servModule.canUnload =
+function (compMgr)
+{
+  dump(">>> unloading test serv.\n");
+  return true;
+}
+
+var servFactory = new Object();
+
+servFactory.createInstance =
+function (outer, iid)
+{
+  if (outer != null)
+    throw Components.results.NS_ERROR_NO_AGGREGATION;
+
+  if (!iid.equals(nsIObserver) &&
+      !iid.equals(nsISupports))
+    throw Components.results.NS_ERROR_NO_INTERFACE;
+
+  return TestServ;
+}
+
+function NSGetModule(compMgr, fileSpec)
+{
+  return servModule;
+}
+
+var TestServ = new nsTestServ();