/*      $NetBSD: linux_hdio.c,v 1.19 2021/09/07 11:43:04 riastradh Exp $        */

/*
* Copyright (c) 2000 Wasabi Systems, Inc.
* All rights reserved.
*
* Written by Frank van der Linden for Wasabi 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. All advertising materials mentioning features or use of this software
*    must display the following acknowledgement:
*      This product includes software developed for the NetBSD Project by
*      Wasabi Systems, Inc.
* 4. The name of Wasabi 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 WASABI 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 WASABI 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.
*/

#include <sys/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: linux_hdio.c,v 1.19 2021/09/07 11:43:04 riastradh Exp $");

#include <sys/param.h>
#include <sys/systm.h>
#include <sys/ioctl.h>
#include <sys/file.h>
#include <sys/filedesc.h>
#include <sys/mount.h>
#include <sys/proc.h>
#include <sys/disklabel.h>

#include <dev/ata/atareg.h>
#include <dev/ic/wdcreg.h>
#include <sys/ataio.h>

#include <sys/syscallargs.h>

#include <compat/linux/common/linux_types.h>
#include <compat/linux/common/linux_ioctl.h>
#include <compat/linux/common/linux_signal.h>
#include <compat/linux/common/linux_util.h>
#include <compat/linux/common/linux_hdio.h>
#include <compat/linux/common/linux_ipc.h>
#include <compat/linux/common/linux_sem.h>

#include <compat/linux/linux_syscallargs.h>

int
linux_ioctl_hdio(struct lwp *l, const struct linux_sys_ioctl_args *uap,
                register_t *retval)
{
       u_long com;
       int error, error1;
       struct file *fp;
       int (*ioctlf)(struct file *, u_long, void *);
       struct atareq req;
       struct disklabel label;
       struct partinfo pi;
       struct linux_hd_geometry hdg;
       struct linux_hd_big_geometry hdg_big;

       if ((fp = fd_getfile(SCARG(uap, fd))) == NULL)
               return (EBADF);

       com = SCARG(uap, com);
       ioctlf = fp->f_ops->fo_ioctl;
       retval[0] = error = 0;

       switch (com) {
       case LINUX_HDIO_OBSOLETE_IDENTITY:
       case LINUX_HDIO_GET_IDENTITY:
               req.flags = ATACMD_READ;
               req.command = WDCC_IDENTIFY;
               req.databuf = SCARG(uap, data);
               /*
                * 142 is the size of the old structure used by Linux,
                * which doesn't seem to be defined anywhere anymore.
                * The new function should return the entire 512 byte area.
                */
               req.datalen = com == LINUX_HDIO_GET_IDENTITY ? 512 : 142;
               req.timeout = 1000;
               error = ioctlf(fp, ATAIOCCOMMAND, &req);
               if (error != 0)
                       break;
               if (req.retsts != ATACMD_OK)
                       error = EIO;
               break;
       case LINUX_HDIO_GETGEO:
               error = linux_machdepioctl(l, uap, retval);
               if (error == 0)
                       break;
               error = ioctlf(fp, DIOCGDINFO, &label);
               error1 = ioctlf(fp, DIOCGPARTINFO, &pi);
               if (error != 0 && error1 != 0) {
                       error = error1;
                       break;
               }
               memset(&hdg, 0, sizeof(hdg));
               hdg.start = error1 != 0 ? pi.pi_offset : 0;
               hdg.heads = label.d_ntracks;
               hdg.cylinders = label.d_ncylinders;
               hdg.sectors = label.d_nsectors;
               error = copyout(&hdg, SCARG(uap, data), sizeof hdg);
               break;
       case LINUX_HDIO_GETGEO_BIG:
               error = linux_machdepioctl(l, uap, retval);
               if (error == 0)
                       break;
               /* FALLTHROUGH */
       case LINUX_HDIO_GETGEO_BIG_RAW:
               error = ioctlf(fp, DIOCGDINFO, &label);
               error1 = ioctlf(fp, DIOCGPARTINFO, &pi);
               if (error != 0 && error1 != 0) {
                       error = error1;
                       break;
               }
               memset(&hdg_big, 0, sizeof(hdg_big));
               hdg_big.start = error1 != 0 ? pi.pi_offset : 0;
               hdg_big.heads = label.d_ntracks;
               hdg_big.cylinders = label.d_ncylinders;
               hdg_big.sectors = label.d_nsectors;
               error = copyout(&hdg_big, SCARG(uap, data), sizeof hdg_big);
               break;
       case LINUX_HDIO_GET_UNMASKINTR:
       case LINUX_HDIO_GET_MULTCOUNT:
       case LINUX_HDIO_GET_KEEPSETTINGS:
       case LINUX_HDIO_GET_32BIT:
       case LINUX_HDIO_GET_NOWERR:
       case LINUX_HDIO_GET_DMA:
       case LINUX_HDIO_GET_NICE:
       case LINUX_HDIO_DRIVE_RESET:
       case LINUX_HDIO_TRISTATE_HWIF:
       case LINUX_HDIO_DRIVE_TASK:
       case LINUX_HDIO_DRIVE_CMD:
       case LINUX_HDIO_SET_MULTCOUNT:
       case LINUX_HDIO_SET_UNMASKINTR:
       case LINUX_HDIO_SET_KEEPSETTINGS:
       case LINUX_HDIO_SET_32BIT:
       case LINUX_HDIO_SET_NOWERR:
       case LINUX_HDIO_SET_DMA:
       case LINUX_HDIO_SET_PIO_MODE:
       case LINUX_HDIO_SCAN_HWIF:
       case LINUX_HDIO_SET_NICE:
       case LINUX_HDIO_UNREGISTER_HWIF:
               error = EINVAL;
       }

       fd_putfile(SCARG(uap, fd));

       return error;
}