local nmap = require "nmap"
local shortport = require "shortport"
local stdnse = require "stdnse"
local table = require "table"
local xdmcp = require "xdmcp"

description = [[
Requests an XDMCP (X display manager control protocol) session and lists supported authentication and authorization mechanisms.
]]

---
-- @usage
-- nmap -sU -p 177 --script xdmcp-discover <ip>
--
-- @output
-- PORT    STATE         SERVICE
-- 177/udp open|filtered xdmcp
-- | xdmcp-discover: 
-- |   Session id: 0x0000703E
-- |   Authorization name: MIT-MAGIC-COOKIE-1
-- |_  Authorization data: c282137c9bf8e2af88879e6eaa922326
--

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


portrule = shortport.port_or_service(177, "xdmcp", "udp")

local mutex = nmap.mutex("xdmcp-discover")
local function fail(err) return ("\n  ERROR: %s"):format(err or "") end
	

action = function(host, port)

	local DISPLAY_ID = 1
	local result = {}

	local helper = xdmcp.Helper:new(host, port)
	local status = helper:connect()
	if ( not(status) ) then
		return fail("Failed to connect to server")
	end

	local status, response = helper:createSession(nil, 
		{"MIT-MAGIC-COOKIE-1", "XDM-AUTHORIZATION-1"}, DISPLAY_ID)
	
	if ( not(status) ) then
		return fail("Failed to create xdmcp session")
	end

	table.insert(result, ("Session id: 0x%.8X"):format(response.session_id))
	if ( response.auth_name and 0 < #response.auth_name ) then
		table.insert(result, ("Authentication name: %s"):format(response.auth_name))
	end
	if ( response.auth_data and 0 < #response.auth_data ) then
		table.insert(result, ("Authentication data: %s"):format(stdnse.tohex(response.auth_data)))
	end
	if ( response.authr_name and 0 < #response.authr_name ) then
		table.insert(result, ("Authorization name: %s"):format(response.authr_name))
	end
	if ( response.authr_data and 0 < #response.authr_data ) then
		table.insert(result, ("Authorization data: %s"):format(stdnse.tohex(response.authr_data)))
	end
	return stdnse.format_output(true, result)
end
