summaryrefslogtreecommitdiffstats
path: root/toolkit/components/url-classifier/tests/mochitest/update.sjs
blob: f3984b2a6f9bafe943d1df8c360877326640c820 (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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
const CC = Components.Constructor;
const BinaryInputStream = CC(
  "@mozilla.org/binaryinputstream;1",
  "nsIBinaryInputStream",
  "setInputStream"
);

function handleRequest(request, response) {
  var query = {};
  request.queryString.split("&").forEach(function (val) {
    var idx = val.indexOf("=");
    query[val.slice(0, idx)] = unescape(val.slice(idx + 1));
  });

  // Store fullhash in the server side.
  if ("list" in query && "fullhash" in query) {
    // In the server side we will store:
    // 1. All the full hashes for a given list
    // 2. All the lists we have right now
    // data is separate by '\n'
    let list = query.list;
    let hashes = getState(list);

    let hash = atob(query.fullhash);
    hashes += hash + "\n";
    setState(list, hashes);

    let lists = getState("lists");
    if (!lists.includes(list)) {
      lists += list + "\n";
      setState("lists", lists);
    }

    return;
  }

  var body = new BinaryInputStream(request.bodyInputStream);
  var avail;
  var bytes = [];

  while ((avail = body.available()) > 0) {
    Array.prototype.push.apply(bytes, body.readByteArray(avail));
  }

  var responseBody = parseV2Request(bytes);

  response.setHeader("Content-Type", "text/plain", false);
  response.write(responseBody);
}

function parseV2Request(bytes) {
  var table = String.fromCharCode.apply(this, bytes).slice(0, -2);

  var ret = "";
  getState("lists")
    .split("\n")
    .forEach(function (list) {
      if (list == table) {
        var completions = getState(list).split("\n");
        ret += "n:1000\n";
        ret += "i:" + list + "\n";
        ret += "a:1:32:" + 32 * (completions.length - 1) + "\n";

        for (var completion of completions) {
          ret += completion;
        }
      }
    });

  return ret;
}