simplify the file id format
git-svn-id: https://weed-fs.googlecode.com/svn/trunk@19 282b0af5-e82d-9cf1-ede4-77906d7719d0
This commit is contained in:
@@ -5,17 +5,16 @@ import (
|
||||
"io/ioutil"
|
||||
"http"
|
||||
"log"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type Needle struct {
|
||||
Cookie uint32 "random number to mitigate brute force lookups"
|
||||
Key uint64 "file id"
|
||||
Size uint32 "Data size"
|
||||
Data []byte "The actual file data"
|
||||
Checksum int32 "CRC32 to check integrity"
|
||||
Padding []byte "Aligned to 8 bytes"
|
||||
Cookie uint32 "random number to mitigate brute force lookups"
|
||||
Key uint64 "file id"
|
||||
Size uint32 "Data size"
|
||||
Data []byte "The actual file data"
|
||||
Checksum int32 "CRC32 to check integrity"
|
||||
Padding []byte "Aligned to 8 bytes"
|
||||
}
|
||||
|
||||
func NewNeedle(r *http.Request) (n *Needle) {
|
||||
@@ -33,30 +32,31 @@ func NewNeedle(r *http.Request) (n *Needle) {
|
||||
return
|
||||
}
|
||||
func (n *Needle) ParsePath(path string) {
|
||||
a := strings.Split(path, "_")
|
||||
log.Println("cookie", a[0], "key", a[1], "altKey", a[2])
|
||||
cookie, _ := strconv.Atoui(a[0])
|
||||
n.Cookie = uint32(cookie)
|
||||
n.Key, _ = strconv.Atoui64(a[1])
|
||||
if len(path) != 16 {
|
||||
return
|
||||
}
|
||||
bytes := []byte(path)
|
||||
n.Cookie = BytesToUint32(bytes[12:16])
|
||||
n.Key = BytesToUint64(bytes[4:12])
|
||||
}
|
||||
func (n *Needle) Append(w io.Writer) {
|
||||
header := make([]byte, 16)
|
||||
uint32toBytes(header[0:4], n.Cookie)
|
||||
uint64toBytes(header[4:12], n.Key)
|
||||
Uint32toBytes(header[0:4], n.Cookie)
|
||||
Uint64toBytes(header[4:12], n.Key)
|
||||
n.Size = uint32(len(n.Data))
|
||||
uint32toBytes(header[12:16], n.Size)
|
||||
Uint32toBytes(header[12:16], n.Size)
|
||||
w.Write(header)
|
||||
w.Write(n.Data)
|
||||
rest := 8 - ((n.Size + 16 + 4) % 8)
|
||||
uint32toBytes(header[0:4], uint32(n.Checksum))
|
||||
Uint32toBytes(header[0:4], uint32(n.Checksum))
|
||||
w.Write(header[0 : rest+4])
|
||||
}
|
||||
func (n *Needle) Read(r io.Reader, size uint32) {
|
||||
bytes := make([]byte, size+16+4)
|
||||
r.Read(bytes)
|
||||
n.Cookie = bytesToUint32(bytes[0:4])
|
||||
n.Key = bytesToUint64(bytes[4:12])
|
||||
n.Size = bytesToUint32(bytes[12:16])
|
||||
n.Cookie = BytesToUint32(bytes[0:4])
|
||||
n.Key = BytesToUint64(bytes[4:12])
|
||||
n.Size = BytesToUint32(bytes[12:16])
|
||||
n.Data = bytes[16 : 16+size]
|
||||
n.Checksum = int32(bytesToUint32(bytes[16+size : 16+size+4]))
|
||||
n.Checksum = int32(BytesToUint32(bytes[16+size : 16+size+4]))
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user