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