git-svn-id: http://svn.ipfire.org/svn/ipfire/1.4/source@36 ea5c0bd1-69bd-2848-81d8...

[people/teissler/ipfire-2.x.git] / src / install+setup / install / install2.c

/* IPCop install2 program.\r
 *\r
 * This program is distributed under the terms of the GNU General Public\r
 * Licence.  See the file COPYING for details.\r
 *\r
 * (c) Lawrence Manning, 2001\r
 * (c) Franck Bourdonnec, 2006\r
 * Contains update/restore code\r
 * \r
 * $Id: install2.c,v 1.1.2.3 2006/01/31 00:51:50 franck78 Exp $\r
 * \r
 */\r
#include "install.h"\r
\r
FILE *flog = NULL;\r
char *mylog;\r
char **ctr;\r
\r
/* \r
    To include a translated string in the final installer, you must reference\r
    it here with a simplr comment. This save a lot a space in the installer\r
*/\r
\r
/* TR_BUILDING_INITRD */\r
/* TR_HELPLINE */\r
/* TR_SKIP */\r
/* TR_RESTORE_CONFIGURATION */\r
/* TR_RESTORE */\r
/* TR_OK */\r
/* TR_CANCEL */\r
/* TR_ERROR */\r
/* TR_INSTALLING_FILES */\r
/* TR_FAILED_TO_FIND */\r
/* TR_UNABLE_TO_INSTALL_FILES */\r
/* TR_LOADING_PCMCIA */\r
\r
//libsmooth\r
/* TR_INTERFACE */\r
/* TR_ENTER_THE_IP_ADDRESS_INFORMATION */\r
/* TR_STATIC */\r
/* TR_DHCP_HOSTNAME */\r
/* TR_IP_ADDRESS_PROMPT */\r
/* TR_NETMASK_PROMPT */\r
/* TR_INVALID_FIELDS */\r
/* TR_IP_ADDRESS_CR */\r
/* TR_NETWORK_MASK_CR */\r
/* TR_DHCP_HOSTNAME_CR */\r
/* TR_LOOKING_FOR_NIC */\r
/* TR_MANUAL */\r
/* TR_SELECT_NETWORK_DRIVER */\r
/* TR_SELECT_NETWORK_DRIVER_LONG */\r
/* TR_UNABLE_TO_LOAD_DRIVER_MODULE */\r
/* TR_THIS_DRIVER_MODULE_IS_ALREADY_LOADED */\r
/* TR_MODULE_PARAMETERS */\r
/* TR_LOADING_MODULE */\r
/* TR_MODULE_NAME_CANNOT_BE_BLANK */\r
\r
//upgrade 120\r
/* TR_UNABLE_TO_OPEN_SETTINGS_FILE */\r
/* TR_DOMAINNAME */\r
/* TR_ENTER_DOMAINNAME */\r
/* TR_DOMAINNAME_CANNOT_CONTAIN_SPACES */\r
/* TR_UNABLE_TO_MOUNT_PROC_FILESYSTEM */\r
/* TR_UNABLE_TO_WRITE_ETC_FSTAB */\r
\r
// dir to find files, chrooted or not...\r
#define TMP_EXTRACT_CH "/tmp/ipcop"\r
#define TMP_EXTRACT "/harddisk" TMP_EXTRACT_CH\r
#define MOUNT_BACKUP_CH "/mnt/usb"\r
#define MOUNT_BACKUP "/harddisk" MOUNT_BACKUP_CH\r
/*\r
    return 0 when dev contains a backup set\r
    leave dev mounted\r
*/\r
int try_mount (char *dev, char *testfile) {\r
	char commandstring[STRING_SIZE];\r
	mysystem("/bin/umount " MOUNT_BACKUP);\r
	sprintf(commandstring, "/bin/mount -t vfat -o ro %s " MOUNT_BACKUP, dev);\r
	mysystem(commandstring);\r
\r
	/*verify it's what we want */\r
	sprintf(commandstring, MOUNT_BACKUP "/%s.dat", testfile);\r
	FILE *handle = fopen(commandstring, "r");\r
	if (handle == NULL) {\r
		return 1;   /* bad disk ! */\r
	}\r
	fclose(handle);\r
	\r
	handle = fopen(MOUNT_BACKUP "/backup.key", "r");\r
	if (handle == NULL) {\r
		return 1;   /* bad disk ! */\r
	}\r
	fclose(handle);\r
	return 0; //success\r
}\r
\r
/* try to mount usb device until backup.tgz is found except the \r
    destination device (scsi names are identical with usb key)\r
    check "sda sdb sdc sdd"\r
*/\r
int mountbackup (char *testfile, char *destination_device) {\r
	char sourcedev[30];\r
	char i,j;\r
	for (i = 'a'; i < 'e'; i++) {\r
		sprintf (sourcedev,"/dev/sd%c ",i);\r
		if (strcmp (destination_device, sourcedev) != 0) {\r
			if (!try_mount (sourcedev, testfile)) return 0;\r
		}\r
		for (j = '1'; j < '5'; j++) {\r
			sourcedev[8] = j;\r
			if (strcmp (destination_device, sourcedev) != 0) {\r
				if (!try_mount (sourcedev, testfile)) return 0;\r
			}\r
		}\r
	}\r
	return 1;\r
}\r
\r
int floppy_locate() {\r
	/* Temporarily mount /proc under /harddisk/proc,\r
           run updfstab to locate the floppy, and unmount /harddisk/proc\r
           again.  This should be run each time the user tries to restore\r
           so it can properly detect removable devices */\r
	if (mysystem("/bin/mount -n -t proc /proc /harddisk/proc")) {\r
		errorbox(ctr[TR_UNABLE_TO_MOUNT_PROC_FILESYSTEM]);\r
		return 1;\r
	}\r
	if (mysystem("/bin/chroot /harddisk /usr/sbin/updfstab")) {\r
		errorbox(ctr[TR_UNABLE_TO_WRITE_ETC_FSTAB]);\r
	    	return 1;\r
	}\r
	mysystem("/bin/umount /harddisk/proc");\r
	return 0;\r
}\r
\r
/* Check the SQUID acl file exists, if not use our 1.4 copy */\r
void fixup_squidacl() {\r
	FILE *aclreadfile;\r
	if ((aclreadfile = fopen ("/harddisk" CONFIG_ROOT "/proxy/acl", "r"))) {\r
		unlink ("/harddisk" CONFIG_ROOT "/proxy/acl-1.4");\r
		fclose(aclreadfile);\r
	} else {\r
		rename ("/harddisk" CONFIG_ROOT "/proxy/acl-1.4",\r
			"/harddisk" CONFIG_ROOT "/proxy/acl");\r
	}\r
	chown  ("/harddisk" CONFIG_ROOT "/proxy/acl", 99, 99);\r
}\r
/* if we detected SCSI then fixup */\r
void fixup_initrd() {\r
	FILE *handle;\r
	char line[STRING_SIZE];\r
	char commandstring[STRING_SIZE];\r
	\r
        if (!(handle = fopen("/scsidriver", "r"))) \r
	    return;\r
\r
	char *driver;\r
       	fgets(line, STRING_SIZE-1, handle);\r
    	fclose(handle);\r
	line[strlen(line) - 1] = 0;\r
	driver = strtok(line, ".");\r
	fprintf(flog, "Detected SCSI driver %s\n", driver);\r
	if (!strlen(driver) > 1) \r
		return;\r
		\r
	fprintf(flog, "Fixing up ipcoprd.img\n");\r
	mysystem("/bin/chroot /harddisk /sbin/modprobe loop");\r
	mkdir("/harddisk/initrd", S_IRWXU|S_IRWXG|S_IRWXO);\r
  	sprintf(commandstring, "/bin/chroot /harddisk /sbin/mkinitrd"\r
				" --with=scsi_mod --with=%s --with=sd_mod"\r
				" --with=sr_mod --with=libata"\r
				" --with=ataraid /boot/ipcoprd.img "KERNEL_VERSION,\r
				driver );\r
	runcommandwithstatus(commandstring, ctr[TR_BUILDING_INITRD]);\r
#ifdef __i386__\r
	sprintf(commandstring, "/bin/chroot /harddisk /sbin/mkinitrd"\r
				" --with=scsi_mod --with=%s --with=sd_mod"\r
				" --with=sr_mod --with=libata"\r
				" --with=ataraid /boot/ipcoprd-smp.img "KERNEL_VERSION"-smp",\r
				driver );\r
	runcommandwithstatus(commandstring, ctr[TR_BUILDING_INITRD]);\r
	mysystem("/bin/chroot /harddisk /bin/mv /boot/grub/scsigrub.conf /boot/grub/grub.conf");\r
#endif\r
#ifdef __alpha__\r
	runcommandwithstatus("/bin/chroot /harddisk /bin/mv /boot/etc/scsiaboot.conf /boot/etc/aboot.conf", ctr[TR_BUILDING_INITRD]);\r
#endif\r
}\r
/* when backup is ready in tmpdir, move files to definitive location */\r
void do_copy_files(int upgrade_level) {\r
	mysystem("/bin/chroot /harddisk /bin/cp -af "TMP_EXTRACT_CH"/. /");\r
	/* Upgrade necessary files from v1.2 to v1.3 to v1.4 */\r
	switch (upgrade_level) {\r
	case 1:\r
		upgrade_v12_v13();\r
		upgrade_v130_v140();\r
	case 2: //some 1.4 files format changed\r
		//between 1.4.0 & 1.4.11  If possible de determine backup/version\r
		//the update code should go here\r
	}\r
}\r
\r
int main(int argc, char *argv[]) {\r
#define LANG argv[1]\r
#define DEST_DEV argv[2]\r
#define WGET argv[3]\r
\r
#ifdef	LANG_EN_ONLY\r
        char **langtrs[] = { en_tr, NULL };\r
#else\r
	char **langtrs[] = { bz_tr, cs_tr, da_tr, de_tr, en_tr, es_tr, fr_tr, el_tr, it_tr, la_tr, hu_tr, nl_tr, no_tr, pl_tr, pt_tr, sk_tr, so_tr, fi_tr, sv_tr, tr_tr, vi_tr, NULL };\r
#endif\r
	char message[1000];\r
	char title[STRING_SIZE];\r
	char commandstring[STRING_SIZE];\r
\r
	setlocale (LC_ALL, "");	\r
	/* Log file/terminal stuff. */\r
	mylog = "/dev/tty2";\r
	ctr = langtrs[ atoi(LANG) ];\r
\r
	if (!(flog = fopen(mylog, "w+")))\r
	{\r
		printf("Couldn't open log terminal\n");\r
		return 0;\r
	}\r
	fprintf(flog, "Install2 program started.\n");\r
	newtInit();\r
	newtCls();\r
	strcpy (title, NAME " v" VERSION " - " SLOGAN);\r
	newtDrawRootText(14, 0, title);\r
	newtPushHelpLine(ctr[TR_HELPLINE]);\r
/*\r
	// build now the device node\r
	runcommandwithstatus("echo 'cd /dev; ./make_devices'>/harddisk/X;" \r
				    "chroot /harddisk chmod +x /X;"\r
				    "chroot /harddisk /X;"\r
				    "chroot /harddisk rm /X"				    \r
				    , ctr[TR_INSTALLING_FILES]);\r
*/\r
	/* working dirs... */\r
    	mkdir(MOUNT_BACKUP, S_IRWXU|S_IRWXG|S_IRWXO);\r
\r
	//create the GUI screen and objects\r
	newtComponent form, header, labelfile, labelkey, file, key, radio0, radio1, radio2, radio3, radio4, ok;\r
\r
	newtCenteredWindow (55,20,ctr[TR_RESTORE]);\r
	form = newtForm (NULL, NULL,0);\r
\r
	sprintf(message, ctr[TR_RESTORE_CONFIGURATION], NAME);\r
	header = newtTextboxReflowed (2,1,message,51,0,0,0);\r
	newtFormAddComponent(form, header);\r
\r
        // The four method of restauration\r
        int start1=1, start2=0, start3=0, start4=0;\r
        radio1 = newtRadiobutton (17, 5, ctr[TR_SKIP], start1, NULL);\r
        radio2 = newtRadiobutton (17, 6, "Floppy (legacy)", start2, radio1);\r
        radio3 = newtRadiobutton (17, 7, "Usb-storage/CDROM", start3, radio2);\r
        if (strcmp(WGET,"none"))\r
            radio4 = newtRadiobutton (17, 8, "HTTP/FTP", start4, radio3);\r
	else\r
            radio4 = NULL;\r
        newtFormAddComponents(form, radio1, radio2, radio3, radio4, NULL);\r
\r
        // The optionnal filename for 'backup'\r
        labelfile=newtTextbox(12, 10, 35, 1, 0);\r
        newtTextboxSetText (labelfile, "Filename");\r
	newtFormAddComponent(form, labelfile);\r
        char *filevalue;\r
        char fileinit[STRING_SIZE] = "backup";\r
        file = newtEntry (17, 11, fileinit, 20, &filevalue, 0);\r
	newtFormAddComponent(form, file);\r
\r
        // The optionnal password for the key\r
        labelkey=newtTextbox(12, 13, 35, 1, 0);\r
        newtTextboxSetText (labelkey, "Backup key password");\r
	newtFormAddComponent(form, labelkey);\r
        char *keyvalue;\r
        char keyinit[STRING_SIZE] = "";\r
        key = newtEntry (17, 14, keyinit, 20, &keyvalue, 0);\r
	newtFormAddComponent(form, key);\r
\r
        // The OK button\r
	ok=newtButton (23, 16, ctr[TR_OK]);\r
	newtFormAddComponent(form, ok);\r
\r
	/* loop until succeeds or user skips out */\r
	int retcode = -1;\r
	while ( retcode<0 ) {\r
\r
	    // run the windows\r
	    struct newtExitStruct reponse;\r
	    newtFormRun (form, &reponse);\r
	    radio0 = newtRadioGetCurrent(radio1);\r
            int radio;\r
            radio = radio0 == radio1 ? 1 : radio0 == radio2 ? 2 :  radio0 == radio3 ? 3 : radio0 == radio4 ? 4 : 0;\r
            strcpy(keyinit,keyvalue);	//reuse actual value\r
            strcpy(fileinit,filevalue);\r
\r
	    if (radio==1) {\r
		    retcode = 1;	// no restore: nothing special\r
		    break;		// out of the while loop\r
	    }\r
\r
	    mkdir(TMP_EXTRACT, S_IRWXU|S_IRWXG|S_IRWXO);\r
	    statuswindow(45, 4, title, ctr[TR_INSTALLING_FILES]);\r
	    switch (radio) {\r
	    case 4:	// network\r
		    sprintf(commandstring,"/bin/wget -P " TMP_EXTRACT " %s/%s.dat", WGET, filevalue);\r
		    mysystem (commandstring);\r
		    sprintf(commandstring,"/bin/wget -P " TMP_EXTRACT " %s/%s.key", WGET, filevalue);\r
		    if (mysystem (commandstring)) {\r
	    		    errorbox(ctr[TR_FAILED_TO_FIND]);\r
			    break;\r
		    };\r
		    goto COMMON;\r
	    case 3:	// normal backup\r
		    if (mountbackup( filevalue, DEST_DEV )) {\r
	    		    errorbox(ctr[TR_UNABLE_TO_INSTALL_FILES]);//mess=no device with backup found\r
			    break;\r
		    };\r
		    // link files to a COMMON location\r
		    sprintf (commandstring, "chroot /harddisk ln -s "MOUNT_BACKUP_CH"/%s.dat " TMP_EXTRACT_CH "/%s.dat", filevalue, filevalue);\r
		    mysystem (commandstring);\r
		    sprintf (commandstring, "chroot /harddisk ln -s "MOUNT_BACKUP_CH"/%s.key " TMP_EXTRACT_CH "/%s.key", filevalue, filevalue);\r
		    mysystem (commandstring);\r
\r
	COMMON:	    // DECRYPT THE TARBALL\r
		    // Copy the key to a new location because we decrypt it!	\r
		    if (strcmp(keyvalue, "")) { // password provided: decrypt the key\r
    			    sprintf(commandstring,   "/bin/chroot /harddisk /usr/bin/openssl enc"\r
						    " -a -d -aes256 -salt"\r
						    " -pass pass:%s"\r
						    " -in " TMP_EXTRACT_CH "/%s.key"\r
						    " -out " TMP_EXTRACT_CH "/__tmp.key",\r
						    keyvalue, filevalue);\r
		    } else {			//just copy to new name\r
    			    sprintf(commandstring,  "/bin/chroot /harddisk cp"\r
						    " " TMP_EXTRACT_CH "/%s.key"\r
						    " " TMP_EXTRACT_CH "/__tmp.key",\r
						    filevalue);\r
		    }\r
		    mysystem (commandstring);\r
\r
		    sprintf(commandstring,  "/bin/chroot /harddisk /usr/bin/openssl des3"\r
					    " -d -salt"\r
					    " -in " TMP_EXTRACT_CH "/%s.dat"\r
					    " -out " TMP_EXTRACT_CH "/backup.tgz"\r
					    " -kfile " TMP_EXTRACT_CH "/__tmp.key",\r
					    filevalue);\r
\r
		    if (mysystem (commandstring)) {\r
	    		    errorbox(ctr[TR_UNABLE_TO_INSTALL_FILES]);//mess=decrypt error:invalid key?\r
			    break;\r
		    }\r
		    strcpy(commandstring,   "/bin/chroot /harddisk /bin/tar"\r
					    " -X " CONFIG_ROOT "/backup/exclude.system"\r
					    " -C " TMP_EXTRACT_CH\r
					    " -xzf " TMP_EXTRACT_CH "/backup.tgz");\r
\r
		    if (mysystem(commandstring)) {\r
	    		    errorbox(ctr[TR_UNABLE_TO_INSTALL_FILES]);\r
			    break;\r
		    }\r
		    sprintf(commandstring, TMP_EXTRACT "/%s.dat", filevalue);\r
	    	    unlink(commandstring ); //dont need them anymore\r
	    	    unlink( TMP_EXTRACT "/backup.tgz");\r
		    sprintf(commandstring, TMP_EXTRACT "/%s.key", filevalue);\r
	    	    unlink(commandstring );\r
	    	    unlink( TMP_EXTRACT "/__tmp.key");\r
\r
	      	    /* Now copy to correct location */\r
		    do_copy_files(0);\r
		    retcode = 0; /* successfully restored */\r
		    break;\r
	    case 2:\r
		    // diskette change\r
		    if (floppy_locate()) {\r
			    retcode = 2; // this an error!\r
			    break;\r
		    }\r
\r
		    /* Always extract to /tmp/ipcop for temporary extraction\r
	    	       just in case floppy fails.\r
		       try a compressed backup first because it's quicker to fail.\r
	    	       In exclude.system, files name must be without leading / or \r
	    	       on extraction, name will never match\r
		    */\r
		    sprintf(commandstring,\r
		     "/bin/chroot /harddisk /bin/tar -X " CONFIG_ROOT "/backup/exclude.system -C "TMP_EXTRACT_CH" -xvzf /dev/floppy > %s 2> /dev/null", mylog);\r
		    if (system(commandstring)) {\r
	    	    /* if it's not compressed, try uncompressed first before failing*/\r
	    		sprintf(commandstring,\r
		     "/bin/chroot /harddisk /bin/tar -X " CONFIG_ROOT "/backup/exclude.system -C "TMP_EXTRACT_CH" -xvf /dev/floppy > %s 2> /dev/null", mylog);\r
	    		if (system(commandstring)) {\r
	    		    /* command failed trying to read from floppy */\r
	    		    errorbox(ctr[TR_UNABLE_TO_INSTALL_FILES]);\r
			    break;\r
	    		} \r
	    	    }\r
	      	    /* Now copy to correct location */\r
		    do_copy_files(1);\r
		    retcode = 0; /* successfully restored */\r
	    }//switch\r
	    /* remove possible badly restored files */\r
	    mysystem("/bin/chroot /harddisk /bin/rm -rf " TMP_EXTRACT_CH );\r
	    newtPopWindow(); // close windows\r
	}//while\r
        newtFormDestroy(form);\r
\r
	/* cleanup */\r
	mysystem("/bin/umount " MOUNT_BACKUP);\r
	mysystem("/bin/chroot /harddisk /bin/rmdir " MOUNT_BACKUP_CH);\r
\r
	/* others operations moved from install to install2 */\r
	fixup_squidacl();\r
	fixup_initrd();\r
\r
	fprintf(flog, "Install2 program ended.\n");\r
	fflush(flog);\r
	fclose(flog);\r
	newtFinished();\r
	return retcode;\r
}\r
\r