/*      $NetBSD: sd.c,v 1.13 2024/05/09 15:11:11 tsutsui Exp $  */

/*
* Copyright (c) 1988 University of Utah.
* Copyright (c) 1990, 1993
*      The Regents of the University of California.  All rights reserved.
*
* This code is derived from software contributed to Berkeley by
* Van Jacobson of Lawrence Berkeley Laboratory and the Systems
* Programming Group of the University of Utah Computer Science Department.
*
* 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. Neither the name of the University nor the names of its contributors
*    may be used to endorse or promote products derived from this software
*    without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE REGENTS 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 REGENTS 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.
*
* from: Utah $Hdr: sd.c 1.9 92/12/21$
*
*      @(#)sd.c        8.1 (Berkeley) 6/10/93
*/

/*
* SCSI CCS disk driver
*/

#include <sys/param.h>
#include <sys/disklabel.h>

#include <lib/libsa/stand.h>

#include <hp300/stand/common/samachdep.h>
#include <hp300/stand/common/conf.h>

#define _IOCTL_
#include <hp300/stand/common/scsireg.h>
#include <hp300/stand/common/scsivar.h>

struct  sdminilabel {
       u_short npart;
       u_long  offset[MAXPARTITIONS];
};

struct  sd_softc {
       int     sc_ctlr;
       int     sc_unit;
       int     sc_part;
       char    sc_retry;
       char    sc_alive;
       short   sc_blkshift;
       struct  sdminilabel sc_pinfo;
#ifdef SUPPORT_CD
       uint8_t sc_type;
#endif
};

#define SDRETRY         2

static int sdinit(int ,int);
static int sdgetinfo(struct sd_softc *);

struct  disklabel sdlabel;
struct sd_softc sd_softc[NSCSI][NSD];

static int
sdinit(int ctlr, int unit)
{
       struct sd_softc *ss = &sd_softc[ctlr][unit];
       u_char stat;
#ifdef SUPPORT_CD
       struct scsi_inquiry inqbuf;
#endif
       int capbuf[2];

       stat = scsi_test_unit_rdy(ctlr, unit);
       if (stat) {
               /* drive may be doing RTZ - wait a bit */
               if (stat == STS_CHECKCOND) {
                       DELAY(1000000);
                       stat = scsi_test_unit_rdy(ctlr, unit);
               }
               if (stat) {
                       printf("sd(%d,%d,0,0): init failed (stat=%x)\n",
                              ctlr, unit, stat);
                       return 0;
               }
       }
#ifdef SUPPORT_CD
       /*
        * try to get the disk type.
        */
       memset(&inqbuf, 0, sizeof(inqbuf));
       stat = scsi_inquiry(ctlr, unit, (u_char *)&inqbuf, sizeof(inqbuf));
       if (stat == 0) {
               /* to fake a disklabel on CD-ROM */
               ss->sc_type = inqbuf.type & SID_TYPE;
       } else {
               /* assume a disk by default */
               ss->sc_type = T_DIRECT;
       }
#endif
       /*
        * try to get the drive block size.
        */
       capbuf[0] = 0;
       capbuf[1] = 0;
       stat = scsi_read_capacity(ctlr, unit,
                                 (u_char *)capbuf, sizeof(capbuf));
       if (stat == 0) {
               if (capbuf[1] > DEV_BSIZE)
                       for (; capbuf[1] > DEV_BSIZE; capbuf[1] >>= 1)
                               ++ss->sc_blkshift;
       }
       ss->sc_alive = 1;
       return 1;
}

char io_buf[MAXBSIZE];

static int
sdgetinfo(struct sd_softc *ss)
{
       struct sdminilabel *pi = &ss->sc_pinfo;
       struct disklabel *lp = &sdlabel;
       char *msg;
       int err, savepart;
       size_t i;

       memset((void *)lp, 0, sizeof *lp);
       lp->d_secsize = (DEV_BSIZE << ss->sc_blkshift);

       /* Disklabel is always from RAW_PART. */
       savepart = ss->sc_part;
       ss->sc_part = RAW_PART;
       err = sdstrategy(ss, F_READ, LABELSECTOR,
           lp->d_secsize ? lp->d_secsize : DEV_BSIZE, io_buf, &i);
       ss->sc_part = savepart;

       if (err) {
               printf("sdgetinfo: sdstrategy error %d\n", err);
               return 0;
       }

       msg = getdisklabel(io_buf, lp);
       if (msg) {
#ifdef SUPPORT_CD
               if (ss->sc_type == T_CDROM) {
                       /* assume a whole disk region is ISO9660 */
                       pi->npart = 3;
                       pi->offset[0] = 0;
                       pi->offset[1] = -1;
                       pi->offset[2] = 0;
               } else
#endif
               {
                       printf("sd(%d,%d,%d): WARNING: %s\n",
                           ss->sc_ctlr, ss->sc_unit, ss->sc_part, msg);
                       pi->npart = 3;
                       pi->offset[0] = pi->offset[1] = -1;
                       pi->offset[2] = 0;
               }
       } else {
               pi->npart = lp->d_npartitions;
               for (i = 0; i < pi->npart; i++)
                       pi->offset[i] = lp->d_partitions[i].p_size == 0 ?
                               -1 : lp->d_partitions[i].p_offset;
       }
       return 1;
}

int
sdopen(struct open_file *f, ...)
{
       va_list ap;
       int ctlr, unit, part;
       struct sd_softc *ss;

       va_start(ap, f);
       ctlr = va_arg(ap, int);
       unit = va_arg(ap, int);
       part = va_arg(ap, int);
       va_end(ap);

#ifdef SD_DEBUG
       if (debug)
       printf("sdopen: ctlr=%d unit=%d part=%d\n",
           ctlr, unit, part);
#endif

       if (ctlr >= NSCSI || scsialive(ctlr) == 0)
               return EADAPT;
       if (unit >= NSD)
               return ECTLR;
       ss = &sd_softc[ctlr][unit];
       ss->sc_part = part;
       ss->sc_unit = unit;
       ss->sc_ctlr = ctlr;
       if (ss->sc_alive == 0) {
               if (sdinit(ctlr, unit) == 0)
                       return ENXIO;
               if (sdgetinfo(ss) == 0)
                       return ERDLAB;
       }
       if (part != RAW_PART &&     /* always allow RAW_PART to be opened */
           (part >= ss->sc_pinfo.npart || ss->sc_pinfo.offset[part] == -1))
               return EPART;
       f->f_devdata = (void *)ss;
       return 0;
}

int
sdclose(struct open_file *f)
{
       struct sd_softc *ss = f->f_devdata;

       /*
        * Mark the disk `not alive' so that the disklabel
        * will be re-loaded at next open.
        */
       memset(ss, 0, sizeof(sd_softc));
       f->f_devdata = NULL;

       return 0;
}

int
sdstrategy(void *devdata, int func, daddr_t dblk, size_t size, void *v_buf,
   size_t *rsize)
{
       struct sd_softc *ss = devdata;
       uint8_t *buf = v_buf;
       int ctlr = ss->sc_ctlr;
       int unit = ss->sc_unit;
       u_int nblk = size >> ss->sc_blkshift;
       daddr_t blk;
       int stat;

       if (size == 0)
               return 0;

       /*
        * Don't do partition translation on the `raw partition'.
        */
       blk = (dblk + ((ss->sc_part == RAW_PART) ? 0 :
           ss->sc_pinfo.offset[ss->sc_part])) >> ss->sc_blkshift;

       ss->sc_retry = 0;

#ifdef SD_DEBUG
       if (debug)
       printf("sdstrategy(%d,%d): size=%d blk=%d nblk=%d\n",
           ctlr, unit, size, blk, nblk);
#endif

retry:
       if (func == F_READ)
               stat = scsi_tt_read(ctlr, unit, buf, size, blk, nblk);
       else
               stat = scsi_tt_write(ctlr, unit, buf, size, blk, nblk);
       if (stat) {
               printf("sd(%d,%d,%d): block=%x, error=0x%x\n",
                      ctlr, unit, ss->sc_part, blk, stat);
               if (++ss->sc_retry > SDRETRY)
                       return EIO;
               goto retry;
       }
       *rsize = size;

       return 0;
}