[PATCH 02/14] ubd: untagle discard vs write zeroes not support handling

[Christoph Hellwig <hch@xxxxxx>]

Discard and Write Zeroes are different operation and implemented
by different fallocate opcodes for ubd.  If one fails the other one
can work and vice versa.

Split the code to disable the operations in ubd_handler to only
disable the operation that actually failed.

Fixes: 50109b5a03b4 ("um: Add support for DISCARD in the UBD Driver")
Signed-off-by: Christoph Hellwig <hch@xxxxxx>
Reviewed-by: Bart Van Assche <bvanassche@xxxxxxx>
Reviewed-by: Damien Le Moal <dlemoal@xxxxxxxxxx>
---
 arch/um/drivers/ubd_kern.c | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/arch/um/drivers/ubd_kern.c b/arch/um/drivers/ubd_kern.c
index 0c9542d58c01b7..093c87879d08ba 100644
--- a/arch/um/drivers/ubd_kern.c
+++ b/arch/um/drivers/ubd_kern.c
@@ -449,10 +449,11 @@ static int bulk_req_safe_read(
 
 static void ubd_end_request(struct io_thread_req *io_req)
 {
-       if (io_req->error == BLK_STS_NOTSUPP &&
-           req_op(io_req->req) == REQ_OP_DISCARD) {
-               blk_queue_max_discard_sectors(io_req->req->q, 0);
-               blk_queue_max_write_zeroes_sectors(io_req->req->q, 0);
+       if (io_req->error == BLK_STS_NOTSUPP) {
+               if (req_op(io_req->req) == REQ_OP_DISCARD)
+                       blk_queue_max_discard_sectors(io_req->req->q, 0);
+               else if (req_op(io_req->req) == REQ_OP_WRITE_ZEROES)
+                       blk_queue_max_write_zeroes_sectors(io_req->req->q, 0);
        }
        blk_mq_end_request(io_req->req, io_req->error);
        kfree(io_req);
-- 
2.43.0