]> git.ipfire.org Git - ipfire-2.x.git/blob - src/installer/main.c
77c1d67d73021c493863796b23951a4e376fc65c
[ipfire-2.x.git] / 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
11 #include <assert.h>
12 #include <errno.h>
13 #include <stdio.h>
14 #include <stdlib.h>
15 #include <string.h>
16 #include <sys/mount.h>
17
18 #include "hw.h"
19 #include "install.h"
20
21 // Translation
22 #include <libintl.h>
23 #define _(x) dgettext("installer", x)
24
25 #define INST_FILECOUNT 21000
26 #define UNATTENDED_CONF "/cdrom/boot/unattended.conf"
27 #define LICENSE_FILE "/cdrom/COPYING"
28
29 FILE *flog = NULL;
30 char *mylog;
31
32 extern char url[STRING_SIZE];
33
34 static int newtChecklist(const char* title, const char* message,
35 unsigned int width, unsigned int height, unsigned int num_entries,
36 const char** entries, int* states) {
37 int ret;
38 const int list_height = 4;
39
40 char cbstates[num_entries];
41
42 for (unsigned int i = 0; i < num_entries; i++) {
43 cbstates[i] = states[i] ? '*' : ' ';
44 }
45
46 newtCenteredWindow(width, height, title);
47
48 newtComponent textbox = newtTextbox(1, 1, width - 2, height - 6 - list_height,
49 NEWT_FLAG_WRAP);
50 newtTextboxSetText(textbox, message);
51
52 int top = newtTextboxGetNumLines(textbox) + 2;
53
54 newtComponent form = newtForm(NULL, NULL, 0);
55
56 newtComponent sb = NULL;
57 if (list_height < num_entries) {
58 sb = newtVerticalScrollbar(
59 width - 4, top + 1, list_height,
60 NEWT_COLORSET_CHECKBOX, NEWT_COLORSET_ACTCHECKBOX);
61
62 newtFormAddComponent(form, sb);
63 }
64
65 newtComponent subform = newtForm(sb, NULL, 0);
66 newtFormSetBackground(subform, NEWT_COLORSET_CHECKBOX);
67
68 newtFormSetHeight(subform, list_height);
69 newtFormSetWidth(subform, width - 10);
70
71 for (unsigned int i = 0; i < num_entries; i++) {
72 newtComponent cb = newtCheckbox(4, top + i, entries[i], cbstates[i],
73 NULL, &cbstates[i]);
74
75 newtFormAddComponent(subform, cb);
76 }
77
78 newtFormAddComponents(form, textbox, subform, NULL);
79
80 newtComponent btn_okay = newtButton((width - 18) / 3, height - 4, _("OK"));
81 newtComponent btn_cancel = newtButton((width - 18) / 3 * 2 + 9, height - 4, _("Cancel"));
82 newtFormAddComponents(form, btn_okay, btn_cancel, NULL);
83
84 newtComponent answer = newtRunForm(form);
85
86 if ((answer == NULL) || (answer == btn_cancel)) {
87 ret = -1;
88 } else {
89 ret = 0;
90
91 for (unsigned int i = 0; i < num_entries; i++) {
92 states[i] = (cbstates[i] != ' ');
93
94 if (states[i])
95 ret++;
96 }
97 }
98
99 newtFormDestroy(form);
100 newtPopWindow();
101
102 return ret;
103 }
104
105 static int newtWinOkCancel(const char* title, const char* message, int width, int height,
106 const char* btn_txt_ok, const char* btn_txt_cancel) {
107 int ret = 1;
108
109 newtCenteredWindow(width, height, title);
110
111 newtComponent form = newtForm(NULL, NULL, 0);
112
113 newtComponent textbox = newtTextbox(1, 1, width - 2, height - 6, NEWT_FLAG_WRAP);
114 newtTextboxSetText(textbox, message);
115 newtFormAddComponent(form, textbox);
116
117 newtComponent btn_ok = newtButton((width - 16) / 3, height - 4, btn_txt_ok);
118 newtComponent btn_cancel = newtButton((width - 16) / 3 * 2 + 9, height - 4,
119 btn_txt_cancel);
120
121 newtFormAddComponents(form, btn_ok, btn_cancel, NULL);
122
123 newtComponent answer = newtRunForm(form);
124
125 if (answer == btn_ok) {
126 ret = 0;
127 }
128
129 newtFormDestroy(form);
130 newtPopWindow();
131
132 return ret;
133 }
134
135 static int newtLicenseBox(const char* title, const char* text, int width, int height) {
136 int ret = 1;
137
138 newtCenteredWindow(width, height, title);
139
140 newtComponent form = newtForm(NULL, NULL, 0);
141
142 newtComponent textbox = newtTextbox(1, 1, width - 2, height - 7,
143 NEWT_FLAG_WRAP|NEWT_FLAG_SCROLL);
144 newtTextboxSetText(textbox, text);
145 newtFormAddComponent(form, textbox);
146
147 char choice;
148 newtComponent checkbox = newtCheckbox(3, height - 3, _("I accept this license"),
149 ' ', " *", &choice);
150
151 newtComponent btn = newtButton(width - 15, height - 4, _("OK"));
152
153 newtFormAddComponents(form, checkbox, btn, NULL);
154
155 newtComponent answer = newtRunForm(form);
156 if (answer == btn && choice == '*')
157 ret = 0;
158
159 newtFormDestroy(form);
160 newtPopWindow();
161
162 return ret;
163 }
164
165 int write_lang_configs(const char *lang) {
166 struct keyvalue *kv = initkeyvalues();
167
168 /* default stuff for main/settings. */
169 replacekeyvalue(kv, "LANGUAGE", lang);
170 replacekeyvalue(kv, "HOSTNAME", SNAME);
171 replacekeyvalue(kv, "THEME", "ipfire");
172 writekeyvalues(kv, "/harddisk" CONFIG_ROOT "/main/settings");
173 freekeyvalues(kv);
174
175 return 1;
176 }
177
178 static char* get_system_release() {
179 char system_release[STRING_SIZE] = "\0";
180
181 FILE* f = fopen("/etc/system-release", "r");
182 if (f) {
183 fgets(system_release, sizeof(system_release), f);
184 fclose(f);
185 }
186
187 return strdup(system_release);
188 }
189
190 static char* center_string(const char* str, int width) {
191 unsigned int str_len = strlen(str);
192
193 unsigned int indent_length = (width - str_len) / 2;
194 char indent[indent_length + 1];
195
196 for (unsigned int i = 0; i < indent_length; i++) {
197 indent[i] = ' ';
198 }
199 indent[indent_length] = '\0';
200
201 char* string = NULL;
202 if (asprintf(&string, "%s%s", indent, str) < 0)
203 return NULL;
204
205 return string;
206 }
207
208 #define DEFAULT_LANG "English"
209 #define NUM_LANGS 8
210
211 static struct lang {
212 const char* code;
213 char* name;
214 } languages[NUM_LANGS + 1] = {
215 { "nl", "Dutch (Nederlands)" },
216 { "en", "English" },
217 { "fr", "French (Français)" },
218 { "de", "German (Deutsch)" },
219 { "pl", "Polish (Polski)" },
220 { "ru", "Russian (Русский)" },
221 { "es", "Spanish (Español)" },
222 { "tr", "Turkish (Türkçe)" },
223 { NULL, NULL },
224 };
225
226 int main(int argc, char *argv[]) {
227 struct hw* hw = hw_init();
228
229 // Read /etc/system-release
230 char* system_release = get_system_release();
231
232 char discl_msg[40000] = "Disclaimer\n";
233
234 char* sourcedrive = NULL;
235 int rc = 0;
236 char commandstring[STRING_SIZE];
237 int choice;
238 char shortlangname[10];
239 char message[STRING_SIZE];
240 char title[STRING_SIZE];
241 int allok = 0;
242 FILE *handle, *cmdfile, *copying;
243 char line[STRING_SIZE];
244
245 int unattended = 0;
246 int serialconsole = 0;
247 struct keyvalue *unattendedkv = initkeyvalues();
248 char restore_file[STRING_SIZE] = "";
249
250 setlocale (LC_ALL, "");
251 sethostname( SNAME , 10);
252
253 /* Log file/terminal stuff. */
254 if (argc >= 2)
255 {
256 if (!(flog = fopen(argv[1], "w+")))
257 return 0;
258 }
259 else
260 return 0;
261
262 mylog = argv[1];
263
264 fprintf(flog, "Install program started.\n");
265
266 newtInit();
267 newtCls();
268
269 // Determine the size of the screen
270 int screen_cols = 0;
271 int screen_rows = 0;
272
273 newtGetScreenSize(&screen_cols, &screen_rows);
274
275 // Draw title
276 char* roottext = center_string(system_release, screen_cols);
277 newtDrawRootText(0, 0, roottext);
278
279 sprintf (title, "%s %s - %s", NAME, VERSION, SLOGAN);
280
281 if (! (cmdfile = fopen("/proc/cmdline", "r")))
282 {
283 fprintf(flog, "Couldn't open commandline: /proc/cmdline\n");
284 } else {
285 fgets(line, STRING_SIZE, cmdfile);
286
287 // check if we have to make an unattended install
288 if (strstr (line, "unattended") != NULL) {
289 unattended = 1;
290 runcommandwithstatus("/bin/sleep 10", title, "WARNING: Unattended installation will start in 10 seconds...");
291 }
292 // check if we have to patch for serial console
293 if (strstr (line, "console=ttyS0") != NULL) {
294 serialconsole = 1;
295 }
296 }
297
298 // Load common modules
299 mysystem("/sbin/modprobe vfat"); // USB key
300 hw_stop_all_raid_arrays();
301
302 if (!unattended) {
303 // Language selection
304 char* langnames[NUM_LANGS + 1];
305
306 for (unsigned int i = 0; i < NUM_LANGS; i++) {
307 if (strcmp(languages[i].name, DEFAULT_LANG) == 0)
308 choice = i;
309
310 langnames[i] = languages[i].name;
311 }
312 langnames[NUM_LANGS] = NULL;
313
314 rc = newtWinMenu(_("Language selection"), _("Select the language you wish to use for the installation."),
315 50, 5, 5, 8, langnames, &choice, _("OK"), NULL);
316
317 assert(choice <= NUM_LANGS);
318
319 fprintf(flog, "Selected language: %s (%s)\n", languages[choice].name, languages[choice].code);
320 setlocale(LC_ALL, languages[choice].code);
321 }
322
323 char* helpline = center_string(_("<Tab>/<Alt-Tab> between elements | <Space> selects | <F12> next screen"), screen_cols);
324 newtPushHelpLine(helpline);
325
326 if (!unattended) {
327 snprintf(message, sizeof(message),
328 _("Welcome to the %s installation program. "
329 "Selecting Cancel on any of the following screens will reboot the computer."), NAME);
330 newtWinMessage(title, _("Start installation"), message);
331 }
332
333 /* Search for a source drive that holds the right
334 * version of the image we are going to install. */
335 sourcedrive = hw_find_source_medium(hw);
336
337 fprintf(flog, "Source drive: %s\n", sourcedrive);
338 if (!sourcedrive) {
339 newtWinMessage(title, _("OK"), _("No local source media found. Starting download."));
340 runcommandwithstatus("/bin/downloadsource.sh", title, _("Downloading installation image ..."));
341 if ((handle = fopen("/tmp/source_device", "r")) == NULL) {
342 errorbox(_("Download error"));
343 goto EXIT;
344 }
345
346 fgets(sourcedrive, 5, handle);
347 fclose(handle);
348 }
349
350 assert(sourcedrive);
351
352 int r = hw_mount(sourcedrive, SOURCE_MOUNT_PATH, "iso9660", MS_RDONLY);
353 if (r) {
354 fprintf(flog, "Could not mount %s to %s\n", sourcedrive, SOURCE_MOUNT_PATH);
355 fprintf(flog, strerror(errno));
356 exit(1);
357 }
358
359 /* load unattended configuration */
360 if (unattended) {
361 fprintf(flog, "unattended: Reading unattended.conf\n");
362
363 (void) readkeyvalues(unattendedkv, UNATTENDED_CONF);
364 findkey(unattendedkv, "RESTORE_FILE", restore_file);
365 }
366
367 if (!unattended) {
368 // Read the license file.
369 if (!(copying = fopen(LICENSE_FILE, "r"))) {
370 sprintf(discl_msg, "Could not open license file: %s\n", LICENSE_FILE);
371 fprintf(flog, discl_msg);
372 } else {
373 fread(discl_msg, 1, 40000, copying);
374 fclose(copying);
375
376 if (newtLicenseBox(title, discl_msg, 75, 20)) {
377 errorbox(_("License not accepted!"));
378
379 goto EXIT;
380 }
381 }
382 }
383
384 int part_type = HW_PART_TYPE_NORMAL;
385
386 // Scan for disks to install on.
387 struct hw_disk** disks = hw_find_disks(hw);
388
389 struct hw_disk** selected_disks = NULL;
390 unsigned int num_selected_disks = 0;
391
392 // Check how many disks have been found and what
393 // we can do with them.
394 unsigned int num_disks = hw_count_disks(disks);
395
396 while (1) {
397 // no harddisks found
398 if (num_disks == 0) {
399 errorbox(_("No hard disk found."));
400 goto EXIT;
401
402 // exactly one disk has been found
403 } else if (num_disks == 1) {
404 selected_disks = hw_select_disks(disks, NULL);
405
406 // more than one usable disk has been found and
407 // the user needs to choose what to do with them
408 } else {
409 const char* disk_names[num_disks];
410 int disk_selection[num_disks];
411
412 for (unsigned int i = 0; i < num_disks; i++) {
413 disk_names[i] = &disks[i]->description;
414 disk_selection[i] = 0;
415 }
416
417 while (!selected_disks) {
418 rc = newtChecklist(_("Disk Selection"),
419 _("Select the disk(s) you want to install IPFire on. "
420 "First those will be partitioned, and then the partitions will have a filesystem put on them.\n\n"
421 "ALL DATA ON THE DISK WILL BE DESTROYED."),
422 50, 20, num_disks, disk_names, disk_selection);
423
424 // Error
425 if (rc < 0) {
426 goto EXIT;
427
428 // Nothing has been selected
429 } else if (rc == 0) {
430 errorbox(_("No disk has been selected.\n\n"
431 "Please select one or more disks you want to install IPFire on."));
432
433 } else {
434 selected_disks = hw_select_disks(disks, disk_selection);
435 }
436 }
437 }
438
439 num_selected_disks = hw_count_disks(selected_disks);
440
441 if (num_selected_disks == 1) {
442 snprintf(message, sizeof(message),
443 _("The installation program will now prepare the chosen harddisk:\n\n %s\n\n"
444 "Do you agree to continue?"), (*selected_disks)->description);
445 rc = newtWinOkCancel(_("Disk Setup"), message, 50, 10,
446 _("Delete all data"), _("Cancel"));
447
448 if (rc == 0)
449 break;
450
451 } else if (num_selected_disks == 2) {
452 snprintf(message, sizeof(message),
453 _("The installation program will now set up a RAID configuration on the selected harddisks:\n\n %s\n %s\n\n"
454 "Do you agree to continue?"), (*selected_disks)->description, (*selected_disks + 1)->description);
455 rc = newtWinOkCancel(_("RAID Setup"), message, 50, 14,
456 _("Delete all data"), _("Cancel"));
457
458 if (rc == 0) {
459 part_type = HW_PART_TYPE_RAID1;
460
461 break;
462 }
463
464 // Currently not supported
465 } else {
466 errorbox(_("You disk configuration is currently not supported."));
467 }
468
469 if (selected_disks) {
470 hw_free_disks(selected_disks);
471 selected_disks = NULL;
472 }
473 }
474
475 hw_free_disks(disks);
476
477 struct hw_destination* destination = hw_make_destination(part_type, selected_disks);
478
479 if (!destination) {
480 errorbox(_("Your harddisk is too small."));
481 goto EXIT;
482 }
483
484 fprintf(flog, "Destination drive: %s\n", destination->path);
485 fprintf(flog, " bootldr: %s (%lluMB)\n", destination->part_bootldr, BYTES2MB(destination->size_bootldr));
486 fprintf(flog, " boot : %s (%lluMB)\n", destination->part_boot, BYTES2MB(destination->size_boot));
487 fprintf(flog, " swap : %s (%lluMB)\n", destination->part_swap, BYTES2MB(destination->size_swap));
488 fprintf(flog, " root : %s (%lluMB)\n", destination->part_root, BYTES2MB(destination->size_root));
489 fprintf(flog, " data : %s (%lluMB)\n", destination->part_data, BYTES2MB(destination->size_data));
490
491 // Warn the user if there is not enough space to create a swap partition
492 if (!unattended && !*destination->part_swap) {
493 rc = newtWinChoice(title, _("OK"), _("Cancel"),
494 _("Your harddisk is very small, but you can continue with an very small swap. (Use with caution)."));
495
496 if (rc != 1)
497 goto EXIT;
498 }
499
500 // Filesystem selection
501 if (!unattended) {
502 struct filesystems {
503 int fstype;
504 const char* description;
505 } filesystems[] = {
506 { HW_FS_EXT4, _("ext4 Filesystem") },
507 { HW_FS_EXT4_WO_JOURNAL, _("ext4 Filesystem without journal") },
508 { HW_FS_XFS, _("XFS Filesystem") },
509 { HW_FS_REISERFS, _("ReiserFS Filesystem") },
510 { 0, NULL },
511 };
512 unsigned int num_filesystems = sizeof(filesystems) / sizeof(*filesystems);
513
514 char* fs_names[num_filesystems];
515 int fs_choice = 0;
516 for (unsigned int i = 0; i < num_filesystems; i++) {
517 if (HW_FS_DEFAULT == filesystems[i].fstype)
518 fs_choice = i;
519
520 fs_names[i] = filesystems[i].description;
521 }
522
523 rc = newtWinMenu(_("Filesystem Selection"), _("Please choose your filesystem:"),
524 50, 5, 5, 6, fs_names, &fs_choice, _("OK"), _("Cancel"), NULL);
525
526 if (rc == 2)
527 goto EXIT;
528
529 destination->filesystem = filesystems[fs_choice].fstype;
530 }
531
532 // Setting up RAID if needed.
533 if (destination->is_raid) {
534 statuswindow(60, 4, title, _("Building RAID..."));
535
536 rc = hw_setup_raid(destination);
537 if (rc) {
538 errorbox(_("Unable to build the RAID."));
539 goto EXIT;
540 }
541
542 newtPopWindow();
543 }
544
545 // Execute the partitioning...
546 statuswindow(60, 4, title, _("Partitioning disk..."));
547
548 rc = hw_create_partitions(destination);
549 if (rc) {
550 errorbox(_("Unable to partition the disk."));
551 goto EXIT;
552 }
553
554 newtPopWindow();
555
556 // Execute the formatting...
557 statuswindow(60, 4, title, _("Creating filesystems..."));
558
559 rc = hw_create_filesystems(destination);
560 if (rc) {
561 errorbox(_("Unable to create filesystems."));
562 goto EXIT;
563 }
564
565 rc = hw_mount_filesystems(destination, DESTINATION_MOUNT_PATH);
566 if (rc) {
567 errorbox(_("Unable to mount filesystems."));
568 goto EXIT;
569 }
570
571 newtPopWindow();
572
573 // Extract files...
574 snprintf(commandstring, STRING_SIZE,
575 "/bin/tar -C /harddisk -xvf /cdrom/distro.img --lzma 2>/dev/null");
576
577 if (runcommandwithprogress(60, 4, title, commandstring, INST_FILECOUNT,
578 _("Installing the system..."))) {
579 errorbox(_("Unable to install the system."));
580 goto EXIT;
581 }
582
583 // Write fstab
584 rc = hw_write_fstab(destination);
585 if (rc) {
586 fprintf(flog, "Could not write /etc/fstab\n");
587 goto EXIT;
588 }
589
590 /* Save language und local settings */
591 write_lang_configs(shortlangname);
592
593 /* Build cache lang file */
594 snprintf(commandstring, STRING_SIZE, "/usr/sbin/chroot /harddisk /usr/bin/perl -e \"require '" CONFIG_ROOT "/lang.pl'; &Lang::BuildCacheLang\"");
595 if (runcommandwithstatus(commandstring, title, _("Installing the language cache..."))) {
596 errorbox(_("Unable to install the language cache."));
597 goto EXIT;
598 }
599
600 // Installing bootloader...
601 statuswindow(60, 4, title, _("Installing the bootloader..."));
602
603 rc = hw_install_bootloader(destination);
604 if (rc) {
605 errorbox(_("Unable to install the bootloader."));
606 goto EXIT;
607 }
608
609 newtPopWindow();
610
611 /* Serial console ? */
612 if (serialconsole) {
613 /* grub */
614 replace("/harddisk/boot/grub/grub.conf", "splashimage", "#splashimage");
615 replace("/harddisk/boot/grub/grub.conf", "#serial", "serial");
616 replace("/harddisk/boot/grub/grub.conf", "#terminal", "terminal");
617 replace("/harddisk/boot/grub/grub.conf", " panic=10 ", " console=ttyS0,115200n8 panic=10 ");
618
619 /*inittab*/
620 replace("/harddisk/etc/inittab", "1:2345:respawn:", "#1:2345:respawn:");
621 replace("/harddisk/etc/inittab", "2:2345:respawn:", "#2:2345:respawn:");
622 replace("/harddisk/etc/inittab", "3:2345:respawn:", "#3:2345:respawn:");
623 replace("/harddisk/etc/inittab", "4:2345:respawn:", "#4:2345:respawn:");
624 replace("/harddisk/etc/inittab", "5:2345:respawn:", "#5:2345:respawn:");
625 replace("/harddisk/etc/inittab", "6:2345:respawn:", "#6:2345:respawn:");
626 replace("/harddisk/etc/inittab", "#7:2345:respawn:", "7:2345:respawn:");
627 }
628
629 /* Set marker that the user has already accepted the gpl */
630 mysystem("/usr/bin/touch /harddisk/var/ipfire/main/gpl_accepted");
631
632 /* Copy restore file from cdrom */
633 if (unattended && (strlen(restore_file) > 0)) {
634 fprintf(flog, "unattended: Copy restore file\n");
635 snprintf(commandstring, STRING_SIZE,
636 "cp /cdrom/%s /harddisk/var/ipfire/backup", restore_file);
637 mysystem(commandstring);
638 }
639
640 // Umount source drive and eject
641 hw_umount(SOURCE_MOUNT_PATH);
642
643 snprintf(commandstring, STRING_SIZE, "/usr/bin/eject %s", sourcedrive);
644 mysystem(commandstring);
645
646 if (!unattended) {
647 snprintf(message, sizeof(message), _("%s was successfully installed. "
648 "Please remove any installation mediums from this system. "
649 "Setup will now run where you may configure networking and the system passwords. "
650 "After Setup has been completed, you should point your web browser at https://%s:444 "
651 "(or whatever you name your %s), and configure dialup networking (if required) and "
652 "remote access."), NAME, SNAME, NAME);
653 newtWinMessage(_("Congratulations!"), _("Reboot"), message);
654 }
655
656 allok = 1;
657
658 EXIT:
659 fprintf(flog, "Install program ended.\n");
660
661 if (!(allok))
662 newtWinMessage(title, _("OK"), _("Press Ok to reboot."));
663
664 if (allok) {
665 fflush(flog);
666 fclose(flog);
667 }
668
669 newtFinished();
670
671 // Free resources
672 free(system_release);
673 free(roottext);
674 free(helpline);
675
676 free(sourcedrive);
677
678 if (destination) {
679 hw_sync();
680
681 hw_umount_filesystems(destination, DESTINATION_MOUNT_PATH);
682 free(destination);
683 }
684
685 hw_stop_all_raid_arrays();
686
687 if (selected_disks)
688 hw_free_disks(selected_disks);
689
690 hw_free(hw);
691
692 fcloseall();
693
694 if (allok == 1)
695 return 0;
696
697 return 1;
698 }