sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-06-15 18:43:16 +03:00
Code Issues Packages Projects Releases Wiki Activity

minor update

Browse Source
This commit is contained in:
Miroslav Stampar 2012-03-28 13:46:31 +00:00
parent 7fd64df167
commit 7d131d1fb1
3 changed files with 28 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
_sqlmap.py
View File

@ -128,6 +128,3 @@ def main():
# Reference: http://stackoverflow.com/questions/1635080/terminate-a-multi-thread-python-program # Reference: http://stackoverflow.com/questions/1635080/terminate-a-multi-thread-python-program
if hasattr(conf, "threads") and conf.threads > 1: if hasattr(conf, "threads") and conf.threads > 1:
os._exit(0) os._exit(0)
if __name__ == "__main__":
main()

35
lib/request/dns.py
View File

@ -7,8 +7,10 @@ Copyright (c) 2006-2012 sqlmap developers (http://www.sqlmap.org/)
See the file 'doc/COPYING' for copying permission See the file 'doc/COPYING' for copying permission
""" """
import os
import socket import socket
import threading import threading
import time
class DNSQuery: class DNSQuery:
""" """
@ -34,21 +36,22 @@ class DNSQuery:
j = ord(raw[i]) j = ord(raw[i])
def response(self, resolution): def response(self, resolution):
retval = "" retVal = ""
if self._query: if self._query:
retval += self._raw[:2] + "\x81\x80" retVal += self._raw[:2] + "\x81\x80"
retval += self._raw[4:6] + self._raw[4:6] + "\x00\x00\x00\x00" # Questions and Answers Counts retVal += self._raw[4:6] + self._raw[4:6] + "\x00\x00\x00\x00" # Questions and Answers Counts
retval += self._raw[12:] # Original Domain Name Question retVal += self._raw[12:] # Original Domain Name Question
retval += "\xc0\x0c" # Pointer to domain name retVal += "\xc0\x0c" # Pointer to domain name
retval += "\x00\x01\x00\x01\x00\x00\x00\x3c\x00\x04" # Response type, ttl and resource data length -> 4 bytes retVal += "\x00\x01\x00\x01\x00\x00\x00\x3c\x00\x04" # Response type, ttl and resource data length -> 4 bytes
retval += "".join(chr(int(_)) for _ in resolution.split('.')) # 4 bytes of IP retVal += "".join(chr(int(_)) for _ in resolution.split('.')) # 4 bytes of IP
return retval return retVal
class DNSServer: class DNSServer:
def __init__(self): def __init__(self):
self._requests = [] self._requests = []
self._lock = threading.Lock()
def run(self): def run(self):
def _(): def _():
@ -60,9 +63,25 @@ class DNSServer:
data, addr = s.recvfrom(1024) data, addr = s.recvfrom(1024)
_ = DNSQuery(data) _ = DNSQuery(data)
s.sendto(_.response("127.0.0.1"), addr) s.sendto(_.response("127.0.0.1"), addr)
self._lock.acquire()
self._requests.append(_._query) self._requests.append(_._query)
self._lock.release()
finally: finally:
s.close() s.close()
thread = threading.Thread(target=_) thread = threading.Thread(target=_)
thread.start() thread.start()
if __name__ == "__main__":
server = DNSServer()
try:
server.run()
while True:
server._lock.acquire()
for _ in server._requests[:]:
print _
server._requests = []
server._lock.release()
time.sleep(1)
except KeyboardInterrupt:
os._exit(0)

2
sqlmap.py
View File

@ -13,7 +13,7 @@ PYVERSION = sys.version.split()[0]
if PYVERSION >= "3" or PYVERSION < "2.6": if PYVERSION >= "3" or PYVERSION < "2.6":
exit("[CRITICAL] incompatible Python version detected ('%s'). For successfully running sqlmap you'll have to use version 2.6 or 2.7 (visit \"http://www.python.org/download/\")" % PYVERSION) exit("[CRITICAL] incompatible Python version detected ('%s'). For successfully running sqlmap you'll have to use version 2.6 or 2.7 (visit \"http://www.python.org/download/\")" % PYVERSION)
else: elif __name__ == "__main__":
from _sqlmap import main from _sqlmap import main
# import needed for proper working of --profile switch # import needed for proper working of --profile switch
from lib.controller.controller import start from lib.controller.controller import start