From 46bd7808625543bb42d1c1f20ff5599b837d1ca8 Mon Sep 17 00:00:00 2001
From: Aldo Cortesi <aldo@nullcube.com>
Date: Thu, 9 Feb 2012 12:09:40 +1300
Subject: Gracefully handle invalid data format passed to -r flag.

---
 libmproxy/console/__init__.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'libmproxy/console')

diff --git a/libmproxy/console/__init__.py b/libmproxy/console/__init__.py
index 608742a4..76968ae5 100644
--- a/libmproxy/console/__init__.py
+++ b/libmproxy/console/__init__.py
@@ -640,7 +640,10 @@ class ConsoleMaster(flow.FlowMaster):
             fr = flow.FlowReader(f)
         except IOError, v:
             return v.strerror
-        flow.FlowMaster.load_flows(self, fr)
+        try:
+            flow.FlowMaster.load_flows(self, fr)
+        except flow.FlowReadError, v:
+            return v.strerror
         f.close()
         if self.conn_list_view:
             self.sync_list_view()
-- 
cgit v1.2.3