/*
uploader -- simplest form file uploader
-Copyright (C) 2018-2020 Sergey Matveev <stargrave@stargrave.org>
+Copyright (C) 2018-2021 Sergey Matveev <stargrave@stargrave.org>
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
"os/exec"
"strconv"
"strings"
+ "sync"
"time"
"go.cypherpunks.ru/recfile"
- "go.cypherpunks.ru/tai64n"
+ "go.cypherpunks.ru/tai64n/v2"
"golang.org/x/crypto/blake2b"
"golang.org/x/net/netutil"
)
var (
Index = template.Must(template.New("index").Parse(`<html>
<head><title>Upload</title></head><body>
-<pre>
Example command line usage:
-
- $ curl -F file=@somedata.tar.gpg [-F comment="optional comment"] http://.../upload/
- $ b2sum -a blake2b somedata.tar.gpg # to verify checksum
+<pre>
+$ curl -F file=@somedata.tar.gpg [-F comment="optional comment"] http://.../upload/
+$ b2sum somedata.tar.gpg # to verify BLAKE2b-512 checksum
</pre>
<form enctype="multipart/form-data" action="/upload/" method="post">
<label for="file">File to upload:</label><br/>
</form></body></html>`))
NotifyFromAddr *string
NotifyToAddr *string
+ Jobs sync.WaitGroup
)
-func notify(tai, filename string, size int64, comment string) {
- if *NotifyToAddr == "" {
- return
- }
- var rec bytes.Buffer
- w := recfile.NewWriter(&rec)
- w.WriteFields(
- recfile.Field{Name: "TAI64N", Value: tai},
- recfile.Field{Name: "Size", Value: strconv.FormatInt(size, 10)},
- recfile.Field{Name: "Filename", Value: filename},
- )
- w.WriteFieldMultiline("Comment", strings.Split(comment, "\n"))
- cmd := exec.Command(SendmailCmd, *NotifyToAddr)
- cmd.Stdin = io.MultiReader(
- strings.NewReader(fmt.Sprintf(
- `From: %s
-To: %s
-Subject: %s
-MIME-Version: 1.0
-Content-Type: text/plain; charset=utf-8
-Content-Transfer-Encoding: base64
-
-`,
- *NotifyFromAddr,
- *NotifyToAddr,
- mime.BEncoding.Encode("UTF-8", fmt.Sprintf("%s (%d KiB)", filename, size/1024)),
- )),
- strings.NewReader(base64.StdEncoding.EncodeToString(rec.Bytes())),
- )
- cmd.Run()
-}
-
func upload(w http.ResponseWriter, r *http.Request) {
log.Println(r.RemoteAddr, "connected")
if r.Method == http.MethodGet {
}
t := time.Now()
ts := new(tai64n.TAI64N)
- tai64n.FromTime(t, ts)
- tai := ts.Encode()[1:]
+ ts.FromTime(t)
+ tai := tai64n.Encode(ts[:])[1:]
fnOrig := p.FileName()
fd, err := os.OpenFile(tai+".part", os.O_WRONLY|os.O_CREATE|os.O_EXCL, 0666)
if err != nil {
log.Println(r.RemoteAddr, tai, fnOrig, n, sum, err)
return
}
+
var rec bytes.Buffer
wr := recfile.NewWriter(&rec)
if _, err = wr.WriteFields(
log.Println(r.RemoteAddr, tai, fnOrig, n, sum, err)
return
}
- io.Copy(w, &rec)
- log.Println(r.RemoteAddr, tai, fnOrig, n, sum)
- p, err = mr.NextPart()
- if err != nil || p.FormName() != CommentFieldName {
- go notify(fnOrig, tai, n, "")
+ if _, err = w.Write(rec.Bytes()); err == nil {
+ log.Println(r.RemoteAddr, tai, fnOrig, n, sum)
+ } else {
+ log.Println(r.RemoteAddr, tai, fnOrig, n, sum, err)
return
}
- comment, err := ioutil.ReadAll(p)
- if err != nil || len(comment) == 0 {
- go notify(tai, fnOrig, n, "")
+
+ if _, err = wr.WriteFields(
+ recfile.Field{Name: "Filename", Value: fnOrig},
+ ); err != nil {
+ log.Println(r.RemoteAddr, tai, fnOrig, n, sum, err)
+ return
+ }
+
+ var commentLines []string
+ p, err = mr.NextPart()
+ if err == nil && p.FormName() == CommentFieldName {
+ comment, err := ioutil.ReadAll(p)
+ if err == nil && len(comment) > 0 {
+ commentLines = strings.Split(string(comment), "\n")
+ wr.WriteFieldMultiline("Comment", commentLines)
+ }
+ }
+
+ ioutil.WriteFile(tai+".rec", rec.Bytes(), os.FileMode(0666))
+ if *NotifyToAddr == "" {
return
}
- ioutil.WriteFile(tai+".txt", comment, os.FileMode(0666))
- go notify(tai, fnOrig, n, string(comment))
+ cmd := exec.Command(SendmailCmd, *NotifyToAddr)
+ cmd.Stdin = io.MultiReader(
+ strings.NewReader(fmt.Sprintf(
+ `From: %s
+To: %s
+Subject: %s
+MIME-Version: 1.0
+Content-Type: text/plain; charset=utf-8
+Content-Transfer-Encoding: base64
+
+`,
+ *NotifyFromAddr,
+ *NotifyToAddr,
+ mime.BEncoding.Encode("UTF-8", fmt.Sprintf("%s (%d KiB)", fnOrig, n/1024)),
+ )),
+ strings.NewReader(base64.StdEncoding.EncodeToString(rec.Bytes())),
+ )
+ Jobs.Add(1)
+ go func() {
+ cmd.Run()
+ Jobs.Done()
+ }()
}
func main() {
+ doUCSPI := flag.Bool("ucspi", false, "Work as UCSPI-TCP service")
bind := flag.String("bind", "[::]:8086", "Address to bind to")
conns := flag.Int("conns", 2, "Maximal number of connections")
- NotifyFromAddr = flag.String("notify-from", "uploader@example.com", "Address notifications are send to")
- NotifyToAddr = flag.String("notify-to", "", "Address notifications are send from")
+ NotifyFromAddr = flag.String("notify-from", "uploader@example.com", "Address notifications are sent to")
+ NotifyToAddr = flag.String("notify-to", "", "Address notifications are sent from")
flag.Parse()
+ log.SetFlags(log.Lshortfile)
+ if !*doUCSPI {
+ log.SetOutput(os.Stdout)
+ }
if len(*NotifyFromAddr) == 0 && len(*NotifyToAddr) > 0 {
log.Fatalln("notify-from address can not be empty, if notify-to is set")
}
- ln, err := net.Listen("tcp", *bind)
- if err != nil {
- log.Fatalln(err)
- }
- log.Println("listening", *bind)
- ln = netutil.LimitListener(ln, *conns)
s := &http.Server{
ReadHeaderTimeout: 10 * time.Second,
MaxHeaderBytes: 10 * (1 << 10),
}
http.HandleFunc("/upload/", upload)
- s.Serve(ln)
+ if *doUCSPI {
+ s.SetKeepAlivesEnabled(false)
+ ln := &UCSPI{}
+ s.ConnState = connStater
+ err := s.Serve(ln)
+ if _, ok := err.(UCSPIAlreadyAccepted); !ok {
+ log.Fatalln(err)
+ }
+ Jobs.Wait()
+ } else {
+ ln, err := net.Listen("tcp", *bind)
+ if err != nil {
+ log.Fatalln(err)
+ }
+ log.Println("listening", *bind)
+ ln = netutil.LimitListener(ln, *conns)
+ if err = s.Serve(ln); err != nil {
+ log.Fatalln(err)
+ }
+ }
}