NoRefer/background.js at master · c475/NoRefer · GitHub

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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
var getTokens = function (content) {
    var ret = {
        regex: {},
        normal: {}
    };

    content = content.split('\n').filter(function (e) {
        return !!e;
    });

    for (var i = 0; i < content.length; i++) {
        var key;
        var value = null;
        var regex = null;

        if (content[i].indexOf(':') !== -1) {
            key = content[i].split(':')[0];
            value = content[i].substr(key.length + 1);
            if (value[0] === ' ') {
                value = content[i].substr(key.length + 2);
            }
        } else {
            key = content[i];
        }

        if (key[0] === '#') {
            key = key.substr(1);
            regex = new RegExp(key);
            ret.regex[key] = {
                val: value,
                regex: regex
            };
        } else {
            ret.normal[key] = {
                val: value,
            };
        }

    }

    return ret;
};

chrome.webRequest.onBeforeSendHeaders.addListener(function (details) {
    var filter = localStorage.getItem('filter');

    if (!filter) {
        return;
    }

    var tokens = getTokens(filter);
    var newHeaders = [];

    if (Object.keys(tokens.regex).length) {
        for (var i = 0; i < details.requestHeaders.length; i++) {
            var h = {};
            var match = false;

            for (var key in tokens.regex) {
                if (tokens.regex[key].regex.test(details.requestHeaders[i].name)) {
                    if (tokens.regex[key].val) {
                        h.name = details.requestHeaders[i].name;
                        h.value = tokens.regex[key].val;
                        details.requestHeaders[i].ignore = true;
                        newHeaders.push(h);
                    }

                    match = true;
                    break;
                }
            }
        }
    }

    for (var i = 0; i < details.requestHeaders.length; i++) {
        var h = {};
        var headerMatch = tokens.normal[details.requestHeaders[i].name];

        if (details.requestHeaders[i].ignore) {
            continue;
        }

        if (headerMatch) {
            if (headerMatch.val) {
                h.name = details.requestHeaders[i].name;
                h.value = headerMatch.val
                newHeaders.push(h);
            }

            delete tokens.normal[details.requestHeaders[i].name];

        } else {
            newHeaders.push(details.requestHeaders[i]);
        }
    }

    for (var key in tokens.normal) {
        if (tokens.normal[key].val !== null) {
            newHeaders.push({
                name: key,
                value: tokens.normal[key].val
            });
        }
    }

    return {
        requestHeaders: newHeaders
    };

}, {

    urls: [
        '<all_urls>'
    ],

}, [

    'blocking',
    'requestHeaders',
    'extraHeaders'

]);