Line data Source code
1 : #include "../burp.h"
2 : #include "sdirs.h"
3 : #include "../alloc.h"
4 : #include "../conf.h"
5 : #include "../fsops.h"
6 : #include "../lock.h"
7 : #include "../log.h"
8 : #include "../prepend.h"
9 : #include "timestamp.h"
10 :
11 308 : struct sdirs *sdirs_alloc(void)
12 : {
13 308 : return (struct sdirs *)calloc_w(1, sizeof(struct sdirs), __func__);
14 : }
15 :
16 309 : static int do_lock_dirs(struct sdirs *sdirs,
17 : const char *cname, const char *conf_lockdir)
18 : {
19 309 : int ret=-1;
20 309 : char *lockbase=NULL;
21 309 : char *lockfile=NULL;
22 309 : if(conf_lockdir)
23 : {
24 2 : if(!(sdirs->lockdir=strdup_w(conf_lockdir, __func__))
25 1 : || !(lockbase=prepend_s(sdirs->lockdir, cname)))
26 : goto end;
27 : }
28 : else
29 : {
30 616 : if(!(sdirs->lockdir=strdup_w(sdirs->client, __func__))
31 308 : || !(lockbase=strdup_w(sdirs->client, __func__)))
32 : goto end;
33 : }
34 618 : if(!(lockfile=prepend_s(lockbase, "lockfile"))
35 309 : || !(sdirs->lock=lock_alloc_and_init(lockfile)))
36 : goto end;
37 309 : ret=0;
38 : end:
39 309 : free_w(&lockbase);
40 309 : free_w(&lockfile);
41 309 : return ret;
42 : }
43 :
44 : static int free_prepend_s(char **dst, const char *a, const char *b)
45 : {
46 23 : free_w(dst);
47 23 : return !(*dst=prepend_s(a, b));
48 : }
49 :
50 5 : int sdirs_get_real_manifest(struct sdirs *sdirs, enum protocol protocol)
51 : {
52 : return free_prepend_s(&sdirs->rmanifest,
53 10 : sdirs->rworking, protocol==PROTO_1?"manifest.gz":"manifest");
54 : }
55 :
56 7 : int sdirs_get_real_working_from_symlink(struct sdirs *sdirs)
57 : {
58 7 : ssize_t len=0;
59 7 : char real[256]="";
60 14 : if((len=readlink(sdirs->working, real, sizeof(real)-1))<0)
61 : {
62 : logp("Could not readlink %s: %s\n",
63 0 : sdirs->working, strerror(errno));
64 0 : return -1;
65 : }
66 7 : real[len]='\0';
67 14 : if(free_prepend_s(&sdirs->rworking, sdirs->client, real)
68 14 : || free_prepend_s(&sdirs->treepath,
69 7 : sdirs->rworking, DATA_DIR "/" TREE_DIR))
70 : return -1;
71 7 : return 0;
72 : }
73 :
74 2 : int sdirs_create_real_working(struct sdirs *sdirs, const char *timestamp_format)
75 : {
76 2 : char tstmp[64]="";
77 2 : char fname[64]="";
78 :
79 4 : if(build_path_w(sdirs->working)
80 2 : || timestamp_get_new(sdirs,
81 2 : tstmp, sizeof(tstmp), fname, sizeof(fname), timestamp_format)
82 4 : || free_prepend_s(&sdirs->rworking, sdirs->client, fname)
83 4 : || free_prepend_s(&sdirs->treepath,
84 2 : sdirs->rworking, DATA_DIR "/" TREE_DIR))
85 : return -1;
86 :
87 : // Add the working symlink before creating the directory.
88 : // This is because bedup checks the working symlink before
89 : // going into a directory. If the directory got created first,
90 : // bedup might go into it in the moment before the symlink
91 : // gets added.
92 2 : if(symlink(fname, sdirs->working)) // relative link to the real work dir
93 : {
94 : logp("could not point working symlink to: %s\n",
95 0 : sdirs->rworking);
96 0 : return -1;
97 : }
98 2 : if(mkdir(sdirs->rworking, 0777))
99 : {
100 0 : logp("could not mkdir for next backup: %s\n", sdirs->rworking);
101 0 : unlink(sdirs->working);
102 0 : return -1;
103 : }
104 2 : if(timestamp_write(sdirs->timestamp, tstmp))
105 : {
106 : logp("unable to write timestamp %s to %s\n",
107 0 : tstmp, sdirs->timestamp);
108 0 : return -1;
109 : }
110 :
111 : return 0;
112 : }
113 :
114 309 : static int do_common_dirs(struct sdirs *sdirs, const char *manual_delete)
115 : {
116 618 : if(!(sdirs->working=prepend_s(sdirs->client, "working"))
117 309 : || !(sdirs->finishing=prepend_s(sdirs->client, "finishing"))
118 309 : || !(sdirs->current=prepend_s(sdirs->client, "current"))
119 309 : || !(sdirs->currenttmp=prepend_s(sdirs->client, "current.tmp"))
120 309 : || !(sdirs->timestamp=prepend_s(sdirs->working, "timestamp"))
121 309 : || !(sdirs->phase1data=prepend_s(sdirs->working, "phase1.gz"))
122 309 : || !(sdirs->changed=prepend_s(sdirs->working, "changed"))
123 618 : || !(sdirs->unchanged=prepend_s(sdirs->working, "unchanged")))
124 : return -1;
125 309 : if(manual_delete)
126 : {
127 0 : if(!(sdirs->deleteme=strdup_w(manual_delete, __func__)))
128 : return -1;
129 : }
130 : else
131 : {
132 309 : if(!(sdirs->deleteme=prepend_s(sdirs->client, "deleteme")))
133 : return -1;
134 : }
135 : return 0;
136 : }
137 :
138 : // Maybe should be in a protocol1 directory.
139 129 : static int do_protocol1_dirs(struct sdirs *sdirs, const char *cname,
140 : const char *manual_delete)
141 : {
142 258 : if(!(sdirs->clients=strdup_w(sdirs->base, __func__))
143 129 : || !(sdirs->client=prepend_s(sdirs->clients, cname))
144 129 : || do_common_dirs(sdirs, manual_delete)
145 129 : || !(sdirs->currentdata=prepend_s(sdirs->current, DATA_DIR))
146 129 : || !(sdirs->manifest=prepend_s(sdirs->working, "manifest.gz"))
147 129 : || !(sdirs->datadirtmp=prepend_s(sdirs->working, "data.tmp"))
148 129 : || !(sdirs->cmanifest=prepend_s(sdirs->current, "manifest.gz"))
149 129 : || !(sdirs->cincexc=prepend_s(sdirs->current, "incexc"))
150 258 : || !(sdirs->deltmppath=prepend_s(sdirs->working, "deltmppath")))
151 : return -1;
152 : // sdirs->rworking gets set later.
153 : // sdirs->treepath gets set later.
154 129 : return 0;
155 : }
156 :
157 180 : static int do_protocol2_dirs(struct sdirs *sdirs,
158 : const char *cname, const char *dedup_group, const char *manual_delete)
159 : {
160 180 : if(!dedup_group)
161 : {
162 0 : logp("dedup_group unset in %s\n", __func__);
163 0 : return -1;
164 : }
165 360 : if(!(sdirs->dedup=prepend_s(sdirs->base, dedup_group))
166 180 : || !(sdirs->clients=prepend_s(sdirs->dedup, "clients"))
167 180 : || !(sdirs->client=prepend_s(sdirs->clients, cname))
168 180 : || !(sdirs->dindex=prepend_s(sdirs->client, "dindex"))
169 180 : || !(sdirs->dfiles=prepend_s(sdirs->client, "dfiles"))
170 180 : || do_common_dirs(sdirs, manual_delete)
171 180 : || !(sdirs->data=prepend_s(sdirs->dedup, DATA_DIR))
172 180 : || !(sdirs->cfiles=prepend_s(sdirs->data, "cfiles"))
173 180 : || !(sdirs->global_sparse=prepend_s(sdirs->data, "sparse"))
174 180 : || !(sdirs->champlock=prepend_s(sdirs->data, "cc.lock"))
175 180 : || !(sdirs->champsock=prepend_s(sdirs->data, "cc.sock"))
176 180 : || !(sdirs->champlog=prepend_s(sdirs->data, "cc.log"))
177 180 : || !(sdirs->manifest=prepend_s(sdirs->working, "manifest"))
178 360 : || !(sdirs->cmanifest=prepend_s(sdirs->current, "manifest")))
179 : return -1;
180 : // sdirs->rworking gets set later.
181 : // sdirs->rmanifest gets set later.
182 180 : return 0;
183 : }
184 :
185 16 : int sdirs_init_from_confs(struct sdirs *sdirs, struct conf **confs)
186 : {
187 : return sdirs_init(
188 : sdirs,
189 : get_protocol(confs),
190 16 : get_string(confs[OPT_DIRECTORY]),
191 16 : get_string(confs[OPT_CNAME]),
192 16 : get_string(confs[OPT_CLIENT_LOCKDIR]),
193 16 : get_string(confs[OPT_DEDUP_GROUP]),
194 16 : get_string(confs[OPT_MANUAL_DELETE])
195 16 : );
196 : }
197 :
198 309 : int sdirs_init(struct sdirs *sdirs, enum protocol protocol,
199 : const char *directory, const char *cname, const char *conf_lockdir,
200 : const char *dedup_group, const char *manual_delete)
201 : {
202 309 : if(!directory)
203 : {
204 0 : logp("directory unset in %s\n", __func__);
205 0 : goto error;
206 : }
207 :
208 309 : if(!(sdirs->base=strdup_w(directory, __func__)))
209 : goto error;
210 :
211 309 : sdirs->protocol=protocol;
212 :
213 309 : if(protocol==PROTO_1)
214 : {
215 129 : if(do_protocol1_dirs(sdirs, cname, manual_delete))
216 : goto error;
217 : }
218 : else
219 : {
220 180 : if(do_protocol2_dirs(sdirs, cname, dedup_group, manual_delete))
221 : goto error;
222 : }
223 :
224 309 : if(do_lock_dirs(sdirs, cname, conf_lockdir)) goto error;
225 :
226 : return 0;
227 : error:
228 : return -1;
229 : }
230 :
231 309 : void sdirs_free_content(struct sdirs *sdirs)
232 : {
233 309 : free_w(&sdirs->base);
234 309 : free_w(&sdirs->dedup);
235 309 : free_w(&sdirs->champlock);
236 309 : free_w(&sdirs->champsock);
237 309 : free_w(&sdirs->champlog);
238 309 : free_w(&sdirs->data);
239 309 : free_w(&sdirs->clients);
240 309 : free_w(&sdirs->client);
241 :
242 309 : free_w(&sdirs->working);
243 309 : free_w(&sdirs->rworking);
244 309 : free_w(&sdirs->finishing);
245 309 : free_w(&sdirs->current);
246 309 : free_w(&sdirs->currenttmp);
247 309 : free_w(&sdirs->deleteme);
248 309 : free_w(&sdirs->dindex);
249 309 : free_w(&sdirs->dfiles);
250 309 : free_w(&sdirs->cfiles);
251 309 : free_w(&sdirs->global_sparse);
252 :
253 309 : free_w(&sdirs->timestamp);
254 309 : free_w(&sdirs->changed);
255 309 : free_w(&sdirs->unchanged);
256 309 : free_w(&sdirs->manifest);
257 309 : free_w(&sdirs->rmanifest);
258 309 : free_w(&sdirs->cmanifest);
259 309 : free_w(&sdirs->phase1data);
260 :
261 309 : free_w(&sdirs->lockdir);
262 309 : lock_free(&sdirs->lock);
263 :
264 : // Protocol1 directories.
265 309 : free_w(&sdirs->currentdata);
266 309 : free_w(&sdirs->datadirtmp);
267 309 : free_w(&sdirs->cincexc);
268 309 : free_w(&sdirs->deltmppath);
269 309 : free_w(&sdirs->treepath);
270 309 : }
271 :
272 347 : void sdirs_free(struct sdirs **sdirs)
273 : {
274 694 : if(!sdirs || !*sdirs) return;
275 308 : sdirs_free_content(*sdirs);
276 :
277 308 : free_v((void **)sdirs);
278 : }
|
