1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
|
# Copyright (C) 2007 Canonical Ltd
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
"""
FTP test server.
Based on medusa: http://www.amk.ca/python/code/medusa.html
"""
import asyncore
import errno
import os
import select
import stat
import threading
import medusa
import medusa.filesys
import medusa.ftp_server
from bzrlib import (
tests,
trace,
transport,
)
class test_authorizer(object):
"""A custom Authorizer object for running the test suite.
The reason we cannot use dummy_authorizer, is because it sets the
channel to readonly, which we don't always want to do.
"""
def __init__(self, root):
self.root = root
# If secured_user is set secured_password will be checked
self.secured_user = None
self.secured_password = None
def authorize(self, channel, username, password):
"""Return (success, reply_string, filesystem)"""
channel.persona = -1, -1
if username == 'anonymous':
channel.read_only = 1
else:
channel.read_only = 0
# Check secured_user if set
if (self.secured_user is not None
and username == self.secured_user
and password != self.secured_password):
return 0, 'Password invalid.', None
else:
return 1, 'OK.', medusa.filesys.os_filesystem(self.root)
class ftp_channel(medusa.ftp_server.ftp_channel):
"""Customized ftp channel"""
def log(self, message):
"""Redirect logging requests."""
trace.mutter('ftp_channel: %s', message)
def log_info(self, message, type='info'):
"""Redirect logging requests."""
trace.mutter('ftp_channel %s: %s', type, message)
def cmd_rnfr(self, line):
"""Prepare for renaming a file."""
self._renaming = line[1]
self.respond('350 Ready for RNTO')
# TODO: jam 20060516 in testing, the ftp server seems to
# check that the file already exists, or it sends
# 550 RNFR command failed
def cmd_rnto(self, line):
"""Rename a file based on the target given.
rnto must be called after calling rnfr.
"""
if not self._renaming:
self.respond('503 RNFR required first.')
pfrom = self.filesystem.translate(self._renaming)
self._renaming = None
pto = self.filesystem.translate(line[1])
if os.path.exists(pto):
self.respond('550 RNTO failed: file exists')
return
try:
os.rename(pfrom, pto)
except (IOError, OSError), e:
# TODO: jam 20060516 return custom responses based on
# why the command failed
# (bialix 20070418) str(e) on Python 2.5 @ Windows
# sometimes don't provide expected error message;
# so we obtain such message via os.strerror()
self.respond('550 RNTO failed: %s' % os.strerror(e.errno))
except:
self.respond('550 RNTO failed')
# For a test server, we will go ahead and just die
raise
else:
self.respond('250 Rename successful.')
def cmd_size(self, line):
"""Return the size of a file
This is overloaded to help the test suite determine if the
target is a directory.
"""
filename = line[1]
if not self.filesystem.isfile(filename):
if self.filesystem.isdir(filename):
self.respond('550 "%s" is a directory' % (filename,))
else:
self.respond('550 "%s" is not a file' % (filename,))
else:
self.respond('213 %d'
% (self.filesystem.stat(filename)[stat.ST_SIZE]),)
def cmd_mkd(self, line):
"""Create a directory.
Overloaded because default implementation does not distinguish
*why* it cannot make a directory.
"""
if len (line) != 2:
self.command_not_understood(''.join(line))
else:
path = line[1]
try:
self.filesystem.mkdir (path)
self.respond ('257 MKD command successful.')
except (IOError, OSError), e:
# (bialix 20070418) str(e) on Python 2.5 @ Windows
# sometimes don't provide expected error message;
# so we obtain such message via os.strerror()
self.respond ('550 error creating directory: %s' %
os.strerror(e.errno))
except:
self.respond ('550 error creating directory.')
class ftp_server(medusa.ftp_server.ftp_server):
"""Customize the behavior of the Medusa ftp_server.
There are a few warts on the ftp_server, based on how it expects
to be used.
"""
_renaming = None
ftp_channel_class = ftp_channel
def __init__(self, *args, **kwargs):
trace.mutter('Initializing ftp_server: %r, %r', args, kwargs)
medusa.ftp_server.ftp_server.__init__(self, *args, **kwargs)
def log(self, message):
"""Redirect logging requests."""
trace.mutter('ftp_server: %s', message)
def log_info(self, message, type='info'):
"""Override the asyncore.log_info so we don't stipple the screen."""
trace.mutter('ftp_server %s: %s', type, message)
class FTPServer(transport.Server):
"""Common code for FTP server facilities."""
def __init__(self):
self._root = None
self._ftp_server = None
self._port = None
self._async_thread = None
# ftp server logs
self.logs = []
def get_url(self):
"""Calculate an ftp url to this server."""
return 'ftp://foo:bar@localhost:%d/' % (self._port)
# def get_bogus_url(self):
# """Return a URL which cannot be connected to."""
# return 'ftp://127.0.0.1:1'
def log(self, message):
"""This is used by medusa.ftp_server to log connections, etc."""
self.logs.append(message)
def setUp(self, vfs_server=None):
from bzrlib.transport.local import LocalURLServer
assert vfs_server is None or isinstance(vfs_server, LocalURLServer), \
"FTPServer currently assumes local transport, got %s" % vfs_server
self._root = os.getcwdu()
self._ftp_server = ftp_server(
authorizer=test_authorizer(root=self._root),
ip='localhost',
port=0, # bind to a random port
resolver=None,
logger_object=self # Use FTPServer.log() for messages
)
self._port = self._ftp_server.getsockname()[1]
# Don't let it loop forever, or handle an infinite number of requests.
# In this case it will run for 1000s, or 10000 requests
self._async_thread = threading.Thread(
target=FTPServer._asyncore_loop_ignore_EBADF,
kwargs={'timeout':0.1, 'count':10000})
self._async_thread.setDaemon(True)
self._async_thread.start()
def tearDown(self):
"""See bzrlib.transport.Server.tearDown."""
self._ftp_server.close()
asyncore.close_all()
self._async_thread.join()
@staticmethod
def _asyncore_loop_ignore_EBADF(*args, **kwargs):
"""Ignore EBADF during server shutdown.
We close the socket to get the server to shutdown, but this causes
select.select() to raise EBADF.
"""
try:
asyncore.loop(*args, **kwargs)
# FIXME: If we reach that point, we should raise an exception
# explaining that the 'count' parameter in setUp is too low or
# testers may wonder why their test just sits there waiting for a
# server that is already dead. Note that if the tester waits too
# long under pdb the server will also die.
except select.error, e:
if e.args[0] != errno.EBADF:
raise
|