bug not detecting EOF correctly

mrvirginis · Nov 9, 2018 · 5827497 · 5827497
1 parent e68ca23
commit 5827497
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/wfuzz/plugins/payloads/file.py b/src/wfuzz/plugins/payloads/file.py
@@ -33,11 +33,11 @@ def __init__(self, params):
         self.__count = None
 
     def __next__(self):
-        line = self.f.readline().strip()
-        if line == '':
+        line = self.f.readline()
+        if not line:
             self.f.close()
             raise StopIteration
-        return line
+        return line.strip()
 
     def count(self):
         if self.__count is None: