From 7767cd3b50ecd52e1961850954e3c6ff7472d944 Mon Sep 17 00:00:00 2001 From: tx-tomcat Date: Sat, 1 Jun 2024 00:18:19 +0700 Subject: [PATCH] [move][move-linter] implement unnecessary while loop rules --- .../src/linters/shift_overflow.rs | 109 ------------------ .../src/linters/unnecessary_while_loop.rs | 78 +++++++++++++ .../tests/custom_rules/shift_overflow.exp | 24 ---- .../tests/custom_rules/shift_overflow.move | 9 -- .../correct_unnecessary_while_loop.move | 12 ++ .../incorrect_unnecessary_while_loop.exp | 13 +++ .../incorrect_unnecessary_while_loop.move | 12 ++ .../tests/linter/suppressed_lints.move | 11 ++ 8 files changed, 126 insertions(+), 142 deletions(-) delete mode 100644 external-crates/move/crates/move-compiler/src/linters/shift_overflow.rs create mode 100644 external-crates/move/crates/move-compiler/src/linters/unnecessary_while_loop.rs delete mode 100644 external-crates/move/crates/move-compiler/tests/custom_rules/shift_overflow.exp delete mode 100644 external-crates/move/crates/move-compiler/tests/custom_rules/shift_overflow.move create mode 100644 external-crates/move/crates/move-compiler/tests/linter/correct_unnecessary_while_loop.move create mode 100644 external-crates/move/crates/move-compiler/tests/linter/incorrect_unnecessary_while_loop.exp create mode 100644 external-crates/move/crates/move-compiler/tests/linter/incorrect_unnecessary_while_loop.move diff --git a/external-crates/move/crates/move-compiler/src/linters/shift_overflow.rs b/external-crates/move/crates/move-compiler/src/linters/shift_overflow.rs deleted file mode 100644 index a66d97525fb8e1..00000000000000 --- a/external-crates/move/crates/move-compiler/src/linters/shift_overflow.rs +++ /dev/null @@ -1,109 +0,0 @@ -//! Detect potential overflow scenarios where the number of bits being shifted exceeds the bit width of -//! the variable being shifted, which could lead to unintended behavior or loss of data. If such a -//! potential overflow is detected, a warning is generated to alert the developer. -use crate::{ - diag, - diagnostics::{ - codes::{custom, DiagnosticInfo, Severity}, - WarningFilters, - }, - expansion::ast::Value_, - naming::ast::{BuiltinTypeName_, TypeName_, Type_}, - parser::ast::BinOp_, - shared::{program_info::TypingProgramInfo, CompilationEnv}, - typing::{ - ast::{self as T, UnannotatedExp_}, - visitor::{TypingVisitorConstructor, TypingVisitorContext}, - }, -}; -use move_ir_types::location::Loc; -use std::str::FromStr; - -use super::{LinterDiagCategory, LINTER_DEFAULT_DIAG_CODE, LINT_WARNING_PREFIX}; - -const SHIFT_OPERATION_OVERFLOW_DIAG: DiagnosticInfo = custom( - LINT_WARNING_PREFIX, - Severity::Warning, - LinterDiagCategory::ShiftOperationOverflow as u8, - LINTER_DEFAULT_DIAG_CODE, - "Potential overflow detected. The number of bits being shifted exceeds the bit width of the variable being shifted.", -); - -pub struct ShiftOperationOverflow; - -pub struct Context<'a> { - env: &'a mut CompilationEnv, -} - -impl TypingVisitorConstructor for ShiftOperationOverflow { - type Context<'a> = Context<'a>; - - fn context<'a>( - env: &'a mut CompilationEnv, - _program_info: &'a TypingProgramInfo, - _program: &T::Program_, - ) -> Self::Context<'a> { - Context { env } - } -} - -impl TypingVisitorContext for Context<'_> { - fn visit_exp_custom(&mut self, exp: &mut T::Exp) -> bool { - // Check if the expression is a binary operation and if it is a shift operation. - if let UnannotatedExp_::BinopExp(lhs, op, _, rhs) = &exp.exp.value { - // can't do let UnannotatedExp_::BinopExp(lhs, BinOp_::Shl | BinOp_::Shr, _, rhs) = &exp.exp.value else { return }; - // because the op is Spanned and not BinOp_ - if matches!(op.value, BinOp_::Shl | BinOp_::Shr) { - match ( - get_bit_width(&lhs.ty.value), - get_shift_amount(&rhs.exp.value), - ) { - (Some(bit_width), Some(shift_amount)) if shift_amount >= bit_width => { - report_overflow(self.env, shift_amount, bit_width, op.loc); - } - _ => (), - } - } - } - false - } - fn add_warning_filter_scope(&mut self, filter: WarningFilters) { - self.env.add_warning_filter_scope(filter) - } - - fn pop_warning_filter_scope(&mut self) { - self.env.pop_warning_filter_scope() - } -} - -fn get_bit_width(ty: &Type_) -> Option { - ty.builtin_name().and_then(|typ| match typ.value { - BuiltinTypeName_::U8 => Some(8), - BuiltinTypeName_::U16 => Some(16), - BuiltinTypeName_::U32 => Some(32), - BuiltinTypeName_::U64 => Some(64), - BuiltinTypeName_::U128 => Some(128), - BuiltinTypeName_::U256 => Some(256), - _ => None, - }) -} - -fn get_shift_amount(value: &UnannotatedExp_) -> Option { - if let UnannotatedExp_::Value(v) = value { - match &v.value { - Value_::U8(v) => Some(*v as u128), - _ => None, - } - } else { - None - } -} - -fn report_overflow(env: &mut CompilationEnv, shift_amount: u128, bit_width: u128, loc: Loc) { - let msg = format!( - "The {} of bits being shifted exceeds the {} bit width of the variable being shifted.", - shift_amount, bit_width - ); - let diag = diag!(SHIFT_OPERATION_OVERFLOW_DIAG, (loc, msg)); - env.add_diag(diag); -} diff --git a/external-crates/move/crates/move-compiler/src/linters/unnecessary_while_loop.rs b/external-crates/move/crates/move-compiler/src/linters/unnecessary_while_loop.rs new file mode 100644 index 00000000000000..68e400dd33e2a7 --- /dev/null +++ b/external-crates/move/crates/move-compiler/src/linters/unnecessary_while_loop.rs @@ -0,0 +1,78 @@ +//! Encourages replacing `while(true)` with `loop` for infinite loops in Move for clarity and conciseness. +//! Identifies `while(true)` patterns, suggesting a more idiomatic approach using `loop`. +//! Aims to enhance code readability and adherence to Rust idioms. +use crate::{ + diag, + diagnostics::{ + codes::{custom, DiagnosticInfo, Severity}, + WarningFilters, + }, + expansion::ast::Value_, + shared::CompilationEnv, + typing::{ + ast::{self as T, UnannotatedExp_}, + visitor::{TypingVisitorConstructor, TypingVisitorContext}, + }, +}; +use move_ir_types::location::Loc; + +use super::{LinterDiagnosticCategory, LINT_WARNING_PREFIX, WHILE_TRUE_TO_LOOP_DIAG_CODE}; + +const WHILE_TRUE_TO_LOOP_DIAG: DiagnosticInfo = custom( + LINT_WARNING_PREFIX, + Severity::Warning, + LinterDiagnosticCategory::Complexity as u8, + WHILE_TRUE_TO_LOOP_DIAG_CODE, + "", +); + +pub struct WhileTrueToLoop; + +pub struct Context<'a> { + env: &'a mut CompilationEnv, +} + +impl TypingVisitorConstructor for WhileTrueToLoop { + type Context<'a> = Context<'a>; + + fn context<'a>(env: &'a mut CompilationEnv, _program: &T::Program) -> Self::Context<'a> { + Context { env } + } +} + +impl TypingVisitorContext for Context<'_> { + fn visit_exp_custom(&mut self, exp: &mut T::Exp) -> bool { + if let UnannotatedExp_::While(_, cond, _) = &exp.exp.value { + if is_condition_always_true(&cond.exp.value) { + report_while_true_to_loop(self.env, exp.exp.loc); + } + } + false + } + fn add_warning_filter_scope(&mut self, filter: WarningFilters) { + self.env.add_warning_filter_scope(filter) + } + + fn pop_warning_filter_scope(&mut self) { + self.env.pop_warning_filter_scope() + } +} + +fn is_condition_always_true(condition: &UnannotatedExp_) -> bool { + if let UnannotatedExp_::Value(val) = condition { + if let Value_::Bool(b) = &val.value { + return *b; + } + } + false +} +fn report_while_true_to_loop(env: &mut CompilationEnv, loc: Loc) { + let diag = diag!( + WHILE_TRUE_TO_LOOP_DIAG, + ( + loc, + "Detected `while(true) {}` loop. Consider replacing with `loop {}`" + ) + ); + env.add_diag(diag); +} diff --git a/external-crates/move/crates/move-compiler/tests/custom_rules/shift_overflow.exp b/external-crates/move/crates/move-compiler/tests/custom_rules/shift_overflow.exp deleted file mode 100644 index ed6efd6a2ea675..00000000000000 --- a/external-crates/move/crates/move-compiler/tests/custom_rules/shift_overflow.exp +++ /dev/null @@ -1,24 +0,0 @@ -warning[Lint W00001]: Potential overflow detected. The number of bits being shifted exceeds the bit width of the variable being shifted. - ┌─ tests/custom_rules/shift_overflow.move:5:20 - │ -5 │ let _b = x << 64; // Should not raise an issue - │ ^^ The 64 of bits being shifted exceeds the 64 bit width of the variable being shifted. - │ - = This warning can be suppressed with '#[allow(lint(shift_overflow))]' applied to the 'module' or module member ('const', 'fun', or 'struct') - -warning[Lint W00001]: Potential overflow detected. The number of bits being shifted exceeds the bit width of the variable being shifted. - ┌─ tests/custom_rules/shift_overflow.move:6:20 - │ -6 │ let _b = x << 65; // Should raise an issue - │ ^^ The 65 of bits being shifted exceeds the 64 bit width of the variable being shifted. - │ - = This warning can be suppressed with '#[allow(lint(shift_overflow))]' applied to the 'module' or module member ('const', 'fun', or 'struct') - -warning[Lint W00001]: Potential overflow detected. The number of bits being shifted exceeds the bit width of the variable being shifted. - ┌─ tests/custom_rules/shift_overflow.move:7:20 - │ -7 │ let _b = x >> 66; // Should raise an issue - │ ^^ The 66 of bits being shifted exceeds the 64 bit width of the variable being shifted. - │ - = This warning can be suppressed with '#[allow(lint(shift_overflow))]' applied to the 'module' or module member ('const', 'fun', or 'struct') - diff --git a/external-crates/move/crates/move-compiler/tests/custom_rules/shift_overflow.move b/external-crates/move/crates/move-compiler/tests/custom_rules/shift_overflow.move deleted file mode 100644 index e5dc71e6f6563e..00000000000000 --- a/external-crates/move/crates/move-compiler/tests/custom_rules/shift_overflow.move +++ /dev/null @@ -1,9 +0,0 @@ -module 0x42::M { - - fun func1(x: u64) { - let _b = x << 24; - let _b = x << 64; // Should raise an issue - let _b = x << 65; // Should raise an issue - let _b = x >> 66; // Should raise an issue - } -} \ No newline at end of file diff --git a/external-crates/move/crates/move-compiler/tests/linter/correct_unnecessary_while_loop.move b/external-crates/move/crates/move-compiler/tests/linter/correct_unnecessary_while_loop.move new file mode 100644 index 00000000000000..83b947f60a5b34 --- /dev/null +++ b/external-crates/move/crates/move-compiler/tests/linter/correct_unnecessary_while_loop.move @@ -0,0 +1,12 @@ +module 0x42::M { + + public fun finite_loop() { + let counter = 0; + loop { + if(counter == 10) { + break + }; + counter = counter + 1; + } + } +} \ No newline at end of file diff --git a/external-crates/move/crates/move-compiler/tests/linter/incorrect_unnecessary_while_loop.exp b/external-crates/move/crates/move-compiler/tests/linter/incorrect_unnecessary_while_loop.exp new file mode 100644 index 00000000000000..cc05539ff014dd --- /dev/null +++ b/external-crates/move/crates/move-compiler/tests/linter/incorrect_unnecessary_while_loop.exp @@ -0,0 +1,13 @@ +warning[Lint W01013]: + ┌─ tests/linter/incorrect_unnecessary_while_loop.move:5:9 + │ + 5 │ ╭ while (true) { + 6 │ │ if(counter == 10) { + 7 │ │ break + 8 │ │ }; + 9 │ │ counter = counter + 1; +10 │ │ } + │ ╰─────────^ Detected `while(true) {}` loop. Consider replacing with `loop {}` + │ + = This warning can be suppressed with '#[allow(lint(unnecessary_while_loop))]' applied to the 'module' or module member ('const', 'fun', or 'struct') + diff --git a/external-crates/move/crates/move-compiler/tests/linter/incorrect_unnecessary_while_loop.move b/external-crates/move/crates/move-compiler/tests/linter/incorrect_unnecessary_while_loop.move new file mode 100644 index 00000000000000..9284248e43f98e --- /dev/null +++ b/external-crates/move/crates/move-compiler/tests/linter/incorrect_unnecessary_while_loop.move @@ -0,0 +1,12 @@ +module 0x42::M { + + public fun finite_loop() { + let counter = 0; + while (true) { + if(counter == 10) { + break + }; + counter = counter + 1; + } + } +} diff --git a/external-crates/move/crates/move-compiler/tests/linter/suppressed_lints.move b/external-crates/move/crates/move-compiler/tests/linter/suppressed_lints.move index a75210a0364a70..20984770d03850 100644 --- a/external-crates/move/crates/move-compiler/tests/linter/suppressed_lints.move +++ b/external-crates/move/crates/move-compiler/tests/linter/suppressed_lints.move @@ -2,4 +2,15 @@ module 0x42::M { #[allow(lint(constant_naming))] const Another_BadName: u64 = 42; // Should trigger a warning + + #[allow(lint(unnecessary_while_loop))] + public fun finite_loop() { + let counter = 0; + while (true) { + if(counter == 10) { + break + }; + counter = counter + 1; + } + } }