Changes in uspace/app/sysinst/sysinst.c [9b95b964:11782da] in mainline
- File:
-
- 1 edited
Legend:
- Unmodified
- Added
- Removed
-
uspace/app/sysinst/sysinst.c
r9b95b964 r11782da 1 1 /* 2 * Copyright (c) 202 4Jiri Svoboda2 * Copyright (c) 2025 Jiri Svoboda 3 3 * All rights reserved. 4 4 * … … 41 41 #include <errno.h> 42 42 #include <fdisk.h> 43 #include <futil.h> 44 #include <gfx/render.h> 45 #include <io/log.h> 43 46 #include <loc.h> 44 47 #include <stdio.h> … … 46 49 #include <str.h> 47 50 #include <str_error.h> 51 #include <system.h> 52 #include <ui/msgdialog.h> 53 #include <ui/ui.h> 54 #include <ui/window.h> 48 55 #include <vfs/vfs.h> 49 56 #include <vol.h> 50 57 51 #include "futil.h"52 58 #include "grub.h" 53 59 #include "rdimg.h" 54 60 #include "volume.h" 61 #include "sysinst.h" 62 63 #define NAME "sysinst" 55 64 56 65 /** Device to install to … … 63 72 */ 64 73 #define DEFAULT_DEV_0 "devices/\\hw\\sys\\00:01.1\\c0d0" 65 #define DEFAULT_DEV_1 "devices/\\hw\\sys\\ 00:01.0\\ata1\\c0d0"74 #define DEFAULT_DEV_1 "devices/\\hw\\sys\\ide1\\c0d0" 66 75 //#define DEFAULT_DEV "devices/\\hw\\pci0\\00:01.2\\uhci_rh\\usb01_a1\\mass-storage0\\l0" 67 76 /** Volume label for the new file system */ … … 80 89 #define BOOT_BLOCK_IDX 0 /* MBR */ 81 90 91 #define CFG_FILES_SRC "/cfg" 92 #define CFG_FILES_DEST MOUNT_POINT "/cfg" 93 82 94 static const char *default_devs[] = { 83 95 DEFAULT_DEV_0, … … 92 104 }; 93 105 106 static fibril_mutex_t shutdown_lock; 107 static fibril_condvar_t shutdown_cv; 108 static bool shutdown_stopped; 109 static bool shutdown_failed; 110 111 static void sysinst_shutdown_complete(void *); 112 static void sysinst_shutdown_failed(void *); 113 114 static system_cb_t sysinst_system_cb = { 115 .shutdown_complete = sysinst_shutdown_complete, 116 .shutdown_failed = sysinst_shutdown_failed 117 }; 118 119 static void wnd_close(ui_window_t *, void *); 120 static errno_t bg_wnd_paint(ui_window_t *, void *); 121 122 static ui_window_cb_t bg_window_cb = { 123 .close = wnd_close, 124 .paint = bg_wnd_paint 125 }; 126 127 static ui_window_cb_t progress_window_cb = { 128 .close = wnd_close 129 }; 130 131 static void sysinst_confirm_button(ui_msg_dialog_t *, void *, unsigned); 132 static void sysinst_confirm_close(ui_msg_dialog_t *, void *); 133 134 static ui_msg_dialog_cb_t sysinst_confirm_cb = { 135 .button = sysinst_confirm_button, 136 .close = sysinst_confirm_close 137 }; 138 139 static errno_t sysinst_restart_dlg_create(sysinst_t *); 140 static void sysinst_restart_dlg_button(ui_msg_dialog_t *, void *, unsigned); 141 static void sysinst_restart_dlg_close(ui_msg_dialog_t *, void *); 142 143 static ui_msg_dialog_cb_t sysinst_restart_dlg_cb = { 144 .button = sysinst_restart_dlg_button, 145 .close = sysinst_restart_dlg_close 146 }; 147 148 static int sysinst_start(sysinst_t *); 149 static errno_t sysinst_restart(sysinst_t *); 150 static void sysinst_progress_destroy(sysinst_progress_t *); 151 static void sysinst_action(sysinst_t *, const char *); 152 static void sysinst_error(sysinst_t *, const char *); 153 static void sysinst_debug(sysinst_t *, const char *); 154 155 static void sysinst_futil_copy_file(void *, const char *, const char *); 156 static void sysinst_futil_create_dir(void *, const char *); 157 158 static futil_cb_t sysinst_futil_cb = { 159 .copy_file = sysinst_futil_copy_file, 160 .create_dir = sysinst_futil_create_dir 161 }; 162 163 static void sysinst_error_msg_button(ui_msg_dialog_t *, void *, unsigned); 164 static void sysinst_error_msg_close(ui_msg_dialog_t *, void *); 165 166 static ui_msg_dialog_cb_t sysinst_error_msg_cb = { 167 .button = sysinst_error_msg_button, 168 .close = sysinst_error_msg_close 169 }; 170 171 /** Close window. 172 * 173 * @param window Window 174 * @param arg Argument (sysinst_t *) 175 */ 176 static void wnd_close(ui_window_t *window, void *arg) 177 { 178 (void)window; 179 (void)arg; 180 } 181 182 /** Paint background window. 183 * 184 * @param window Window 185 * @param arg Argument (sysinst_t *) 186 */ 187 static errno_t bg_wnd_paint(ui_window_t *window, void *arg) 188 { 189 sysinst_t *sysinst = (sysinst_t *)arg; 190 gfx_rect_t app_rect; 191 gfx_context_t *gc; 192 errno_t rc; 193 194 gc = ui_window_get_gc(window); 195 196 rc = gfx_set_color(gc, sysinst->bg_color); 197 if (rc != EOK) 198 return rc; 199 200 ui_window_get_app_rect(window, &app_rect); 201 202 rc = gfx_fill_rect(gc, &app_rect); 203 if (rc != EOK) 204 return rc; 205 206 rc = gfx_update(gc); 207 if (rc != EOK) 208 return rc; 209 210 return EOK; 211 } 212 213 /** Installation confirm dialog OK button press. 214 * 215 * @param dialog Message dialog 216 * @param arg Argument (sysinst_t *) 217 * @param btn Button number 218 * @param earg Entry argument 219 */ 220 static void sysinst_confirm_button(ui_msg_dialog_t *dialog, void *arg, 221 unsigned btn) 222 { 223 sysinst_t *sysinst = (sysinst_t *) arg; 224 225 ui_msg_dialog_destroy(dialog); 226 227 switch (btn) { 228 case 0: 229 /* OK */ 230 sysinst_start(sysinst); 231 break; 232 default: 233 /* Cancel */ 234 ui_quit(sysinst->ui); 235 break; 236 } 237 } 238 239 /** Installation confirm dialog close request. 240 * 241 * @param dialog Message dialog 242 * @param arg Argument (sysinst_t *) 243 */ 244 static void sysinst_confirm_close(ui_msg_dialog_t *dialog, void *arg) 245 { 246 sysinst_t *sysinst = (sysinst_t *) arg; 247 248 ui_msg_dialog_destroy(dialog); 249 ui_quit(sysinst->ui); 250 } 251 252 /** Restart system dialog OK button press. 253 * 254 * @param dialog Message dialog 255 * @param arg Argument (sysinst_t *) 256 * @param btn Button number 257 * @param earg Entry argument 258 */ 259 static void sysinst_restart_dlg_button(ui_msg_dialog_t *dialog, void *arg, 260 unsigned btn) 261 { 262 sysinst_t *sysinst = (sysinst_t *) arg; 263 264 ui_msg_dialog_destroy(dialog); 265 266 (void)sysinst; 267 268 switch (btn) { 269 case 0: 270 /* OK */ 271 (void)sysinst_restart(sysinst); 272 break; 273 default: 274 /* Cancel */ 275 ui_quit(sysinst->ui); 276 break; 277 } 278 } 279 280 /** Restat system dialog close request. 281 * 282 * @param dialog Message dialog 283 * @param arg Argument (sysinst_t *) 284 */ 285 static void sysinst_restart_dlg_close(ui_msg_dialog_t *dialog, void *arg) 286 { 287 sysinst_t *sysinst = (sysinst_t *) arg; 288 289 ui_msg_dialog_destroy(dialog); 290 ui_quit(sysinst->ui); 291 } 292 293 /** Installation error message dialog button press. 294 * 295 * @param dialog Message dialog 296 * @param arg Argument (sysinst_t *) 297 * @param bnum Button number 298 */ 299 static void sysinst_error_msg_button(ui_msg_dialog_t *dialog, 300 void *arg, unsigned bnum) 301 { 302 sysinst_t *sysinst = (sysinst_t *) arg; 303 304 ui_msg_dialog_destroy(dialog); 305 ui_quit(sysinst->ui); 306 } 307 308 /** Installation error message dialog close request. 309 * 310 * @param dialog Message dialog 311 * @param arg Argument (shutdown_dlg_t *) 312 */ 313 static void sysinst_error_msg_close(ui_msg_dialog_t *dialog, void *arg) 314 { 315 sysinst_t *sysinst = (sysinst_t *) arg; 316 317 ui_msg_dialog_destroy(dialog); 318 ui_quit(sysinst->ui); 319 } 320 321 /** Create error message dialog. 322 * 323 * @param sysinst System installer 324 * @return EOK on success or an error code 325 */ 326 static errno_t sysinst_error_msg_create(sysinst_t *sysinst) 327 { 328 ui_msg_dialog_params_t params; 329 ui_msg_dialog_t *dialog; 330 errno_t rc; 331 332 ui_msg_dialog_params_init(¶ms); 333 params.caption = "Error"; 334 params.text = sysinst->errmsg; 335 params.flags |= umdf_topmost | umdf_center; 336 337 rc = ui_msg_dialog_create(sysinst->ui, ¶ms, &dialog); 338 if (rc != EOK) 339 return rc; 340 341 ui_msg_dialog_set_cb(dialog, &sysinst_error_msg_cb, (void *)sysinst); 342 343 return EOK; 344 } 345 346 /** Called when futil is starting to copy a file. 347 * 348 * @param arg Argument (sysinst_t *) 349 * @param src Source path 350 * @param dest Destination path 351 */ 352 static void sysinst_futil_copy_file(void *arg, const char *src, 353 const char *dest) 354 { 355 sysinst_t *sysinst = (sysinst_t *)arg; 356 char buf[128]; 357 358 (void)src; 359 snprintf(buf, sizeof(buf), "Copying %s.", dest); 360 sysinst_action(sysinst, buf); 361 } 362 363 /** Called when futil is about to create a directory. 364 * 365 * @param arg Argument (sysinst_t *) 366 * @param dest Destination path 367 */ 368 static void sysinst_futil_create_dir(void *arg, const char *dest) 369 { 370 sysinst_t *sysinst = (sysinst_t *)arg; 371 char buf[128]; 372 373 snprintf(buf, sizeof(buf), "Creating %s.", dest); 374 sysinst_action(sysinst, buf); 375 } 376 377 /** System shutdown complete. 378 * 379 * @param arg Argument (shutdown_t *) 380 */ 381 static void sysinst_shutdown_complete(void *arg) 382 { 383 (void)arg; 384 385 fibril_mutex_lock(&shutdown_lock); 386 shutdown_stopped = true; 387 shutdown_failed = false; 388 fibril_condvar_broadcast(&shutdown_cv); 389 fibril_mutex_unlock(&shutdown_lock); 390 } 391 392 /** System shutdown failed. 393 * 394 * @param arg Argument (not used) 395 */ 396 static void sysinst_shutdown_failed(void *arg) 397 { 398 (void)arg; 399 400 fibril_mutex_lock(&shutdown_lock); 401 shutdown_stopped = true; 402 shutdown_failed = true; 403 fibril_condvar_broadcast(&shutdown_cv); 404 fibril_mutex_unlock(&shutdown_lock); 405 } 406 94 407 /** Check the if the destination device exists. 95 408 * … … 113 426 /** Label the destination device. 114 427 * 428 * @param sysinst System installer 115 429 * @param dev Disk device to label 116 430 * @param psvc_id Place to store service ID of the created partition … … 118 432 * @return EOK on success or an error code 119 433 */ 120 static errno_t sysinst_label_dev(const char *dev, service_id_t *psvc_id) 434 static errno_t sysinst_label_dev(sysinst_t *sysinst, const char *dev, 435 service_id_t *psvc_id) 121 436 { 122 437 fdisk_t *fdisk; … … 129 444 errno_t rc; 130 445 131 printf("sysinst_label_dev(): get service ID '%s'\n", dev); 446 sysinst_debug(sysinst, "sysinst_label_dev(): get service ID"); 447 132 448 rc = loc_service_get_id(dev, &sid, 0); 133 449 if (rc != EOK) 134 450 return rc; 135 451 136 printf("sysinst_label_dev(): open device\n");452 sysinst_debug(sysinst, "sysinst_label_dev(): open device"); 137 453 138 454 rc = fdisk_create(&fdisk); 139 455 if (rc != EOK) { 140 printf("Error initializing fdisk.\n");456 sysinst_error(sysinst, "Error initializing fdisk."); 141 457 return rc; 142 458 } … … 144 460 rc = fdisk_dev_open(fdisk, sid, &fdev); 145 461 if (rc != EOK) { 146 printf("Error opening device.\n");147 return rc; 148 } 149 150 printf("sysinst_label_dev(): create mount directory\n");462 sysinst_error(sysinst, "Error opening device."); 463 return rc; 464 } 465 466 sysinst_debug(sysinst, "sysinst_label_dev(): create mount directory"); 151 467 152 468 rc = vfs_link_path(MOUNT_POINT, KIND_DIRECTORY, NULL); 153 if (rc != EOK) 154 return rc; 155 156 printf("sysinst_label_dev(): create label\n"); 469 if (rc != EOK) { 470 sysinst_error(sysinst, "Error creating mount directory."); 471 return rc; 472 } 473 474 sysinst_debug(sysinst, "sysinst_label_dev(): create label"); 157 475 158 476 rc = fdisk_label_create(fdev, lt_mbr); 159 477 if (rc != EOK) { 160 printf("Error creating label: %s.\n", str_error(rc));161 return rc; 162 } 163 164 printf("sysinst_label_dev(): create partition\n");478 sysinst_error(sysinst, "Error creating label."); 479 return rc; 480 } 481 482 sysinst_debug(sysinst, "sysinst_label_dev(): create partition"); 165 483 166 484 rc = fdisk_part_get_max_avail(fdev, spc_pri, &capa); 167 485 if (rc != EOK) { 168 printf("Error getting available capacity: %s.\n", str_error(rc)); 486 sysinst_error(sysinst, 487 "Error getting available capacity."); 169 488 return rc; 170 489 } … … 179 498 rc = fdisk_part_create(fdev, &pspec, &part); 180 499 if (rc != EOK) { 181 printf("Error creating partition.\n");500 sysinst_error(sysinst, "Error creating partition."); 182 501 return rc; 183 502 } … … 185 504 rc = fdisk_part_get_info(part, &pinfo); 186 505 if (rc != EOK) { 187 printf("Error getting partition information.\n");188 return rc; 189 } 190 191 printf("sysinst_label_dev(): OK\n");506 sysinst_error(sysinst, "Error getting partition information."); 507 return rc; 508 } 509 510 sysinst_debug(sysinst, "sysinst_label_dev(): OK"); 192 511 *psvc_id = pinfo.svc_id; 193 512 return EOK; … … 196 515 /** Set up system volume structure. 197 516 * 198 * @return EOK on success or an error code 199 */ 200 static errno_t sysinst_setup_sysvol(void) 517 * @param sysinst System installer 518 * @return EOK on success or an error code 519 */ 520 static errno_t sysinst_setup_sysvol(sysinst_t *sysinst) 201 521 { 202 522 errno_t rc; … … 215 535 rc = vfs_link_path(path, KIND_DIRECTORY, NULL); 216 536 if (rc != EOK) { 217 printf("Error creating directory '%s'.\n", path);537 sysinst_error(sysinst, "Error creating directory."); 218 538 goto error; 219 539 } … … 226 546 free(path); 227 547 path = NULL; 548 549 /* Copy initial configuration files */ 550 rc = futil_rcopy_contents(sysinst->futil, CFG_FILES_SRC, 551 CFG_FILES_DEST); 552 if (rc != EOK) 553 return rc; 228 554 229 555 return EOK; … … 238 564 * @return EOK on success or an error code 239 565 */ 240 static errno_t sysinst_copy_boot_files(void) 241 { 242 errno_t rc; 243 244 printf("sysinst_copy_boot_files(): copy bootloader files\n"); 245 rc = futil_rcopy_contents(BOOT_FILES_SRC, MOUNT_POINT); 246 if (rc != EOK) 247 return rc; 248 249 printf("sysinst_copy_boot_files(): OK\n"); 566 static errno_t sysinst_copy_boot_files(sysinst_t *sysinst) 567 { 568 errno_t rc; 569 570 log_msg(LOG_DEFAULT, LVL_NOTE, 571 "sysinst_copy_boot_files(): copy bootloader files"); 572 rc = futil_rcopy_contents(sysinst->futil, BOOT_FILES_SRC, MOUNT_POINT); 573 if (rc != EOK) 574 return rc; 575 576 sysinst_debug(sysinst, "sysinst_copy_boot_files(): OK"); 250 577 return EOK; 251 578 } … … 253 580 /** Set up configuration in the initial RAM disk. 254 581 * 255 * @return EOK on success or an error code 256 */ 257 static errno_t sysinst_customize_initrd(void) 582 * @param sysinst System installer 583 * @return EOK on success or an error code 584 */ 585 static errno_t sysinst_customize_initrd(sysinst_t *sysinst) 258 586 { 259 587 errno_t rc; … … 267 595 rc = rd_img_open(MOUNT_POINT "/boot/initrd.img", &rdpath, &rd); 268 596 if (rc != EOK) { 269 printf("Error opening initial RAM disk image.\n");270 goto error; 271 } 272 273 rv = asprintf(&path, "%s%s", rdpath, "/cfg/ volsrv.sif");597 sysinst_error(sysinst, "Error opening initial RAM disk image."); 598 goto error; 599 } 600 601 rv = asprintf(&path, "%s%s", rdpath, "/cfg/initvol.sif"); 274 602 if (rv < 0) { 275 603 rc = ENOMEM; … … 277 605 } 278 606 279 printf("Configuring volume server.\n"); 607 sysinst_debug(sysinst, "Configuring volume server."); 608 280 609 rc = vol_volumes_create(path, &volumes); 281 610 if (rc != EOK) { 282 printf("Error creating volume server configuration.\n"); 611 sysinst_error(sysinst, 612 "Error creating volume server configuration."); 283 613 rc = EIO; 284 614 goto error; 285 615 } 286 616 287 printf("Configuring volume server: look up volume\n");617 sysinst_debug(sysinst, "Configuring volume server: look up volume"); 288 618 rc = vol_volume_lookup_ref(volumes, INST_VOL_LABEL, &volume); 289 619 if (rc != EOK) { 290 printf("Error creating volume server configuration.\n"); 620 sysinst_error(sysinst, 621 "Error creating volume server configuration."); 291 622 rc = EIO; 292 623 goto error; 293 624 } 294 625 295 printf("Configuring volume server: set mount point\n");626 sysinst_debug(sysinst, "Configuring volume server: set mount point"); 296 627 rc = vol_volume_set_mountp(volume, INST_VOL_MP); 297 628 if (rc != EOK) { 298 printf("Error creating system partition configuration.\n"); 629 sysinst_error(sysinst, 630 "Error creating system partition configuration."); 299 631 rc = EIO; 300 632 goto error; 301 633 } 302 634 303 printf("Configuring volume server: delete reference\n"); 635 rc = vol_volumes_sync(volumes); 636 if (rc != EOK) { 637 sysinst_error(sysinst, "Error saving volume confiuration."); 638 goto error; 639 } 640 641 log_msg(LOG_DEFAULT, LVL_NOTE, 642 "Configuring volume server: delete reference"); 304 643 vol_volume_del_ref(volume); 305 644 volume = NULL; 306 printf("Configuring volume server: destroy volumes object\n"); 645 log_msg(LOG_DEFAULT, LVL_NOTE, 646 "Configuring volume server: destroy volumes object"); 307 647 vol_volumes_destroy(volumes); 308 648 volumes = NULL; … … 310 650 rc = rd_img_close(rd); 311 651 if (rc != EOK) { 312 printf("Error closing initial RAM disk image.\n");652 sysinst_error(sysinst, "Error closing initial RAM disk image."); 313 653 rc = EIO; 314 654 goto error; … … 353 693 * Install Grub's boot blocks. 354 694 * 695 * @param sysinst System installer 355 696 * @param devp Disk device 356 697 * @return EOK on success or an error code 357 698 */ 358 static errno_t sysinst_copy_boot_blocks( const char *devp)699 static errno_t sysinst_copy_boot_blocks(sysinst_t *sysinst, const char *devp) 359 700 { 360 701 void *boot_img; … … 370 711 errno_t rc; 371 712 372 printf("sysinst_copy_boot_blocks: Read boot block image.\n"); 373 rc = futil_get_file(BOOT_FILES_SRC "/boot/grub/i386-pc/boot.img", 713 log_msg(LOG_DEFAULT, LVL_NOTE, 714 "sysinst_copy_boot_blocks: Read boot block image."); 715 716 rc = futil_get_file(sysinst->futil, 717 BOOT_FILES_SRC "/boot/grub/i386-pc/boot.img", 374 718 &boot_img, &boot_img_size); 375 719 if (rc != EOK || boot_img_size != 512) 376 720 return EIO; 377 721 378 printf("sysinst_copy_boot_blocks: Read GRUB core image.\n"); 379 rc = futil_get_file(BOOT_FILES_SRC "/boot/grub/i386-pc/core.img", 722 log_msg(LOG_DEFAULT, LVL_NOTE, 723 "sysinst_copy_boot_blocks: Read GRUB core image."); 724 725 rc = futil_get_file(sysinst->futil, 726 BOOT_FILES_SRC "/boot/grub/i386-pc/core.img", 380 727 &core_img, &core_img_size); 381 728 if (rc != EOK) 382 729 return EIO; 383 730 384 printf("sysinst_copy_boot_blocks: get service ID.\n"); 731 log_msg(LOG_DEFAULT, LVL_NOTE, 732 "sysinst_copy_boot_blocks: get service ID."); 733 385 734 rc = loc_service_get_id(devp, &sid, 0); 386 735 if (rc != EOK) 387 736 return rc; 388 737 389 printf("sysinst_copy_boot_blocks: block_init.\n"); 738 log_msg(LOG_DEFAULT, LVL_NOTE, 739 "sysinst_copy_boot_blocks: block_init."); 740 390 741 rc = block_init(sid); 391 742 if (rc != EOK) 392 743 return rc; 393 744 394 printf("sysinst_copy_boot_blocks: get block size\n"); 745 log_msg(LOG_DEFAULT, LVL_NOTE, 746 "sysinst_copy_boot_blocks: get block size"); 747 395 748 rc = block_get_bsize(sid, &bsize); 396 749 if (rc != EOK) … … 398 751 399 752 if (bsize != 512) { 400 printf("Device block size != 512.\n");753 sysinst_error(sysinst, "Device block size != 512."); 401 754 return EIO; 402 755 } 403 756 404 printf("sysinst_copy_boot_blocks: read boot block\n"); 757 log_msg(LOG_DEFAULT, LVL_NOTE, 758 "sysinst_copy_boot_blocks: read boot block"); 759 405 760 rc = block_read_direct(sid, BOOT_BLOCK_IDX, 1, bbuf); 406 761 if (rc != EOK) … … 417 772 --bl; 418 773 if ((void *)bl < core_img) { 419 printf("No block terminator in core image.\n"); 774 sysinst_error(sysinst, 775 "No block terminator in core image."); 420 776 return EIO; 421 777 } … … 431 787 set_unaligned_u64le(bbuf + grub_boot_machine_kernel_sector, core_start); 432 788 433 printf("sysinst_copy_boot_blocks: write boot block\n"); 789 log_msg(LOG_DEFAULT, LVL_NOTE, 790 "sysinst_copy_boot_blocks: write boot block"); 791 434 792 rc = block_write_direct(sid, BOOT_BLOCK_IDX, 1, bbuf); 435 793 if (rc != EOK) 436 794 return EIO; 437 795 438 printf("sysinst_copy_boot_blocks: write core blocks\n"); 796 log_msg(LOG_DEFAULT, LVL_NOTE, 797 "sysinst_copy_boot_blocks: write core blocks"); 798 439 799 /* XXX Must pad last block with zeros */ 440 800 rc = block_write_direct(sid, core_start, core_blocks, core_img); … … 442 802 return EIO; 443 803 444 printf("sysinst_copy_boot_blocks: OK.\n"); 804 log_msg(LOG_DEFAULT, LVL_NOTE, 805 "sysinst_copy_boot_blocks: OK."); 806 445 807 return EOK; 446 808 } … … 448 810 /** Eject installation volume. 449 811 * 812 * @param sysinst System installer 450 813 * @param psvc_id Partition service ID 451 */ 452 static errno_t sysinst_eject_dev(service_id_t part_id) 814 * @return EOK on success or an error code 815 */ 816 static errno_t sysinst_eject_dev(sysinst_t *sysinst, service_id_t part_id) 453 817 { 454 818 vol_t *vol = NULL; … … 457 821 rc = vol_create(&vol); 458 822 if (rc != EOK) { 459 printf("Error contacting volume service.\n");823 sysinst_error(sysinst, "Error contacting volume service."); 460 824 goto out; 461 825 } 462 826 463 rc = vol_part_eject(vol, part_id );464 if (rc != EOK) { 465 printf("Error ejecting volume.\n");827 rc = vol_part_eject(vol, part_id, vef_physical); 828 if (rc != EOK) { 829 sysinst_error(sysinst, "Error ejecting volume."); 466 830 goto out; 467 831 } … … 473 837 } 474 838 839 /** Restart the system. 840 * 841 * @param sysinst System installer 842 * @return EOK on success or an error code 843 */ 844 static errno_t sysinst_restart(sysinst_t *sysinst) 845 { 846 errno_t rc; 847 system_t *system; 848 849 fibril_mutex_initialize(&shutdown_lock); 850 fibril_condvar_initialize(&shutdown_cv); 851 shutdown_stopped = false; 852 shutdown_failed = false; 853 854 rc = system_open(SYSTEM_DEFAULT, &sysinst_system_cb, NULL, &system); 855 if (rc != EOK) { 856 sysinst_error(sysinst, 857 "Failed opening system control service."); 858 return rc; 859 } 860 861 rc = system_restart(system); 862 if (rc != EOK) { 863 system_close(system); 864 sysinst_error(sysinst, "Failed requesting system restart."); 865 return rc; 866 } 867 868 fibril_mutex_lock(&shutdown_lock); 869 sysinst_debug(sysinst, "The system is shutting down..."); 870 871 while (!shutdown_stopped) 872 fibril_condvar_wait(&shutdown_cv, &shutdown_lock); 873 874 if (shutdown_failed) { 875 sysinst_error(sysinst, "Shutdown failed."); 876 system_close(system); 877 return rc; 878 } 879 880 log_msg(LOG_DEFAULT, LVL_NOTE, 881 "Shutdown complete. It is now safe to remove power."); 882 883 /* Sleep forever */ 884 while (true) 885 fibril_condvar_wait(&shutdown_cv, &shutdown_lock); 886 887 fibril_mutex_unlock(&shutdown_lock); 888 889 system_close(system); 890 return 0; 891 892 } 893 475 894 /** Install system to a device. 476 895 * 896 * @parma sysinst System installer 477 897 * @param dev Device to install to. 478 898 * @return EOK on success or an error code 479 899 */ 480 static errno_t sysinst_install( const char *dev)900 static errno_t sysinst_install(sysinst_t *sysinst, const char *dev) 481 901 { 482 902 errno_t rc; 483 903 service_id_t psvc_id; 484 904 485 rc = sysinst_label_dev(dev, &psvc_id); 486 if (rc != EOK) 487 return rc; 488 489 printf("FS created and mounted. Creating system directory structure.\n"); 490 rc = sysinst_setup_sysvol(); 491 if (rc != EOK) 492 return rc; 493 494 printf("Directories created. Copying boot files.\n"); 495 rc = sysinst_copy_boot_files(); 496 if (rc != EOK) 497 return rc; 498 499 printf("Boot files done. Configuring the system.\n"); 500 rc = sysinst_customize_initrd(); 501 if (rc != EOK) 502 return rc; 503 504 printf("Boot files done. Installing boot blocks.\n"); 505 rc = sysinst_copy_boot_blocks(dev); 506 if (rc != EOK) 507 return rc; 508 509 printf("Ejecting device.\n"); 510 rc = sysinst_eject_dev(psvc_id); 511 if (rc != EOK) 512 return rc; 513 514 return EOK; 515 } 516 517 int main(int argc, char *argv[]) 518 { 905 sysinst_action(sysinst, "Creating device label and file system."); 906 907 rc = sysinst_label_dev(sysinst, dev, &psvc_id); 908 if (rc != EOK) 909 return rc; 910 911 sysinst_action(sysinst, "Creating system directory structure."); 912 rc = sysinst_setup_sysvol(sysinst); 913 if (rc != EOK) 914 return rc; 915 916 sysinst_action(sysinst, "Copying boot files."); 917 rc = sysinst_copy_boot_files(sysinst); 918 if (rc != EOK) 919 return rc; 920 921 sysinst_action(sysinst, "Configuring the system."); 922 rc = sysinst_customize_initrd(sysinst); 923 if (rc != EOK) 924 return rc; 925 926 sysinst_action(sysinst, "Installing boot blocks."); 927 rc = sysinst_copy_boot_blocks(sysinst, dev); 928 if (rc != EOK) 929 return rc; 930 931 sysinst_action(sysinst, "Ejecting device."); 932 rc = sysinst_eject_dev(sysinst, psvc_id); 933 if (rc != EOK) 934 return rc; 935 936 return EOK; 937 } 938 939 /** Installation fibril. 940 * 941 * @param arg Argument (sysinst_t *) 942 * @return EOK on success or an error code 943 */ 944 static errno_t sysinst_install_fibril(void *arg) 945 { 946 sysinst_t *sysinst = (sysinst_t *)arg; 519 947 unsigned i; 520 948 errno_t rc; 949 950 (void)sysinst; 521 951 522 952 i = 0; … … 525 955 if (rc == EOK) 526 956 break; 957 ++i; 527 958 } 528 959 529 960 if (default_devs[i] == NULL) { 530 printf("Cannot determine installation device.\n"); 961 sysinst_error(sysinst, "Cannot determine installation device."); 962 rc = ENOENT; 963 goto error; 964 } 965 966 rc = sysinst_install(sysinst, default_devs[i]); 967 if (rc != EOK) 968 goto error; 969 970 sysinst_progress_destroy(sysinst->progress); 971 rc = sysinst_restart_dlg_create(sysinst); 972 if (rc != EOK) 973 goto error; 974 975 return EOK; 976 error: 977 ui_lock(sysinst->ui); 978 sysinst_progress_destroy(sysinst->progress); 979 (void)sysinst_error_msg_create(sysinst); 980 ui_unlock(sysinst->ui); 981 return rc; 982 } 983 984 /** Create installation progress window. 985 * 986 * @param sysinst System installer 987 * @param rprogress Place to store pointer to new progress window 988 * @return EOK on success or an error code 989 */ 990 static errno_t sysinst_progress_create(sysinst_t *sysinst, 991 sysinst_progress_t **rprogress) 992 { 993 ui_wnd_params_t params; 994 ui_window_t *window = NULL; 995 gfx_rect_t rect; 996 gfx_rect_t arect; 997 ui_resource_t *ui_res; 998 sysinst_progress_t *progress; 999 ui_fixed_t *fixed = NULL; 1000 errno_t rc; 1001 1002 ui_wnd_params_init(¶ms); 1003 params.caption = "System Installation"; 1004 params.style &= ~ui_wds_titlebar; 1005 params.flags |= ui_wndf_topmost; 1006 params.placement = ui_wnd_place_center; 1007 if (ui_is_textmode(sysinst->ui)) { 1008 params.rect.p0.x = 0; 1009 params.rect.p0.y = 0; 1010 params.rect.p1.x = 64; 1011 params.rect.p1.y = 5; 1012 } else { 1013 params.rect.p0.x = 0; 1014 params.rect.p0.y = 0; 1015 params.rect.p1.x = 500; 1016 params.rect.p1.y = 60; 1017 } 1018 1019 progress = calloc(1, sizeof(sysinst_progress_t)); 1020 if (progress == NULL) { 1021 rc = ENOMEM; 1022 sysinst_error(sysinst, "Out of memory."); 1023 goto error; 1024 } 1025 1026 rc = ui_window_create(sysinst->ui, ¶ms, &window); 1027 if (rc != EOK) { 1028 sysinst_error(sysinst, "Error creating window."); 1029 goto error; 1030 } 1031 1032 ui_window_set_cb(window, &progress_window_cb, (void *)sysinst); 1033 1034 ui_res = ui_window_get_res(window); 1035 1036 rc = ui_fixed_create(&fixed); 1037 if (rc != EOK) { 1038 sysinst_error(sysinst, "Error creating fixed layout."); 1039 goto error; 1040 } 1041 1042 rc = ui_label_create(ui_res, "Installing system. Please wait...", 1043 &progress->label); 1044 if (rc != EOK) { 1045 sysinst_error(sysinst, "Error creating label."); 1046 goto error; 1047 } 1048 1049 ui_window_get_app_rect(window, &arect); 1050 1051 if (ui_is_textmode(sysinst->ui)) { 1052 rect.p0.x = arect.p0.x; 1053 rect.p0.y = arect.p0.y; 1054 rect.p1.x = arect.p1.x; 1055 rect.p1.y = 2; 1056 } else { 1057 rect.p0.x = arect.p0.x; 1058 rect.p0.y = arect.p0.y; 1059 rect.p1.x = arect.p1.x; 1060 rect.p1.y = 30; 1061 } 1062 ui_label_set_rect(progress->label, &rect); 1063 ui_label_set_halign(progress->label, gfx_halign_center); 1064 ui_label_set_valign(progress->label, gfx_valign_center); 1065 1066 rc = ui_fixed_add(fixed, ui_label_ctl(progress->label)); 1067 if (rc != EOK) { 1068 sysinst_error(sysinst, "Error adding control to layout."); 1069 ui_label_destroy(progress->label); 1070 progress->label = NULL; 1071 goto error; 1072 } 1073 1074 rc = ui_label_create(ui_res, "", 1075 &progress->action); 1076 if (rc != EOK) { 1077 sysinst_error(sysinst, "Error creating label."); 1078 goto error; 1079 } 1080 1081 if (ui_is_textmode(sysinst->ui)) { 1082 rect.p0.x = arect.p0.x; 1083 rect.p0.y = 3; 1084 rect.p1.x = arect.p1.x; 1085 rect.p1.y = arect.p1.y; 1086 } else { 1087 rect.p0.x = arect.p0.x; 1088 rect.p0.y = 30; 1089 rect.p1.x = arect.p1.x; 1090 rect.p1.y = arect.p1.y; 1091 } 1092 ui_label_set_rect(progress->action, &rect); 1093 ui_label_set_halign(progress->action, gfx_halign_center); 1094 ui_label_set_valign(progress->action, gfx_valign_center); 1095 1096 rc = ui_fixed_add(fixed, ui_label_ctl(progress->action)); 1097 if (rc != EOK) { 1098 sysinst_error(sysinst, "Error adding control to layout."); 1099 ui_label_destroy(progress->label); 1100 progress->label = NULL; 1101 goto error; 1102 } 1103 1104 ui_window_add(window, ui_fixed_ctl(fixed)); 1105 fixed = NULL; 1106 1107 rc = ui_window_paint(window); 1108 if (rc != EOK) { 1109 sysinst_error(sysinst, "Error painting window."); 1110 goto error; 1111 } 1112 1113 progress->window = window; 1114 progress->fixed = fixed; 1115 *rprogress = progress; 1116 return EOK; 1117 error: 1118 if (progress != NULL && progress->fixed != NULL) 1119 ui_fixed_destroy(progress->fixed); 1120 if (window != NULL) 1121 ui_window_destroy(window); 1122 if (progress != NULL) 1123 free(progress); 1124 return rc; 1125 } 1126 1127 /** Destroy installation progress window. 1128 * 1129 * @param sysinst System installer 1130 * @param rprogress Place to store pointer to new progress window 1131 * @return EOK on success or an error code 1132 */ 1133 static void sysinst_progress_destroy(sysinst_progress_t *progress) 1134 { 1135 if (progress == NULL) 1136 return; 1137 1138 ui_window_destroy(progress->window); 1139 free(progress); 1140 } 1141 1142 /** Set current action message. 1143 * 1144 * @param sysinst System installer 1145 * @param action Action text 1146 */ 1147 static void sysinst_action(sysinst_t *sysinst, const char *action) 1148 { 1149 if (sysinst->progress == NULL) 1150 return; 1151 1152 ui_label_set_text(sysinst->progress->action, action); 1153 ui_label_paint(sysinst->progress->action); 1154 log_msg(LOG_DEFAULT, LVL_NOTE, "%s", action); 1155 } 1156 1157 /** Set current error message. 1158 * 1159 * @param sysinst System installer 1160 * @param errmsg Error message 1161 */ 1162 static void sysinst_error(sysinst_t *sysinst, const char *errmsg) 1163 { 1164 str_cpy(sysinst->errmsg, sizeof(sysinst->errmsg), errmsg); 1165 log_msg(LOG_DEFAULT, LVL_ERROR, errmsg); 1166 } 1167 1168 /** Log a debug message. 1169 * 1170 * @param sysinst System installer 1171 * @param errmsg Error message 1172 */ 1173 static void sysinst_debug(sysinst_t *sysinst, const char *msg) 1174 { 1175 log_msg(LOG_DEFAULT, LVL_ERROR, msg); 1176 } 1177 1178 /** Start system installation. 1179 * 1180 * @param sysinst System installer 1181 * @return EOK on success or an error code 1182 */ 1183 static int sysinst_start(sysinst_t *sysinst) 1184 { 1185 errno_t rc; 1186 fid_t fid; 1187 1188 rc = sysinst_progress_create(sysinst, &sysinst->progress); 1189 if (rc != EOK) 1190 return rc; 1191 1192 fid = fibril_create(sysinst_install_fibril, (void *)sysinst); 1193 if (fid == 0) { 1194 sysinst_error(sysinst, "Out of memory."); 1195 return ENOMEM; 1196 } 1197 1198 fibril_add_ready(fid); 1199 return EOK; 1200 } 1201 1202 /** Create installation confirmation dialog. 1203 * 1204 * @param sysinst System installer 1205 * @return EOK on success or an error code 1206 */ 1207 static errno_t sysinst_confirm_create(sysinst_t *sysinst) 1208 { 1209 ui_msg_dialog_params_t params; 1210 ui_msg_dialog_t *dialog; 1211 errno_t rc; 1212 1213 ui_msg_dialog_params_init(¶ms); 1214 params.caption = "System installation"; 1215 params.text = "This will install HelenOS to your computer. Continue?"; 1216 params.choice = umdc_ok_cancel; 1217 params.flags |= umdf_topmost | umdf_center; 1218 1219 rc = ui_msg_dialog_create(sysinst->ui, ¶ms, &dialog); 1220 if (rc != EOK) 1221 return rc; 1222 1223 ui_msg_dialog_set_cb(dialog, &sysinst_confirm_cb, sysinst); 1224 return EOK; 1225 } 1226 1227 /** Create restart dialog. 1228 * 1229 * @param sysinst System installer 1230 * @return EOK on success or an error code 1231 */ 1232 static errno_t sysinst_restart_dlg_create(sysinst_t *sysinst) 1233 { 1234 ui_msg_dialog_params_t params; 1235 ui_msg_dialog_t *dialog; 1236 errno_t rc; 1237 1238 ui_msg_dialog_params_init(¶ms); 1239 params.caption = "Restart System"; 1240 params.text = "Installation complete. Restart the system?"; 1241 params.choice = umdc_ok_cancel; 1242 params.flags |= umdf_topmost | umdf_center; 1243 1244 rc = ui_msg_dialog_create(sysinst->ui, ¶ms, &dialog); 1245 if (rc != EOK) 1246 return rc; 1247 1248 ui_msg_dialog_set_cb(dialog, &sysinst_restart_dlg_cb, sysinst); 1249 return EOK; 1250 } 1251 1252 /** Run system installer on display. 1253 * 1254 * @param display_spec Display specification 1255 * @return EOK on success or an error code 1256 */ 1257 static errno_t sysinst_run(const char *display_spec) 1258 { 1259 ui_t *ui = NULL; 1260 sysinst_t *sysinst; 1261 ui_wnd_params_t params; 1262 errno_t rc; 1263 1264 sysinst = calloc(1, sizeof(sysinst_t)); 1265 if (sysinst == NULL) 1266 return ENOMEM; 1267 1268 rc = futil_create(&sysinst_futil_cb, (void *)sysinst, &sysinst->futil); 1269 if (rc != EOK) { 1270 printf("Out of memory.\n"); 1271 goto error; 1272 } 1273 1274 rc = ui_create(display_spec, &ui); 1275 if (rc != EOK) { 1276 printf("Error creating UI on display %s.\n", display_spec); 1277 goto error; 1278 } 1279 1280 sysinst->ui = ui; 1281 1282 ui_wnd_params_init(¶ms); 1283 params.caption = "System Installation"; 1284 params.style &= ~ui_wds_decorated; 1285 params.placement = ui_wnd_place_full_screen; 1286 params.flags |= ui_wndf_topmost | ui_wndf_nofocus; 1287 1288 rc = ui_window_create(sysinst->ui, ¶ms, &sysinst->bgwindow); 1289 if (rc != EOK) { 1290 printf("Error creating window.\n"); 1291 goto error; 1292 } 1293 1294 ui_window_set_cb(sysinst->bgwindow, &bg_window_cb, (void *)sysinst); 1295 1296 if (ui_is_textmode(sysinst->ui)) { 1297 rc = gfx_color_new_ega(0x17, &sysinst->bg_color); 1298 if (rc != EOK) { 1299 printf("Error allocating color.\n"); 1300 goto error; 1301 } 1302 } else { 1303 rc = gfx_color_new_rgb_i16(0x8000, 0xc800, 0xffff, &sysinst->bg_color); 1304 if (rc != EOK) { 1305 printf("Error allocating color.\n"); 1306 goto error; 1307 } 1308 } 1309 1310 rc = ui_window_paint(sysinst->bgwindow); 1311 if (rc != EOK) { 1312 printf("Error painting window.\n"); 1313 goto error; 1314 } 1315 1316 (void)sysinst_confirm_create(sysinst); 1317 1318 ui_run(ui); 1319 1320 if (sysinst->bgwindow != NULL) 1321 ui_window_destroy(sysinst->bgwindow); 1322 if (sysinst->system != NULL) 1323 system_close(sysinst->system); 1324 gfx_color_delete(sysinst->bg_color); 1325 ui_destroy(ui); 1326 free(sysinst); 1327 return EOK; 1328 error: 1329 if (sysinst->futil != NULL) 1330 futil_destroy(sysinst->futil); 1331 if (sysinst->system != NULL) 1332 system_close(sysinst->system); 1333 if (sysinst->bg_color != NULL) 1334 gfx_color_delete(sysinst->bg_color); 1335 if (sysinst->bgwindow != NULL) 1336 ui_window_destroy(sysinst->bgwindow); 1337 if (ui != NULL) 1338 ui_destroy(ui); 1339 free(sysinst); 1340 return rc; 1341 } 1342 1343 static void print_syntax(void) 1344 { 1345 printf("Syntax: " NAME " [-d <display-spec>]\n"); 1346 } 1347 1348 int main(int argc, char *argv[]) 1349 { 1350 const char *display_spec = UI_ANY_DEFAULT; 1351 errno_t rc; 1352 int i; 1353 1354 i = 1; 1355 while (i < argc && argv[i][0] == '-') { 1356 if (str_cmp(argv[i], "-d") == 0) { 1357 ++i; 1358 if (i >= argc) { 1359 printf("Argument missing.\n"); 1360 print_syntax(); 1361 return 1; 1362 } 1363 1364 display_spec = argv[i++]; 1365 } else { 1366 printf("Invalid option '%s'.\n", argv[i]); 1367 print_syntax(); 1368 return 1; 1369 } 1370 } 1371 1372 if (i < argc) { 1373 print_syntax(); 531 1374 return 1; 532 1375 } 533 1376 534 return sysinst_install(default_devs[i]); 1377 if (log_init(NAME) != EOK) { 1378 printf(NAME ": Failed to initialize logging.\n"); 1379 return 1; 1380 } 1381 1382 rc = sysinst_run(display_spec); 1383 if (rc != EOK) 1384 return 1; 1385 1386 return 0; 535 1387 } 536 1388
Note:
See TracChangeset
for help on using the changeset viewer.