##// END OF EJS Templates
Backport PR #4720: never use ssh multiplexer in tunnels...
MinRK -
Show More
@@ -1,356 +1,356 b''
1 """Basic ssh tunnel utilities, and convenience functions for tunneling
1 """Basic ssh tunnel utilities, and convenience functions for tunneling
2 zeromq connections.
2 zeromq connections.
3
3
4 Authors
4 Authors
5 -------
5 -------
6 * Min RK
6 * Min RK
7 """
7 """
8
8
9 #-----------------------------------------------------------------------------
9 #-----------------------------------------------------------------------------
10 # Copyright (C) 2010-2011 The IPython Development Team
10 # Copyright (C) 2010-2011 The IPython Development Team
11 #
11 #
12 # Distributed under the terms of the BSD License. The full license is in
12 # Distributed under the terms of the BSD License. The full license is in
13 # the file COPYING, distributed as part of this software.
13 # the file COPYING, distributed as part of this software.
14 #-----------------------------------------------------------------------------
14 #-----------------------------------------------------------------------------
15
15
16
16
17
17
18 #-----------------------------------------------------------------------------
18 #-----------------------------------------------------------------------------
19 # Imports
19 # Imports
20 #-----------------------------------------------------------------------------
20 #-----------------------------------------------------------------------------
21
21
22 from __future__ import print_function
22 from __future__ import print_function
23
23
24 import os,sys, atexit
24 import os,sys, atexit
25 import signal
25 import signal
26 import socket
26 import socket
27 from multiprocessing import Process
27 from multiprocessing import Process
28 from getpass import getpass, getuser
28 from getpass import getpass, getuser
29 import warnings
29 import warnings
30
30
31 try:
31 try:
32 with warnings.catch_warnings():
32 with warnings.catch_warnings():
33 warnings.simplefilter('ignore', DeprecationWarning)
33 warnings.simplefilter('ignore', DeprecationWarning)
34 import paramiko
34 import paramiko
35 except ImportError:
35 except ImportError:
36 paramiko = None
36 paramiko = None
37 else:
37 else:
38 from forward import forward_tunnel
38 from forward import forward_tunnel
39
39
40 try:
40 try:
41 from IPython.external import pexpect
41 from IPython.external import pexpect
42 except ImportError:
42 except ImportError:
43 pexpect = None
43 pexpect = None
44
44
45 #-----------------------------------------------------------------------------
45 #-----------------------------------------------------------------------------
46 # Code
46 # Code
47 #-----------------------------------------------------------------------------
47 #-----------------------------------------------------------------------------
48
48
49 # select_random_ports copied from IPython.parallel.util
49 # select_random_ports copied from IPython.parallel.util
50 _random_ports = set()
50 _random_ports = set()
51
51
52 def select_random_ports(n):
52 def select_random_ports(n):
53 """Selects and return n random ports that are available."""
53 """Selects and return n random ports that are available."""
54 ports = []
54 ports = []
55 for i in xrange(n):
55 for i in xrange(n):
56 sock = socket.socket()
56 sock = socket.socket()
57 sock.bind(('', 0))
57 sock.bind(('', 0))
58 while sock.getsockname()[1] in _random_ports:
58 while sock.getsockname()[1] in _random_ports:
59 sock.close()
59 sock.close()
60 sock = socket.socket()
60 sock = socket.socket()
61 sock.bind(('', 0))
61 sock.bind(('', 0))
62 ports.append(sock)
62 ports.append(sock)
63 for i, sock in enumerate(ports):
63 for i, sock in enumerate(ports):
64 port = sock.getsockname()[1]
64 port = sock.getsockname()[1]
65 sock.close()
65 sock.close()
66 ports[i] = port
66 ports[i] = port
67 _random_ports.add(port)
67 _random_ports.add(port)
68 return ports
68 return ports
69
69
70
70
71 #-----------------------------------------------------------------------------
71 #-----------------------------------------------------------------------------
72 # Check for passwordless login
72 # Check for passwordless login
73 #-----------------------------------------------------------------------------
73 #-----------------------------------------------------------------------------
74
74
75 def try_passwordless_ssh(server, keyfile, paramiko=None):
75 def try_passwordless_ssh(server, keyfile, paramiko=None):
76 """Attempt to make an ssh connection without a password.
76 """Attempt to make an ssh connection without a password.
77 This is mainly used for requiring password input only once
77 This is mainly used for requiring password input only once
78 when many tunnels may be connected to the same server.
78 when many tunnels may be connected to the same server.
79
79
80 If paramiko is None, the default for the platform is chosen.
80 If paramiko is None, the default for the platform is chosen.
81 """
81 """
82 if paramiko is None:
82 if paramiko is None:
83 paramiko = sys.platform == 'win32'
83 paramiko = sys.platform == 'win32'
84 if not paramiko:
84 if not paramiko:
85 f = _try_passwordless_openssh
85 f = _try_passwordless_openssh
86 else:
86 else:
87 f = _try_passwordless_paramiko
87 f = _try_passwordless_paramiko
88 return f(server, keyfile)
88 return f(server, keyfile)
89
89
90 def _try_passwordless_openssh(server, keyfile):
90 def _try_passwordless_openssh(server, keyfile):
91 """Try passwordless login with shell ssh command."""
91 """Try passwordless login with shell ssh command."""
92 if pexpect is None:
92 if pexpect is None:
93 raise ImportError("pexpect unavailable, use paramiko")
93 raise ImportError("pexpect unavailable, use paramiko")
94 cmd = 'ssh -f '+ server
94 cmd = 'ssh -f '+ server
95 if keyfile:
95 if keyfile:
96 cmd += ' -i ' + keyfile
96 cmd += ' -i ' + keyfile
97 cmd += ' exit'
97 cmd += ' exit'
98 p = pexpect.spawn(cmd)
98 p = pexpect.spawn(cmd)
99 while True:
99 while True:
100 try:
100 try:
101 p.expect('[Pp]assword:', timeout=.1)
101 p.expect('[Pp]assword:', timeout=.1)
102 except pexpect.TIMEOUT:
102 except pexpect.TIMEOUT:
103 continue
103 continue
104 except pexpect.EOF:
104 except pexpect.EOF:
105 return True
105 return True
106 else:
106 else:
107 return False
107 return False
108
108
109 def _try_passwordless_paramiko(server, keyfile):
109 def _try_passwordless_paramiko(server, keyfile):
110 """Try passwordless login with paramiko."""
110 """Try passwordless login with paramiko."""
111 if paramiko is None:
111 if paramiko is None:
112 msg = "Paramiko unavaliable, "
112 msg = "Paramiko unavaliable, "
113 if sys.platform == 'win32':
113 if sys.platform == 'win32':
114 msg += "Paramiko is required for ssh tunneled connections on Windows."
114 msg += "Paramiko is required for ssh tunneled connections on Windows."
115 else:
115 else:
116 msg += "use OpenSSH."
116 msg += "use OpenSSH."
117 raise ImportError(msg)
117 raise ImportError(msg)
118 username, server, port = _split_server(server)
118 username, server, port = _split_server(server)
119 client = paramiko.SSHClient()
119 client = paramiko.SSHClient()
120 client.load_system_host_keys()
120 client.load_system_host_keys()
121 client.set_missing_host_key_policy(paramiko.WarningPolicy())
121 client.set_missing_host_key_policy(paramiko.WarningPolicy())
122 try:
122 try:
123 client.connect(server, port, username=username, key_filename=keyfile,
123 client.connect(server, port, username=username, key_filename=keyfile,
124 look_for_keys=True)
124 look_for_keys=True)
125 except paramiko.AuthenticationException:
125 except paramiko.AuthenticationException:
126 return False
126 return False
127 else:
127 else:
128 client.close()
128 client.close()
129 return True
129 return True
130
130
131
131
132 def tunnel_connection(socket, addr, server, keyfile=None, password=None, paramiko=None, timeout=60):
132 def tunnel_connection(socket, addr, server, keyfile=None, password=None, paramiko=None, timeout=60):
133 """Connect a socket to an address via an ssh tunnel.
133 """Connect a socket to an address via an ssh tunnel.
134
134
135 This is a wrapper for socket.connect(addr), when addr is not accessible
135 This is a wrapper for socket.connect(addr), when addr is not accessible
136 from the local machine. It simply creates an ssh tunnel using the remaining args,
136 from the local machine. It simply creates an ssh tunnel using the remaining args,
137 and calls socket.connect('tcp://localhost:lport') where lport is the randomly
137 and calls socket.connect('tcp://localhost:lport') where lport is the randomly
138 selected local port of the tunnel.
138 selected local port of the tunnel.
139
139
140 """
140 """
141 new_url, tunnel = open_tunnel(addr, server, keyfile=keyfile, password=password, paramiko=paramiko, timeout=timeout)
141 new_url, tunnel = open_tunnel(addr, server, keyfile=keyfile, password=password, paramiko=paramiko, timeout=timeout)
142 socket.connect(new_url)
142 socket.connect(new_url)
143 return tunnel
143 return tunnel
144
144
145
145
146 def open_tunnel(addr, server, keyfile=None, password=None, paramiko=None, timeout=60):
146 def open_tunnel(addr, server, keyfile=None, password=None, paramiko=None, timeout=60):
147 """Open a tunneled connection from a 0MQ url.
147 """Open a tunneled connection from a 0MQ url.
148
148
149 For use inside tunnel_connection.
149 For use inside tunnel_connection.
150
150
151 Returns
151 Returns
152 -------
152 -------
153
153
154 (url, tunnel): The 0MQ url that has been forwarded, and the tunnel object
154 (url, tunnel): The 0MQ url that has been forwarded, and the tunnel object
155 """
155 """
156
156
157 lport = select_random_ports(1)[0]
157 lport = select_random_ports(1)[0]
158 transport, addr = addr.split('://')
158 transport, addr = addr.split('://')
159 ip,rport = addr.split(':')
159 ip,rport = addr.split(':')
160 rport = int(rport)
160 rport = int(rport)
161 if paramiko is None:
161 if paramiko is None:
162 paramiko = sys.platform == 'win32'
162 paramiko = sys.platform == 'win32'
163 if paramiko:
163 if paramiko:
164 tunnelf = paramiko_tunnel
164 tunnelf = paramiko_tunnel
165 else:
165 else:
166 tunnelf = openssh_tunnel
166 tunnelf = openssh_tunnel
167
167
168 tunnel = tunnelf(lport, rport, server, remoteip=ip, keyfile=keyfile, password=password, timeout=timeout)
168 tunnel = tunnelf(lport, rport, server, remoteip=ip, keyfile=keyfile, password=password, timeout=timeout)
169 return 'tcp://127.0.0.1:%i'%lport, tunnel
169 return 'tcp://127.0.0.1:%i'%lport, tunnel
170
170
171 def openssh_tunnel(lport, rport, server, remoteip='127.0.0.1', keyfile=None, password=None, timeout=60):
171 def openssh_tunnel(lport, rport, server, remoteip='127.0.0.1', keyfile=None, password=None, timeout=60):
172 """Create an ssh tunnel using command-line ssh that connects port lport
172 """Create an ssh tunnel using command-line ssh that connects port lport
173 on this machine to localhost:rport on server. The tunnel
173 on this machine to localhost:rport on server. The tunnel
174 will automatically close when not in use, remaining open
174 will automatically close when not in use, remaining open
175 for a minimum of timeout seconds for an initial connection.
175 for a minimum of timeout seconds for an initial connection.
176
176
177 This creates a tunnel redirecting `localhost:lport` to `remoteip:rport`,
177 This creates a tunnel redirecting `localhost:lport` to `remoteip:rport`,
178 as seen from `server`.
178 as seen from `server`.
179
179
180 keyfile and password may be specified, but ssh config is checked for defaults.
180 keyfile and password may be specified, but ssh config is checked for defaults.
181
181
182 Parameters
182 Parameters
183 ----------
183 ----------
184
184
185 lport : int
185 lport : int
186 local port for connecting to the tunnel from this machine.
186 local port for connecting to the tunnel from this machine.
187 rport : int
187 rport : int
188 port on the remote machine to connect to.
188 port on the remote machine to connect to.
189 server : str
189 server : str
190 The ssh server to connect to. The full ssh server string will be parsed.
190 The ssh server to connect to. The full ssh server string will be parsed.
191 user@server:port
191 user@server:port
192 remoteip : str [Default: 127.0.0.1]
192 remoteip : str [Default: 127.0.0.1]
193 The remote ip, specifying the destination of the tunnel.
193 The remote ip, specifying the destination of the tunnel.
194 Default is localhost, which means that the tunnel would redirect
194 Default is localhost, which means that the tunnel would redirect
195 localhost:lport on this machine to localhost:rport on the *server*.
195 localhost:lport on this machine to localhost:rport on the *server*.
196
196
197 keyfile : str; path to public key file
197 keyfile : str; path to public key file
198 This specifies a key to be used in ssh login, default None.
198 This specifies a key to be used in ssh login, default None.
199 Regular default ssh keys will be used without specifying this argument.
199 Regular default ssh keys will be used without specifying this argument.
200 password : str;
200 password : str;
201 Your ssh password to the ssh server. Note that if this is left None,
201 Your ssh password to the ssh server. Note that if this is left None,
202 you will be prompted for it if passwordless key based login is unavailable.
202 you will be prompted for it if passwordless key based login is unavailable.
203 timeout : int [default: 60]
203 timeout : int [default: 60]
204 The time (in seconds) after which no activity will result in the tunnel
204 The time (in seconds) after which no activity will result in the tunnel
205 closing. This prevents orphaned tunnels from running forever.
205 closing. This prevents orphaned tunnels from running forever.
206 """
206 """
207 if pexpect is None:
207 if pexpect is None:
208 raise ImportError("pexpect unavailable, use paramiko_tunnel")
208 raise ImportError("pexpect unavailable, use paramiko_tunnel")
209 ssh="ssh "
209 ssh="ssh "
210 if keyfile:
210 if keyfile:
211 ssh += "-i " + keyfile
211 ssh += "-i " + keyfile
212
212
213 if ':' in server:
213 if ':' in server:
214 server, port = server.split(':')
214 server, port = server.split(':')
215 ssh += " -p %s" % port
215 ssh += " -p %s" % port
216
216
217 cmd = "%s -f -L 127.0.0.1:%i:%s:%i %s sleep %i" % (
217 cmd = "%s -f -S none -L 127.0.0.1:%i:%s:%i %s sleep %i" % (
218 ssh, lport, remoteip, rport, server, timeout)
218 ssh, lport, remoteip, rport, server, timeout)
219 tunnel = pexpect.spawn(cmd)
219 tunnel = pexpect.spawn(cmd)
220 failed = False
220 failed = False
221 while True:
221 while True:
222 try:
222 try:
223 tunnel.expect('[Pp]assword:', timeout=.1)
223 tunnel.expect('[Pp]assword:', timeout=.1)
224 except pexpect.TIMEOUT:
224 except pexpect.TIMEOUT:
225 continue
225 continue
226 except pexpect.EOF:
226 except pexpect.EOF:
227 if tunnel.exitstatus:
227 if tunnel.exitstatus:
228 print (tunnel.exitstatus)
228 print (tunnel.exitstatus)
229 print (tunnel.before)
229 print (tunnel.before)
230 print (tunnel.after)
230 print (tunnel.after)
231 raise RuntimeError("tunnel '%s' failed to start"%(cmd))
231 raise RuntimeError("tunnel '%s' failed to start"%(cmd))
232 else:
232 else:
233 return tunnel.pid
233 return tunnel.pid
234 else:
234 else:
235 if failed:
235 if failed:
236 print("Password rejected, try again")
236 print("Password rejected, try again")
237 password=None
237 password=None
238 if password is None:
238 if password is None:
239 password = getpass("%s's password: "%(server))
239 password = getpass("%s's password: "%(server))
240 tunnel.sendline(password)
240 tunnel.sendline(password)
241 failed = True
241 failed = True
242
242
243 def _split_server(server):
243 def _split_server(server):
244 if '@' in server:
244 if '@' in server:
245 username,server = server.split('@', 1)
245 username,server = server.split('@', 1)
246 else:
246 else:
247 username = getuser()
247 username = getuser()
248 if ':' in server:
248 if ':' in server:
249 server, port = server.split(':')
249 server, port = server.split(':')
250 port = int(port)
250 port = int(port)
251 else:
251 else:
252 port = 22
252 port = 22
253 return username, server, port
253 return username, server, port
254
254
255 def paramiko_tunnel(lport, rport, server, remoteip='127.0.0.1', keyfile=None, password=None, timeout=60):
255 def paramiko_tunnel(lport, rport, server, remoteip='127.0.0.1', keyfile=None, password=None, timeout=60):
256 """launch a tunner with paramiko in a subprocess. This should only be used
256 """launch a tunner with paramiko in a subprocess. This should only be used
257 when shell ssh is unavailable (e.g. Windows).
257 when shell ssh is unavailable (e.g. Windows).
258
258
259 This creates a tunnel redirecting `localhost:lport` to `remoteip:rport`,
259 This creates a tunnel redirecting `localhost:lport` to `remoteip:rport`,
260 as seen from `server`.
260 as seen from `server`.
261
261
262 If you are familiar with ssh tunnels, this creates the tunnel:
262 If you are familiar with ssh tunnels, this creates the tunnel:
263
263
264 ssh server -L localhost:lport:remoteip:rport
264 ssh server -L localhost:lport:remoteip:rport
265
265
266 keyfile and password may be specified, but ssh config is checked for defaults.
266 keyfile and password may be specified, but ssh config is checked for defaults.
267
267
268
268
269 Parameters
269 Parameters
270 ----------
270 ----------
271
271
272 lport : int
272 lport : int
273 local port for connecting to the tunnel from this machine.
273 local port for connecting to the tunnel from this machine.
274 rport : int
274 rport : int
275 port on the remote machine to connect to.
275 port on the remote machine to connect to.
276 server : str
276 server : str
277 The ssh server to connect to. The full ssh server string will be parsed.
277 The ssh server to connect to. The full ssh server string will be parsed.
278 user@server:port
278 user@server:port
279 remoteip : str [Default: 127.0.0.1]
279 remoteip : str [Default: 127.0.0.1]
280 The remote ip, specifying the destination of the tunnel.
280 The remote ip, specifying the destination of the tunnel.
281 Default is localhost, which means that the tunnel would redirect
281 Default is localhost, which means that the tunnel would redirect
282 localhost:lport on this machine to localhost:rport on the *server*.
282 localhost:lport on this machine to localhost:rport on the *server*.
283
283
284 keyfile : str; path to public key file
284 keyfile : str; path to public key file
285 This specifies a key to be used in ssh login, default None.
285 This specifies a key to be used in ssh login, default None.
286 Regular default ssh keys will be used without specifying this argument.
286 Regular default ssh keys will be used without specifying this argument.
287 password : str;
287 password : str;
288 Your ssh password to the ssh server. Note that if this is left None,
288 Your ssh password to the ssh server. Note that if this is left None,
289 you will be prompted for it if passwordless key based login is unavailable.
289 you will be prompted for it if passwordless key based login is unavailable.
290 timeout : int [default: 60]
290 timeout : int [default: 60]
291 The time (in seconds) after which no activity will result in the tunnel
291 The time (in seconds) after which no activity will result in the tunnel
292 closing. This prevents orphaned tunnels from running forever.
292 closing. This prevents orphaned tunnels from running forever.
293
293
294 """
294 """
295 if paramiko is None:
295 if paramiko is None:
296 raise ImportError("Paramiko not available")
296 raise ImportError("Paramiko not available")
297
297
298 if password is None:
298 if password is None:
299 if not _try_passwordless_paramiko(server, keyfile):
299 if not _try_passwordless_paramiko(server, keyfile):
300 password = getpass("%s's password: "%(server))
300 password = getpass("%s's password: "%(server))
301
301
302 p = Process(target=_paramiko_tunnel,
302 p = Process(target=_paramiko_tunnel,
303 args=(lport, rport, server, remoteip),
303 args=(lport, rport, server, remoteip),
304 kwargs=dict(keyfile=keyfile, password=password))
304 kwargs=dict(keyfile=keyfile, password=password))
305 p.daemon=False
305 p.daemon=False
306 p.start()
306 p.start()
307 atexit.register(_shutdown_process, p)
307 atexit.register(_shutdown_process, p)
308 return p
308 return p
309
309
310 def _shutdown_process(p):
310 def _shutdown_process(p):
311 if p.is_alive():
311 if p.is_alive():
312 p.terminate()
312 p.terminate()
313
313
314 def _paramiko_tunnel(lport, rport, server, remoteip, keyfile=None, password=None):
314 def _paramiko_tunnel(lport, rport, server, remoteip, keyfile=None, password=None):
315 """Function for actually starting a paramiko tunnel, to be passed
315 """Function for actually starting a paramiko tunnel, to be passed
316 to multiprocessing.Process(target=this), and not called directly.
316 to multiprocessing.Process(target=this), and not called directly.
317 """
317 """
318 username, server, port = _split_server(server)
318 username, server, port = _split_server(server)
319 client = paramiko.SSHClient()
319 client = paramiko.SSHClient()
320 client.load_system_host_keys()
320 client.load_system_host_keys()
321 client.set_missing_host_key_policy(paramiko.WarningPolicy())
321 client.set_missing_host_key_policy(paramiko.WarningPolicy())
322
322
323 try:
323 try:
324 client.connect(server, port, username=username, key_filename=keyfile,
324 client.connect(server, port, username=username, key_filename=keyfile,
325 look_for_keys=True, password=password)
325 look_for_keys=True, password=password)
326 # except paramiko.AuthenticationException:
326 # except paramiko.AuthenticationException:
327 # if password is None:
327 # if password is None:
328 # password = getpass("%s@%s's password: "%(username, server))
328 # password = getpass("%s@%s's password: "%(username, server))
329 # client.connect(server, port, username=username, password=password)
329 # client.connect(server, port, username=username, password=password)
330 # else:
330 # else:
331 # raise
331 # raise
332 except Exception as e:
332 except Exception as e:
333 print ('*** Failed to connect to %s:%d: %r' % (server, port, e))
333 print ('*** Failed to connect to %s:%d: %r' % (server, port, e))
334 sys.exit(1)
334 sys.exit(1)
335
335
336 # Don't let SIGINT kill the tunnel subprocess
336 # Don't let SIGINT kill the tunnel subprocess
337 signal.signal(signal.SIGINT, signal.SIG_IGN)
337 signal.signal(signal.SIGINT, signal.SIG_IGN)
338
338
339 try:
339 try:
340 forward_tunnel(lport, remoteip, rport, client.get_transport())
340 forward_tunnel(lport, remoteip, rport, client.get_transport())
341 except KeyboardInterrupt:
341 except KeyboardInterrupt:
342 print ('SIGINT: Port forwarding stopped cleanly')
342 print ('SIGINT: Port forwarding stopped cleanly')
343 sys.exit(0)
343 sys.exit(0)
344 except Exception as e:
344 except Exception as e:
345 print ("Port forwarding stopped uncleanly: %s"%e)
345 print ("Port forwarding stopped uncleanly: %s"%e)
346 sys.exit(255)
346 sys.exit(255)
347
347
348 if sys.platform == 'win32':
348 if sys.platform == 'win32':
349 ssh_tunnel = paramiko_tunnel
349 ssh_tunnel = paramiko_tunnel
350 else:
350 else:
351 ssh_tunnel = openssh_tunnel
351 ssh_tunnel = openssh_tunnel
352
352
353
353
354 __all__ = ['tunnel_connection', 'ssh_tunnel', 'openssh_tunnel', 'paramiko_tunnel', 'try_passwordless_ssh']
354 __all__ = ['tunnel_connection', 'ssh_tunnel', 'openssh_tunnel', 'paramiko_tunnel', 'try_passwordless_ssh']
355
355
356
356
General Comments 0
You need to be logged in to leave comments. Login now