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
|
# Copyright (C) 2005 by Canonical Ltd
# -*- coding: utf-8 -*-
# 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
"""Tests for bzr setting permissions.
Files in the branch control directory (.bzr or .bzr/branch) should inherit
the .bzr directory permissions.
So if the directory is group writable, the files and subdirs should be as well.
"""
# TODO: jam 20051215 Currently the default behavior for 'bzr branch' is just
# defined by the local umask. This isn't terrible, is it
# the truly desired behavior?
import os
import sys
import stat
from StringIO import StringIO
from bzrlib.branch import Branch
from bzrlib.bzrdir import BzrDir
from bzrlib.lockable_files import LockableFiles
from bzrlib.tests import TestCaseWithTransport, TestSkipped
from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer
from bzrlib.transport import get_transport
from bzrlib.workingtree import WorkingTree
# TODO RBC consolidate the helper methods here and in tests/test_permissions.py
def chmod_r(base, file_mode, dir_mode):
"""Recursively chmod from a base directory"""
assert os.path.isdir(base)
os.chmod(base, dir_mode)
for root, dirs, files in os.walk(base):
for d in dirs:
p = os.path.join(root, d)
os.chmod(p, dir_mode)
for f in files:
p = os.path.join(root, f)
os.chmod(p, file_mode)
def check_mode_r(test, base, file_mode, dir_mode, include_base=True):
"""Check that all permissions match
:param test: The TestCase being run
:param base: The path to the root directory to check
:param file_mode: The mode for all files
:param dir_mode: The mode for all directories
:param include_base: If false, only check the subdirectories
"""
assert os.path.isdir(base)
t = get_transport(".")
if include_base:
test.assertTransportMode(t, base, dir_mode)
for root, dirs, files in os.walk(base):
for d in dirs:
p = os.path.join(root, d)
test.assertTransportMode(t, p, dir_mode)
for f in files:
p = os.path.join(root, f)
test.assertTransportMode(t, p, file_mode)
class TestPermissions(TestCaseWithTransport):
def test_new_branch(self):
if sys.platform == 'win32':
raise TestSkipped('chmod has no effect on win32')
# also, these are BzrBranch format specific things..
os.mkdir('a')
mode = stat.S_IMODE(os.stat('a').st_mode)
t = self.make_branch_and_tree('.')
b = t.branch
self.assertEqualMode(mode, b.control_files._dir_mode)
self.assertEqualMode(mode & ~07111, b.control_files._file_mode)
os.mkdir('b')
os.chmod('b', 02777)
b = BzrDir.create('b').create_branch()
self.assertEqualMode(02777, b.control_files._dir_mode)
self.assertEqualMode(00666, b.control_files._file_mode)
check_mode_r(self, b.control_files._transport.base, 00666, 02777)
os.mkdir('c')
os.chmod('c', 02750)
b = BzrDir.create('c').create_branch()
self.assertEqualMode(02750, b.control_files._dir_mode)
self.assertEqualMode(00640, b.control_files._file_mode)
check_mode_r(self, b.control_files._transport.base, 00640, 02750)
os.mkdir('d')
os.chmod('d', 0700)
b = BzrDir.create('d').create_branch()
self.assertEqualMode(0700, b.control_files._dir_mode)
self.assertEqualMode(0600, b.control_files._file_mode)
check_mode_r(self, b.control_files._transport.base, 00600, 00700)
|