diff --git a/webapp/xss_trainer/levels/contrib.py b/webapp/xss_trainer/levels/contrib.py index ab79f34..3701199 100644 --- a/webapp/xss_trainer/levels/contrib.py +++ b/webapp/xss_trainer/levels/contrib.py @@ -18,8 +18,11 @@ class EscapeChars(meta.BaseLevel): author = "Sharkmoos" def sanitise(self, data): - # This is more of a level 2/3 difficulty - payload = (data.replace("'", "\\'")).replace('"', '\\"') + try: + int(data) + payload = (data.replace("'", "\\'")).replace('"', '\\"') + except ValueError: + payload = "You must enter a string payload for this level" return payload class Encoding(meta.BaseLevel):