-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.go
117 lines (95 loc) · 2.43 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
package main
import (
"crypto/tls"
"flag"
"fmt"
"io"
"log"
"net/http"
"os"
"path"
"strings"
)
var(
serverDirectory *string
uploadDirectory *string
)
func main(){
dir,err := os.Getwd()
if err != nil{
log.Fatal(err)
}
port := flag.String("p", "8080", "Listening port.")
serverDirectory = flag.String("d", dir, "Serve directory.")
uploadDirectory = flag.String("u", dir, "Custom uploads directory ( Default is CWD )")
isTLS := flag.Bool("tls", false, "Enable HTTPS.")
flag.Parse()
fmt.Println("[+] Listening on port", *port, "...")
fmt.Println("[+] Serving directory:", *serverDirectory)
fmt.Println("[+] Uploads directory:", *uploadDirectory)
http.HandleFunc("/upload", uploadHandler)
http.Handle("/", http.FileServer(http.Dir(*serverDirectory)))
host := fmt.Sprintf(":%s", *port)
// TODO clean this bit up
if *isTLS {
cert, key, err := GenerateCert()
if err != nil{
log.Fatal(err)
}
fcert, err := tls.X509KeyPair(cert, key)
tlsConfig := &tls.Config{
Certificates: []tls.Certificate{fcert},
}
server := http.Server{
Addr: host,
Handler: logRequest(http.DefaultServeMux),
TLSConfig: tlsConfig,
}
log.Fatal(server.ListenAndServeTLS("",""))
}else{
log.Fatal(http.ListenAndServe(host, logRequest(http.DefaultServeMux)))
}
}
//curl -F [email protected] http://localhost:8080/upload
func uploadHandler(w http.ResponseWriter, r *http.Request){
if r.Method != "POST"{
w.Write([]byte(`
<html>
<head>
<title>Upload</title>
</head>
<body>
<form enctype="multipart/form-data" action="/upload" method="POST">
<input type="file" name="file" />
<input type="submit" value="upload" />
</form>
</body>
</html>
`))
return
}
r.ParseMultipartForm(32 << 20)
file, handler, err := r.FormFile("file")
if err != nil {
log.Println(err)
return
}
fileName := path.Join(*uploadDirectory, path.Base(handler.Filename))
fh, err := os.Create(fileName)
defer fh.Close()
if err != nil{
log.Println(err)//?fatal???
return
}
io.Copy(fh, file)
ipAddr := strings.Split(r.RemoteAddr, ":")[0]
fmt.Printf("%s %s %s %s\n", ipAddr, "UPLOAD", handler.Filename, fileName)
w.Write([]byte("File uploaded."))
}
func logRequest(handler http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
ipAddr := strings.Split(r.RemoteAddr, ":")[0]//remove remote port.
fmt.Printf("%s %s %s\n", ipAddr, r.Method, r.URL)
handler.ServeHTTP(w, r)
})
}