From 458cc20cebae8ebe7d614993d3e579e4164ffe4b Mon Sep 17 00:00:00 2001
From: Oran Nadler <orann@guardicore.com>
Date: Wed, 7 Mar 2018 00:03:55 -0800
Subject: [PATCH] small fixes

---
 monkey_island/cc/resources/pthmap.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/monkey_island/cc/resources/pthmap.py b/monkey_island/cc/resources/pthmap.py
index 3e3b78549..3ad75d366 100644
--- a/monkey_island/cc/resources/pthmap.py
+++ b/monkey_island/cc/resources/pthmap.py
@@ -466,7 +466,10 @@ class PassTheHashMap(object):
         SIDs = set()
         
         for m in self.machines:
-            SIDs.add(m.GetSidBySecret(secret))
+            sid = m.GetSidBySecret(secret)
+            
+            if sid:
+                SIDs.add(sid)
         
         return SIDs
     
@@ -650,7 +653,7 @@ def main():
         print """<h3>SIDs that use that secret</h3>"""
         print """<ul>"""
         for sid in pth.GetSidsBySecret(secret):
-            print """<li><a href="#{sid}">{username} ({sid})</a></li>""".format(username=m.GetUsernameBySid(sid), sid=sid)
+            print """<li><a href="#{sid}">{username} ({sid})</a></li>""".format(username=pth.GetUsernameBySid(sid), sid=sid)
         print """</ul>"""
         
         print """<h3>Attackable Machines with that secret</h3>"""