diff options
Diffstat (limited to 'tools/fexc.c')
-rw-r--r-- | tools/fexc.c | 330 |
1 files changed, 330 insertions, 0 deletions
diff --git a/tools/fexc.c b/tools/fexc.c new file mode 100644 index 0000000..9f603dd --- /dev/null +++ b/tools/fexc.c @@ -0,0 +1,330 @@ +/* + * Copyright (C) 2012 Alejandro Mery <amery@geeks.cl> + * + * 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/>. + */ + +#include "fexc.h" + +#include <errno.h> +#include <libgen.h> +#include <stdlib.h> +#include <string.h> +#include <sys/mman.h> +#include <sys/stat.h> +#include <unistd.h> +#include <fcntl.h> + +#define pr_info(...) errf("fexc: " __VA_ARGS__) +#define pr_err(...) errf("E: fexc: " __VA_ARGS__) + +enum script_format { + FEX_SCRIPT_FORMAT, + BIN_SCRIPT_FORMAT, + UBOOT_HEADER_FORMAT, +}; + +/* + */ +static inline char *read_all(int fd, const char *filename, size_t *size) +{ + size_t buf_size = 4096, count = 0; + char *p, *buf = malloc(buf_size); + if (!buf) { + pr_err("%s: %s\n", "malloc", strerror(errno)); + return NULL; + } + p = buf; + while (1) { + ssize_t rc = read(fd, p, buf_size-count); + if (rc == 0) + break; + else if (rc > 0) { + count += rc; + p += rc; + + if (count == buf_size) { + char *new; + buf_size *= 2; + new = realloc(buf, buf_size); + if (!new) { + pr_err("%s: %s\n", "realloc", + strerror(errno)); + free(buf); + return NULL; + } else if (new != buf) { + buf = new; + p = buf + count; + } + } + } else if (errno != EAGAIN && errno != EINTR) { + pr_err("%s: %s: %s\n", filename, + "read", strerror(errno)); + free(buf); + return NULL; + } + } + + *size = count; + return buf; +} + +/* + */ +static inline int script_parse(enum script_format format, + const char *filename, + struct script *script) +{ + int ret = 0; + switch (format) { + case FEX_SCRIPT_FORMAT: { + FILE *in = stdin; + if (!filename) + filename = "<stdin>"; + else if ((in = fopen(filename, "r")) == NULL) { + pr_err("%s: %s\n", filename, strerror(errno)); + break; + } + ret = script_parse_fex(in, filename, script); + fclose(in); + }; break; + case BIN_SCRIPT_FORMAT: { + int in = 0; /* stdin */ + struct stat sb; + void *bin = NULL; + size_t bin_size; + int allocated = 1; + + if (!filename) + filename = "<stdin>"; + else if ((in = open(filename, O_RDONLY)) < 0) { + pr_err("%s: %s\n", filename, strerror(errno)); + break; + } + + if (fstat(in, &sb) == -1) { + pr_err("%s: %s: %s\n", filename, + "fstat", strerror(errno)); + goto bin_close; + } else if (S_ISREG(sb.st_mode)) { + /* regular file, mmap it */ + bin = mmap(0, sb.st_size, PROT_READ, MAP_SHARED, in, 0); + if (bin == MAP_FAILED) { + pr_err("%s: %s: %s\n", filename, + "mmap", strerror(errno)); + goto bin_close; + } + bin_size = sb.st_size; + allocated = 0; + } else { + /* something else... just read it all! */ + bin = read_all(in, filename, &bin_size); + if (bin == NULL) + goto bin_close; + allocated = 1; + } + + ret = script_decompile_bin(bin, bin_size, filename, script); + if (allocated) + free(bin); + else if (munmap(bin, bin_size) == -1) { + pr_err("%s: %s: %s\n", filename, + "munmap", strerror(errno)); + } +bin_close: + close(in); + }; break; + case UBOOT_HEADER_FORMAT: /* not valid input */ + ; + } + return ret; +} +static inline int script_generate(enum script_format format, + const char *filename, + struct script *script) +{ + int ret = 0; + static int (*text_gen[3]) (FILE *, const char *, struct script *) = { + [FEX_SCRIPT_FORMAT] = script_generate_fex, + [UBOOT_HEADER_FORMAT] = script_generate_uboot, + }; + + if (text_gen[format]) { + FILE *out = stdout; + + if (!filename) + filename = "<stdout>"; + else if ((out = fopen(filename, "w")) == NULL) { + pr_err("%s: %s\n", filename, strerror(errno)); + goto done; + } + + ret = text_gen[format](out, filename, script); + fclose(out); + } else { + int out = 1; /* stdout */ + size_t sections, entries, bin_size; + void *bin; + + if (!filename) + filename = "<stdout>"; + else if ((out = open(filename, O_WRONLY|O_CREAT|O_TRUNC, 0666)) < 0) { + pr_err("%s: %s\n", filename, strerror(errno)); + goto done; + } + + bin_size = script_bin_size(script, §ions, &entries); + bin = calloc(1, bin_size); + if (!bin) + pr_err("%s: %s\n", "malloc", strerror(errno)); + else if (script_generate_bin(bin, bin_size, script, sections, entries)) { + char *p = bin; + while(bin_size) { + ssize_t wc = write(out, p, bin_size); + + if (wc>0) { + p += wc; + bin_size -= wc; + } else if (wc < 0 && errno != EINTR) { + pr_err("%s: %s: %s\n", filename, + "write", strerror(errno)); + break; + } + } + ret = (bin_size == 0); + } + free(bin); + close(out); + } +done: + return ret; +} + +/* + */ +static inline void app_usage(const char *arg0, int mode) +{ + errf("Usage: %s [-vq]%s[<input> [<output>]]\n", arg0, + mode ? " " : " [-I <infmt>] [-O <outfmt>] "); + + if (mode == 0) + fputs("\ninfmt: fex, bin (default:fex)" + "\noutfmt: fex, bin, uboot (default:bin)\n", + stderr); +} + +static inline int app_choose_mode(char *arg0) +{ + const char *name = basename(arg0); + if (strcmp(name, "fex2bin") == 0) + return 1; + else if (strcmp(name, "bin2fex") == 0) + return 2; + else + return 0; +} + +/* + */ +int main(int argc, char *argv[]) +{ + static const char *formats[] = { "fex", "bin", "uboot", NULL }; + enum script_format infmt=FEX_SCRIPT_FORMAT; + enum script_format outfmt=BIN_SCRIPT_FORMAT; + const char *filename[] = { NULL /*stdin*/, NULL /*stdout*/}; + struct script *script; + + int app_mode = app_choose_mode(argv[0]); + + const char *opt_string = "I:O:vq?"; + if (app_mode != 0) opt_string += 4; /* disallow -I and -O */ + int opt, ret = 1; + int verbose = 0; + + if (app_mode == 2) { /* bin2fex */ + infmt = BIN_SCRIPT_FORMAT; + outfmt = FEX_SCRIPT_FORMAT; + } + + while ((opt = getopt(argc, argv, opt_string)) != -1) { + switch (opt) { + case 'I': + infmt=0; + for (const char **f = formats; *f; f++, infmt++) { + if (strcmp(*f, optarg) == 0) + break; + } + switch (infmt) { + case FEX_SCRIPT_FORMAT: + case BIN_SCRIPT_FORMAT: + break; + default: + errf("%s: invalid format -- \"%s\"\n", + argv[0], optarg); + goto show_usage; + } + break; + case 'O': + outfmt=0; + for (const char **f = formats; *f; f++, outfmt++) { + if (strcmp(*f, optarg) == 0) + break; + } + if (!formats[outfmt]) { + errf("%s: invalid format -- \"%s\"\n", + argv[0], optarg); + goto show_usage; + } + break; + case 'v': + verbose++; + break; + case 'q': + verbose--; + break; + default: +show_usage: + app_usage(argv[0], app_mode); + goto done; + } + } + + switch (argc - optind) { + case 2: + filename[1] = argv[optind+1]; /* out */ + case 1: + if (strcmp(argv[optind], "-") != 0) + filename[0] = argv[optind]; /* in */ + case 0: + break; + default: + goto show_usage; + } + + if (verbose>0) + errf("%s: from %s:%s to %s:%s\n", argv[0], + formats[infmt], filename[0]?filename[0]:"<stdin>", + formats[outfmt], filename[1]?filename[1]:"<stdout>"); + + if ((script = script_new()) == NULL) { + perror("malloc"); + goto done; + } else if (script_parse(infmt, filename[0], script) && + script_generate(outfmt, filename[1], script)) { + ret = 0; + } + script_delete(script); +done: + return ret; +} |