MIME/Multipart 上传可能导致文件损坏

使用 HTTP Go 客户端上传图片时，MIME/Multipart 导致文件损坏。问题在于客户端的 `mimetype.DetectReader(file)` 调用读取了文件的一部分，而后续的 `io.Copy(pict, file)` 调用读取了文件的其余部分。这导致服务器端 MIME 类型检测失败，图像标题损坏。解决方案是将这两个调用之间的文件偏移量重置为 0，以读取整个文件。

我写了一个服务器，有一个上传图片的路由。这是一个接收几个参数的表单：title、description、visibility 和 picture。该页面还使用 authentication 标头。

func uploadpictureroute(prv *services.provider) http.handlerfunc {
    return func(w http.responsewriter, r *http.request) {
        user, err := auth.validaterequest(prv, w, r)
        if auth.responderror(w, err) {
            return
        }

        r.parsemultipartform(10 << 20) // 10 meg max

        title := r.formvalue("title")
        desc := r.formvalue("description")
        visib := r.formvalue("visibility")
        visibint, err := strconv.atoi(visib)
        visibility := int8(visibint) // visibility can be either 0, 1, 2

        if err != nil {
            w.writeheader(http.statusbadrequest)
        }

        file, _, err := r.formfile("picture")
        if err != nil {
            w.writeheader(http.statusbadrequest)
            return
        }
        defer file.close()

        mimetype, _, err := mimetype.detectreader(file) // package gabriel-vasile/mimetype
        if err != nil {
            w.writeheader(http.statusbadrequest)
            return
        }

        if !utils.isvalidmimetype(mimetype) { // basically just comparing to image/png, image/jpg. crashes here
            w.writeheader(http.statusbadrequest)
            return
        }

        parentfolder := prv.picturepath + "/" + strconv.formatint(*user.id, 10) + "/"

        _, err = os.stat(parentfolder)
        if os.isnotexist(err) {
            err = os.mkdirall(parentfolder, os.modeperm)
            if err != nil {
                w.writeheader(http.statusinternalservererror)
                return
            }
        }

        pict := model.picture{
            title:       title,
            description: desc,
            creator:     &user,
            visibility:  visibility,
            ext:         utils.getextformimetype(mimetype),
        }

        pict, err = dal.createpicture(prv, pict)
        if err != nil {
            w.writeheader(http.statusinternalservererror)
            return
        }

        outputfile, err := os.create(parentfolder + strconv.formatint(*pict.id, 10) + "." + pict.ext)
        if err != nil {
            w.writeheader(http.statusinternalservererror)
            dal.deletepicture(prv, pict)
        }
        defer outputfile.close()

        _, err = io.copy(outputfile, file)
        if err != nil {
            w.writeheader(http.statusinternalservererror)
            return
        }
// responding 
    }
}

使用 postman 效果很好，它可以正确上传文件，一切都按预期进行。

但是，我未能编写一个 go http 客户端来上传文件：

type uploadResponse struct {
    URLID string
}

func main() {
    filename := "filename.png"

    file, err := os.Open(filename)
    if err != nil {
        fmt.Println(err)
    }
    defer file.Close()

    // Just to be sure we're really at the start of the file
    _, err = file.Seek(0, io.SeekStart)
    if err != nil {
        fmt.Println("Can't read the file")
        return
    }

    mime, _, err := mimetype.DetectReader(file)
    if err != nil {
        fmt.Println("Can't read the file")
        return
    }

    if !api.IsMimetypeAllowed(mime) { // Just check if it's a jpg/png/gif. This works on the client
        fmt.Println("This filetype can't be uploaded to a Scinna server.")
        fmt.Println("Please convert this file to jpeg, png or gif.")

        return
    }


    body := &bytes.Buffer{}
    writer := multipart.NewWriter(body)

    title, _ := writer.CreateFormField("title")
    title.Write([]byte("My picture title"))

    desc, _ := writer.CreateFormField("description")
    desc.Write([]byte("My picture description"))

    visib, _ := writer.CreateFormField("visibility")
    visib.Write([]byte("0"))

    pict, _ := writer.CreateFormFile("picture", file.Name())
    _, err = io.Copy(pict, file)

    writer.Close()

    r, err := http.NewRequest("POST", "https://myapi.local/pictures", body)
    if err != nil {
        panic(err)
    }
    r.Header.Add("Content-Type", writer.FormDataContentType())
    r.Header.Add("Authorization", "Bearer "+config.Token)

    client := &http.Client{}
    client.Do(r)
}

此代码会导致服务器崩溃。图片的 mimetype 变成 application/octet-stream 并且图像标题被破坏（它仍然在某些编辑器中打开，但 eyesofgnome 基本上说图片不是 jpg/png 文件，因为它无法在开始）

如何修复http go客户端才能成功上传图片？

解决方案

对 mimetype.detectreader(file) 的调用读取部分文件。对 _, err = io.copy(pict, file) 的调用会读取文件的其余部分。要读取整个文件，请返回到调用 io.copy 之前的偏移量 0。

文件在偏移量 0 处打开。调用 open 后无需立即查找偏移量 0。

通过交换调用顺序来修复问题：

...

mime, _, err := mimetype.detectreader(file)
if err != nil {
    fmt.println("can't read the file")
    return
}

// rewind to the start of the file
_, err = file.seek(0, io.seekstart)
if err != nil {
    fmt.println("can't read the file")
    return
}

...

服务器也有类似的问题。检测类型后回退：

mimeType, _, err := mimetype.DetectReader(file) // Package gabriel-vasile/mimetype
if err != nil {
    w.WriteHeader(http.StatusBadRequest)
    return
}

// Rewind to the start of the file
_, err = file.Seek(0, io.SeekStart)
if err != nil {
    w.WriteHeader(http.StatusInternalServerError)
    return
}

我找到了解决方案：

UploadFile

克隆这个，如果你发现任何问题，请在这里评论。

到这里，我们也就讲完了《MIME/Multipart 上传可能导致文件损坏》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！