"""M2Crypto client-side FTP/TLS.

This implementation complies with draft-murray-auth-ftp-ssl-07.txt.

Example:

>>> import xftpslib
>>> f = ftpslib.FTP_TLS()
>>> f.connect('', 9021)
'220 spinnaker.dyndns.org M2Crypto (Medusa) FTP/TLS server v0.07 ready.'
>>> f.auth_tls()
>>> f.set_pasv(0)
>>> f.login('ftp', 'ngps@')
'230 Ok.'
>>> f.retrlines('LIST')
-rw-rw-r--   1 0        198          2326 Jul  3  1996 apache_pb.gif
drwxrwxr-x   7 0        198          1536 Oct 10  2000 manual
drwxrwxr-x   2 0        198           512 Oct 31  2000 modpy
drwxrwxr-x   2 0        198           512 Oct 31  2000 bobo
drwxr-xr-x   2 0        198         14336 May 28 15:54 postgresql
drwxr-xr-x   4 100      198           512 May 16 17:19 home
drwxr-xr-x   7 100      100          3584 Sep 23  2000 openacs
drwxr-xr-x  10 0        0             512 Aug  5  2000 python1.5
-rw-r--r--   1 100      198           326 Jul 29 03:29 index.html
drwxr-xr-x  12 0        0             512 May 31 17:08 python2.1
'226 Transfer complete'
>>> f.quit()
'221 Goodbye.'
>>>
"""

# Python
from ftplib import *
#from ftplib import parse150, parse227
#from ftplib import error_reply, error_temp, error_perm, error_proto
import socket, time

##import SSL
##DEFAULT_PROTOCOL='sslv23'

class FTP_TLS(FTP):

    """Python OO interface to client-side FTP/TLS."""

    def __init__(self, host=None, key_file=None, cert_file=None):
        """Initialise the client. If 'host' is supplied, connect to it."""
##        if ssl_ctx is not None:
##            self.ssl_ctx = ssl_ctx
##        else:
##            self.ssl_ctx = SSL.Context(DEFAULT_PROTOCOL)
        self.key_file = key_file
        self.cert_file = cert_file
        
        if host:
            self.connect(host)
        self.prot = 0

    def auth_tls(self):
        """Secure the control connection per AUTH TLS, aka AUTH TLS-C."""
        self.voidcmd('AUTH TLS')
        #self.voidcmd('AUTH SSL')
        ssl = socket.ssl(self.sock, self.key_file, self.cert_file)
        import httplib
        self.sock = httplib.FakeSocket(self.sock, ssl)
        self.file = self.sock.makefile('rb')

    def auth_ssl(self):
        """Secure the control connection per AUTH SSL, aka AUTH TLS-P."""
        raise NotImplementedError

    def prot_p(self):
        """Set up secure data connection."""
        self.voidcmd('PBSZ 0')
        self.voidcmd('PROT P')
        self.prot = 1
            
    def prot_c(self):
        """Set up data connection in the clear."""
        self.voidcmd('PROT C')
        self.prot = 0

    def ntransfercmd(self, cmd, rest=None):
        """Initiate a data transfer."""
        conn, size = FTP.ntransfercmd(self, cmd, rest)
        if self.prot:
            ssl = socket.ssl(conn, self.key_file, self.cert_file)
            import httplib
            conn = httplib.FakeSocket(conn, ssl)
            
            ##conn = socket.ssl(conn) #SSL.Connection(self.ssl_ctx, conn)
            ##conn.setup_ssl()
            ##conn.set_connect_state()
            ##conn.set_session(self.sock.get_session())
            ##conn.connect_ssl()
        return conn, size
 
    def retrbinary(self, cmd, callback, blocksize=8192, rest=None):
        """Retrieve data in binary mode.

        `cmd' is a RETR command.  `callback' is a callback function is
        called for each block.  No more than `blocksize' number of
        bytes will be read from the socket.  Optional `rest' is passed
        to transfercmd().

        A new port is created for you.  Return the response code.
        """
        self.voidcmd('TYPE I')
        conn = self.transfercmd(cmd, rest)
        fp = conn.makefile('rb')
        while 1:
            #data = conn.recv(blocksize)
            data = fp.read()    #blocksize)
            if not data:
                break
            callback(data)
        fp.close()
        conn.close()
        return self.voidresp()

def test():
    server,port,user,pwd=map( raw_input, ('server','port','user','pwd') )   #'ftp.xxxx.xx',21,'yyyy','zzzz'
    verbose=1
    import StringIO
    f=StringIO.StringIO('aaaaaaaaaaaaaaa')
    ftp = FTP_TLS()
    ftp.set_debuglevel( verbose )
    ftp.connect(server, port)
    ftp.auth_tls();  ftp.prot_p()
    ftp.login(user,pwd)
    ftp.set_pasv( 1 )
    ftp.retrlines('LIST')
    ftp.storbinary("STOR file0.txt", f)

if __name__=='__main__':
    test()