/*      $NetBSD: check.c,v 1.6 2009/03/14 21:04:06 dsl Exp $    */

/*
* Copyright (c) 1999 The NetBSD Foundation, Inc.
* All rights reserved.
*
* This code is derived from software contributed to The NetBSD Foundation
* by Julian Coleman, Waldi Ravens and Leo Weppelman.
*
* 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.
*
* THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
* ``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 THE FOUNDATION OR CONTRIBUTORS
* 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.
*/

#include "privahdi.h"

/* Partitions which have errors */
int     ahdi_errp1;
int     ahdi_errp2;

/*
* Check partitions for consistency :
*      GEM vs BGM
*      Start sector > 2
*      End sector < secperunit
*      other partition overlap
*      auxiliary root overlap
*      Number of partitions in root/auxiliary root
*/
int
ahdi_checklabel (struct ahdi_ptable *ptable)
{
       int             i, j, rcount, acount;
       u_int32_t       i_end, j_end;

       ahdi_errp1 = -1;
       ahdi_errp2 = -1;

       if (ptable->nparts < 1 || ptable->nparts > MAXPARTITIONS)
               return (-2);

       rcount = 0;
       acount = 0;

       for (i = 0; i < ptable->nparts; i++) {

               /* GEM vs BGM */
               if (ptable->parts[i].size > 32768) {
                       if (AHDI_MKPID (ptable->parts[i].id[0],
                           ptable->parts[i].id[1], ptable->parts[i].id[2])
                           == AHDI_PID_GEM) {
                               ahdi_errp1 = i;
                               return (-3);
                       }
               } else {
                       if (AHDI_MKPID (ptable->parts[i].id[0],
                           ptable->parts[i].id[1], ptable->parts[i].id[2])
                           == AHDI_PID_BGM) {
                               ahdi_errp1 = i;
                               return (-3);
                       }
               }

               /* Need 2 free sectors at start for root and bad sector list */
               if (ptable->parts[i].start < 2) {
                       ahdi_errp1 = i;
                       return (-4);
               }

               i_end = ptable->parts[i].start + ptable->parts[i].size - 1;

               /* Check partition does not extend past end of disk */
               if (i_end >= ptable->secperunit) {
                       ahdi_errp1 = i;
                       return (-5);
               }

               for (j = i + 1; j < ptable->nparts; j++) {
                       /* Check for overlap with other partitions */
                       j_end = ptable->parts[j].start + ptable->parts[j].size
                           - 1;
                       if ((ptable->parts[j].start >= ptable->parts[i].start
                           && ptable->parts[j].start <= i_end) ||
                           (j_end >= ptable->parts[i].start &&
                            j_end <= i_end)) {
                               ahdi_errp1 = i;
                               ahdi_errp2 = j;
                               return (-6);
                       }
                       /* Check number of partitions in auxiliary root */
                       if (ptable->parts[i].root &&
                           ptable->parts[i].root == ptable->parts[j].root) {
                               ahdi_errp1 = i;
                               ahdi_errp2 = j;
                               return (-9);
                       }
               }

               for (j = i; j < ptable->nparts; j++)
                       /* Check for overlap with auxiliary root(s) */
                       if (ptable->parts[j].root >= ptable->parts[i].start &&
                           ptable->parts[j].root <= i_end) {
                               ahdi_errp1 = i;
                               return (-7);
                       }

               /* Count partitions in root/auxiliary roots */
               if (ptable->parts[i].root)
                       acount++;
               else
                       rcount++;

       }
       /* Check number of partitions in root sector */
       if (acount)
               rcount++;
       if (rcount > 4)
               return (-8);
       else
               return (1);
}