Allow generator to be used for sr function issue #31

phaethon · Nov 12, 2015 · ee66c1d · ee66c1d
1 parent 0cc599c
commit ee66c1d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scapy/base_classes.py b/scapy/base_classes.py
@@ -12,6 +12,7 @@
 ################
 
 import re,random,socket
+from types import GeneratorType
 import scapy.config
 from . import error
 
@@ -22,7 +23,7 @@ def __iter__(self):
 class SetGen(Gen):
     def __init__(self, col, _iterpacket=1):
         self._iterpacket=_iterpacket
-        if type(col) is list:
+        if type(col) is list or isinstance(col, GeneratorType):
             self.col = col
         elif isinstance(col, BasePacketList):
             self.col = list(col)