/*      $NetBSD: install.c,v 1.25 2023/01/06 18:13:40 martin Exp $      */

/*
* Copyright 1997 Piermont Information Systems Inc.
* All rights reserved.
*
* Written by Philip A. Nelson for Piermont Information Systems Inc.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
*    notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
*    notice, this list of conditions and the following disclaimer in the
*    documentation and/or other materials provided with the distribution.
* 3. The name of Piermont Information Systems Inc. may not be used to endorse
*    or promote products derived from this software without specific prior
*    written permission.
*
* THIS SOFTWARE IS PROVIDED BY PIERMONT INFORMATION SYSTEMS INC. ``AS IS''
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL PIERMONT INFORMATION SYSTEMS INC. BE
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
* THE POSSIBILITY OF SUCH DAMAGE.
*
*/

/* install.c -- system installation. */

#include <sys/param.h>
#include <stdio.h>
#include <curses.h>
#include "defs.h"
#include "msg_defs.h"
#include "menu_defs.h"

/*
* helper function: call the md pre/post disklabel callback for all involved
* inner partitions and write them back.
* return net result
*/
static bool
write_all_parts(struct install_partition_desc *install)
{
       struct disk_partitions **allparts, *parts;
#ifndef NO_CLONES
       struct selected_partition *src;
#endif
       size_t num_parts, i, j;
       bool found, res;

       /* pessimistic assumption: all partitions on different devices */
       allparts = calloc(install->num + install->num_write_back,
           sizeof(*allparts));
       if (allparts == NULL)
               return false;

       /* collect all different partition sets */
       num_parts = 0;
       for (i = 0; i < install->num_write_back; i++) {
               parts = install->write_back[i];
               if (parts == NULL)
                       continue;
               found = false;
               for (j = 0; j < num_parts; j++) {
                       if (allparts[j] == parts) {
                               found = true;
                               break;
                       }
               }
               if (found)
                       continue;
               allparts[num_parts++] = parts;
       }
       for (i = 0; i < install->num; i++) {
               parts = install->infos[i].parts;
               if (parts == NULL)
                       continue;
               found = false;
               for (j = 0; j < num_parts; j++) {
                       if (allparts[j] == parts) {
                               found = true;
                               break;
                       }
               }
               if (found)
                       continue;
               allparts[num_parts++] = parts;
       }

       /* do multiple phases, abort anytime and go out, returning res */

       res = true;

       /* phase 1: pre disklabel - used to write MBR and similar */
       for (i = 0; i < num_parts; i++) {
               if (!md_pre_disklabel(install, allparts[i])) {
                       res = false;
                       goto out;
               }
       }

       /* phase 2: write our partitions (used to be: disklabel) */
       for (i = 0; i < num_parts; i++) {
               if (!allparts[i]->pscheme->write_to_disk(allparts[i])) {
                       res = false;
                       goto out;
               }
       }

       /* phase 3: now we may have a first chance to enable swap space */
       set_swap_if_low_ram(install);

#ifndef NO_CLONES
       /* phase 4: copy any cloned partitions data (if requested) */
       for (i = 0; i < install->num; i++) {
               if ((install->infos[i].flags & PUIFLG_CLONE_PARTS) == 0
                   || install->infos[i].clone_src == NULL
                   || !install->infos[i].clone_src->with_data)
                       continue;
               src = &install->infos[i].clone_src
                   ->selection[install->infos[i].clone_ndx];
               clone_partition_data(install->infos[i].parts,
                   install->infos[i].cur_part_id,
                   src->parts, src->id);
       }
#endif

       /* phase 5: post disklabel (used for updating boot loaders) */
       for (i = 0; i < num_parts; i++) {
               if (!md_post_disklabel(install, allparts[i])) {
                       res = false;
                       goto out;
               }
       }

out:
       free(allparts);

       return res;
}

/* Do the system install. */

void
do_install(void)
{
       int find_disks_ret;
       int retcode = 0, res;
       struct install_partition_desc install = {};

#ifndef NO_PARTMAN
       partman_go = -1;
#else
       partman_go = 0;
#endif

#ifndef DEBUG
       msg_display(MSG_installusure);
       if (!ask_noyes(NULL))
               return;
#endif

       memset(&install, 0, sizeof install);

       /* Create and mount partitions */
       find_disks_ret = find_disks(msg_string(MSG_install), false);
       if (partman_go == 1) {
               if (partman(&install) < 0) {
                       hit_enter_to_continue(MSG_abort_part, NULL);
                       return;
               }
       } else if (find_disks_ret < 0)
               return;
       else {
       /* Classical partitioning wizard */
               partman_go = 0;
               clear();
               refresh();

               if (check_swap(pm->diskdev, 0) > 0) {
                       hit_enter_to_continue(MSG_swapactive, NULL);
                       if (check_swap(pm->diskdev, 1) < 0) {
                               hit_enter_to_continue(MSG_swapdelfailed, NULL);
                               if (!debug)
                                       return;
                       }
               }

               for (;;) {
                       if (md_get_info(&install)) {
                               res = md_make_bsd_partitions(&install);
                               if (res == -1) {
                                       pm->parts = NULL;
                                       continue;
                               } else if (res == 1) {
                                       break;
                               }
                       }
                       hit_enter_to_continue(MSG_abort_inst, NULL);
                       goto error;
               }

               /* Last chance ... do you really want to do this? */
               clear();
               refresh();
               msg_fmt_display(MSG_lastchance, "%s", pm->diskdev);
               if (!ask_noyes(NULL))
                       goto error;

               if ((!pm->no_part && !write_all_parts(&install)) ||
                   make_filesystems(&install) ||
                   make_fstab(&install) != 0 ||
                   md_post_newfs(&install) != 0)
                       goto error;
       }

       /* Unpack the distribution. */
       process_menu(MENU_distset, &retcode);
       if (retcode == 0)
               goto error;
       if (get_and_unpack_sets(0, MSG_disksetupdone,
           MSG_extractcomplete, MSG_abortinst) != 0)
               goto error;

       if (md_post_extract(&install, false) != 0)
               goto error;

       root_pw_setup();
#if CHECK_ENTROPY
       do_add_entropy();
#endif
       do_configmenu(&install);

       sanity_check();

       md_cleanup_install(&install);

       hit_enter_to_continue(MSG_instcomplete, NULL);

error:
       free_install_desc(&install);
}