local creds = require "creds"
local nmap = require "nmap"
local shortport = require "shortport"
local stdnse = require "stdnse"
local bin = require "bin"
local string = require "string"
local table = require "table"

local cassandra = stdnse.silent_require "cassandra"

description = [[
Attempts to get basic info and server status from a Cassandra database.

For more information about Cassandra, see:
http://cassandra.apache.org/
]]

---
-- @usage
-- nmap -p 9160 <ip> --script=cassandra-info
-- 
-- @output
-- PORT     STATE SERVICE   REASON
-- 9160/tcp open  cassandra syn-ack
-- | cassandra-info: 
-- |   Cluster name: Test Cluster
-- |_  Version: 19.10.0
-- 

-- version 0.1
-- Created 14/09/2012 - v0.1 - created by Vlatko Kosturjak <kost@linux.hr>

author = "Vlatko Kosturjak"
license = "Same as Nmap--See http://nmap.org/book/man-legal.html"
categories = {"default", "discovery", "safe"}

dependencies = {"cassandra-brute"}

portrule = shortport.port_or_service({9160}, {"cassandra"})

function action(host,port)

	local socket = nmap.new_socket()
        local cassinc = 2 -- cmd/resp starts at 2
	
	-- set a reasonable timeout value
	socket:set_timeout(10000)
	-- do some exception  / cleanup
	local catch = function()
		socket:close()
	end
	
	local try = nmap.new_try(catch)

	try( socket:connect(host, port) )
	
        local results = {}

	-- uglyness to allow creds.cassandra to work, as the port is not recognized
	-- as cassandra even when service scan was run, taken from mongodb
	local ps = port.service
	port.service = 'cassandra'
	local c = creds.Credentials:new(creds.ALL_DATA, host, port)
	for cred in c:getCredentials(creds.State.VALID + creds.State.PARAM) do
		local status, err = cassandra.login(socket, cred.user, cred.pass)
                table.insert(results, ("Using credentials: %s"):format(cred.user.."/"..cred.pass))
		if ( not(status) ) then
			return err
		end
	end
	port.service = ps

        local status, val = cassandra.describe_cluster_name(socket,cassinc)
        if (not(status)) then
          return "Error getting cluster name: " .. val
        end
        cassinc = cassinc + 1
        port.version.name ='cassandra'
        port.version.product='Cassandra'
        port.version.name_confidence = 10
        nmap.set_port_version(host,port)
        table.insert(results, ("Cluster name: %s"):format(val))

        local status, val = cassandra.describe_version(socket,cassinc)
        if (not(status)) then
          return "Error getting version: " .. val
        end
        cassinc = cassinc + 1
        port.version.product='Cassandra ('..val..')'
        nmap.set_port_version(host,port)
        table.insert(results, ("Version: %s"):format(val))

        return stdnse.format_output(true, results)
end