/*
uploader -- simplest form file uploader
-Copyright (C) 2018 Sergey Matveev <stargrave@stargrave.org>
+Copyright (C) 2018-2019 Sergey Matveev <stargrave@stargrave.org>
*/
package main
"encoding/hex"
"flag"
"fmt"
+ "html/template"
"io"
"log"
"net"
)
const (
- WriteBufSize = 1 << 20
+ WriteBufSize = 1 << 20
+ FileFieldName = "fileupload"
)
-func upload(w http.ResponseWriter, r *http.Request) {
- log.Println(r.RemoteAddr, "connected")
- if r.Method == http.MethodGet {
- w.Write([]byte(`<html>
+var (
+ Index = template.Must(template.New("index").Parse(`<html>
<head><title>Upload</title></head><body>
<form enctype="multipart/form-data" action="/upload/" method="post">
-<input type="file" name="fileupload" /><input type="submit" />
+<input type="file" name="{{.}}" /><input type="submit" />
</form></body></html>`))
+)
+
+func upload(w http.ResponseWriter, r *http.Request) {
+ log.Println(r.RemoteAddr, "connected")
+ if r.Method == http.MethodGet {
+ if err := Index.Execute(w, FileFieldName); err != nil {
+ log.Println(r.RemoteAddr, err)
+ }
return
}
mr, err := r.MultipartReader()
log.Println(r.RemoteAddr, err)
return
}
- if p.FormName() != "fileupload" {
+ if p.FormName() != FileFieldName {
log.Println(r.RemoteAddr, "non file form field")
return
}
+ h, err := blake2s.New256(nil)
+ if err != nil {
+ panic(err)
+ }
fn := time.Now().Format(time.RFC3339Nano)
- fd, err := os.OpenFile(fn, os.O_WRONLY|os.O_CREATE, 0600)
+ fd, err := os.OpenFile(fn+".part", os.O_WRONLY|os.O_CREATE, 0600)
if err != nil {
log.Println(r.RemoteAddr, fn, p.FileName(), err)
return
}
- defer fd.Close()
- h, err := blake2s.New256(nil)
- if err != nil {
- panic(err)
- }
fdBuf := bufio.NewWriterSize(fd, WriteBufSize)
mw := io.MultiWriter(fdBuf, h)
n, err := io.Copy(mw, p)
if err != nil {
log.Println(r.RemoteAddr, fn, p.FileName(), err)
+ fd.Close()
return
}
if err = fdBuf.Flush(); err != nil {
log.Println(r.RemoteAddr, fn, p.FileName(), err)
+ fd.Close()
return
}
+ fd.Close()
sum := hex.EncodeToString(h.Sum(nil))
+ if err = os.Rename(fn+".part", fn); err != nil {
+ log.Println(r.RemoteAddr, fn, p.FileName(), n, sum, err)
+ return
+ }
fmt.Fprintf(w, "bytes: %d\nBLAKE2s: %s\n", n, sum)
log.Println(r.RemoteAddr, fn, p.FileName(), n, sum)
}