From 657cca128e671e0042da26ec94b4063debe0f187 Mon Sep 17 00:00:00 2001
From: linxin <linxin@geedgenetworks.com>
Date: Wed, 19 Jul 2023 17:29:10 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dcoredump-tool=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E8=AF=BB=E5=8F=96=E4=B8=8D=E5=85=A8=E7=9A=84=E9=97=AE?=
 =?UTF-8?q?=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 coredump-handler/coredump-handler.go | 15 +++++++++++++++
 coredump-tool/coredump-tool.go       |  2 +-
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/coredump-handler/coredump-handler.go b/coredump-handler/coredump-handler.go
index 554980d..c4c57a9 100644
--- a/coredump-handler/coredump-handler.go
+++ b/coredump-handler/coredump-handler.go
@@ -359,6 +359,21 @@ func main() {
 	//write coredump info
 	err = writeCoreConfig(coredump_config)
 	if err != nil {
+		journal.Print(journal.PriErr, err.Error())
+	}
+	flag, err = isDiskSufficient(pipe_config)
+	if err != nil {
+		journal.Print(journal.PriErr, "Can't judge disk's space is sufficient or not. "+err.Error())
+		return
+	}
+	if !flag {
+		journal.Print(journal.PriErr, "Disk space exceeds limit after write coredump!")
+		err := os.RemoveAll(pipe_config.Storage)
+		if err != nil {
+			journal.Print(journal.PriErr, err.Error())
+			return
+		}
 		journal.Print(journal.PriInfo, err.Error())
+		return
 	}
 }
diff --git a/coredump-tool/coredump-tool.go b/coredump-tool/coredump-tool.go
index e968f77..06b4cd5 100644
--- a/coredump-tool/coredump-tool.go
+++ b/coredump-tool/coredump-tool.go
@@ -70,7 +70,7 @@ func WalkDirectory(dir string) {
 			})
 
 			if err != nil {
-				return err
+				fmt.Printf("Error walking directory %s: %v\n", path, err)
 			}
 		}