Fixed poll/select on write subdevice.

diff --git a/comedi/comedi_fops.c b/comedi/comedi_fops.c
index 2e6f5885eb6f1eeaaee639002fc4aee7891e75a1..886ed7a1b66eb9aa19d903c9c26974ee1d9b36bf 100644 (file)
--- a/comedi/comedi_fops.c
+++ b/comedi/comedi_fops.c
@@ -1471,11 +1471,12 @@ static unsigned int comedi_poll(struct file *file, poll_table * wait)
        write_subdev = comedi_get_write_subdevice(dev, minor);
        if (write_subdev) {
                poll_wait(file, &write_subdev->async->wait_head, wait);
+               comedi_buf_write_alloc(write_subdev->async, write_subdev->async->prealloc_bufsz);
                if (!write_subdev->busy
                        || !(comedi_get_subdevice_runflags(write_subdev) &
                                SRF_RUNNING)
-                       || comedi_buf_write_n_available(write_subdev->async) >
-                       0) {
+                       || comedi_buf_write_n_allocated(write_subdev->async) >=
+                       bytes_per_sample(write_subdev->async->subdevice)) {
                        mask |= POLLOUT | POLLWRNORM;
                }
        }