123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165 |
- /**
- * u3-tool - U3 USB stick manager
- * Copyright (C) 2007 Daviedev, daviedev@users.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 of the License, or
- * (at your option) any later version.
- *
- * This program 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 this program; if not, write to the Free Software Foundation, Inc.,
- * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
- */
- #include "u3_scsi.h"
- #include "u3_error.h"
- #include <sys/types.h>
- #include <sys/stat.h>
- #include <sys/ioctl.h>
- #include <stdlib.h>
- #include <fcntl.h>
- #include <string.h>
- #include <unistd.h>
- #include <errno.h>
- #include <linux/cdrom.h>
- #include <scsi/sg.h>
- #include "sg_err.h"
- #define U3_TIMEOUT 2000 //2000 millisecs == 2 seconds
- int u3_open(u3_handle_t *device, const char *which)
- {
- int k;
- int sg_fd;
- u3_set_error(device, "");
- device->dev = NULL;
- if ((sg_fd = open(which, O_RDWR)) < 0) {
- // Nautilus sends a eject ioctl to the device when unmounting.
- // Just have to close the tray ;-)
- if (errno == ENOMEDIUM) {
- if ((sg_fd = open(which, O_RDONLY | O_NONBLOCK)) < 0) {
- u3_set_error(device, "%s", strerror(errno));
- return U3_FAILURE;
- }
- ioctl(sg_fd, CDROMCLOSETRAY);
- close(sg_fd);
- if ((sg_fd = open(which, O_RDWR)) < 0) {
- u3_set_error(device, "%s", strerror(errno));
- return U3_FAILURE;
- }
- } else {
- u3_set_error(device, "%s", strerror(errno));
- return U3_FAILURE;
- }
- }
- // It is prudent to check we have a sg device by trying an ioctl
- if (ioctl(sg_fd, SG_GET_VERSION_NUM, &k) < 0) {
- close(sg_fd);
- if (k < 30000) {
- u3_set_error(device, "device is not an SG device");
- } else {
- u3_set_error(device, "SG driver to old");
- }
- return U3_FAILURE;
- }
- if ((device->dev = malloc(sizeof(int))) == NULL) {
- close(sg_fd);
- u3_set_error(device, "Failed allocating memory for file descriptor");
- return U3_FAILURE;
- }
-
- *((int *)device->dev) = sg_fd;
- return U3_SUCCESS;
- }
- void u3_close(u3_handle_t *device)
- {
- int *sg_fd = (int *)device->dev;
- close(*sg_fd);
- }
- int u3_send_cmd(u3_handle_t *device, uint8_t cmd[U3_CMD_LEN],
- int dxfer_direction, int dxfer_length, uint8_t *dxfer_data,
- uint8_t *status)
- {
- sg_io_hdr_t io_hdr;
- unsigned char sense_buf[32];
- int *sg_fd = (int *)device->dev;
- // translate dxfer_direction
- switch (dxfer_direction) {
- case U3_DATA_NONE:
- dxfer_direction = SG_DXFER_NONE;
- break;
- case U3_DATA_TO_DEV:
- dxfer_direction = SG_DXFER_TO_DEV;
- break;
- case U3_DATA_FROM_DEV:
- dxfer_direction = SG_DXFER_FROM_DEV;
- break;
- }
- // Prepare command
- memset(&io_hdr, 0, sizeof(sg_io_hdr_t));
- io_hdr.interface_id = 'S'; // fixed
- io_hdr.cmd_len = U3_CMD_LEN; // length of command in bytes
- // io_hdr.iovec_count = 0; // don't use iovector stuff
- io_hdr.mx_sb_len = sizeof(sense_buf); // sense buffer size. do we use this???
- io_hdr.dxfer_direction = dxfer_direction; // send data to device
- io_hdr.dxfer_len = dxfer_length; // Size of data transfered
- io_hdr.dxferp = dxfer_data; // Data buffer to transfer
- io_hdr.cmdp = cmd; // Command buffer to execute
- io_hdr.sbp = sense_buf; // Sense buffer
- io_hdr.timeout = U3_TIMEOUT; // timeout
- // io_hdr.flags = 0; // take defaults: indirect IO, etc
- // io_hdr.pack_id = 0; // internal packet. used by te program to recognize packets
- // io_hdr.usr_ptr = NULL; // user data
- // preform ioctl on device
- if (ioctl(*sg_fd, SG_IO, &io_hdr) < 0) {
- u3_set_error(device, "Failed executing scsi command: "
- "SG_IO ioctl Failed");
- return U3_FAILURE;
- }
- // evaluate result
- if ((io_hdr.info & SG_INFO_OK_MASK) != SG_INFO_OK) {
- if (io_hdr.host_status == SG_ERR_DID_OK &&
- (io_hdr.driver_status & SG_ERR_DRIVER_SENSE))
- {
- //
- // The usb-storage driver automatically request sense
- // data if a command fails. So this state isn't really
- // a error but only indicates that the sense data in
- // the buffer is fresh. Currently we aren't don't use
- // this sense data, we only wanna know if a command
- // failed or not. so ignore this...
- //
- } else {
- u3_set_error(device, "Failed executing scsi command: "
- "Status (S:0x%x,H:0x%x,D:0x%x)", io_hdr.status,
- io_hdr.host_status, io_hdr.driver_status);
- return U3_FAILURE;
- }
- }
- *status = io_hdr.masked_status;
- return U3_SUCCESS;
- }
|