/*      $NetBSD: md.c,v 1.13 2025/05/28 06:06:53 andvar Exp $ */

/*
* Copyright 1997 Piermont Information Systems Inc.
* All rights reserved.
*
* Based on code 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.
*/

/* md.c -- mac68k machine specific routines */

#include <stdio.h>
#include <util.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/ioctl.h>
#include <sys/utsname.h>
#include <machine/int_fmtio.h>

#include "defs.h"
#include "md.h"
#include "msg_defs.h"
#include "menu_defs.h"

static int      stricmp(const char *c1, const char *c2);
static void     setpartition(struct apple_part_map_entry *, char *, int);
static int      getFreeLabelEntry(char *);
static char     *getFstype(struct apple_part_map_entry *, int, char *);
static char     *getUse(struct apple_part_map_entry *, int, char *);
static char     *getName(struct apple_part_map_entry *, int, char *);
static int      findStdType(int, char *, int, int *, int);
static int      check_for_errors(void);
static int      edit_diskmap(void);
#ifdef MD_DEBUG_SORT_MERGE
static int      md_debug_dump(char *);
#endif

int blk_size;

MAP_TYPE map_types[] = {
       {MAP_RESERVED, APPLE_PART_TYPE_DRIVER},
       {MAP_RESERVED, APPLE_PART_TYPE_DRIVER43},
       {MAP_RESERVED, APPLE_PART_TYPE_DRIVERATA},
       {MAP_RESERVED, APPLE_PART_TYPE_FWB_COMPONENT},
       {MAP_MACOS,    APPLE_PART_TYPE_MAC},
       {MAP_NETBSD,   APPLE_PART_TYPE_NETBSD},
       {MAP_RESERVED, APPLE_PART_TYPE_PARTMAP},
       {MAP_OTHER,    APPLE_PART_TYPE_SCRATCH},
       {MAP_NETBSD,   APPLE_PART_TYPE_UNIX},
       {MAP_EOL,      NULL}
};

MAP map = {0, 0, 0, 0, 0, 0, 0, 0, {0}, NULL};

struct apple_part_map_entry new_map[] =
{
       { APPLE_PART_MAP_ENTRY_MAGIC, 0xa5a5, 6, 1, NEW_MAP_SIZE & 0x7e,
         "Apple", "Apple_Partition_Map", 0, NEW_MAP_SIZE, 0x37,
         0, 0, 0, 0, 0, 0, 0, {0}, {0}, {0}},
       { APPLE_PART_MAP_ENTRY_MAGIC, 0, 6, 64, 32,
         "Macintosh", "Apple_Driver", 0, 0, 0x37,
         0, 0, 0, 0, 0, 0, 0, {0}, {0}, {0}},
       { APPLE_PART_MAP_ENTRY_MAGIC, 0, 6, 96, 64,
         "Macintosh", "Apple_Driver43", 0, 0, 0x37,
         0, 0, 0, 0, 0, 0, 0, {0}, {0}, {0}},
       { APPLE_PART_MAP_ENTRY_MAGIC, 0, 6, 160, 64,
         "Macintosh", "Apple_Driver_ATA", 0, 0, 0x37,
         0, 0, 0, 0, 0, 0, 0, {0}, {0}, {0}},
       { APPLE_PART_MAP_ENTRY_MAGIC, 0, 6, 224, 4096,
         "untitled", "Apple_HFS", 0, 0, 0x37,
         0, 0, 0, 0, 0, 0, 0, {0}, {0}, {0}},
       { APPLE_PART_MAP_ENTRY_MAGIC, 0, 6,4320, 0,
         "untitled", "Apple_Free", 0, 0, 0x37,
         0, 0, 0, 0, 0, 0, 0, {0}, {0}, {0}}
};

void
md_init(void)
{
}

void
md_init_set_status(int flags)
{
      struct utsname instsys;

       (void)flags;

       /*
        * Get the name of the Install Kernel we are running under and
        * enable the installation of the corresponding GENERIC kernel.
        *
        * Note:  In md.h the two kernels are disabled.  If they are
        *        enabled there the logic here needs to be switched.
        */
       uname(&instsys);
       if (strstr(instsys.version, "(INSTALLSBC)"))
               /*
                * Running the SBC Installation Kernel, so enable GENERICSBC
                */
               set_kernel_set(SET_KERNEL_2);
       else
               /*
                * Running the GENERIC Installation Kernel, so enable GENERIC
                */
               set_kernel_set(SET_KERNEL_1);
}

bool
md_get_info(struct install_partition_desc *install)
{
       struct disklabel disklabel;
       int fd, i;
       char dev_name[100];
       struct apple_part_map_entry block;

       snprintf(dev_name, sizeof(dev_name), "/dev/r%s%c",
               pm->diskdev, 'a' + getrawpartition());

       /*
        * Open the disk as a raw device
        */
       fd = open(dev_name, O_RDONLY, 0);
       if (fd < 0) {
               endwin();
               fprintf (stderr, "Can't open %s\n", dev_name);
               exit(1);
       }
       /*
        * Try to get the default disklabel info for the device
        */
       if (ioctl(fd, DIOCGDINFO, &disklabel) == -1) {
               endwin();
               fprintf (stderr, "Can't read disklabel on %s\n", dev_name);
               close(fd);
               exit(1);
       }
       /*
        * Get the disk parameters from the disk driver.  It should have
        *  obtained them by querying the disk itself.
        */
       blk_size = disklabel.d_secsize;
       pm->dlcyl = disklabel.d_ncylinders;
       pm->dlhead = disklabel.d_ntracks;
       pm->dlsec = disklabel.d_nsectors;
       /*
        * Just in case, initialize the structures we'll need if we
        *  need to completely initialize the disk.
        */
       pm->dlsize = disklabel.d_secperunit;
/*
* XXX this code is broken: it accesses off the end of new_map[],
* because NEW_MAP_SIZE is substantially larger than the number of
* entries in new_map[]. Based on the description of struct
* apple_part_map_entry in sys/bootblock.h, and the usage of it in
* new_map[], NEW_MAP_SIZE is expected to be a block count, not an
* entry count. As far I can tell the logic here is just wrong; it
* needs someone with platform knowledge to sort it out.
*
* Note that nothing uses the data this writes into new_map[] so
* disabling it should have no adverse consequences.
*
*   - dholland 20160530
*/
#if 0 /* XXX broken */
       for (i=0;i<NEW_MAP_SIZE;i++) {
          if (i > 0)
               new_map[i].pmPyPartStart = new_map[i-1].pmPyPartStart +
                       new_map[i-1].pmPartBlkCnt;
          new_map[i].pmDataCnt = new_map[i].pmPartBlkCnt;
          if (new_map[i].pmPartBlkCnt == 0) {
               new_map[i].pmPartBlkCnt = pm->dlsize;
               new_map[i].pmDataCnt = pm->dlsize;
               break;
          }
          pm->dlsize -= new_map[i].pmPartBlkCnt;
       }
#endif /* 0 - broken */
       pm->dlsize = disklabel.d_secperunit;
#if 0
       msg_fmt_display(MSG_dldebug, "%d%d%d%d%d",
           blk_size, pm->dlcyl, pm->dlhead, pm->dlsec, pm->dlsize);
       process_menu(MENU_ok, NULL);
#endif
       map.size = 0;
       /*
        * Verify the disk has been initialized for MacOS use by checking
        *  to see if the disk have a Boot Block
        */
       if (lseek(fd, (off_t)0 * blk_size, SEEK_SET) < 0 ||
           read(fd,  &block, sizeof(block)) - sizeof(block) != 0 ||
           block.pmSig != 0x4552) {
            process_menu(MENU_nodiskmap, NULL);
       }
       else {
          /*
           * Scan for the Partition Map entry that describes the Partition
           *  Map itself.  We need to know the number of blocks allocated
           *  to it and the number currently in use.
           */
          for (i=0;i<MAXMAXPARTITIONS;i++) {
               lseek(fd, (off_t)(i+1) * blk_size, SEEK_SET);
               read(fd, &block, sizeof(block));
               if (stricmp("Apple_partition_map", (char *)block.pmPartType) == 0) {
                   map.size = block.pmPartBlkCnt;
                   map.in_use_cnt = block.pmMapBlkCnt;
                   map.blk = (struct apple_part_map_entry *)malloc(map.size * blk_size);
                   break;
               }
           }
           lseek(fd, (off_t)1 * blk_size, SEEK_SET);
           read(fd, map.blk, map.size * blk_size);
       }
       close(fd);

       return edit_diskmap();
}

/*
* md back-end code for menu-driven BSD disklabel editor.
*/
int
md_make_bsd_partitions(struct install_partition_desc *install)
{
       int i, j, rv;
       EBZB *bzb;
       struct disk_part_info info;
       uint fs_type;
       const char *mountpoint;
       part_id pid;
       size_t ndx;

       /*
        * Scan for any problems and report them before continuing.
        *  The user can abort installation and we'll take them back
        *  to the main menu; continue ignoring the warnings, or
        *  ask to reedit the Disk Partition Map.
        */
       while (1) {
           if (check_for_errors()) {
               process_menu (MENU_sanity, &rv);
               if (rv < 0)
                   return 0;
               else if (rv)
                   break;
               edit_diskmap();
           } else
               break;
       }

       /* Start with empty fake disklabel partitions */
       pm->parts->pscheme->delete_all_partitions(pm->parts);

       /*
        * Now, scan through the Disk Partition Map and transfer the
        *  information into the incore disklabel.
        */
       for (i=0;i<map.usable_cnt;i++) {
           j = map.mblk[i];
           bzb = (EBZB *)&map.blk[j].pmBootArgs[0];
           if (bzb->flags.part) {
               mountpoint = NULL;
               fs_type = FS_UNUSED;
               switch (whichType(&map.blk[j])) {
                   case HFS_PART:
                       fs_type = FS_HFS;
                       mountpoint = (const char*)bzb->mount_point;
                       break;
                   case ROOT_PART:
                   case UFS_PART:
                       fs_type = FS_BSDFFS;
                       mountpoint = (const char*)bzb->mount_point;
                       break;
                   case SWAP_PART:
                       fs_type = FS_SWAP;
                       break;
                   case SCRATCH_PART:
                       fs_type = FS_OTHER;
                       mountpoint = (const char*)bzb->mount_point;
                       break;
                   default:
                       continue;
               }
               if (fs_type != FS_UNUSED) {
                       memset(&info, 0, sizeof info);
                       info.start = map.blk[j].pmPyPartStart;
                       info.size = map.blk[j].pmPartBlkCnt;
                       info.fs_type = fs_type;
                       info.last_mounted = mountpoint;
                       info.nat_type = pm->parts->pscheme->get_fs_part_type(
                           PT_root, fs_type, 0);
                       pid = pm->parts->pscheme->add_outer_partition(pm->parts,
                           &info, NULL);
                       if (pid == NO_PART)
                               return 0;
               }
           }
       }

       /* Disk name  - don't bother asking, just use the physical name*/
       pm->parts->pscheme->set_disk_pack_name(pm->parts, pm->diskdev);

       /* Write the converted partitions */
       if (!pm->parts->pscheme->write_to_disk(pm->parts))
               return 0;

       /* now convert to install info */
       if (!install_desc_from_parts(install, pm->parts))
               return 0;

       /* set newfs flag for all FFS partitions */
       for (ndx = 0; ndx < install->num; ndx++) {
               if (install->infos[ndx].fs_type == FS_BSDFFS &&
                   install->infos[ndx].size > 0 &&
                   (install->infos[ndx].instflags & PUIINST_MOUNT))
                       install->infos[ndx].instflags |= PUIINST_NEWFS;
       }

       return 1;
}

/*
* any additional partition validation
*/
bool
md_check_partitions(struct install_partition_desc *install)
{
       return true;
}

/*
* hook called before writing new disklabel.
*/
bool
md_pre_disklabel(struct install_partition_desc *install,
   struct disk_partitions *parts)
{
   int fd;
   char dev_name[100];
   struct disklabel lp;
   Block0 new_block0 = {APPLE_DRVR_MAP_MAGIC, 512,
                        0, 0, 0, 0, 0, 0, 0, 0, {0}};

   /*
    * Danger Will Robinson!  We're about to turn that nice MacOS disk
    *  into an expensive doorstop...
    */
   printf ("%s", msg_string (MSG_dodiskmap));

   snprintf (dev_name, sizeof(dev_name), "/dev/r%sc", pm->diskdev);
   /*
    * Open the disk as a raw device
    */
   if ((fd = open(dev_name, O_WRONLY, 0)) < 0) {
       endwin();
       fprintf(stderr, "Can't open %s to rewrite the Disk Map\n", dev_name);
       exit (1);
   }
   /*
    *  First check the pmSigPad field of the first block in the incore
    *  Partition Map.  It should be zero, but if it's 0xa5a5 that means
    *  we need to write out Block0 too.
    */
   if (map.blk[0].pmSigPad == 0xa5a5) {
       if (lseek (fd, (off_t)0 * blk_size, SEEK_SET) < 0) {
           endwin();
           fprintf (stderr, "Can't position to write Block0\n");
           close (fd);
           exit (1);
       }
       new_block0.sbBlkCount = pm->dlsize;             /* Set disk size */
       if (write (fd, &new_block0, blk_size) != blk_size) {
           endwin();
           fprintf (stderr, "I/O error writing Block0\n");
           close (fd);
           exit (1);
       }
       map.blk[0].pmSigPad = 0;
   }
   if (lseek (fd, (off_t)1 * blk_size, SEEK_SET) < 0) {
       endwin();
       fprintf (stderr, "Can't position disk to rewrite Disk Map\n");
       close (fd);
       exit (1);
   }
   if (write (fd, map.blk, map.size * blk_size) != (map.size * blk_size)) {
       endwin();
       fprintf(stderr, "I/O error writing Disk Map\n");
       close (fd);
       exit (1);
   }
   fsync(fd);
   /*
    * Well, if we get here the dirty deed has been done.
    *
    * Now we need to force the incore disk table to get updated. This
    * should be done by disklabel -- which is normally called right after
    * we return -- but may be commented out for the mac68k port. We'll
    * instead update the incore table by forcing a dummy write here. This
    * relies on a change in the mac68k-specific write_disklabel() routine.
    * If that change doesn't exist nothing bad happens here. If disklabel
    * properly updates the ondisk and incore labels everything still
    * works. Only if we fail here and if disklabel fails are we in
    * in a state where we've updated the disk but not the incore and
    * a reboot is necessary.
    *
    * First, we grab a copy of the incore label as it existed before
    * we did anything to it. Then we invoke the "write label" ioctl to
    * rewrite it to disk. As a result, the ondisk partition map is
    * re-read and the incore label is reconstructed from it. If
    * disklabel() is then called to update again, either that fails
    * because the mac68k port doesn't support native disklabels, or it
    * succeeds and writes out a new ondisk copy.
    */
   ioctl(fd, DIOCGDINFO, &lp);    /* Get the current disk label */
   ioctl(fd, DIOCWDINFO, &lp);    /* Write it out again */

   close (fd);
   return true;
}

/*
* hook called after writing disklabel to new target disk.
*/
bool
md_post_disklabel(struct install_partition_desc *install,
   struct disk_partitions *parts)
{
#if 0   // XXX
   struct disklabel updated_label;
   int fd, i, no_match;
   char dev_name[100], buf[80];
   const char *fst[] = {"free", "swap", " v6 ", " v7 ", "sysv", "v71k",
                       " v8 ", "ffs ", "dos ", "lfs ", "othr", "hpfs",
                       "9660", "boot", "ados", "hfs ", "fcor", "ex2f",
                       "ntfs", "raid", "ccd "};

   snprintf(dev_name, sizeof(dev_name), "/dev/r%sc", pm->diskdev);
   /*
    * Open the disk as a raw device
    */
   if ((fd = open(dev_name, O_RDONLY, 0)) < 0)
      return false;
   /*
    * Get the "new" label to see if we were successful.  If we aren't
    *  we'll return an error to keep from destroying the user's disk.
    */
   ioctl(fd, DIOCGDINFO, &updated_label);
   close(fd);
   /*
    * Make sure the in-core label matches the on-disk one
    */
   no_match = 0;
   for (i=0;i<MAXPARTITIONS;i++) {
       if (i > updated_label.d_npartitions)
          break;
       if (pm->bsdlabel[i].pi_size != updated_label.d_partitions[i].p_size)
          no_match = 1;
       if (pm->bsdlabel[i].pi_size) {
          if (pm->bsdlabel[i].pi_offset != updated_label.d_partitions[i].p_offset)
              no_match = 1;
          if (pm->bsdlabel[i].pi_fstype != updated_label.d_partitions[i].p_fstype)
              no_match = 1;
       }
       if (no_match)
          break;
   }
   /*
    * If the labels don't match, tell the user why
    */
   if (no_match) {
      msg_clear();
      msg_display(MSG_label_error);
      msg_fmt_table_add(MSG_dump_line, "%s",
          " in-core: offset      size type on-disk: offset      size type");
      for (i=0;i<MAXPARTITIONS;i++) {
          snprintf(buf, sizeof(buf),
             " %c:%13.8x%10.8x%5s%16.8x%10.8x%5s", i+'a',
             pm->bsdlabel[i].pi_offset, pm->bsdlabel[i].pi_size,
             fst[pm->bsdlabel[i].pi_fstype],
             updated_label.d_partitions[i].p_offset,
             updated_label.d_partitions[i].p_size,
             fst[updated_label.d_partitions[i].p_fstype]);
          msg_fmt_table_add(MSG_dump_line, "%s", buf);
      }
      process_menu(MENU_ok2, NULL);
   }
   return no_match == 0;
#else
       return true;
#endif
}

/*
* hook called after upgrade() or install() has finished setting
* up the target disk but immediately before the user is given the
* ``disks are now set up'' message.
*/
int
md_post_newfs(struct install_partition_desc *install)
{
       return 0;
}

int
md_post_extract(struct install_partition_desc *install, bool upgrade)
{
       return 0;
}

void
md_cleanup_install(struct install_partition_desc *install)
{
#ifndef DEBUG
       enable_rc_conf();
#endif
}

int
md_pre_update(struct install_partition_desc *install)
{
       return 1;
}

/* Upgrade support */
int
md_update(struct install_partition_desc *install)
{
       md_post_newfs(install);
       return 1;
}

/*
* Compare lexigraphically two strings
*/
static int
stricmp(s1, s2)
       const char *s1;
       const char *s2;
{
       char c1, c2;

       while (1) {
           c1 = tolower((unsigned char)*s1++);
           c2 = tolower((unsigned char)*s2++);
           if (c1 < c2) return -1;
           if (c1 > c2) return 1;
           if (c1 == 0) return 0;
       }
}

static void
setpartition(part, in_use, slot)
       struct apple_part_map_entry *part;
       char in_use[];
       int slot;
{
       EBZB *bzb;

       bzb = (EBZB *)&part->pmBootArgs[0];
       in_use[slot] = 1;
       bzb->flags.used = 1;
       bzb->flags.part = 'a' + slot;
}

/*
* Find an entry in a use array that is unused and return it or
*  -1 if no entry is available
*/
static int
getFreeLabelEntry(slots)
       char *slots;
{
       int i;

       for ( i = 0; i < MAXPARTITIONS; i++) {
               if (i != RAW_PART && slots[i] == 0)
                       return i;
       }
       return -1;
}

/*
* Figure out what type type of the given partition is and return it.
*/
int
whichType(part)
       struct apple_part_map_entry *part;
{
       MAP_TYPE *map_entry = (MAP_TYPE *)&map_types;
       EBZB *bzb;
       char partyp[32];
       int type, maxsiz, entry_type = MAP_OTHER;

       bzb = (EBZB *)&part->pmBootArgs[0];
       if (part->pmSig != APPLE_PART_MAP_ENTRY_MAGIC)
           return 0;
       maxsiz = sizeof(part->pmPartType);
       if (maxsiz > (int)sizeof(partyp))
           maxsiz = sizeof(partyp);
       strncpy(partyp, (char *)part->pmPartType, maxsiz);
       partyp[maxsiz-1] = '\0';

       /*
        * Find out how to treat the partition type under NetBSD
        */
       while (map_entry->type != MAP_EOL) {
           if (stricmp(map_entry->name, partyp) == 0) {
               entry_type = map_entry->type;
               break;
           }
           map_entry++;
       }

       /*
        * Now classify the use for NetBSD
        */
       if (entry_type == MAP_RESERVED)
               type = 0;
       else if (entry_type == MAP_NETBSD) {
           if (bzb->magic != APPLE_BZB_MAGIC)
               type = 0;
           else if (bzb->type == APPLE_BZB_TYPEFS) {
               if (bzb->flags.root)
                   type = ROOT_PART;
               else if (bzb->flags.usr)
                   type = UFS_PART;
               else
                   type = SCRATCH_PART;
           } else if (bzb->type == APPLE_BZB_TYPESWAP)
               type = SWAP_PART;
           else
               type = SCRATCH_PART;
       } else if (entry_type == MAP_MACOS)
           type = HFS_PART;
       else
           type = SCRATCH_PART;
       return type;
}

static char *
getFstype(part, len_type, type)
       struct apple_part_map_entry *part;
       int len_type;
       char *type;
{
       *type = '\0';
       switch(whichType(part)) {
           case ROOT_PART:
           case UFS_PART:
               strncpy(type, "4.2BSD", len_type);
               break;
           case SWAP_PART:
               strncpy(type, "swap", len_type);
               break;
           case HFS_PART:
               strncpy(type, "HFS", len_type);
               break;
           case SCRATCH_PART:
           default:
               break;
       }
       return (type);
}

static char *
getUse(part, len_use, use)
       struct apple_part_map_entry *part;
       int len_use;
       char *use;
{
       EBZB *bzb;
       char partyp[32];

       *use = '\0';
       bzb = (EBZB *)&part->pmBootArgs[0];
       switch(whichType(part)) {
           case ROOT_PART:
               if (bzb->flags.usr)
                   strncpy(use, "Root&Usr", len_use);
               else
                   strncpy(use, "Root", len_use);
               break;
           case UFS_PART:
               strncpy(use, "Usr", len_use);
               break;
           case SWAP_PART:
               break;
           case HFS_PART:
               strncpy(use, "MacOS", len_use);
               break;
           case SCRATCH_PART:
               strncpy(partyp, (char *)part->pmPartType, sizeof(partyp));
               partyp[sizeof(partyp)-1] = '\0';
               if (stricmp("Apple_Free", partyp) == 0)
                   strncpy(use, "Free", len_use);
               else if (stricmp("Apple_Scratch", partyp) == 0)
                   strncpy(use, "Scratch", len_use);
               else if (stricmp("Apple_MFS", partyp) == 0)
                   strncpy(use, "MFS", len_use);
               else if (stricmp("Apple_PRODOS", partyp) == 0)
                   strncpy(use, "PRODOS", len_use);
               else
                   strncpy(use, "unknown", len_use);
           default:
               break;
       }
       return(use);
}

static char *
getName(part, len_name, name)
       struct apple_part_map_entry *part;
       int len_name;
       char *name;
{
       EBZB *bzb;
       int fd;
       off_t seek;
       char dev_name[100], macosblk[512];

       *name = '\0';
       bzb = (EBZB *)&part->pmBootArgs[0];
       switch(whichType(part)) {
           case SCRATCH_PART:
           case ROOT_PART:
           case UFS_PART:
               strncpy(name, (char *)bzb->mount_point, len_name);
               break;
           case SWAP_PART:
               break;
           case HFS_PART:
               /*
                * OK, this is stupid but it's damn nice to know!
                */
               snprintf (dev_name, sizeof(dev_name), "/dev/r%sc", pm->diskdev);
               /*
                * Open the disk as a raw device
                */
               if ((fd = open(dev_name, O_RDONLY, 0)) >= 0) {
                   seek = (off_t)part->pmPyPartStart + (off_t)2;
                   seek *= (off_t)blk_size;
                   lseek(fd, seek, SEEK_SET);
                   read(fd, &macosblk, sizeof(macosblk));
                   macosblk[37+32] = '\0';
                   strncpy(name, (char *)bzb->mount_point, len_name);
                   strncat(name, " (", len_name-strlen(name));
                   strncat(name, &macosblk[37], len_name-strlen(name));
                   strncat(name, ")", len_name-strlen(name));
                   close(fd);
               }
               break;
           default:
               break;
       }
       return(name);
}

/*
* Find the first occurrence of a Standard Type partition and
*  mark it for use along with the default mount slot.
*/
static int
findStdType(num_parts, in_use, type, count, alt)
       int num_parts;
       char in_use[];
       int type;
       int *count;
       int alt;
{
       EBZB *bzb;
       int i;

       for (i = 0; i < num_parts; i++) {
               bzb = (EBZB *)&map.blk[i].pmBootArgs[0];
               if (whichType(&map.blk[i]) != type || bzb->flags.used)
                       continue;
               if (type == ROOT_PART) {
                       if (alt >= 0 && alt != bzb->cluster)
                               continue;
                       setpartition(&map.blk[i], in_use, 0);
                       strcpy ((char *)bzb->mount_point, "/");
                       *count += 1;
               } else if (type == UFS_PART) {
                       if (alt >= 0 && alt != bzb->cluster)
                               continue;
                       setpartition(&map.blk[i], in_use, 6);
                       if (bzb->mount_point[0] == '\0')
                               strcpy ((char *)bzb->mount_point, "/usr");
                       *count += 1;
               } else if (type == SWAP_PART) {
                       setpartition(&map.blk[i], in_use, 1);
                       *count += 1;
               }
               return 0;
       }
       return -1;
}

/*
* Reset the flags and reserved fields in the selected partition.
* This functions isn't called to process any of the reserved partitions
* where the boot code for MacOS is stored, so (hopefully) we won't
* do more damage that we're trying to avoid.  Eventually the NetBSD
* Boot Code will need to go into a partition too, but that should go
* into a reserved partition as well.  I'd suggest using a partition
* named something like "NetBSD_Boot" with a pmPartName of "Macintosh".
* The Apple Start Manager (in ROM) will then recognize the partition
* as the one containing the system bootstrip for the volume.
*/
void
reset_part_flags(part)
       struct apple_part_map_entry *part;
{
       EBZB *bzb;

       /*
        * Clear out the MacOS fields that might be used for booting just
        *  in case we've clobbered the boot code.
        */
       part->pmLgDataStart = 0;
       part->pmPartStatus = 0x77;  /* make sure the partition shows up */
       part->pmLgBootStart = 0;
       part->pmBootSize = 0;
       part->pmBootLoad = 0;
       part->pmBootLoad2 = 0;
       part->pmBootEntry = 0;
       part->pmBootEntry2 = 0;
       part->pmBootCksum = 0;

       /*
        * Clear out all the NetBSD fields too.  We only clear out the ones
        *  that should get reset during our processing.
        */
       bzb = (EBZB *)&part->pmBootArgs[0];
       bzb->magic = 0;
       bzb->cluster = 0;
       bzb->inode = 0;
       bzb->type = 0;
       bzb->flags.root = 0;
       bzb->flags.usr = 0;
       bzb->flags.crit = 0;
       bzb->flags.slice = 0;
       bzb->flags.used = 0;
       return;
}

/*
* sortmerge:
*  1) Moves all the Partition Map entries to the front of the Map.
*     This is required because some disk formatters leave holes.
*  2) Sorts all entries by ascending start block number.
*     Needed so the NetBSD algorithm for finding partitions works
*     consistently from a user perspective.
*  3) Collapse multiple adjected "free" entries into a single entry.
*  4) Identify the NetBSD mount_points.
*/
void
sortmerge(void)
{
   struct apple_part_map_entry tmp_blk;
   char in_use[MAXPARTITIONS];
   int i, j;
   EBZB *bzb;

   /*
    * Step 1, squeeze out the holes that some disk formatters leave in
    *  the Map.  Also convert any "old" Map entries to the new entry
    *  type. Also clear out our used flag which is used to indicte
    *  we've mapped the partition.
    */
   map.in_use_cnt = 0;
   for (i=0;i<map.size-1;i++) {
       if (map.blk[i].pmSig == 0x5453)
           map.blk[i].pmSig = APPLE_PART_MAP_ENTRY_MAGIC;
       if (map.blk[i].pmSig != APPLE_PART_MAP_ENTRY_MAGIC) {
           for (j=i+1;j<map.size;j++) {
               if (map.blk[j].pmSig == 0x5453)
                   map.blk[j].pmSig = APPLE_PART_MAP_ENTRY_MAGIC;
               if (map.blk[j].pmSig == APPLE_PART_MAP_ENTRY_MAGIC) {
                   memcpy (&map.blk[i], &map.blk[j], sizeof(map.blk[i]));
                   map.blk[j].pmSig = 0;
                   break;
               }
           }
       } else {
           map.in_use_cnt += 1;
           bzb = (EBZB *)&map.blk[i].pmBootArgs[0];
           bzb->flags.used = 0;
           bzb->flags.part = 0;
       }
   }

   /*
    * Step 2, sort by ascending starting block number.  Since
    *         we've already removed the holes we only need to
    *         deal with the in_use count of blocks.
    */
   for (i=0;i<map.in_use_cnt-1;i++) {
       for (j=i+1;j<map.in_use_cnt;j++) {
           if (map.blk[i].pmPyPartStart > map.blk[j].pmPyPartStart) {
               memcpy (&tmp_blk, &map.blk[i], sizeof(tmp_blk));
               memcpy (&map.blk[i], &map.blk[j], sizeof(map.blk[i]));
               memcpy (&map.blk[j], &tmp_blk, sizeof(map.blk[j]));
           }
       }
   }

   /*
    * Step 3, merge adjacent free space
    */
   for (i=0;i<map.in_use_cnt-1;i++) {
       if (stricmp("Apple_Free", (char *)map.blk[i].pmPartType) == 0 &&
           stricmp("Apple_Free", (char *)map.blk[i+1].pmPartType) == 0) {
           map.blk[i].pmPartBlkCnt += map.blk[i+1].pmPartBlkCnt;
           map.blk[i].pmDataCnt += map.blk[i+1].pmDataCnt;
           map.blk[i+1].pmSig = 0;
           for (j=i+1;j<map.in_use_cnt-1;j++) {
               memcpy (&map.blk[j], &map.blk[j+1], sizeof(map.blk[j]));
               map.blk[j+1].pmSig = 0;
           }
           map.in_use_cnt -= 1;
       }
   }

   /*
    * Step 4, try to identify the mount points for the partitions
    *         and adjust the pmMapBlkCnt in each Map entry.  Set
    *         up the display array for the non-reserved partitions,
    *         and count the number of NetBSD usable partitions
    */
   map.hfs_cnt = 0;
   map.root_cnt = 0;
   map.swap_cnt = 0;
   map.usr_cnt = 0;
   map.usable_cnt = 0;
   /*
    * Clear out record of partition slots already in use
    */
   memset(&in_use, 0, sizeof(in_use));
   for (i=0,j=0;i<map.in_use_cnt;i++) {
       map.blk[i].pmSig = APPLE_PART_MAP_ENTRY_MAGIC;
       map.blk[i].pmMapBlkCnt = map.in_use_cnt;
       /*
        * Since MAXPARTITIONS == 8 for mac68k, and we do not display
        * the c partition, we only need 7 partition slots on the screen.
        * If/when MAXPARTITIONS is changed, the "Edit Disk Partition Map"
        * needs to be a scrollable view of the partition table.
        */
       if (whichType(&map.blk[i]) && (j < MAXPARTITIONS - 1)) {
               map.mblk[j++] = i;
               map.usable_cnt += 1;
       }
   }
   /*
    * Fill in standard partitions.  Look for a Cluster "0" first and use
    *  it, otherwise take any Cluster value.
    */
   if (findStdType(map.in_use_cnt, in_use, ROOT_PART, &map.root_cnt, 0))
       findStdType(map.in_use_cnt, in_use, ROOT_PART, &map.root_cnt, -1);
   if (findStdType(map.in_use_cnt, in_use, UFS_PART, &map.usr_cnt, 0))
       findStdType(map.in_use_cnt, in_use, UFS_PART, &map.usr_cnt, -1);
   if (findStdType(map.in_use_cnt, in_use, SWAP_PART, &map.swap_cnt, 0))
       findStdType(map.in_use_cnt, in_use, SWAP_PART, &map.swap_cnt, -1);

#ifdef MD_DEBUG_SORT_MERGE
       md_debug_dump("After marking Standard Types");
#endif
   /*
    * Now fill in the remaining partitions counting them by type and
    *  assigning them the slot the where the kernel should map them.
    * This will be where they are displayed in the Edit Map.
    */
   for (i=0; i < map.in_use_cnt; i++) {
       bzb = (EBZB *)&map.blk[i].pmBootArgs[0];
       if (bzb->flags.used == 0) {
           if ((j = getFreeLabelEntry(in_use)) < 0)
               break;
           switch (whichType(&map.blk[i])) {
               case ROOT_PART:
                   map.root_cnt += 1;
                   setpartition(&map.blk[i], in_use, j);
                   break;
               case UFS_PART:
                   map.usr_cnt += 1;
                   setpartition(&map.blk[i], in_use, j);
                   break;
               case SWAP_PART:
                   map.swap_cnt += 1;
                   setpartition(&map.blk[i], in_use, j);
                   break;
               case HFS_PART:
                   map.hfs_cnt += 1;
                   setpartition(&map.blk[i], in_use, j);
                   break;
               case SCRATCH_PART:
                   setpartition(&map.blk[i], in_use, j);
               default:
                   break;
           }
       }
   }
#ifdef MD_DEBUG_SORT_MERGE
       md_debug_dump("After sort merge");
#endif
   return;
}

void
disp_selected_part(sel)
       int sel;
{
       int i,j;
       char fstyp[16], use[16], name[32];
       EBZB *bzb;

       msg_table_add(MSG_mac68k_part_header);
       for (i=0;i<map.usable_cnt;i++) {
           if (i == sel) msg_standout();
           j = map.mblk[i];
           getFstype(&map.blk[j], sizeof(fstyp), fstyp);
           getUse(&map.blk[j], sizeof(use), use);
           getName(&map.blk[j], sizeof(name), name);
           bzb = (EBZB *)&map.blk[j].pmBootArgs[0];
           msg_fmt_table_add(MSG_part_row, "%s%c%d%d%s%s%s",
               pm->diskdev,
               bzb->flags.part, map.blk[j].pmPyPartStart,
                map.blk[j].pmPartBlkCnt, fstyp, use, name);
           if (i == sel) msg_standend();
       }
       return;
}

/*
* check for any anomalies on the requested setup
*/
static int
check_for_errors()
{
   int i, j;
   int errs = 0;

   errs = (!map.root_cnt) || (map.root_cnt > 1) || (!map.swap_cnt) ||
          (map.swap_cnt > 1);

   for (i=0;i<map.usable_cnt;i++) {
       j = map.mblk[i];
       if (map.blk[j].pmPyPartStart > pm->dlsize)
               errs++;
       if ((map.blk[j].pmPyPartStart + map.blk[j].pmPartBlkCnt) > pm->dlsize + 1)
               errs++;
   }
   return(errs);
}

/*
* check for and report anomalies on the requested setup
*/
void
report_errors()
{
   int i, j;
   int errs = 0;
   EBZB *bzb;

   if (!map.root_cnt) {
       msg_display_add(MSG_disksetup_no_root);
       errs++;
   }
   if (map.root_cnt > 1) {
       msg_display_add(MSG_disksetup_multiple_roots);
       errs++;
   }
   if (!map.swap_cnt) {
       msg_display_add(MSG_disksetup_no_swap);
       errs++;
   }
   if (map.swap_cnt > 1) {
       msg_display_add(MSG_disksetup_multiple_swaps);
       errs++;
   }
   for (i=0;i<map.usable_cnt;i++) {
       j = map.mblk[i];
       if (map.blk[j].pmPyPartStart > pm->dlsize) {
           bzb = (EBZB *)&map.blk[j].pmBootArgs[0];
           msg_fmt_display_add(MSG_disksetup_part_beginning, "%s%c",
               pm->diskdev, bzb->flags.part);
           errs++;
       }
       if ((map.blk[j].pmPyPartStart + map.blk[j].pmPartBlkCnt) > pm->dlsize) {
           bzb = (EBZB *)&map.blk[j].pmBootArgs[0];
           msg_fmt_display_add(MSG_disksetup_part_size, "%s%c",
               pm->diskdev, bzb->flags.part);
           errs++;
       }
   }
   if (!errs)
       msg_display_add(MSG_disksetup_noerrors);
   return;
}

static int
edit_diskmap(void)
{
   int i;

       /* Ask full/part */
       msg_fmt_display (MSG_mac68k_fullpart, "%s", pm->diskdev);
       process_menu (MENU_fullpart, NULL);

       map.selected = 0;
       sortmerge();

       /* If blowing away the whole disk, let user know if there
        *  are any active disk partitions */
       if (usefull) {
           if (map.usable_cnt > (map.root_cnt+map.swap_cnt+map.usr_cnt)) {
               msg_display (MSG_mac68k_ovrwrite);
               if (!ask_noyes(NULL)) {
                       endwin();
                       return 0;
               }
           }
           /*
            * mark all non-reserved partitions as "free"
            *  then sort and merge the map into something sensible
            */
           for (i=0;i<map.size;i++)
               if (whichType(&map.blk[i]))
                   strcpy ((char *)map.blk[i].pmPartType, "Apple_Free");
           sortmerge();
       }
       process_menu (MENU_editparttable, NULL);
       return 1;
}

#ifdef MD_DEBUG_SORT_MERGE
static int
md_debug_dump(title)
       char *title;
{
       char buf[96], type;
       char fstyp[16], use[16], name[64];
       int i, j, rv;
       EBZB *bzb;

       msg_clear();
       sprintf(buf, "Apple Disk Partition Map: %s", title);
       msg_fmt_table_add(MSG_dump_line, "%s", buf);
       msg_fmt_table_add(MSG_dump_line, "%s",
          "slot      base   fstype        use name");
       for (i=0;i<map.in_use_cnt;i++) {
          j = whichType(&map.blk[i]);
          getFstype(&map.blk[i], sizeof(fstyp), fstyp);
          getUse(&map.blk[i], sizeof(use), use);
          getName(&map.blk[i], sizeof(name), name);
          bzb = (EBZB *) &map.blk[i].pmBootArgs[0];
          type = bzb->flags.part;
          if (type < 'a' || type > 'h') type = '?';
          if (j == 0) strcpy (name, "reserved for Apple");
          sprintf(buf, " %02d:%c %08x %8s %10s %s", i+1,  type,
               map.blk[i].pmPyPartStart, fstyp, use, name);
          msg_fmt_table_add(MSG_dump_line, "%s", buf);
       }
       process_menu(MENU_okabort, &rv);
       msg_clear();
       return rv;
}
#endif /* MD_DEBUG_SORT_MERGE */

int
md_pre_mount(struct install_partition_desc *install, size_t ndx)
{
       return 0;
}

bool
md_parts_use_wholedisk(struct disk_partitions *parts)
{
       return parts_use_wholedisk(parts, 0, NULL);
}

#ifdef HAVE_GPT
bool
md_gpt_post_write(struct disk_partitions *parts, part_id root_id,
   bool root_is_new, part_id efi_id, bool efi_is_new)
{
       /* no GPT boot support, nothing needs to be done here */
       return true;
}
#endif