read database from partition

1 files changed, 78 insertions, 7 deletions

diff --git a/core/storage.go b/core/storage.go
index 8c76c52..ba4faf6 100644
--- a/core/storage.go
+++ b/core/storage.go
@@ -16,6 +16,7 @@ type Header struct {
 	Signature      string
 	FilesystemSize int
 	TableSize      int
+	TableUnit      int
 }
 
 type Node struct {
@@ -42,6 +43,7 @@ func ReadHeader(partition string) (Header, error) {
 	Signature := make([]byte, 302)
 	FileSystemSize := make([]byte, 4)
 	TableSize := make([]byte, 4)
+	TableUnit := make([]byte, 1)
 
 	_, err = reader.Read(MagicNumber)
 	MagicNum := binary.BigEndian.Uint16(MagicNumber)
@@ -62,19 +64,88 @@ func ReadHeader(partition string) (Header, error) {
 	if err != nil {
 		return Header{}, err
 	}
+	_, err = reader.Read(TableUnit)
+	if err != nil {
+		return Header{}, err
+	}
 
 	header.Signature = string(Signature)
 	header.FilesystemSize = int(binary.BigEndian.Uint16(FileSystemSize))
-	header.TableSize = int(binary.BigEndian.Uint16(TableSize))
+	header.TableSize = int(binary.BigEndian.Uint32(TableSize))
+	switch TableUnit[0] {
+	case 0:
+		header.TableUnit = 1
+	case 1:
+		header.TableUnit = 1000
+	case 2:
+		header.TableUnit = 1000000
+	case 3:
+		header.TableUnit = 1000000000
+	case 4:
+		header.TableUnit = 1000000000000
+	case 5:
+		header.TableUnit = 1000000000000000
+	default:
+		return Header{}, fmt.Errorf("Unknown TableUnit %d", TableUnit)
+	}
 	return header, nil
 }
 
-func OpenDB() (*bolt.DB, error) {
-	_, exist := os.Stat("my.db") // TODO: use configuration file for db path
+func ReadDB(partition string) (string, error) {
+	_, exist := os.Stat(partition)
+	if os.IsNotExist(exist) {
+		return "", fmt.Errorf("Cannot find partition %s", partition)
+	}
+	part, err := os.Open(partition)
+	if err != nil {
+		return "", err
+	}
+	defer part.Close()
+	reader := bufio.NewReader(part)
+
+	_, err = reader.Read(make([]byte, 313))
+	if err != nil {
+		fmt.Println(err)
+		return "", err
+	}
+
+	header, err := ReadHeader(partition)
+	if err != nil {
+		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)
+	if err != nil {
+		return "", err
+	}
+
+	temp, err := os.MkdirTemp("", "*-fsverify")
+	if err != nil {
+		return "", err
+	}
+
+	fmt.Println("DB Path:")
+	fmt.Println(temp)
+	fmt.Println()
+	err = os.WriteFile(temp+"/verify.db", db, 0777)
+	if err != nil {
+		return "", err
+	}
+
+	//defer os.RemoveAll(temp)
+	return temp + "/verify.db", err
+}
+
+func OpenDB(dbpath string) (*bolt.DB, error) {
+	_, exist := os.Stat(dbpath)
 	if os.IsNotExist(exist) {
-		os.Create("my.db")
+		os.Create(dbpath)
 	}
-	db, err := bolt.Open("my.db", 0777, nil)
+	db, err := bolt.Open(dbpath, 0777, nil)
 	if err != nil {
 		return nil, err
 	}
@@ -85,7 +156,7 @@ func AddNode(node Node, db *bolt.DB) error {
 	var err error
 	var deferDB bool
 	if db == nil {
-		db, err = OpenDB()
+		db, err = OpenDB("my.db")
 		if err != nil {
 			return err
 		}
@@ -114,7 +185,7 @@ func GetNode(checksum string, db *bolt.DB) (Node, error) {
 	var err error
 	var deferDB bool
 	if db == nil {
-		db, err = OpenDB()
+		db, err = OpenDB("my.db")
 		if err != nil {
 			return Node{}, err
 		}