/*
 * Guillaume Cottenceau (was gc@mandrakesoft.com)
 *
 * Copyright 2000-2004 Mandrakesoft
 *
 *
 * This software may be freely redistributed under the terms of the GNU
 * public license.
 *
 * 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.
 *
 */

/*
 * Portions from Erik Troan (ewt@redhat.com)
 *
 * Copyright 1996 Red Hat Software 
 *
 */

#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdio.h>
#include <sys/ioctl.h>
#include <sys/mount.h>
#include <string.h>
#include <errno.h>
#include <ctype.h>
#include <stdarg.h>
#include <signal.h>
#include <linux/unistd.h>
_syscall2(int,pivot_root,const char *,new_root,const char *,put_old)

#include "stage1.h"

#include "log.h"
#include "probing.h"
#include "frontend.h"
#include "modules.h"
#include "tools.h"
#include "automatic.h"
#include "mount.h"
#include "insmod.h"
#include "thirdparty.h"

#ifdef ENABLE_PCMCIA
#include "pcmcia_/pcmcia.h"
#endif

#ifndef DISABLE_CDROM
#include "cdrom.h"
#endif

#ifndef DISABLE_NETWORK
#include "network.h"
#endif

#ifndef DISABLE_DISK
#include "disk.h"
#endif


/************************************************************
 * globals */



void fatal_error(char *msg)
{
	printf("FATAL ERROR IN STAGE1: %s\n\nI can't recover from this.\nYou may reboot your system.\n", msg);
	while (1);
}


/************************************************************
 * special frontend functs
 * (the principle is to not pollute frontend code with stage1-specific stuff) */

void stg1_error_message(char *msg, ...)
{
	va_list args;
	va_start(args, msg);
	unset_automatic();
	verror_message(msg, args);
	va_end(args);
}

void stg1_fatal_message(char *msg, ...)
{
	va_list args;
	va_start(args, msg);
	unset_automatic();
	verror_message(msg, args);
	va_end(args);
        exit(1);
}

void stg1_info_message(char *msg, ...)
{
	va_list args;
	va_start(args, msg);
	if (IS_AUTOMATIC) {
		vlog_message(msg, args);
		return;
	}
	vinfo_message(msg, args);
	va_end(args);
}


#ifdef SPAWN_SHELL
static pid_t shell_pid = 0;

/************************************************************
 * spawns a shell on console #2 */
static void spawn_shell(void)
{
	int fd;
	char * shell_name[] = { "/tmp/sh", NULL };

	log_message("spawning a shell");

	if (!IS_TESTING) {
		fd = open("/dev/tty2", O_RDWR);
		if (fd == -1) {
			log_message("cannot open /dev/tty2 -- no shell will be provided");
			return;
		}
		else if (access(shell_name[0], X_OK)) {
			log_message("cannot open shell - %s doesn't exist", shell_name[0]);
			return;
		}
		
		if (!(shell_pid = fork())) {
			dup2(fd, 0);
			dup2(fd, 1);
			dup2(fd, 2);
			
			close(fd);
			setsid();
			if (ioctl(0, TIOCSCTTY, NULL))
				log_perror("could not set new controlling tty");

			execv(shell_name[0], shell_name);
			log_message("execve of %s failed: %s", shell_name[0], strerror(errno));
			exit(-1);
		}
		
		close(fd);
	}
}
#endif

#ifdef SPAWN_INTERACTIVE
char * interactive_fifo = "/tmp/stage1-fifo";
static pid_t interactive_pid = 0;

/* spawns my small interactive on console #6 */
static void spawn_interactive(void)
{
	int fd;
	char * dev = "/dev/tty6";

	printf("spawning my interactive on %s\n", dev);

	if (!IS_TESTING) {
		fd = open(dev, O_RDWR);
		if (fd == -1) {
			printf("cannot open %s -- no interactive\n", dev);
			return;
		}

		if (mkfifo(interactive_fifo, O_RDWR)) {
			printf("cannot create fifo -- no interactive\n");
			return;
		}
		
		if (!(interactive_pid = fork())) {
			int fif_out;

			dup2(fd, 0);
			dup2(fd, 1);
			dup2(fd, 2);
			
			close(fd);
			setsid();
			if (ioctl(0, TIOCSCTTY, NULL))
				perror("could not set new controlling tty");

			fif_out = open(interactive_fifo, O_WRONLY);
			printf("Please enter your command (availables: [+,-] [rescue]).\n");
				
			while (1) {
				char s[50];
				int i = 0;
				printf("? ");
				fflush(stdout);
				read(0, &(s[i++]), 1);
				fcntl(0, F_SETFL, O_NONBLOCK);
				while (read(0, &(s[i++]), 1) > 0 && i < sizeof(s));
				fcntl(0, F_SETFL, 0);
				write(fif_out, s, i-2);
				printf("Ok.\n");
			}
		}
		
		close(fd);
	}
}
#endif


#ifdef ENABLE_PCMCIA
static void handle_pcmcia(void)
{
        char * pcmcia_adapter;
	if (kernel_version() == 2) {
		stg1_error_message("We now use kernel pcmcia support and this won't work with a 2.2 kernel.");
		return;
	}

	pcmcia_adapter = pcmcia_probe();
	if (!pcmcia_adapter) {
		log_message("no pcmcia adapter found");
		return;
	}
	my_insmod("pcmcia_core", ANY_DRIVER_TYPE, NULL, 0);
	my_insmod(pcmcia_adapter, ANY_DRIVER_TYPE, NULL, 0);
	/* ds is an alias for pcmcia in recent 2.6 kernels
           but we don't have modules.alias in install, so try to load both */
	my_insmod("ds", ANY_DRIVER_TYPE, NULL, 0);
	my_insmod("pcmcia", ANY_DRIVER_TYPE, NULL, 0);
	
        /* call to cardmgr takes time, let's use the wait message */
	wait_message("Enabling PCMCIA extension cards...");
	log_message("cardmgr rc: %d", cardmgr_call());
	remove_wait_message();

        add_to_env("PCMCIA", pcmcia_adapter);
}
#endif


/************************************************************
 */

static void method_select_and_prepare(void)
{
	enum return_type results;
	char * choice;
	char * means[10], * means_auto[10];
	int i;

#ifndef DISABLE_DISK
	char * disk_install = "Hard disk"; char * disk_install_auto = "disk";
#endif
#ifndef DISABLE_CDROM
	char * cdrom_install = "CDROM drive"; char * cdrom_install_auto = "cdrom";
#endif
#ifndef DISABLE_NETWORK
	char * network_nfs_install = "NFS server"; char * network_nfs_install_auto = "nfs";
	char * network_ftp_install = "FTP server"; char * network_ftp_install_auto = "ftp";
	char * network_http_install = "HTTP server"; char * network_http_install_auto = "http";
#endif
	char * thirdparty_install = "Load third party modules"; char * thirdparty_install_auto = "thirdparty";

	i = 0;
#ifndef DISABLE_NETWORK
	means[i] = network_nfs_install; means_auto[i++] = network_nfs_install_auto;
	means[i] = network_ftp_install; means_auto[i++] = network_ftp_install_auto;
	means[i] = network_http_install; means_auto[i++] = network_http_install_auto;
#endif
#ifndef DISABLE_CDROM
	means[i] = cdrom_install; means_auto[i++] = cdrom_install_auto;
        allow_additional_modules_floppy = 0;
#endif
#ifndef DISABLE_DISK
	means[i] = disk_install; means_auto[i++] = disk_install_auto;
        allow_additional_modules_floppy = 0;
#endif
	means[i] = thirdparty_install; means_auto[i++] = thirdparty_install_auto;
	means[i] = NULL;

	unlink(IMAGE_LOCATION);
	rmdir(IMAGE_LOCATION); /* useful if we change the method, eg: we have automatic:cdrom but go back to nfs */

	results = ask_from_list_auto("Please choose the installation method.", means, &choice, "method", means_auto);

	if (results != RETURN_OK)
		return method_select_and_prepare();

#ifndef DISABLE_CDROM
	if (!strcmp(choice, cdrom_install))
		results = cdrom_prepare();
#endif
        
#ifndef DISABLE_DISK
	if (!strcmp(choice, disk_install))
		results = disk_prepare();
#endif
	
#ifndef DISABLE_NETWORK
	if (!strcmp(choice, network_nfs_install))
		results = nfs_prepare();

#ifndef MANDRAKE_MOVE
	if (!strcmp(choice, network_ftp_install))
		results = ftp_prepare();
	
	if (!strcmp(choice, network_http_install))
		results = http_prepare();
#endif
#endif

	if (!strcmp(choice, thirdparty_install)) {
		thirdparty_load_modules();
		return method_select_and_prepare();
        }

	if (results != RETURN_OK)
		return method_select_and_prepare();

        /* try to find third party modules on the install media */
        thirdparty_load_media_modules();
}

static enum return_type create_initial_fs_symlinks(char* symlinks)
{
        FILE *f;
        char buf[5000];

        if (scall(!(f = fopen(symlinks, "rb")), "fopen"))
                return RETURN_ERROR;
        while (fgets(buf, sizeof(buf), f)) {
                char oldpath[500], newpath[500], newpathfinal[500];
                buf[strlen(buf)-1] = '\0';  // trim \n
                if (sscanf(buf, "%s %s", oldpath, newpath) != 2) {
                        sprintf(oldpath, "%s%s", STAGE2_LOCATION_ROOTED, buf);
                        sprintf(newpathfinal, "%s%s", SLASH_LOCATION, buf);
                } else {
                        sprintf(newpathfinal, "%s%s", SLASH_LOCATION, newpath);
                }
                log_message("creating symlink %s -> %s", oldpath, newpathfinal);
                if (scall(symlink(oldpath, newpathfinal), "symlink"))
                        return RETURN_ERROR;
        }
        fclose(f);
        return RETURN_OK;
}

static enum return_type create_initial_fs_devices(char* devices)
{
        FILE *f;
        char buf[5000];

        // need to create the few devices needed to start up stage2 in a decent manner, we can't symlink or they will keep CD busy
        if (scall(mkdir(SLASH_LOCATION "/dev", 0755), "mkdir"))
                return RETURN_ERROR;
        if (scall(!(f = fopen(devices, "rb")), "fopen"))
                return RETURN_ERROR;
        while (fgets(buf, sizeof(buf), f)) {
                char name[500], path[500], type;
                int major, minor;
                sscanf(buf, "%s %c %d %d", name, &type, &major, &minor);
                sprintf(path, "%s%s", SLASH_LOCATION, name);
                log_message("creating device %s %c %d %d", path, type, major, minor);
                if (scall(mknod(path, (type == 'c' ? S_IFCHR : S_IFBLK) | 0600, makedev(major, minor)), "mknod"))
                        return RETURN_ERROR;
        }
        fclose(f);
        return RETURN_OK;
}

#ifdef MANDRAKE_MOVE
static enum return_type handle_move_clp(char* clp_name, char* live, char* location_live, char* location_mount, int* is_symlink, int preload)
{
	if (mount_clp_may_preload(clp_name, location_mount, preload) == RETURN_OK) {
		return RETURN_OK;
	} else {	
		char *full_live = asprintf_("%s%s", location_live, live);
                log_message("no %s found (or disabled), trying to fallback on plain tree", clp_name);
                if (!access(full_live, R_OK)) {
                        if (scall(symlink(location_live, location_mount), "symlink"))
                                return RETURN_ERROR;
                        *is_symlink = 1;
                        return RETURN_OK;
                } else {
                        log_message("move: can't find %s nor %s, proceeding hoping files will be there", clp_name, full_live);
                        return RETURN_OK;
                }
        }
}

int mandrake_move_post(void)
{
        int boot__real_is_symlink_to_raw = 0;
        int always__real_is_symlink_to_raw = 0;
        int totem__real_is_symlink_to_raw = 0;
        int main__real_is_symlink_to_raw = 0;

        if (handle_move_clp("live_tree_boot.clp", "/usr/bin/runstage2.pl",
                       IMAGE_LOCATION "/live_tree_boot", BOOT_LOCATION,
                       &boot__real_is_symlink_to_raw, 1) != RETURN_OK)
                return RETURN_ERROR;

        if (handle_move_clp("live_tree_always.clp", "/bin/bash",
                       IMAGE_LOCATION "/live_tree_always", ALWAYS_LOCATION,
                       &always__real_is_symlink_to_raw, 1) != RETURN_OK)
                return RETURN_ERROR;

        if (handle_move_clp("live_tree_totem.clp", "/usr/bin/totem",
                       IMAGE_LOCATION "/live_tree_totem", TOTEM_LOCATION,
                       &totem__real_is_symlink_to_raw, 1) != RETURN_OK)
                return RETURN_ERROR;

        if (handle_move_clp("live_tree.clp", "/etc/fstab",
                       IMAGE_LOCATION "/live_tree", STAGE2_LOCATION,
                       &main__real_is_symlink_to_raw, 0) != RETURN_OK)
                return RETURN_ERROR;
       
        // in case we didn't mount any clp, because gzloop.o is not available later in /lib/modules
	my_insmod("gzloop", ANY_DRIVER_TYPE, NULL, 0);
        
        // hardcoded :(
        if (!access(TOTEM_LOCATION, R_OK)) {
                if (scall(symlink("/image_totem/usr", SLASH_LOCATION "/usr"), "symlink"))
                        return RETURN_ERROR;
        } else
                // need a fallback in case we don't use image_totem.clp nor live_tree_totem, but we're in -u mode
                if (scall(symlink(STAGE2_LOCATION_ROOTED "/usr", SLASH_LOCATION "/usr"), "symlink"))
                        return RETURN_ERROR;

        if (create_initial_fs_symlinks(STAGE2_LOCATION "/move/symlinks") != RETURN_OK ||
	    create_initial_fs_devices(STAGE2_LOCATION "/move/devices") != RETURN_OK)
                return RETURN_ERROR;

        if (boot__real_is_symlink_to_raw) {
                if (scall(unlink(BOOT_LOCATION), "unlink"))
                        return RETURN_ERROR;
                if (scall(symlink(IMAGE_LOCATION_REL "/live_tree_boot", BOOT_LOCATION), "symlink"))
                        return RETURN_ERROR;
        }

        if (always__real_is_symlink_to_raw) {
                if (scall(unlink(ALWAYS_LOCATION), "unlink"))
                        return RETURN_ERROR;
                if (scall(symlink(IMAGE_LOCATION_REL "/live_tree_always", ALWAYS_LOCATION), "symlink"))
                        return RETURN_ERROR;
        }

        if (totem__real_is_symlink_to_raw) {
                if (scall(unlink(TOTEM_LOCATION), "unlink"))
                        return RETURN_ERROR;
                if (scall(symlink(IMAGE_LOCATION_REL "/live_tree_totem", TOTEM_LOCATION), "symlink"))
                        return RETURN_ERROR;
        }

        if (main__real_is_symlink_to_raw) {
                if (scall(unlink(STAGE2_LOCATION), "unlink"))
                        return RETURN_ERROR;
                if (scall(symlink(IMAGE_LOCATION_REL "/live_tree", STAGE2_LOCATION), "symlink"))
                        return RETURN_ERROR;
        }
	return RETURN_OK;
}
#endif

int do_pivot_root(void)
{
        int fd;
        char rootdev[] = "0x0100"; 

        if (IS_DEBUGSTAGE1)
                while (1);

        log_message("pivot_rooting");
        // trick so that kernel won't try to mount the root device when initrd exits
        if (scall((fd = open("/proc/sys/kernel/real-root-dev", O_WRONLY)) < 0, "open"))
                return RETURN_ERROR;
        if (scall(write(fd, rootdev, strlen(rootdev)) != (signed)strlen(rootdev), "write")) {
                close(fd);
                return RETURN_ERROR;
        }
        close(fd);

        if (scall(mkdir(SLASH_LOCATION "/stage1", 0755), "mkdir"))
                return RETURN_ERROR;

        if (scall(pivot_root(SLASH_LOCATION, SLASH_LOCATION "/stage1"), "pivot_root"))
                return RETURN_ERROR;

        return RETURN_OK;
}

void finish_preparing(void)
{
#ifdef MANDRAKE_MOVE
	if (mandrake_move_post() != RETURN_OK)
		stg1_fatal_message("Fatal error when launching MandrakeMove.");
#else
	mkdir(SLASH_LOCATION "/etc", 0755);
	mkdir(SLASH_LOCATION "/var", 0755);
	if (IS_RESCUE) {
                if (create_initial_fs_symlinks(STAGE2_LOCATION "/usr/share/symlinks") != RETURN_OK)
                        stg1_fatal_message("Fatal error finishing initialization.");

	} else {
                if (create_initial_fs_symlinks(STAGE2_LOCATION "/usr/share/symlinks") != RETURN_OK ||
		    create_initial_fs_devices(STAGE2_LOCATION "/usr/share/devices") != RETURN_OK)
                        stg1_fatal_message("Fatal error finishing initialization.");
	}
#endif

	/* /tmp/syslog is used by the second init, so it must be copied now, not in stage2 */
	/* we remove it to ensure the old one is not copied over it in stage2 */
	copy_file("/tmp/syslog", SLASH_LOCATION "/tmp/syslog", NULL);
	unlink("/tmp/syslog");
	copy_file("/etc/resolv.conf", SLASH_LOCATION "/etc/resolv.conf", NULL);
	mkdir(SLASH_LOCATION "/modules", 0755);
	copy_file("/modules/modules.dep", SLASH_LOCATION "/modules/modules.dep", NULL);

	if (!IS_RESCUE) {
		copy_file(STAGE2_LOCATION "/etc/init", SLASH_LOCATION "/etc/init", NULL);
		chmod(SLASH_LOCATION "/etc/init", 0755);
	}
                
	umount("/tmp/tmpfs");
	do_pivot_root();

	if (file_size(IS_RESCUE ? "/sbin/init" : "/etc/init") == -1)
		stg1_fatal_message("Fatal error giving hand to second stage.");

#ifdef SPAWN_SHELL
	if (shell_pid != 0) {
                int fd;
		kill(shell_pid, 9);
		fd = open("/dev/tty2", O_RDWR);
                write(fd, "Killed\n", 7);
                close(fd);
        }
#endif
}

int main(int argc __attribute__ ((unused)), char **argv __attribute__ ((unused)), char **env)
{
#ifdef ENABLE_NETWORK_STANDALONE
	open_log();
	init_frontend("");

	unlink("/etc/resolv.conf"); /* otherwise it is read-only */
	set_param(MODE_AUTOMATIC);
	grab_automatic_params("network:dhcp");

	intf_select_and_up();
	finish_frontend();
	return 0;
#else
	if (getenv("DEBUGSTAGE1")) {
		set_param(MODE_DEBUGSTAGE1);
		set_param(MODE_TESTING);
        }

	if (!IS_TESTING) {
		mkdir(SLASH_LOCATION, 0755);
		if (scall(mount("none", SLASH_LOCATION, "tmpfs", MS_MGC_VAL, NULL), "mount tmpfs"))
			fatal_error("Fatal error initializing.");
		mkdir(SLASH_LOCATION "/tmp", 0755);
	}

#ifdef SPAWN_INTERACTIVE
	spawn_interactive();
#endif

	open_log();
	log_message("welcome to the " DISTRIB_NAME " install (mdk-stage1, version " DISTRIB_VERSION " built " __DATE__ " " __TIME__")");
	process_cmdline();
#ifdef SPAWN_SHELL
	spawn_shell();
#endif
	init_modules_insmoding();
	init_frontend("Welcome to " DISTRIB_DESCR ", " __DATE__ " " __TIME__);

        /* load usb interface as soon as possible, helps usb mouse detection in stage2 */
	probe_that_type(USB_CONTROLLERS, BUS_USB);

	if (IS_THIRDPARTY)
		thirdparty_load_modules();

#ifdef ENABLE_PCMCIA
	if (!IS_NOAUTO)
		handle_pcmcia();
#endif
        
	if (IS_CHANGEDISK)
		stg1_info_message("You are starting the installation with an alternate booting method. "
				  "Please change your disk, and insert the Installation disk.");

	if (IS_RESCUE && total_memory() < MEM_LIMIT_RESCUE) {
		stg1_error_message("You are starting the rescue with a low memory configuration. "
				   "Our experience shows that your system may crash at any point "
				   "or lock up for no apparent reason. Continue at "
				   "your own risk. Alternatively, you may reboot your system now.");
	}

#ifdef MANDRAKE_MOVE
	if (total_memory() < MEM_LIMIT_MOVE)
		stg1_info_message(DISTRIB_NAME " typically needs more than %d Mbytes of memory (detected %d Mbytes). You may proceed, but the machine may crash or lock up for no apparent reason. Continue at your own risk. Alternatively, you may reboot your system now.",
				   MEM_LIMIT_MOVE, total_memory());
#endif
        method_select_and_prepare();

#ifndef MANDRAKE_MOVE
	if (access(STAGE2_LOCATION, R_OK) != 0)
		if (symlink(IMAGE_LOCATION_REL "/" LIVE_LOCATION_REL, STAGE2_LOCATION) != 0)
			log_perror("symlink from " IMAGE_LOCATION_REL "/" LIVE_LOCATION_REL " to " STAGE2_LOCATION " failed");
#endif

#ifdef SPAWN_INTERACTIVE
	if (interactive_pid != 0)
		kill(interactive_pid, 9);
#endif

	finish_preparing();

	finish_frontend();
	close_log();

	if (IS_TESTING)
		return 0;
	else
		return 66;
#endif
}