sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-03-03 11:45:46 +03:00
Code Issues Packages Projects Releases Wiki Activity

Minor update for an Issue #105

Browse Source
This commit is contained in:
Miroslav Stampar 2012-07-18 14:09:04 +02:00
parent 2656b8fc51
commit 4fc462c4d9
1 changed files with 3 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
lib/core/wordlist.py
View File

@ -29,10 +29,7 @@ class Wordlist:
if self.index > len(self.filenames):
raise StopIteration
elif self.index == len(self.filenames):
if not self.proc_id:
self.iter = iter(self.custom)
else:
raise StopIteration
self.iter = iter(self.custom)
else:
current = self.filenames[self.index]
self.fp = open(current, "r")
@ -48,15 +45,13 @@ class Wordlist:
def next(self):
retVal = None
while True:
self.counter += 1
try:
retVal = self.iter.next().rstrip()
except StopIteration:
self.adjust()
retVal = self.iter.next().rstrip()
if not self.proc_count:
break
self.counter += 1
if self.counter % self.proc_count == self.proc_id:
if not self.proc_count or self.counter % self.proc_count == self.proc_id:
break
return retVal