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
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
 
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
19
19
"""Tests for bzr setting permissions.
 
 
28
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
 
 
29
# TODO: jam 20051215 Currently the default behavior for 'bzr branch' is just 
 
30
30
#                    defined by the local umask. This isn't terrible, is it
 
31
31
#                    the truly desired behavior?
 
 
39
39
from bzrlib.branch import Branch
 
40
40
from bzrlib.bzrdir import BzrDir
 
 
41
from bzrlib.lockable_files import LockableFiles, TransportLock
 
41
42
from bzrlib.tests import TestCaseWithTransport, TestSkipped
 
42
43
from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer
 
43
44
from bzrlib.transport import get_transport
 
 
47
48
def chmod_r(base, file_mode, dir_mode):
 
48
49
    """Recursively chmod from a base directory"""
 
 
50
    assert os.path.isdir(base)
 
49
51
    os.chmod(base, dir_mode)
 
50
52
    for root, dirs, files in os.walk(base):
 
 
65
67
    :param dir_mode: The mode for all directories
 
66
68
    :param include_base: If false, only check the subdirectories
 
 
70
    assert os.path.isdir(base)
 
68
71
    t = get_transport(".")
 
70
73
        test.assertTransportMode(t, base, dir_mode)
 
 
97
100
        # although we are modifying the filesystem
 
98
101
        # underneath the objects, they are not locked, and thus it must
 
99
 
        # be safe for most operations. But here we want to observe a
 
 
102
        # be safe for most operations. But here we want to observe a 
 
100
103
        # mode change in the control bits, which current do not refresh
 
101
104
        # when a new lock is taken out.
 
102
105
        t = WorkingTree.open('.')
 
104
107
        self.assertEqualMode(0755, b.control_files._dir_mode)
 
105
108
        self.assertEqualMode(0644, b.control_files._file_mode)
 
106
 
        self.assertEqualMode(0755, b.bzrdir._get_dir_mode())
 
107
 
        self.assertEqualMode(0644, b.bzrdir._get_file_mode())
 
109
110
        # Modifying a file shouldn't break the permissions
 
110
111
        open('a', 'wb').write('foo2\n')
 
 
126
127
        self.assertEqualMode(0775, b.control_files._dir_mode)
 
127
128
        self.assertEqualMode(0664, b.control_files._file_mode)
 
128
 
        self.assertEqualMode(0775, b.bzrdir._get_dir_mode())
 
129
 
        self.assertEqualMode(0664, b.bzrdir._get_file_mode())
 
131
130
        open('a', 'wb').write('foo3\n')
 
 
137
136
        t.commit('new c')
 
138
137
        check_mode_r(self, '.bzr', 0664, 0775)
 
140
 
    def test_new_files_group_sticky_bit(self):
 
141
 
        if sys.platform == 'win32':
 
142
 
            raise TestSkipped('chmod has no effect on win32')
 
143
 
        elif sys.platform == 'darwin' or sys.platform.startswith('freebsd'):
 
144
 
            # OS X (and FreeBSD) create temp dirs with the 'wheel' group, which
 
145
 
            # users are not likely to be in, and this prevents us from setting
 
147
 
            os.chown(self.test_dir, os.getuid(), os.getgid())
 
149
 
        t = self.make_branch_and_tree('.')
 
152
139
        # Test the group sticky bit
 
153
140
        # Recursively update the modes of all files
 
154
141
        chmod_r('.bzr', 0664, 02775)
 
 
158
145
        self.assertEqualMode(02775, b.control_files._dir_mode)
 
159
146
        self.assertEqualMode(0664, b.control_files._file_mode)
 
160
 
        self.assertEqualMode(02775, b.bzrdir._get_dir_mode())
 
161
 
        self.assertEqualMode(0664, b.bzrdir._get_file_mode())
 
163
148
        open('a', 'wb').write('foo4\n')
 
 
169
154
        t.commit('new d')
 
170
155
        check_mode_r(self, '.bzr', 0664, 02775)
 
 
157
    def test_disable_set_mode(self):
 
 
158
        # TODO: jam 20051215 Ultimately, this test should probably test that
 
 
159
        #                    extra chmod calls aren't being made
 
 
161
            transport = get_transport(self.get_url())
 
 
162
            transport.put_bytes('my-lock', '')
 
 
163
            lockable = LockableFiles(transport, 'my-lock', TransportLock)
 
 
164
            self.assertNotEqual(None, lockable._dir_mode)
 
 
165
            self.assertNotEqual(None, lockable._file_mode)
 
 
167
            LockableFiles._set_dir_mode = False
 
 
168
            transport = get_transport('.')
 
 
169
            lockable = LockableFiles(transport, 'my-lock', TransportLock)
 
 
170
            self.assertEqual(None, lockable._dir_mode)
 
 
171
            self.assertNotEqual(None, lockable._file_mode)
 
 
173
            LockableFiles._set_file_mode = False
 
 
174
            transport = get_transport('.')
 
 
175
            lockable = LockableFiles(transport, 'my-lock', TransportLock)
 
 
176
            self.assertEqual(None, lockable._dir_mode)
 
 
177
            self.assertEqual(None, lockable._file_mode)
 
 
179
            LockableFiles._set_dir_mode = True
 
 
180
            transport = get_transport('.')
 
 
181
            lockable = LockableFiles(transport, 'my-lock', TransportLock)
 
 
182
            self.assertNotEqual(None, lockable._dir_mode)
 
 
183
            self.assertEqual(None, lockable._file_mode)
 
 
185
            LockableFiles._set_file_mode = True
 
 
186
            transport = get_transport('.')
 
 
187
            lockable = LockableFiles(transport, 'my-lock', TransportLock)
 
 
188
            self.assertNotEqual(None, lockable._dir_mode)
 
 
189
            self.assertNotEqual(None, lockable._file_mode)
 
 
191
            LockableFiles._set_dir_mode = True
 
 
192
            LockableFiles._set_file_mode = True
 
173
195
class TestSftpPermissions(TestCaseWithSFTPServer):
 
 
197
219
        b_local = t.branch
 
198
220
        self.assertEqualMode(0755, b_local.control_files._dir_mode)
 
199
221
        self.assertEqualMode(0644, b_local.control_files._file_mode)
 
200
 
        self.assertEqualMode(0755, b_local.bzrdir._get_dir_mode())
 
201
 
        self.assertEqualMode(0644, b_local.bzrdir._get_file_mode())
 
204
224
        sftp_url = self.get_url('sftp')
 
 
212
232
        b_sftp = Branch.open(sftp_url)
 
213
233
        self.assertEqualMode(0755, b_sftp.control_files._dir_mode)
 
214
234
        self.assertEqualMode(0644, b_sftp.control_files._file_mode)
 
215
 
        self.assertEqualMode(0755, b_sftp.bzrdir._get_dir_mode())
 
216
 
        self.assertEqualMode(0644, b_sftp.bzrdir._get_file_mode())
 
218
236
        open('local/a', 'wb').write('foo2\n')
 
219
237
        t_local.commit('foo2')
 
 
235
253
        b_sftp = Branch.open(sftp_url)
 
236
254
        self.assertEqualMode(0775, b_sftp.control_files._dir_mode)
 
237
255
        self.assertEqualMode(0664, b_sftp.control_files._file_mode)
 
238
 
        self.assertEqualMode(0775, b_sftp.bzrdir._get_dir_mode())
 
239
 
        self.assertEqualMode(0664, b_sftp.bzrdir._get_file_mode())
 
241
257
        open('local/a', 'wb').write('foo3\n')
 
242
258
        t_local.commit('foo3')
 
 
266
282
            t.put_bytes('b', 'txt', mode=0666)
 
267
283
            self.assertTransportMode(t, 'b', 0666)
 
269
 
            t._get_sftp().mkdir('c', mode=0777)
 
 
285
            t._sftp.mkdir('c', mode=0777)
 
270
286
            self.assertTransportMode(t, 'c', 0777 &~umask)
 
272
288
            t.mkdir('d', mode=0777)