2 uploader -- simplest form file uploader
3 Copyright (C) 2018 Sergey Matveev <stargrave@stargrave.org>
20 "golang.org/x/crypto/blake2s"
21 "golang.org/x/net/netutil"
25 WriteBufSize = 1 << 20
28 func upload(w http.ResponseWriter, r *http.Request) {
29 log.Println(r.RemoteAddr, "connected")
30 if r.Method == http.MethodGet {
31 w.Write([]byte(`<html>
32 <head><title>Upload</title></head><body>
33 <form enctype="multipart/form-data" action="/upload/" method="post">
34 <input type="file" name="fileupload" /><input type="submit" />
35 </form></body></html>`))
38 mr, err := r.MultipartReader()
40 log.Println(r.RemoteAddr, err)
43 p, err := mr.NextPart()
45 log.Println(r.RemoteAddr, err)
48 if p.FormName() != "fileupload" {
49 log.Println(r.RemoteAddr, "non file form field")
52 fn := time.Now().Format(time.RFC3339Nano)
53 fd, err := os.OpenFile(fn, os.O_WRONLY|os.O_CREATE, 0600)
55 log.Println(r.RemoteAddr, fn, p.FileName(), err)
59 h, err := blake2s.New256(nil)
63 fdBuf := bufio.NewWriterSize(fd, WriteBufSize)
64 mw := io.MultiWriter(fdBuf, h)
65 n, err := io.Copy(mw, p)
67 log.Println(r.RemoteAddr, fn, p.FileName(), err)
70 if err = fdBuf.Flush(); err != nil {
71 log.Println(r.RemoteAddr, fn, p.FileName(), err)
74 sum := hex.EncodeToString(h.Sum(nil))
75 fmt.Fprintf(w, "bytes: %d\nBLAKE2s: %s\n", n, sum)
76 log.Println(r.RemoteAddr, fn, p.FileName(), n, sum)
80 bind := flag.String("bind", "[::]:8086", "Address to bind to")
81 conns := flag.Int("conns", 2, "Maximal number of connections")
83 ln, err := net.Listen("tcp", *bind)
87 log.Println("listening", *bind)
88 ln = netutil.LimitListener(ln, *conns)
90 ReadHeaderTimeout: 10 * time.Second,
91 MaxHeaderBytes: 10 * (1 << 10),
93 http.HandleFunc("/upload/", upload)