cow/main.go

package main

import (
	"os"
	"os/signal"
	"runtime"
	"syscall"
	// "syscall"
	// "runtime/pprof"
)

var sigChan = make(chan os.Signal, 1)

// var cpuprofile = flag.String("cpuprofile", "", "write cpu profile to file")

func sigHandler() {
	for sig := range sigChan {
		info.Printf("%v caught, exit\n", sig)
		writeDomainSet()
		break
	}
	os.Exit(0)
}

var hasParentProxy = false

func main() {
	// Parse flags after load config to allow override options in config
	cmdLineConfig := parseCmdLineConfig()
	parseConfig(cmdLineConfig.RcFile)
	// need to update config
	updateConfig(cmdLineConfig)

	initLog()

	initProxyServerAddr()
	initSocksServer()
	initShadowSocks()

	if !hasSocksServer && !hasShadowSocksServer {
		info.Println("no socks/shadowsocks server, can't handle blocked sites")
	} else {
		hasParentProxy = true
	}

	setSelfURL()

	if config.PrintVer {
		printVersion()
		os.Exit(0)
	}

	loadDomainSet()
	/*
		if *cpuprofile != "" {
			f, err := os.Create(*cpuprofile)
			if err != nil {
				info.Println(err)
				os.Exit(1)
			}
			pprof.StartCPUProfile(f)
			signal.Notify(c, os.Interrupt)
			go func() {
				for sig := range c {
					info.Printf("captured %v, stopping profiler and exiting..", sig)
					pprof.StopCPUProfile()
					os.Exit(0)
				}
			}()
		}
	*/

	runtime.GOMAXPROCS(config.Core)

	signal.Notify(sigChan, syscall.SIGINT)
	signal.Notify(sigChan, syscall.SIGTERM)
	go sigHandler()

	go runSSH()

	py := NewProxy(config.ListenAddr)
	py.Serve()
}