aboutsummaryrefslogtreecommitdiff
path: root/go
diff options
context:
space:
mode:
authorChris Lu <chris.lu@gmail.com>2013-07-15 17:26:00 -0700
committerChris Lu <chris.lu@gmail.com>2013-07-15 17:26:00 -0700
commit5e3b6e968a930a735fcb208bfeda372fe9ba7efc (patch)
tree643b3e6eb5175da86c8009d5c254cb06c1fab03b /go
parentdd66193bfbc36bb43bace01e2f88fe52f983c838 (diff)
downloadseaweedfs-5e3b6e968a930a735fcb208bfeda372fe9ba7efc.tar.xz
seaweedfs-5e3b6e968a930a735fcb208bfeda372fe9ba7efc.zip
escaping upload file name special characters
if already gzipped, avoid gzipping again
Diffstat (limited to 'go')
-rw-r--r--go/operation/upload_content.go4
-rw-r--r--go/weed/upload.go7
-rw-r--r--go/weed/volume.go14
3 files changed, 16 insertions, 9 deletions
diff --git a/go/operation/upload_content.go b/go/operation/upload_content.go
index e606ef7e3..b73ed0a66 100644
--- a/go/operation/upload_content.go
+++ b/go/operation/upload_content.go
@@ -21,11 +21,13 @@ type UploadResult struct {
Error string
}
+var fileNameEscaper = strings.NewReplacer("\\", "\\\\", "\"", "\\\"")
+
func Upload(uploadUrl string, filename string, reader io.Reader, isGzipped bool) (*UploadResult, error) {
body_buf := bytes.NewBufferString("")
body_writer := multipart.NewWriter(body_buf)
h := make(textproto.MIMEHeader)
- h.Set("Content-Disposition", fmt.Sprintf(`form-data; name="file"; filename="%s"`, filename))
+ h.Set("Content-Disposition", fmt.Sprintf(`form-data; name="file"; filename="%s"`, fileNameEscaper.Replace(filename)))
h.Set("Content-Type", mime.TypeByExtension(strings.ToLower(filepath.Ext(filename))))
if isGzipped {
h.Set("Content-Encoding", "gzip")
diff --git a/go/weed/upload.go b/go/weed/upload.go
index 8e75b387b..7256093cf 100644
--- a/go/weed/upload.go
+++ b/go/weed/upload.go
@@ -78,7 +78,12 @@ func upload(filename string, server string, fid string) (int, error) {
debug("Failed to stat file:", filename)
return 0, fiErr
}
- ret, e := operation.Upload("http://"+server+"/"+fid+"?ts="+strconv.Itoa(int(fi.ModTime().Unix())), path.Base(filename), fh, false)
+ filename = path.Base(filename)
+ isGzipped := path.Ext(filename) == ".gz"
+ if isGzipped {
+ filename = filename[0:len(filename)-3]
+ }
+ ret, e := operation.Upload("http://"+server+"/"+fid+"?ts="+strconv.Itoa(int(fi.ModTime().Unix())), filename, fh, isGzipped)
if e != nil {
return 0, e
}
diff --git a/go/weed/volume.go b/go/weed/volume.go
index dbc0ee547..a22babd45 100644
--- a/go/weed/volume.go
+++ b/go/weed/volume.go
@@ -272,13 +272,13 @@ func parseURLPath(path string) (vid, fid, filename, ext string) {
vid, fid, filename = parts[1], parts[2], parts[3]
ext = filepath.Ext(filename)
case 2:
- parts := strings.Split(path, "/")
- vid, fid = parts[1], parts[2]
- dotIndex := strings.LastIndex(fid, ".")
- if dotIndex > 0 {
- ext = fid[dotIndex:]
- fid = fid[0:dotIndex]
- }
+ parts := strings.Split(path, "/")
+ vid, fid = parts[1], parts[2]
+ dotIndex := strings.LastIndex(fid, ".")
+ if dotIndex > 0 {
+ ext = fid[dotIndex:]
+ fid = fid[0:dotIndex]
+ }
default:
sepIndex := strings.LastIndex(path, "/")
commaIndex := strings.LastIndex(path[sepIndex:], ",")