From 5bd45a7642d35d3174345937587e8ce88f13715f Mon Sep 17 00:00:00 2001
From: david <david@e0a8ed71-7df4-0310-8962-fdc924857419>
Date: Thu, 16 Jul 2009 23:24:42 +0000
Subject: [PATCH] pjl-ready-message.nse: Add more error checking, shorten the
 output to one line, and display the previous message also when the message is
 changed.

---
 scripts/pjl-ready-message.nse | 56 +++++++++++++++++++++--------------
 1 file changed, 33 insertions(+), 23 deletions(-)

diff --git a/scripts/pjl-ready-message.nse b/scripts/pjl-ready-message.nse
index a12276b46..e88aca4ac 100644
--- a/scripts/pjl-ready-message.nse
+++ b/scripts/pjl-ready-message.nse
@@ -10,10 +10,7 @@ message and changes it to the message given.
 -- @arg pjl_ready_message Ready message to display.
 -- @output
 -- 9100/tcp open  jetdirect
--- |   hprdymsg: "Printer display initially read: "<initial message>"
--- |_ "p0wn3d pr1nt3r" was set as the display for printer at printer.ip.address
--- |_ "Re-polling printer to check that message was successful"
--- |_ "Current printer display message: "p0wn3d pr1nt3r"
+-- |_ pjl-ready-message: "READY" changed to "p0wn3d pr1nt3r"
 -- @usage
 -- nmap --script=pjl-ready-message.nse \
 --   --script-args='pjl_ready_message="your message here"'
@@ -31,20 +28,13 @@ portrule = shortport.port_or_service(9100, "jetdirect")
 local function parse_response(response)
 	local msg
 	local line
-	local found=0
 	
 	for line in response:gmatch(".-\n") do
 		msg = line:match("^DISPLAY=\"(.*)\"")
 		if msg then
-			found=1
-			break
+			return msg
 		end
 	end
-	if found==1 then
-		return(msg)
-	else
-		return("No Response from Printer")
-	end
 end
 
 action = function(host, port)
@@ -67,14 +57,28 @@ action = function(host, port)
 	response,data=socket:receive()  
 	if not response then			--send an initial probe. If no response, send nothing further. 
 		socket:close()
-		return("No response from printer")
-	else
-	--The following block will check for an argument from the command line and if there isn't one, it will attempt to return the current display and quit out
-		if not nmap.registry.args.pjl_ready_message then
-			status=parse_response(data)
-			return("Current Display: \""..status.."\"")
-		else	--There is a command-line arg if you got here. Set up the new display message for injection. 
-			rdymsgarg=nmap.registry.args.pjl_ready_message
+		if nmap.verbosity() > 0 then
+			return "No response from printer: "..data
+		else
+			return nil
+		end
+	end
+
+	status = parse_response(data)
+	if not status then
+		if nmap.verbosity() > 0 then
+			return "Error reading printer response: "..data
+		else
+			return nil
+		end
+	end
+
+	rdymsgarg = nmap.registry.args.pjl_ready_message
+	if not rdymsgarg then
+		if status then
+			return "\""..status.."\"" 
+		else
+			return nil
 		end
 	end
 	
@@ -83,11 +87,17 @@ action = function(host, port)
 	
 	try(socket:send(statusmsg))		--this block gets the status again for comparison
 	response,data=socket:receive()
+	if not response then
+		socket:close()
+		return "\""..status.."\""
+	end
 	newstatus=parse_response(data)
+	if not newstatus then
+		socket:close()
+		return "\""..status.."\""
+	end
 	
 	socket:close()
 	
-	local outstring
-	outstring = "\""..rdymsgarg.."\"".." was set as the display for printer at "..host.ip.."\r\n".."Re-polling printer to check that message was successful...\r\n".."Current printer display message: \""..newstatus.."\""
-	return(outstring)
+	return "\""..status.."\" changed to \""..newstatus.."\""
 end