shell_read_until_token should work for any requested idx now

git-svn-id: file:///home/svn/framework3/trunk@8665 4d416f70-5f16-0410-b530-b9f4589650da

lib/msf/base/sessions/command_shell.rb

@@ -132,16 +132,21 @@ class CommandShell
 		if (not (select([rstream], nil, nil, 5)))
 			return nil
 		end
+		if (wanted_idx == 0)
+			parts_needed = 2
+		else
+			parts_needed = 1 + (wanted_idx * 2)
+		end
 
 		# Read until we get the token or timeout.
 		buf = ''
 		idx = nil
 		while (tmp = rstream.get_once(-1, 1))
 			buf << tmp
-			
+
 			# see if we have the wanted idx
 			parts = buf.split(token, -1)
-			if (parts.length == 1+(wanted_idx*2))
+			if (parts.length == parts_needed)
 				# cause another prompt to appear (just in case)
 				shell_write("\n")
 				return parts[wanted_idx]