archive_write.c revision 368707
1/*- 2 * Copyright (c) 2003-2010 Tim Kientzle 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR 15 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 16 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 17 * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT, 18 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 19 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 20 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 21 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 23 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 24 */ 25 26#include "archive_platform.h" 27__FBSDID("$FreeBSD: stable/11/contrib/libarchive/libarchive/archive_write.c 368707 2020-12-16 22:25:20Z mm $"); 28 29/* 30 * This file contains the "essential" portions of the write API, that 31 * is, stuff that will essentially always be used by any client that 32 * actually needs to write an archive. Optional pieces have been, as 33 * far as possible, separated out into separate files to reduce 34 * needlessly bloating statically-linked clients. 35 */ 36 37#ifdef HAVE_SYS_WAIT_H 38#include <sys/wait.h> 39#endif 40#ifdef HAVE_ERRNO_H 41#include <errno.h> 42#endif 43#ifdef HAVE_LIMITS_H 44#include <limits.h> 45#endif 46#include <stdio.h> 47#ifdef HAVE_STDLIB_H 48#include <stdlib.h> 49#endif 50#ifdef HAVE_STRING_H 51#include <string.h> 52#endif 53#include <time.h> 54#ifdef HAVE_UNISTD_H 55#include <unistd.h> 56#endif 57 58#include "archive.h" 59#include "archive_entry.h" 60#include "archive_private.h" 61#include "archive_write_private.h" 62 63static struct archive_vtable *archive_write_vtable(void); 64 65static int _archive_filter_code(struct archive *, int); 66static const char *_archive_filter_name(struct archive *, int); 67static int64_t _archive_filter_bytes(struct archive *, int); 68static int _archive_write_filter_count(struct archive *); 69static int _archive_write_close(struct archive *); 70static int _archive_write_free(struct archive *); 71static int _archive_write_header(struct archive *, struct archive_entry *); 72static int _archive_write_finish_entry(struct archive *); 73static ssize_t _archive_write_data(struct archive *, const void *, size_t); 74 75struct archive_none { 76 size_t buffer_size; 77 size_t avail; 78 char *buffer; 79 char *next; 80}; 81 82static struct archive_vtable * 83archive_write_vtable(void) 84{ 85 static struct archive_vtable av; 86 static int inited = 0; 87 88 if (!inited) { 89 av.archive_close = _archive_write_close; 90 av.archive_filter_bytes = _archive_filter_bytes; 91 av.archive_filter_code = _archive_filter_code; 92 av.archive_filter_name = _archive_filter_name; 93 av.archive_filter_count = _archive_write_filter_count; 94 av.archive_free = _archive_write_free; 95 av.archive_write_header = _archive_write_header; 96 av.archive_write_finish_entry = _archive_write_finish_entry; 97 av.archive_write_data = _archive_write_data; 98 inited = 1; 99 } 100 return (&av); 101} 102 103/* 104 * Allocate, initialize and return an archive object. 105 */ 106struct archive * 107archive_write_new(void) 108{ 109 struct archive_write *a; 110 unsigned char *nulls; 111 112 a = (struct archive_write *)calloc(1, sizeof(*a)); 113 if (a == NULL) 114 return (NULL); 115 a->archive.magic = ARCHIVE_WRITE_MAGIC; 116 a->archive.state = ARCHIVE_STATE_NEW; 117 a->archive.vtable = archive_write_vtable(); 118 /* 119 * The value 10240 here matches the traditional tar default, 120 * but is otherwise arbitrary. 121 * TODO: Set the default block size from the format selected. 122 */ 123 a->bytes_per_block = 10240; 124 a->bytes_in_last_block = -1; /* Default */ 125 126 /* Initialize a block of nulls for padding purposes. */ 127 a->null_length = 1024; 128 nulls = (unsigned char *)calloc(1, a->null_length); 129 if (nulls == NULL) { 130 free(a); 131 return (NULL); 132 } 133 a->nulls = nulls; 134 return (&a->archive); 135} 136 137/* 138 * Set the block size. Returns 0 if successful. 139 */ 140int 141archive_write_set_bytes_per_block(struct archive *_a, int bytes_per_block) 142{ 143 struct archive_write *a = (struct archive_write *)_a; 144 archive_check_magic(&a->archive, ARCHIVE_WRITE_MAGIC, 145 ARCHIVE_STATE_NEW, "archive_write_set_bytes_per_block"); 146 a->bytes_per_block = bytes_per_block; 147 return (ARCHIVE_OK); 148} 149 150/* 151 * Get the current block size. -1 if it has never been set. 152 */ 153int 154archive_write_get_bytes_per_block(struct archive *_a) 155{ 156 struct archive_write *a = (struct archive_write *)_a; 157 archive_check_magic(&a->archive, ARCHIVE_WRITE_MAGIC, 158 ARCHIVE_STATE_ANY, "archive_write_get_bytes_per_block"); 159 return (a->bytes_per_block); 160} 161 162/* 163 * Set the size for the last block. 164 * Returns 0 if successful. 165 */ 166int 167archive_write_set_bytes_in_last_block(struct archive *_a, int bytes) 168{ 169 struct archive_write *a = (struct archive_write *)_a; 170 archive_check_magic(&a->archive, ARCHIVE_WRITE_MAGIC, 171 ARCHIVE_STATE_ANY, "archive_write_set_bytes_in_last_block"); 172 a->bytes_in_last_block = bytes; 173 return (ARCHIVE_OK); 174} 175 176/* 177 * Return the value set above. -1 indicates it has not been set. 178 */ 179int 180archive_write_get_bytes_in_last_block(struct archive *_a) 181{ 182 struct archive_write *a = (struct archive_write *)_a; 183 archive_check_magic(&a->archive, ARCHIVE_WRITE_MAGIC, 184 ARCHIVE_STATE_ANY, "archive_write_get_bytes_in_last_block"); 185 return (a->bytes_in_last_block); 186} 187 188/* 189 * dev/ino of a file to be rejected. Used to prevent adding 190 * an archive to itself recursively. 191 */ 192int 193archive_write_set_skip_file(struct archive *_a, la_int64_t d, la_int64_t i) 194{ 195 struct archive_write *a = (struct archive_write *)_a; 196 archive_check_magic(&a->archive, ARCHIVE_WRITE_MAGIC, 197 ARCHIVE_STATE_ANY, "archive_write_set_skip_file"); 198 a->skip_file_set = 1; 199 a->skip_file_dev = d; 200 a->skip_file_ino = i; 201 return (ARCHIVE_OK); 202} 203 204/* 205 * Allocate and return the next filter structure. 206 */ 207struct archive_write_filter * 208__archive_write_allocate_filter(struct archive *_a) 209{ 210 struct archive_write *a = (struct archive_write *)_a; 211 struct archive_write_filter *f; 212 213 f = calloc(1, sizeof(*f)); 214 f->archive = _a; 215 f->state = ARCHIVE_WRITE_FILTER_STATE_NEW; 216 if (a->filter_first == NULL) 217 a->filter_first = f; 218 else 219 a->filter_last->next_filter = f; 220 a->filter_last = f; 221 return f; 222} 223 224/* 225 * Write data to a particular filter. 226 */ 227int 228__archive_write_filter(struct archive_write_filter *f, 229 const void *buff, size_t length) 230{ 231 int r; 232 /* Never write to non-open filters */ 233 if (f->state != ARCHIVE_WRITE_FILTER_STATE_OPEN) 234 return(ARCHIVE_FATAL); 235 if (length == 0) 236 return(ARCHIVE_OK); 237 if (f->write == NULL) 238 /* If unset, a fatal error has already occurred, so this filter 239 * didn't open. We cannot write anything. */ 240 return(ARCHIVE_FATAL); 241 r = (f->write)(f, buff, length); 242 f->bytes_written += length; 243 return (r); 244} 245 246/* 247 * Recursive function for opening the filter chain 248 * Last filter is opened first 249 */ 250static int 251__archive_write_open_filter(struct archive_write_filter *f) 252{ 253 int ret; 254 255 ret = ARCHIVE_OK; 256 if (f->next_filter != NULL) 257 ret = __archive_write_open_filter(f->next_filter); 258 if (ret != ARCHIVE_OK) 259 return (ret); 260 if (f->state != ARCHIVE_WRITE_FILTER_STATE_NEW) 261 return (ARCHIVE_FATAL); 262 if (f->open == NULL) { 263 f->state = ARCHIVE_WRITE_FILTER_STATE_OPEN; 264 return (ARCHIVE_OK); 265 } 266 ret = (f->open)(f); 267 if (ret == ARCHIVE_OK) 268 f->state = ARCHIVE_WRITE_FILTER_STATE_OPEN; 269 else 270 f->state = ARCHIVE_WRITE_FILTER_STATE_FATAL; 271 return (ret); 272} 273 274/* 275 * Open all filters 276 */ 277static int 278__archive_write_filters_open(struct archive_write *a) 279{ 280 return (__archive_write_open_filter(a->filter_first)); 281} 282 283/* 284 * Close all filtes 285 */ 286static int 287__archive_write_filters_close(struct archive_write *a) 288{ 289 struct archive_write_filter *f; 290 int ret, ret1; 291 ret = ARCHIVE_OK; 292 for (f = a->filter_first; f != NULL; f = f->next_filter) { 293 /* Do not close filters that are not open */ 294 if (f->state == ARCHIVE_WRITE_FILTER_STATE_OPEN) { 295 if (f->close != NULL) { 296 ret1 = (f->close)(f); 297 if (ret1 < ret) 298 ret = ret1; 299 if (ret1 == ARCHIVE_OK) { 300 f->state = 301 ARCHIVE_WRITE_FILTER_STATE_CLOSED; 302 } else { 303 f->state = 304 ARCHIVE_WRITE_FILTER_STATE_FATAL; 305 } 306 } else 307 f->state = ARCHIVE_WRITE_FILTER_STATE_CLOSED; 308 } 309 } 310 return (ret); 311} 312 313int 314__archive_write_output(struct archive_write *a, const void *buff, size_t length) 315{ 316 return (__archive_write_filter(a->filter_first, buff, length)); 317} 318 319int 320__archive_write_nulls(struct archive_write *a, size_t length) 321{ 322 if (length == 0) 323 return (ARCHIVE_OK); 324 325 while (length > 0) { 326 size_t to_write = length < a->null_length ? length : a->null_length; 327 int r = __archive_write_output(a, a->nulls, to_write); 328 if (r < ARCHIVE_OK) 329 return (r); 330 length -= to_write; 331 } 332 return (ARCHIVE_OK); 333} 334 335static int 336archive_write_client_open(struct archive_write_filter *f) 337{ 338 struct archive_write *a = (struct archive_write *)f->archive; 339 struct archive_none *state; 340 void *buffer; 341 size_t buffer_size; 342 int ret; 343 344 f->bytes_per_block = archive_write_get_bytes_per_block(f->archive); 345 f->bytes_in_last_block = 346 archive_write_get_bytes_in_last_block(f->archive); 347 buffer_size = f->bytes_per_block; 348 349 state = (struct archive_none *)calloc(1, sizeof(*state)); 350 buffer = (char *)malloc(buffer_size); 351 if (state == NULL || buffer == NULL) { 352 free(state); 353 free(buffer); 354 archive_set_error(f->archive, ENOMEM, 355 "Can't allocate data for output buffering"); 356 return (ARCHIVE_FATAL); 357 } 358 359 state->buffer_size = buffer_size; 360 state->buffer = buffer; 361 state->next = state->buffer; 362 state->avail = state->buffer_size; 363 f->data = state; 364 365 if (a->client_opener == NULL) 366 return (ARCHIVE_OK); 367 ret = a->client_opener(f->archive, a->client_data); 368 if (ret != ARCHIVE_OK) { 369 free(state->buffer); 370 free(state); 371 f->data = NULL; 372 } 373 return (ret); 374} 375 376static int 377archive_write_client_write(struct archive_write_filter *f, 378 const void *_buff, size_t length) 379{ 380 struct archive_write *a = (struct archive_write *)f->archive; 381 struct archive_none *state = (struct archive_none *)f->data; 382 const char *buff = (const char *)_buff; 383 ssize_t remaining, to_copy; 384 ssize_t bytes_written; 385 386 remaining = length; 387 388 /* 389 * If there is no buffer for blocking, just pass the data 390 * straight through to the client write callback. In 391 * particular, this supports "no write delay" operation for 392 * special applications. Just set the block size to zero. 393 */ 394 if (state->buffer_size == 0) { 395 while (remaining > 0) { 396 bytes_written = (a->client_writer)(&a->archive, 397 a->client_data, buff, remaining); 398 if (bytes_written <= 0) 399 return (ARCHIVE_FATAL); 400 remaining -= bytes_written; 401 buff += bytes_written; 402 } 403 return (ARCHIVE_OK); 404 } 405 406 /* If the copy buffer isn't empty, try to fill it. */ 407 if (state->avail < state->buffer_size) { 408 /* If buffer is not empty... */ 409 /* ... copy data into buffer ... */ 410 to_copy = ((size_t)remaining > state->avail) ? 411 state->avail : (size_t)remaining; 412 memcpy(state->next, buff, to_copy); 413 state->next += to_copy; 414 state->avail -= to_copy; 415 buff += to_copy; 416 remaining -= to_copy; 417 /* ... if it's full, write it out. */ 418 if (state->avail == 0) { 419 char *p = state->buffer; 420 size_t to_write = state->buffer_size; 421 while (to_write > 0) { 422 bytes_written = (a->client_writer)(&a->archive, 423 a->client_data, p, to_write); 424 if (bytes_written <= 0) 425 return (ARCHIVE_FATAL); 426 if ((size_t)bytes_written > to_write) { 427 archive_set_error(&(a->archive), 428 -1, "write overrun"); 429 return (ARCHIVE_FATAL); 430 } 431 p += bytes_written; 432 to_write -= bytes_written; 433 } 434 state->next = state->buffer; 435 state->avail = state->buffer_size; 436 } 437 } 438 439 while ((size_t)remaining >= state->buffer_size) { 440 /* Write out full blocks directly to client. */ 441 bytes_written = (a->client_writer)(&a->archive, 442 a->client_data, buff, state->buffer_size); 443 if (bytes_written <= 0) 444 return (ARCHIVE_FATAL); 445 buff += bytes_written; 446 remaining -= bytes_written; 447 } 448 449 if (remaining > 0) { 450 /* Copy last bit into copy buffer. */ 451 memcpy(state->next, buff, remaining); 452 state->next += remaining; 453 state->avail -= remaining; 454 } 455 return (ARCHIVE_OK); 456} 457 458static int 459archive_write_client_free(struct archive_write_filter *f) 460{ 461 struct archive_write *a = (struct archive_write *)f->archive; 462 463 if (a->client_freer) 464 (*a->client_freer)(&a->archive, a->client_data); 465 a->client_data = NULL; 466 467 /* Clear passphrase. */ 468 if (a->passphrase != NULL) { 469 memset(a->passphrase, 0, strlen(a->passphrase)); 470 free(a->passphrase); 471 a->passphrase = NULL; 472 } 473 474 return (ARCHIVE_OK); 475} 476 477static int 478archive_write_client_close(struct archive_write_filter *f) 479{ 480 struct archive_write *a = (struct archive_write *)f->archive; 481 struct archive_none *state = (struct archive_none *)f->data; 482 ssize_t block_length; 483 ssize_t target_block_length; 484 ssize_t bytes_written; 485 int ret = ARCHIVE_OK; 486 487 /* If there's pending data, pad and write the last block */ 488 if (state->next != state->buffer) { 489 block_length = state->buffer_size - state->avail; 490 491 /* Tricky calculation to determine size of last block */ 492 if (a->bytes_in_last_block <= 0) 493 /* Default or Zero: pad to full block */ 494 target_block_length = a->bytes_per_block; 495 else 496 /* Round to next multiple of bytes_in_last_block. */ 497 target_block_length = a->bytes_in_last_block * 498 ( (block_length + a->bytes_in_last_block - 1) / 499 a->bytes_in_last_block); 500 if (target_block_length > a->bytes_per_block) 501 target_block_length = a->bytes_per_block; 502 if (block_length < target_block_length) { 503 memset(state->next, 0, 504 target_block_length - block_length); 505 block_length = target_block_length; 506 } 507 bytes_written = (a->client_writer)(&a->archive, 508 a->client_data, state->buffer, block_length); 509 ret = bytes_written <= 0 ? ARCHIVE_FATAL : ARCHIVE_OK; 510 } 511 if (a->client_closer) 512 (*a->client_closer)(&a->archive, a->client_data); 513 free(state->buffer); 514 free(state); 515 516 /* Clear the close handler myself not to be called again. */ 517 f->state = ARCHIVE_WRITE_FILTER_STATE_CLOSED; 518 return (ret); 519} 520 521/* 522 * Open the archive using the current settings. 523 */ 524int 525archive_write_open2(struct archive *_a, void *client_data, 526 archive_open_callback *opener, archive_write_callback *writer, 527 archive_close_callback *closer, archive_free_callback *freer) 528{ 529 struct archive_write *a = (struct archive_write *)_a; 530 struct archive_write_filter *client_filter; 531 int ret, r1; 532 533 archive_check_magic(&a->archive, ARCHIVE_WRITE_MAGIC, 534 ARCHIVE_STATE_NEW, "archive_write_open"); 535 archive_clear_error(&a->archive); 536 537 a->client_writer = writer; 538 a->client_opener = opener; 539 a->client_closer = closer; 540 a->client_freer = freer; 541 a->client_data = client_data; 542 543 client_filter = __archive_write_allocate_filter(_a); 544 client_filter->open = archive_write_client_open; 545 client_filter->write = archive_write_client_write; 546 client_filter->close = archive_write_client_close; 547 client_filter->free = archive_write_client_free; 548 549 ret = __archive_write_filters_open(a); 550 if (ret < ARCHIVE_WARN) { 551 r1 = __archive_write_filters_close(a); 552 __archive_write_filters_free(_a); 553 return (r1 < ret ? r1 : ret); 554 } 555 556 a->archive.state = ARCHIVE_STATE_HEADER; 557 if (a->format_init) 558 ret = (a->format_init)(a); 559 return (ret); 560} 561 562int 563archive_write_open(struct archive *_a, void *client_data, 564 archive_open_callback *opener, archive_write_callback *writer, 565 archive_close_callback *closer) 566{ 567 return archive_write_open2(_a, client_data, opener, writer, 568 closer, NULL); 569} 570 571/* 572 * Close out the archive. 573 */ 574static int 575_archive_write_close(struct archive *_a) 576{ 577 struct archive_write *a = (struct archive_write *)_a; 578 int r = ARCHIVE_OK, r1 = ARCHIVE_OK; 579 580 archive_check_magic(&a->archive, ARCHIVE_WRITE_MAGIC, 581 ARCHIVE_STATE_ANY | ARCHIVE_STATE_FATAL, 582 "archive_write_close"); 583 if (a->archive.state == ARCHIVE_STATE_NEW 584 || a->archive.state == ARCHIVE_STATE_CLOSED) 585 return (ARCHIVE_OK); /* Okay to close() when not open. */ 586 587 archive_clear_error(&a->archive); 588 589 /* Finish the last entry if a finish callback is specified */ 590 if (a->archive.state == ARCHIVE_STATE_DATA 591 && a->format_finish_entry != NULL) 592 r = ((a->format_finish_entry)(a)); 593 594 /* Finish off the archive. */ 595 /* TODO: have format closers invoke compression close. */ 596 if (a->format_close != NULL) { 597 r1 = (a->format_close)(a); 598 if (r1 < r) 599 r = r1; 600 } 601 602 /* Finish the compression and close the stream. */ 603 r1 = __archive_write_filters_close(a); 604 if (r1 < r) 605 r = r1; 606 607 if (a->archive.state != ARCHIVE_STATE_FATAL) 608 a->archive.state = ARCHIVE_STATE_CLOSED; 609 return (r); 610} 611 612static int 613_archive_write_filter_count(struct archive *_a) 614{ 615 struct archive_write *a = (struct archive_write *)_a; 616 struct archive_write_filter *p = a->filter_first; 617 int count = 0; 618 while(p) { 619 count++; 620 p = p->next_filter; 621 } 622 return count; 623} 624 625void 626__archive_write_filters_free(struct archive *_a) 627{ 628 struct archive_write *a = (struct archive_write *)_a; 629 int r = ARCHIVE_OK, r1; 630 631 while (a->filter_first != NULL) { 632 struct archive_write_filter *next 633 = a->filter_first->next_filter; 634 if (a->filter_first->free != NULL) { 635 r1 = (*a->filter_first->free)(a->filter_first); 636 if (r > r1) 637 r = r1; 638 } 639 free(a->filter_first); 640 a->filter_first = next; 641 } 642 a->filter_last = NULL; 643} 644 645/* 646 * Destroy the archive structure. 647 * 648 * Be careful: user might just call write_new and then write_free. 649 * Don't assume we actually wrote anything or performed any non-trivial 650 * initialization. 651 */ 652static int 653_archive_write_free(struct archive *_a) 654{ 655 struct archive_write *a = (struct archive_write *)_a; 656 int r = ARCHIVE_OK, r1; 657 658 if (_a == NULL) 659 return (ARCHIVE_OK); 660 /* It is okay to call free() in state FATAL. */ 661 archive_check_magic(&a->archive, ARCHIVE_WRITE_MAGIC, 662 ARCHIVE_STATE_ANY | ARCHIVE_STATE_FATAL, "archive_write_free"); 663 if (a->archive.state != ARCHIVE_STATE_FATAL) 664 r = archive_write_close(&a->archive); 665 666 /* Release format resources. */ 667 if (a->format_free != NULL) { 668 r1 = (a->format_free)(a); 669 if (r1 < r) 670 r = r1; 671 } 672 673 __archive_write_filters_free(_a); 674 675 /* Release various dynamic buffers. */ 676 free((void *)(uintptr_t)(const void *)a->nulls); 677 archive_string_free(&a->archive.error_string); 678 if (a->passphrase != NULL) { 679 /* A passphrase should be cleaned. */ 680 memset(a->passphrase, 0, strlen(a->passphrase)); 681 free(a->passphrase); 682 } 683 a->archive.magic = 0; 684 __archive_clean(&a->archive); 685 free(a); 686 return (r); 687} 688 689/* 690 * Write the appropriate header. 691 */ 692static int 693_archive_write_header(struct archive *_a, struct archive_entry *entry) 694{ 695 struct archive_write *a = (struct archive_write *)_a; 696 int ret, r2; 697 698 archive_check_magic(&a->archive, ARCHIVE_WRITE_MAGIC, 699 ARCHIVE_STATE_DATA | ARCHIVE_STATE_HEADER, "archive_write_header"); 700 archive_clear_error(&a->archive); 701 702 if (a->format_write_header == NULL) { 703 archive_set_error(&(a->archive), -1, 704 "Format must be set before you can write to an archive."); 705 a->archive.state = ARCHIVE_STATE_FATAL; 706 return (ARCHIVE_FATAL); 707 } 708 709 /* In particular, "retry" and "fatal" get returned immediately. */ 710 ret = archive_write_finish_entry(&a->archive); 711 if (ret == ARCHIVE_FATAL) { 712 a->archive.state = ARCHIVE_STATE_FATAL; 713 return (ARCHIVE_FATAL); 714 } 715 if (ret < ARCHIVE_OK && ret != ARCHIVE_WARN) 716 return (ret); 717 718 if (a->skip_file_set && 719 archive_entry_dev_is_set(entry) && 720 archive_entry_ino_is_set(entry) && 721 archive_entry_dev(entry) == (dev_t)a->skip_file_dev && 722 archive_entry_ino64(entry) == a->skip_file_ino) { 723 archive_set_error(&a->archive, 0, 724 "Can't add archive to itself"); 725 return (ARCHIVE_FAILED); 726 } 727 728 /* Format and write header. */ 729 r2 = ((a->format_write_header)(a, entry)); 730 if (r2 == ARCHIVE_FAILED) { 731 return (ARCHIVE_FAILED); 732 } 733 if (r2 == ARCHIVE_FATAL) { 734 a->archive.state = ARCHIVE_STATE_FATAL; 735 return (ARCHIVE_FATAL); 736 } 737 if (r2 < ret) 738 ret = r2; 739 740 a->archive.state = ARCHIVE_STATE_DATA; 741 return (ret); 742} 743 744static int 745_archive_write_finish_entry(struct archive *_a) 746{ 747 struct archive_write *a = (struct archive_write *)_a; 748 int ret = ARCHIVE_OK; 749 750 archive_check_magic(&a->archive, ARCHIVE_WRITE_MAGIC, 751 ARCHIVE_STATE_HEADER | ARCHIVE_STATE_DATA, 752 "archive_write_finish_entry"); 753 if (a->archive.state & ARCHIVE_STATE_DATA 754 && a->format_finish_entry != NULL) 755 ret = (a->format_finish_entry)(a); 756 a->archive.state = ARCHIVE_STATE_HEADER; 757 return (ret); 758} 759 760/* 761 * Note that the compressor is responsible for blocking. 762 */ 763static ssize_t 764_archive_write_data(struct archive *_a, const void *buff, size_t s) 765{ 766 struct archive_write *a = (struct archive_write *)_a; 767 const size_t max_write = INT_MAX; 768 769 archive_check_magic(&a->archive, ARCHIVE_WRITE_MAGIC, 770 ARCHIVE_STATE_DATA, "archive_write_data"); 771 /* In particular, this catches attempts to pass negative values. */ 772 if (s > max_write) 773 s = max_write; 774 archive_clear_error(&a->archive); 775 return ((a->format_write_data)(a, buff, s)); 776} 777 778static struct archive_write_filter * 779filter_lookup(struct archive *_a, int n) 780{ 781 struct archive_write *a = (struct archive_write *)_a; 782 struct archive_write_filter *f = a->filter_first; 783 if (n == -1) 784 return a->filter_last; 785 if (n < 0) 786 return NULL; 787 while (n > 0 && f != NULL) { 788 f = f->next_filter; 789 --n; 790 } 791 return f; 792} 793 794static int 795_archive_filter_code(struct archive *_a, int n) 796{ 797 struct archive_write_filter *f = filter_lookup(_a, n); 798 return f == NULL ? -1 : f->code; 799} 800 801static const char * 802_archive_filter_name(struct archive *_a, int n) 803{ 804 struct archive_write_filter *f = filter_lookup(_a, n); 805 return f != NULL ? f->name : NULL; 806} 807 808static int64_t 809_archive_filter_bytes(struct archive *_a, int n) 810{ 811 struct archive_write_filter *f = filter_lookup(_a, n); 812 return f == NULL ? -1 : f->bytes_written; 813} 814