PenTest/sqlmap
1
0
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-12-06 12:41:30 +00:00
Code Issues 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
Download Patch File Download Diff File Expand all files Collapse all files

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)