From 3e31771637cb87e2cae2092121562b47ea13cd6e Mon Sep 17 00:00:00 2001
From: Robey Pointer <robey@lag.net>
Date: Thu, 4 Mar 2004 08:21:45 +0000
Subject: [project @
 Arch-1:robey@lag.net--2003-public%secsh--dev--1.0--patch-27] add BufferedFile
 abstraction SFTP client mode is mostly functional.  there are probably still
 some bugs but most of the operations on "file" objects have survived my
 simple tests.

BufferedFile wraps a simpler stream in something that looks like a python
file (and can even handle seeking if the stream underneath supports it).
it's meant to be subclassed.  most of it is ripped out of what used to be
ChannelFile so i can reuse it for sftp -- ChannelFile is now tiny.

SFTP and Message are now exported.

fixed util.format_binary_line to not quote spaces.
---
 demo_sftp.py | 157 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 157 insertions(+)
 create mode 100755 demo_sftp.py

(limited to 'demo_sftp.py')

diff --git a/demo_sftp.py b/demo_sftp.py
new file mode 100755
index 00000000..1f6cd3a6
--- /dev/null
+++ b/demo_sftp.py
@@ -0,0 +1,157 @@
+#!/usr/bin/python
+
+import sys, os, socket, threading, getpass, logging, time, base64, select, termios, tty, traceback
+import paramiko
+
+
+#####   utility functions
+
+def load_host_keys():
+    filename = os.environ['HOME'] + '/.ssh/known_hosts'
+    keys = {}
+    try:
+        f = open(filename, 'r')
+    except Exception, e:
+        print '*** Unable to open host keys file (%s)' % filename
+        return
+    for line in f:
+        keylist = line.split(' ')
+        if len(keylist) != 3:
+            continue
+        hostlist, keytype, key = keylist
+        hosts = hostlist.split(',')
+        for host in hosts:
+            if not keys.has_key(host):
+                keys[host] = {}
+            keys[host][keytype] = base64.decodestring(key)
+    f.close()
+    return keys
+
+
+#####   main demo
+
+# setup logging
+l = logging.getLogger("paramiko")
+l.setLevel(logging.DEBUG)
+if len(l.handlers) == 0:
+    f = open('demo.log', 'w')
+    lh = logging.StreamHandler(f)
+    lh.setFormatter(logging.Formatter('%(levelname)-.3s [%(asctime)s] %(name)s: %(message)s', '%Y%m%d:%H%M%S'))
+    l.addHandler(lh)
+
+
+username = ''
+if len(sys.argv) > 1:
+    hostname = sys.argv[1]
+    if hostname.find('@') >= 0:
+        username, hostname = hostname.split('@')
+else:
+    hostname = raw_input('Hostname: ')
+if len(hostname) == 0:
+    print '*** Hostname required.'
+    sys.exit(1)
+port = 22
+if hostname.find(':') >= 0:
+    hostname, portstr = hostname.split(':')
+    port = int(portstr)
+
+# now connect
+try:
+    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    sock.connect((hostname, port))
+except Exception, e:
+    print '*** Connect failed: ' + str(e)
+    traceback.print_exc()
+    sys.exit(1)
+
+try:
+    event = threading.Event()
+    t = paramiko.Transport(sock)
+    t.start_client(event)
+    # print repr(t)
+    event.wait(15)
+    if not t.is_active():
+        print '*** SSH negotiation failed.'
+        sys.exit(1)
+    # print repr(t)
+
+    keys = load_host_keys()
+    keytype, hostkey = t.get_remote_server_key()
+    if not keys.has_key(hostname):
+        print '*** WARNING: Unknown host key!'
+    elif not keys[hostname].has_key(keytype):
+        print '*** WARNING: Unknown host key!'
+    elif keys[hostname][keytype] != hostkey:
+        print '*** WARNING: Host key has changed!!!'
+        sys.exit(1)
+    else:
+        print '*** Host key OK.'
+
+    event.clear()
+
+    # get username
+    if username == '':
+        default_username = getpass.getuser()
+        username = raw_input('Username [%s]: ' % default_username)
+        if len(username) == 0:
+            username = default_username
+
+    # ask for what kind of authentication to try
+    default_auth = 'p'
+    auth = raw_input('Auth by (p)assword, (r)sa key, or (d)ss key? [%s] ' % default_auth)
+    if len(auth) == 0:
+        auth = default_auth
+
+    if auth == 'r':
+        key = paramiko.RSAKey()
+        default_path = os.environ['HOME'] + '/.ssh/id_rsa'
+        path = raw_input('RSA key [%s]: ' % default_path)
+        if len(path) == 0:
+            path = default_path
+        try:
+            key.read_private_key_file(path)
+        except paramiko.PasswordRequiredException:
+            password = getpass.getpass('RSA key password: ')
+            key.read_private_key_file(path, password)
+        t.auth_publickey(username, key, event)
+    elif auth == 'd':
+        key = paramiko.DSSKey()
+        default_path = os.environ['HOME'] + '/.ssh/id_dsa'
+        path = raw_input('DSS key [%s]: ' % default_path)
+        if len(path) == 0:
+            path = default_path
+        try:
+            key.read_private_key_file(path)
+        except paramiko.PasswordRequiredException:
+            password = getpass.getpass('DSS key password: ')
+            key.read_private_key_file(path, password)
+        t.auth_key(username, key, event)
+    else:
+        pw = getpass.getpass('Password for %s@%s: ' % (username, hostname))
+        t.auth_password(username, pw, event)
+
+    event.wait(10)
+    # print repr(t)
+    if not t.is_authenticated():
+        print '*** Authentication failed. :('
+        t.close()
+        sys.exit(1)
+
+    chan = t.open_session()
+    chan.invoke_subsystem('sftp')
+    print '*** SFTP...'
+    sftp = paramiko.SFTP(chan)
+    print repr(sftp.listdir('/tmp'))
+
+    chan.close()
+    t.close()
+
+except Exception, e:
+    print '*** Caught exception: ' + str(e.__class__) + ': ' + str(e)
+    traceback.print_exc()
+    try:
+        t.close()
+    except:
+        pass
+    sys.exit(1)
+
-- 
cgit v1.2.3