From 19fcec84d8d7d21e796c7624e521b60d28ee21ed Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Sun, 7 Apr 2024 20:45:59 +0200
Subject: Adding upstream version 16.2.11+ds.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 src/script/smr_benchmark/linearCopy.sh    | 91 +++++++++++++++++++++++++++++++
 src/script/smr_benchmark/linearSMRCopy.sh | 69 +++++++++++++++++++++++
 2 files changed, 160 insertions(+)
 create mode 100755 src/script/smr_benchmark/linearCopy.sh
 create mode 100755 src/script/smr_benchmark/linearSMRCopy.sh

(limited to 'src/script/smr_benchmark')

diff --git a/src/script/smr_benchmark/linearCopy.sh b/src/script/smr_benchmark/linearCopy.sh
new file mode 100755
index 000000000..416a7e742
--- /dev/null
+++ b/src/script/smr_benchmark/linearCopy.sh
@@ -0,0 +1,91 @@
+#!/usr/bin/env bash
+
+# copy a linear file from srcFile to destination disk in a loop until writeSize MBs is written
+# destinationDisk is a SMR Host Aware Disk eg. /dev/sdb
+
+if [ "$#" -lt 3 ]; then
+	echo "Usage ./linearCopy.sh srcFile destinationDisk writeSize(MB)"
+	exit
+fi
+
+if [ "$(id -u)" != "0" ]; then
+	echo "Please run as sudo user"
+	exit
+fi
+
+srcFile=$1
+destDisk=$2
+writeSize=$3
+verbose=true
+
+if [ -f time ]; then
+	rm -rf time
+fi
+
+#chunkSize=4096 # in bytes
+chunkSize=1048576 # in bytes
+fileSize=`stat --printf="%s" $srcFile`
+
+numChunksInFile=`echo "$fileSize * (1048576 / $chunkSize)" | bc`
+chunksLeft=$(( $(($writeSize * 1048576)) / $chunkSize))
+
+
+echo "fileSize = $fileSize"
+
+if [ "$(($fileSize % 512))" -ne 0 ]; then
+	echo "$srcFile not 512 byte aligned"
+	exit
+fi
+
+if [ "$(($chunkSize % 512))" -ne 0 ]; then
+	echo "$chunkSize not 512 byte aligned"
+	exit
+fi
+
+if [ "$fileSize" -lt "$chunkSize" ]; then
+	echo "filesize $fileSize should be greater than chunkSize $chunkSize"
+	exit
+fi
+
+
+numFileChunks=$(($fileSize / $chunkSize))
+if [ $verbose == true ]; then
+	echo "numFileChunks = $numFileChunks"
+fi
+
+smrLBAStart=33554432 # TODO query from SMR Drive
+#smrLBAStart=37224448
+
+offset=$(( $smrLBAStart / $(( $chunkSize / 512)) ))
+
+if [ $verbose == true ]; then
+	echo "chunksLeft = $chunksLeft, offset = $offset"
+fi
+
+chunkNum=0
+
+while [ "$chunksLeft" -gt 0 ]; do
+	chunkNum=$(($chunkNum + 1))
+	if [ $verbose == true ]; then
+		echo "CHUNK $chunkNum `date +%H:%M:%S`" >> time
+	fi
+	dd if=$srcFile of=$destDisk seek=$offset bs=$chunkSize 2> tmp 
+	cat tmp | grep MB >> time # > /dev/null 2>&1
+	if [ $verbose == true ]; then
+		echo "chunksLeft = $chunksLeft, offset = $offset"
+	fi
+	chunksLeft=$(($chunksLeft - $numFileChunks))
+	offset=$(($offset + $numFileChunks))
+done
+
+if [ -f tmp ]; then
+	rm tmp
+fi
+
+if [ $verbose == false ]; then
+	rm time
+else
+	echo "Time Stamp for Chunk Writes"
+	cat time
+	rm time
+fi
diff --git a/src/script/smr_benchmark/linearSMRCopy.sh b/src/script/smr_benchmark/linearSMRCopy.sh
new file mode 100755
index 000000000..1ff2695c6
--- /dev/null
+++ b/src/script/smr_benchmark/linearSMRCopy.sh
@@ -0,0 +1,69 @@
+#! /usr/bin/env bash
+
+# copy a linear file from srcFile to destination SMRDisk in a loop until writeSize MBs is written
+# SMRDisk is the SMR Host Aware / Host Managed Disk eg. /dev/sdb
+
+usage(){
+	echo "linearSMRCopy.sh <srcFile> <SMRDisk> <writeSize (MB)>"
+}
+
+if [ "$#" -lt 3 ]; then
+	usage
+	exit
+fi
+
+if [ "$(id -u)" != "0" ]; then
+	echo "Please run as sudo user"
+	exit
+fi
+
+if which zbc_open_zone > /dev/null 2>&1 && which zbc_read_zone > /dev/null 2>&1 && which zbc_write_zone > /dev/null 2>&1 ; then
+	echo "libzbc commands present... refreshing zones"
+	# reset all write pointers before starting to write
+	sudo zbc_reset_write_ptr /dev/sdb -1
+else
+	echo "libzbc commands not detected. Please install libzbc"
+	exit
+fi
+
+srcFile=$1
+SMRDisk=$2
+writeSize=$3
+iosize=10240
+
+numberOfSectors=$(($writeSize * 2048))
+
+smrZoneStart=33554432 # TODO query this from SMR drive
+
+#dd if=$srcFile of=$destDisk seek=$smrZoneStart bs=512
+
+fileSize=`stat --printf="%s" $srcFile`
+
+if [ "$(($fileSize % 512))" -ne 0 ]; then
+	echo "$srcFile not 512 byte aligned"
+	exit
+fi
+
+sectorsLeftToWrite=$(($fileSize / 512))
+
+znum=64 # TODO query this from SMR Drive
+
+zoneLength=524288 # number of sectors in each zone TODO query from SMR drive
+
+writeOffset=$smrZoneStart
+
+sectorsLeftToWrite=$numberOfSectors
+
+echo "write begin sectors Left = $sectorsLeftToWrite, writeOffset = $writeOffset zone Num = $znum"
+
+while [ "$sectorsLeftToWrite" -gt 0 ]; 
+do
+	sudo zbc_open_zone $SMRDisk $znum 
+        sudo time zbc_write_zone -f $srcFile -loop $SMRDisk $znum $iosize
+	sudo zbc_close_zone /dev/sdb $znum
+	writeOffset=$(($writeOffset+$zoneLength))
+	znum=$(($znum+1))
+	sectorsLeftToWrite=$(($sectorsLeftToWrite - $zoneLength))
+done
+
+echo "write end sectors Left = $sectorsLeftToWrite, writeOffset = $writeOffset zone Num = $znum"
-- 
cgit v1.2.3