1
# Copyright (C) 2005 by Canonical Ltd
1
# Copyright (C) 2005 Canonical Ltd
2
2
# -*- coding: utf-8 -*-
4
4
# This program is free software; you can redistribute it and/or modify
5
5
# it under the terms of the GNU General Public License as published by
6
6
# the Free Software Foundation; either version 2 of the License, or
7
7
# (at your option) any later version.
9
9
# This program is distributed in the hope that it will be useful,
10
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12
12
# GNU General Public License for more details.
14
14
# You should have received a copy of the GNU General Public License
15
15
# along with this program; if not, write to the Free Software
16
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
24
24
In the future, when we have Repository/Branch/Checkout information, the
25
25
permissions should be inherited individually, rather than all be the same.
27
TODO: jam 20051215 There are no tests for ftp yet, because we have no ftp server
28
TODO: jam 20051215 Currently the default behavior for 'bzr branch' is just
29
defined by the local umask. This isn't terrible, is it
30
the truly desired behavior?
28
# TODO: jam 20051215 There are no tests for ftp yet, because we have no ftp server
29
# TODO: jam 20051215 Currently the default behavior for 'bzr branch' is just
30
# defined by the local umask. This isn't terrible, is it
31
# the truly desired behavior?
36
from cStringIO import StringIO
37
39
from bzrlib.branch import Branch
38
from bzrlib.tests import TestCaseInTempDir, TestSkipped
40
from bzrlib.bzrdir import BzrDir
41
from bzrlib.lockable_files import LockableFiles, TransportLock
42
from bzrlib.tests import TestCaseWithTransport, TestSkipped
39
43
from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer
40
from bzrlib.tests.test_transport import check_mode
44
from bzrlib.transport import get_transport
45
from bzrlib.workingtree import WorkingTree
43
48
def chmod_r(base, file_mode, dir_mode):
44
49
"""Recursively chmod from a base directory"""
45
assert os.path.isdir(base)
46
50
os.chmod(base, dir_mode)
47
51
for root, dirs, files in os.walk(base):
62
66
:param dir_mode: The mode for all directories
63
67
:param include_base: If false, only check the subdirectories
65
assert os.path.isdir(base)
69
t = get_transport(".")
67
check_mode(test, base, dir_mode)
71
test.assertTransportMode(t, base, dir_mode)
68
72
for root, dirs, files in os.walk(base):
70
p = os.path.join(root, d)
71
check_mode(test, p, dir_mode)
74
p = '/'.join([urllib.quote(x) for x in root.split('/\\') + [d]])
75
test.assertTransportMode(t, p, dir_mode)
73
77
p = os.path.join(root, f)
74
check_mode(test, p, file_mode)
77
def assertEqualMode(test, mode, mode_test):
78
test.assertEqual(mode, mode_test,
79
'mode mismatch %o != %o' % (mode, mode_test))
82
class TestPermissions(TestCaseInTempDir):
78
p = '/'.join([urllib.quote(x) for x in root.split('/\\') + [f]])
79
test.assertTransportMode(t, p, file_mode)
82
class TestPermissions(TestCaseWithTransport):
84
84
def test_new_files(self):
85
85
if sys.platform == 'win32':
86
86
raise TestSkipped('chmod has no effect on win32')
88
b = Branch.initialize(u'.')
88
t = self.make_branch_and_tree('.')
90
90
open('a', 'wb').write('foo\n')
91
# ensure check_mode_r works with capital-letter file-ids like TREE_ROOT
94
# Delete them because we are modifying the filesystem underneath them
96
95
chmod_r('.bzr', 0644, 0755)
97
96
check_mode_r(self, '.bzr', 0644, 0755)
101
assertEqualMode(self, 0755, b.control_files._dir_mode)
102
assertEqualMode(self, 0644, b.control_files._file_mode)
98
# although we are modifying the filesystem
99
# underneath the objects, they are not locked, and thus it must
100
# be safe for most operations. But here we want to observe a
101
# mode change in the control bits, which current do not refresh
102
# when a new lock is taken out.
103
t = WorkingTree.open('.')
105
self.assertEqualMode(0755, b.control_files._dir_mode)
106
self.assertEqualMode(0644, b.control_files._file_mode)
107
self.assertEqualMode(0755, b.bzrdir._get_dir_mode())
108
self.assertEqualMode(0644, b.bzrdir._get_file_mode())
104
110
# Modifying a file shouldn't break the permissions
105
111
open('a', 'wb').write('foo2\n')
113
119
t.commit('new b')
114
120
check_mode_r(self, '.bzr', 0644, 0755)
117
122
# Recursively update the modes of all files
118
123
chmod_r('.bzr', 0664, 0775)
119
124
check_mode_r(self, '.bzr', 0664, 0775)
122
assertEqualMode(self, 0775, b.control_files._dir_mode)
123
assertEqualMode(self, 0664, b.control_files._file_mode)
125
t = WorkingTree.open('.')
127
self.assertEqualMode(0775, b.control_files._dir_mode)
128
self.assertEqualMode(0664, b.control_files._file_mode)
129
self.assertEqualMode(0775, b.bzrdir._get_dir_mode())
130
self.assertEqualMode(0664, b.bzrdir._get_file_mode())
125
132
open('a', 'wb').write('foo3\n')
131
138
t.commit('new c')
132
139
check_mode_r(self, '.bzr', 0664, 0775)
141
def test_new_files_group_sticky_bit(self):
142
if sys.platform == 'win32':
143
raise TestSkipped('chmod has no effect on win32')
144
elif sys.platform == 'darwin':
145
# OS X creates temp dirs with the 'wheel' group, which users are
146
# not likely to be in, and this prevents us from setting the sgid
148
os.chown(self.test_dir, os.getuid(), os.getgid())
150
t = self.make_branch_and_tree('.')
134
153
# Test the group sticky bit
136
154
# Recursively update the modes of all files
137
155
chmod_r('.bzr', 0664, 02775)
138
156
check_mode_r(self, '.bzr', 0664, 02775)
141
assertEqualMode(self, 02775, b.control_files._dir_mode)
142
assertEqualMode(self, 0664, b.control_files._file_mode)
157
t = WorkingTree.open('.')
159
self.assertEqualMode(02775, b.control_files._dir_mode)
160
self.assertEqualMode(0664, b.control_files._file_mode)
161
self.assertEqualMode(02775, b.bzrdir._get_dir_mode())
162
self.assertEqualMode(0664, b.bzrdir._get_file_mode())
144
164
open('a', 'wb').write('foo4\n')
150
170
t.commit('new d')
151
171
check_mode_r(self, '.bzr', 0664, 02775)
153
def test_disable_set_mode(self):
154
# TODO: jam 20051215 Ultimately, this test should probably test that
155
# extra chmod calls aren't being made
156
import bzrlib.lockable_files
158
b = Branch.initialize(u'.')
159
self.assertNotEqual(None, b.control_files._dir_mode)
160
self.assertNotEqual(None, b.control_files._file_mode)
162
bzrlib.lockable_files.LockableFiles._set_dir_mode = False
163
b = Branch.open(u'.')
164
self.assertEqual(None, b.control_files._dir_mode)
165
self.assertNotEqual(None, b.control_files._file_mode)
167
bzrlib.lockable_files.LockableFiles._set_file_mode = False
168
b = Branch.open(u'.')
169
self.assertEqual(None, b.control_files._dir_mode)
170
self.assertEqual(None, b.control_files._file_mode)
172
bzrlib.lockable_files.LockableFiles._set_dir_mode = True
173
b = Branch.open(u'.')
174
self.assertNotEqual(None, b.control_files._dir_mode)
175
self.assertEqual(None, b.control_files._file_mode)
177
bzrlib.lockable_files.LockableFiles._set_file_mode = True
178
b = Branch.open(u'.')
179
self.assertNotEqual(None, b.control_files._dir_mode)
180
self.assertNotEqual(None, b.control_files._file_mode)
182
bzrlib.lockable_files.LockableFiles._set_dir_mode = True
183
bzrlib.lockable_files.LockableFiles._set_file_mode = True
185
def test_new_branch(self):
186
if sys.platform == 'win32':
187
raise TestSkipped('chmod has no effect on win32')
190
mode = stat.S_IMODE(os.stat('a').st_mode)
191
b = Branch.initialize('a')
192
assertEqualMode(self, mode, b.control_files._dir_mode)
193
assertEqualMode(self, mode & ~07111, b.control_files._file_mode)
197
b = Branch.initialize('b')
198
assertEqualMode(self, 02777, b.control_files._dir_mode)
199
assertEqualMode(self, 00666, b.control_files._file_mode)
200
check_mode_r(self, 'b/.bzr', 00666, 02777)
204
b = Branch.initialize('c')
205
assertEqualMode(self, 02750, b.control_files._dir_mode)
206
assertEqualMode(self, 00640, b.control_files._file_mode)
207
check_mode_r(self, 'c/.bzr', 00640, 02750)
211
b = Branch.initialize('d')
212
assertEqualMode(self, 0700, b.control_files._dir_mode)
213
assertEqualMode(self, 0600, b.control_files._file_mode)
214
check_mode_r(self, 'd/.bzr', 00600, 0700)
217
174
class TestSftpPermissions(TestCaseWithSFTPServer):
222
179
# Though it would be nice to test that SFTP to a server
223
180
# which does support chmod has the right effect
225
from bzrlib.transport.sftp import SFTPTransport
227
# We don't actually use it directly, we just want to
228
# keep the connection open, since StubSFTPServer only
229
# allows 1 connection
231
_transport = SFTPTransport(self._sftp_url)
182
# bodge around for stubsftpserver not letting use connect
184
_t = get_transport(self.get_url())
233
186
os.mkdir('local')
234
b_local = Branch.initialize(u'local')
235
t_local = b_local.working_tree()
187
t_local = self.make_branch_and_tree('local')
188
b_local = t_local.branch
236
189
open('local/a', 'wb').write('foo\n')
238
191
t_local.commit('foo')
240
193
# Delete them because we are modifying the filesystem underneath them
242
194
chmod_r('local/.bzr', 0644, 0755)
243
195
check_mode_r(self, 'local/.bzr', 0644, 0755)
245
b_local = Branch.open(u'local')
246
t_local = b_local.working_tree()
247
assertEqualMode(self, 0755, b_local.control_files._dir_mode)
248
assertEqualMode(self, 0644, b_local.control_files._file_mode)
197
t = WorkingTree.open('local')
199
self.assertEqualMode(0755, b_local.control_files._dir_mode)
200
self.assertEqualMode(0644, b_local.control_files._file_mode)
201
self.assertEqualMode(0755, b_local.bzrdir._get_dir_mode())
202
self.assertEqualMode(0644, b_local.bzrdir._get_file_mode())
251
# Why does self._sftp_url end with a slash????
252
sftp_url = self._sftp_url + 'sftp'
253
b_sftp = Branch.initialize(sftp_url)
205
sftp_url = self.get_url('sftp')
206
b_sftp = BzrDir.create_branch_and_repo(sftp_url)
255
208
b_sftp.pull(b_local)
258
211
check_mode_r(self, 'sftp/.bzr', 0644, 0755)
260
213
b_sftp = Branch.open(sftp_url)
261
assertEqualMode(self, 0755, b_sftp.control_files._dir_mode)
262
assertEqualMode(self, 0644, b_sftp.control_files._file_mode)
214
self.assertEqualMode(0755, b_sftp.control_files._dir_mode)
215
self.assertEqualMode(0644, b_sftp.control_files._file_mode)
216
self.assertEqualMode(0755, b_sftp.bzrdir._get_dir_mode())
217
self.assertEqualMode(0644, b_sftp.bzrdir._get_file_mode())
264
219
open('local/a', 'wb').write('foo2\n')
265
220
t_local.commit('foo2')
279
234
check_mode_r(self, 'sftp/.bzr', 0664, 0775)
281
236
b_sftp = Branch.open(sftp_url)
282
assertEqualMode(self, 0775, b_sftp.control_files._dir_mode)
283
assertEqualMode(self, 0664, b_sftp.control_files._file_mode)
237
self.assertEqualMode(0775, b_sftp.control_files._dir_mode)
238
self.assertEqualMode(0664, b_sftp.control_files._file_mode)
239
self.assertEqualMode(0775, b_sftp.bzrdir._get_dir_mode())
240
self.assertEqualMode(0664, b_sftp.bzrdir._get_file_mode())
285
242
open('local/a', 'wb').write('foo3\n')
286
243
t_local.commit('foo3')
301
258
original_umask = os.umask(umask)
304
from bzrlib.transport.sftp import SFTPTransport
306
t = SFTPTransport(self._sftp_url)
261
t = get_transport(self.get_url())
307
262
# Direct access should be masked by umask
308
263
t._sftp_open_exclusive('a', mode=0666).write('foo\n')
309
check_mode(self, 'a', 0666 &~umask)
264
self.assertTransportMode(t, 'a', 0666 &~umask)
311
266
# but Transport overrides umask
312
t.put('b', 'txt', mode=0666)
313
check_mode(self, 'b', 0666)
267
t.put_bytes('b', 'txt', mode=0666)
268
self.assertTransportMode(t, 'b', 0666)
315
t._sftp.mkdir('c', mode=0777)
316
check_mode(self, 'c', 0777 &~umask)
270
t._get_sftp().mkdir('c', mode=0777)
271
self.assertTransportMode(t, 'c', 0777 &~umask)
318
273
t.mkdir('d', mode=0777)
319
check_mode(self, 'd', 0777)
274
self.assertTransportMode(t, 'd', 0777)
321
276
os.umask(original_umask)