From: Thomas Wouters <thomas@python.org>
Date: Mon, 17 Mar 2003 11:24:29 +0000 (+0000)
Subject: binascii_a2b_base64: Properly return an empty string if the input was all
X-Git-Tag: v2.3c1~1450
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=9e1c192525f0382ec1e8710c2f8e1dcef389ff77;p=thirdparty%2FPython%2Fcpython.git

binascii_a2b_base64: Properly return an empty string if the input was all
    invalid, rather than returning a string of random garbage of the
    estimated result length. Closes SF patch #703471 by Hye-Shik Chang.

Will backport to 2.2-maint (consider it done.)
---

diff --git a/Lib/test/test_binascii.py b/Lib/test/test_binascii.py
index f1f8b33f59d1..2764c3c5e52a 100755
--- a/Lib/test/test_binascii.py
+++ b/Lib/test/test_binascii.py
@@ -69,6 +69,10 @@ for line in map(addnoise, lines):
     res = res + b
 verify(res == testdata)
 
+# Test base64 with just invalid characters, which should return
+# empty strings. TBD: shouldn't it raise an exception instead ?
+verify(binascii.a2b_base64(fillers) == '')
+
 # Test uu
 print "uu test"
 MAX_UU = 45
diff --git a/Modules/binascii.c b/Modules/binascii.c
index c56d528ef47f..05964c9351cc 100644
--- a/Modules/binascii.c
+++ b/Modules/binascii.c
@@ -408,9 +408,16 @@ binascii_a2b_base64(PyObject *self, PyObject *args)
 		return NULL;
 	}
 
-	/* and set string size correctly */
+	/* And set string size correctly. If the result string is empty
+	** (because the input was all invalid) return the shared empty
+	** string instead; _PyString_Resize() won't do this for us.
+	*/
 	if (bin_len > 0)
 		_PyString_Resize(&rv, bin_len);
+	else {
+		Py_DECREF(rv);
+		rv = PyString_FromString("");
+	}
 	return rv;
 }