← Back to branch summary

~ubuntu-branches/ubuntu/wily/pyzmq/wily

« back to all changes in this revision

Viewing changes to zmq/ssh/forward.py

  • Committer: Package Import Robot
  • Author(s): Julian Taylor
  • Date: 2013-02-24 19:23:15 UTC
  • mfrom: (1.2.1) (9 sid)
  • mto: This revision was merged to the branch mainline in revision 10.
  • Revision ID: package-import@ubuntu.com-20130224192315-qhmwp3m3ymk8r60d
Tags: 2.2.0.1-1
* New upstream release
* relicense debian packaging to LGPL-3
* update watch file to use github directly
  thanks to Bart Martens for the file
* add autopkgtests
* drop obsolete DM-Upload-Allowed
* bump standard to 3.9.4, no changes required

Show diffs side-by-side

added added

removed removed

Lines of Context:
zmq/ssh/forward.py
 
1
#------------------------------------------------------------------------------
 
2
#
 
3
# This file is adapted from a paramiko demo, and thus licensed under LGPL 2.1.
 
4
# Original Copyright (C) 2003-2007  Robey Pointer <robeypointer@gmail.com>
 
5
# Edits Copyright (C) 2010 The IPython Team
 
6
#
 
7
# Paramiko is free software; you can redistribute it and/or modify it under the
 
8
# terms of the GNU Lesser General Public License as published by the Free
 
9
# Software Foundation; either version 2.1 of the License, or (at your option)
 
10
# any later version.
 
11
#
 
12
# Paramiko is distrubuted in the hope that it will be useful, but WITHOUT ANY
 
13
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
 
14
# A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
 
15
# details.
 
16
#
 
17
# You should have received a copy of the GNU Lesser General Public License
 
18
# along with Paramiko; if not, write to the Free Software Foundation, Inc.,
 
19
# 51 Franklin Street, Fifth Floor, Boston, MA  02111-1301  USA.
 
20
#------------------------------------------------------------------------------
 
21
 
 
22
"""
 
23
Sample script showing how to do local port forwarding over paramiko.
 
24
 
 
25
This script connects to the requested SSH server and sets up local port
 
26
forwarding (the openssh -L option) from a local port through a tunneled
 
27
connection to a destination reachable from the SSH server machine.
 
28
"""
 
29
 
 
30
import sys
 
31
import logging
 
32
import select
 
33
import SocketServer
 
34
 
 
35
logger = logging.getLogger('ssh')
 
36
 
 
37
class ForwardServer (SocketServer.ThreadingTCPServer):
 
38
    daemon_threads = True
 
39
    allow_reuse_address = True
 
40
    
 
41
 
 
42
class Handler (SocketServer.BaseRequestHandler):
 
43
 
 
44
    def handle(self):
 
45
        try:
 
46
            chan = self.ssh_transport.open_channel('direct-tcpip',
 
47
                                                   (self.chain_host, self.chain_port),
 
48
                                                   self.request.getpeername())
 
49
        except Exception as e:
 
50
            logger.debug('Incoming request to %s:%d failed: %s' % (self.chain_host,
 
51
                                                              self.chain_port,
 
52
                                                              repr(e)))
 
53
            return
 
54
        if chan is None:
 
55
            logger.debug('Incoming request to %s:%d was rejected by the SSH server.' %
 
56
                    (self.chain_host, self.chain_port))
 
57
            return
 
58
 
 
59
        logger.debug('Connected!  Tunnel open %r -> %r -> %r' % (self.request.getpeername(),
 
60
                                                            chan.getpeername(), (self.chain_host, self.chain_port)))
 
61
        while True:
 
62
            r, w, x = select.select([self.request, chan], [], [])
 
63
            if self.request in r:
 
64
                data = self.request.recv(1024)
 
65
                if len(data) == 0:
 
66
                    break
 
67
                chan.send(data)
 
68
            if chan in r:
 
69
                data = chan.recv(1024)
 
70
                if len(data) == 0:
 
71
                    break
 
72
                self.request.send(data)
 
73
        chan.close()
 
74
        self.request.close()
 
75
        logger.debug('Tunnel closed ')
 
76
 
 
77
 
 
78
def forward_tunnel(local_port, remote_host, remote_port, transport):
 
79
    # this is a little convoluted, but lets me configure things for the Handler
 
80
    # object.  (SocketServer doesn't give Handlers any way to access the outer
 
81
    # server normally.)
 
82
    class SubHander (Handler):
 
83
        chain_host = remote_host
 
84
        chain_port = remote_port
 
85
        ssh_transport = transport
 
86
    ForwardServer(('127.0.0.1', local_port), SubHander).serve_forever()
 
87
 
 
88
 
 
89
__all__ = ['forward_tunnel']