src/installer/main.c

   1 /* SmoothWall install program.
   2  *
   3  * This program is distributed under the terms of the GNU General Public
   4  * Licence.  See the file COPYING for details.
   5  *
   6  * (c) Lawrence Manning, 2001
   7  * Contains main entry point, and misc functions.6
   8  *
   9  */
  10 #define _GNU_SOURCE
  11
  12 #include <assert.h>
  13 #include <errno.h>
  14 #include <libsmooth.h>
  15 #include <stdio.h>
  16 #include <stdlib.h>
  17 #include <string.h>
  18 #include <sys/mount.h>
  19
  20 #include "hw.h"
  21
  22 // Translation
  23 #include <libintl.h>
  24 #define _(x) dgettext("installer", x)
  25
  26 #define INST_FILECOUNT 24800
  27 #define LICENSE_FILE    "/cdrom/COPYING"
  28 #define SOURCE_TEMPFILE "/tmp/downloads/image.iso"
  29
  30 extern char url[STRING_SIZE];
  31
  32 static int newtChecklist(const char* title, const char* message,
  33                 unsigned int width, unsigned int height, unsigned int num_entries,
  34                 const char** entries, int* states) {
  35         int ret;
  36         const int list_height = 4;
  37
  38         char cbstates[num_entries];
  39
  40         for (unsigned int i = 0; i < num_entries; i++) {
  41                 cbstates[i] = states[i] ? '*' : ' ';
  42         }
  43
  44         newtCenteredWindow(width, height, title);
  45
  46         newtComponent textbox = newtTextbox(1, 1, width - 2, height - 6 - list_height,
  47                 NEWT_FLAG_WRAP);
  48         newtTextboxSetText(textbox, message);
  49
  50         int top = newtTextboxGetNumLines(textbox) + 2;
  51
  52         newtComponent form = newtForm(NULL, NULL, 0);
  53
  54         newtComponent sb = NULL;
  55         if (list_height < num_entries) {
  56                 sb = newtVerticalScrollbar(
  57                         width - 4, top + 1, list_height,
  58                         NEWT_COLORSET_CHECKBOX, NEWT_COLORSET_ACTCHECKBOX);
  59
  60                 newtFormAddComponent(form, sb);
  61         }
  62
  63         newtComponent subform = newtForm(sb, NULL, 0);
  64         newtFormSetBackground(subform, NEWT_COLORSET_CHECKBOX);
  65
  66         newtFormSetHeight(subform, list_height);
  67         newtFormSetWidth(subform, width - 10);
  68
  69         for (unsigned int i = 0; i < num_entries; i++) {
  70                 newtComponent cb = newtCheckbox(4, top + i, entries[i], cbstates[i],
  71                         NULL, &cbstates[i]);
  72
  73                 newtFormAddComponent(subform, cb);
  74         }
  75
  76         newtFormAddComponents(form, textbox, subform, NULL);
  77
  78         newtComponent btn_okay   = newtButton((width - 18) / 3, height - 4, _("OK"));
  79         newtComponent btn_cancel = newtButton((width - 18) / 3 * 2 + 9, height - 4, _("Cancel"));
  80         newtFormAddComponents(form, btn_okay, btn_cancel, NULL);
  81
  82         newtComponent answer = newtRunForm(form);
  83
  84         if ((answer == NULL) || (answer == btn_cancel)) {
  85                 ret = -1;
  86         } else {
  87                 ret = 0;
  88
  89                 for (unsigned int i = 0; i < num_entries; i++) {
  90                         states[i] = (cbstates[i] != ' ');
  91
  92                         if (states[i])
  93                                 ret++;
  94                 }
  95         }
  96
  97         newtFormDestroy(form);
  98         newtPopWindow();
  99
 100         return ret;
 101 }
 102
 103 static int newtWinOkCancel(const char* title, const char* message, int width, int height,
 104                 const char* btn_txt_ok, const char* btn_txt_cancel) {
 105         int ret = 1;
 106
 107         unsigned int btn_width_ok = strlen(btn_txt_ok);
 108         unsigned int btn_width_cancel = strlen(btn_txt_cancel);
 109
 110         // Maybe make the box wider to fix both buttons inside
 111         unsigned int min_width = btn_width_ok + btn_width_cancel + 5;
 112         if (width < min_width)
 113                 width = min_width;
 114
 115         unsigned int btn_pos_ok = (width / 3) - (btn_width_ok / 2) - 1;
 116         unsigned int btn_pos_cancel = (width * 2 / 3) - (btn_width_cancel / 2) - 1;
 117
 118         // Move buttons a bit if they overlap
 119         while ((btn_pos_ok + btn_width_ok + 5) > btn_pos_cancel) {
 120                 // Move the cancel button to the right if there is enough space left
 121                 if ((btn_pos_cancel + btn_width_cancel + 2) < width) {
 122                         ++btn_pos_cancel;
 123                         continue;
 124                 }
 125
 126                 // Move the OK button to the left if possible
 127                 if (btn_pos_ok > 1) {
 128                         --btn_pos_ok;
 129                         continue;
 130                 }
 131
 132                 // If they still overlap, we cannot fix the situtation
 133                 // and break. Should actually never get here, because we
 134                 // adjust the width of the window earlier.
 135                 break;
 136         }
 137
 138         newtCenteredWindow(width, height, title);
 139
 140         newtComponent form = newtForm(NULL, NULL, 0);
 141
 142         newtComponent textbox = newtTextbox(1, 1, width - 2, height - 6, NEWT_FLAG_WRAP);
 143         newtTextboxSetText(textbox, message);
 144         newtFormAddComponent(form, textbox);
 145
 146         newtComponent btn_ok = newtButton(btn_pos_ok, height - 4, btn_txt_ok);
 147         newtComponent btn_cancel = newtButton(btn_pos_cancel, height - 4, btn_txt_cancel);
 148
 149         newtFormAddComponents(form, btn_ok, btn_cancel, NULL);
 150
 151         newtComponent answer = newtRunForm(form);
 152
 153         if (answer == btn_ok) {
 154                 ret = 0;
 155         }
 156
 157         newtFormDestroy(form);
 158         newtPopWindow();
 159
 160         return ret;
 161 }
 162
 163 static int newtLicenseBox(const char* title, const char* text, int width, int height) {
 164         int ret = 1;
 165
 166         newtCenteredWindow(width, height, title);
 167
 168         newtComponent form = newtForm(NULL, NULL, 0);
 169
 170         newtComponent textbox = newtTextbox(1, 1, width - 2, height - 7,
 171                 NEWT_FLAG_WRAP|NEWT_FLAG_SCROLL);
 172         newtTextboxSetText(textbox, text);
 173         newtFormAddComponent(form, textbox);
 174
 175         char choice;
 176         newtComponent checkbox = newtCheckbox(3, height - 3, _("I accept this license"),
 177                 ' ', " *", &choice);
 178
 179         newtComponent btn = newtButton(width - 15, height - 4, _("OK"));
 180
 181         newtFormAddComponents(form, checkbox, btn, NULL);
 182
 183         newtComponent answer = newtRunForm(form);
 184         if (answer == btn && choice == '*')
 185                 ret = 0;
 186
 187         newtFormDestroy(form);
 188         newtPopWindow();
 189
 190         return ret;
 191 }
 192
 193 int write_lang_configs(char* lang) {
 194         struct keyvalue *kv = initkeyvalues();
 195
 196         /* default stuff for main/settings. */
 197         replacekeyvalue(kv, "LANGUAGE", lang);
 198         replacekeyvalue(kv, "HOSTNAME", DISTRO_SNAME);
 199         replacekeyvalue(kv, "THEME", "ipfire");
 200         writekeyvalues(kv, "/harddisk" CONFIG_ROOT "/main/settings");
 201         freekeyvalues(kv);
 202
 203         return 1;
 204 }
 205
 206 static char* get_system_release() {
 207         char system_release[STRING_SIZE] = "\0";
 208
 209         FILE* f = fopen("/etc/system-release", "r");
 210         if (f) {
 211                 fgets(system_release, sizeof(system_release), f);
 212                 fclose(f);
 213         }
 214
 215         return strdup(system_release);
 216 }
 217
 218 static char* center_string(const char* str, int width) {
 219         if (!str)
 220                 return NULL;
 221
 222         char* string = NULL;
 223         unsigned int str_len = strlen(str);
 224
 225         if (str_len == width) {
 226                 string = strdup(str);
 227
 228         } else if (str_len > width) {
 229                 string = strdup(str);
 230                 string[width - 1] = '\0';
 231
 232         } else {
 233                 unsigned int indent_length = (width - str_len) / 2;
 234                 char indent[indent_length + 1];
 235
 236                 for (unsigned int i = 0; i < indent_length; i++) {
 237                         indent[i] = ' ';
 238                 }
 239                 indent[indent_length] = '\0';
 240
 241                 if (asprintf(&string, "%s%s", indent, str) < 0)
 242                         return NULL;
 243         }
 244
 245         return string;
 246 }
 247
 248 #define DEFAULT_LANG "en.utf8"
 249 #define NUM_LANGS 13
 250
 251 static struct lang {
 252         const char* code;
 253         char* name;
 254 } languages[NUM_LANGS + 1] = {
 255         { "fa.utf8",    "فارسی (Persian)" },
 256         { "da.utf8",    "Dansk (Danish)" },
 257         { "es.utf8",    "Español (Spanish)" },
 258         { "en.utf8",    "English" },
 259         { "fr.utf8",    "Français (French)" },
 260         { "hr.utf8",    "Hrvatski (Croatian)" },
 261         { "it.utf8",    "Italiano (Italian)" },
 262         { "de.utf8",    "Deutsch (German)" },
 263         { "nl.utf8",    "Nederlands (Dutch)" },
 264         { "pl.utf8",    "Polski (Polish)" },
 265         { "pt.utf8",    "Portuguese (Brasil)" },
 266         { "ru.utf8",    "Русский (Russian)" },
 267         { "tr.utf8",    "Türkçe (Turkish)" },
 268         { NULL, NULL },
 269 };
 270
 271 static struct config {
 272         int unattended;
 273         int serial_console;
 274         int novga;
 275         int require_networking;
 276         int perform_download;
 277         int disable_swap;
 278         char download_url[STRING_SIZE];
 279         char postinstall[STRING_SIZE];
 280         char* language;
 281 } config = {
 282         .unattended = 0,
 283         .serial_console = 0,
 284         .novga = 0,
 285         .require_networking = 0,
 286         .perform_download = 0,
 287         .disable_swap = 0,
 288         .download_url = DOWNLOAD_URL,
 289         .postinstall = "\0",
 290         .language = DEFAULT_LANG,
 291 };
 292
 293 static void parse_command_line(struct config* c) {
 294         char buffer[STRING_SIZE];
 295         char cmdline[STRING_SIZE];
 296
 297         FILE* f = fopen("/proc/cmdline", "r");
 298         if (!f)
 299                 return;
 300
 301         int r = fread(&cmdline, 1, sizeof(cmdline) - 1, f);
 302         if (r > 0) {
 303                 char* token = strtok(cmdline, " ");
 304
 305                 while (token) {
 306                         strncpy(buffer, token, sizeof(buffer));
 307                         char* val = buffer;
 308                         char* key = strsep(&val, "=");
 309
 310                         // serial console
 311                         if ((strcmp(key, "console") == 0) && (strncmp(val, "ttyS", 4) == 0))
 312                                 c->serial_console = 1;
 313
 314                         // novga
 315                         else if (strcmp(key, "novga") == 0)
 316                                 c->novga = 1;
 317
 318                         // enable networking?
 319                         else if (strcmp(token, "installer.net") == 0)
 320                                 c->require_networking = 1;
 321
 322                         // unattended mode
 323                         else if (strcmp(token, "installer.unattended") == 0)
 324                                 c->unattended = 1;
 325
 326                         // disable swap
 327                         else if (strcmp(token, "installer.disable-swap") == 0)
 328                                 c->disable_swap = 1;
 329
 330                         // download url
 331                         else if (strcmp(key, "installer.download-url") == 0) {
 332                                 strncpy(c->download_url, val, sizeof(c->download_url));
 333                                 c->perform_download = 1;
 334
 335                                 // Require networking for the download
 336                                 c->require_networking = 1;
 337
 338                         // postinstall script
 339                         } else if (strcmp(key, "installer.postinstall") == 0) {
 340                                 strncpy(c->postinstall, val, sizeof(c->postinstall));
 341
 342                                 // Require networking for the download
 343                                 c->require_networking = 1;
 344                         }
 345
 346                         token = strtok(NULL, " ");
 347                 }
 348         }
 349
 350         fclose(f);
 351 }
 352
 353 int main(int argc, char *argv[]) {
 354         struct hw* hw = hw_init();
 355         const char* logfile = NULL;
 356
 357         // Read /etc/system-release
 358         char* system_release = get_system_release();
 359
 360         char discl_msg[40000] = "Disclaimer\n";
 361
 362         char* sourcedrive = NULL;
 363         int rc = 0;
 364         char commandstring[STRING_SIZE];
 365         int choice;
 366         char message[STRING_SIZE];
 367         char title[STRING_SIZE];
 368         int allok = 0;
 369         FILE *copying;
 370
 371         setlocale(LC_ALL, "");
 372         sethostname(DISTRO_SNAME, 10);
 373
 374         /* Log file/terminal stuff. */
 375         FILE* flog = NULL;
 376         if (argc >= 2) {
 377                 logfile = argv[1];
 378
 379                 if (!(flog = fopen(logfile, "w+")))
 380                         return 0;
 381         } else {
 382                 return 0;
 383         }
 384
 385         fprintf(flog, "Install program started.\n");
 386         if (hw->efi)
 387                 fprintf(flog, "EFI mode enabled\n");
 388
 389         newtInit();
 390         newtCls();
 391
 392         // Determine the size of the screen
 393         int screen_cols = 0;
 394         int screen_rows = 0;
 395
 396         newtGetScreenSize(&screen_cols, &screen_rows);
 397
 398         // Draw title
 399         char* roottext = center_string(system_release, screen_cols);
 400         if (roottext)
 401                 newtDrawRootText(0, 0, roottext);
 402
 403         snprintf(title, sizeof(title), "%s - %s", DISTRO_NAME, DISTRO_SLOGAN);
 404
 405         // Parse parameters from the kernel command line
 406         parse_command_line(&config);
 407
 408         if (config.unattended) {
 409                 splashWindow(title, _("Warning: Unattended installation will start in 10 seconds..."), 10);
 410         }
 411
 412         // Load common modules
 413         mysystem(logfile, "/sbin/modprobe vfat"); // USB key
 414         hw_stop_all_raid_arrays(logfile);
 415
 416         if (!config.unattended) {
 417                 // Language selection
 418                 char* langnames[NUM_LANGS + 1];
 419
 420                 for (unsigned int i = 0; i < NUM_LANGS; i++) {
 421                         if (strcmp(languages[i].code, DEFAULT_LANG) == 0)
 422                                 choice = i;
 423
 424                         langnames[i] = languages[i].name;
 425                 }
 426                 langnames[NUM_LANGS] = NULL;
 427
 428                 rc = newtWinMenu(_("Language selection"), _("Select the language you wish to use for the installation."),
 429                         50, 5, 5, 8, langnames, &choice, _("OK"), NULL);
 430
 431                 assert(choice <= NUM_LANGS);
 432
 433                 fprintf(flog, "Selected language: %s (%s)\n", languages[choice].name, languages[choice].code);
 434                 config.language = languages[choice].code;
 435
 436                 setlocale(LC_ALL, config.language);
 437                 setenv("LANGUAGE", config.language, 1);
 438         }
 439
 440         // Set helpline
 441         char* helpline = NULL;
 442         if (config.unattended)
 443                 helpline = center_string(_("Unattended mode"), screen_cols);
 444         else
 445                 helpline = center_string(_("<Tab>/<Alt-Tab> between elements | <Space> selects | <F12> next screen"), screen_cols);
 446
 447         if (helpline)
 448                 newtPushHelpLine(helpline);
 449
 450         if (!config.unattended) {
 451                 snprintf(message, sizeof(message),
 452                         _("Welcome to the %s installation program.\n\n"
 453                         "Selecting Cancel on any of the following screens will reboot the computer."), DISTRO_NAME);
 454                 newtWinMessage(title, _("Start installation"), message);
 455         }
 456
 457         /* Search for a source drive that holds the right
 458          * version of the image we are going to install. */
 459         if (!config.perform_download) {
 460                 sourcedrive = hw_find_source_medium(hw);
 461                 fprintf(flog, "Source drive: %s\n", sourcedrive);
 462         }
 463
 464         /* If we could not find a source drive, we will try
 465          * downloading the install image */
 466         if (!sourcedrive)
 467                 config.perform_download = 1;
 468
 469         if (config.perform_download) {
 470                 if (!config.unattended) {
 471                         // Show the right message to the user
 472                         char reason[STRING_SIZE];
 473                         if (config.perform_download) {
 474                                 snprintf(reason, sizeof(reason),
 475                                         _("The installer will now try downloading the installation image."));
 476                         } else {
 477                                 snprintf(reason, sizeof(reason),
 478                                         _("No source drive could be found.\n\n"
 479                                         "You can try downloading the required installation image."));
 480                         }
 481                         snprintf(message, sizeof(message), "%s %s", reason,
 482                                 _("Please make sure to connect your machine to a network and "
 483                                 "the installer will try connect to acquire an IP address."));
 484
 485                         rc = newtWinOkCancel(title, message, 55, 12,
 486                                 _("Download installation image"), _("Cancel"));
 487
 488                         if (rc != 0)
 489                                 goto EXIT;
 490                 }
 491
 492                 // Make sure that we enable networking before download
 493                 config.require_networking = 1;
 494         }
 495
 496         // Try starting the networking if we require it
 497         if (config.require_networking) {
 498                 while (1) {
 499                         statuswindow(60, 4, title, _("Trying to start networking (DHCP)..."));
 500
 501                         rc = hw_start_networking(logfile);
 502                         newtPopWindow();
 503
 504                         // Networking was successfully started
 505                         if (rc == 0) {
 506                                 break;
 507
 508                         // An error happened, ask the user what to do
 509                         } else {
 510                                 rc = newtWinOkCancel(title, _("Networking could not be started "
 511                                         "but is required to go on with the installation.\n\n"
 512                                         "Please connect your machine to a network with a "
 513                                         "DHCP server and retry."), 50, 10, _("Retry"), _("Cancel"));
 514
 515                                 if (rc)
 516                                         goto EXIT;
 517                         }
 518                 }
 519
 520                 // Download the image if required
 521                 if (config.perform_download) {
 522                         fprintf(flog, "Download URL: %s\n", config.download_url);
 523                         snprintf(commandstring, sizeof(commandstring), "/usr/bin/downloadsource.sh %s %s",
 524                                 SOURCE_TEMPFILE, config.download_url);
 525
 526                         while (!sourcedrive) {
 527                                 rc = runcommandwithstatus(commandstring, title, _("Downloading installation image..."), logfile);
 528
 529                                 FILE* f = fopen(SOURCE_TEMPFILE, "r");
 530                                 if (f) {
 531                                         sourcedrive = strdup(SOURCE_TEMPFILE);
 532                                         fclose(f);
 533                                 } else {
 534                                         char reason[STRING_SIZE] = "-";
 535                                         if (rc == 2)
 536                                                 snprintf(reason, sizeof(STRING_SIZE), _("MD5 checksum mismatch"));
 537
 538                                         snprintf(message, sizeof(message),
 539                                                 _("The installation image could not be downloaded.\n  Reason: %s\n\n%s"),
 540                                                 reason, config.download_url);
 541
 542                                         rc = newtWinOkCancel(title, message, 75, 12, _("Retry"), _("Cancel"));
 543                                         if (rc)
 544                                                 goto EXIT;
 545                                 }
 546                         }
 547                 }
 548         }
 549
 550         assert(sourcedrive);
 551
 552         int r = hw_mount(sourcedrive, SOURCE_MOUNT_PATH, "iso9660", MS_RDONLY);
 553         if (r) {
 554                 snprintf(message, sizeof(message), _("Could not mount %s to %s:\n  %s\n"),
 555                         sourcedrive, SOURCE_MOUNT_PATH, strerror(errno));
 556                 errorbox(message);
 557                 goto EXIT;
 558         }
 559
 560         if (!config.unattended) {
 561                 // Read the license file.
 562                 if (!(copying = fopen(LICENSE_FILE, "r"))) {
 563                         sprintf(discl_msg, "Could not open license file: %s\n", LICENSE_FILE);
 564                         fprintf(flog, "%s", discl_msg);
 565                 } else {
 566                         fread(discl_msg, 1, 40000, copying);
 567                         fclose(copying);
 568
 569                         if (newtLicenseBox(_("License Agreement"), discl_msg, 75, 20)) {
 570                                 errorbox(_("License not accepted!"));
 571
 572                                 goto EXIT;
 573                         }
 574                 }
 575         }
 576
 577         int part_type = HW_PART_TYPE_NORMAL;
 578
 579         // Scan for disks to install on.
 580         struct hw_disk** disks = hw_find_disks(hw, sourcedrive);
 581
 582         struct hw_disk** selected_disks = NULL;
 583         unsigned int num_selected_disks = 0;
 584
 585         // Check how many disks have been found and what
 586         // we can do with them.
 587         unsigned int num_disks = hw_count_disks((const struct hw_disk**)disks);
 588
 589         while (1) {
 590                 // no harddisks found
 591                 if (num_disks == 0) {
 592                         errorbox(_("No hard disk found."));
 593                         goto EXIT;
 594
 595                 // exactly one disk has been found
 596                 // or if we are running in unattended mode, we will select
 597                 // the first disk and go with that one
 598                 } else if ((num_disks == 1) || (config.unattended && num_disks >= 1)) {
 599                         selected_disks = hw_select_first_disk((const struct hw_disk**)disks);
 600
 601                 // more than one usable disk has been found and
 602                 // the user needs to choose what to do with them
 603                 } else {
 604                         const char* disk_names[num_disks];
 605                         int disk_selection[num_disks];
 606
 607                         for (unsigned int i = 0; i < num_disks; i++) {
 608                                 disk_names[i] = disks[i]->description;
 609                                 disk_selection[i] = 0;
 610                         }
 611
 612                         while (!selected_disks) {
 613                                 rc = newtChecklist(_("Disk Selection"),
 614                                         _("Select the disk(s) you want to install IPFire on. "
 615                                         "First those will be partitioned, and then the partitions will have a filesystem put on them.\n\n"
 616                                         "ALL DATA ON THE DISK WILL BE DESTROYED."),
 617                                         50, 20, num_disks, disk_names, disk_selection);
 618
 619                                 // Error
 620                                 if (rc < 0) {
 621                                         goto EXIT;
 622
 623                                 // Nothing has been selected
 624                                 } else if (rc == 0) {
 625                                         errorbox(_("No disk has been selected.\n\n"
 626                                                 "Please select one or more disks you want to install IPFire on."));
 627
 628                                 } else {
 629                                         selected_disks = hw_select_disks(disks, disk_selection);
 630                                 }
 631                         }
 632                 }
 633
 634                 // Don't print the auto-selected harddisk setup in
 635                 // unattended mode.
 636                 if (config.unattended)
 637                         break;
 638
 639                 num_selected_disks = hw_count_disks((const struct hw_disk**)selected_disks);
 640
 641                 if (num_selected_disks == 1) {
 642                         snprintf(message, sizeof(message),
 643                                 _("The installation program will now prepare the chosen harddisk:\n\n  %s\n\n"
 644                                 "Do you agree to continue?"), (*selected_disks)->description);
 645                         rc = newtWinOkCancel(_("Disk Setup"), message, 50, 10,
 646                                 _("Delete all data"), _("Cancel"));
 647
 648                         if (rc == 0)
 649                                 break;
 650
 651                 } else if (num_selected_disks == 2) {
 652                         snprintf(message, sizeof(message),
 653                                 _("The installation program will now set up a RAID configuration on the selected harddisks:\n\n  %s\n  %s\n\n"
 654                                 "Do you agree to continue?"), selected_disks[0]->description, selected_disks[1]->description);
 655                         rc = newtWinOkCancel(_("RAID Setup"), message, 50, 14,
 656                                 _("Delete all data"), _("Cancel"));
 657
 658                         if (rc == 0) {
 659                                 part_type = HW_PART_TYPE_RAID1;
 660
 661                                 break;
 662                         }
 663
 664                 // Currently not supported
 665                 } else {
 666                         errorbox(_("Your disk configuration is currently not supported."));
 667                         fprintf(flog, "Num disks selected: %d\n", num_selected_disks);
 668                 }
 669
 670                 if (selected_disks) {
 671                         hw_free_disks(selected_disks);
 672                         selected_disks = NULL;
 673                 }
 674         }
 675
 676         hw_free_disks(disks);
 677
 678         struct hw_destination* destination = hw_make_destination(hw, part_type,
 679                 selected_disks, config.disable_swap);
 680
 681         if (!destination) {
 682                 errorbox(_("Your harddisk is too small."));
 683                 goto EXIT;
 684         }
 685
 686         fprintf(flog, "Destination drive: %s\n", destination->path);
 687         fprintf(flog, "  bootldr: %s (%lluMB)\n", destination->part_bootldr, BYTES2MB(destination->size_bootldr));
 688         fprintf(flog, "  boot   : %s (%lluMB)\n", destination->part_boot, BYTES2MB(destination->size_boot));
 689         fprintf(flog, "  ESP    : %s (%lluMB)\n", destination->part_boot_efi, BYTES2MB(destination->size_boot_efi));
 690         fprintf(flog, "  swap   : %s (%lluMB)\n", destination->part_swap, BYTES2MB(destination->size_swap));
 691         fprintf(flog, "  root   : %s (%lluMB)\n", destination->part_root, BYTES2MB(destination->size_root));
 692         fprintf(flog, "Memory   : %lluMB\n", BYTES2MB(hw_memory()));
 693
 694         // Warn the user if there is not enough space to create a swap partition
 695         if (!config.unattended) {
 696                 if (!config.disable_swap && !*destination->part_swap) {
 697                         rc = newtWinChoice(title, _("OK"), _("Cancel"),
 698                                 _("Your harddisk is very small, but you can continue without a swap partition."));
 699
 700                         if (rc != 1)
 701                                 goto EXIT;
 702                 }
 703         }
 704
 705         // Filesystem selection
 706         if (!config.unattended) {
 707                 struct filesystems {
 708                         int fstype;
 709                         char* description;
 710                 } filesystems[] = {
 711                         { HW_FS_EXT4,            _("ext4 Filesystem") },
 712                         { HW_FS_EXT4_WO_JOURNAL, _("ext4 Filesystem without journal") },
 713                         { HW_FS_XFS,             _("XFS Filesystem") },
 714                         { HW_FS_REISERFS,        _("ReiserFS Filesystem") },
 715                         { 0, NULL },
 716                 };
 717                 unsigned int num_filesystems = sizeof(filesystems) / sizeof(*filesystems);
 718
 719                 char* fs_names[num_filesystems];
 720                 int fs_choice = 0;
 721                 for (unsigned int i = 0; i < num_filesystems; i++) {
 722                         if (HW_FS_DEFAULT == filesystems[i].fstype)
 723                                 fs_choice = i;
 724
 725                         fs_names[i] = filesystems[i].description;
 726                 }
 727
 728                 rc = newtWinMenu(_("Filesystem Selection"), _("Please choose your filesystem:"),
 729                         50, 5, 5, 6, fs_names, &fs_choice, _("OK"), _("Cancel"), NULL);
 730
 731                 if (rc == 2)
 732                         goto EXIT;
 733
 734                 destination->filesystem = filesystems[fs_choice].fstype;
 735         }
 736
 737         // Setting up RAID if needed.
 738         if (destination->is_raid) {
 739                 statuswindow(60, 4, title, _("Building RAID..."));
 740
 741                 rc = hw_setup_raid(destination, logfile);
 742                 if (rc) {
 743                         errorbox(_("Unable to build the RAID."));
 744                         goto EXIT;
 745                 }
 746
 747                 newtPopWindow();
 748         } else {
 749                 // We will have to destroy all RAID setups that may have
 750                 // been on the devices that we want to use now.
 751                 hw_destroy_raid_superblocks(destination, logfile);
 752         }
 753
 754         // Execute the partitioning...
 755         statuswindow(60, 4, title, _("Partitioning disk..."));
 756
 757         rc = hw_create_partitions(destination, logfile);
 758         if (rc) {
 759                 errorbox(_("Unable to partition the disk."));
 760                 goto EXIT;
 761         }
 762
 763         newtPopWindow();
 764
 765         // Execute the formatting...
 766         statuswindow(60, 4, title, _("Creating filesystems..."));
 767
 768         rc = hw_create_filesystems(destination, logfile);
 769         if (rc) {
 770                 errorbox(_("Unable to create filesystems."));
 771                 goto EXIT;
 772         }
 773
 774         rc = hw_mount_filesystems(destination, DESTINATION_MOUNT_PATH);
 775         if (rc) {
 776                 errorbox(_("Unable to mount filesystems."));
 777                 goto EXIT;
 778         }
 779
 780         newtPopWindow();
 781
 782         // Extract files...
 783         snprintf(commandstring, STRING_SIZE,
 784                 "/bin/tar -C /harddisk -xvf /cdrom/distro.img --xz 2>/dev/null");
 785
 786         if (runcommandwithprogress(60, 4, title, commandstring, INST_FILECOUNT,
 787                         _("Installing the system..."), logfile)) {
 788                 errorbox(_("Unable to install the system."));
 789                 goto EXIT;
 790         }
 791
 792         // Write fstab
 793         rc = hw_write_fstab(destination);
 794         if (rc) {
 795                 fprintf(flog, "Could not write /etc/fstab\n");
 796                 goto EXIT;
 797         }
 798
 799         /* Save language und local settings */
 800         write_lang_configs(config.language);
 801
 802         /* Build cache lang file */
 803         snprintf(commandstring, STRING_SIZE, "/usr/sbin/chroot /harddisk /usr/bin/perl -e \"require '" CONFIG_ROOT "/lang.pl'; &Lang::BuildCacheLang\"");
 804         if (runcommandwithstatus(commandstring, title, _("Installing the language cache..."), logfile)) {
 805                 errorbox(_("Unable to install the language cache."));
 806                 goto EXIT;
 807         }
 808
 809         // Installing bootloader...
 810         statuswindow(60, 4, title, _("Installing the bootloader..."));
 811
 812         /* Serial console ? */
 813         if (config.serial_console) {
 814                 /* grub */
 815                 FILE* f = fopen(DESTINATION_MOUNT_PATH "/etc/default/grub", "a");
 816                 if (!f) {
 817                         errorbox(_("Unable to open /etc/default/grub for writing."));
 818                         goto EXIT;
 819                 }
 820
 821                 fprintf(f, "GRUB_TERMINAL=\"serial\"\n");
 822                 fprintf(f, "GRUB_SERIAL_COMMAND=\"serial --unit=0 --speed=%d\"\n", SERIAL_BAUDRATE);
 823                 fclose(f);
 824
 825                 replace(DESTINATION_MOUNT_PATH "/etc/default/grub", "panic=10", "panic=10 console=ttyS0,115200n8");
 826         }
 827
 828         /* novga */
 829         if (config.novga) {
 830                 /* grub */
 831                 FILE* f = fopen(DESTINATION_MOUNT_PATH "/etc/default/grub", "a");
 832                 if (!f) {
 833                         errorbox(_("Unable to open /etc/default/grub for writing."));
 834                         goto EXIT;
 835                 }
 836
 837                 fprintf(f, "GRUB_GFXMODE=\"none\"\n");
 838                 fclose(f);
 839         }
 840
 841         rc = hw_install_bootloader(hw, destination, logfile);
 842         if (rc) {
 843                 errorbox(_("Unable to install the bootloader."));
 844                 goto EXIT;
 845         }
 846
 847         newtPopWindow();
 848
 849         /* Set marker that the user has already accepted the GPL if the license has been shown
 850          * in the installation process. In unatteded mode, the user will be presented the
 851          * license when he or she logs on to the web user interface for the first time. */
 852         if (!config.unattended)
 853                 mysystem(logfile, "/usr/bin/touch /harddisk/var/ipfire/main/gpl_accepted");
 854
 855         /* Copy restore file from cdrom */
 856         char* backup_file = hw_find_backup_file(logfile, SOURCE_MOUNT_PATH);
 857         if (backup_file) {
 858                 rc = 0;
 859                 if (!config.unattended) {
 860                         rc = newtWinOkCancel(title, _("A backup file has been found on the installation image.\n\n"
 861                                 "Do you want to restore the backup?"), 50, 10, _("Yes"), _("No"));
 862                 }
 863
 864                 if (rc == 0) {
 865                         rc = hw_restore_backup(logfile, backup_file, DESTINATION_MOUNT_PATH);
 866
 867                         if (rc) {
 868                                 errorbox(_("An error occured when the backup file was restored."));
 869                                 goto EXIT;
 870                         }
 871                 }
 872
 873                 free(backup_file);
 874         }
 875
 876         // Download and execute the postinstall script
 877         if (*config.postinstall) {
 878                 snprintf(commandstring, sizeof(commandstring),
 879                         "/usr/bin/execute-postinstall.sh %s %s", DESTINATION_MOUNT_PATH, config.postinstall);
 880
 881                 if (runcommandwithstatus(commandstring, title, _("Running post-install script..."), logfile)) {
 882                         errorbox(_("Post-install script failed."));
 883                         goto EXIT;
 884                 }
 885         }
 886
 887         // Umount the destination drive
 888         statuswindow(60, 4, title, _("Umounting filesystems..."));
 889
 890         rc = hw_umount_filesystems(destination, DESTINATION_MOUNT_PATH);
 891         if (rc) {
 892                 // Show an error message if filesystems could not be umounted properly
 893                 snprintf(message, sizeof(message),
 894                         _("Could not umount all filesystems successfully:\n\n  %s"), strerror(errno));
 895                 errorbox(message);
 896                 goto EXIT;
 897         }
 898
 899         // Umount source drive and eject
 900         hw_umount(SOURCE_MOUNT_PATH);
 901
 902         // Free downloaded ISO image
 903         if (strcmp(sourcedrive, SOURCE_TEMPFILE) == 0) {
 904                 rc = unlink(sourcedrive);
 905                 if (rc)
 906                         fprintf(flog, "Could not free downloaded ISO image: %s\n", sourcedrive);
 907
 908         // or eject real images
 909         } else {
 910                 snprintf(commandstring, STRING_SIZE, "/usr/bin/eject %s", sourcedrive);
 911                 mysystem(logfile, commandstring);
 912         }
 913         newtPopWindow();
 914
 915         // Stop the RAID array if we are using RAID
 916         if (destination->is_raid)
 917                 hw_stop_all_raid_arrays(logfile);
 918
 919         // Show a short message that the installation went well and
 920         // wait a moment so that all disk caches get flushed.
 921         if (config.unattended) {
 922                 splashWindow(title, _("Unattended installation has finished. The system will be shutting down in a moment..."), 5);
 923
 924         } else {
 925                 snprintf(message, sizeof(message), _(
 926                         "%s was successfully installed!\n\n"
 927                         "Please remove any installation mediums from this system and hit the reboot button. "
 928                         "Once the system has restarted you will be asked to setup networking and system passwords. "
 929                         "After that, you should point your web browser at https://%s:444 (or what ever you name "
 930                         "your %s) for the web configuration console."), DISTRO_NAME, DISTRO_SNAME, DISTRO_NAME);
 931                 newtWinMessage(_("Congratulations!"), _("Reboot"), message);
 932         }
 933
 934         allok = 1;
 935
 936 EXIT:
 937         fprintf(flog, "Install program ended.\n");
 938         fflush(flog);
 939         fclose(flog);
 940
 941         if (!allok)
 942                 newtWinMessage(title, _("OK"), _("Setup has failed. Press Ok to reboot."));
 943
 944         newtFinished();
 945
 946         // Free resources
 947         if (system_release)
 948                 free(system_release);
 949
 950         if (roottext)
 951                 free(roottext);
 952
 953         if (helpline)
 954                 free(helpline);
 955
 956         if (sourcedrive)
 957                 free(sourcedrive);
 958
 959         if (destination)
 960                 free(destination);
 961
 962         hw_stop_all_raid_arrays(logfile);
 963
 964         if (selected_disks)
 965                 hw_free_disks(selected_disks);
 966
 967         if (hw)
 968                 hw_free(hw);
 969
 970         fcloseall();
 971
 972         if (allok == 1)
 973                 return 0;
 974
 975         return 1;
 976 }