-
Notifications
You must be signed in to change notification settings - Fork 4
/
main.go
108 lines (88 loc) · 2.47 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
package main
import (
"flag"
"fmt"
rawLog "log"
"os"
"github.com/kardianos/mitmproxy/addon"
"github.com/kardianos/mitmproxy/cert"
"github.com/kardianos/mitmproxy/proxy"
"github.com/kardianos/mitmproxy/web"
log "github.com/sirupsen/logrus"
)
type Config struct {
debug int
version bool
certPath string
addr string
webAddr string
ssl_insecure bool
dump string // dump filename
dumpLevel int // dump level
mapperDir string
}
func loadConfig() *Config {
config := new(Config)
flag.IntVar(&config.debug, "debug", 0, "debug mode: 1 - print debug log, 2 - show debug from")
flag.BoolVar(&config.version, "version", false, "show version")
flag.StringVar(&config.addr, "addr", ":9080", "proxy listen addr")
flag.StringVar(&config.webAddr, "web_addr", ":9081", "web interface listen addr")
flag.BoolVar(&config.ssl_insecure, "ssl_insecure", false, "not verify upstream server SSL/TLS certificates.")
flag.StringVar(&config.dump, "dump", "", "dump filename")
flag.IntVar(&config.dumpLevel, "dump_level", 0, "dump level: 0 - header, 1 - header + body")
flag.StringVar(&config.mapperDir, "mapper_dir", "", "mapper files dirpath")
flag.StringVar(&config.certPath, "cert_path", "", "path of generate cert files")
flag.Parse()
return config
}
func main() {
config := loadConfig()
if config.debug > 0 {
rawLog.SetFlags(rawLog.LstdFlags | rawLog.Lshortfile)
log.SetLevel(log.DebugLevel)
} else {
log.SetLevel(log.InfoLevel)
}
if config.debug == 2 {
log.SetReportCaller(true)
}
log.SetOutput(os.Stdout)
log.SetFormatter(&log.TextFormatter{
FullTimestamp: true,
})
l, err := cert.NewPathLoader(config.certPath)
if err != nil {
log.Fatal(err)
}
ca, err := cert.New(l)
if err != nil {
log.Fatal(err)
}
opts := &proxy.Options{
Debug: config.debug,
Addr: config.addr,
StreamLargeBodies: 1024 * 1024 * 5,
InsecureSkipVerifyTLS: config.ssl_insecure,
CA: ca,
}
p, err := proxy.NewProxy(opts)
if err != nil {
log.Fatal(err)
}
if config.version {
fmt.Println("go-mitmproxy: " + p.Version)
os.Exit(0)
}
log.Infof("go-mitmproxy version %v\n", p.Version)
p.AddAddon(&addon.LogAddon{})
p.AddAddon(web.NewWebAddon(config.webAddr))
if config.dump != "" {
dumper := addon.NewDumperWithFilename(config.dump, config.dumpLevel)
p.AddAddon(dumper)
}
if config.mapperDir != "" {
mapper := addon.NewMapper(config.mapperDir)
p.AddAddon(mapper)
}
log.Fatal(p.Start())
}