Line data Source code
1 : #include "../burp.h"
2 : #include "../alloc.h"
3 : #include "../cmd.h"
4 : #include "../fsops.h"
5 : #include "../fzp.h"
6 : #include "../hexmap.h"
7 : #include "../log.h"
8 : #include "../msg.h"
9 : #include "../prepend.h"
10 : #include "../protocol2/blk.h"
11 : #include "../sbuf.h"
12 : #include "manio.h"
13 : #include "protocol2/champ_chooser/champ_chooser.h"
14 : #include "protocol2/dpth.h"
15 :
16 : #define MANIO_MODE_READ "rb"
17 : #define MANIO_MODE_WRITE "wb"
18 : #define MANIO_MODE_APPEND "ab"
19 :
20 : static void man_off_t_free_content(man_off_t *offset)
21 : {
22 246 : if(!offset) return;
23 13992 : free_w(&offset->fpath);
24 13992 : free_w(&offset->ppath);
25 : }
26 :
27 14020 : void man_off_t_free(man_off_t **offset)
28 : {
29 14020 : if(!offset || !*offset) return;
30 27492 : man_off_t_free_content(*offset);
31 13746 : free_v((void **)offset);
32 : }
33 :
34 : static man_off_t *man_off_t_alloc(void)
35 : {
36 13746 : return (man_off_t *)calloc_w(1, sizeof(man_off_t), __func__);
37 : }
38 :
39 11 : static int init_write_hooks(struct manio *manio,
40 : const char *hook_dir, const char *rmanifest)
41 : {
42 11 : if(!(manio->hook_dir=strdup_w(hook_dir, __func__))
43 11 : || !(manio->rmanifest=strdup_w(rmanifest, __func__))
44 11 : || !(manio->hook_sort=(uint64_t *)calloc_w(MANIFEST_SIG_MAX,
45 : sizeof(uint64_t), __func__)))
46 : return -1;
47 : return 0;
48 : }
49 :
50 11 : static int init_write_dindex(struct manio *manio, const char *dir)
51 : {
52 11 : if(!(manio->dindex_dir=strdup_w(dir, __func__))
53 11 : || !(manio->dindex_sort=(uint64_t *)calloc_w(MANIFEST_SIG_MAX,
54 : sizeof(uint64_t), __func__)))
55 : return -1;
56 : return 0;
57 : }
58 :
59 : static int is_single_file(struct manio *manio)
60 : {
61 1610 : return manio->protocol==PROTO_1 || manio->phase==1;
62 : }
63 :
64 1078 : static char *get_next_fpath(struct manio *manio, man_off_t *offset)
65 : {
66 : static char tmp[32];
67 2156 : if(is_single_file(manio))
68 430 : return strdup_w(manio->manifest, __func__);
69 648 : snprintf(tmp, sizeof(tmp), "%08" PRIX64, offset->fcount++);
70 648 : return prepend_s(manio->manifest, tmp);
71 : }
72 :
73 978 : static int manio_open_next_fpath(struct manio *manio)
74 : {
75 : static struct stat statp;
76 978 : man_off_t *offset=manio->offset;
77 :
78 978 : free_w(&offset->ppath);
79 978 : offset->ppath=offset->fpath;
80 978 : if(!(offset->fpath=get_next_fpath(manio, offset)))
81 : return -1;
82 :
83 978 : if(!strcmp(manio->mode, MANIO_MODE_READ)
84 1210 : && lstat(offset->fpath, &statp))
85 : return 0;
86 :
87 818 : if(build_path_w(offset->fpath))
88 : return -1;
89 818 : switch(manio->phase)
90 : {
91 : case 2:
92 630 : if(!(manio->fzp=fzp_open(offset->fpath,
93 630 : manio->mode))) return -1;
94 630 : return 0;
95 : case 1:
96 : case 3:
97 : default:
98 188 : if(!(manio->fzp=fzp_gzopen(offset->fpath,
99 188 : manio->mode))) return -1;
100 188 : return 0;
101 : }
102 : }
103 :
104 46 : static int manio_open_last_fpath(struct manio *manio)
105 : {
106 46 : int max=-1;
107 92 : if(is_single_file(manio))
108 10 : return manio_open_next_fpath(manio);
109 36 : if(get_highest_entry(manio->manifest, &max, 8))
110 : return -1;
111 36 : if(max<0) max=0;
112 36 : manio->offset->fcount=(uint64_t)max;
113 36 : return manio_open_next_fpath(manio);
114 : }
115 :
116 : static struct manio *manio_alloc(void)
117 : {
118 756 : return (struct manio *)calloc_w(1, sizeof(struct manio), __func__);
119 : }
120 :
121 756 : static struct manio *do_manio_open(const char *manifest, const char *mode,
122 : enum protocol protocol, int phase)
123 : {
124 756 : struct manio *manio=NULL;
125 756 : if(!(manio=manio_alloc())
126 756 : || !(manio->manifest=strdup_w(manifest, __func__))
127 756 : || !(manio->mode=strdup_w(mode, __func__))
128 1512 : || !(manio->offset=man_off_t_alloc()))
129 : goto error;
130 756 : manio->protocol=protocol;
131 756 : manio->phase=phase;
132 756 : if(!strcmp(manio->mode, MANIO_MODE_APPEND))
133 : {
134 46 : if(manio->phase!=2)
135 : {
136 0 : logp("manio append mode only works for phase 2.\n");
137 0 : logp("%s has phase: %d\n",
138 0 : manio->manifest, manio->phase);
139 0 : goto error;
140 : }
141 46 : if(manio_open_last_fpath(manio))
142 : goto error;
143 : }
144 : else
145 : {
146 710 : if(manio_open_next_fpath(manio))
147 : goto error;
148 : }
149 756 : return manio;
150 : error:
151 0 : manio_close(&manio);
152 0 : return NULL;
153 : }
154 :
155 39 : struct manio *manio_open(const char *manifest, const char *mode,
156 : enum protocol protocol)
157 : {
158 39 : return do_manio_open(manifest, mode, protocol, 0);
159 : }
160 :
161 144 : struct manio *manio_open_phase1(const char *manifest, const char *mode,
162 : enum protocol protocol)
163 : {
164 144 : return do_manio_open(manifest, mode, protocol, 1);
165 : }
166 :
167 554 : struct manio *manio_open_phase2(const char *manifest, const char *mode,
168 : enum protocol protocol)
169 : {
170 554 : return do_manio_open(manifest, mode, protocol, 2);
171 : }
172 :
173 19 : struct manio *manio_open_phase3(const char *manifest, const char *mode,
174 : enum protocol protocol, const char *rmanifest)
175 : {
176 19 : struct manio *manio=NULL;
177 :
178 19 : if(!(manio=do_manio_open(manifest, mode, protocol, 3)))
179 : goto end;
180 :
181 19 : if(protocol==PROTO_2 && rmanifest)
182 : {
183 11 : char *hooksdir=NULL;
184 11 : char *dindexdir=NULL;
185 11 : if(!(hooksdir=prepend_s(manifest, "hooks"))
186 11 : || !(dindexdir=prepend_s(manifest, "dindex"))
187 11 : || init_write_hooks(manio, hooksdir, rmanifest)
188 11 : || init_write_dindex(manio, dindexdir))
189 0 : manio_close(&manio);
190 11 : free_w(&hooksdir);
191 11 : free_w(&dindexdir);
192 : }
193 :
194 : end:
195 19 : return manio;
196 : }
197 :
198 756 : static void manio_free_content(struct manio *manio)
199 : {
200 756 : if(!manio) return;
201 756 : man_off_t_free(&manio->offset);
202 756 : free_w(&manio->manifest);
203 756 : free_w(&manio->mode);
204 756 : free_w(&manio->rmanifest);
205 756 : free_w(&manio->hook_dir);
206 756 : free_v((void **)&manio->hook_sort);
207 756 : free_w(&manio->dindex_dir);
208 756 : free_v((void **)&manio->dindex_sort);
209 756 : memset(manio, 0, sizeof(struct manio));
210 : }
211 :
212 : #ifndef UTEST
213 : static
214 : #endif
215 36 : int write_hook_header(struct fzp *fzp, const char *rmanifest, const char *msg)
216 : {
217 36 : int ret=0;
218 36 : char *tmp=NULL;
219 36 : if(!(tmp=prepend_s(rmanifest, msg))
220 36 : || send_msg_fzp(fzp, CMD_MANIFEST, tmp, strlen(tmp)))
221 : ret=-1;
222 36 : free_w(&tmp);
223 36 : return ret;
224 : }
225 :
226 377847 : static int uint64_t_sort(const void *a, const void *b)
227 : {
228 377847 : uint64_t *x=(uint64_t *)a;
229 377847 : uint64_t *y=(uint64_t *)b;
230 377847 : if(*x>*y) return 1;
231 185793 : if(*x<*y) return -1;
232 0 : return 0;
233 : }
234 :
235 : static char *get_fcount_path(struct manio *manio)
236 : {
237 21 : return prepend_s(manio->manifest, "fcount");
238 : }
239 :
240 : // Backup phase4 needs to know the fcount, so leave a file behind that
241 : // contains it (otherwise phase4 will have to read and sort the directory
242 : // contents).
243 20 : static int manio_write_fcount(struct manio *manio)
244 : {
245 20 : int ret=-1;
246 20 : struct fzp *fzp=NULL;
247 20 : char *path=NULL;
248 :
249 40 : if(!(path=get_fcount_path(manio))
250 20 : || !(fzp=fzp_open(path, "wb")))
251 : goto end;
252 20 : if(fzp_printf(fzp, "%08" PRIX64 "\n", manio->offset->fcount)!=9)
253 : {
254 0 : logp("Short write when writing to %s\n", path);
255 0 : goto end;
256 : }
257 : ret=0;
258 : end:
259 20 : if(fzp_close(&fzp))
260 : {
261 0 : logp("Could not close file pointer to %s\n", path);
262 0 : ret=-1;
263 : }
264 20 : free_w(&path);
265 20 : return ret;
266 : }
267 :
268 1 : int manio_read_fcount(struct manio *manio)
269 : {
270 1 : int ret=-1;
271 : size_t s;
272 1 : struct fzp *fzp=NULL;
273 1 : char *path=NULL;
274 1 : char buf[16]="";
275 2 : if(!(path=get_fcount_path(manio))
276 1 : || !(fzp=fzp_open(path, "rb")))
277 : goto end;
278 1 : if(!fzp_gets(fzp, buf, sizeof(buf)))
279 : {
280 0 : logp("fzp_gets on %s failed\n", path);
281 0 : goto end;
282 : }
283 1 : s=strlen(buf);
284 1 : if(s!=9)
285 : {
286 0 : logp("data in %s is not the right length (%lu!=9)\n", path,
287 : (unsigned long)s);
288 0 : goto end;
289 : }
290 1 : manio->offset->fcount=strtoul(buf, NULL, 16);
291 1 : ret=0;
292 : end:
293 1 : fzp_close(&fzp);
294 1 : free_w(&path);
295 1 : return ret;
296 : }
297 :
298 1085 : static int sort_and_write_hooks(struct manio *manio)
299 : {
300 : int i;
301 1085 : int ret=-1;
302 1085 : struct fzp *fzp=NULL;
303 1085 : char msg[32]="";
304 1085 : char *path=NULL;
305 1085 : int hook_count=manio->hook_count;
306 1085 : uint64_t *hook_sort=manio->hook_sort;
307 1085 : if(!hook_sort) return 0;
308 :
309 20 : snprintf(msg, sizeof(msg), "%08" PRIX64, manio->offset->fcount-1);
310 20 : if(!(path=prepend_s(manio->hook_dir, msg))
311 20 : || build_path_w(path)
312 20 : || !(fzp=fzp_gzopen(path, MANIO_MODE_WRITE)))
313 : goto end;
314 :
315 20 : qsort(hook_sort, hook_count, sizeof(uint64_t), uint64_t_sort);
316 :
317 20 : if(write_hook_header(fzp, manio->rmanifest, msg)) goto end;
318 2208 : for(i=0; i<hook_count; i++)
319 : {
320 : // Do not bother with duplicates.
321 2208 : if(i && hook_sort[i]==hook_sort[i-1])
322 0 : continue;
323 :
324 2208 : if(to_fzp_fingerprint(fzp, hook_sort[i]))
325 : goto end;
326 : }
327 20 : if(fzp_close(&fzp))
328 : {
329 0 : logp("Error closing %s in %s: %s\n",
330 0 : path, __func__, strerror(errno));
331 0 : goto end;
332 : }
333 20 : if(manio_write_fcount(manio)) goto end;
334 20 : manio->hook_count=0;
335 20 : ret=0;
336 : end:
337 20 : fzp_close(&fzp);
338 20 : free_w(&path);
339 20 : return ret;
340 : }
341 :
342 1085 : static int sort_and_write_dindex(struct manio *manio)
343 : {
344 : int i;
345 1085 : int ret=-1;
346 1085 : struct fzp *fzp=NULL;
347 1085 : char msg[32]="";
348 1085 : char *path=NULL;
349 : struct iobuf wbuf;
350 : struct blk blk;
351 1085 : int dindex_count=manio->dindex_count;
352 1085 : uint64_t *dindex_sort=manio->dindex_sort;
353 1085 : if(!dindex_sort) return 0;
354 :
355 20 : snprintf(msg, sizeof(msg), "%08" PRIX64, manio->offset->fcount-1);
356 20 : if(!(path=prepend_s(manio->dindex_dir, msg))
357 20 : || build_path_w(path)
358 20 : || !(fzp=fzp_gzopen(path, MANIO_MODE_WRITE)))
359 : goto end;
360 :
361 20 : qsort(dindex_sort, dindex_count, sizeof(uint64_t), uint64_t_sort);
362 :
363 34899 : for(i=0; i<dindex_count; i++)
364 : {
365 : // Do not bother with duplicates.
366 34879 : if(i && dindex_sort[i]==dindex_sort[i-1])
367 0 : continue;
368 :
369 34879 : blk.savepath=dindex_sort[i];
370 34879 : blk_to_iobuf_savepath(&blk, &wbuf);
371 34879 : if(iobuf_send_msg_fzp(&wbuf, fzp)) return -1;
372 : }
373 20 : if(fzp_close(&fzp))
374 : {
375 0 : logp("Error closing %s in %s: %s\n",
376 0 : path, __func__, strerror(errno));
377 0 : goto end;
378 : }
379 20 : manio->dindex_count=0;
380 20 : ret=0;
381 : end:
382 20 : fzp_close(&fzp);
383 20 : free_w(&path);
384 20 : return ret;
385 : }
386 :
387 1085 : static int sort_and_write_hooks_and_dindex(struct manio *manio)
388 : {
389 1085 : return sort_and_write_hooks(manio)
390 1085 : || sort_and_write_dindex(manio);
391 : }
392 :
393 924 : int manio_close(struct manio **manio)
394 : {
395 924 : int ret=0;
396 : // int fd;
397 924 : if(!manio || !*manio) return ret;
398 756 : if(sort_and_write_hooks_and_dindex(*manio))
399 0 : ret=-1;
400 : /*
401 : There is no gzfileno()
402 : if((fd=fzp_fileno((*manio)->fzp))<0)
403 : {
404 : logp("Could not get fileno in %s for %s: %s\n", __func__,
405 : (*manio)->manifest, strerror(errno));
406 : ret=-1;
407 : }
408 : // Should probably have a flush before fsync too.
409 : if(fsync(fd))
410 : {
411 : logp("Error in fsync in %s for %s: %s\n", __func__,
412 : (*manio)->manifest, strerror(errno));
413 : ret=-1;
414 : }
415 : */
416 756 : if(fzp_close(&((*manio)->fzp)))
417 0 : ret=-1;
418 756 : sync();
419 756 : manio_free_content(*manio);
420 756 : free_v((void **)manio);
421 756 : return ret;
422 : }
423 :
424 : // Return -1 for error, 0 for stuff read OK, 1 for end of files.
425 115436 : int manio_read_with_blk(struct manio *manio, struct sbuf *sb, struct blk *blk)
426 : {
427 : while(1)
428 : {
429 115594 : if(!manio->fzp)
430 : {
431 179 : if(manio_open_next_fpath(manio)) goto error;
432 179 : if(!manio->fzp) return 1; // No more files to read.
433 : }
434 :
435 115463 : switch(sbuf_fill_from_file(sb, manio->fzp, blk))
436 : {
437 : case 0: return 0; // Got something.
438 : case 1: break; // Keep going.
439 : default: goto error; // Error.
440 : }
441 :
442 : // Reached the end of the current file.
443 : // Maybe there is another file to continue with.
444 286 : if(sort_and_write_hooks_and_dindex(manio)
445 286 : || fzp_close(&manio->fzp)) goto error;
446 :
447 572 : if(is_single_file(manio)) return 1;
448 : }
449 :
450 : error:
451 : return -1;
452 : }
453 :
454 54944 : int manio_read(struct manio *manio, struct sbuf *sb)
455 : {
456 54944 : return manio_read_with_blk(manio, sb, NULL);
457 : }
458 :
459 43 : static int reset_sig_count_and_close(struct manio *manio)
460 : {
461 43 : if(sort_and_write_hooks_and_dindex(manio)) return -1;
462 43 : if(fzp_close(&manio->fzp)) return -1;
463 43 : manio->sig_count=0;
464 43 : if(manio_open_next_fpath(manio)) return -1;
465 43 : return 0;
466 : }
467 :
468 : #ifndef UTEST
469 : static
470 : #endif
471 3574 : int manio_find_boundary(uint8_t *md5sum)
472 : {
473 : int i;
474 : uint8_t x;
475 : uint8_t y;
476 3574 : uint8_t b4=0;
477 :
478 : // I am currently making it look for four of the same consecutive
479 : // characters in the md5sum.
480 56854 : for(i=0; i<MD5_DIGEST_LENGTH-1; i++)
481 : {
482 53333 : x=md5sum[i]>>4;
483 53333 : y=md5sum[i]&0x0F;
484 53333 : if(x==y)
485 : {
486 3303 : if(x!=md5sum[i+1]>>4)
487 3106 : continue;
488 197 : if(i && x==b4)
489 : return 1;
490 175 : if(x==(md5sum[i+1]&0x0F))
491 : return 1;
492 : }
493 : b4=y;
494 : }
495 : return 0;
496 : }
497 :
498 : // After conditions are met, close the file currently being written to.
499 : // Allow the number of signatures to be vary between MANIFEST_SIG_MIN and
500 : // MANIFEST_SIG_MAX. This will hopefully allow fewer candidate manifests
501 : // generated, since the boundary will be able to vary.
502 187793 : static int check_sig_count(struct manio *manio, struct blk *blk)
503 : {
504 187793 : manio->sig_count++;
505 :
506 187793 : if(manio->sig_count<MANIFEST_SIG_MIN)
507 : return 0; // Not yet time to close.
508 :
509 3558 : if(manio->sig_count>=MANIFEST_SIG_MAX)
510 0 : return reset_sig_count_and_close(manio); // Time to close.
511 :
512 : // At this point, dynamically decide based on the current msg.
513 3558 : if(manio_find_boundary(blk->md5sum))
514 43 : return reset_sig_count_and_close(manio); // Time to close.
515 : return 0;
516 : }
517 :
518 187793 : static int write_sig_msg(struct manio *manio, struct blk *blk)
519 : {
520 : struct iobuf wbuf;
521 187793 : if(!manio->fzp && manio_open_next_fpath(manio)) return -1;
522 187793 : blk_to_iobuf_sig_and_savepath(blk, &wbuf);
523 187793 : if(iobuf_send_msg_fzp(&wbuf, manio->fzp)) return -1;
524 187793 : return check_sig_count(manio, blk);
525 : }
526 :
527 187793 : int manio_write_sig_and_path(struct manio *manio, struct blk *blk)
528 : {
529 187793 : if(manio->protocol==PROTO_1) return 0;
530 187793 : if(manio->hook_sort && blk_fingerprint_is_hook(blk))
531 : {
532 : // Add to list of hooks for this manifest chunk.
533 2208 : manio->hook_sort[manio->hook_count++]=blk->fingerprint;
534 : }
535 187793 : if(manio->dindex_sort)
536 : {
537 35171 : uint64_t savepath=blk->savepath;
538 35171 : savepath &= 0xFFFFFFFFFFFF0000ULL;
539 : // Ignore obvious duplicates.
540 35171 : if(!manio->dindex_count
541 35152 : || manio->dindex_sort[manio->dindex_count-1]!=savepath)
542 : {
543 : // Add to list of dindexes for this manifest chunk.
544 34879 : manio->dindex_sort[manio->dindex_count++]=savepath;
545 : }
546 : }
547 187793 : return write_sig_msg(manio, blk);
548 : }
549 :
550 91874 : int manio_write_sbuf(struct manio *manio, struct sbuf *sb)
551 : {
552 91874 : if(!manio->fzp && manio_open_next_fpath(manio)) return -1;
553 91874 : return sbuf_to_manifest(sb, manio->fzp);
554 : }
555 :
556 : // Return -1 on error, 0 on OK, 1 for srcmanio finished.
557 18 : int manio_copy_entry(struct sbuf *csb, struct sbuf *sb,
558 : struct manio *srcmanio, struct manio *dstmanio)
559 : {
560 : int ars;
561 : struct iobuf copy;
562 : struct blk *blk;
563 18 : if(!(blk=blk_alloc()))
564 : goto error;
565 :
566 : // Use the most recent stat for the new manifest.
567 18 : if(dstmanio)
568 : {
569 18 : if(manio_write_sbuf(dstmanio, sb)) goto error;
570 18 : if(dstmanio->protocol==PROTO_1)
571 : {
572 8 : sbuf_free_content(csb);
573 8 : blk_free(&blk);
574 8 : return 0;
575 : }
576 : }
577 :
578 10 : copy.len=csb->path.len;
579 10 : copy.cmd=csb->path.cmd;
580 10 : if(!(copy.buf=strdup_w(csb->path.buf, __func__)))
581 : goto error;
582 : while(1)
583 : {
584 30 : if((ars=manio_read_with_blk(srcmanio, csb, blk))<0)
585 : goto error;
586 30 : else if(ars>0)
587 : {
588 : // Finished.
589 5 : sbuf_free_content(csb);
590 5 : blk_free(&blk);
591 5 : iobuf_free_content(©);
592 5 : return 1;
593 : }
594 :
595 : // Got something.
596 25 : if(iobuf_pathcmp(&csb->path, ©))
597 : {
598 : // Found the next entry.
599 5 : iobuf_free_content(©);
600 5 : blk_free(&blk);
601 5 : return 0;
602 : }
603 20 : if(dstmanio)
604 : {
605 20 : if(!dstmanio->fzp
606 0 : && manio_open_next_fpath(dstmanio))
607 : goto error;
608 :
609 20 : if(csb->endfile.buf)
610 : {
611 10 : if(iobuf_send_msg_fzp(&csb->endfile,
612 : dstmanio->fzp)) goto error;
613 : }
614 : else
615 : {
616 : // Should have the next signature.
617 : // Write it to the destination manifest.
618 10 : if(manio_write_sig_and_path(dstmanio, blk))
619 : goto error;
620 : }
621 : }
622 : }
623 :
624 : error:
625 0 : blk_free(&blk);
626 0 : iobuf_free_content(©);
627 0 : return -1;
628 : }
629 :
630 0 : int manio_forward_through_sigs(struct sbuf *csb, struct manio *manio)
631 : {
632 : // Call manio_copy_entry with nothing to write to, so
633 : // that we forward through the sigs in manio.
634 0 : return manio_copy_entry(csb, NULL, manio, NULL);
635 : }
636 :
637 12990 : man_off_t *manio_tell(struct manio *manio)
638 : {
639 12990 : man_off_t *offset=NULL;
640 12990 : if(!manio->fzp)
641 : {
642 0 : logp("manio_tell called on null fzp\n");
643 0 : goto error;
644 : }
645 12990 : if(!(offset=man_off_t_alloc())
646 12990 : || !(offset->fpath=strdup_w(manio->offset->fpath, __func__))
647 12990 : || (offset->offset=fzp_tell(manio->fzp))<0)
648 : goto error;
649 12990 : offset->fcount=manio->offset->fcount;
650 12990 : return offset;
651 : error:
652 0 : man_off_t_free(&offset);
653 0 : return NULL;
654 : }
655 :
656 246 : int manio_seek(struct manio *manio, man_off_t *offset)
657 : {
658 246 : fzp_close(&manio->fzp);
659 246 : if(!(manio->fzp=fzp_gzopen(offset->fpath, manio->mode))
660 246 : || fzp_seek(manio->fzp, offset->offset, SEEK_SET))
661 : return -1;
662 492 : man_off_t_free_content(manio->offset);
663 246 : if(!(manio->offset->fpath=strdup_w(offset->fpath, __func__)))
664 : return -1;
665 246 : manio->offset->offset=offset->offset;
666 246 : manio->offset->fcount=offset->fcount;
667 246 : return 0;
668 : }
669 :
670 100 : static int remove_trailing_files(struct manio *manio, man_off_t *offset)
671 : {
672 100 : int ret=-1;
673 100 : char *fpath=NULL;
674 : struct stat statp;
675 : while(1)
676 : {
677 100 : free_w(&fpath);
678 100 : if(!(fpath=get_next_fpath(manio, offset)))
679 : goto end;
680 200 : if(lstat(fpath, &statp)) break;
681 0 : if(!S_ISREG(statp.st_mode))
682 : goto end;
683 0 : if(recursive_delete(fpath))
684 : goto end;
685 : }
686 : ret=0;
687 : end:
688 100 : free_w(&fpath);
689 100 : return ret;
690 : }
691 :
692 200 : int manio_close_and_truncate(struct manio **manio,
693 : man_off_t *offset, int compression)
694 : {
695 200 : int ret=-1;
696 200 : errno=0;
697 400 : if(!is_single_file(*manio)
698 100 : && remove_trailing_files(*manio, offset))
699 : goto end;
700 200 : if(manio_close(manio)) goto end;
701 200 : if(fzp_truncate(offset->fpath, FZP_FILE, offset->offset, compression))
702 : {
703 0 : logp("Could not fzp_truncate %s in %s(): %s\n",
704 : offset->fpath, __func__, strerror(errno));
705 0 : goto end;
706 : }
707 : ret=0;
708 : end:
709 200 : return ret;
710 : }
|