From f4a47f3cb32f1a0cddd1d9c35c636ff32fba286e Mon Sep 17 00:00:00 2001
From: VakarisZ <vakarisz@yahoo.com>
Date: Mon, 27 May 2019 09:49:40 +0300
Subject: [PATCH] Request exception handling

---
 .../post_breach/actions/users_custom_pba.py          | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/monkey/infection_monkey/post_breach/actions/users_custom_pba.py b/monkey/infection_monkey/post_breach/actions/users_custom_pba.py
index f8ec48c7e..f92035a57 100644
--- a/monkey/infection_monkey/post_breach/actions/users_custom_pba.py
+++ b/monkey/infection_monkey/post_breach/actions/users_custom_pba.py
@@ -41,10 +41,14 @@ class UsersPBA(PBA):
         :return: True if successful, false otherwise
         """
 
-        pba_file_contents = requests.get("https://%s/api/pba/download/%s" %
-                                         (WormConfiguration.current_server, filename),
-                                         verify=False,
-                                         proxies=ControlClient.proxies)
+        try:
+            pba_file_contents = requests.get("https://%s/api/pba/download/%s" %
+                                             (WormConfiguration.current_server, filename),
+                                             verify=False,
+                                             proxies=ControlClient.proxies)
+        except requests.exceptions.RequestException:
+            return False
+        
         if not pba_file_contents.content:
             LOG.error("Island didn't respond with post breach file.")
             return False