aboutsummaryrefslogtreecommitdiffstats
path: root/lib/HttpRequestHeaders.jsm
blob: 062400bbce92d05894936992c0206f1d17762196 (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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
/*******************************************************************************

    ηMatrix - a browser extension to black/white list requests.
    Copyright (C) 2014-2019 The uMatrix/uBlock Origin authors
    Copyright (C) 2019-2020 Alessio Vanni

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see {http://www.gnu.org/licenses/}.

    Home: https://libregit.org/heckyel/ematrix
    uMatrix Home: https://github.com/gorhill/uMatrix
*/

'use strict';

var EXPORTED_SYMBOLS = ['HTTPRequestHeaders'];

var junkyard = [];

var HTTPRequestHeaders = function (channel) {
    this.init(channel);
};

HTTPRequestHeaders.factory = function (channel) {
    let entry = junkyard.pop();
    if (entry) {
        return entry.init(channel);
    }

    return new HTTPRequestHeaders(channel);
}

HTTPRequestHeaders.prototype.init = function (channel) {
    this.channel = channel;
    this.headers = new Array();
    this.originalHeaderNames = new Array();

    channel.visitRequestHeaders({
        visitHeader: function (name, value) {
            this.headers.push({name: name, value: value});
            this.originalHeaderNames.push(name);
        }.bind(this)
    });

    return this;
};

HTTPRequestHeaders.prototype.dispose = function () {
    this.channel = null;
    this.headers = null;
    this.originalHeaderNames = null;
    junkyard.push(this);
};

HTTPRequestHeaders.prototype.update = function () {
    let newHeaderNames = new Set();
    for (let header of this.headers) {
        this.setHeader(header.name, header.value, true);
        newHeaderNames.add(header.name);
    }

    //Clear any headers that were removed
    for (let name of this.originalHeaderNames) {
        if (!newHeaderNames.has(name)) {
            this.channel.setRequestHeader(name, '', false);
        }
    }
};

HTTPRequestHeaders.prototype.getHeader = function (name) {
    try {
        return this.channel.getRequestHeader(name);
    } catch (e) {
        // Ignore
    }

    return '';
};

HTTPRequestHeaders.prototype.setHeader = function (name, newValue, create) {
    let oldValue = this.getHeader(name);
    if (newValue === oldValue) {
        return false;
    }

    if (oldValue === '' && create !== true) {
        return false;
    }

    this.channel.setRequestHeader(name, newValue, false);
    return true;
};