util/cbfstool/extract.c

   1 /*
   2  * cbfstool
   3  *
   4  * Copyright (C) 2009 Myles Watson <mylesgw@gmail.com>
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; version 2 of the License.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA, 02110-1301 USA
  18  */
  19
  20 #include <ctype.h>
  21 #include <stdlib.h>
  22 #include <sys/stat.h>
  23 #include <unistd.h>
  24 #include <fcntl.h>
  25 #include "cbfstool.h"
  26
  27 static int extract_blob(struct rom *rom, const char *filename, const char *name)
  28 {
  29         void *buf;
  30         int fd, ret;
  31         unsigned long size;
  32
  33         ret = rom_extract(rom, name, &buf, &size);
  34
  35         if (ret == -1)
  36                 return -1;
  37
  38         fd = open(filename, O_RDWR | O_CREAT | O_TRUNC, S_IRUSR | S_IWUSR);
  39
  40         if (fd == -1) {
  41                 ERROR("Could not open %s: %m\n", filename);
  42                 return -1;
  43         }
  44
  45         if (write(fd, buf, size) != size) {
  46                 ERROR("Couldn't write %ld bytes!\n", size);
  47                 ret = -1;
  48         }
  49
  50         close(fd);
  51
  52         return ret;
  53 }
  54
  55 void extract_usage(void)
  56 {
  57         printf("extract [FILE] [NAME] \textract a component\n");
  58 }
  59
  60 int extract_handler(struct rom *rom, int argc, char **argv)
  61 {
  62         if (argc < 2) {
  63                 extract_usage();
  64                 return -1;
  65         }
  66
  67         if (!rom_exists(rom)) {
  68                 ERROR("Can't find the ROM!\n");
  69                 return -1;
  70         }
  71
  72         return extract_blob(rom, argv[0], argv[1]);
  73 }
  74