From 57e3add2f90ad1900be2e9d1cd8019f6aa5ad440 Mon Sep 17 00:00:00 2001 From: axtloss Date: Fri, 2 Feb 2024 15:55:53 +0100 Subject: fix database extraction --- core/storage.go | 15 +++++---------- 1 file changed, 5 insertions(+), 10 deletions(-) diff --git a/core/storage.go b/core/storage.go index ba4faf6..b3e1b53 100644 --- a/core/storage.go +++ b/core/storage.go @@ -5,6 +5,7 @@ import ( "encoding/binary" "encoding/json" "fmt" + "io" "os" "strings" @@ -114,11 +115,9 @@ func ReadDB(partition string) (string, error) { fmt.Println(err) return "", err } - fmt.Println("Header:") - fmt.Println(header.TableSize) - fmt.Println(header.TableUnit) + db := make([]byte, header.TableSize*header.TableUnit) - _, err = reader.Read(db) + _, err = io.ReadFull(reader, db) if err != nil { return "", err } @@ -128,16 +127,12 @@ func ReadDB(partition string) (string, error) { return "", err } - fmt.Println("DB Path:") - fmt.Println(temp) - fmt.Println() - err = os.WriteFile(temp+"/verify.db", db, 0777) + err = os.WriteFile(temp+"/verify.db", db, 0700) if err != nil { return "", err } - //defer os.RemoveAll(temp) - return temp + "/verify.db", err + return temp + "/verify.db", nil } func OpenDB(dbpath string) (*bolt.DB, error) { -- cgit v1.2.3