Use new provided function instead of an iteration through all allocated
blocks.
Signed-off-by: NJan Kara <jack@suse.cz>
Signed-off-by: NJens Axboe <axboe@fb.com>