python: ntacls, fix a leftover that is not in the try/except branch
[samba.git] / source4 / scripting / python / samba / ntacls.py
1 #!/usr/bin/python
2
3 # Unix SMB/CIFS implementation.
4 # Copyright (C) Matthieu Patou <mat@matws.net> 2009-2010
5 #
6 #
7 # This program is free software; you can redistribute it and/or modify
8 # it under the terms of the GNU General Public License as published by
9 # the Free Software Foundation; either version 3 of the License, or
10 # (at your option) any later version.
11 #
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 # GNU General Public License for more details.
16 #
17 # You should have received a copy of the GNU General Public License
18 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
19 #
20
21 import os
22 import tdb
23 import samba.xattr_native, samba.xattr_tdb
24 from samba.dcerpc import security, xattr
25 from samba.ndr import ndr_pack, ndr_unpack
26 class XattrBackendError(Exception):
27     """A generic xattr backend error."""
28
29 def checkset_backend(lp,backend,eadbfile):
30         if backend != None:
31                 if backend == "native":
32                         lp.set("posix:eadb","")
33                 elif backend == "tdb":
34                         if eadbfile != None:
35                                 lp.set("posix:eadb",eadbfile)
36                         else:
37                                 os.path.abspath(os.path.join(lp.get("private dir"),"eadb.tdb"))
38                 else:
39                         raise XattrBackendError("Unvalid xattr backend choice %s"%backend)
40
41 def getntacl(lp,file,backend=None,eadbfile=None):
42         try:
43                 checkset_backend(lp,backend,eadbfile)
44         except:
45                 raise
46         eadbname = lp.get("posix:eadb")
47         if eadbname != None and eadbname != "" :
48                 try:
49                         attribute = samba.xattr_tdb.wrap_getxattr(eadbname,file,xattr.XATTR_NTACL_NAME)
50                 except:
51                         print "Fail to open %s"%eadbname
52                         attribute = samba.xattr_native.wrap_getxattr(file,xattr.XATTR_NTACL_NAME)
53         else:
54                 attribute = samba.xattr_native.wrap_getxattr(file,xattr.XATTR_NTACL_NAME)
55         ntacl = ndr_unpack(xattr.NTACL,attribute)
56         return ntacl
57
58 def setntacl(lp,file,sddl,domsid,backend=None,eadbfile=None):
59         try:
60                 checkset_backend(lp,backend,eadbfile)
61         except:
62                 raise
63         ntacl=xattr.NTACL()
64         ntacl.version = 1
65         sid=security.dom_sid(domsid)
66         sd = security.descriptor.from_sddl(sddl, sid)
67         ntacl.info = sd
68         eadbname = lp.get("posix:eadb")
69         if eadbname != None  and eadbname != "":
70                 try:
71                         attribute = samba.xattr_tdb.wrap_setxattr(eadbname,file,xattr.XATTR_NTACL_NAME,ndr_pack(ntacl))
72                 except:
73                         print "Fail to open %s"%eadbname
74                         attribute = samba.xattr_native.wrap_setxattr(file,xattr.XATTR_NTACL_NAME,ndr_pack(ntacl))
75         else:
76                 attribute = samba.xattr_native.wrap_setxattr(file,xattr.XATTR_NTACL_NAME,ndr_pack(ntacl))
77
78 # Takes the access mask of a DS ACE and transform them in a File ACE mask
79 def ldapmask2filemask(ldm):
80         RIGHT_DS_CREATE_CHILD     = 0x00000001
81         RIGHT_DS_DELETE_CHILD     = 0x00000002
82         RIGHT_DS_LIST_CONTENTS    = 0x00000004
83         ACTRL_DS_SELF             = 0x00000008
84         RIGHT_DS_READ_PROPERTY    = 0x00000010
85         RIGHT_DS_WRITE_PROPERTY   = 0x00000020
86         RIGHT_DS_DELETE_TREE      = 0x00000040
87         RIGHT_DS_LIST_OBJECT      = 0x00000080
88         RIGHT_DS_CONTROL_ACCESS   = 0x00000100
89         FILE_READ_DATA            = 0x0001
90         FILE_LIST_DIRECTORY       = 0x0001
91         FILE_WRITE_DATA           = 0x0002
92         FILE_ADD_FILE             = 0x0002
93         FILE_APPEND_DATA          = 0x0004
94         FILE_ADD_SUBDIRECTORY     = 0x0004
95         FILE_CREATE_PIPE_INSTANCE = 0x0004
96         FILE_READ_EA              = 0x0008
97         FILE_WRITE_EA             = 0x0010
98         FILE_EXECUTE              = 0x0020
99         FILE_TRAVERSE             = 0x0020
100         FILE_DELETE_CHILD         = 0x0040
101         FILE_READ_ATTRIBUTES      = 0x0080
102         FILE_WRITE_ATTRIBUTES     = 0x0100
103         DELETE                    = 0x00010000
104         READ_CONTROL              = 0x00020000
105         WRITE_DAC                 = 0x00040000
106         WRITE_OWNER               = 0x00080000
107         SYNCHRONIZE               = 0x00100000
108         STANDARD_RIGHTS_ALL       = 0x001F0000
109
110         filemask = ldm & STANDARD_RIGHTS_ALL
111         #filemask = 0
112
113         if( (ldm & RIGHT_DS_READ_PROPERTY) and (ldm & RIGHT_DS_LIST_CONTENTS) ):
114                 filemask = filemask | (SYNCHRONIZE | FILE_LIST_DIRECTORY |\
115                                                                 FILE_READ_ATTRIBUTES | FILE_READ_EA |\
116                                                                 FILE_READ_DATA | FILE_EXECUTE)
117
118         if( (ldm & RIGHT_DS_WRITE_PROPERTY) ):
119                 filemask = filemask | (SYNCHRONIZE | FILE_WRITE_DATA |\
120                                                                 FILE_APPEND_DATA | FILE_WRITE_EA |\
121                                                                 FILE_WRITE_ATTRIBUTES | FILE_ADD_FILE |\
122                                                                 FILE_ADD_SUBDIRECTORY)
123
124         if( (ldm & RIGHT_DS_CREATE_CHILD) ):
125                 filemask = filemask | (FILE_ADD_SUBDIRECTORY | FILE_ADD_FILE)
126
127         if( (ldm & RIGHT_DS_DELETE_CHILD) ):
128                 filemask = filemask | FILE_DELETE_CHILD
129
130         return filemask
131
132 # This function takes an the SDDL representation of a DS
133 # ACL and return the SDDL representation of this ACL adapted
134 # for files. It's used for Policy object provision
135
136 def dsacl2fsacl(dssddl,domsid):
137         sid = security.dom_sid(domsid)
138         ref = security.descriptor.from_sddl(dssddl,sid)
139         fdescr = security.descriptor()
140         fdescr.owner_sid = ref.owner_sid
141         fdescr.group_sid = ref.group_sid
142         fdescr.type = ref.type
143         fdescr.revision = ref.revision
144         fdescr.sacl = ref.sacl
145         aces = ref.dacl.aces
146         for i in range(0,len(aces)):
147                 ace = aces[i]
148                 if not ace.type &  security.SEC_ACE_TYPE_ACCESS_ALLOWED_OBJECT and str(ace.trustee) != security.SID_BUILTIN_PREW2K:
149                 #       if fdescr.type & security.SEC_DESC_DACL_AUTO_INHERITED:
150                         ace.flags = ace.flags | security.SEC_ACE_FLAG_OBJECT_INHERIT | security.SEC_ACE_FLAG_CONTAINER_INHERIT
151                         if str(ace.trustee) == security.SID_CREATOR_OWNER:
152                                 # For Creator/Owner the IO flag is set as this ACE has only a sense for child objects
153                                 ace.flags = ace.flags | security.SEC_ACE_FLAG_INHERIT_ONLY
154                         ace.access_mask =  ldapmask2filemask(ace.access_mask)
155                         fdescr.dacl_add(ace)
156
157         return fdescr.as_sddl(sid)