From bf7bd86934ae61a75f9c0872570adb1788d1924d Mon Sep 17 00:00:00 2001 From: mxve <68632137+mxve@users.noreply.github.com> Date: Sat, 30 Mar 2024 18:48:11 +0100 Subject: [PATCH] add length check to arg_value --- src/main.rs | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/main.rs b/src/main.rs index 2c1d910..499ae93 100644 --- a/src/main.rs +++ b/src/main.rs @@ -487,9 +487,12 @@ fn setup_env() { } fn arg_value(args: &[String], arg: &str) -> Option { - args.iter() - .position(|r| r == arg) - .map(|e| args[e + 1].clone()) + if let Some(e) = args.iter().position(|r| r == arg) { + if e + 1 < args.len() { + return Some(args[e + 1].clone()); + } + } + None } fn arg_bool(args: &[String], arg: &str) -> bool {