28
import SocketServer as socketserver
35
from bzrlib.transport import (
37
from ..transport import (
38
from bzrlib.tests import test_server
41
class StubServer (paramiko.ServerInterface):
43
def __init__(self, test_case):
40
from . import test_server
43
class StubServer(paramiko.ServerInterface):
45
def __init__(self, test_case_server):
44
46
paramiko.ServerInterface.__init__(self)
45
self._test_case = test_case
47
self.log = test_case_server.log
47
49
def check_auth_password(self, username, password):
49
self._test_case.log('sftpserver - authorizing: %s' % (username,))
51
self.log('sftpserver - authorizing: %s' % (username,))
50
52
return paramiko.AUTH_SUCCESSFUL
52
54
def check_channel_request(self, kind, chanid):
54
'sftpserver - channel request: %s, %s' % (kind, chanid))
55
self.log('sftpserver - channel request: %s, %s' % (kind, chanid))
55
56
return paramiko.OPEN_SUCCEEDED
58
class StubSFTPHandle (paramiko.SFTPHandle):
59
class StubSFTPHandle(paramiko.SFTPHandle):
61
63
return paramiko.SFTPAttributes.from_stat(
62
64
os.fstat(self.readfile.fileno()))
64
66
return paramiko.SFTPServer.convert_errno(e.errno)
66
68
def chattr(self, attr):
265
class SocketListener(threading.Thread):
267
def __init__(self, callback):
268
threading.Thread.__init__(self)
269
self._callback = callback
270
self._socket = socket.socket()
271
self._socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
272
self._socket.bind(('localhost', 0))
273
self._socket.listen(1)
274
self.host, self.port = self._socket.getsockname()[:2]
275
self._stop_event = threading.Event()
278
# called from outside this thread
279
self._stop_event.set()
280
# use a timeout here, because if the test fails, the server thread may
281
# never notice the stop_event.
287
readable, writable_unused, exception_unused = \
288
select.select([self._socket], [], [], 0.1)
289
if self._stop_event.isSet():
291
if len(readable) == 0:
294
s, addr_unused = self._socket.accept()
295
# because the loopback socket is inline, and transports are
296
# never explicitly closed, best to launch a new thread.
297
threading.Thread(target=self._callback, args=(s,)).start()
298
except socket.error, x:
299
sys.excepthook(*sys.exc_info())
300
trace.warning('Socket error during accept() '
301
'within unit test server thread: %r' % x)
303
# probably a failed test; unit test thread will log the
305
sys.excepthook(*sys.exc_info())
307
'Exception from within unit test server thread: %r' % x)
310
284
class SocketDelay(object):
311
285
"""A socket decorator to make TCP appear slower.
382
356
return bytes_sent
385
class SFTPServer(test_server.TestServer):
359
class TestingSFTPConnectionHandler(socketserver.BaseRequestHandler):
362
self.wrap_for_latency()
363
tcs = self.server.test_case_server
364
ptrans = paramiko.Transport(self.request)
365
self.paramiko_transport = ptrans
366
# Set it to a channel under 'bzr' so that we get debug info
367
ptrans.set_log_channel('brz.paramiko.transport')
368
ptrans.add_server_key(tcs.get_host_key())
369
ptrans.set_subsystem_handler('sftp', paramiko.SFTPServer,
370
StubSFTPServer, root=tcs._root,
371
home=tcs._server_homedir)
372
server = tcs._server_interface(tcs)
373
# This blocks until the key exchange has been done
374
ptrans.start_server(None, server)
377
# Wait for the conversation to finish, when the paramiko.Transport
379
# TODO: Consider timing out after XX seconds rather than hanging.
380
# Also we could check paramiko_transport.active and possibly
381
# paramiko_transport.getException().
382
self.paramiko_transport.join()
384
def wrap_for_latency(self):
385
tcs = self.server.test_case_server
387
# Give the socket (which the request really is) a latency adding
389
self.request = SocketDelay(self.request, tcs.add_latency)
392
class TestingSFTPWithoutSSHConnectionHandler(TestingSFTPConnectionHandler):
395
self.wrap_for_latency()
396
# Re-import these as locals, so that they're still accessible during
397
# interpreter shutdown (when all module globals get set to None, leading
398
# to confusing errors like "'NoneType' object has no attribute 'error'".
399
class FakeChannel(object):
400
def get_transport(self):
402
def get_log_channel(self):
403
return 'brz.paramiko'
406
def get_hexdump(self):
411
tcs = self.server.test_case_server
412
sftp_server = paramiko.SFTPServer(
413
FakeChannel(), 'sftp', StubServer(tcs), StubSFTPServer,
414
root=tcs._root, home=tcs._server_homedir)
415
self.sftp_server = sftp_server
416
sys_stderr = sys.stderr # Used in error reporting during shutdown
418
sftp_server.start_subsystem(
419
'sftp', None, ssh.SocketAsChannelAdapter(self.request))
420
except socket.error as e:
421
if (len(e.args) > 0) and (e.args[0] == errno.EPIPE):
422
# it's okay for the client to disconnect abruptly
423
# (bug in paramiko 1.6: it should absorb this exception)
427
except Exception as e:
428
# This typically seems to happen during interpreter shutdown, so
429
# most of the useful ways to report this error won't work.
430
# Writing the exception type, and then the text of the exception,
431
# seems to be the best we can do.
432
# FIXME: All interpreter shutdown errors should have been related
433
# to daemon threads, cleanup needed -- vila 20100623
434
sys_stderr.write('\nEXCEPTION %r: ' % (e.__class__,))
435
sys_stderr.write('%s\n\n' % (e,))
438
self.sftp_server.finish_subsystem()
441
class TestingSFTPServer(test_server.TestingThreadingTCPServer):
443
def __init__(self, server_address, request_handler_class, test_case_server):
444
test_server.TestingThreadingTCPServer.__init__(
445
self, server_address, request_handler_class)
446
self.test_case_server = test_case_server
449
class SFTPServer(test_server.TestingTCPServerInAThread):
386
450
"""Common code for SFTP server facilities."""
388
452
def __init__(self, server_interface=StubServer):
453
self.host = '127.0.0.1'
455
super(SFTPServer, self).__init__((self.host, self.port),
457
TestingSFTPConnectionHandler)
389
458
self._original_vendor = None
391
self._server_homedir = None
392
self._listener = None
394
459
self._vendor = ssh.ParamikoVendor()
395
460
self._server_interface = server_interface
461
self._host_key = None
398
463
self.add_latency = 0
465
self._server_homedir = None
400
468
def _get_sftp_url(self, path):
401
469
"""Calculate an sftp url to this server for path."""
402
return 'sftp://foo:bar@%s:%d/%s' % (self._listener.host,
403
self._listener.port, path)
470
return "sftp://foo:bar@%s:%s/%s" % (self.host, self.port, path)
405
472
def log(self, message):
406
473
"""StubServer uses this to log when a new server is created."""
407
474
self.logs.append(message)
409
def _run_server_entry(self, sock):
410
"""Entry point for all implementations of _run_server.
412
If self.add_latency is > 0.000001 then sock is given a latency adding
415
if self.add_latency > 0.000001:
416
sock = SocketDelay(sock, self.add_latency)
417
return self._run_server(sock)
419
def _run_server(self, s):
420
ssh_server = paramiko.Transport(s)
421
key_file = osutils.pathjoin(self._homedir, 'test_rsa.key')
422
f = open(key_file, 'w')
423
f.write(STUB_SERVER_KEY)
425
host_key = paramiko.RSAKey.from_private_key_file(key_file)
426
ssh_server.add_server_key(host_key)
427
server = self._server_interface(self)
428
ssh_server.set_subsystem_handler('sftp', paramiko.SFTPServer,
429
StubSFTPServer, root=self._root,
430
home=self._server_homedir)
431
event = threading.Event()
432
ssh_server.start_server(event, server)
476
def create_server(self):
477
server = self.server_class((self.host, self.port),
478
self.request_handler_class,
482
def get_host_key(self):
483
if self._host_key is None:
484
key_file = osutils.pathjoin(self._homedir, 'test_rsa.key')
485
f = open(key_file, 'w')
487
f.write(STUB_SERVER_KEY)
490
self._host_key = paramiko.RSAKey.from_private_key_file(key_file)
491
return self._host_key
435
493
def start_server(self, backing_server=None):
436
494
# XXX: TODO: make sftpserver back onto backing_server rather than local
442
500
'the local current working directory.' % (backing_server,))
443
501
self._original_vendor = ssh._ssh_vendor_manager._cached_ssh_vendor
444
502
ssh._ssh_vendor_manager._cached_ssh_vendor = self._vendor
445
# FIXME: the following block should certainly just be self._homedir =
446
# osutils.getcwd() but that fails badly on Unix -- vila 20100224
503
self._homedir = osutils.getcwd()
447
504
if sys.platform == 'win32':
448
# Win32 needs to use the UNICODE api
449
self._homedir = os.getcwdu()
505
# Normalize the path or it will be wrongly escaped
506
self._homedir = osutils.normpath(self._homedir)
451
# But Linux SFTP servers should just deal in bytestreams
452
self._homedir = os.getcwd()
508
self._homedir = self._homedir
453
509
if self._server_homedir is None:
454
510
self._server_homedir = self._homedir
456
512
if sys.platform == 'win32':
458
self._listener = SocketListener(self._run_server_entry)
459
self._listener.setDaemon(True)
460
self._listener.start()
514
super(SFTPServer, self).start_server()
462
516
def stop_server(self):
463
self._listener.stop()
464
ssh._ssh_vendor_manager._cached_ssh_vendor = self._original_vendor
518
super(SFTPServer, self).stop_server()
520
ssh._ssh_vendor_manager._cached_ssh_vendor = self._original_vendor
466
522
def get_bogus_url(self):
467
"""See bzrlib.transport.Server.get_bogus_url."""
468
# this is chosen to try to prevent trouble with proxies, wierd dns, etc
523
"""See breezy.transport.Server.get_bogus_url."""
524
# this is chosen to try to prevent trouble with proxies, weird dns, etc
469
525
# we bind a random socket, so that we get a guaranteed unused port
470
526
# we just never listen on that port
471
527
s = socket.socket()
491
547
def __init__(self):
492
548
super(SFTPServerWithoutSSH, self).__init__()
493
549
self._vendor = ssh.LoopbackVendor()
495
def _run_server(self, sock):
496
# Re-import these as locals, so that they're still accessible during
497
# interpreter shutdown (when all module globals get set to None, leading
498
# to confusing errors like "'NoneType' object has no attribute 'error'".
499
class FakeChannel(object):
500
def get_transport(self):
502
def get_log_channel(self):
506
def get_hexdump(self):
511
server = paramiko.SFTPServer(
512
FakeChannel(), 'sftp', StubServer(self), StubSFTPServer,
513
root=self._root, home=self._server_homedir)
515
server.start_subsystem(
516
'sftp', None, ssh.SocketAsChannelAdapter(sock))
517
except socket.error, e:
518
if (len(e.args) > 0) and (e.args[0] == errno.EPIPE):
519
# it's okay for the client to disconnect abruptly
520
# (bug in paramiko 1.6: it should absorb this exception)
525
# This typically seems to happen during interpreter shutdown, so
526
# most of the useful ways to report this error are won't work.
527
# Writing the exception type, and then the text of the exception,
528
# seems to be the best we can do.
530
sys.stderr.write('\nEXCEPTION %r: ' % (e.__class__,))
531
sys.stderr.write('%s\n\n' % (e,))
532
server.finish_subsystem()
550
self.request_handler_class = TestingSFTPWithoutSSHConnectionHandler
535
556
class SFTPAbsoluteServer(SFTPServerWithoutSSH):
536
557
"""A test server for sftp transports, using absolute urls."""
538
559
def get_url(self):
539
"""See bzrlib.transport.Server.get_url."""
560
"""See breezy.transport.Server.get_url."""
540
561
homedir = self._homedir
541
562
if sys.platform != 'win32':
542
563
# Remove the initial '/' on all platforms but win32