1 files changed, 137 insertions, 0 deletions
diff --git a/dom/tests/mochitest/beacon/beacon-handler.sjs b/dom/tests/mochitest/beacon/beacon-handler.sjs
new file mode 100644
index 0000000000..6a1beff303
--- /dev/null
+++ b/dom/tests/mochitest/beacon/beacon-handler.sjs
@@ -0,0 +1,137 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set ts=8 sts=2 et sw=2 tw=80: */
+/* 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/. */
+
+const CC = Components.Constructor;
+const BinaryInputStream = CC("@mozilla.org/binaryinputstream;1",
+                             "nsIBinaryInputStream",
+                             "setInputStream");
+
+function DEBUG(str)
+{
+  // dump("********** " + str + "\n");
+}
+
+function setOurState(data) {
+  x = { data: data, QueryInterface: function(iid) { return this } };
+  x.wrappedJSObject = x;
+  setObjectState("beacon-handler", x);
+  DEBUG("our state is " + data);
+}
+
+function getOurState() {
+  var data;
+  getObjectState("beacon-handler", function(x) {
+    // x can be null if no one has set any state yet
+    if (x) {
+      data = x.wrappedJSObject.data;
+    }
+  });
+  return data;
+}
+
+function handleRequest(request, response) {
+  DEBUG("Entered request handler");
+  response.setHeader("Cache-Control", "no-cache", false);
+
+  function finishControlResponse(response) {
+    DEBUG("********* sending out the control GET response");
+    var data = getState("beaconData");
+    var mimetype = getState("beaconMimetype");
+    DEBUG("GET was sending : " + data + "\n");
+    DEBUG("GET was sending : " + mimetype + "\n");
+    var result = {
+      "data": data,
+      "mimetype": mimetype,
+    };
+    response.write(JSON.stringify(result));
+    setOurState(null);
+  }
+
+  if (request.method == "GET") {
+    DEBUG(" ------------ GET --------------- ");
+    response.setHeader("Content-Type", "application/json", false);
+    switch (request.queryString) {
+    case "getLastBeaconCors":
+      // Allow CORS responses of the last beacon
+      var originHeader = request.getHeader("origin");
+      response.setHeader("Access-Control-Allow-Headers", "content-type", false);
+      response.setHeader("Access-Control-Allow-Methods", "POST, GET", false);
+      response.setHeader("Access-Control-Allow-Origin", originHeader, false);
+      response.setHeader("Access-Control-Allow-Credentials", "true", false);
+    case "getLastBeacon":
+      var state = getOurState();
+      if (state === "unblocked") {
+        finishControlResponse(response);
+      } else {
+        DEBUG("GET has  arrived, but POST has not, blocking response!");
+        setOurState(response);
+        response.processAsync();
+      }
+      break;
+    default:
+      response.setStatusLine(request.httpVersion, 400, "Bad Request");
+      break;
+    }
+    return;
+  }
+
+  if (request.method == "POST") {
+    DEBUG(" ------------ POST --------------- ");
+    var body = new BinaryInputStream(request.bodyInputStream);
+    var avail;
+    var bytes = [];
+
+    while ((avail = body.available()) > 0) {
+      Array.prototype.push.apply(bytes, body.readByteArray(avail));
+    }
+
+    var data = "";
+    for (var i=0; i < bytes.length; i++) {
+      // We are only passing strings at this point.
+      if (bytes[i] < 32) continue;
+      var charcode = String.fromCharCode(bytes[i]);
+      data += charcode;
+    }
+
+    var mimetype = "";
+    if (request.hasHeader("Content-Type")) {
+      mimetype = request.getHeader("Content-Type");
+    }
+
+    // check to see if this is form data.
+    if (mimetype.indexOf("multipart/form-data") != -1) {
+
+      // trim the mime type to make testing easier.
+      mimetype = "multipart/form-data";
+      // Extract only the form-data name.
+
+      var pattern = /; name=\"(.+)\";/;
+      data = data.split(pattern)[1];
+    }
+
+    DEBUG("**********   POST was sending : " + data + "\n");
+    DEBUG("**********   POST was sending : " + mimetype + "\n");
+    setState("beaconData", data);
+    setState("beaconMimetype", mimetype);
+
+    response.setHeader("Content-Type", "text/plain", false);
+    response.write('ok');
+
+    var blockedResponse = getOurState();
+    if (typeof(blockedResponse) == "object" && blockedResponse) {
+      DEBUG("GET is already pending, finishing!");
+      finishControlResponse(blockedResponse);
+      blockedResponse.finish();
+    } else {
+      DEBUG("GET has not arrived, marking it as unblocked");
+      setOurState("unblocked");
+    }
+
+    return;
+  }
+
+  response.setStatusLine(request.httpVersion, 402, "Bad Request");
+}