From 0427a20e68cb4f7ad466206d97b4ccb33f5a58da Mon Sep 17 00:00:00 2001
From: Phil <phil@secdev.org>
Date: Tue, 10 Feb 2009 14:04:01 +0100
Subject: [PATCH] Added default random size parameter to RandString and RandBin
classes (at last!)
---
scapy/volatile.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/scapy/volatile.py b/scapy/volatile.py
index 17a66e35..8c8b4533 100644
--- a/scapy/volatile.py
+++ b/scapy/volatile.py
@@ -144,9 +144,11 @@ class RandChoice(RandField):
return random.choice(self._choice)
class RandString(RandField):
- def __init__(self, size, chars="ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"):
- self.chars = chars
+ def __init__(self, size=None, chars="ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"):
+ if size is None:
+ size = RandNumExpo(0.01)
self.size = size
+ self.chars = chars
def _fix(self):
s = ""
for i in range(self.size):
@@ -154,7 +156,7 @@ class RandString(RandField):
return s
class RandBin(RandString):
- def __init__(self, size):
+ def __init__(self, size=None):
RandString.__init__(self, size, "".join(map(chr,range(256))))
--
GitLab