usb: usbtmc: Add ioctl USBTMC_IOCTL_AUTO_ABORT

Add ioctl USBTMC_IOCTL_AUTO_ABORT to configure auto_abort for
each specific file handle.

Signed-off-by: Guido Kiener <guido.kiener@rohde-schwarz.com>
Reviewed-by: Steve Bayless <steve_bayless@keysight.com>
Signed-off-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>
This commit is contained in:
Guido Kiener 2018-09-12 10:51:00 +02:00 committed by Greg Kroah-Hartman
parent 8409e96f01
commit ec34d08eff
2 changed files with 17 additions and 7 deletions

View File

@ -136,6 +136,7 @@ struct usbtmc_file_data {
u8 eom_val; u8 eom_val;
u8 term_char; u8 term_char;
bool term_char_enabled; bool term_char_enabled;
bool auto_abort;
spinlock_t err_lock; /* lock for errors */ spinlock_t err_lock; /* lock for errors */
@ -201,6 +202,7 @@ static int usbtmc_open(struct inode *inode, struct file *filp)
file_data->timeout = USBTMC_TIMEOUT; file_data->timeout = USBTMC_TIMEOUT;
file_data->term_char = data->TermChar; file_data->term_char = data->TermChar;
file_data->term_char_enabled = data->TermCharEnabled; file_data->term_char_enabled = data->TermCharEnabled;
file_data->auto_abort = data->auto_abort;
file_data->eom_val = 1; file_data->eom_val = 1;
INIT_LIST_HEAD(&file_data->file_elem); INIT_LIST_HEAD(&file_data->file_elem);
@ -1376,7 +1378,7 @@ static ssize_t usbtmc_read(struct file *filp, char __user *buf,
retval = send_request_dev_dep_msg_in(file_data, count); retval = send_request_dev_dep_msg_in(file_data, count);
if (retval < 0) { if (retval < 0) {
if (data->auto_abort) if (file_data->auto_abort)
usbtmc_ioctl_abort_bulk_out(data); usbtmc_ioctl_abort_bulk_out(data);
goto exit; goto exit;
} }
@ -1401,7 +1403,7 @@ static ssize_t usbtmc_read(struct file *filp, char __user *buf,
if (retval < 0) { if (retval < 0) {
dev_dbg(dev, "Unable to read data, error %d\n", retval); dev_dbg(dev, "Unable to read data, error %d\n", retval);
if (data->auto_abort) if (file_data->auto_abort)
usbtmc_ioctl_abort_bulk_in(data); usbtmc_ioctl_abort_bulk_in(data);
goto exit; goto exit;
} }
@ -1411,21 +1413,21 @@ static ssize_t usbtmc_read(struct file *filp, char __user *buf,
/* Sanity checks for the header */ /* Sanity checks for the header */
if (actual < USBTMC_HEADER_SIZE) { if (actual < USBTMC_HEADER_SIZE) {
dev_err(dev, "Device sent too small first packet: %u < %u\n", actual, USBTMC_HEADER_SIZE); dev_err(dev, "Device sent too small first packet: %u < %u\n", actual, USBTMC_HEADER_SIZE);
if (data->auto_abort) if (file_data->auto_abort)
usbtmc_ioctl_abort_bulk_in(data); usbtmc_ioctl_abort_bulk_in(data);
goto exit; goto exit;
} }
if (buffer[0] != 2) { if (buffer[0] != 2) {
dev_err(dev, "Device sent reply with wrong MsgID: %u != 2\n", buffer[0]); dev_err(dev, "Device sent reply with wrong MsgID: %u != 2\n", buffer[0]);
if (data->auto_abort) if (file_data->auto_abort)
usbtmc_ioctl_abort_bulk_in(data); usbtmc_ioctl_abort_bulk_in(data);
goto exit; goto exit;
} }
if (buffer[1] != data->bTag_last_write) { if (buffer[1] != data->bTag_last_write) {
dev_err(dev, "Device sent reply with wrong bTag: %u != %u\n", buffer[1], data->bTag_last_write); dev_err(dev, "Device sent reply with wrong bTag: %u != %u\n", buffer[1], data->bTag_last_write);
if (data->auto_abort) if (file_data->auto_abort)
usbtmc_ioctl_abort_bulk_in(data); usbtmc_ioctl_abort_bulk_in(data);
goto exit; goto exit;
} }
@ -1440,7 +1442,7 @@ static ssize_t usbtmc_read(struct file *filp, char __user *buf,
if (n_characters > this_part) { if (n_characters > this_part) {
dev_err(dev, "Device wants to return more data than requested: %u > %zu\n", n_characters, count); dev_err(dev, "Device wants to return more data than requested: %u > %zu\n", n_characters, count);
if (data->auto_abort) if (file_data->auto_abort)
usbtmc_ioctl_abort_bulk_in(data); usbtmc_ioctl_abort_bulk_in(data);
goto exit; goto exit;
} }
@ -1582,7 +1584,7 @@ static ssize_t usbtmc_write(struct file *filp, const char __user *buf,
if (retval < 0) { if (retval < 0) {
dev_err(&data->intf->dev, dev_err(&data->intf->dev,
"Unable to send data, error %d\n", retval); "Unable to send data, error %d\n", retval);
if (data->auto_abort) if (file_data->auto_abort)
usbtmc_ioctl_abort_bulk_out(data); usbtmc_ioctl_abort_bulk_out(data);
goto exit; goto exit;
} }
@ -2091,6 +2093,7 @@ static long usbtmc_ioctl(struct file *file, unsigned int cmd, unsigned long arg)
struct usbtmc_file_data *file_data; struct usbtmc_file_data *file_data;
struct usbtmc_device_data *data; struct usbtmc_device_data *data;
int retval = -EBADRQC; int retval = -EBADRQC;
__u8 tmp_byte;
file_data = file->private_data; file_data = file->private_data;
data = file_data->data; data = file_data->data;
@ -2207,6 +2210,12 @@ static long usbtmc_ioctl(struct file *file, unsigned int cmd, unsigned long arg)
(__u8 __user *)arg); (__u8 __user *)arg);
break; break;
case USBTMC_IOCTL_AUTO_ABORT:
retval = get_user(tmp_byte, (unsigned char __user *)arg);
if (retval == 0)
file_data->auto_abort = !!tmp_byte;
break;
case USBTMC_IOCTL_CANCEL_IO: case USBTMC_IOCTL_CANCEL_IO:
retval = usbtmc_ioctl_cancel_io(file_data); retval = usbtmc_ioctl_cancel_io(file_data);
break; break;

View File

@ -99,6 +99,7 @@ struct usbtmc_message {
#define USBTMC488_IOCTL_WAIT_SRQ _IOW(USBTMC_IOC_NR, 23, __u32) #define USBTMC488_IOCTL_WAIT_SRQ _IOW(USBTMC_IOC_NR, 23, __u32)
#define USBTMC_IOCTL_MSG_IN_ATTR _IOR(USBTMC_IOC_NR, 24, __u8) #define USBTMC_IOCTL_MSG_IN_ATTR _IOR(USBTMC_IOC_NR, 24, __u8)
#define USBTMC_IOCTL_AUTO_ABORT _IOW(USBTMC_IOC_NR, 25, __u8)
/* Cancel and cleanup asynchronous calls */ /* Cancel and cleanup asynchronous calls */
#define USBTMC_IOCTL_CANCEL_IO _IO(USBTMC_IOC_NR, 35) #define USBTMC_IOCTL_CANCEL_IO _IO(USBTMC_IOC_NR, 35)