Merge pull request #116 from mvschaik/patch-1

Limit memory allocation of get_bytes to 1MB
2013-02-27 15:31:20 -08:00 · 2013-02-27 15:31:20 -08:00 · 37d0247301
parent e761502e8e 3bbcf808d8
commit 37d0247301
1 changed files with 2 additions and 1 deletions
--- a/paramiko/message.py
+++ b/paramiko/message.py
@ -110,7 +110,8 @@ class Message (object):
        @rtype: string
        """
        b = self.packet.read(n)
-        if len(b) < n:
+        max_pad_size = 1<<20  # Limit padding to 1 MB
+        if len(b) < n and n < max_pad_size:
            return b + '\x00' * (n - len(b))
        return b