Several Fixes

VainlyStrain · Jul 3, 2020 · 3dc4606 · 3dc4606
1 parent d1416d1
commit 3dc4606
Show file tree

Hide file tree

Showing 9 changed files with 12 additions and 0 deletions.
diff --git a/core/methods/__pycache__/cookie.cpython-37.pyc b/core/methods/__pycache__/cookie.cpython-37.pyc
diff --git a/core/methods/__pycache__/filecheck.cpython-37.pyc b/core/methods/__pycache__/filecheck.cpython-37.pyc
diff --git a/core/methods/__pycache__/inpath.cpython-37.pyc b/core/methods/__pycache__/inpath.cpython-37.pyc
diff --git a/core/methods/__pycache__/loot.cpython-37.pyc b/core/methods/__pycache__/loot.cpython-37.pyc
diff --git a/core/methods/__pycache__/query.cpython-37.pyc b/core/methods/__pycache__/query.cpython-37.pyc
diff --git a/core/methods/cookie.py b/core/methods/cookie.py
@@ -55,6 +55,8 @@ def readCookie(url):
     return (cookie, selectedpart)
 
 def determine_payloads_cookie(url, cookie, selected ,verbose, depth, paylist, file):
+    if not url.endswith("/"):
+        url += "/"
     s = session()
     payloads = []
     nullbytes = []
@@ -94,6 +96,8 @@ def determine_payloads_cookie(url, cookie, selected ,verbose, depth, paylist, fi
     return (payloads, nullbytes)
 
 def cookie_attack(url, cookie, selected, files, dirs, depth, verbose, dl, selected_payloads, selected_nullbytes):
+    if not url.endswith("/"):
+        url += "/"
     s = session()
     found=[]
     urls = []

diff --git a/core/methods/filecheck.py b/core/methods/filecheck.py
@@ -22,6 +22,8 @@
 def filecheck(r, con2, payload):
     con = r.content
     conn = str(con).lower()
+    if r.encoding == None:
+      r.encoding = "utf-8"
     txt = r.text.lower()
     check = (con != con2 and "[<a href='function.main'>function.main</a>" not in conn and "[<a href='function.include'>function.include</a>" not in conn 
     and ("failed opening" not in conn and "for inclusion" not in conn) and "failed to open stream:" not in conn and "open_basedir restriction in effect" not in conn 

diff --git a/core/methods/inpath.py b/core/methods/inpath.py
@@ -77,6 +77,8 @@ def inpath(url,url2,keyword,files,dirs,depth,verbose,dl,selected_payloads, selec
 
 
 def determine_payloads_inpath(url,url2,keyword,verbose,depth,paylist,file):
+    if not url.endswith("/"):
+        url += "/"
     payloads = []
     nullbytes = []
     s = session()

diff --git a/core/methods/query.py b/core/methods/query.py
@@ -26,6 +26,8 @@
 
 
 def query(url,url2,keyword,files,dirs,depth,verbose,dl, selected_payloads, selected_nullbytes):
+    if not url.endswith("/"):
+        url += "/"
     found=[]
     urls = []
     s = session()
@@ -78,6 +80,8 @@ def query(url,url2,keyword,files,dirs,depth,verbose,dl, selected_payloads, selec
     return (found, urls)
 
 def determine_payloads_query(url,url2,keyword,verbose,depth,paylist, file):
+    if not url.endswith("/"):
+        url += "/"
     payloads = []
     nullbytes = []
     s = session()