contrib/libarchive/libarchive/archive_read_set_format.c

   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_TAR:
  77       strcpy(str, "tar");
  78       break;
  79     case ARCHIVE_FORMAT_XAR:
  80       strcpy(str, "xar");
  81       break;
  82     case ARCHIVE_FORMAT_ZIP:
  83       strcpy(str, "zip");
  84       break;
  85     default:
  86       archive_set_error(&a->archive, ARCHIVE_ERRNO_PROGRAMMER,
  87           "Invalid format code specified");
  88       return (ARCHIVE_FATAL);
  89   }
  90
  91   slots = sizeof(a->formats) / sizeof(a->formats[0]);
  92   a->format = &(a->formats[0]);
  93   for (i = 0; i < slots; i++, a->format++) {
  94     if (!a->format->name || !strcmp(a->format->name, str))
  95       break;
  96   }
  97   if (!a->format->name || strcmp(a->format->name, str))
  98   {
  99     archive_set_error(&a->archive, ARCHIVE_ERRNO_PROGRAMMER,
 100         "Internal error: Unable to set format");
 101     r1 = (ARCHIVE_FATAL);
 102   }
 103
 104   return (r1 < r2) ? r1 : r2;
 105 }