109 lines
3.6 KiB
C
109 lines
3.6 KiB
C
/* ata_pthru.c - ATA pass through for ata.mod. */
|
||
/*
|
||
* GRUB -- GRand Unified Bootloader
|
||
* Copyright (C) 2009 Free Software Foundation, Inc.
|
||
*
|
||
* GRUB is free software: you can redistribute it and/or modify
|
||
* it under the terms of the GNU General Public License as published by
|
||
* the Free Software Foundation, either version 3 of the License, or
|
||
* (at your option) any later version.
|
||
*
|
||
* GRUB is distributed in the hope that it will be useful,
|
||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
* GNU General Public License for more details.
|
||
*
|
||
* You should have received a copy of the GNU General Public License
|
||
* along with GRUB. If not, see <http://www.gnu.org/licenses/>.
|
||
*/
|
||
|
||
#include <grub/ata.h>
|
||
#include <grub/disk.h>
|
||
#include <grub/dl.h>
|
||
#include <grub/mm.h>
|
||
|
||
GRUB_MOD_LICENSE ("GPLv3+");
|
||
|
||
/* ATA pass through support, used by hdparm.mod. */
|
||
static grub_err_t
|
||
grub_ata_pass_through (grub_disk_t disk,
|
||
struct grub_disk_ata_pass_through_parms *parms)
|
||
{
|
||
if (disk->dev->id != GRUB_DISK_DEVICE_ATA_ID)
|
||
return grub_error (GRUB_ERR_BAD_DEVICE,
|
||
"device not accessed via ata.mod");
|
||
|
||
struct grub_ata_device *dev = (struct grub_ata_device *) disk->data;
|
||
|
||
if (! (parms->size == 0 || parms->size == GRUB_DISK_SECTOR_SIZE))
|
||
return grub_error (GRUB_ERR_NOT_IMPLEMENTED_YET,
|
||
"ATA multi-sector read and DATA OUT not implemented");
|
||
|
||
grub_dprintf ("ata", "ata_pass_through: cmd=0x%x, features=0x%x, sectors=0x%x\n",
|
||
parms->taskfile[GRUB_ATA_REG_CMD],
|
||
parms->taskfile[GRUB_ATA_REG_FEATURES],
|
||
parms->taskfile[GRUB_ATA_REG_SECTORS]);
|
||
grub_dprintf ("ata", "lba_high=0x%x, lba_mid=0x%x, lba_low=0x%x, size=%d\n",
|
||
parms->taskfile[GRUB_ATA_REG_LBAHIGH],
|
||
parms->taskfile[GRUB_ATA_REG_LBAMID],
|
||
parms->taskfile[GRUB_ATA_REG_LBALOW], parms->size);
|
||
|
||
/* Set registers. */
|
||
grub_ata_regset (dev, GRUB_ATA_REG_DISK, 0xE0 | dev->device << 4
|
||
| (parms->taskfile[GRUB_ATA_REG_DISK] & 0xf));
|
||
if (grub_ata_check_ready (dev))
|
||
return grub_errno;
|
||
|
||
int i;
|
||
for (i = GRUB_ATA_REG_FEATURES; i <= GRUB_ATA_REG_LBAHIGH; i++)
|
||
grub_ata_regset (dev, i, parms->taskfile[i]);
|
||
|
||
/* Start command. */
|
||
grub_ata_regset (dev, GRUB_ATA_REG_CMD, parms->taskfile[GRUB_ATA_REG_CMD]);
|
||
|
||
/* Wait for !BSY. */
|
||
if (grub_ata_wait_not_busy (dev, GRUB_ATA_TOUT_DATA))
|
||
return grub_errno;
|
||
|
||
/* Check status. */
|
||
grub_int8_t sts = grub_ata_regget (dev, GRUB_ATA_REG_STATUS);
|
||
grub_dprintf ("ata", "status=0x%x\n", sts);
|
||
|
||
/* Transfer data. */
|
||
if ((sts & (GRUB_ATA_STATUS_DRQ | GRUB_ATA_STATUS_ERR)) == GRUB_ATA_STATUS_DRQ)
|
||
{
|
||
if (parms->size != GRUB_DISK_SECTOR_SIZE)
|
||
return grub_error (GRUB_ERR_READ_ERROR, "DRQ unexpected");
|
||
grub_ata_pio_read (dev, parms->buffer, GRUB_DISK_SECTOR_SIZE);
|
||
}
|
||
|
||
/* Return registers. */
|
||
for (i = GRUB_ATA_REG_ERROR; i <= GRUB_ATA_REG_STATUS; i++)
|
||
parms->taskfile[i] = grub_ata_regget (dev, i);
|
||
|
||
grub_dprintf ("ata", "status=0x%x, error=0x%x, sectors=0x%x\n",
|
||
parms->taskfile[GRUB_ATA_REG_STATUS],
|
||
parms->taskfile[GRUB_ATA_REG_ERROR],
|
||
parms->taskfile[GRUB_ATA_REG_SECTORS]);
|
||
|
||
if (parms->taskfile[GRUB_ATA_REG_STATUS]
|
||
& (GRUB_ATA_STATUS_DRQ | GRUB_ATA_STATUS_ERR))
|
||
return grub_error (GRUB_ERR_READ_ERROR, "ATA passthrough failed");
|
||
|
||
return GRUB_ERR_NONE;
|
||
}
|
||
|
||
|
||
|
||
GRUB_MOD_INIT(ata_pthru)
|
||
{
|
||
/* Register ATA pass through function. */
|
||
grub_disk_ata_pass_through = grub_ata_pass_through;
|
||
}
|
||
|
||
GRUB_MOD_FINI(ata_pthru)
|
||
{
|
||
if (grub_disk_ata_pass_through == grub_ata_pass_through)
|
||
grub_disk_ata_pass_through = NULL;
|
||
}
|