From 24aec0d6976c36c36f524a1ea7bd11fdc82712fb 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 --- .../crates/move-compiler/src/linters/mod.rs | 32 +++-- .../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 -- .../incorrect_unnecessary_while_loop.exp | 50 ++++++++ .../incorrect_unnecessary_while_loop.move | 54 +++++++++ 7 files changed, 205 insertions(+), 151 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/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/mod.rs b/external-crates/move/crates/move-compiler/src/linters/mod.rs index 01ab4d3c2902f2..49a516fe8bff81 100644 --- a/external-crates/move/crates/move-compiler/src/linters/mod.rs +++ b/external-crates/move/crates/move-compiler/src/linters/mod.rs @@ -8,6 +8,7 @@ use crate::{ linters::constant_naming::ConstantNamingVisitor, typing::visitor::TypingVisitor, }; pub mod constant_naming; +mod unnecessary_while_loop; #[derive(Debug, Clone, Copy, PartialEq, Eq)] pub enum LintLevel { @@ -34,16 +35,26 @@ pub const ALLOW_ATTR_CATEGORY: &str = "lint"; pub const LINT_WARNING_PREFIX: &str = "Lint "; pub const CONSTANT_NAMING_FILTER_NAME: &str = "constant_naming"; pub const CONSTANT_NAMING_DIAG_CODE: u8 = 1; +pub const WHILE_TRUE_TO_LOOP_FILTER_NAME: &str = "while_true_to_loop"; +pub const WHILE_TRUE_TO_LOOP_DIAG_CODE: u8 = 4; pub fn known_filters() -> (Option, Vec) { ( Some(ALLOW_ATTR_CATEGORY.into()), - vec![WarningFilter::code( - Some(LINT_WARNING_PREFIX), - LinterDiagnosticCategory::Style as u8, - CONSTANT_NAMING_DIAG_CODE, - Some(CONSTANT_NAMING_FILTER_NAME), - )], + vec![ + WarningFilter::code( + Some(LINT_WARNING_PREFIX), + LinterDiagnosticCategory::Style as u8, + CONSTANT_NAMING_DIAG_CODE, + Some(CONSTANT_NAMING_FILTER_NAME), + ), + WarningFilter::code( + Some(LINT_WARNING_PREFIX), + LinterDiagnosticCategory::Complexity as u8, + WHILE_TRUE_TO_LOOP_DIAG_CODE, + Some(WHILE_TRUE_TO_LOOP_FILTER_NAME), + ), + ], ) } @@ -51,9 +62,12 @@ pub fn linter_visitors(level: LintLevel) -> Vec { match level { LintLevel::None | LintLevel::Default => vec![], LintLevel::All => { - vec![constant_naming::ConstantNamingVisitor::visitor( - ConstantNamingVisitor, - )] + vec![ + constant_naming::ConstantNamingVisitor::visitor(ConstantNamingVisitor), + unnecessary_while_loop::WhileTrueToLoop::visitor( + unnecessary_while_loop::WhileTrueToLoop, + ), + ] } } } 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..53954c54875544 --- /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 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 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 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/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..91d540de2e3af4 --- /dev/null +++ b/external-crates/move/crates/move-compiler/tests/linter/incorrect_unnecessary_while_loop.exp @@ -0,0 +1,50 @@ +warning[Lint W01004]: + ┌─ tests/linter/incorrect_unnecessary_while_loop.move:6:9 + │ +6 │ ╭ while (true) { +7 │ │ // This loop will run forever +8 │ │ } + │ ╰─────────^ Detected `while(true) {}` loop. Consider replacing with `loop {}` + │ + = This warning can be suppressed with '#[allow(lint(while_true_to_loop))]' applied to the 'module' or module member ('const', 'fun', or 'struct') + +warning[Lint W01004]: + ┌─ tests/linter/incorrect_unnecessary_while_loop.move:14:9 + │ +14 │ ╭ while (true) { +15 │ │ if (counter == 10) { +16 │ │ break +17 │ │ }; +18 │ │ counter = counter + 1; +19 │ │ } + │ ╰─────────^ Detected `while(true) {}` loop. Consider replacing with `loop {}` + │ + = This warning can be suppressed with '#[allow(lint(while_true_to_loop))]' applied to the 'module' or module member ('const', 'fun', or 'struct') + +warning[Lint W01004]: + ┌─ tests/linter/incorrect_unnecessary_while_loop.move:43:9 + │ +43 │ ╭ while (true) { +44 │ │ if (vector::length(&vec1) == 5) break; +45 │ │ vector::push_back(&mut vec1, vector::length(&vec1)); +46 │ │ }; + │ ╰─────────^ Detected `while(true) {}` loop. Consider replacing with `loop {}` + │ + = This warning can be suppressed with '#[allow(lint(while_true_to_loop))]' applied to the 'module' or module member ('const', 'fun', or 'struct') + +error[E07001]: referential transparency violated + ┌─ tests/linter/incorrect_unnecessary_while_loop.move:45:57 + │ +45 │ vector::push_back(&mut vec1, vector::length(&vec1)); + │ --------- ^^^^^ Invalid borrow of variable 'vec1' + │ │ + │ It is still being mutably borrowed by this reference + +error[E07001]: referential transparency violated + ┌─ tests/linter/incorrect_unnecessary_while_loop.move:51:57 + │ +51 │ vector::push_back(&mut vec2, vector::length(&vec2)); + │ --------- ^^^^^ Invalid borrow of variable 'vec2' + │ │ + │ It is still being mutably borrowed by this reference + 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..7e7922d791d149 --- /dev/null +++ b/external-crates/move/crates/move-compiler/tests/linter/incorrect_unnecessary_while_loop.move @@ -0,0 +1,54 @@ +module 0x42::loop_test { + use std::vector; + + // This function should trigger the linter + public fun infinite_loop() { + while (true) { + // This loop will run forever + } + } + + // This function should also trigger the linter + public fun finite_loop() { + let counter = 0; + while (true) { + if (counter == 10) { + break + }; + counter = counter + 1; + } + } + + // This function should not trigger the linter + public fun correct_infinite_loop() { + loop { + // This is the correct way to write an infinite loop + } + } + + // This function should not trigger the linter + public fun while_with_condition(n: u64) { + let i = 0; + while (i < n) { + i = i + 1; + } + } + + // This function uses both `while(true)` and `loop` for comparison + public fun mixed_loops() { + let vec1 = vector::empty(); + let vec2 = vector::empty(); + + // This should trigger the linter + while (true) { + if (vector::length(&vec1) == 5) break; + vector::push_back(&mut vec1, vector::length(&vec1)); + }; + + // This should not trigger the linter + loop { + if (vector::length(&vec2) == 5) break; + vector::push_back(&mut vec2, vector::length(&vec2)); + }; + } +}