sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2024-11-22 01:26:42 +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
if hasattr(conf, "threads") and conf.threads > 1:
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
"""
import os
import socket
import threading
import time
class DNSQuery:
"""
@ -34,21 +36,22 @@ class DNSQuery:
j = ord(raw[i])
def response(self, resolution):
retval = ""
retVal = ""
if self._query:
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[12:] # Original Domain Name Question
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 += "".join(chr(int(_)) for _ in resolution.split('.')) # 4 bytes of IP
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[12:] # Original Domain Name Question
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 += "".join(chr(int(_)) for _ in resolution.split('.')) # 4 bytes of IP
return retval
return retVal
class DNSServer:
def __init__(self):
self._requests = []
self._lock = threading.Lock()
def run(self):
def _():
@ -60,9 +63,25 @@ class DNSServer:
data, addr = s.recvfrom(1024)
_ = DNSQuery(data)
s.sendto(_.response("127.0.0.1"), addr)
self._lock.acquire()
self._requests.append(_._query)
self._lock.release()
finally:
s.close()
thread = threading.Thread(target=_)
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":
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
# import needed for proper working of --profile switch
from lib.controller.controller import start