Extended socket.htonl and ntohl to accept longs.

Fixes SF bug #568322.

The code should raise an OverflowError if the long is > 32 bits, even
on platforms where sizeof(long) > 4.

1 files changed, 11 insertions, 0 deletions

diff --git a/Lib/test/test_socket.py b/Lib/test/test_socket.py
index 423df76..44b42ff 100644
--- a/Lib/test/test_socket.py
+++ b/Lib/test/test_socket.py
@@ -247,6 +247,17 @@ class GeneralModuleTests(unittest.TestCase):
         except socket.error:
             pass
 
+    def testNtoH(self):
+        def twice(f):
+            def g(x):
+                return f(f(x))
+            return g
+        for i in (0, 1, 0xffff0000, 2L, (2**32L) - 1):
+            self.assertEqual(i, twice(socket.htonl)(i))
+            self.assertEqual(i, twice(socket.ntohl)(i))
+        self.assertRaises(OverflowError, socket.htonl, 2L**34)
+        self.assertRaises(OverflowError, socket.ntohl, 2L**34)
+
     def testGetServByName(self):
         """Testing getservbyname()."""
         if hasattr(socket, 'getservbyname'):