From mboxrd@z Thu Jan 1 00:00:00 1970 Received: from smtp.kernel.org (aws-us-west-2-korg-mail-1.web.codeaurora.org [10.30.226.201]) (using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits)) (No client certificate requested) by smtp.subspace.kernel.org (Postfix) with ESMTPS id 263F417E6 for ; Sat, 22 Oct 2022 01:48:20 +0000 (UTC) Received: by smtp.kernel.org (Postfix) with ESMTPSA id C183FC433C1 for ; Sat, 22 Oct 2022 01:48:19 +0000 (UTC) Authentication-Results: smtp.kernel.org; dkim=pass (1024-bit key) header.d=zx2c4.com header.i=@zx2c4.com header.b="MsQLprMd"; dkim=pass (1024-bit key) header.d=zx2c4.com header.i=@zx2c4.com header.b="RYModmzz" DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=zx2c4.com; s=20210105; t=1666403298; h=from:from:reply-to:subject:subject:date:date:message-id:message-id: to:to:cc:cc:mime-version:mime-version: content-transfer-encoding:content-transfer-encoding:resent-to: resent-from:resent-message-id:in-reply-to:in-reply-to: references:references; bh=6V7n7WLPPuhmK3sEMGuUnn+DVRADTUodeMZgVjm6mzw=; b=MsQLprMdb/GwOf9rMrFGwQIQWzfwQCe1Ip2lJumcBbi8vVpUAYLB41WNbz7vlZOkt3cgDo HenT+AWxzEvtnpKdBLAep0xcAPuD3ka5hBFMMX4T5P5baUaK6R9DPdSkZy1sEpAnqnxChn +5ZLBI6ljR2RNx9tW5SBFahjCr7opsw= Received: by mail.zx2c4.com (ZX2C4 Mail Server) with ESMTPSA id 357a981d (TLSv1.3:TLS_AES_256_GCM_SHA384:256:NO) for ; Sat, 22 Oct 2022 01:48:18 +0000 (UTC) Resent-From: "Jason A. Donenfeld" Resent-Date: Fri, 21 Oct 2022 21:48:18 -0400 Resent-Message-ID: Resent-To: patches@lists.linux.dev Received: by 2002:ab0:3311:0:b0:3e5:cdde:6278 with SMTP id r17csp1682608uao; Fri, 21 Oct 2022 18:44:32 -0700 (PDT) X-Google-Smtp-Source: AMsMyM5rkmgvLyv1tZWH/rOoaO36YCgiBz4+b5oMkwMwcx1A3m0/ApTgixq5WxfKfPQfbOoZZUCS X-Received: by 2002:a05:620a:131b:b0:6ec:bce7:99a9 with SMTP id o27-20020a05620a131b00b006ecbce799a9mr15988700qkj.84.1666403071799; Fri, 21 Oct 2022 18:44:31 -0700 (PDT) ARC-Seal: i=1; a=rsa-sha256; t=1666403071; cv=none; d=google.com; s=arc-20160816; b=bwrXgns/yCtE7lotuOg9ATNYbjV5H9KslI1jnavk1AYDioVkFFbE6al6Oa/hPxOWUa 31VMWJR4W3nfCN67HG3FjaTc6ArHXDzc09PeYHaGiR5ccJYpzpblfpTL4jMhHXNNRV3F XGyoREZLWtwG2x4owiM104cR5KJ+GxFAHb0FImbMY/M7SIizb0ecJ0V5mJUGnSkKwbtj C8xUHsYnAN5E6hUnjvfnOoJmz6BLeJP1mucl2QHLPQPo6z16GGrYQX5BiaRog8t7+XcT JVcgrPNMCrAfW5pmZknmmwGpWzLuSvlQ9AfeZ9mqoR6VfwI4sxJkY/JEtSxdfI2XPo7Q vNug== ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=arc-20160816; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:dkim-signature; bh=6V7n7WLPPuhmK3sEMGuUnn+DVRADTUodeMZgVjm6mzw=; b=lsX4V2987K618lRihU0AM3EW3x9K7DonkMytFnVlGZ6uIWwot07HQkhq+CEKx/plK5 b0k1jwYEJSYjVCUaGauVkEXdvnGt5qtQ2eFABwSTw8nqH4lNkYMIUfZYNLhbLBaPBEsN emB4eUFKQmutJlFmrnBRrK2lGELJSPmFoAyLlvN2o6/MEB+HY1Eq/VqFY5p+/RPFhbfK RUv2GX4efoTkSYTz1U6QnQVx3lTlaSQ30WlouZmJX73sjjzQNxnd/LjUTSmq8fLUeK8s rYA/caJj2BuGdgknvdWLlOwYsuoGb7kf4oapLvUYdIraNQAuQI5McYiK3asNaRFE63s0 MwPg== ARC-Authentication-Results: i=1; mx.google.com; dkim=pass header.i=@zx2c4.com header.s=20210105 header.b=RYModmzz; spf=pass (google.com: domain of jason@zx2c4.com designates 104.131.123.232 as permitted sender) smtp.mailfrom=Jason@zx2c4.com; dmarc=pass (p=QUARANTINE sp=QUARANTINE dis=NONE) header.from=zx2c4.com Received: from mail.zx2c4.com (mail.zx2c4.com. [104.131.123.232]) by mx.google.com with ESMTPS id t26-20020ac8761a000000b0039cbb3f4c8fsi12471333qtq.591.2022.10.21.18.44.31 for (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Fri, 21 Oct 2022 18:44:31 -0700 (PDT) Received-SPF: pass (google.com: domain of jason@zx2c4.com designates 104.131.123.232 as permitted sender) client-ip=104.131.123.232; Authentication-Results: mx.google.com; dkim=pass header.i=@zx2c4.com header.s=20210105 header.b=RYModmzz; spf=pass (google.com: domain of jason@zx2c4.com designates 104.131.123.232 as permitted sender) smtp.mailfrom=Jason@zx2c4.com; dmarc=pass (p=QUARANTINE sp=QUARANTINE dis=NONE) header.from=zx2c4.com DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=zx2c4.com; s=20210105; t=1666403065; h=from:from:reply-to:subject:subject:date:date:message-id:message-id: to:to:cc:cc:mime-version:mime-version: content-transfer-encoding:content-transfer-encoding: in-reply-to:in-reply-to:references:references; bh=6V7n7WLPPuhmK3sEMGuUnn+DVRADTUodeMZgVjm6mzw=; b=RYModmzzec1OgskkcQmS88KlPnl8bEDxVs4NFDzK61wcmW5QvikH5xeZo68AnjN6PonqCT u/znWy0ao8IKhXHXs/gq7RjG6sZSoZ1g/F1NTIf3MbxHxZ33HyaUZgMQA9GjldEfSrFsFn 1AU24c29vQR9f+tA0s5hbhp1IEc7HeQ= Received: by mail.zx2c4.com (ZX2C4 Mail Server) with ESMTPSA id bec1a5aa (TLSv1.3:TLS_AES_256_GCM_SHA384:256:NO); Sat, 22 Oct 2022 01:44:25 +0000 (UTC) From: "Jason A. Donenfeld" To: linux-kernel@vger.kernel.org Cc: "Jason A. Donenfeld" , Kees Cook , Greg Kroah-Hartman , Jakub Kicinski , Russell King , Catalin Marinas , Thomas Bogendoerfer , Heiko Carstens , Herbert Xu , =?UTF-8?q?Christoph=20B=C3=B6hmwalder?= , Jani Nikula , Jason Gunthorpe , Sakari Ailus , "Martin K . Petersen" , Theodore Ts'o , Andreas Dilger , Jaegeuk Kim , Richard Weinberger , "Darrick J . Wong" , SeongJae Park , Thomas Gleixner , Andrew Morton , Michael Ellerman , Helge Deller , netdev@vger.kernel.org, linux-crypto@vger.kernel.org, linux-block@vger.kernel.org, linux-fsdevel@vger.kernel.org, linux-media@vger.kernel.org, linux-arm-kernel@lists.infradead.org, loongarch@lists.linux.dev, linux-mips@vger.kernel.org, linuxppc-dev@lists.ozlabs.org, linux-mmc@vger.kernel.org, linux-parisc@vger.kernel.org Subject: [PATCH v1 4/5] treewide: use get_random_u32_{above,below}() instead of manual loop Date: Fri, 21 Oct 2022 21:44:02 -0400 Message-Id: <20221022014403.3881893-5-Jason@zx2c4.com> In-Reply-To: <20221022014403.3881893-1-Jason@zx2c4.com> References: <20221022014403.3881893-1-Jason@zx2c4.com> Precedence: bulk X-Mailing-List: patches@lists.linux.dev List-Id: List-Subscribe: List-Unsubscribe: MIME-Version: 1.0 Content-Transfer-Encoding: 8bit These cases were done with this Coccinelle: @@ expression E; identifier I; @@ - do { ... - I = get_random_u32(); ... - } while (I > E); + I = get_random_u32_below(E + 1); @@ expression E; identifier I; @@ - do { ... - I = get_random_u32(); ... - } while (I >= E); + I = get_random_u32_below(E); @@ expression E; identifier I; @@ - do { ... - I = get_random_u32(); ... - } while (I < E); + I = get_random_u32_above(E - 1); @@ expression E; identifier I; @@ - do { ... - I = get_random_u32(); ... - } while (I <= E); + I = get_random_u32_above(E); @@ identifier I; @@ - do { ... - I = get_random_u32(); ... - } while (!I); + I = get_random_u32_above(0); @@ identifier I; @@ - do { ... - I = get_random_u32(); ... - } while (I == 0); + I = get_random_u32_above(0); @@ expression E; identifier I; @@ - E + 1 + get_random_u32_below(U32_MAX - E) + get_random_u32_above(E) Signed-off-by: Jason A. Donenfeld --- fs/ext4/mmp.c | 8 +------- lib/test_fprobe.c | 5 +---- lib/test_kprobes.c | 5 +---- net/ipv6/output_core.c | 8 +------- net/vmw_vsock/af_vsock.c | 3 +-- 5 files changed, 5 insertions(+), 24 deletions(-) diff --git a/fs/ext4/mmp.c b/fs/ext4/mmp.c index 588cb09c5291..4681fff6665f 100644 --- a/fs/ext4/mmp.c +++ b/fs/ext4/mmp.c @@ -262,13 +262,7 @@ void ext4_stop_mmpd(struct ext4_sb_info *sbi) */ static unsigned int mmp_new_seq(void) { - u32 new_seq; - - do { - new_seq = get_random_u32(); - } while (new_seq > EXT4_MMP_SEQ_MAX); - - return new_seq; + return get_random_u32_below(EXT4_MMP_SEQ_MAX + 1); } /* diff --git a/lib/test_fprobe.c b/lib/test_fprobe.c index e0381b3ec410..1fb56cf5e5ce 100644 --- a/lib/test_fprobe.c +++ b/lib/test_fprobe.c @@ -144,10 +144,7 @@ static unsigned long get_ftrace_location(void *func) static int fprobe_test_init(struct kunit *test) { - do { - rand1 = get_random_u32(); - } while (rand1 <= div_factor); - + rand1 = get_random_u32_above(div_factor); target = fprobe_selftest_target; target2 = fprobe_selftest_target2; target_ip = get_ftrace_location(target); diff --git a/lib/test_kprobes.c b/lib/test_kprobes.c index eeb1d728d974..1c95e5719802 100644 --- a/lib/test_kprobes.c +++ b/lib/test_kprobes.c @@ -339,10 +339,7 @@ static int kprobes_test_init(struct kunit *test) stacktrace_target = kprobe_stacktrace_target; internal_target = kprobe_stacktrace_internal_target; stacktrace_driver = kprobe_stacktrace_driver; - - do { - rand1 = get_random_u32(); - } while (rand1 <= div_factor); + rand1 = get_random_u32_above(div_factor); return 0; } diff --git a/net/ipv6/output_core.c b/net/ipv6/output_core.c index 2685c3f15e9d..b5205311f372 100644 --- a/net/ipv6/output_core.c +++ b/net/ipv6/output_core.c @@ -15,13 +15,7 @@ static u32 __ipv6_select_ident(struct net *net, const struct in6_addr *dst, const struct in6_addr *src) { - u32 id; - - do { - id = get_random_u32(); - } while (!id); - - return id; + return get_random_u32_above(0); } /* This function exists only for tap drivers that must support broken diff --git a/net/vmw_vsock/af_vsock.c b/net/vmw_vsock/af_vsock.c index f185f57c34e7..9cfed7fd8740 100644 --- a/net/vmw_vsock/af_vsock.c +++ b/net/vmw_vsock/af_vsock.c @@ -626,8 +626,7 @@ static int __vsock_bind_connectible(struct vsock_sock *vsk, struct sockaddr_vm new_addr; if (!port) - port = LAST_RESERVED_PORT + 1 + - get_random_u32_below(U32_MAX - LAST_RESERVED_PORT); + port = get_random_u32_above(LAST_RESERVED_PORT); vsock_addr_init(&new_addr, addr->svm_cid, addr->svm_port); -- 2.38.1