From fe981e5014e4fef769f50480da6b421723dc1aca Mon Sep 17 00:00:00 2001
From: djalal <djalal@e0a8ed71-7df4-0310-8962-fdc924857419>
Date: Thu, 30 Jun 2011 22:14:58 +0000
Subject: [PATCH] o [NSE] Added an ftp.connect() methode.

---
 nselib/ftp.lua | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/nselib/ftp.lua b/nselib/ftp.lua
index d3b52a0a4..3d0e7c5d6 100644
--- a/nselib/ftp.lua
+++ b/nselib/ftp.lua
@@ -5,6 +5,33 @@
 
 module(... or "ftp", package.seeall)
 
+local comm    = require 'comm'
+local stdnse  = require 'stdnse'
+
+local ERROR_MESSAGES = {
+  ["EOF"]     = "connection closed",
+  ["TIMEOUT"] = "connection timeout",
+  ["ERROR"]   = "failed to receive data"
+}
+
+--- Connects to the FTP server based on the provided options.
+--
+-- @param host The host table
+-- @param port The port table
+-- @param opts The connection option table, possible options:
+--    timeout: generic timeout value
+--    recv_before: receive data before returning
+-- @return socket The socket descriptor, or nil on errors
+-- @return response The response received on success and when
+--    the recv_before is set, or the error message on failures.
+connect = function(host, port, opts)
+  local socket, _, _, ret = comm.tryssl(host, port, '', opts)
+  if not socket then
+    return socket, (ERROR_MESSAGES[ret] or 'unspecified error')
+  end
+  return socket, ret
+end
+
 ---
 -- Read an FTP reply and return the numeric code and the message. See RFC 959,
 -- section 4.2.