boot_linux.c

/*
 *
 * Copyright (C) 2021 BigfootACA <bigfoot@classfun.cn>
 *
 * SPDX-License-Identifier: LGPL-3.0-or-later
 *
 */

#ifdef ENABLE_GUI
#ifdef ENABLE_UEFI
#define _GNU_SOURCE
#include<stddef.h>
#include<stdlib.h>
#include<stdbool.h>
#include<Library/BaseLib.h>
#include<Library/MemoryAllocationLib.h>
#include<Library/UefiBootServicesTableLib.h>
#include"gui.h"
#include"list.h"
#include"confd.h"
#include"logger.h"
#include"linux_boot.h"
#include"filesystem.h"
#include"gui/tools.h"
#include"gui/sysbar.h"
#include"gui/msgbox.h"
#include"gui/activity.h"
#include"gui/filepicker.h"
#define TAG "linux-boot"

static const char*base="gui.boot_linux";
static linux_config*boot_cfg=NULL;
struct boot_linux{
	lv_obj_t*box,*ok,*cancel;
	lv_obj_t*btn_abootimg,*btn_kernel,*btn_initrd,*btn_dtb;
	lv_obj_t*clr_abootimg,*clr_kernel,*clr_initrd,*clr_dtb,*clr_cmdline;
	lv_obj_t*abootimg,*kernel,*initrd,*dtb;
	lv_obj_t*cmdline,*use_uefi;
};

static int boot_linux_get_focus(struct gui_activity*d){
	struct boot_linux*am=d->data;
	if(!am)return 0;
	lv_group_add_obj(gui_grp,am->clr_abootimg);
	lv_group_add_obj(gui_grp,am->abootimg);
	lv_group_add_obj(gui_grp,am->btn_abootimg);
	lv_group_add_obj(gui_grp,am->clr_kernel);
	lv_group_add_obj(gui_grp,am->kernel);
	lv_group_add_obj(gui_grp,am->btn_kernel);
	lv_group_add_obj(gui_grp,am->clr_initrd);
	lv_group_add_obj(gui_grp,am->initrd);
	lv_group_add_obj(gui_grp,am->btn_initrd);
	lv_group_add_obj(gui_grp,am->clr_dtb);
	lv_group_add_obj(gui_grp,am->dtb);
	lv_group_add_obj(gui_grp,am->btn_dtb);
	lv_group_add_obj(gui_grp,am->clr_cmdline);
	lv_group_add_obj(gui_grp,am->cmdline);
	lv_group_add_obj(gui_grp,am->use_uefi);
	lv_group_add_obj(gui_grp,am->ok);
	lv_group_add_obj(gui_grp,am->cancel);
	return 0;
}

static int boot_linux_lost_focus(struct gui_activity*d){
	struct boot_linux*am=d->data;
	if(!am)return 0;
	lv_group_remove_obj(am->clr_abootimg);
	lv_group_remove_obj(am->abootimg);
	lv_group_remove_obj(am->btn_abootimg);
	lv_group_remove_obj(am->clr_kernel);
	lv_group_remove_obj(am->kernel);
	lv_group_remove_obj(am->btn_kernel);
	lv_group_remove_obj(am->clr_initrd);
	lv_group_remove_obj(am->initrd);
	lv_group_remove_obj(am->btn_initrd);
	lv_group_remove_obj(am->clr_dtb);
	lv_group_remove_obj(am->dtb);
	lv_group_remove_obj(am->btn_dtb);
	lv_group_remove_obj(am->clr_cmdline);
	lv_group_remove_obj(am->cmdline);
	lv_group_remove_obj(am->use_uefi);
	lv_group_remove_obj(am->ok);
	lv_group_remove_obj(am->cancel);
	return 0;
}

extern int guiapp_main(int argc,char**argv);

static void do_boot(){
	linux_boot*boot;
	if(!boot_cfg)return;
	if(!(boot=linux_boot_new(boot_cfg)))return;
	if((linux_boot_prepare(boot))!=0)return;
	if((linux_boot_execute(boot))!=0)return;
	linux_boot_free(boot);
	return;
}

static int after_exit(void*d __attribute__((unused))){
	gui_run=true;
	logger_set_console(confd_get_boolean("boot.console_log",true));
	do_boot();
	logger_set_console(false);
	tlog_warn("boot failed");
	gBS->Stall(10*1000*1000);
	lv_obj_invalidate(lv_scr_act());
	return guiapp_main(1,(char*[]){"guiapp",NULL});
}

static bool set_path(lv_obj_t*ta,linux_load_from*from,const char*key){
	int r;
	fsh*f=NULL;
	const char*txt;
	if(!ta||!from)return false;
	if(!(txt=lv_textarea_get_text(ta)))return false;
	if(!txt[0])return true;
	confd_set_string_base(base,key,(char*)txt);
	r=fs_open(NULL,&f,txt,FILE_FLAG_READ);
	if(r!=0){
		msgbox_alert(
			"Load file '%s' failed: %s",
			txt,strerror(r)
		);
		tlog_warn(
			"load file '%s' failed: %s",
			txt,strerror(r)
		);
		return false;
	}
	from->enabled=true;
	from->type=FROM_FILE_SYSTEM_HANDLE;
	from->fsh=f;
	tlog_debug("loaded %s from '%s'",key,txt);
	return true;
}

static void ok_cb(lv_event_t*e){
	struct boot_linux*am=e->user_data;
	if(!am||e->target!=am->ok)return;
	linux_load_from*f=malloc(sizeof(linux_load_from));
	linux_config*cfg=linux_config_new();
	if(!cfg||!f){
		if(f)free(f);
		if(cfg)linux_config_free(cfg);
		return;
	}
	memset(f,0,sizeof(linux_load_from));
	AsciiStrCpyS(cfg->tag,sizeof(cfg->tag)-1,"<FROM_GUI>");
	set_path(am->abootimg,&cfg->abootimg,"abootimg");
	set_path(am->kernel,&cfg->kernel,"kernel");
	set_path(am->initrd,f,"initrd");
	set_path(am->dtb,&cfg->dtb,"dtb");
	list_obj_add_new(&cfg->initrd,f);
	const char*cmdline=lv_textarea_get_text(am->cmdline);
	if(cmdline&&cmdline[0]){
		confd_set_string_base(base,"cmdline",(char*)cmdline);
		strncpy(cfg->cmdline,cmdline,sizeof(cfg->cmdline)-1);
	}
	cfg->use_uefi=lv_obj_is_checked(am->use_uefi);
	confd_set_boolean_base(base,"use_uefi",cfg->use_uefi);
	if(!cfg->abootimg.enabled&&!cfg->kernel.enabled){
		msgbox_alert("No kernel specified");
		linux_config_free(cfg);
		return;
	}
	if(cfg->abootimg.enabled&&cfg->kernel.enabled){
		msgbox_alert("Conflict options");
		linux_config_free(cfg);
		return;
	}
	boot_cfg=cfg;
	gui_run_and_exit(after_exit);
}

static int do_cleanup(struct gui_activity*act){
	struct boot_linux*am=act->data;
	if(!am)return 0;
	free(am);
	act->data=NULL;
	return 0;
}

static int draw_boot_linux(struct gui_activity*act){
	char path[PATH_MAX];
	struct boot_linux*am=malloc(sizeof(struct boot_linux));
	if(!am)return -ENOMEM;
	memset(am,0,sizeof(struct boot_linux));
	act->data=am;
	am->box=lv_draw_dialog_box(act->page,NULL,"Boot Linux");
	lv_draw_input(am->box,"Android Boot Image",     NULL,&am->clr_abootimg,&am->abootimg,&am->btn_abootimg);
	lv_draw_input(am->box,"Linux Kernel",           NULL,&am->clr_kernel,  &am->kernel,  &am->btn_kernel);
	lv_draw_input(am->box,"Ramdisk (initramfs)",    NULL,&am->clr_initrd,  &am->initrd,  &am->btn_initrd);
	lv_draw_input(am->box,"Device Tree Blob (dtb)", NULL,&am->clr_dtb,     &am->dtb,     &am->btn_dtb);
	lv_draw_input(am->box,"Kernel Commandline",     NULL,&am->clr_cmdline, &am->cmdline, NULL);
	lv_textarea_set_one_line(am->cmdline,false);
	am->use_uefi=lv_draw_checkbox(am->box,"Boot with UEFI runtime",false,NULL,NULL);
	lv_draw_btns_ok_cancel(am->box,&am->ok,&am->cancel,ok_cb,am);

	#define set_text(tag) \
	lv_textarea_set_text(\
		am->tag,\
		confd_get_sstring_base(\
			base,#tag,"",\
			path,sizeof(path)\
		)\
	);
	set_text(abootimg);
	set_text(kernel);
	set_text(initrd);
	set_text(dtb);
	set_text(cmdline);
	lv_obj_set_checked(
		am->use_uefi,
		confd_get_boolean_base(
			base,"use_uefi",false
		)
	);
	return 0;
}

struct gui_register guireg_boot_linux={
	.name="boot-linux",
	.title="Boot Linux",
	.show_app=true,
	.open_file=false,
	.quiet_exit=do_cleanup,
	.get_focus=boot_linux_get_focus,
	.lost_focus=boot_linux_lost_focus,
	.draw=draw_boot_linux,
	.back=true,
	.mask=true,
};
#endif
#endif