The Tor Browser / file comparison
comparison: dom/tests/mochitest/beacon/beacon-handler.sjs
dom/tests/mochitest/beacon/beacon-handler.sjs
- changeset 0
- 6474c204b198
equal
deleted
inserted
replaced
| -1:000000000000 | 0:268df037edc0 |
|---|---|
| 1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ | |
| 2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */ | |
| 3 /* This Source Code Form is subject to the terms of the Mozilla Public | |
| 4 * License, v. 2.0. If a copy of the MPL was not distributed with this | |
| 5 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ | |
| 6 | |
| 7 const CC = Components.Constructor; | |
| 8 const BinaryInputStream = CC("@mozilla.org/binaryinputstream;1", | |
| 9 "nsIBinaryInputStream", | |
| 10 "setInputStream"); | |
| 11 | |
| 12 function DEBUG(str) | |
| 13 { | |
| 14 // dump("********** " + str + "\n"); | |
| 15 } | |
| 16 | |
| 17 function setOurState(data) { | |
| 18 x = { data: data, QueryInterface: function(iid) { return this } }; | |
| 19 x.wrappedJSObject = x; | |
| 20 setObjectState("beacon-handler", x); | |
| 21 DEBUG("our state is " + data); | |
| 22 } | |
| 23 | |
| 24 function getOurState() { | |
| 25 var data; | |
| 26 getObjectState("beacon-handler", function(x) { | |
| 27 // x can be null if no one has set any state yet | |
| 28 if (x) { | |
| 29 data = x.wrappedJSObject.data; | |
| 30 } | |
| 31 }); | |
| 32 return data; | |
| 33 } | |
| 34 | |
| 35 function handleRequest(request, response) { | |
| 36 DEBUG("Entered request handler"); | |
| 37 response.setHeader("Cache-Control", "no-cache", false); | |
| 38 | |
| 39 function finishControlResponse(response) { | |
| 40 DEBUG("********* sending out the control GET response"); | |
| 41 var data = getState("beaconData"); | |
| 42 var mimetype = getState("beaconMimetype"); | |
| 43 DEBUG("GET was sending : " + data + "\n"); | |
| 44 DEBUG("GET was sending : " + mimetype + "\n"); | |
| 45 var result = { | |
| 46 "data": data, | |
| 47 "mimetype": mimetype, | |
| 48 }; | |
| 49 response.write(JSON.stringify(result)); | |
| 50 setOurState(null); | |
| 51 } | |
| 52 | |
| 53 if (request.method == "GET") { | |
| 54 DEBUG(" ------------ GET --------------- "); | |
| 55 response.setHeader("Content-Type", "application/json", false); | |
| 56 switch (request.queryString) { | |
| 57 case "getLastBeacon": | |
| 58 var state = getOurState(); | |
| 59 if (state === "unblocked") { | |
| 60 finishControlResponse(response); | |
| 61 } else { | |
| 62 DEBUG("GET has arrived, but POST has not, blocking response!"); | |
| 63 setOurState(response); | |
| 64 response.processAsync(); | |
| 65 } | |
| 66 break; | |
| 67 default: | |
| 68 response.setStatusLine(request.httpVersion, 400, "Bad Request"); | |
| 69 break; | |
| 70 } | |
| 71 return; | |
| 72 } | |
| 73 | |
| 74 if (request.method == "POST") { | |
| 75 DEBUG(" ------------ POST --------------- "); | |
| 76 var body = new BinaryInputStream(request.bodyInputStream); | |
| 77 var avail; | |
| 78 var bytes = []; | |
| 79 | |
| 80 while ((avail = body.available()) > 0) { | |
| 81 Array.prototype.push.apply(bytes, body.readByteArray(avail)); | |
| 82 } | |
| 83 | |
| 84 var data = ""; | |
| 85 for (var i=0; i < bytes.length; i++) { | |
| 86 // We are only passing strings at this point. | |
| 87 if (bytes[i] < 32) continue; | |
| 88 var charcode = String.fromCharCode(bytes[i]); | |
| 89 data += charcode; | |
| 90 } | |
| 91 | |
| 92 var mimetype = request.getHeader("Content-Type"); | |
| 93 | |
| 94 // check to see if this is form data. | |
| 95 if (mimetype.indexOf("multipart/form-data") != -1) { | |
| 96 | |
| 97 // trim the mime type to make testing easier. | |
| 98 mimetype = "multipart/form-data"; | |
| 99 // Extract only the form-data name. | |
| 100 | |
| 101 var pattern = /; name=\"(.+)\";/; | |
| 102 data = data.split(pattern)[1]; | |
| 103 } | |
| 104 | |
| 105 DEBUG("********** POST was sending : " + data + "\n"); | |
| 106 DEBUG("********** POST was sending : " + mimetype + "\n"); | |
| 107 setState("beaconData", data); | |
| 108 setState("beaconMimetype", mimetype); | |
| 109 | |
| 110 response.setHeader("Content-Type", "text/plain", false); | |
| 111 response.write('ok'); | |
| 112 | |
| 113 var blockedResponse = getOurState(); | |
| 114 if (typeof(blockedResponse) == "object" && blockedResponse) { | |
| 115 DEBUG("GET is already pending, finishing!"); | |
| 116 finishControlResponse(blockedResponse); | |
| 117 blockedResponse.finish(); | |
| 118 } else { | |
| 119 DEBUG("GET has not arrived, marking it as unblocked"); | |
| 120 setOurState("unblocked"); | |
| 121 } | |
| 122 | |
| 123 return; | |
| 124 } | |
| 125 | |
| 126 response.setStatusLine(request.httpVersion, 402, "Bad Request"); | |
| 127 } |
