ba83dcf7e91ab608ec6d4dbf0692d3e4eaec35e2
[nivanova/samba-autobuild/.git] / source4 / scripting / bin / samba_gpoupdate
1 #!/usr/bin/env python
2 # Copyright Luke Morrison <luc785@.hotmail.com> July 2013
3 # Co-Edited by Matthieu Pattou July 2013 from original August 2013
4 # Edited by Garming Sam Feb. 2014
5 # Edited by Luke Morrison April 2014
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 '''This script reads a log file of previous GPO, gets all GPO from sysvol
21 and sorts them by container. Then, it applies the ones that haven't been
22 applied, have changed, or is in the right container'''
23
24 import os
25 import fcntl
26 import sys
27 import tempfile
28 import subprocess
29
30 sys.path.insert(0, "bin/python")
31
32 import samba
33 import optparse
34 from samba import getopt as options
35 from samba.gpclass import *
36 from samba.net import Net
37 from samba.dcerpc import nbt
38 from samba import smb
39
40
41 # Finds all GPO Files ending in inf
42 def gp_path_list(path):
43
44     GPO_LIST = []
45     for ext in gp_extensions:
46         GPO_LIST.append((ext, ext.list(path)))
47     return GPO_LIST
48
49
50 def gpo_parser(GPO_LIST, ldb, conn, attr_log):
51     '''The API method to parse the GPO
52     :param GPO_LIST:
53     :param ldb: Live instance of an LDB object AKA Samba
54     :param conn: Live instance of a CIFS connection
55     :param attr_log: backlog path for GPO and attribute to be written
56     no return except a newly updated Samba
57     '''
58
59     ret = False
60     for entry in GPO_LIST:
61         (ext, thefile) = entry
62         if ret == False:
63             ret = ext.parse(thefile, ldb, conn, attr_log)
64         else:
65             temp = ext.parse(thefile, ldb, conn, attr_log)
66     return ret
67
68
69 class GPOServiceSetup:
70     def __init__(self):
71         """Initialize all components necessary to return instances of
72         a Samba lp context (smb.conf) and Samba LDB context
73         """
74
75         self.parser = optparse.OptionParser("samba_gpoupdate [options]")
76         self.sambaopts = options.SambaOptions(self.parser)
77         self.credopts = None
78         self.opts = None
79         self.args = None
80         self.lp = None
81         self.smbconf = None
82         self.creds = None
83         self.url = None
84
85     # Setters or Initializers
86     def init_parser(self):
87         '''Get the command line options'''
88         self.parser.add_option_group(self.sambaopts)
89         self.parser.add_option_group(options.VersionOptions(self.parser))
90         self.init_credopts()
91         self.parser.add_option("-H", dest="url", help="URL for the samdb")
92         self.parser.add_option_group(self.credopts)
93
94     def init_argsopts(self):
95         '''Set the options and the arguments'''
96         (opts, args) = self.parser.parse_args()
97
98         self.opts = opts
99         self.args = args
100
101     def init_credopts(self):
102         '''Set Credential operations'''
103         self.credopts = options.CredentialsOptions(self.parser)
104
105     def init_lp(self):
106         '''Set the loadparm context'''
107         self.lp = self.sambaopts.get_loadparm()
108         self.smbconf = self.lp.configfile
109         if (not self.opts.url):
110             self.url = self.lp.samdb_url()
111         else:
112             self.url = self.opts.url
113
114     def init_session(self):
115         '''Initialize the session'''
116         self.creds = self.credopts.get_credentials(self.lp,
117             fallback_machine=True)
118         self.session = system_session()
119
120     def InitializeService(self):
121         '''Inializer for the thread'''
122         self.init_parser()
123         self.init_argsopts()
124         self.init_lp()
125         self.init_session()
126
127     # Getters
128     def Get_LDB(self):
129         '''Return a live instance of Samba'''
130         SambaDB = SamDB(self.url, session_info=self.session,
131             credentials=self.creds, lp=self.lp)
132         return SambaDB
133
134     def Get_lp_Content(self):
135         '''Return an instance of a local lp context'''
136         return self.lp
137
138     def Get_Creds(self):
139         '''Return an instance of a local creds'''
140         return self.creds
141
142
143 def GetBackLog(sys_log):
144     """Reads BackLog and makes thread aware of which GPO are unchanged or empty
145     :param String sys_log: path to backLog
146     :return Dictionary previous_scanned_version: {Unedited GPO: Version Number}
147     *NOTE on Version below
148     """
149     previous_scanned_version = {}
150     if os.path.isfile(sys_log):
151         previous_scanned_version = scan_log(sys_log)
152         return previous_scanned_version
153     else:
154         return None
155
156 # Set up the GPO service
157 GPOService = GPOServiceSetup()
158 GPOService.InitializeService()
159
160 # Get the Samba Instance
161 test_ldb = GPOService.Get_LDB()
162
163 # Get The lp context
164 lp = GPOService.Get_lp_Content()
165
166 # Get the CREDS
167 creds = GPOService.Get_Creds()
168
169 # Read the readable backLog into a hashmap
170 # then open writable backLog in same location
171 BackLoggedGPO = None
172 sys_log = '%s/%s' % (lp.get("path", "sysvol"), 'syslog.txt')
173 attr_log = '%s/%s' % (lp.get("path", "sysvol"), 'attrlog.txt')
174 BackLoggedGPO = GetBackLog(sys_log)
175
176
177 BackLog = open(sys_log, "w")
178
179
180 # We need to know writable DC to setup SMB connection
181 net = Net(creds=creds, lp=lp)
182 cldap_ret = net.finddc(domain=lp.get('realm'), flags=(nbt.NBT_SERVER_LDAP |
183     nbt.NBT_SERVER_DS))
184 dc_hostname = cldap_ret.pdc_dns_name
185
186 try:
187     conn = smb.SMB(dc_hostname, 'sysvol', lp=lp, creds=creds)
188 except Exception, e:
189     raise Exception("Error connecting to '%s' using SMB" % dc_hostname, e)
190
191 # Get the dn of the domain, and the dn of readable/writable DC
192 global_dn = test_ldb.domain_dn()
193 DC_OU = "OU=Domain Controllers" + ',' + global_dn
194
195 # Set up a List of the GUID for all GPO's
196 guid_list = [x['name'] for x in conn.list('%s/Policies' % lp.get("realm").lower())]
197 SYSV_PATH = '%s/%s/%s' % (lp.get("path", "sysvol"), lp.get("realm"), 'Policies')
198
199 hierarchy_gpos = establish_hierarchy(test_ldb, guid_list, DC_OU, global_dn)
200 change_backlog = False
201
202 # Take a local list of all current GPO list and run it against previous GPO's
203 # to see if something has changed. If so reset default and re-apply GPO.
204 Applicable_GPO = []
205 for i in hierarchy_gpos:
206     Applicable_GPO += i
207
208 # Flag gets set when
209 GPO_Changed = False
210 GPO_Deleted = check_deleted(Applicable_GPO, BackLoggedGPO)
211 if (GPO_Deleted):
212     # Null the backlog
213     BackLoggedGPO = {}
214     # Reset defaults then overwrite them
215     Reset_Defaults(test_ldb)
216     GPO_Changed = False
217
218 for guid_eval in hierarchy_gpos:
219     guid = guid_eval[0]
220     gp_extensions = [gp_sec_ext()]
221     local_path = '%s/Policies' % lp.get("realm").lower() + '/' + guid + '/'
222     version = gpo.gpo_get_sysvol_gpt_version(lp.get("path", "sysvol") + '/' + local_path)[1]
223     gpolist = gp_path_list(local_path)
224     if(version != BackLoggedGPO.get(guid)):
225         GPO_Changed = True
226     # If the GPO has a dn that is applicable to Samba
227     if guid_eval[1]:
228         # If it has a GPO file that could apply to Samba
229         if gpolist[0][1]:
230             # If it we have not read it before and is not empty
231             # Rewrite entire logfile here
232             if  (version != 0) and GPO_Changed == True:
233                 change_backlog = gpo_parser(gpolist, test_ldb, conn, attr_log)
234
235     BackLog.write('%s %i\n' % (guid, version))