summaryrefslogtreecommitdiffstats
path: root/bootloader.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'bootloader.cpp')
-rw-r--r--bootloader.cpp130
1 files changed, 127 insertions, 3 deletions
diff --git a/bootloader.cpp b/bootloader.cpp
index d80c5e793..13bc39167 100644
--- a/bootloader.cpp
+++ b/bootloader.cpp
@@ -14,6 +14,10 @@
* limitations under the License.
*/
+extern "C" {
+#include "mtdutils/mtdutils.h"
+}
+
#include <errno.h>
#include <fcntl.h>
#include <inttypes.h>
@@ -22,26 +26,51 @@
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
-
-#include <fs_mgr.h>
+#include <stdlib.h>
#include "bootloader.h"
#include "common.h"
#include "mtdutils/mtdutils.h"
-#include "roots.h"
+//#include "roots.h"
#include "unique_fd.h"
+// fake Volume struct that allows us to use the AOSP code easily
+struct Volume
+{
+ char fs_type[8];
+ char blk_device[256];
+};
+
+static Volume misc;
+
+void set_misc_device(const char* type, const char* name) {
+ strlcpy(misc.fs_type, type, sizeof(misc.fs_type));
+ if (strlen(name) >= sizeof(misc.blk_device)) {
+ LOGE("New device name of '%s' is too large for bootloader.cpp\n", name);
+ } else {
+ strcpy(misc.blk_device, name);
+ }
+}
+
static int get_bootloader_message_mtd(bootloader_message* out, const Volume* v);
static int set_bootloader_message_mtd(const bootloader_message* in, const Volume* v);
static int get_bootloader_message_block(bootloader_message* out, const Volume* v);
static int set_bootloader_message_block(const bootloader_message* in, const Volume* v);
int get_bootloader_message(bootloader_message* out) {
+#if 0
Volume* v = volume_for_path("/misc");
if (v == nullptr) {
LOGE("Cannot load volume /misc!\n");
return -1;
}
+#else
+ Volume* v = &misc;
+ if (v->fs_type[0] == 0) {
+ LOGI("Not using /misc, not defined in fstab.\n");
+ return -1;
+ }
+#endif
if (strcmp(v->fs_type, "mtd") == 0) {
return get_bootloader_message_mtd(out, v);
} else if (strcmp(v->fs_type, "emmc") == 0) {
@@ -52,11 +81,19 @@ int get_bootloader_message(bootloader_message* out) {
}
int set_bootloader_message(const bootloader_message* in) {
+#if 0
Volume* v = volume_for_path("/misc");
if (v == nullptr) {
LOGE("Cannot load volume /misc!\n");
return -1;
}
+#else
+ Volume* v = &misc;
+ if (v->fs_type[0] == 0) {
+ LOGI("Not using /misc, not defined in fstab.\n");
+ return -1;
+ }
+#endif
if (strcmp(v->fs_type, "mtd") == 0) {
return set_bootloader_message_mtd(in, v);
} else if (strcmp(v->fs_type, "emmc") == 0) {
@@ -174,7 +211,11 @@ static int get_bootloader_message_block(bootloader_message* out,
LOGE("failed to open \"%s\": %s\n", v->blk_device, strerror(errno));
return -1;
}
+#ifdef BOARD_RECOVERY_BLDRMSG_OFFSET
+ fseek(f, BOARD_RECOVERY_BLDRMSG_OFFSET, SEEK_SET);
+#endif
bootloader_message temp;
+
int count = fread(&temp, sizeof(temp), 1, f);
if (count != 1) {
LOGE("failed to read \"%s\": %s\n", v->blk_device, strerror(errno));
@@ -197,6 +238,10 @@ static int set_bootloader_message_block(const bootloader_message* in,
return -1;
}
+#ifdef BOARD_RECOVERY_BLDRMSG_OFFSET
+ lseek(fd.get(), BOARD_RECOVERY_BLDRMSG_OFFSET, SEEK_SET);
+#endif
+
size_t written = 0;
const uint8_t* start = reinterpret_cast<const uint8_t*>(in);
size_t total = sizeof(*in);
@@ -216,3 +261,82 @@ static int set_bootloader_message_block(const bootloader_message* in,
}
return 0;
}
+
+
+static const char *COMMAND_FILE = "/cache/recovery/command";
+static const int MAX_ARG_LENGTH = 4096;
+static const int MAX_ARGS = 100;
+
+// command line args come from, in decreasing precedence:
+// - the actual command line
+// - the bootloader control block (one per line, after "recovery")
+// - the contents of COMMAND_FILE (one per line)
+void
+get_args(int *argc, char ***argv) {
+ struct bootloader_message boot;
+ memset(&boot, 0, sizeof(boot));
+ get_bootloader_message(&boot); // this may fail, leaving a zeroed structure
+
+ if (boot.command[0] != 0 && boot.command[0] != 255) {
+ LOGI("Boot command: %.*s\n", (int)sizeof(boot.command), boot.command);
+ }
+
+ if (boot.status[0] != 0 && boot.status[0] != 255) {
+ LOGI("Boot status: %.*s\n", (int)sizeof(boot.status), boot.status);
+ }
+
+ // --- if arguments weren't supplied, look in the bootloader control block
+ if (*argc <= 1) {
+ boot.recovery[sizeof(boot.recovery) - 1] = '\0'; // Ensure termination
+ const char *arg = strtok(boot.recovery, "\n");
+ if (arg != NULL && !strcmp(arg, "recovery")) {
+ *argv = (char **) malloc(sizeof(char *) * MAX_ARGS);
+ (*argv)[0] = strdup(arg);
+ for (*argc = 1; *argc < MAX_ARGS; ++*argc) {
+ if ((arg = strtok(NULL, "\n")) == NULL) break;
+ (*argv)[*argc] = strdup(arg);
+ }
+ LOGI("Got arguments from boot message\n");
+ } else if (boot.recovery[0] != 0 && boot.recovery[0] != 255) {
+ LOGE("Bad boot message\n\"%.20s\"\n", boot.recovery);
+ }
+ }
+
+ // --- if that doesn't work, try the command file
+ if (*argc <= 1) {
+ FILE *fp = fopen(COMMAND_FILE, "r");
+ if (fp != NULL) {
+ char *token;
+ char *argv0 = (*argv)[0];
+ *argv = (char **) malloc(sizeof(char *) * MAX_ARGS);
+ (*argv)[0] = argv0; // use the same program name
+
+ char buf[MAX_ARG_LENGTH];
+ for (*argc = 1; *argc < MAX_ARGS; ++*argc) {
+ if (!fgets(buf, sizeof(buf), fp)) break;
+ token = strtok(buf, "\r\n");
+ if (token != NULL) {
+ (*argv)[*argc] = strdup(token); // Strip newline.
+ } else {
+ --*argc;
+ }
+ }
+
+ fflush(fp);
+ if (ferror(fp)) LOGE("Error in %s\n(%s)\n", COMMAND_FILE, strerror(errno));
+ fclose(fp);
+ LOGI("Got arguments from %s\n", COMMAND_FILE);
+ }
+ }
+
+ // --> write the arguments we have back into the bootloader control block
+ // always boot into recovery after this (until finish_recovery() is called)
+ strlcpy(boot.command, "boot-recovery", sizeof(boot.command));
+ strlcpy(boot.recovery, "recovery\n", sizeof(boot.recovery));
+ int i;
+ for (i = 1; i < *argc; ++i) {
+ strlcat(boot.recovery, (*argv)[i], sizeof(boot.recovery));
+ strlcat(boot.recovery, "\n", sizeof(boot.recovery));
+ }
+ set_bootloader_message(&boot);
+}