From f0de0522738a66779bd8c793d03046b024f22bd9 Mon Sep 17 00:00:00 2001
From: dchakro <35454738+dchakro@users.noreply.github.com>
Date: Thu, 7 Jul 2022 09:13:33 +0300
Subject: [PATCH] asks file size as input

---
 storage_speedtest.sh | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/storage_speedtest.sh b/storage_speedtest.sh
index 78fcde1..f8dc156 100755
--- a/storage_speedtest.sh
+++ b/storage_speedtest.sh
@@ -1,5 +1,6 @@
-#!/bin/bash
+#!/usr/bin/env bash
 # Created by Alexander aka BioDranik <me@alex.bio> in Minsk, Belarus
+# Modifications by Deepankar Chakroborty (https://github.com/dchakro)
 
 set -eu
 
@@ -10,8 +11,12 @@ fi
 
 SD_DIR=$1
 FILENAME="$SD_DIR/file_speed_test.deleteme"
-SIZE=1000 # in MB
-
+read -p "Enter Filesize (Megabytes): " SIZE
+if [[ -n ${SIZE//[0-9]/} ]]; then
+    echo "Improper input! File size set to 512MB."
+    SIZE=512 # in MB
+fi
+# 
 if [[ "$OSTYPE" == "darwin"* ]]; then
   WRITE_FILE_COMMAND="mkfile ${SIZE}m $FILENAME"
   CLEAR_CACHE_COMMAND="purge"