summaryrefslogtreecommitdiffstats
path: root/toolkit/components/url-classifier/tests/mochitest/threathit.sjs
blob: a083a2e24775544cbbc1da6f51c3dd0696643fa7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
/* Any copyright is dedicated to the Public Domain.
 * http://creativecommons.org/publicdomain/zero/1.0/ */

const CC = Components.Constructor;
const BinaryInputStream = CC(
  "@mozilla.org/binaryinputstream;1",
  "nsIBinaryInputStream",
  "setInputStream"
);

function handleRequest(request, response) {
  let params = new URLSearchParams(request.queryString);
  var action = params.get("action");

  var responseBody;

  // Store data in the server side.
  if (action == "store") {
    // In the server side we will store:
    // All the full hashes or update for a given list
    let state = params.get("list") + params.get("type");
    let dataStr = params.get("data");
    setState(state, dataStr);
  } else if (action == "get") {
    let state = params.get("list") + params.get("type");
    responseBody = atob(getState(state));
    response.setStatusLine(request.httpVersion, 200, "OK");
    response.bodyOutputStream.write(responseBody, responseBody.length);
  } else if (action == "report") {
    let state = params.get("list") + "report";
    let requestUrl =
      request.scheme +
      "://" +
      request.host +
      ":" +
      request.port +
      request.path +
      "?" +
      request.queryString;
    setState(state, requestUrl);
  } else if (action == "getreport") {
    let state = params.get("list") + "report";
    responseBody = getState(state);
    response.setHeader("Content-Type", "text/plain", false);
    response.write(responseBody);
  }
}