src: add built-in ogrelive grub2 boot mode templateHEADv1.2.5-26master

Add built-in ogrelive grub2 boot mode template and use it to generate the grub2
boot file for each computer for the new live system.

Use grub2 http to download vmlinuz, initrd.img.

Use live support to download filesystem.squashfs through http.

check for vmlinuz in cache, as usual, otherwise fallback to network boot.

1 files changed, 119 insertions, 0 deletions

diff --git a/src/boot.c b/src/boot.c
new file mode 100644
index 0000000..c58f332
--- /dev/null
+++ b/src/boot.c
@@ -0,0 +1,119 @@
+/*
+ * Copyright (C) 2020-2024 Soleta Networks <info@soleta.eu>
+ *
+ * This program is free software: you can redistribute it and/or modify it under
+ * the terms of the GNU Affero General Public License as published by the
+ * Free Software Foundation; either version 3 of the License, or
+ * (at your option) any later version.
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <time.h>
+#include <syslog.h>
+#include <sys/stat.h>
+#include <errno.h>
+#include "boot.h"
+#include "utils.h"
+
+#define OGRELIVE_TEMPLATE					\
+"# Autogenerated by ogserver on %s - do not edit this file!\n"	\
+"set ogserver=%s\n"						\
+"set ogrepo=%s\n"						\
+"set ogreliveversion=%s\n"					\
+"set ogrelivedir=ogrelive/$ogreliveversion\n"			\
+"set username=%s\n"						\
+"set passwd=%s\n"						\
+"\n"								\
+"set timeout=0\n"						\
+"set timeout_style=hidden\n"					\
+"\n"								\
+"echo \"checking for vmlinuz in cache...\"\n"			\
+"set root=''\n"							\
+"search --file --set root /boot/$ogreliveversion/vmlinuz\n"	\
+"if [ \"$root\" == \"\" ]; then\n"				\
+"    echo \"no vmlinuz found in cache, booting from network\"\n" \
+"    set default=1;\n"						\
+"else\n"							\
+"    echo \"vmlinuz found, booting from cache\"\n"		\
+"    set default=0;\n"						\
+"fi\n"								\
+"\n"								\
+"menuentry \"ogReLive HTTP boot\" {\n"				\
+"    insmod http\n"						\
+"    insmod net\n"						\
+"    net_bootp\n"						\
+"    linux (http,$ogrepo)/$ogrelivedir/vmlinuz ro boot=live quiet splash oglivedir=$ogrelivedir ip=$ogserver ogrepo=$ogrepo username=$username passwd=$passwd fetch=http://$ogrepo/$ogrelivedir/filesystem.squashfs\n" \
+"    initrd (http,$ogrepo)/$ogrelivedir/initrd.img\n"		\
+"    boot\n"							\
+"}\n"
+
+/* Same as OG_TFTP_BOOT_UEFI. */
+#define OG_BOOT_GRUB_DIR	"/opt/opengnsys/tftpboot/grub"
+
+static void ogrelive_grub2_file_path(char *grub2_filename,
+				     size_t grub2_filename_size,
+				     const char *mac)
+{
+        snprintf(grub2_filename, grub2_filename_size,
+		 OG_BOOT_GRUB_DIR"/01-%c%c:%c%c:%c%c:%c%c:%c%c:%c%c",
+		 mac[0], mac[1], mac[2], mac[3], mac[4], mac[5],
+		 mac[6], mac[7], mac[8], mac[9], mac[10], mac[11]);
+        str_tolower(grub2_filename);
+}
+
+int ogrelive_generate_grub2_file(const struct og_boot_cfg *cfg, const char *mac)
+{
+	char grub2_tmp_filename[] = "/tmp/grub_template_XXXXXX";
+	char grub2_filename[FILENAME_MAX] = {};
+	time_t now = time(NULL);
+	char *date;
+	FILE *fp;
+	int fd;
+
+	date = asctime(localtime(&now));
+	date[strlen(date) - 1] = '\0';
+
+	fd = mkstemp(grub2_tmp_filename);
+	if (fd < 0) {
+		syslog(LOG_ERR, "Cannot open %s to %s (%s:%u)\n",
+		       grub2_tmp_filename, grub2_filename, __FILE__, __LINE__);
+		return -1;
+	}
+
+	fp = fdopen(fd, "w+");
+	if (!fp) {
+		close(fd);
+		syslog(LOG_ERR, "Cannot open %s to %s (%s:%u)\n",
+		       grub2_tmp_filename, grub2_filename, __FILE__, __LINE__);
+		return -1;
+	}
+
+	fprintf(fp, OGRELIVE_TEMPLATE, date,
+		cfg->ogserver, cfg->ogrepo, cfg->ogrelivedir, cfg->username, cfg->passwd);
+
+	fclose(fp);
+
+	if (unlink(grub2_filename) < 0) {
+		if (errno != ENOENT) {
+			syslog(LOG_ERR, "Failed to delete %s (%s:%u)\n",
+			       grub2_filename, __FILE__, __LINE__);
+			return -1;
+		}
+	}
+
+	ogrelive_grub2_file_path(grub2_filename, sizeof(grub2_filename), mac);
+
+	if (rename(grub2_tmp_filename, grub2_filename) < 0) {
+		unlink(grub2_tmp_filename);
+		syslog(LOG_ERR, "Failed to rename %s to %s (%s:%u)\n",
+		       grub2_tmp_filename, grub2_filename, __FILE__, __LINE__);
+		return -1;
+	}
+
+	chmod(grub2_filename, 0644);
+
+	return 0;
+}