archive_read_set_format.c 3.2 KB

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