Permalink
Browse files

Merge pull request #62 from Nythiennzo/master

Fix bug on injectString size input.
  • Loading branch information...
codingo committed Dec 5, 2017
2 parents 9bcf017 + b8852e2 commit 9cd931198cce2cca182900b1e5ea9353f32fb8da
Showing with 18 additions and 2 deletions.
  1. +18 −2 nsmweb.py
View
@@ -81,7 +81,15 @@ def getApps(webPort,victim,uri,https,verb,requestHeaders):
if appUp == True:
injectSize = raw_input("Baseline test-Enter random string size: ")
sizeSelect = True
while sizeSelect:
injectSize = raw_input("Baseline test-Enter random string size: ")
if injectSize.isdigit():
sizeSelect = False
else:
print "Invalid! The size should be an integer."
injectString = randInjString(int(injectSize))
print "Using " + injectString + " for injection testing.\n"
@@ -429,7 +437,15 @@ def postApps(victim,webPort,uri,https,verb,postData,requestHeaders):
raw_input("Something went wrong. Press enter to return to the main menu...")
return
injectSize = raw_input("Baseline test-Enter random string size: ")
sizeSelect = True
while sizeSelect:
injectSize = raw_input("Baseline test-Enter random string size: ")
if injectSize.isdigit():
sizeSelect = False
else:
print "Invalid! The size should be an integer."
injectString = randInjString(int(injectSize))
print "Using " + injectString + " for injection testing.\n"

0 comments on commit 9cd9311

Please sign in to comment.