diff -r b239b7d9f752 Lib/test/test_xdrlib.py --- a/Lib/test/test_xdrlib.py Sat Sep 10 10:08:13 2011 +0300 +++ b/Lib/test/test_xdrlib.py Sat Sep 17 12:58:40 2011 +0200 @@ -51,8 +51,32 @@ up.done() self.assertRaises(EOFError, up.unpack_uint) +class ConversionErrorTest(unittest.TestCase): + + def setUp(self): + self.packer = xdrlib.Packer() + + def assertRaisesConversion(self, *args): + self.assertRaises(xdrlib.ConversionError, *args) + + def test_pack_int(self): + self.assertRaisesConversion(self.packer.pack_int, 'string') + + def test_pack_uint(self): + self.assertRaisesConversion(self.packer.pack_uint, 'string') + + def test_float(self): + self.assertRaisesConversion(self.packer.pack_float, 'string') + + def test_double(self): + self.assertRaisesConversion(self.packer.pack_double, 'string') + + def test_uhyper(self): + self.assertRaisesConversion(self.packer.pack_uhyper, 'string') + def test_main(): support.run_unittest(XDRTest) + support.run_unittest(ConversionErrorTest) if __name__ == "__main__": test_main() diff -r b239b7d9f752 Lib/xdrlib.py --- a/Lib/xdrlib.py Sat Sep 10 10:08:13 2011 +0300 +++ b/Lib/xdrlib.py Sat Sep 17 12:58:40 2011 +0200 @@ -31,6 +31,13 @@ class ConversionError(Error): pass +def raising_conversion_error(function): + def result(self, value): + try: + return function(self, value) + except struct.error as e: + raise ConversionError(e) + return result class Packer: @@ -47,9 +54,11 @@ # backwards compatibility get_buf = get_buffer + @raising_conversion_error def pack_uint(self, x): self.__buf.write(struct.pack('>L', x)) + @raising_conversion_error def pack_int(self, x): self.__buf.write(struct.pack('>l', x)) @@ -60,20 +69,21 @@ else: self.__buf.write(b'\0\0\0\0') def pack_uhyper(self, x): - self.pack_uint(x>>32 & 0xffffffff) - self.pack_uint(x & 0xffffffff) + try: + self.pack_uint(x>>32 & 0xffffffff) + self.pack_uint(x & 0xffffffff) + except (TypeError, struct.error) as e: + raise ConversionError(e) pack_hyper = pack_uhyper + @raising_conversion_error def pack_float(self, x): - try: self.__buf.write(struct.pack('>f', x)) - except struct.error as msg: - raise ConversionError(msg) + self.__buf.write(struct.pack('>f', x)) + @raising_conversion_error def pack_double(self, x): - try: self.__buf.write(struct.pack('>d', x)) - except struct.error as msg: - raise ConversionError(msg) + self.__buf.write(struct.pack('>d', x)) def pack_fstring(self, n, s): if n < 0: