124 	struct blkdev_dio *dio = bio->bi_private;  in blkdev_bio_end_io()  local
125 	bool should_dirty = dio->flags & DIO_SHOULD_DIRTY;  in blkdev_bio_end_io()
127 	if (bio->bi_status && !dio->bio.bi_status)  in blkdev_bio_end_io()
128 		dio->bio.bi_status = bio->bi_status;  in blkdev_bio_end_io()
130 	if (atomic_dec_and_test(&dio->ref)) {  in blkdev_bio_end_io()
131 		if (!(dio->flags & DIO_IS_SYNC)) {  in blkdev_bio_end_io()
132 			struct kiocb *iocb = dio->iocb;  in blkdev_bio_end_io()
137 			if (likely(!dio->bio.bi_status)) {  in blkdev_bio_end_io()
138 				ret = dio->size;  in blkdev_bio_end_io()
141 				ret = blk_status_to_errno(dio->bio.bi_status);  in blkdev_bio_end_io()
144 			dio->iocb->ki_complete(iocb, ret);  in blkdev_bio_end_io()
145 			bio_put(&dio->bio);  in blkdev_bio_end_io()
147 			struct task_struct *waiter = dio->waiter;  in blkdev_bio_end_io()
149 			WRITE_ONCE(dio->waiter, NULL);  in blkdev_bio_end_io()
167 	struct blkdev_dio *dio;  in __blkdev_direct_IO()  local
181 	dio = container_of(bio, struct blkdev_dio, bio);  in __blkdev_direct_IO()
182 	atomic_set(&dio->ref, 1);  in __blkdev_direct_IO()
191 		dio->flags = DIO_IS_SYNC;  in __blkdev_direct_IO()
192 		dio->waiter = current;  in __blkdev_direct_IO()
194 		dio->flags = 0;  in __blkdev_direct_IO()
195 		dio->iocb = iocb;  in __blkdev_direct_IO()
198 	dio->size = 0;  in __blkdev_direct_IO()
200 		dio->flags |= DIO_SHOULD_DIRTY;  in __blkdev_direct_IO()
206 		bio->bi_private = dio;  in __blkdev_direct_IO()
236 			if (dio->flags & DIO_SHOULD_DIRTY)  in __blkdev_direct_IO()
241 		dio->size += bio->bi_iter.bi_size;  in __blkdev_direct_IO()
249 		atomic_inc(&dio->ref);  in __blkdev_direct_IO()
261 		if (!READ_ONCE(dio->waiter))  in __blkdev_direct_IO()
268 		ret = blk_status_to_errno(dio->bio.bi_status);  in __blkdev_direct_IO()
270 		ret = dio->size;  in __blkdev_direct_IO()
272 	bio_put(&dio->bio);  in __blkdev_direct_IO()
278 	struct blkdev_dio *dio = container_of(bio, struct blkdev_dio, bio);  in blkdev_bio_end_io_async()  local
279 	struct kiocb *iocb = dio->iocb;  in blkdev_bio_end_io_async()
285 		ret = dio->size;  in blkdev_bio_end_io_async()
293 	if (dio->flags & DIO_SHOULD_DIRTY) {  in blkdev_bio_end_io_async()
308 	struct blkdev_dio *dio;  in __blkdev_direct_IO_async()  local
320 	dio = container_of(bio, struct blkdev_dio, bio);  in __blkdev_direct_IO_async()
321 	dio->flags = 0;  in __blkdev_direct_IO_async()
322 	dio->iocb = iocb;  in __blkdev_direct_IO_async()
342 	dio->size = bio->bi_iter.bi_size;  in __blkdev_direct_IO_async()
346 			dio->flags |= DIO_SHOULD_DIRTY;  in __blkdev_direct_IO_async()