|
@ -2,13 +2,10 @@ package storage |
|
|
|
|
|
|
|
|
import ( |
|
|
import ( |
|
|
"encoding/hex" |
|
|
"encoding/hex" |
|
|
"errors" |
|
|
|
|
|
"fmt" |
|
|
"fmt" |
|
|
"io" |
|
|
|
|
|
"io/ioutil" |
|
|
"io/ioutil" |
|
|
"mime" |
|
|
"mime" |
|
|
"net/http" |
|
|
"net/http" |
|
|
"os" |
|
|
|
|
|
"pkg/util" |
|
|
"pkg/util" |
|
|
"strconv" |
|
|
"strconv" |
|
|
"strings" |
|
|
"strings" |
|
@ -89,45 +86,6 @@ func (n *Needle) ParsePath(fid string) { |
|
|
} |
|
|
} |
|
|
} |
|
|
} |
|
|
} |
|
|
} |
|
|
func (n *Needle) Append(w io.Writer) uint32 { |
|
|
|
|
|
header := make([]byte, 16) |
|
|
|
|
|
util.Uint32toBytes(header[0:4], n.Cookie) |
|
|
|
|
|
util.Uint64toBytes(header[4:12], n.Id) |
|
|
|
|
|
n.Size = uint32(len(n.Data)) |
|
|
|
|
|
util.Uint32toBytes(header[12:16], n.Size) |
|
|
|
|
|
w.Write(header) |
|
|
|
|
|
w.Write(n.Data) |
|
|
|
|
|
rest := 8 - ((n.Size + 16 + 4) % 8) |
|
|
|
|
|
util.Uint32toBytes(header[0:4], n.Checksum.Value()) |
|
|
|
|
|
w.Write(header[0 : rest+4]) |
|
|
|
|
|
return n.Size |
|
|
|
|
|
} |
|
|
|
|
|
func (n *Needle) Read(r io.Reader, size uint32, version Version) (int, error) { |
|
|
|
|
|
bytes := make([]byte, size+16+4) |
|
|
|
|
|
ret, e := r.Read(bytes) |
|
|
|
|
|
n.Cookie = util.BytesToUint32(bytes[0:4]) |
|
|
|
|
|
n.Id = util.BytesToUint64(bytes[4:12]) |
|
|
|
|
|
n.Size = util.BytesToUint32(bytes[12:16]) |
|
|
|
|
|
n.Data = bytes[16 : 16+size] |
|
|
|
|
|
checksum := util.BytesToUint32(bytes[16+size : 16+size+4]) |
|
|
|
|
|
if checksum != NewCRC(n.Data).Value() { |
|
|
|
|
|
return 0, errors.New("CRC error! Data On Disk Corrupted!") |
|
|
|
|
|
} |
|
|
|
|
|
return ret, e |
|
|
|
|
|
} |
|
|
|
|
|
func ReadNeedle(r *os.File) (*Needle, uint32) { |
|
|
|
|
|
n := new(Needle) |
|
|
|
|
|
bytes := make([]byte, 16) |
|
|
|
|
|
count, e := r.Read(bytes) |
|
|
|
|
|
if count <= 0 || e != nil { |
|
|
|
|
|
return nil, 0 |
|
|
|
|
|
} |
|
|
|
|
|
n.Cookie = util.BytesToUint32(bytes[0:4]) |
|
|
|
|
|
n.Id = util.BytesToUint64(bytes[4:12]) |
|
|
|
|
|
n.Size = util.BytesToUint32(bytes[12:16]) |
|
|
|
|
|
rest := 8 - ((n.Size + 16 + 4) % 8) |
|
|
|
|
|
return n, n.Size + 4 + rest |
|
|
|
|
|
} |
|
|
|
|
|
func ParseKeyHash(key_hash_string string) (uint64, uint32) { |
|
|
func ParseKeyHash(key_hash_string string) (uint64, uint32) { |
|
|
key_hash_bytes, khe := hex.DecodeString(key_hash_string) |
|
|
key_hash_bytes, khe := hex.DecodeString(key_hash_string) |
|
|
key_hash_len := len(key_hash_bytes) |
|
|
key_hash_len := len(key_hash_bytes) |
|
|