archive_read_set_format.c 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116
  1. /*-
  2. * Copyright (c) 2003-2012 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. #include "archive_platform.h"
  26. #ifdef HAVE_ERRNO_H
  27. #include <errno.h>
  28. #endif
  29. #include "archive.h"
  30. #include "archive_private.h"
  31. #include "archive_read_private.h"
  32. int
  33. archive_read_set_format(struct archive *_a, int code)
  34. {
  35. int r1, r2, slots, i;
  36. char str[10];
  37. struct archive_read *a = (struct archive_read *)_a;
  38. if ((r1 = archive_read_support_format_by_code(_a, code)) < (ARCHIVE_OK))
  39. return r1;
  40. r1 = r2 = (ARCHIVE_OK);
  41. if (a->format)
  42. r2 = (ARCHIVE_WARN);
  43. switch (code & ARCHIVE_FORMAT_BASE_MASK)
  44. {
  45. case ARCHIVE_FORMAT_7ZIP:
  46. strcpy(str, "7zip");
  47. break;
  48. case ARCHIVE_FORMAT_AR:
  49. strcpy(str, "ar");
  50. break;
  51. case ARCHIVE_FORMAT_CAB:
  52. strcpy(str, "cab");
  53. break;
  54. case ARCHIVE_FORMAT_CPIO:
  55. strcpy(str, "cpio");
  56. break;
  57. case ARCHIVE_FORMAT_EMPTY:
  58. strcpy(str, "empty");
  59. break;
  60. case ARCHIVE_FORMAT_ISO9660:
  61. strcpy(str, "iso9660");
  62. break;
  63. case ARCHIVE_FORMAT_LHA:
  64. strcpy(str, "lha");
  65. break;
  66. case ARCHIVE_FORMAT_MTREE:
  67. strcpy(str, "mtree");
  68. break;
  69. case ARCHIVE_FORMAT_RAR:
  70. strcpy(str, "rar");
  71. break;
  72. case ARCHIVE_FORMAT_RAR_V5:
  73. strcpy(str, "rar5");
  74. break;
  75. case ARCHIVE_FORMAT_RAW:
  76. strcpy(str, "raw");
  77. break;
  78. case ARCHIVE_FORMAT_TAR:
  79. strcpy(str, "tar");
  80. break;
  81. case ARCHIVE_FORMAT_WARC:
  82. strcpy(str, "warc");
  83. break;
  84. case ARCHIVE_FORMAT_XAR:
  85. strcpy(str, "xar");
  86. break;
  87. case ARCHIVE_FORMAT_ZIP:
  88. strcpy(str, "zip");
  89. break;
  90. default:
  91. archive_set_error(&a->archive, ARCHIVE_ERRNO_PROGRAMMER,
  92. "Invalid format code specified");
  93. return (ARCHIVE_FATAL);
  94. }
  95. slots = sizeof(a->formats) / sizeof(a->formats[0]);
  96. a->format = &(a->formats[0]);
  97. for (i = 0; i < slots; i++, a->format++) {
  98. if (!a->format->name || !strcmp(a->format->name, str))
  99. break;
  100. }
  101. if (!a->format->name || strcmp(a->format->name, str))
  102. {
  103. archive_set_error(&a->archive, ARCHIVE_ERRNO_PROGRAMMER,
  104. "Internal error: Unable to set format");
  105. r1 = (ARCHIVE_FATAL);
  106. }
  107. return (r1 < r2) ? r1 : r2;
  108. }