From nobody Sat Feb 7 10:15:45 2026 Received: from mail-yb1-f225.google.com (mail-yb1-f225.google.com [209.85.219.225]) (using TLSv1.2 with cipher ECDHE-RSA-AES128-GCM-SHA256 (128/128 bits)) (No client certificate requested) by smtp.subspace.kernel.org (Postfix) with ESMTPS id E9F7F4A1D for ; Mon, 5 May 2025 17:26:31 +0000 (UTC) Authentication-Results: smtp.subspace.kernel.org; arc=none smtp.client-ip=209.85.219.225 ARC-Seal: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1746465994; cv=none; b=bSyIJv1St3HLKltxhFaCxbrAwXzFVR8gSKG3VsAhu57hPI92pLF1vOf8o9ZlqqEVW8AHgfM4o34t4KYsJOErcgn71f75WQ3aqkzw/7gDQh7vuRtTpN0wHsew1fZv4E/joVn2ehFHzhlyFFPdn+b3nJ350NH70qMezwy1PJhATTA= ARC-Message-Signature: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1746465994; c=relaxed/simple; bh=aODsyg3joXcQ287FoVhkIV5aJfVBMDhrX2MPoMsF9/I=; h=From:To:Cc:Subject:Date:Message-ID:MIME-Version; b=hWgE7MmOnOHtDWb7ErcMkj5KnnEchgdONnB34AJktfZw3ItaAzAgASCCIeUJfRjW6IwK+hdrB4VhY22njua5dzp3/yoF/L+0vOI56bGoKXwvKrdk1IWseyLBQY4ckFg8NFhZjj+ag+y6mj+A1Ue93k2iS7ALJ24S1GxNUeyrfAM= ARC-Authentication-Results: i=1; smtp.subspace.kernel.org; dmarc=pass (p=reject dis=none) header.from=purestorage.com; spf=fail smtp.mailfrom=purestorage.com; dkim=pass (2048-bit key) header.d=purestorage.com header.i=@purestorage.com header.b=GbKWrNHz; arc=none smtp.client-ip=209.85.219.225 Authentication-Results: smtp.subspace.kernel.org; dmarc=pass (p=reject dis=none) header.from=purestorage.com Authentication-Results: smtp.subspace.kernel.org; spf=fail smtp.mailfrom=purestorage.com Authentication-Results: smtp.subspace.kernel.org; dkim=pass (2048-bit key) header.d=purestorage.com header.i=@purestorage.com header.b="GbKWrNHz" Received: by mail-yb1-f225.google.com with SMTP id 3f1490d57ef6-e75b6f8c766so40340276.3 for ; Mon, 05 May 2025 10:26:31 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=purestorage.com; s=google2022; t=1746465991; x=1747070791; darn=vger.kernel.org; h=content-transfer-encoding:mime-version:message-id:date:subject:cc :to:from:from:to:cc:subject:date:message-id:reply-to; bh=DDBk+OYzCApSnK8VVZgyIgA3ncZhmRQ/48wNK7Y0ijk=; b=GbKWrNHzT5fOlzhH4jPGQXmfFjRfiWXRy59ev1xoqxCppbyi8GibE+zJcxyLeZXoIM MiqJt4b59XkuZvrmywTbQk0UT2C/hJ4A6o8y8yUSAX+FpaQOpct7UppFRzejtPy836I3 tUUtvxiJF4KWhqD5V27jjxACWBypiU30O/+iX21UIr0EXro2aN+zSUFjfwIfgZTSOoHU 6dI6FCLy1mwNUkQtPHI9bTjlSc0ySERArgUFBw5Ll53jhPtXGpKIPQT9GD4EnXoILd5G Ty+0A3Mpclxtgd2ewYJ/9eM2XEBt0qVA8ah4lKeQ9pTyn8ExHVJt9tyS5xodUUJfAadB jTIQ== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1746465991; x=1747070791; h=content-transfer-encoding:mime-version:message-id:date:subject:cc :to:from:x-gm-message-state:from:to:cc:subject:date:message-id :reply-to; bh=DDBk+OYzCApSnK8VVZgyIgA3ncZhmRQ/48wNK7Y0ijk=; b=HF3aIgfS/VxJMXYP8XKRLF6N1IBlP++WzKwigXa8q6xI8H4qDWTWIrRBhubvhf7CgV OguAwADpFECv3p+YS4n7BofNqrXK95RhDr9E0EQgR8ZC+3wUMQgmkFwH7pIUfnwkiDwi 3RekNCafpbGF+L6iDGKRJFLRthZr+9yRdXs2FQqCYJNJM1u6g+M9IzZ85UQox3l7gn2f yc726e+BW3Fqg3b85YBGU539jBRJwtZzLbKRe0nAWGz9OkVIjPc0JeY1TUH12pKw8YS1 0ZY/NUfbFagSIpN7fkgrCBhDlYdTL7Bbv0u56NE7LqRqCPRVUD4xBQ1S2RWvxrSyxr0+ Lfuw== X-Forwarded-Encrypted: i=1; AJvYcCUGoLd7XodLvBzqDsZtkOISNWxjRXmKoqQB0Ht4S0+qm81fQXqLbFL3XIoyKpye0VIbbgRKF4wXghw8o4E=@vger.kernel.org X-Gm-Message-State: AOJu0YxvjGLcHkdc9ebGNuqTANGNNYq6y2MQS7wxCtbSX667n/M/pgYG cRs1iy0QYdSR7MMydYVlhgnuQwNAj06+/yHbPPb8Kb5l60BA+qBws5sZDK/tLnNqkomVJFPlSd0 oCdJp0HYBjQLLCR95STJqrH8DCaKYgsl45SaO7ruzkn2r9z+1 X-Gm-Gg: ASbGncsm20RNgxJDtMskmLQsz/iYT2pK4biptbWyH3pJpP+kT3rk4U+nrDytGIKUPqa vRRCDRHl1D225GM+3IMwedw67pi3QbA58tB/nNRmPokjlILUjVXBl4UsThZ+1y6BOKhPThb9lSo ZyQwsc2q29N8FI3tu16DeM63N9LwWXJdBlMLQg3PCAHBbHum00t8D/3OjIEs4n01O4R2FQJP5OO DxhJR0FA9CngdKlj6vrse/BtoJjw60p9wahxGid53ggT0oeSPh+R8QqzhFATlR3TVr96I2mN83D idi5Jf1NgAoG+jU2gS0hnIc1Edq1OQ== X-Google-Smtp-Source: AGHT+IF8JQQZ9iJZET0yqNKPYRUGBpoq1VA7CgS5xHctkCNPAgyxmvjK72Iq3Quz/EaHkhR44JzXhXdmNEo9 X-Received: by 2002:a05:690c:7108:b0:6ff:8418:5afb with SMTP id 00721157ae682-708cedf5b65mr78859887b3.5.1746465990790; Mon, 05 May 2025 10:26:30 -0700 (PDT) Received: from c7-smtp-2023.dev.purestorage.com ([2620:125:9017:12:36:3:5:0]) by smtp-relay.gmail.com with ESMTPS id 00721157ae682-708c46f219csm20088657b3.60.2025.05.05.10.26.30 (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Mon, 05 May 2025 10:26:30 -0700 (PDT) X-Relaying-Domain: purestorage.com Received: from dev-csander.dev.purestorage.com (dev-csander.dev.purestorage.com [10.7.70.37]) by c7-smtp-2023.dev.purestorage.com (Postfix) with ESMTP id 2FBB2340350; Mon, 5 May 2025 11:26:30 -0600 (MDT) Received: by dev-csander.dev.purestorage.com (Postfix, from userid 1557716354) id 29BA6E41993; Mon, 5 May 2025 11:26:30 -0600 (MDT) From: Caleb Sander Mateos To: Ming Lei , Jens Axboe Cc: Caleb Sander Mateos , linux-block@vger.kernel.org, linux-kernel@vger.kernel.org Subject: [PATCH v2] ublk: consolidate UBLK_IO_FLAG_OWNED_BY_SRV checks Date: Mon, 5 May 2025 11:26:23 -0600 Message-ID: <20250505172624.1121839-1-csander@purestorage.com> X-Mailer: git-send-email 2.45.2 Precedence: bulk X-Mailing-List: linux-kernel@vger.kernel.org List-Id: List-Subscribe: List-Unsubscribe: MIME-Version: 1.0 Content-Transfer-Encoding: quoted-printable Content-Type: text/plain; charset="utf-8" Every ublk I/O command except UBLK_IO_FETCH_REQ checks that the ublk_io has UBLK_IO_FLAG_OWNED_BY_SRV set. Consolidate the separate checks into a single one in __ublk_ch_uring_cmd(), analogous to those for UBLK_IO_FLAG_ACTIVE and UBLK_IO_FLAG_NEED_GET_DATA. Signed-off-by: Caleb Sander Mateos Reviewed-by: Ming Lei --- v2: rebase drivers/block/ublk_drv.c | 26 ++++++++------------------ 1 file changed, 8 insertions(+), 18 deletions(-) diff --git a/drivers/block/ublk_drv.c b/drivers/block/ublk_drv.c index 2c8f28c0d0a1..3650bab40dd0 100644 --- a/drivers/block/ublk_drv.c +++ b/drivers/block/ublk_drv.c @@ -1924,20 +1924,16 @@ static void ublk_io_release(void *priv) static int ublk_register_io_buf(struct io_uring_cmd *cmd, const struct ublk_queue *ubq, unsigned int tag, unsigned int index, unsigned int issue_flags) { struct ublk_device *ub =3D cmd->file->private_data; - const struct ublk_io *io =3D &ubq->ios[tag]; struct request *req; int ret; =20 if (!ublk_support_zero_copy(ubq)) return -EINVAL; =20 - if (!(io->flags & UBLK_IO_FLAG_OWNED_BY_SRV)) - return -EINVAL; - req =3D __ublk_check_and_get_req(ub, ubq, tag, 0); if (!req) return -EINVAL; =20 ret =3D io_buffer_register_bvec(cmd, req, ublk_io_release, index, @@ -1949,21 +1945,16 @@ static int ublk_register_io_buf(struct io_uring_cmd= *cmd, =20 return 0; } =20 static int ublk_unregister_io_buf(struct io_uring_cmd *cmd, - const struct ublk_queue *ubq, unsigned int tag, + const struct ublk_queue *ubq, unsigned int index, unsigned int issue_flags) { - const struct ublk_io *io =3D &ubq->ios[tag]; - if (!ublk_support_zero_copy(ubq)) return -EINVAL; =20 - if (!(io->flags & UBLK_IO_FLAG_OWNED_BY_SRV)) - return -EINVAL; - return io_buffer_unregister_bvec(cmd, index, issue_flags); } =20 static int ublk_fetch(struct io_uring_cmd *cmd, struct ublk_queue *ubq, struct ublk_io *io, __u64 buf_addr) @@ -2013,16 +2004,12 @@ static int ublk_fetch(struct io_uring_cmd *cmd, str= uct ublk_queue *ubq, =20 static int ublk_commit_and_fetch(const struct ublk_queue *ubq, struct ublk_io *io, struct io_uring_cmd *cmd, const struct ublksrv_io_cmd *ub_cmd) { - struct request *req; - - if (!(io->flags & UBLK_IO_FLAG_OWNED_BY_SRV)) - return -EINVAL; + struct request *req =3D io->req; =20 - req =3D io->req; if (ublk_need_map_io(ubq)) { /* * COMMIT_AND_FETCH_REQ has to provide IO buffer if * NEED GET DATA is not enabled or it is Read IO. */ @@ -2102,10 +2089,15 @@ static int __ublk_ch_uring_cmd(struct io_uring_cmd = *cmd, if (io->flags & UBLK_IO_FLAG_ACTIVE) { ret =3D -EBUSY; goto out; } =20 + /* only UBLK_IO_FETCH_REQ is allowed if io is not OWNED_BY_SRV */ + if (!(io->flags & UBLK_IO_FLAG_OWNED_BY_SRV) && + _IOC_NR(cmd_op) !=3D UBLK_IO_FETCH_REQ) + goto out; + /* * ensure that the user issues UBLK_IO_NEED_GET_DATA * iff the driver have set the UBLK_IO_FLAG_NEED_GET_DATA. */ if ((!!(io->flags & UBLK_IO_FLAG_NEED_GET_DATA)) @@ -2119,11 +2111,11 @@ static int __ublk_ch_uring_cmd(struct io_uring_cmd = *cmd, ret =3D -EINVAL; switch (_IOC_NR(cmd_op)) { case UBLK_IO_REGISTER_IO_BUF: return ublk_register_io_buf(cmd, ubq, tag, ub_cmd->addr, issue_flags); case UBLK_IO_UNREGISTER_IO_BUF: - return ublk_unregister_io_buf(cmd, ubq, tag, ub_cmd->addr, issue_flags); + return ublk_unregister_io_buf(cmd, ubq, ub_cmd->addr, issue_flags); case UBLK_IO_FETCH_REQ: ret =3D ublk_fetch(cmd, ubq, io, ub_cmd->addr); if (ret) goto out; break; @@ -2131,12 +2123,10 @@ static int __ublk_ch_uring_cmd(struct io_uring_cmd = *cmd, ret =3D ublk_commit_and_fetch(ubq, io, cmd, ub_cmd); if (ret) goto out; break; case UBLK_IO_NEED_GET_DATA: - if (!(io->flags & UBLK_IO_FLAG_OWNED_BY_SRV)) - goto out; io->addr =3D ub_cmd->addr; if (!ublk_get_data(ubq, io)) return -EIOCBQUEUED; =20 return UBLK_IO_RES_OK; --=20 2.45.2