mirrors/uquic
1
0
Fork 0
mirror of https://github.com/refraction-networking/uquic.git synced 2025-04-04 12:47:36 +03:00
Code Issues Projects Releases Packages Wiki Activity

add an upload handler to the demo server

Browse source
This commit is contained in:
Marten Seemann 2016-08-20 22:20:43 +07:00
parent 9de08f8913
commit c2af049b8a
No known key found for this signature in database
GPG key ID: 3603F40B121FCDEA
1 changed files with 40 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

40
example/main.go
View file

@ -1,12 +1,15 @@
package main
import (
"crypto/md5"
"crypto/tls"
"errors"
"flag"
"fmt"
"io"
"io/ioutil"
"log"
"mime/multipart"
"net/http"
"strings"
"sync"
@ -29,6 +32,11 @@ func (b *binds) Set(v string) error {
return nil
}
// Size is needed by the /demo/upload handler to determine the size of the uploaded file
type Size interface {
Size() int64
}
func init() {
http.HandleFunc("/demo/tile", func(w http.ResponseWriter, r *http.Request) {
// Small 40x40 png
@ -58,6 +66,38 @@ func init() {
}
w.Write(body)
})
// accept file uploads and return the MD5 of the uploaded file
// maximum accepted file size is 1 GB
http.HandleFunc("/demo/upload", func(w http.ResponseWriter, r *http.Request) {
if r.Method == http.MethodPost {
var err error
err = r.ParseMultipartForm(1 << 30) // 1 GB
if err == nil {
var file multipart.File
file, _, err = r.FormFile("uploadfile")
if err == nil {
var size int64
if sizeInterface, ok := file.(Size); ok {
size = sizeInterface.Size()
b := make([]byte, size)
file.Read(b)
md5 := md5.Sum(b)
fmt.Fprintf(w, "%x", md5)
return
}
err = errors.New("Couldn't get uploaded file size.")
}
}
if err != nil {
utils.Infof("Error receiving upload: %#v", err)
}
}
io.WriteString(w, `<html><body><form action="/demo/upload" method="post" enctype="multipart/form-data">
<input type="file" name="uploadfile"><br>
<input type="submit">
</form></body></html>`)
})
}
func main() {