/*

* Copyright (C) 2012 Henrik Nordstrom <henrik@henriknordstrom.net>

*

* This program is free software: you can redistribute it and/or modify

* it under the terms of the GNU General Public License as published by

* the Free Software Foundation, either version 2 of the License, or

* (at your option) any later version.

*

* This program is distributed in the hope that it will be useful,

* but WITHOUT ANY WARRANTY; without even the implied warranty of

* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

* GNU General Public License for more details.

*

* You should have received a copy of the GNU General Public License

* along with this program. If not, see <http://www.gnu.org/licenses/>.

*/

/* Needs _BSD_SOURCE for htole and letoh */

#define _BSD_SOURCE

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <endian.h>

#include <unistd.h>

struct phoenix_ptable {

} ptable;

static int save_part(struct phoenix_ptable *ptable, int part, const char *dest, FILE *in)

{

int l = strlen(dest) + 16;

char outname[l];

FILE *out;

char *buf = NULL;

int ret = 0;

snprintf(outname, l, dest, part);

if (part > ptable->parts) {

fprintf(stderr, "ERROR: Part index out of range\n");

return -1;

}

buf = malloc(ptable->part[part].size);

if (!buf)

goto err;

if (strcmp(outname, "-") != 0)

out = fopen(outname, "wb");

else

out = stdout;

if (!out)

goto err;

if (fseek(in, ptable->part[part].start * 0x200, SEEK_SET) == -1)

goto err;

if (fread(buf, ptable->part[part].size, 1, in) != 1)

goto err;

if (fwrite(buf, ptable->part[part].size, 1, out) != 1)

goto err;

ret = 0;

_exit:

if (buf)

free(buf);

if (out != stdout)

fclose(out);

return ret;

err:

perror(NULL);

ret = -1;

goto _exit;

}

static void usage(char **argv)

{

printf("Usage: %s [options] [phoenix_image]\n"

" -v verbose\n"

" -q quiet\n"

" -p N part number\n"

" -o X destination directory, file or pattern (%%d for part number)\n"

" -s save all parts\n"

, argv[0]

);

}

int main(int argc, char **argv)

{

int i;

FILE *in = stdin;

int verbose = 1;

int save_parts = 0;

int part = -1;

int opt;

const char *dest = "%d.img";

while ((opt = getopt(argc, argv, "vqso:p:?")) != -1) {

switch(opt) {

case 'v':

verbose++;

break;

case 'q':

if (verbose)

verbose--;

break;

case 'o':

dest = optarg;

save_parts = 1;

break;

case 'p':

save_parts = 1;

part = atoi(optarg);

break;

case 's':

save_parts = 1;

break;

default:

usage(argv);

exit(1);

break;

}

}

if (save_parts && !strchr(dest, '%')) {

const char *t = dest;

if (!*t)

t = "./";

if (t[strlen(t)-1] == '/' || !part) {

int l = strlen(t) + strlen("/%d.img") + 1;

char *tmp = malloc(l);

snprintf(tmp, l, "%s/%%d.img", optarg);

t = tmp;

}

dest = t;

}

if (argc > optind + 1) {

usage(argv);

exit(1);

}

if (optind < argc ) {

in = fopen(argv[optind], "rb");

}

fseek(in, 0x1C00, SEEK_CUR);

fread(&ptable, 1, 0x400, in);

if (strncmp(ptable.signature, "PHOENIX_CARD_IMG", 16) != 0) {

fprintf(stderr, "ERROR: Not a phoenix image\n");

exit(1);

}

if (verbose > 1) {

printf("???? : %08x\n", le32toh(ptable.unknown1));

printf("Parts : %d\n", le16toh(ptable.parts));

printf("???? : %08x\n", le16toh(ptable.unknown2));

printf("pad : %02x%02x%02x%02x%02x%02x%02x%02x\n", ptable.pad[0], ptable.pad[1], ptable.pad[2], ptable.pad[3], ptable.pad[4], ptable.pad[5], ptable.pad[6], ptable.pad[7]);

printf("\n");

}

for (i = 0; i < le16toh(ptable.parts); i++) {

if (verbose && (part == -1 || part == i)) {

printf("part %d:\n", i);

printf("\tstart: 0x%08x (%u / 0x%08x)\n", le32toh(ptable.part[i].start)*512, le32toh(ptable.part[i].start), le32toh(ptable.part[i].start));

printf("\tsize : %u\n", le32toh(ptable.part[i].size));

printf("\t?????: %08x\n", le32toh(ptable.part[i].unknown));

if (verbose > 1 || le32toh(ptable.part[i].sig) != 0x00646461)

printf("\tsig??: %08x\n", le32toh(ptable.part[i].sig));

printf("\n");

}

if (save_parts && (part == -1 || part == i)) {

save_part(&ptable, i, dest, in);

}

}

}