staging: comedi: comedi_buf: make comedi_buf_write_samples() add samples that fit

This function currently fails if the number of samples to add would
overflow the async buffer. Modify it to add the samples that fit so
at least some of the sample data is returned to the user.

Signed-off-by: H Hartley Sweeten <hsweeten@visionengravers.com>
Reviewed-by: Ian Abbott <abbotti@mev.co.uk>
Signed-off-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>

diff --git a/drivers/staging/comedi/comedi_buf.c b/drivers/staging/comedi/comedi_buf.c
index 6dd87cb2a8af13de793909cdcb95bc1f2f282ab0..b43424d611ad7ca3154b404b33541155fdee128a 100644 (file)
--- a/drivers/staging/comedi/comedi_buf.c
+++ b/drivers/staging/comedi/comedi_buf.c
@@ -484,12 +484,16 @@ unsigned int comedi_buf_write_samples(struct comedi_subdevice *s,
        unsigned int max_samples;
        unsigned int nbytes;
 
-       /* make sure there is enought room in the buffer for all the samples */
+       /*
+        * Make sure there is enough room in the buffer for all the samples.
+        * If not, clamp the nsamples to the number that will fit, flag the
+        * buffer overrun and add the samples that fit.
+        */
        max_samples = comedi_buf_write_n_available(s) / bytes_per_sample(s);
        if (nsamples > max_samples) {
                dev_warn(s->device->class_dev, "buffer overrun\n");
                s->async->events |= COMEDI_CB_OVERFLOW;
-               return 0;
+               nsamples = max_samples;
        }
 
        if (nsamples == 0)