package storage import ( "fmt" "io/ioutil" "mime" "net/http" "path" "strconv" "strings" "time" "github.com/chrislusf/seaweedfs/weed/glog" "github.com/chrislusf/seaweedfs/weed/images" "github.com/chrislusf/seaweedfs/weed/operation" ) const ( NeedleHeaderSize = 16 //should never change this NeedlePaddingSize = 8 NeedleChecksumSize = 4 MaxPossibleVolumeSize = 4 * 1024 * 1024 * 1024 * 8 ) /* * A Needle means a uploaded and stored file. * Needle file size is limited to 4GB for now. */ type Needle struct { Cookie uint32 `comment:"random number to mitigate brute force lookups"` Id uint64 `comment:"needle id"` Size uint32 `comment:"sum of DataSize,Data,NameSize,Name,MimeSize,Mime"` DataSize uint32 `comment:"Data size"` //version2 Data []byte `comment:"The actual file data"` Flags byte `comment:"boolean flags"` //version2 NameSize uint8 //version2 Name []byte `comment:"maximum 256 characters"` //version2 MimeSize uint8 //version2 Mime []byte `comment:"maximum 256 characters"` //version2 LastModified uint64 //only store LastModifiedBytesLength bytes, which is 5 bytes to disk Ttl *TTL Checksum CRC `comment:"CRC32 to check integrity"` Padding []byte `comment:"Aligned to 8 bytes"` rawBlock *Block // underlying supporing []byte, fetched and released into a pool } func (n *Needle) String() (str string) { str = fmt.Sprintf("Cookie:%d, Id:%d, Size:%d, DataSize:%d, Name: %s, Mime: %s", n.Cookie, n.Id, n.Size, n.DataSize, n.Name, n.Mime) return } func ParseUpload(r *http.Request) ( fileName string, data []byte, mimeType string, isGzipped bool, modifiedTime uint64, ttl *TTL, isChunkedFile bool, e error) { form, fe := r.MultipartReader() if fe != nil { glog.V(0).Infoln("MultipartReader [ERROR]", fe) e = fe return } //first multi-part item part, fe := form.NextPart() if fe != nil { glog.V(0).Infoln("Reading Multi part [ERROR]", fe) e = fe return } fileName = part.FileName() if fileName != "" { fileName = path.Base(fileName) } data, e = ioutil.ReadAll(part) if e != nil { glog.V(0).Infoln("Reading Content [ERROR]", e) return } //if the filename is empty string, do a search on the other multi-part items for fileName == "" { part2, fe := form.NextPart() if fe != nil { break // no more or on error, just safely break } fName := part2.FileName() //found the first multi-part has filename if fName != "" { data2, fe2 := ioutil.ReadAll(part2) if fe2 != nil { glog.V(0).Infoln("Reading Content [ERROR]", fe2) e = fe2 return } //update data = data2 fileName = path.Base(fName) break } } dotIndex := strings.LastIndex(fileName, ".") ext, mtype := "", "" if dotIndex > 0 { ext = strings.ToLower(fileName[dotIndex:]) mtype = mime.TypeByExtension(ext) } contentType := part.Header.Get("Content-Type") if contentType != "" && mtype != contentType { mimeType = contentType //only return mime type if not deductable mtype = contentType } if part.Header.Get("Content-Encoding") == "gzip" { isGzipped = true } else if operation.IsGzippable(ext, mtype) { if data, e = operation.GzipData(data); e != nil { return } isGzipped = true } if ext == ".gz" { isGzipped = true } if strings.HasSuffix(fileName, ".gz") && !strings.HasSuffix(fileName, ".tar.gz") { fileName = fileName[:len(fileName)-3] } modifiedTime, _ = strconv.ParseUint(r.FormValue("ts"), 10, 64) ttl, _ = ReadTTL(r.FormValue("ttl")) isChunkedFile, _ = strconv.ParseBool(r.FormValue("cm")) return } func NewNeedle(r *http.Request, fixJpgOrientation bool) (n *Needle, e error) { fname, mimeType, isGzipped, isChunkedFile := "", "", false, false n = new(Needle) fname, n.Data, mimeType, isGzipped, n.LastModified, n.Ttl, isChunkedFile, e = ParseUpload(r) if e != nil { return } if len(fname) < 256 { n.Name = []byte(fname) n.SetHasName() } if len(mimeType) < 256 { n.Mime = []byte(mimeType) n.SetHasMime() } if isGzipped { n.SetGzipped() } if n.LastModified == 0 { n.LastModified = uint64(time.Now().Unix()) } n.SetHasLastModifiedDate() if n.Ttl != EMPTY_TTL { n.SetHasTtl() } if isChunkedFile { n.SetIsChunkManifest() } if fixJpgOrientation { loweredName := strings.ToLower(fname) if mimeType == "image/jpeg" || strings.HasSuffix(loweredName, ".jpg") || strings.HasSuffix(loweredName, ".jpeg") { n.Data = images.FixJpgOrientation(n.Data) } } n.Checksum = NewCRC(n.Data) commaSep := strings.LastIndex(r.URL.Path, ",") dotSep := strings.LastIndex(r.URL.Path, ".") fid := r.URL.Path[commaSep+1:] if dotSep > 0 { fid = r.URL.Path[commaSep+1 : dotSep] } e = n.ParsePath(fid) return } func (n *Needle) ParsePath(fid string) (err error) { length := len(fid) if length <= 8 { return fmt.Errorf("Invalid fid: %s", fid) } delta := "" deltaIndex := strings.LastIndex(fid, "_") if deltaIndex > 0 { fid, delta = fid[0:deltaIndex], fid[deltaIndex+1:] } n.Id, n.Cookie, err = ParseKeyHash(fid) if err != nil { return err } if delta != "" { if d, e := strconv.ParseUint(delta, 10, 64); e == nil { n.Id += d } else { return e } } return err } func ParseKeyHash(key_hash_string string) (uint64, uint32, error) { if len(key_hash_string) <= 8 { return 0, 0, fmt.Errorf("KeyHash is too short.") } if len(key_hash_string) > 24 { return 0, 0, fmt.Errorf("KeyHash is too long.") } split := len(key_hash_string) - 8 key, err := strconv.ParseUint(key_hash_string[:split], 16, 64) if err != nil { return 0, 0, fmt.Errorf("Parse key error: %v", err) } hash, err := strconv.ParseUint(key_hash_string[split:], 16, 32) if err != nil { return 0, 0, fmt.Errorf("Parse hash error: %v", err) } return key, uint32(hash), nil }