r19002: Add a very simple synchronisation function to download the full GPO.
[sfrench/samba-autobuild/.git] / source3 / libgpo / gpo_filesync.c
1 /* 
2  *  Unix SMB/CIFS implementation.
3  *  Group Policy Object Support
4  *  Copyright (C) Guenther Deschner 2006
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 2 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, write to the Free Software
18  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
19  */
20
21 #include "includes.h"
22
23 struct sync_context {
24         TALLOC_CTX *mem_ctx;
25         struct cli_state *cli;
26         char *remote_path;
27         char *local_path;
28         pstring mask;
29         uint16 attribute;
30 };
31
32 static void gpo_sync_func(const char *mnt,
33                            file_info *info,
34                            const char *mask,
35                            void *state);
36
37 NTSTATUS gpo_copy_file(TALLOC_CTX *mem_ctx,
38                        struct cli_state *cli,
39                        const char *nt_path,
40                        const char *unix_path)
41 {
42         NTSTATUS result;
43         int fnum, fd;
44         char *data = NULL;
45         static int io_bufsize = 64512;
46         int read_size = io_bufsize;
47         off_t start = 0;
48         off_t nread = 0;
49
50         if ((fnum = cli_open(cli, nt_path, O_RDONLY, DENY_NONE)) == -1) {
51                 result = NT_STATUS_NO_SUCH_FILE;
52                 goto out;
53         }
54
55         if ((fd = sys_open(unix_path, O_WRONLY|O_CREAT|O_TRUNC, 0644)) == -1) {
56                 result = map_nt_error_from_unix(errno);
57                 goto out;
58         }
59          
60         if ((data = (char *)SMB_MALLOC(read_size)) == NULL) {
61                 result = NT_STATUS_NO_MEMORY;
62                 goto out;
63         }
64
65         while (1) {
66
67                 int n = cli_read(cli, fnum, data, nread + start, read_size);
68
69                 if (n <= 0)
70                         break;
71
72                 if (write(fd, data, n) != n) {
73                         break;
74                 }
75
76                 nread += n;
77         }
78
79         result = NT_STATUS_OK;
80
81  out:
82         SAFE_FREE(data);
83         if (fnum) {
84                 cli_close(cli, fnum);
85         }
86         if (fd) {
87                 close(fd);
88         }
89
90         return result;
91 }
92
93 /****************************************************************
94  copy dir
95 ****************************************************************/
96
97 static NTSTATUS gpo_copy_dir(const char *unix_path)
98 {
99         if ((mkdir(unix_path, 0644)) < 0 && errno != EEXIST) {
100                 return NT_STATUS_ACCESS_DENIED;
101         }
102
103         return NT_STATUS_OK;
104 }
105
106 /****************************************************************
107  sync files
108 ****************************************************************/
109
110 static BOOL gpo_sync_files(struct sync_context *ctx)
111 {
112         DEBUG(3,("calling cli_list with mask: %s\n", ctx->mask));
113
114         if (cli_list(ctx->cli, ctx->mask, ctx->attribute, gpo_sync_func, ctx) == -1) {
115                 DEBUG(1,("listing [%s] failed with error: %s\n", 
116                         ctx->mask, cli_errstr(ctx->cli)));
117                 return False;
118         }
119
120         return True;
121 }
122
123 /****************************************************************
124  syncronisation call back
125 ****************************************************************/
126
127 static void gpo_sync_func(const char *mnt,
128                           file_info *info,
129                           const char *mask,
130                           void *state)
131 {
132         NTSTATUS result;
133         struct sync_context *ctx;
134         fstring nt_filename, unix_filename;
135         fstring nt_dir, unix_dir;
136         char *old_nt_dir, *old_unix_dir;
137
138         ctx = (struct sync_context *)state;
139
140         if (strequal(info->name, ".") || strequal(info->name, "..")) {
141                 return;
142         }
143
144         DEBUG(5,("gpo_sync_func: got mask: [%s], name: [%s]\n", 
145                 mask, info->name));
146
147         if (info->mode & aDIR) {
148
149                 DEBUG(3,("got dir: [%s]\n", info->name));
150
151                 fstrcpy(nt_dir, ctx->remote_path);
152                 fstrcat(nt_dir, "\\");
153                 fstrcat(nt_dir, info->name);
154
155                 fstrcpy(unix_dir, ctx->local_path);
156                 fstrcat(unix_dir, "/");
157                 fstrcat(unix_dir, info->name);
158
159                 result = gpo_copy_dir(unix_dir);
160                 if (!NT_STATUS_IS_OK(result)) {
161                         DEBUG(1,("failed to copy dir: %s\n", nt_errstr(result)));
162                 }
163
164                 old_nt_dir = ctx->remote_path;
165                 ctx->remote_path = nt_dir;
166                 
167                 old_unix_dir = ctx->local_path;
168                 ctx->local_path = talloc_strdup(ctx->mem_ctx, unix_dir);
169
170                 pstrcpy(ctx->mask, nt_dir);
171                 pstrcat(ctx->mask, "\\*");
172
173                 if (!gpo_sync_files(ctx)) {
174                         DEBUG(0,("could not sync files\n"));
175                 }
176
177                 ctx->remote_path = old_nt_dir;
178                 ctx->local_path = old_unix_dir;
179                 return;
180         }
181
182         DEBUG(3,("got file: [%s]\n", info->name));
183
184         fstrcpy(nt_filename, ctx->remote_path);
185         fstrcat(nt_filename, "\\");
186         fstrcat(nt_filename, info->name);
187
188         fstrcpy(unix_filename, ctx->local_path);
189         fstrcat(unix_filename, "/");
190         fstrcat(unix_filename, info->name);
191
192         result = gpo_copy_file(ctx->mem_ctx, ctx->cli, nt_filename, unix_filename);
193         if (!NT_STATUS_IS_OK(result)) {
194                 DEBUG(1,("failed to copy file: %s\n", nt_errstr(result)));
195         }
196 }
197
198
199 /****************************************************************
200  list a remote directory and download recursivly
201 ****************************************************************/
202
203 NTSTATUS gpo_sync_directories(TALLOC_CTX *mem_ctx, 
204                               struct cli_state *cli, 
205                               const char *nt_path, 
206                               const char *local_path)
207 {
208         struct sync_context ctx;
209
210         ctx.mem_ctx     = mem_ctx;
211         ctx.cli         = cli;
212         ctx.remote_path = CONST_DISCARD(char *, nt_path);
213         ctx.local_path  = CONST_DISCARD(char *, local_path);
214         ctx.attribute   = (aSYSTEM | aHIDDEN | aDIR);
215
216         pstrcpy(ctx.mask, nt_path);
217         pstrcat(ctx.mask, "\\*");
218
219         if (!gpo_sync_files(&ctx)) {
220                 return NT_STATUS_NO_SUCH_FILE;
221         }
222
223         return NT_STATUS_OK;
224 }