1 files changed, 257 insertions, 0 deletions
diff --git a/kexec/arch/x86_64/kexec-elf-x86_64.c b/kexec/arch/x86_64/kexec-elf-x86_64.c
new file mode 100644
index 0000000..7f9540a
--- /dev/null
+++ b/kexec/arch/x86_64/kexec-elf-x86_64.c
@@ -0,0 +1,257 @@
+/*
+ * kexec: Linux boots Linux
+ *
+ * Copyright (C) 2003-2005  Eric Biederman (ebiederm@xmission.com)
+ *
+ * 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., 675 Mass Ave, Cambridge, MA 02139, USA.
+ */
+
+#define _GNU_SOURCE
+#include <stddef.h>
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+#include <errno.h>
+#include <limits.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <unistd.h>
+#include <getopt.h>
+#include <elf.h>
+#include <x86/x86-linux.h>
+#include "../../kexec.h"
+#include "../../kexec-syscall.h"
+#include "../../kexec-elf.h"
+#include "../../kexec-elf-boot.h"
+#include "../i386/x86-linux-setup.h"
+#include "kexec-x86_64.h"
+#include "../i386/kexec-x86.h"
+#include "../i386/crashdump-x86.h"
+#include <arch/options.h>
+
+int elf_x86_64_probe(const char *buf, off_t len)
+{
+	return elf_x86_any_probe(buf, len, CORE_TYPE_ELF64);
+}
+
+void elf_x86_64_usage(void)
+{
+	printf(	"    --command-line=STRING Set the kernel command line to STRING\n"
+		"    --append=STRING       Set the kernel command line to STRING\n"
+		"    --reuse-cmdline       Use kernel command line from running system.\n"
+		"    --initrd=FILE         Use FILE as the kernel's initial ramdisk.\n"
+		"    --ramdisk=FILE        Use FILE as the kernel's initial ramdisk.\n"
+		"    --args-linux          Pass linux kernel style options\n"
+		"    --args-elf            Pass elf boot notes\n"
+		"    --args-none           Jump directly from the kernel\n"
+		);
+}
+
+int elf_x86_64_load(int argc, char **argv, const char *buf, off_t len, 
+	struct kexec_info *info)
+{
+	struct mem_ehdr ehdr;
+	const char *append = NULL;
+	char *command_line = NULL, *modified_cmdline;
+	char *tmp_cmdline = NULL;
+	int command_line_len;
+	const char *ramdisk;
+	unsigned long entry, max_addr;
+	int arg_style;
+#define ARG_STYLE_ELF   0
+#define ARG_STYLE_LINUX 1
+#define ARG_STYLE_NONE  2
+	int opt;
+	int result = 0;
+	const char *error_msg = NULL;
+
+	/* See options.h and add any new options there too! */
+	static const struct option options[] = {
+		KEXEC_ARCH_OPTIONS
+		{ "command-line",	1, NULL, OPT_APPEND },
+		{ "append",		1, NULL, OPT_APPEND },
+		{ "reuse-cmdline",	0, NULL, OPT_REUSE_CMDLINE },
+		{ "initrd",		1, NULL, OPT_RAMDISK },
+		{ "ramdisk",		1, NULL, OPT_RAMDISK },
+		{ "args-elf",		0, NULL, OPT_ARGS_ELF },
+		{ "args-linux",		0, NULL, OPT_ARGS_LINUX },
+		{ "args-none",		0, NULL, OPT_ARGS_NONE },
+		{ 0, 			0, NULL, 0 },
+	};
+
+	static const char short_options[] = KEXEC_ARCH_OPT_STR "";
+
+	/*
+	 * Parse the command line arguments
+	 */
+	arg_style = ARG_STYLE_ELF;
+	modified_cmdline = 0;
+	ramdisk = 0;
+	while((opt = getopt_long(argc, argv, short_options, options, 0)) != -1) {
+		switch(opt) {
+		default:
+			/* Ignore core options */
+			if (opt < OPT_ARCH_MAX) {
+				break;
+			}
+			fprintf(stderr, "Unknown option: opt: %d\n", opt);
+		case OPT_APPEND:
+			append = optarg;
+			break;
+		case OPT_REUSE_CMDLINE:
+			tmp_cmdline = get_command_line();
+			break;
+		case OPT_RAMDISK:
+			ramdisk = optarg;
+			break;
+		case OPT_ARGS_ELF: 
+			arg_style = ARG_STYLE_ELF;
+			break;
+		case OPT_ARGS_LINUX:
+			arg_style = ARG_STYLE_LINUX;
+			break;
+		case OPT_ARGS_NONE:
+#ifdef __x86_64__
+			arg_style = ARG_STYLE_NONE;
+#else
+			die("--args-none only works on arch x86_64\n");
+#endif
+			break;
+		}
+	}
+	command_line = concat_cmdline(tmp_cmdline, append);
+	if (tmp_cmdline)
+		free(tmp_cmdline);
+	command_line_len = 0;
+	if (command_line) {
+		command_line_len = strlen(command_line) +1;
+	}
+
+	/* Need to append some command line parameters internally in case of
+	 * taking crash dumps.
+	 */
+	if (info->kexec_flags & KEXEC_ON_CRASH) {
+		modified_cmdline = xmalloc(COMMAND_LINE_SIZE);
+		memset((void *)modified_cmdline, 0, COMMAND_LINE_SIZE);
+		if (command_line) {
+			strncpy(modified_cmdline, command_line,
+						COMMAND_LINE_SIZE);
+			modified_cmdline[COMMAND_LINE_SIZE - 1] = '\0';
+		}
+	}
+
+	/* Load the ELF executable */
+	elf_exec_build_load(info, &ehdr, buf, len, 0);
+
+	entry = ehdr.e_entry;
+	max_addr = elf_max_addr(&ehdr);
+
+	/* Do we want arguments? */
+	if (arg_style != ARG_STYLE_NONE) {
+		/* Load the setup code */
+		elf_rel_build_load(info, &info->rhdr, purgatory, purgatory_size,
+			0, ULONG_MAX, 1, 0);
+	}
+	if (arg_style == ARG_STYLE_NONE) {
+		info->entry = (void *)entry;
+
+	}
+	else if (arg_style == ARG_STYLE_ELF) {
+		unsigned long note_base;
+		struct entry64_regs regs;
+
+		/* Setup the ELF boot notes */
+		note_base = elf_boot_notes(info, max_addr, command_line, command_line_len);
+
+		/* Initialize the registers */
+		elf_rel_get_symbol(&info->rhdr, "entry64_regs", &regs, sizeof(regs));
+		regs.rdi = note_base;   /* The notes (arg1) */
+		regs.rip = entry;       /* The entry point */
+		regs.rsp = elf_rel_get_addr(&info->rhdr, "stack_end"); /* Stack, unused */
+		elf_rel_set_symbol(&info->rhdr, "entry64_regs", &regs, sizeof(regs));
+
+		if (ramdisk) {
+			error_msg = "Ramdisks not supported with generic elf arguments";
+			goto out;
+		}
+	}
+	else if (arg_style == ARG_STYLE_LINUX) {
+		struct x86_linux_faked_param_header *hdr;
+		unsigned long param_base;
+		char *ramdisk_buf;
+		off_t ramdisk_length;
+		struct entry64_regs regs;
+		int rc=0;
+
+		/* Get the linux parameter header */
+		hdr = xmalloc(sizeof(*hdr));
+		param_base = add_buffer(info, hdr, sizeof(*hdr), sizeof(*hdr),
+			16, 0, max_addr, 1);
+
+		/* Initialize the parameter header */
+		memset(hdr, 0, sizeof(*hdr));
+		init_linux_parameters(&hdr->hdr);
+
+		/* Add a ramdisk to the current image */
+		ramdisk_buf = 0;
+		ramdisk_length = 0;
+		if (ramdisk) {
+			ramdisk_buf = slurp_file(ramdisk, &ramdisk_length);
+		}
+
+		/* If panic kernel is being loaded, additional segments need
+		 * to be created. */
+		if (info->kexec_flags & KEXEC_ON_CRASH) {
+			rc = load_crashdump_segments(info, modified_cmdline,
+							max_addr, 0);
+			if (rc < 0) {
+				result = -1;
+				goto out;
+			}
+			/* Use new command line. */
+			free(command_line);
+			command_line = modified_cmdline;
+			command_line_len = strlen(modified_cmdline) + 1;
+			modified_cmdline = NULL;
+		}
+
+		/* Tell the kernel what is going on */
+		setup_linux_bootloader_parameters(info, &hdr->hdr, param_base, 
+			offsetof(struct x86_linux_faked_param_header, command_line),
+			command_line, command_line_len,
+			ramdisk_buf, ramdisk_length);
+
+		/* Fill in the information bios calls would usually provide */
+		setup_linux_system_parameters(info, &hdr->hdr);
+
+		/* Initialize the registers */
+		elf_rel_get_symbol(&info->rhdr, "entry64_regs", &regs, sizeof(regs));
+		regs.rbx = 0;		/* Bootstrap processor */
+		regs.rsi = param_base;	/* Pointer to the parameters */
+		regs.rip = entry;	/* the entry point */
+		regs.rsp = elf_rel_get_addr(&info->rhdr, "stack_end"); /* Stack, unused */
+		elf_rel_set_symbol(&info->rhdr, "entry64_regs", &regs, sizeof(regs));
+	}
+	else {
+		error_msg = "Unknown argument style\n";
+	}
+
+out:
+	free(command_line);
+	free(modified_cmdline);
+	if (error_msg)
+		die("%s", error_msg);
+	return result;
+}