PenTest/nmap
1
0
Fork 0
mirror of https://github.com/nmap/nmap.git synced 2025-12-18 05:29:02 +00:00
Code Issues Projects Releases Wiki Activity

Whitespace in comm.lua.

Browse Source
This commit is contained in:
david
2011-01-10 00:25:12 +00:00
parent 4e5f8799e1
commit 64ccea886b
1 changed files with 103 additions and 103 deletions
Download Patch File Download Diff File Expand all files Collapse all files

206
nselib/comm.lua
View File

@@ -27,49 +27,49 @@ require 'datafiles'
-- Makes sure that opts exists and the default proto is there -- Makes sure that opts exists and the default proto is there
local initopts = function(opts) local initopts = function(opts)
if not opts then if not opts then
opts = {} opts = {}
end end
if not opts.proto then if not opts.proto then
opts.proto = "tcp" opts.proto = "tcp"
end end
return opts return opts
end end
-- Sets up the socket and connects to host:port -- Sets up the socket and connects to host:port
local setup_connect = function(host, port, opts) local setup_connect = function(host, port, opts)
local sock = nmap.new_socket() local sock = nmap.new_socket()
if opts.timeout then if opts.timeout then
sock:set_timeout(opts.timeout) sock:set_timeout(opts.timeout)
end end
local status, err = sock:connect(host, port, opts.proto) local status, err = sock:connect(host, port, opts.proto)
if not status then if not status then
return status, err return status, err
end end
return true, sock return true, sock
end end
local read = function(sock, opts) local read = function(sock, opts)
local response, status local response, status
if opts.lines then if opts.lines then
status, response = sock:receive_lines(opts.lines) status, response = sock:receive_lines(opts.lines)
return status, response return status, response
end end
if opts.bytes then if opts.bytes then
status, response = sock:receive_bytes(opts.bytes) status, response = sock:receive_bytes(opts.bytes)
return status, response return status, response
end end
status, response = sock:receive() status, response = sock:receive()
return status, response return status, response
end end
--- This function simply connects to the specified port number on the --- This function simply connects to the specified port number on the
@@ -84,14 +84,14 @@ end
-- @return Status (true or false). -- @return Status (true or false).
-- @return Data (if status is true) or error string (if status is false). -- @return Data (if status is true) or error string (if status is false).
get_banner = function(host, port, opts) get_banner = function(host, port, opts)
opts = initopts(opts) opts = initopts(opts)
opts.recv_before = true opts.recv_before = true
local socket, nothing, correct, banner = tryssl(host, port, "", opts) local socket, nothing, correct, banner = tryssl(host, port, "", opts)
if socket then if socket then
socket:close() socket:close()
return true, banner return true, banner
end end
return false, banner return false, banner
end end
--- This function connects to the specified port number on the specified --- This function connects to the specified port number on the specified
@@ -107,28 +107,28 @@ end
-- @return Status (true or false). -- @return Status (true or false).
-- @return Data (if status is true) or error string (if status is false). -- @return Data (if status is true) or error string (if status is false).
exchange = function(host, port, data, opts) exchange = function(host, port, data, opts)
opts = initopts(opts) opts = initopts(opts)
local status, sock = setup_connect(host, port, opts) local status, sock = setup_connect(host, port, opts)
local ret local ret
if not status then if not status then
-- sock is an error message in this case -- sock is an error message in this case
return status, sock return status, sock
end end
status, ret = sock:send(data) status, ret = sock:send(data)
if not status then if not status then
sock:close() sock:close()
return status, ret return status, ret
end end
status, ret = read(sock, opts) status, ret = read(sock, opts)
sock:close() sock:close()
return status, ret return status, ret
end end
--- This function just checks if the provided port number is on a list --- This function just checks if the provided port number is on a list
@@ -137,14 +137,14 @@ end
-- @param port_number The number of the port to check -- @param port_number The number of the port to check
-- @return bool True if port is usually ssl, otherwise false -- @return bool True if port is usually ssl, otherwise false
local function is_ssl(port_number) local function is_ssl(port_number)
local common_ssl_ports = {443, 465, 989, 990, 992, 993, 994, 995, 587, 6697, 6679, 8443} local common_ssl_ports = {443, 465, 989, 990, 992, 993, 994, 995, 587, 6697, 6679, 8443}
local table_size = table.maxn(common_ssl_ports) local table_size = table.maxn(common_ssl_ports)
local i = 0 local i = 0
while i < table_size do while i < table_size do
if port_number == common_ssl_ports[i] then return true end if port_number == common_ssl_ports[i] then return true end
i = i + 1 i = i + 1
end end
return false return false
end end
--- This function returns best protocol order for trying to open a --- This function returns best protocol order for trying to open a
@@ -155,14 +155,14 @@ end
-- @return Best option ("tcp" or "ssl") -- @return Best option ("tcp" or "ssl")
-- @return Worst option ("tcp" or "ssl") -- @return Worst option ("tcp" or "ssl")
local function bestoption(port) local function bestoption(port)
if type(port) == 'table' then if type(port) == 'table' then
if port.version and port.version.service_tunnel and port.version.service_tunnel == "ssl" then return "ssl","tcp" end if port.version and port.version.service_tunnel and port.version.service_tunnel == "ssl" then return "ssl","tcp" end
if port.version and port.version.name_confidence and port.version.name_confidence > 6 then return "tcp","ssl" end if port.version and port.version.name_confidence and port.version.name_confidence > 6 then return "tcp","ssl" end
if is_ssl(port.number) then return "ssl","tcp" end if is_ssl(port.number) then return "ssl","tcp" end
elseif type(port) == 'number' then elseif type(port) == 'number' then
if is_ssl(port) then return "ssl","tcp" end if is_ssl(port) then return "ssl","tcp" end
end end
return "tcp","ssl" return "tcp","ssl"
end end
--- This function opens a connection, sends the first data payload and --- This function opens a connection, sends the first data payload and
@@ -186,51 +186,51 @@ end
-- @return early_resp If opt recv_before is true, returns the value -- @return early_resp If opt recv_before is true, returns the value
-- of the first receive (before sending data) -- of the first receive (before sending data)
local function opencon(host, port, protocol, data, opts) local function opencon(host, port, protocol, data, opts)
local sd = nmap.new_socket() local sd = nmap.new_socket()
-- check for connect_timeout or timeout option -- check for connect_timeout or timeout option
if opts and opts.connect_timeout then if opts and opts.connect_timeout then
sd:set_timeout(opts.connect_timeout) sd:set_timeout(opts.connect_timeout)
elseif opts and opts.timeout then elseif opts and opts.timeout then
sd:set_timeout(opts.timeout) sd:set_timeout(opts.timeout)
else else
sd:set_timeout(8000) sd:set_timeout(8000)
end end
local status = sd:connect(host, port, protocol) local status = sd:connect(host, port, protocol)
if not status then if not status then
sd:close() sd:close()
return nil, nil, nil return nil, nil, nil
end end
-- check for request_timeout or timeout option -- check for request_timeout or timeout option
if opts and opts.request_timeout then if opts and opts.request_timeout then
sd:set_timeout(opts.request_timeout) sd:set_timeout(opts.request_timeout)
elseif opts and opts.timeout then elseif opts and opts.timeout then
sd:set_timeout(opts.timeout) sd:set_timeout(opts.timeout)
else else
sd:set_timeout(8000) sd:set_timeout(8000)
end end
local response, early_resp; local response, early_resp;
if opts and opts.recv_before then status, early_resp = sd:receive() end if opts and opts.recv_before then status, early_resp = sd:receive() end
if #data > 0 then if #data > 0 then
sd:send(data) sd:send(data)
status, response = sd:receive() status, response = sd:receive()
else else
if not opts and opts.recv_before then if not opts and opts.recv_before then
nsedebug.print_debug("Using comm.tryssl without first data payload and recv_first." .. nsedebug.print_debug("Using comm.tryssl without first data payload and recv_first." ..
"\nImpossible to test the connection for the correct protocol!") "\nImpossible to test the connection for the correct protocol!")
end end
response = early_resp response = early_resp
end end
if not status then if not status then
sd:close() sd:close()
return nil, response, early_resp return nil, response, early_resp
end end
return sd, response, early_resp return sd, response, early_resp
end end
--- This function tries to open a connection based on the best --- This function tries to open a connection based on the best
@@ -252,13 +252,13 @@ end
-- @return earlyResp If opt recv_before is true, returns the value -- @return earlyResp If opt recv_before is true, returns the value
-- of the first receive (before sending data) -- of the first receive (before sending data)
function tryssl(host, port, data, opts) function tryssl(host, port, data, opts)
local opt1, opt2 = bestoption(port) local opt1, opt2 = bestoption(port)
local best = opt1 local best = opt1
local sd, response, early_resp = opencon(host, port, opt1, data, opts) local sd, response, early_resp = opencon(host, port, opt1, data, opts)
if not sd then if not sd then
sd, response, early_resp = opencon(host, port, opt2, data, opts) sd, response, early_resp = opencon(host, port, opt2, data, opts)
best = opt2 best = opt2
end end
if not sd then best = "none" end if not sd then best = "none" end
return sd, response, best, early_resp return sd, response, best, early_resp
end end