Ver código fonte

ethdb: return copied value from MemDatabase.Get (#14958)

gary rong 8 anos atrás
pai
commit
73c5aba21f
2 arquivos alterados com 16 adições e 1 exclusões
  1. 15 0
      ethdb/database_test.go
  2. 1 1
      ethdb/memory_database.go

+ 15 - 0
ethdb/database_test.go

@@ -94,6 +94,21 @@ func testPutGet(db ethdb.Database, t *testing.T) {
 		}
 	}
 
+	for _, v := range test_values {
+		orig, err := db.Get([]byte(v))
+		if err != nil {
+			t.Fatalf("get failed: %v", err)
+		}
+		orig[0] = byte(0xff)
+		data, err := db.Get([]byte(v))
+		if err != nil {
+			t.Fatalf("get failed: %v", err)
+		}
+		if !bytes.Equal(data, []byte("?")) {
+			t.Fatalf("get returned wrong result, got %q expected ?", string(data))
+		}
+	}
+
 	for _, v := range test_values {
 		err := db.Delete([]byte(v))
 		if err != nil {

+ 1 - 1
ethdb/memory_database.go

@@ -50,7 +50,7 @@ func (db *MemDatabase) Get(key []byte) ([]byte, error) {
 	defer db.lock.RUnlock()
 
 	if entry, ok := db.db[string(key)]; ok {
-		return entry, nil
+		return common.CopyBytes(entry), nil
 	}
 	return nil, errors.New("not found")
 }