[mirror_smartmontools-debian.git] / os_netbsd.cpp

/*
 * os_netbsd.cpp
 *
 * Home page of code is: http://www.smartmontools.org
 *
 * Copyright (C) 2003-8 Sergey Svishchev <smartmontools-support@lists.sourceforge.net>
 *
 * This program 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 2, or (at your option)
 * any later version.
 *
 * You should have received a copy of the GNU General Public License
 * (for example COPYING); if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 */

#include "config.h"
#include "int64.h"
#include "atacmds.h"
#include "scsicmds.h"
#include "utility.h"
#include "os_netbsd.h"

#include <errno.h>
#include <unistd.h>

const char * os_netbsd_cpp_cvsid = "$Id: os_netbsd.cpp 4320 2016-05-10 13:39:19Z chrfranke $"
  OS_NETBSD_H_CVSID;

enum warnings {
  BAD_SMART, MAX_MSG
};

/* Utility function for printing warnings */
void
printwarning(int msgNo, const char *extra)
{
  static int printed[] = {0, 0};
  static const char *message[] = {
    "Error: SMART Status command failed.\nPlease get assistance from \n" PACKAGE_HOMEPAGE "\nRegister values returned from SMART Status command are:\n",
    PACKAGE_STRING " does not currently support twe(4) and twa(4) devices (3ware Escalade, Apache)\n",
  };

  if (msgNo >= 0 && msgNo <= MAX_MSG) {
    if (!printed[msgNo]) {
      printed[msgNo] = 1;
      pout("%s", message[msgNo]);
      if (extra)
	pout("%s", extra);
    }
  }
  return;
}

static const char *net_dev_prefix = "/dev/r";
static const char *net_dev_ata_disk = "wd";
static const char *net_dev_scsi_disk = "sd";
static const char *net_dev_scsi_tape = "enrst";

/* Guess device type (ATA or SCSI) based on device name */
int
guess_device_type(const char *dev_name)
{
  int len;
  int dev_prefix_len = strlen(net_dev_prefix);

  if (!dev_name || !(len = strlen(dev_name)))
    return CONTROLLER_UNKNOWN;

  if (!strncmp(net_dev_prefix, dev_name, dev_prefix_len)) {
    if (len <= dev_prefix_len)
      return CONTROLLER_UNKNOWN;
    else
      dev_name += dev_prefix_len;
  }
  if (!strncmp(net_dev_ata_disk, dev_name, strlen(net_dev_ata_disk)))
    return CONTROLLER_ATA;

  if (!strncmp(net_dev_scsi_disk, dev_name, strlen(net_dev_scsi_disk)))
    return CONTROLLER_SCSI;

  if (!strncmp(net_dev_scsi_tape, dev_name, strlen(net_dev_scsi_tape)))
    return CONTROLLER_SCSI;

  return CONTROLLER_UNKNOWN;
}

int
get_dev_names(char ***names, const char *prefix)
{
  char *disknames, *p, **mp;
  int n = 0;
  int sysctl_mib[2];
  size_t sysctl_len;

  *names = NULL;

  sysctl_mib[0] = CTL_HW;
  sysctl_mib[1] = HW_DISKNAMES;
  if (-1 == sysctl(sysctl_mib, 2, NULL, &sysctl_len, NULL, 0)) {
    pout("Failed to get value of sysctl `hw.disknames'\n");
    return -1;
  }
  if (!(disknames = (char *)malloc(sysctl_len))) {
    pout("Out of memory constructing scan device list\n");
    return -1;
  }
  if (-1 == sysctl(sysctl_mib, 2, disknames, &sysctl_len, NULL, 0)) {
    pout("Failed to get value of sysctl `hw.disknames'\n");
    return -1;
  }
  if (!(mp = (char **) calloc(strlen(disknames) / 2, sizeof(char *)))) {
    pout("Out of memory constructing scan device list\n");
    return -1;
  }
  for (p = strtok(disknames, " "); p; p = strtok(NULL, " ")) {
    if (strncmp(p, prefix, strlen(prefix))) {
      continue;
    }
    mp[n] = (char *)malloc(strlen(net_dev_prefix) + strlen(p) + 2);
    if (!mp[n]) {
      pout("Out of memory constructing scan device list\n");
      return -1;
    }
    sprintf(mp[n], "%s%s%c", net_dev_prefix, p, 'a' + getrawpartition());
    n++;
  }

  char ** tmp = (char **)realloc(mp, n * (sizeof(char *)));
  if (NULL == tmp) {
    pout("Out of memory constructing scan device list\n");
    free(mp);
    return -1;
  }
  else
    mp = tmp;
  *names = mp;
  return n;
}

int
make_device_names(char ***devlist, const char *name)
{
  if (!strcmp(name, "SCSI"))
    return get_dev_names(devlist, net_dev_scsi_disk);
  else if (!strcmp(name, "ATA"))
    return get_dev_names(devlist, net_dev_ata_disk);
  else
    return 0;
}

int
deviceopen(const char *pathname, char *type)
{
  if (!strcmp(type, "SCSI")) {
    int fd = open(pathname, O_RDWR | O_NONBLOCK);
    if (fd < 0 && errno == EROFS)
      fd = open(pathname, O_RDONLY | O_NONBLOCK);
    return fd;
  } else if (!strcmp(type, "ATA"))
    return open(pathname, O_RDWR | O_NONBLOCK);
  else
    return -1;
}

int
deviceclose(int fd)
{
  return close(fd);
}

int
ata_command_interface(int fd, smart_command_set command, int select, char *data)
{
  struct atareq req;
  unsigned char inbuf[DEV_BSIZE];
  int retval, copydata = 0;

  memset(&req, 0, sizeof(req));
  req.timeout = 1000;

  memset(&inbuf, 0, sizeof(inbuf));

  switch (command) {
  case READ_VALUES:
    req.flags = ATACMD_READ;
    req.features = WDSM_RD_DATA;
    req.command = WDCC_SMART;
    req.databuf = (char *)inbuf;
    req.datalen = sizeof(inbuf);
    req.cylinder = WDSMART_CYL;
    copydata = 1;
    break;
  case READ_THRESHOLDS:
    req.flags = ATACMD_READ;
    req.features = WDSM_RD_THRESHOLDS;
    req.command = WDCC_SMART;
    req.databuf = (char *)inbuf;
    req.datalen = sizeof(inbuf);
    req.cylinder = WDSMART_CYL;
    copydata = 1;
    break;
  case READ_LOG:
    req.flags = ATACMD_READ;
    req.features = ATA_SMART_READ_LOG_SECTOR;	/* XXX missing from wdcreg.h */
    req.command = WDCC_SMART;
    req.databuf = (char *)inbuf;
    req.datalen = sizeof(inbuf);
    req.cylinder = WDSMART_CYL;
    req.sec_num = select;
    req.sec_count = 1;
    copydata = 1;
    break;
  case WRITE_LOG:
    memcpy(inbuf, data, 512);
    req.flags = ATACMD_WRITE;
    req.features = ATA_SMART_WRITE_LOG_SECTOR;	/* XXX missing from wdcreg.h */
    req.command = WDCC_SMART;
    req.databuf = (char *)inbuf;
    req.datalen = sizeof(inbuf);
    req.cylinder = WDSMART_CYL;
    req.sec_num = select;
    req.sec_count = 1;
    break;
  case IDENTIFY:
    req.flags = ATACMD_READ;
    req.command = WDCC_IDENTIFY;
    req.databuf = (char *)inbuf;
    req.datalen = sizeof(inbuf);
    copydata = 1;
    break;
  case PIDENTIFY:
    req.flags = ATACMD_READ;
    req.command = ATAPI_IDENTIFY_DEVICE;
    req.databuf = (char *)inbuf;
    req.datalen = sizeof(inbuf);
    copydata = 1;
    break;
  case ENABLE:
    req.flags = ATACMD_READREG;
    req.features = WDSM_ENABLE_OPS;
    req.command = WDCC_SMART;
    req.cylinder = WDSMART_CYL;
    break;
  case DISABLE:
    req.flags = ATACMD_READREG;
    req.features = WDSM_DISABLE_OPS;
    req.command = WDCC_SMART;
    req.cylinder = WDSMART_CYL;
    break;
  case AUTO_OFFLINE:
    /* NOTE: According to ATAPI 4 and UP, this command is obsolete */
    req.flags = ATACMD_READREG;
    req.features = ATA_SMART_AUTO_OFFLINE;	/* XXX missing from wdcreg.h */
    req.command = WDCC_SMART;
    req.cylinder = WDSMART_CYL;
    req.sec_count = select;
    break;
  case AUTOSAVE:
    req.flags = ATACMD_READREG;
    req.features = ATA_SMART_AUTOSAVE;	/* XXX missing from wdcreg.h */
    req.command = WDCC_SMART;
    req.cylinder = WDSMART_CYL;
    req.sec_count = select;
    break;
  case IMMEDIATE_OFFLINE:
    /* NOTE: According to ATAPI 4 and UP, this command is obsolete */
    req.flags = ATACMD_READREG;
    req.features = ATA_SMART_IMMEDIATE_OFFLINE;	/* XXX missing from wdcreg.h */
    req.command = WDCC_SMART;
    req.cylinder = WDSMART_CYL;
    req.sec_num = select;
    req.sec_count = 1;
    break;
  case STATUS:		/* should return 0 if SMART is enabled at all */
  case STATUS_CHECK:	/* should return 0 if disk's health is ok */
    req.flags = ATACMD_READREG;
    req.features = WDSM_STATUS;
    req.command = WDCC_SMART;
    req.cylinder = WDSMART_CYL;
    break;
  case CHECK_POWER_MODE:
    req.flags = ATACMD_READREG;
    req.command = WDCC_CHECK_PWR;
    break;
  default:
    pout("Unrecognized command %d in ata_command_interface()\n", command);
    errno = ENOSYS;
    return -1;
  }

  if (command == STATUS_CHECK || command == AUTOSAVE || command == AUTO_OFFLINE) {
    char buf[512];

    unsigned const short normal = WDSMART_CYL, failed = 0x2cf4;

    if ((retval = ioctl(fd, ATAIOCCOMMAND, &req))) {
      perror("Failed command");
      return -1;
    }
    if (req.retsts != ATACMD_OK) {
      return -1;
    }
    /* Cyl low and Cyl high unchanged means "Good SMART status" */
    if (req.cylinder == normal)
      return 0;

    /* These values mean "Bad SMART status" */
    if (req.cylinder == failed)
      return 1;

    /* We haven't gotten output that makes sense; 
     * print out some debugging info */
    snprintf(buf, sizeof(buf),
      "CMD=0x%02x\nFR =0x%02x\nNS =0x%02x\nSC =0x%02x\nCL =0x%02x\nCH =0x%02x\nRETURN =0x%04x\n",
      (int) req.command, (int) req.features, (int) req.sec_count, (int) req.sec_num,
      (int) (le16toh(req.cylinder) & 0xff), (int) ((le16toh(req.cylinder) >> 8) & 0xff),
      (int) req.error);
    printwarning(BAD_SMART, buf);
    return 0;
  }

  retval = ioctl(fd, ATAIOCCOMMAND, &req);
  if (retval < 0) {
    perror("Failed command");
    return -1;
  }
  if (req.retsts != ATACMD_OK) {
    return -1;
  }

  if (command == CHECK_POWER_MODE)
    data[0] = req.sec_count;

  if (copydata)
    memcpy(data, inbuf, 512);

  return 0;
}

int
do_scsi_cmnd_io(int fd, struct scsi_cmnd_io * iop, int report)
{
  struct scsireq sc;

  if (report > 0) {
    size_t k;

    const unsigned char *ucp = iop->cmnd;
    const char *np;

    np = scsi_get_opcode_name(ucp[0]);
    pout(" [%s: ", np ? np : "<unknown opcode>");
    for (k = 0; k < iop->cmnd_len; ++k)
      pout("%02x ", ucp[k]);
    if ((report > 1) &&
      (DXFER_TO_DEVICE == iop->dxfer_dir) && (iop->dxferp)) {
      int trunc = (iop->dxfer_len > 256) ? 1 : 0;

      pout("]\n  Outgoing data, len=%d%s:\n", (int) iop->dxfer_len,
	(trunc ? " [only first 256 bytes shown]" : ""));
      dStrHex(iop->dxferp, (trunc ? 256 : iop->dxfer_len), 1);
    } else
      pout("]");
  }
  memset(&sc, 0, sizeof(sc));
  memcpy(sc.cmd, iop->cmnd, iop->cmnd_len);
  sc.cmdlen = iop->cmnd_len;
  sc.databuf = (char *)iop->dxferp;
  sc.datalen = iop->dxfer_len;
  sc.senselen = iop->max_sense_len;
  sc.timeout = iop->timeout == 0 ? 60000 : (1000 * iop->timeout);
  sc.flags =
    (iop->dxfer_dir == DXFER_NONE ? SCCMD_READ :
    (iop->dxfer_dir == DXFER_FROM_DEVICE ? SCCMD_READ : SCCMD_WRITE));

  if (ioctl(fd, SCIOCCOMMAND, &sc) < 0) {
    warn("error sending SCSI ccb");
    return -1;
  }
  iop->resid = sc.datalen - sc.datalen_used;
  iop->scsi_status = sc.status;
  if (iop->sensep) {
    memcpy(iop->sensep, sc.sense, sc.senselen_used);
    iop->resp_sense_len = sc.senselen_used;
  }
  if (report > 0) {
    int trunc;

    pout("  status=0\n");
    trunc = (iop->dxfer_len > 256) ? 1 : 0;

    pout("  Incoming data, len=%d%s:\n", (int) iop->dxfer_len,
      (trunc ? " [only first 256 bytes shown]" : ""));
    dStrHex(iop->dxferp, (trunc ? 256 : iop->dxfer_len), 1);
  }
  switch (sc.retsts) {
    case SCCMD_OK:
      return 0;
    case SCCMD_TIMEOUT:
      return -ETIMEDOUT;
    case SCCMD_BUSY:
      return -EBUSY;
    default:
      return -EIO;
  }
}

/* print examples for smartctl */
void 
print_smartctl_examples()
{
  char p;

  p = 'a' + getrawpartition();
  printf("=================================================== SMARTCTL EXAMPLES =====\n\n");
#ifdef HAVE_GETOPT_LONG
  printf(
    "  smartctl -a /dev/wd0%c                      (Prints all SMART information)\n\n"
    "  smartctl --smart=on --offlineauto=on --saveauto=on /dev/wd0%c\n"
    "                                              (Enables SMART on first disk)\n\n"
    "  smartctl -t long /dev/wd0%c             (Executes extended disk self-test)\n\n"
    "  smartctl --attributes --log=selftest --quietmode=errorsonly /dev/wd0%c\n"
    "                                      (Prints Self-Test & Attribute errors)\n",
    p, p, p, p
    );
#else
  printf(
    "  smartctl -a /dev/wd0%c                     (Prints all SMART information)\n"
    "  smartctl -s on -o on -S on /dev/wd0%c        (Enables SMART on first disk)\n"
    "  smartctl -t long /dev/wd0%c            (Executes extended disk self-test)\n"
    "  smartctl -A -l selftest -q errorsonly /dev/wd0%c"
    "                                      (Prints Self-Test & Attribute errors)\n",
    p, p, p, p
    );
#endif
  return;
}
Commit	Line	Data
832b75ed	1	/*
4d59bff9	2	* os_netbsd.cpp
832b75ed	3	*
a86ec89e	4	* Home page of code is: http://www.smartmontools.org
832b75ed	5	*
34ad0c5f	6	* Copyright (C) 2003-8 Sergey Svishchev <smartmontools-support@lists.sourceforge.net>
832b75ed GG	7	*
	8	* This program is free software; you can redistribute it and/or modify
	9	* it under the terms of the GNU General Public License as published by
	10	* the Free Software Foundation; either version 2, or (at your option)
	11	* any later version.
	12	*
	13	* You should have received a copy of the GNU General Public License
ee38a438 GI	14	* (for example COPYING); if not, write to the Free Software Foundation,
ee38a438 GI	15	* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
832b75ed GG	16	*
	17	*/
	18
	19	#include "config.h"
	20	#include "int64.h"
	21	#include "atacmds.h"
	22	#include "scsicmds.h"
	23	#include "utility.h"
	24	#include "os_netbsd.h"
cfbba5b9 GI	25
cfbba5b9 GI	26	#include <errno.h>
832b75ed GG	27	#include <unistd.h>
832b75ed GG	28
a86ec89e	29	const char * os_netbsd_cpp_cvsid = "$Id: os_netbsd.cpp 4320 2016-05-10 13:39:19Z chrfranke $"
cfbba5b9	30	OS_NETBSD_H_CVSID;
832b75ed	31
832b75ed	32	enum warnings {
a86ec89e	33	BAD_SMART, MAX_MSG
832b75ed GG	34	};
	35
	36	/* Utility function for printing warnings */
	37	void
	38	printwarning(int msgNo, const char *extra)
	39	{
	40	static int printed[] = {0, 0};
	41	static const char *message[] = {
	42	"Error: SMART Status command failed.\nPlease get assistance from \n" PACKAGE_HOMEPAGE "\nRegister values returned from SMART Status command are:\n",
34ad0c5f	43	PACKAGE_STRING " does not currently support twe(4) and twa(4) devices (3ware Escalade, Apache)\n",
832b75ed GG	44	};
	45
	46	if (msgNo >= 0 && msgNo <= MAX_MSG) {
	47	if (!printed[msgNo]) {
	48	printed[msgNo] = 1;
	49	pout("%s", message[msgNo]);
	50	if (extra)
	51	pout("%s", extra);
	52	}
	53	}
	54	return;
	55	}
	56
a86ec89e	57	static const char *net_dev_prefix = "/dev/r";
832b75ed GG	58	static const char *net_dev_ata_disk = "wd";
	59	static const char *net_dev_scsi_disk = "sd";
	60	static const char *net_dev_scsi_tape = "enrst";
	61
34ad0c5f	62	/* Guess device type (ATA or SCSI) based on device name */
832b75ed GG	63	int
	64	guess_device_type(const char *dev_name)
	65	{
	66	int len;
	67	int dev_prefix_len = strlen(net_dev_prefix);
	68
	69	if (!dev_name \|\| !(len = strlen(dev_name)))
	70	return CONTROLLER_UNKNOWN;
	71
	72	if (!strncmp(net_dev_prefix, dev_name, dev_prefix_len)) {
	73	if (len <= dev_prefix_len)
	74	return CONTROLLER_UNKNOWN;
	75	else
	76	dev_name += dev_prefix_len;
	77	}
	78	if (!strncmp(net_dev_ata_disk, dev_name, strlen(net_dev_ata_disk)))
	79	return CONTROLLER_ATA;
	80
	81	if (!strncmp(net_dev_scsi_disk, dev_name, strlen(net_dev_scsi_disk)))
	82	return CONTROLLER_SCSI;
	83
	84	if (!strncmp(net_dev_scsi_tape, dev_name, strlen(net_dev_scsi_tape)))
	85	return CONTROLLER_SCSI;
	86
	87	return CONTROLLER_UNKNOWN;
	88	}
	89
	90	int
	91	get_dev_names(char **names, const char prefix)
	92	{
	93	char disknames, p, **mp;
	94	int n = 0;
	95	int sysctl_mib[2];
	96	size_t sysctl_len;
	97
	98	*names = NULL;
	99
	100	sysctl_mib[0] = CTL_HW;
	101	sysctl_mib[1] = HW_DISKNAMES;
	102	if (-1 == sysctl(sysctl_mib, 2, NULL, &sysctl_len, NULL, 0)) {
	103	pout("Failed to get value of sysctl `hw.disknames'\n");
	104	return -1;
	105	}
4d59bff9	106	if (!(disknames = (char *)malloc(sysctl_len))) {
832b75ed GG	107	pout("Out of memory constructing scan device list\n");
	108	return -1;
	109	}
	110	if (-1 == sysctl(sysctl_mib, 2, disknames, &sysctl_len, NULL, 0)) {
	111	pout("Failed to get value of sysctl `hw.disknames'\n");
	112	return -1;
	113	}
	114	if (!(mp = (char *) calloc(strlen(disknames) / 2, sizeof(char )))) {
	115	pout("Out of memory constructing scan device list\n");
	116	return -1;
	117	}
	118	for (p = strtok(disknames, " "); p; p = strtok(NULL, " ")) {
	119	if (strncmp(p, prefix, strlen(prefix))) {
	120	continue;
	121	}
4d59bff9	122	mp[n] = (char *)malloc(strlen(net_dev_prefix) + strlen(p) + 2);
832b75ed GG	123	if (!mp[n]) {
	124	pout("Out of memory constructing scan device list\n");
	125	return -1;
	126	}
	127	sprintf(mp[n], "%s%s%c", net_dev_prefix, p, 'a' + getrawpartition());
832b75ed GG	128	n++;
	129	}
	130
a86ec89e GI	131	char tmp = (char )realloc(mp, n * (sizeof(char *)));
	132	if (NULL == tmp) {
	133	pout("Out of memory constructing scan device list\n");
	134	free(mp);
	135	return -1;
	136	}
	137	else
	138	mp = tmp;
832b75ed GG	139	*names = mp;
	140	return n;
	141	}
	142
	143	int
	144	make_device_names(char **devlist, const char name)
	145	{
	146	if (!strcmp(name, "SCSI"))
	147	return get_dev_names(devlist, net_dev_scsi_disk);
	148	else if (!strcmp(name, "ATA"))
	149	return get_dev_names(devlist, net_dev_ata_disk);
	150	else
	151	return 0;
	152	}
	153
	154	int
	155	deviceopen(const char pathname, char type)
	156	{
	157	if (!strcmp(type, "SCSI")) {
	158	int fd = open(pathname, O_RDWR \| O_NONBLOCK);
	159	if (fd < 0 && errno == EROFS)
	160	fd = open(pathname, O_RDONLY \| O_NONBLOCK);
	161	return fd;
	162	} else if (!strcmp(type, "ATA"))
	163	return open(pathname, O_RDWR \| O_NONBLOCK);
	164	else
	165	return -1;
	166	}
	167
	168	int
	169	deviceclose(int fd)
	170	{
	171	return close(fd);
	172	}
	173
832b75ed GG	174	int
	175	ata_command_interface(int fd, smart_command_set command, int select, char *data)
	176	{
	177	struct atareq req;
	178	unsigned char inbuf[DEV_BSIZE];
	179	int retval, copydata = 0;
	180
	181	memset(&req, 0, sizeof(req));
34ad0c5f GG	182	req.timeout = 1000;
34ad0c5f GG	183
832b75ed GG	184	memset(&inbuf, 0, sizeof(inbuf));
	185
	186	switch (command) {
	187	case READ_VALUES:
	188	req.flags = ATACMD_READ;
	189	req.features = WDSM_RD_DATA;
	190	req.command = WDCC_SMART;
4d59bff9	191	req.databuf = (char *)inbuf;
832b75ed GG	192	req.datalen = sizeof(inbuf);
832b75ed GG	193	req.cylinder = WDSMART_CYL;
832b75ed GG	194	copydata = 1;
	195	break;
	196	case READ_THRESHOLDS:
	197	req.flags = ATACMD_READ;
	198	req.features = WDSM_RD_THRESHOLDS;
	199	req.command = WDCC_SMART;
4d59bff9	200	req.databuf = (char *)inbuf;
832b75ed GG	201	req.datalen = sizeof(inbuf);
832b75ed GG	202	req.cylinder = WDSMART_CYL;
832b75ed GG	203	copydata = 1;
	204	break;
	205	case READ_LOG:
	206	req.flags = ATACMD_READ;
	207	req.features = ATA_SMART_READ_LOG_SECTOR; /* XXX missing from wdcreg.h */
	208	req.command = WDCC_SMART;
4d59bff9	209	req.databuf = (char *)inbuf;
832b75ed GG	210	req.datalen = sizeof(inbuf);
	211	req.cylinder = WDSMART_CYL;
	212	req.sec_num = select;
	213	req.sec_count = 1;
832b75ed GG	214	copydata = 1;
	215	break;
	216	case WRITE_LOG:
	217	memcpy(inbuf, data, 512);
	218	req.flags = ATACMD_WRITE;
	219	req.features = ATA_SMART_WRITE_LOG_SECTOR; /* XXX missing from wdcreg.h */
	220	req.command = WDCC_SMART;
4d59bff9	221	req.databuf = (char *)inbuf;
832b75ed GG	222	req.datalen = sizeof(inbuf);
	223	req.cylinder = WDSMART_CYL;
	224	req.sec_num = select;
	225	req.sec_count = 1;
832b75ed GG	226	break;
	227	case IDENTIFY:
	228	req.flags = ATACMD_READ;
	229	req.command = WDCC_IDENTIFY;
4d59bff9	230	req.databuf = (char *)inbuf;
832b75ed	231	req.datalen = sizeof(inbuf);
832b75ed GG	232	copydata = 1;
	233	break;
	234	case PIDENTIFY:
	235	req.flags = ATACMD_READ;
	236	req.command = ATAPI_IDENTIFY_DEVICE;
4d59bff9	237	req.databuf = (char *)inbuf;
832b75ed	238	req.datalen = sizeof(inbuf);
832b75ed GG	239	copydata = 1;
	240	break;
	241	case ENABLE:
34ad0c5f	242	req.flags = ATACMD_READREG;
832b75ed GG	243	req.features = WDSM_ENABLE_OPS;
	244	req.command = WDCC_SMART;
	245	req.cylinder = WDSMART_CYL;
832b75ed GG	246	break;
832b75ed GG	247	case DISABLE:
34ad0c5f	248	req.flags = ATACMD_READREG;
832b75ed GG	249	req.features = WDSM_DISABLE_OPS;
	250	req.command = WDCC_SMART;
	251	req.cylinder = WDSMART_CYL;
832b75ed GG	252	break;
	253	case AUTO_OFFLINE:
	254	/* NOTE: According to ATAPI 4 and UP, this command is obsolete */
34ad0c5f	255	req.flags = ATACMD_READREG;
832b75ed GG	256	req.features = ATA_SMART_AUTO_OFFLINE; /* XXX missing from wdcreg.h */
832b75ed GG	257	req.command = WDCC_SMART;
832b75ed	258	req.cylinder = WDSMART_CYL;
34ad0c5f	259	req.sec_count = select;
832b75ed GG	260	break;
832b75ed GG	261	case AUTOSAVE:
34ad0c5f	262	req.flags = ATACMD_READREG;
832b75ed GG	263	req.features = ATA_SMART_AUTOSAVE; /* XXX missing from wdcreg.h */
	264	req.command = WDCC_SMART;
	265	req.cylinder = WDSMART_CYL;
34ad0c5f	266	req.sec_count = select;
832b75ed GG	267	break;
	268	case IMMEDIATE_OFFLINE:
	269	/* NOTE: According to ATAPI 4 and UP, this command is obsolete */
34ad0c5f	270	req.flags = ATACMD_READREG;
832b75ed GG	271	req.features = ATA_SMART_IMMEDIATE_OFFLINE; /* XXX missing from wdcreg.h */
832b75ed GG	272	req.command = WDCC_SMART;
832b75ed GG	273	req.cylinder = WDSMART_CYL;
	274	req.sec_num = select;
	275	req.sec_count = 1;
832b75ed	276	break;
34ad0c5f GG	277	case STATUS: /* should return 0 if SMART is enabled at all */
	278	case STATUS_CHECK: /* should return 0 if disk's health is ok */
	279	req.flags = ATACMD_READREG;
832b75ed GG	280	req.features = WDSM_STATUS;
	281	req.command = WDCC_SMART;
	282	req.cylinder = WDSMART_CYL;
832b75ed GG	283	break;
	284	case CHECK_POWER_MODE:
	285	req.flags = ATACMD_READREG;
	286	req.command = WDCC_CHECK_PWR;
832b75ed GG	287	break;
	288	default:
	289	pout("Unrecognized command %d in ata_command_interface()\n", command);
	290	errno = ENOSYS;
	291	return -1;
	292	}
	293
34ad0c5f	294	if (command == STATUS_CHECK \|\| command == AUTOSAVE \|\| command == AUTO_OFFLINE) {
832b75ed GG	295	char buf[512];
	296
	297	unsigned const short normal = WDSMART_CYL, failed = 0x2cf4;
	298
	299	if ((retval = ioctl(fd, ATAIOCCOMMAND, &req))) {
	300	perror("Failed command");
	301	return -1;
	302	}
34ad0c5f GG	303	if (req.retsts != ATACMD_OK) {
	304	return -1;
	305	}
832b75ed GG	306	/* Cyl low and Cyl high unchanged means "Good SMART status" */
	307	if (req.cylinder == normal)
	308	return 0;
	309
	310	/* These values mean "Bad SMART status" */
	311	if (req.cylinder == failed)
	312	return 1;
	313
	314	/* We haven't gotten output that makes sense;
	315	* print out some debugging info */
	316	snprintf(buf, sizeof(buf),
	317	"CMD=0x%02x\nFR =0x%02x\nNS =0x%02x\nSC =0x%02x\nCL =0x%02x\nCH =0x%02x\nRETURN =0x%04x\n",
	318	(int) req.command, (int) req.features, (int) req.sec_count, (int) req.sec_num,
	319	(int) (le16toh(req.cylinder) & 0xff), (int) ((le16toh(req.cylinder) >> 8) & 0xff),
	320	(int) req.error);
	321	printwarning(BAD_SMART, buf);
	322	return 0;
	323	}
34ad0c5f	324
a86ec89e GI	325	retval = ioctl(fd, ATAIOCCOMMAND, &req);
a86ec89e GI	326	if (retval < 0) {
832b75ed GG	327	perror("Failed command");
	328	return -1;
	329	}
34ad0c5f GG	330	if (req.retsts != ATACMD_OK) {
	331	return -1;
	332	}
	333
832b75ed GG	334	if (command == CHECK_POWER_MODE)
	335	data[0] = req.sec_count;
	336
	337	if (copydata)
	338	memcpy(data, inbuf, 512);
	339
	340	return 0;
	341	}
	342
832b75ed GG	343	int
	344	do_scsi_cmnd_io(int fd, struct scsi_cmnd_io * iop, int report)
	345	{
	346	struct scsireq sc;
	347
	348	if (report > 0) {
	349	size_t k;
	350
	351	const unsigned char *ucp = iop->cmnd;
	352	const char *np;
	353
	354	np = scsi_get_opcode_name(ucp[0]);
	355	pout(" [%s: ", np ? np : "<unknown opcode>");
	356	for (k = 0; k < iop->cmnd_len; ++k)
	357	pout("%02x ", ucp[k]);
	358	if ((report > 1) &&
	359	(DXFER_TO_DEVICE == iop->dxfer_dir) && (iop->dxferp)) {
	360	int trunc = (iop->dxfer_len > 256) ? 1 : 0;
	361
	362	pout("]\n Outgoing data, len=%d%s:\n", (int) iop->dxfer_len,
	363	(trunc ? " [only first 256 bytes shown]" : ""));
	364	dStrHex(iop->dxferp, (trunc ? 256 : iop->dxfer_len), 1);
	365	} else
	366	pout("]");
	367	}
	368	memset(&sc, 0, sizeof(sc));
	369	memcpy(sc.cmd, iop->cmnd, iop->cmnd_len);
	370	sc.cmdlen = iop->cmnd_len;
4d59bff9	371	sc.databuf = (char *)iop->dxferp;
832b75ed GG	372	sc.datalen = iop->dxfer_len;
	373	sc.senselen = iop->max_sense_len;
	374	sc.timeout = iop->timeout == 0 ? 60000 : (1000 * iop->timeout);
	375	sc.flags =
4d59bff9	376	(iop->dxfer_dir == DXFER_NONE ? SCCMD_READ :
832b75ed GG	377	(iop->dxfer_dir == DXFER_FROM_DEVICE ? SCCMD_READ : SCCMD_WRITE));
	378
	379	if (ioctl(fd, SCIOCCOMMAND, &sc) < 0) {
	380	warn("error sending SCSI ccb");
	381	return -1;
	382	}
	383	iop->resid = sc.datalen - sc.datalen_used;
	384	iop->scsi_status = sc.status;
	385	if (iop->sensep) {
	386	memcpy(iop->sensep, sc.sense, sc.senselen_used);
	387	iop->resp_sense_len = sc.senselen_used;
	388	}
	389	if (report > 0) {
	390	int trunc;
	391
	392	pout(" status=0\n");
	393	trunc = (iop->dxfer_len > 256) ? 1 : 0;
	394
	395	pout(" Incoming data, len=%d%s:\n", (int) iop->dxfer_len,
	396	(trunc ? " [only first 256 bytes shown]" : ""));
	397	dStrHex(iop->dxferp, (trunc ? 256 : iop->dxfer_len), 1);
	398	}
34ad0c5f GG	399	switch (sc.retsts) {
	400	case SCCMD_OK:
	401	return 0;
	402	case SCCMD_TIMEOUT:
	403	return -ETIMEDOUT;
	404	case SCCMD_BUSY:
	405	return -EBUSY;
	406	default:
	407	return -EIO;
	408	}
832b75ed GG	409	}
	410
	411	/* print examples for smartctl */
	412	void
	413	print_smartctl_examples()
	414	{
	415	char p;
	416
	417	p = 'a' + getrawpartition();
	418	printf("=================================================== SMARTCTL EXAMPLES =====\n\n");
	419	#ifdef HAVE_GETOPT_LONG
	420	printf(
	421	" smartctl -a /dev/wd0%c (Prints all SMART information)\n\n"
	422	" smartctl --smart=on --offlineauto=on --saveauto=on /dev/wd0%c\n"
	423	" (Enables SMART on first disk)\n\n"
	424	" smartctl -t long /dev/wd0%c (Executes extended disk self-test)\n\n"
	425	" smartctl --attributes --log=selftest --quietmode=errorsonly /dev/wd0%c\n"
	426	" (Prints Self-Test & Attribute errors)\n",
	427	p, p, p, p
	428	);
	429	#else
	430	printf(
	431	" smartctl -a /dev/wd0%c (Prints all SMART information)\n"
	432	" smartctl -s on -o on -S on /dev/wd0%c (Enables SMART on first disk)\n"
	433	" smartctl -t long /dev/wd0%c (Executes extended disk self-test)\n"
	434	" smartctl -A -l selftest -q errorsonly /dev/wd0%c"
	435	" (Prints Self-Test & Attribute errors)\n",
	436	p, p, p, p
	437	);
	438	#endif
	439	return;
	440	}