add upstream cli args

README.md

@@ -67,6 +67,8 @@ Usage of go-mitmproxy:
  map remote config filename
  -ssl_insecure
  not verify upstream server SSL/TLS certificates.
+ -upstream string
+ upstream proxy
  -version
  show go-mitmproxy version
  -web_addr string

README_CN.md

@@ -67,6 +67,8 @@ Usage of go-mitmproxy:
  map remote json配置文件地址
  -ssl_insecure
  不验证上游服务器的 SSL/TLS 证书
+ -upstream string
+ upstream proxy
  -version
  显示 go-mitmproxy 版本
  -web_addr string

cmd/go-mitmproxy/config.go

@@ -25,6 +25,7 @@ func loadConfigFromCli() *Config {
  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.Upstream, "upstream", "", "upstream proxy")
  flag.StringVar(&config.MapRemote, "map_remote", "", "map remote config filename")
  flag.StringVar(&config.MapLocal, "map_local", "", "map local config filename")
  flag.StringVar(&config.filename, "f", "", "read config from the filename")
@@ -63,6 +64,9 @@ func mergeConfigs(fileConfig, cliConfig *Config) *Config {
  if cliConfig.DumpLevel != 0 {
  config.DumpLevel = cliConfig.DumpLevel
  }
+ if cliConfig.Upstream != "" {
+ config.Upstream = cliConfig.Upstream
+ }
  if cliConfig.MapRemote != "" {
  config.MapRemote = cliConfig.MapRemote
  }

cmd/go-mitmproxy/main.go

@@ -24,6 +24,7 @@ type Config struct {
  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
+ Upstream string // upstream proxy
  MapRemote string // map remote config filename
  MapLocal string // map local config filename
 
@@ -53,6 +54,7 @@ func main() {
  StreamLargeBodies: 1024 * 1024 * 5,
  SslInsecure: config.SslInsecure,
  CaRootPath: config.CertPath,
+ Upstream: config.Upstream,
  }
 
  p, err := proxy.NewProxy(opts)