From 6b1cc4ae9776e4b755ad353652466616a74c1ba2 Mon Sep 17 00:00:00 2001
From: Thomas Rientjes <synzvato@protonmail.com>
Date: Sat, 5 Dec 2015 19:26:16 +0100
Subject: [PATCH] Add support for direct requests

---
 lib/request-analyzer.js | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/lib/request-analyzer.js b/lib/request-analyzer.js
index b01e0ac..15616e7 100644
--- a/lib/request-analyzer.js
+++ b/lib/request-analyzer.js
@@ -36,14 +36,17 @@ function isValidCandidate(httpChannel) {
     //noinspection JSUnresolvedVariable
     var whitelistedDomains = require('sdk/simple-prefs').prefs.domainWhitelist.split(";");
 
-    for (var domain in whitelistedDomains) {
+    if (httpChannel.referrer) {
 
-        if (whitelistedDomains[domain] === httpChannel.referrer.host) {
+        for (var domain in whitelistedDomains) {
 
-            // Remove referer header from request.
-            httpChannel.setRequestHeader('Referer', null, false);
+            if (whitelistedDomains[domain] === httpChannel.referrer.host) {
 
-            return false;
+                // Remove referer header from request.
+                httpChannel.setRequestHeader('Referer', null, false);
+
+                return false;
+            }
         }
     }
 
-- 
GitLab