From 04210ef88f0f3ab9719a70d18b5a6b2d8f3fc94f Mon Sep 17 00:00:00 2001
From: david <david@e0a8ed71-7df4-0310-8962-fdc924857419>
Date: Thu, 30 Dec 2010 21:08:24 +0000
Subject: [PATCH] When an entry in stdnse.format_output has multiple lines,
 insert the indent and prefix before each line, not just at the beginning. If
 the indent was ">>>>", then formatting the line "AB\nCD" would result in

| >>>>  AB
|_CD

Now it will be

| >>>>  AB
|_>>>>  CD

Some script were working around this by relying on an invisible blank
first line and manually indenting following lines.
---
 nselib/stdnse.lua | 30 ++++++++++++++++++++++++++----
 1 file changed, 26 insertions(+), 4 deletions(-)

diff --git a/nselib/stdnse.lua b/nselib/stdnse.lua
index 9d9ab36df..18520b788 100644
--- a/nselib/stdnse.lua
+++ b/nselib/stdnse.lua
@@ -426,6 +426,28 @@ local function format_get_indent(indent, at_end)
   return str
 end
 
+local function splitlines(s)
+  local result = {}
+  local i = 0
+
+  while i <= #s do
+    local b, e
+    b, e = string.find(s, "\r?\n", i)
+    if not b then
+      break
+    end
+    result[#result + 1] = string.sub(s, i, b - 1)
+    i = e + 1
+  end
+
+  if i <= #s then
+    result[#result + 1] = string.sub(s, i)
+  end
+
+  return result
+end
+
+
 -- A helper for format_output (see below).
 local function format_output_sub(status, data, indent)
   if (#data == 0) then
@@ -482,10 +504,10 @@ local function format_output_sub(status, data, indent)
       output = output .. format_output_sub(status, value, new_indent)
         
     elseif(type(value) == 'string') then
-      if(i ~= #data) then
-        output = output .. format("%s  %s%s\n", format_get_indent(indent, false), prefix, value)
-      else
-        output = output .. format("%s  %s%s\n", format_get_indent(indent, true), prefix, value)
+      local lines = splitlines(value)
+
+      for j, line in ipairs(lines) do
+        output = output .. format_get_indent(indent, i == #data and j == #lines) .. "  " .. prefix .. line .. "\n"
       end
     end
   end