Add tests for --save-*

[mimedecode.git] / mimedecode.py

diff --git a/mimedecode.py b/mimedecode.py
index c1731d6336c5ab8eff9e5928a61a9cce1181ddc2..50f8a18675a5265700f2f4928208e169dc116f8d 100755 (executable)
--- a/mimedecode.py
+++ b/mimedecode.py
@@ -270,6 +270,14 @@ def _save_message(msg, outstring, save_headers=False, save_body=False):
     ):
         fname = msg.get_param(param, header=header)
         if fname:
+            if isinstance(fname, tuple):
+                fname = fname[2] # Do not recode if it isn't recoded yet
+            try:
+                    for forbidden in chr(0), '/', '\\':
+                        if forbidden in fname:
+                            raise ValueError
+            except ValueError:
+                continue
             fname = '-' + fname
             break
     else: