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.
108 lines
2.8 KiB
Go
108 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func loadConfigFromFile(filename string) (*Config, error) {
|
|
data, err := os.ReadFile(filename)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var config Config
|
|
if err := json.Unmarshal(data, &config); err != nil {
|
|
return nil, err
|
|
}
|
|
return &config, nil
|
|
}
|
|
|
|
func loadConfigFromCli() *Config {
|
|
config := new(Config)
|
|
|
|
flag.BoolVar(&config.version, "version", false, "show go-mitmproxy version")
|
|
flag.StringVar(&config.Addr, "addr", ":9080", "proxy listen addr")
|
|
flag.StringVar(&config.WebAddr, "web_addr", ":9081", "web interface listen addr")
|
|
flag.BoolVar(&config.SslInsecure, "ssl_insecure", false, "not verify upstream server SSL/TLS certificates.")
|
|
flag.Var((*arrayValue)(&config.IgnoreHosts), "ignore_hosts", "a list of ignore hosts")
|
|
flag.Var((*arrayValue)(&config.AllowHosts), "allow_hosts", "a list of allow hosts")
|
|
flag.StringVar(&config.CertPath, "cert_path", "", "path of generate cert files")
|
|
flag.IntVar(&config.Debug, "debug", 0, "debug mode: 1 - print debug log, 2 - show debug from")
|
|
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.filename, "f", "", "read config from the filename")
|
|
flag.Parse()
|
|
|
|
return config
|
|
}
|
|
|
|
func mergeConfigs(fileConfig, cliConfig *Config) *Config {
|
|
config := new(Config)
|
|
*config = *fileConfig
|
|
if cliConfig.Addr != "" {
|
|
config.Addr = cliConfig.Addr
|
|
}
|
|
if cliConfig.WebAddr != "" {
|
|
config.WebAddr = cliConfig.WebAddr
|
|
}
|
|
if cliConfig.SslInsecure {
|
|
config.SslInsecure = cliConfig.SslInsecure
|
|
}
|
|
if len(cliConfig.IgnoreHosts) > 0 {
|
|
config.IgnoreHosts = cliConfig.IgnoreHosts
|
|
}
|
|
if len(cliConfig.AllowHosts) > 0 {
|
|
config.AllowHosts = cliConfig.AllowHosts
|
|
}
|
|
if cliConfig.CertPath != "" {
|
|
config.CertPath = cliConfig.CertPath
|
|
}
|
|
if cliConfig.Debug != 0 {
|
|
config.Debug = cliConfig.Debug
|
|
}
|
|
if cliConfig.Dump != "" {
|
|
config.Dump = cliConfig.Dump
|
|
}
|
|
if cliConfig.DumpLevel != 0 {
|
|
config.DumpLevel = cliConfig.DumpLevel
|
|
}
|
|
if cliConfig.MapperDir != "" {
|
|
config.MapperDir = cliConfig.MapperDir
|
|
}
|
|
return config
|
|
}
|
|
|
|
func loadConfig() *Config {
|
|
cliConfig := loadConfigFromCli()
|
|
if cliConfig.version {
|
|
return cliConfig
|
|
}
|
|
if cliConfig.filename == "" {
|
|
return cliConfig
|
|
}
|
|
|
|
fileConfig, err := loadConfigFromFile(cliConfig.filename)
|
|
if err != nil {
|
|
log.Warnf("read config from %v error %v", cliConfig.filename, err)
|
|
return cliConfig
|
|
}
|
|
return mergeConfigs(fileConfig, cliConfig)
|
|
}
|
|
|
|
// arrayValue 实现了 flag.Value 接口
|
|
type arrayValue []string
|
|
|
|
func (a *arrayValue) String() string {
|
|
return fmt.Sprint(*a)
|
|
}
|
|
|
|
func (a *arrayValue) Set(value string) error {
|
|
*a = append(*a, value)
|
|
return nil
|
|
}
|