From 5f4c92e745409df9de205e54a0de9c255541dcb7 Mon Sep 17 00:00:00 2001
From: spiritlhl <103393591+spiritLHLS@users.noreply.github.com>
Date: Tue, 30 Jul 2024 13:56:52 +0000
Subject: [PATCH] update

---
 disk/disktest.go | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/disk/disktest.go b/disk/disktest.go
index 987a18e..89140b9 100644
--- a/disk/disktest.go
+++ b/disk/disktest.go
@@ -110,7 +110,8 @@ func ddTest1(path, deviceName, blockFile, blockName, blockCount, bs string) stri
 			Logger.Info("Read test error: " + err.Error())
 		}
 	}
-	if err != nil || strings.Contains(tempText, "Invalid argument") || strings.Contains(tempText, "Permission denied") {
+	if strings.Contains(tempText, "Invalid argument") || strings.Contains(tempText, "Permission denied") ||
+		strings.Contains(tempText, "失败") || strings.Contains(tempText, "无效的参数") {
 		if err != nil && EnableLoger {
 			Logger.Info("Read test (first attempt) error: " + err.Error())
 		}
@@ -142,7 +143,8 @@ func ddTest2(blockFile, blockName, blockCount, bs string) string {
 			Logger.Info("execDDTest error for /root/ path: " + err.Error())
 		}
 	}
-	if err != nil || strings.Contains(tempText, "Invalid argument") || strings.Contains(tempText, "Permission denied") {
+	if strings.Contains(tempText, "Invalid argument") || strings.Contains(tempText, "Permission denied") ||
+		strings.Contains(tempText, "失败") || strings.Contains(tempText, "无效的参数") {
 		time.Sleep(1 * time.Second)
 		tempText, err = execDDTest("/dev/zero", "/tmp/"+blockFile, bs, blockCount)
 		defer os.Remove("/tmp/" + blockFile)
@@ -177,6 +179,7 @@ func ddTest2(blockFile, blockName, blockCount, bs string) string {
 			Logger.Info("execDDTest read error for /root/ path: " + err.Error())
 		}
 	}
+	// /dev/null 无法访问，需要替换
 	if strings.Contains(tempText, "Invalid argument") || strings.Contains(tempText, "Permission denied") ||
 		strings.Contains(tempText, "失败") || strings.Contains(tempText, "无效的参数") {
 		time.Sleep(1 * time.Second)
@@ -189,9 +192,6 @@ func ddTest2(blockFile, blockName, blockCount, bs string) string {
 			}
 		}
 	}
-	if EnableLoger {
-		Logger.Info("tempText: " + tempText)
-	}
 	result += parseResultDD(tempText, blockCount)
 	result += "\n"
 	return result