388c1d25b8d45c5ef8a23b8b0e31c5684ec3b62e
[kai/samba-autobuild/.git] / source4 / lib / policy / gp_filesys.c
1 /*
2  *  Unix SMB/CIFS implementation.
3  *  Group Policy Object Support
4  *  Copyright (C) Wilco Baan Hofman 2008-2010
5  *
6  *  This program is free software; you can redistribute it and/or modify
7  *  it under the terms of the GNU General Public License as published by
8  *  the Free Software Foundation; either version 3 of the License, or
9  *  (at your option) any later version.
10  *
11  *  This program is distributed in the hope that it will be useful,
12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  *  GNU General Public License for more details.
15  *
16  *  You should have received a copy of the GNU General Public License
17  *  along with this program; if not, see <http://www.gnu.org/licenses/>.
18  */
19 #include "includes.h"
20 #include "lib/policy/policy.h"
21 #include "libcli/raw/smb.h"
22 #include "libcli/libcli.h"
23 #include "param/param.h"
24 #include "libcli/resolve/resolve.h"
25 #include <sys/stat.h>
26 #include <fcntl.h>
27 #include <unistd.h>
28 #include <dirent.h>
29
30 #define GP_MAX_DEPTH 25
31
32 struct gp_list_state {
33         struct gp_context *gp_ctx;
34         uint8_t depth;
35         const char *cur_rel_path;
36         const char *share_path;
37         const char *local_path;
38 };
39
40 static NTSTATUS gp_do_list(const char *, struct gp_list_state *);
41
42 /* Create a temporary policy directory */
43 static const char *gp_tmpdir(TALLOC_CTX *mem_ctx)
44 {
45         const char *gp_dir = talloc_asprintf(mem_ctx, "%s/policy", tmpdir());
46         struct stat st;
47
48         if (stat(gp_dir, &st) != 0) {
49                 mkdir(gp_dir, 0755);
50         }
51
52         return gp_dir;
53 }
54
55 /* This function is called by the smbcli_list function */
56 static void gp_list_helper (struct clilist_file_info *info, const char *mask, void *list_state_ptr)
57 {
58         struct gp_list_state *state = list_state_ptr;
59         const char *rel_path, *full_remote_path;
60         char *local_rel_path, *full_local_path;
61         unsigned int i;
62         int fh_remote, fh_local;
63         uint8_t *buf;
64         size_t nread = 0;
65         size_t buf_size = 1024;
66
67         /* Get local path by replacing backslashes with slashes */
68         local_rel_path = talloc_strdup(state, state->cur_rel_path);
69         for (i = 0; local_rel_path[i] != '\0'; i++) {
70                 if (local_rel_path[i] == '\\') {
71                         local_rel_path[i] = '/';
72                 }
73         }
74         full_local_path = talloc_asprintf(state, "%s%s/%s", state->local_path, local_rel_path, info->name);
75
76         /* Directory */
77         if (info->attrib & FILE_ATTRIBUTE_DIRECTORY) {
78                 if (state->depth >= GP_MAX_DEPTH)
79                         return;
80                 if (strcmp(info->name, ".") == 0 || strcmp(info->name, "..") == 0)
81                         return;
82
83                 mkdir(full_local_path, 0755);
84
85                 rel_path = talloc_asprintf(state, "%s\\%s", state->cur_rel_path, info->name);
86
87                 /* Recurse into this directory */
88                 gp_do_list(rel_path, state);
89                 return;
90         }
91
92         full_remote_path = talloc_asprintf(state, "%s%s\\%s", state->share_path, state->cur_rel_path, info->name);
93
94         /* Open the remote file */
95         fh_remote = smbcli_open(state->gp_ctx->cli->tree, full_remote_path, O_RDONLY, DENY_NONE);
96         if (fh_remote == -1) {
97                 DEBUG(0, ("Failed to open remote file: %s\n", full_remote_path));
98                 return;
99         }
100
101         /* Open the local file */
102         fh_local = open(full_local_path, O_WRONLY | O_CREAT | O_TRUNC, 0644);
103         if (fh_local == -1) {
104                 DEBUG(0, ("Failed to open local file: %s\n", full_local_path));
105                 return;
106         }
107
108         /* Copy the contents of the file */
109         buf = talloc_zero_array(state, uint8_t, buf_size);
110         while (1) {
111                 int n = smbcli_read(state->gp_ctx->cli->tree, fh_remote, buf, nread, buf_size);
112                 if (n <= 0) {
113                         break;
114                 }
115                 if (write(fh_local, buf, n) != n) {
116                         DEBUG(0, ("Short write while copying file.\n"));
117                         return;
118                 }
119                 nread += n;
120         }
121         /* Close the files */
122         smbcli_close(state->gp_ctx->cli->tree, fh_remote);
123         close(fh_local);
124
125         return;
126 }
127
128 static NTSTATUS gp_do_list (const char *rel_path, struct gp_list_state *state)
129 {
130         uint16_t attributes;
131         int success;
132         char *mask;
133         const char *old_rel_path;
134
135         attributes = FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY;
136
137         /* Update the relative paths, while buffering the parent */
138         old_rel_path = state->cur_rel_path;
139         state->cur_rel_path = rel_path;
140         state->depth++;
141
142         /* Get the current mask */
143         mask = talloc_asprintf(state, "%s%s\\*", state->share_path, rel_path);
144         success = smbcli_list(state->gp_ctx->cli->tree, mask, attributes, gp_list_helper, state);
145         talloc_free(mask);
146
147         /* Go back to the state of the parent */
148         state->cur_rel_path = old_rel_path;
149         state->depth--;
150
151         if (!success)
152                 return NT_STATUS_UNSUCCESSFUL;
153
154         return NT_STATUS_OK;
155 }
156
157 static NTSTATUS gp_cli_connect(struct gp_context *gp_ctx)
158 {
159         struct smbcli_options options;
160         struct smbcli_session_options session_options;
161
162         if (gp_ctx->cli != NULL)
163                 return NT_STATUS_OK;
164
165         gp_ctx->cli = smbcli_state_init(gp_ctx);
166
167         lp_smbcli_options(gp_ctx->lp_ctx, &options);
168         lp_smbcli_session_options(gp_ctx->lp_ctx, &session_options);
169
170
171         return smbcli_full_connection(gp_ctx,
172                         &gp_ctx->cli,
173                         gp_ctx->active_dc.name,
174                         lp_smb_ports(gp_ctx->lp_ctx),
175                         "sysvol",
176                         NULL,
177                         lp_socket_options(gp_ctx->lp_ctx),
178                         gp_ctx->credentials,
179                         lp_resolve_context(gp_ctx->lp_ctx),
180                         gp_ctx->ev_ctx,
181                         &options,
182                         &session_options,
183                         lp_iconv_convenience(gp_ctx->lp_ctx),
184                         lp_gensec_settings(gp_ctx, gp_ctx->lp_ctx));
185
186         return NT_STATUS_OK;
187 }
188
189 NTSTATUS gp_fetch_gpt (struct gp_context *gp_ctx, struct gp_object *gpo, const char **ret_local_path)
190 {
191         TALLOC_CTX *mem_ctx;
192         struct gp_list_state *state;
193         NTSTATUS status;
194         unsigned int i, bkslash_cnt;
195         struct stat st;
196         int rv;
197
198         /* Create a forked memory context, as a base for everything here */
199         mem_ctx = talloc_new(gp_ctx);
200
201
202         if (gp_ctx->cli == NULL) {
203                 status = gp_cli_connect(gp_ctx);
204                 if (!NT_STATUS_IS_OK(status)) {
205                         DEBUG(0, ("Failed to create cli connection to DC\n"));
206                         talloc_free(mem_ctx);
207                         return status;
208                 }
209         }
210
211         /* Prepare the state structure */
212         state = talloc_zero(mem_ctx, struct gp_list_state);
213         state->gp_ctx = gp_ctx;
214         state->local_path = talloc_asprintf(gp_ctx, "%s/%s", gp_tmpdir(mem_ctx), gpo->name);
215
216         /* Get the path from the share down (\\..\..\(this\stuff) */
217         for (i = 0, bkslash_cnt = 0; gpo->file_sys_path[i] != '\0'; i++) {
218                 if (gpo->file_sys_path[i] == '\\')
219                         bkslash_cnt++;
220
221                 if (bkslash_cnt == 4) {
222                         state->share_path = talloc_strdup(mem_ctx, &gpo->file_sys_path[i]);
223                         break;
224                 }
225         }
226
227         /* Create the GPO dir if it does not exist */
228         if (stat(state->local_path, &st) != 0) {
229                 rv = mkdir(state->local_path, 0755);
230                 if (rv < 0) {
231                         DEBUG(0, ("Could not create local path\n"));
232                         talloc_free(mem_ctx);
233                         return NT_STATUS_UNSUCCESSFUL;
234                 }
235         }
236
237         /* Copy the files */
238         status = gp_do_list("", state);
239         if (!NT_STATUS_IS_OK(status)) {
240                 DEBUG(0, ("Could not list GPO files on remote server\n"));
241                 talloc_free(mem_ctx);
242                 return status;
243         }
244
245         /* Return the local path to the gpo */
246         *ret_local_path = state->local_path;
247
248         talloc_free(mem_ctx);
249         return NT_STATUS_OK;
250 }
251
252 static NTSTATUS push_recursive (struct gp_context *gp_ctx, const char *local_path, const char *remote_path, int depth)
253 {
254         DIR *dir;
255         struct dirent *dirent;
256         char *entry_local_path;
257         char *entry_remote_path;
258         int local_fd, remote_fd;
259         int buf[1024];
260         int nread, total_read;
261
262         dir = opendir(local_path);
263         while ((dirent = readdir(dir)) != NULL) {
264                 if (strcmp(dirent->d_name, ".") == 0 || strcmp(dirent->d_name, "..") == 0) {
265                         continue;
266                 }
267
268                 entry_local_path = talloc_asprintf(gp_ctx, "%s/%s", local_path, dirent->d_name);
269                 entry_remote_path = talloc_asprintf(gp_ctx, "%s\\%s", remote_path, dirent->d_name);
270                 if (dirent->d_type == DT_DIR) {
271                         DEBUG(6, ("Pushing directory %s to %s on sysvol\n", entry_local_path, entry_remote_path));
272                         smbcli_mkdir(gp_ctx->cli->tree, entry_remote_path);
273                         if (depth < GP_MAX_DEPTH)
274                                 push_recursive(gp_ctx, entry_local_path, entry_remote_path, depth+1);
275                 } else {
276                         DEBUG(6, ("Pushing file %s to %s on sysvol\n", entry_local_path, entry_remote_path));
277                         remote_fd = smbcli_open(gp_ctx->cli->tree, entry_remote_path, O_WRONLY | O_CREAT, 0);
278                         if (remote_fd < 0) {
279                                 talloc_free(entry_local_path);
280                                 talloc_free(entry_remote_path);
281                                 DEBUG(0, ("Failed to create remote file: %s\n", entry_remote_path));
282                                 return NT_STATUS_UNSUCCESSFUL;
283                         }
284                         local_fd = open(entry_local_path, O_RDONLY);
285                         if (local_fd < 0) {
286                                 talloc_free(entry_local_path);
287                                 talloc_free(entry_remote_path);
288                                 DEBUG(0, ("Failed to open local file: %s\n", entry_local_path));
289                                 return NT_STATUS_UNSUCCESSFUL;
290                         }
291                         total_read = 0;
292                         while ((nread = read(local_fd, &buf, sizeof(buf)))) {
293                                 smbcli_write(gp_ctx->cli->tree, remote_fd, 0, &buf, total_read, nread);
294                                 total_read += nread;
295                         }
296
297                         close(local_fd);
298                         smbcli_close(gp_ctx->cli->tree, remote_fd);
299                 }
300                 talloc_free(entry_local_path);
301                 talloc_free(entry_remote_path);
302         }
303         closedir(dir);
304
305         return NT_STATUS_OK;
306 }
307
308
309
310 NTSTATUS gp_push_gpt(struct gp_context *gp_ctx, const char *local_path, const char *file_sys_path)
311 {
312         NTSTATUS status;
313         unsigned int i, bkslash_cnt;
314         char *share_path;
315
316         if (gp_ctx->cli == NULL) {
317                 status = gp_cli_connect(gp_ctx);
318                 if (!NT_STATUS_IS_OK(status)) {
319                         DEBUG(0, ("Failed to create cli connection to DC\n"));
320                         return status;
321                 }
322         }
323
324         /* Get the path from the share down (\\..\..\(this\stuff) */
325         for (i = 0, bkslash_cnt = 0; file_sys_path[i] != '\0'; i++) {
326                 if (file_sys_path[i] == '\\')
327                         bkslash_cnt++;
328
329                 if (bkslash_cnt == 4) {
330                         share_path = talloc_strdup(gp_ctx, &file_sys_path[i]);
331                         break;
332                 }
333         }
334
335         DEBUG(5, ("Copying %s to %s on sysvol\n", local_path, share_path));
336
337         smbcli_mkdir(gp_ctx->cli->tree, share_path);
338
339         status = push_recursive(gp_ctx, local_path, share_path, 0);
340
341         talloc_free(share_path);
342         return status;
343 }
344
345 NTSTATUS gp_create_gpt(struct gp_context *gp_ctx, const char *name, const char *file_sys_path)
346 {
347         TALLOC_CTX *mem_ctx;
348         const char *tmp_dir, *policy_dir, *tmp_str;
349         int rv;
350         int fd;
351         NTSTATUS status;
352
353         /* Create a forked memory context, as a base for everything here */
354         mem_ctx = talloc_new(gp_ctx);
355
356         tmp_dir = gp_tmpdir(mem_ctx);
357         policy_dir = talloc_asprintf(mem_ctx, "%s/%s", tmp_dir, name);
358         rv = mkdir(policy_dir, 0755);
359         if (rv < 0) {
360                 DEBUG(0, ("Could not create the policy dir: %s\n", policy_dir));
361                 talloc_free(mem_ctx);
362                 return NT_STATUS_UNSUCCESSFUL;
363         }
364
365         tmp_str = talloc_asprintf(mem_ctx, "%s/User", policy_dir);
366         rv = mkdir(tmp_str, 0755);
367         if (rv < 0) {
368                 DEBUG(0, ("Could not create the User dir: %s\n", tmp_str));
369                 talloc_free(mem_ctx);
370                 return NT_STATUS_UNSUCCESSFUL;
371         }
372
373         tmp_str = talloc_asprintf(mem_ctx, "%s/Machine", policy_dir);
374         rv = mkdir(tmp_str, 0755);
375         if (rv < 0) {
376                 DEBUG(0, ("Could not create the Machine dir: %s\n", tmp_str));
377                 talloc_free(mem_ctx);
378                 return NT_STATUS_UNSUCCESSFUL;
379         }
380
381         tmp_str = talloc_asprintf(mem_ctx, "%s/GPT.INI", policy_dir);
382         fd = open(tmp_str, O_CREAT | O_WRONLY, 0644);
383         if (fd < 0) {
384                 DEBUG(0, ("Could not create the GPT.INI: %s\n", tmp_str));
385                 talloc_free(mem_ctx);
386                 return NT_STATUS_UNSUCCESSFUL;
387         }
388
389         rv = write(fd, "[General]\r\nVersion=0\r\n", 23);
390         if (rv != 23) {
391                 DEBUG(0, ("Short write in GPT.INI\n"));
392                 talloc_free(mem_ctx);
393                 return NT_STATUS_UNSUCCESSFUL;
394         }
395
396         close(fd);
397
398
399         status = gp_push_gpt(gp_ctx, policy_dir, file_sys_path);
400         if (!NT_STATUS_IS_OK(status)) {
401                 talloc_free(mem_ctx);
402                 return status;
403         }
404
405         talloc_free(mem_ctx);
406         return NT_STATUS_OK;
407 }