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