Skip to content

Commit 471ad5a

Browse files
committed
Optimization on nmap to accept file pointers (#2)
1 parent b220e71 commit 471ad5a

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

searchsploit.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -539,7 +539,10 @@ def nmapxml(file=""):
539539

540540
# First check whether file exists or use stdin
541541
try:
542-
contentFile = open(file, "r")
542+
if (type(file) == str):
543+
contentFile = open(file, "r")
544+
else:
545+
contentFile = file # if file access, link directly to file pointer
543546
content = contentFile.read()
544547
contentFile.close()
545548
except:
@@ -606,7 +609,10 @@ def nmapgrep(file=""):
606609

607610
# First check whether file exists or use stdin
608611
try:
609-
contentFile = open(file, "r")
612+
if (type(file) == str):
613+
contentFile = open(file, "r")
614+
else:
615+
contentFile = file
610616
content = contentFile.read()
611617
contentFile.close()
612618
except:

0 commit comments

Comments
 (0)