You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
114 lines
2.6 KiB
Go
114 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
rawLog "log"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/lqqyt2423/go-mitmproxy/addon"
|
|
"github.com/lqqyt2423/go-mitmproxy/proxy"
|
|
"github.com/lqqyt2423/go-mitmproxy/web"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type Config struct {
|
|
version bool // show go-mitmproxy version
|
|
|
|
Addr string // proxy listen addr
|
|
WebAddr string // web interface listen addr
|
|
SslInsecure bool // not verify upstream server SSL/TLS certificates.
|
|
IgnoreHosts []string // a list of ignore hosts
|
|
AllowHosts []string // a list of allow hosts
|
|
CertPath string // path of generate cert files
|
|
Debug int // debug mode: 1 - print debug log, 2 - show debug from
|
|
Dump string // dump filename
|
|
DumpLevel int // dump level: 0 - header, 1 - header + body
|
|
MapperDir string // mapper files dirpath
|
|
|
|
filename string // read config from the filename
|
|
}
|
|
|
|
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,
|
|
})
|
|
|
|
opts := &proxy.Options{
|
|
Debug: config.Debug,
|
|
Addr: config.Addr,
|
|
StreamLargeBodies: 1024 * 1024 * 5,
|
|
SslInsecure: config.SslInsecure,
|
|
CaRootPath: config.CertPath,
|
|
}
|
|
|
|
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)
|
|
|
|
if len(config.IgnoreHosts) > 0 {
|
|
p.SetShouldInterceptRule(func(address string) bool {
|
|
return !matchHost(address, config.IgnoreHosts)
|
|
})
|
|
}
|
|
if len(config.AllowHosts) > 0 {
|
|
p.SetShouldInterceptRule(func(address string) bool {
|
|
return matchHost(address, config.AllowHosts)
|
|
})
|
|
}
|
|
|
|
p.AddAddon(&proxy.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())
|
|
}
|
|
|
|
func matchHost(address string, hosts []string) bool {
|
|
hostname, port := splitHostPort(address)
|
|
for _, host := range hosts {
|
|
h, p := splitHostPort(host)
|
|
if h == hostname && (p == "" || p == port) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func splitHostPort(address string) (string, string) {
|
|
index := strings.LastIndex(address, ":")
|
|
if index == -1 {
|
|
return address, ""
|
|
}
|
|
return address[:index], address[index+1:]
|
|
}
|