summaryrefslogtreecommitdiff
path: root/util/romtool/add.c
diff options
context:
space:
mode:
Diffstat (limited to 'util/romtool/add.c')
-rw-r--r--util/romtool/add.c327
1 files changed, 0 insertions, 327 deletions
diff --git a/util/romtool/add.c b/util/romtool/add.c
deleted file mode 100644
index 3ffc74b966..0000000000
--- a/util/romtool/add.c
+++ /dev/null
@@ -1,327 +0,0 @@
-/*
- * romtool
- *
- * Copyright (C) 2008 Jordan Crouse <jordan@cosmicpenguin.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; version 2 of the License.
- *
- * 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, write to the Free Software
- * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA, 02110-1301 USA
- */
-
-#include <ctype.h>
-#include <string.h>
-#include <stdlib.h>
-#include <unistd.h>
-#include <fcntl.h>
-#include <signal.h>
-#include <sys/stat.h>
-#include <sys/mman.h>
-#include <sys/wait.h>
-#include "romtool.h"
-
-#define MAX_PATH 255
-
-static int add_from_fd(struct rom *rom, const char *name, int type, int fd)
-{
- unsigned char *buffer = malloc(16 * 1024);
- unsigned char *ptr = buffer;
-
- int size = 0;
- int aloc = 16 * 1024;
- int remain = 16 * 1024;
- int ret;
-
- if (buffer == NULL)
- return -1;
-
- while (1) {
- ret = read(fd, ptr, remain);
-
- if (ret <= 0)
- break;
-
- ptr += ret;
- remain -= ret;
- size += ret;
-
- if (remain == 0) {
- buffer = realloc(buffer, aloc + 16 * 1024);
-
- if (buffer == NULL) {
- ret = -1;
- break;
- }
-
- ptr = buffer + size;
-
- aloc += (16 * 1024);
- remain = 16 * 1024;
- }
- }
-
- if (ret == -1 || size == 0) {
-
- if (buffer != NULL)
- free(buffer);
-
- return -1;
- }
-
- ret = rom_add(rom, name, buffer, size, type);
- free(buffer);
-
- return ret;
-}
-
-int fork_tool_and_add(struct rom *rom, const char *tool, const char *input,
- const char *name, int type, int argc, char **argv)
-{
- int output[2];
- pid_t pid;
- int ret;
- int status;
- char **toolargs;
- int i;
-
- /* Create the pipe */
-
- if (pipe(output)) {
- ERROR("Couldn't create a pipe: %m\n");
- return -1;
- }
-
- toolargs = (char **)malloc((5 + argc) * sizeof(char *));
-
- if (toolargs == NULL) {
- ERROR("Unable to allocate memory: %m\n");
- return -1;
- }
-
- toolargs[0] = (char *)tool;
-
- /* these are args. So they need a - in front */
- for (i = 0; i < argc; i++) {
- /* I wish I had python */
- char *c = malloc(strlen(argv[i])) + 2;
- c[0] = '-';
- strcpy(&c[1], argv[i]);
- c[strlen(argv[i])+1] = 0;
- toolargs[1 + i] = c;
- }
-
- toolargs[1 + argc] = "-o";
- toolargs[2 + argc] = "-";
- toolargs[3 + argc] = (char *)input;
- toolargs[4 + argc] = NULL;
-
- pid = fork();
-
- if (pid == 0) {
-
- /* Set up stdin/stdout for the child */
-
- dup2(output[1], STDOUT_FILENO);
- close(output[0]);
-
- /* Execute the tool */
- if (execv(tool, toolargs)) {
- ERROR("Unable to execute %s: %m\n", tool);
- exit(-1);
- }
-
- exit(0);
- }
-
- free(toolargs);
-
- close(output[1]);
-
- /* Read from the file */
- ret = add_from_fd(rom, name, type, output[0]);
-
- /* Reap the child */
- waitpid(pid, &status, 0);
-
- if (WIFSIGNALED(status)) {
- kill(pid, WTERMSIG(status));
- ERROR("Error while executing %s\n", tool);
- return -1;
- } else if (WEXITSTATUS(status) != 0) {
- ERROR("Error while executing %s: %d\n", tool,
- (int)WEXITSTATUS(status));
- return -1;
- }
-
- return ret;
-}
-
-static int add_blob(struct rom *rom, const char *filename,
- const char *name, int type)
-{
- void *ptr;
- struct stat s;
- int fd, ret;
-
- if (!strcmp(filename, "-"))
- return add_from_fd(rom, name, type, 0);
-
- fd = open(filename, O_RDONLY);
-
- if (fd == -1) {
- ERROR("Could not open %s: %m\n", filename);
- return -1;
- }
-
- if (fstat(fd, &s)) {
- ERROR("Could not stat %s: %m\n", filename);
- close(fd);
- return -1;
- }
-
- ptr = mmap(0, s.st_size, PROT_READ, MAP_SHARED, fd, 0);
-
- if (ptr == MAP_FAILED) {
- ERROR("Unable to map %s: %m\n", filename);
- close(fd);
- return -1;
- }
-
- ret = rom_add(rom, name, ptr, s.st_size, type);
-
- munmap(ptr, s.st_size);
- close(fd);
-
- return ret;
-}
-
-void add_usage(void)
-{
- printf("add [FILE] [NAME] [TYPE]\tAdd a component\n");
-}
-
-void add_stage_usage(void)
-{
- printf("add-stage [FILE] [NAME] [OPTIONS]\tAdd a stage to the ROM\n");
-}
-
-void add_payload_usage(void)
-{
- printf
- ("add-payload [FILE] [NAME] [OPTIONS]\tAdd a payload to the ROM\n");
-}
-
-int add_handler(struct rom *rom, int argc, char **argv)
-{
- unsigned int type = ROMFS_COMPONENT_NULL;
-
- if (argc < 2) {
- add_usage();
- return -1;
- }
-
- if (!rom_exists(rom)) {
- ERROR("You need to create the ROM before adding files to it\n");
- return -1;
- }
-
- /* There are two ways to specify the type - a string or a number */
-
- if (argc == 3) {
- if (isdigit(*(argv[2])))
- type = strtoul(argv[2], 0, 0);
- }
-
- if (type == ROMFS_COMPONENT_NULL)
- WARN("No file type was given for %s - using default\n",
- argv[0]);
-
- return add_blob(rom, argv[0], argv[1], type);
-}
-
-char *find_tool(char *tool)
-{
- static char toolpath[MAX_PATH];
- extern char romtool_bindir[];
-
- snprintf(toolpath, MAX_PATH - 1, "tools/%s", tool);
- if (!access(toolpath, X_OK))
- return toolpath;
-
- snprintf(toolpath, MAX_PATH - 1, "%s/tools/%s", romtool_bindir, tool);
-
- if (!access(toolpath, X_OK))
- return toolpath;
-
- snprintf(toolpath, MAX_PATH - 1, "%s/%s", romtool_bindir, tool);
-
- if (!access(toolpath, X_OK))
- return toolpath;
-
- strncpy(toolpath, tool, MAX_PATH - 1);
- return toolpath;
-}
-
-/* Invoke the rom-mkpayload utility */
-
-int add_payload_handler(struct rom *rom, int argc, char **argv)
-{
- if (argc < 2) {
- add_payload_usage();
- return -1;
- }
-
- /* Make sure the ROM exists */
-
- if (!rom_exists(rom)) {
- ERROR("You need to create the ROM before adding files to it\n");
- return -1;
- }
-
- /* Check that the incoming file exists */
-
- if (access(argv[0], R_OK)) {
- ERROR("File %s does not exist\n", argv[0]);
- return -1;
- }
-
- return fork_tool_and_add(rom, find_tool("rom-mkpayload"), argv[0],
- argv[1], ROMFS_COMPONENT_PAYLOAD, argc - 2,
- argc > 2 ? &argv[2] : NULL);
-}
-
-/* Invoke the rom-mkstage utility */
-
-int add_stage_handler(struct rom *rom, int argc, char **argv)
-{
- if (argc < 2) {
- add_stage_usage();
- return -1;
- }
-
- /* Make sure the ROM exists */
-
- if (!rom_exists(rom)) {
- ERROR("You need to create the ROM before adding files to it\n");
- return -1;
- }
-
- /* Check that the incoming file exists */
-
- if (access(argv[0], R_OK)) {
- ERROR("File %s does not exist\n", argv[0]);
- return -1;
- }
-
- return fork_tool_and_add(rom, find_tool("rom-mkstage"), argv[0],
- argv[1], ROMFS_COMPONENT_STAGE, argc - 2,
- argc > 2 ? &argv[2] : NULL);
-}