|
@@ -322,23 +322,34 @@ static struct stub_priv *stub_priv_alloc(struct stub_device *sdev,
|
|
|
return priv;
|
|
|
}
|
|
|
|
|
|
-static int get_pipe(struct stub_device *sdev, int epnum, int dir)
|
|
|
+static int get_pipe(struct stub_device *sdev, struct usbip_header *pdu)
|
|
|
{
|
|
|
struct usb_device *udev = sdev->udev;
|
|
|
struct usb_host_endpoint *ep;
|
|
|
struct usb_endpoint_descriptor *epd = NULL;
|
|
|
+ int epnum = pdu->base.ep;
|
|
|
+ int dir = pdu->base.direction;
|
|
|
+
|
|
|
+ if (epnum < 0 || epnum > 15)
|
|
|
+ goto err_ret;
|
|
|
|
|
|
if (dir == USBIP_DIR_IN)
|
|
|
ep = udev->ep_in[epnum & 0x7f];
|
|
|
else
|
|
|
ep = udev->ep_out[epnum & 0x7f];
|
|
|
- if (!ep) {
|
|
|
- dev_err(&sdev->udev->dev, "no such endpoint?, %d\n",
|
|
|
- epnum);
|
|
|
- BUG();
|
|
|
- }
|
|
|
+ if (!ep)
|
|
|
+ goto err_ret;
|
|
|
|
|
|
epd = &ep->desc;
|
|
|
+
|
|
|
+ /* validate transfer_buffer_length */
|
|
|
+ if (pdu->u.cmd_submit.transfer_buffer_length > INT_MAX) {
|
|
|
+ dev_err(&sdev->udev->dev,
|
|
|
+ "CMD_SUBMIT: -EMSGSIZE transfer_buffer_length %d\n",
|
|
|
+ pdu->u.cmd_submit.transfer_buffer_length);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
if (usb_endpoint_xfer_control(epd)) {
|
|
|
if (dir == USBIP_DIR_OUT)
|
|
|
return usb_sndctrlpipe(udev, epnum);
|
|
@@ -361,15 +372,31 @@ static int get_pipe(struct stub_device *sdev, int epnum, int dir)
|
|
|
}
|
|
|
|
|
|
if (usb_endpoint_xfer_isoc(epd)) {
|
|
|
+ /* validate packet size and number of packets */
|
|
|
+ unsigned int maxp, packets, bytes;
|
|
|
+
|
|
|
+ maxp = usb_endpoint_maxp(epd);
|
|
|
+ maxp *= usb_endpoint_maxp_mult(epd);
|
|
|
+ bytes = pdu->u.cmd_submit.transfer_buffer_length;
|
|
|
+ packets = DIV_ROUND_UP(bytes, maxp);
|
|
|
+
|
|
|
+ if (pdu->u.cmd_submit.number_of_packets < 0 ||
|
|
|
+ pdu->u.cmd_submit.number_of_packets > packets) {
|
|
|
+ dev_err(&sdev->udev->dev,
|
|
|
+ "CMD_SUBMIT: isoc invalid num packets %d\n",
|
|
|
+ pdu->u.cmd_submit.number_of_packets);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
if (dir == USBIP_DIR_OUT)
|
|
|
return usb_sndisocpipe(udev, epnum);
|
|
|
else
|
|
|
return usb_rcvisocpipe(udev, epnum);
|
|
|
}
|
|
|
|
|
|
+err_ret:
|
|
|
/* NOT REACHED */
|
|
|
- dev_err(&sdev->udev->dev, "get pipe, epnum %d\n", epnum);
|
|
|
- return 0;
|
|
|
+ dev_err(&sdev->udev->dev, "CMD_SUBMIT: invalid epnum %d\n", epnum);
|
|
|
+ return -1;
|
|
|
}
|
|
|
|
|
|
static void masking_bogus_flags(struct urb *urb)
|
|
@@ -433,7 +460,10 @@ static void stub_recv_cmd_submit(struct stub_device *sdev,
|
|
|
struct stub_priv *priv;
|
|
|
struct usbip_device *ud = &sdev->ud;
|
|
|
struct usb_device *udev = sdev->udev;
|
|
|
- int pipe = get_pipe(sdev, pdu->base.ep, pdu->base.direction);
|
|
|
+ int pipe = get_pipe(sdev, pdu);
|
|
|
+
|
|
|
+ if (pipe == -1)
|
|
|
+ return;
|
|
|
|
|
|
priv = stub_priv_alloc(sdev, pdu);
|
|
|
if (!priv)
|
|
@@ -452,7 +482,8 @@ static void stub_recv_cmd_submit(struct stub_device *sdev,
|
|
|
}
|
|
|
|
|
|
/* allocate urb transfer buffer, if needed */
|
|
|
- if (pdu->u.cmd_submit.transfer_buffer_length > 0) {
|
|
|
+ if (pdu->u.cmd_submit.transfer_buffer_length > 0 &&
|
|
|
+ pdu->u.cmd_submit.transfer_buffer_length <= INT_MAX) {
|
|
|
priv->urb->transfer_buffer =
|
|
|
kzalloc(pdu->u.cmd_submit.transfer_buffer_length,
|
|
|
GFP_KERNEL);
|