From: H Hartley Sweeten Date: Thu, 30 Oct 2014 18:10:50 +0000 (-0700) Subject: staging: comedi: comedi_buf: make comedi_buf_write_samples() add samples that fit X-Git-Tag: firefly_0821_release~176^2~2665^2~490 X-Git-Url: http://demsky.eecs.uci.edu/git/?a=commitdiff_plain;h=eb3a13230b78c3dc580b509e370862c632c90dae;p=firefly-linux-kernel-4.4.55.git 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 Reviewed-by: Ian Abbott Signed-off-by: Greg Kroah-Hartman --- diff --git a/drivers/staging/comedi/comedi_buf.c b/drivers/staging/comedi/comedi_buf.c index 6dd87cb2a8af..b43424d611ad 100644 --- 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)