From 336a71f1623fe1721e8d39f409eb3c36ad590fef Mon Sep 17 00:00:00 2001
From: Bob Mottram <bob@freedombone.net>
Date: Sun, 12 Sep 2021 13:18:27 +0100
Subject: [PATCH] Get referer domain

---
 daemon.py | 23 +++++++++++++++++------
 1 file changed, 17 insertions(+), 6 deletions(-)

diff --git a/daemon.py b/daemon.py
index 254eaebd1..a450d225a 100644
--- a/daemon.py
+++ b/daemon.py
@@ -6647,7 +6647,7 @@ class PubServer(BaseHTTPRequestHandler):
             return
         self._404()
 
-    def _showCachedAvatar(self, callingDomain: str, path: str,
+    def _showCachedAvatar(self, refererDomain: str, path: str,
                           baseDir: str,
                           GETstartTime, GETtimings: {}) -> None:
         """Shows an avatar image obtained from the cache
@@ -6664,7 +6664,7 @@ class PubServer(BaseHTTPRequestHandler):
                 self._set_headers_etag(mediaFilename,
                                        mimeType,
                                        mediaBinary, None,
-                                       self.server.domainFull,
+                                       refererDomain,
                                        False, None)
                 self._write(mediaBinary)
                 self._benchmarkGETtimings(GETstartTime, GETtimings,
@@ -11205,7 +11205,7 @@ class PubServer(BaseHTTPRequestHandler):
                                   'share files shown')
         return True
 
-    def _showAvatarOrBanner(self, callingDomain: str, path: str,
+    def _showAvatarOrBanner(self, refererDomain: str, path: str,
                             baseDir: str, domain: str,
                             GETstartTime, GETtimings: {}) -> bool:
         """Shows an avatar or banner or profile background image
@@ -11259,7 +11259,7 @@ class PubServer(BaseHTTPRequestHandler):
             mediaBinary = avFile.read()
             self._set_headers_etag(avatarFilename, mediaImageType,
                                    mediaBinary, None,
-                                   None, True,
+                                   refererDomain, True,
                                    lastModifiedTimeStr)
             self._write(mediaBinary)
         self._benchmarkGETtimings(GETstartTime, GETtimings,
@@ -11627,6 +11627,17 @@ class PubServer(BaseHTTPRequestHandler):
                     self._400()
                     return
 
+        # which domain is the GET request coming from?
+        refererDomain = None
+        if self.headers.get('referer'):
+            refererDomain, refererPort = \
+                getDomainFromActor(self.headers['referer'])
+            refererDomain = getFullDomain(refererDomain, refererPort)
+        elif self.headers.get('Referer'):
+            refererDomain, refererPort = \
+                getDomainFromActor(self.headers['Referer'])
+            refererDomain = getFullDomain(refererDomain, refererPort)
+
         if self._blockedUserAgent(callingDomain):
             self._400()
             return
@@ -13027,7 +13038,7 @@ class PubServer(BaseHTTPRequestHandler):
         # cached avatar images
         # Note that this comes before the busy flag to avoid conflicts
         if self.path.startswith('/avatars/'):
-            self._showCachedAvatar(self.server.domainFull, self.path,
+            self._showCachedAvatar(refererDomain, self.path,
                                    self.server.baseDir,
                                    GETstartTime, GETtimings)
             return
@@ -13038,7 +13049,7 @@ class PubServer(BaseHTTPRequestHandler):
 
         # show avatar or background image
         # Note that this comes before the busy flag to avoid conflicts
-        if self._showAvatarOrBanner(callingDomain, self.path,
+        if self._showAvatarOrBanner(refererDomain, self.path,
                                     self.server.baseDir,
                                     self.server.domain,
                                     GETstartTime, GETtimings):