enhance WebServer's decission if AppleTV is real...

iBaa · Jun 12, 2013 · 96f02fa · 96f02fa
1 parent 35e16bf
commit 96f02fa
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/WebServer.py b/WebServer.py
@@ -69,8 +69,10 @@ def do_GET(self):
             dprint(__name__, 2, "cleaned path:\n{0}", self.path)
             dprint(__name__, 2, "request options:\n{0}", options)
 
-            if self.headers['Host'] == g_param['HostToIntercept'] and \
-               self.headers['User-Agent'].startswith("iTunes-AppleTV"):
+            if 'Host' in self.headers and \
+               'User-Agent' in self.headers and \
+               self.headers['Host'] == g_param['HostToIntercept'] and \
+               'AppleTV' in self.headers['User-Agent']:
 
                 # recieve simple logging messages from the ATV
                 if self.path.endswith("&atvlogger"):