diff --git a/scapy/base_classes.py b/scapy/base_classes.py index ae833ea..bace90d 100644 --- 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)