golang tcp 2 unix socket proxy

来源：互联网发布：淘宝创建追评失败编辑：程序博客网时间：2024/06/06 03:19

想将mysql 的TCP 封死，所有外部链接由我的proxy来控制，so 写了一个 tcp 转 unix socket 的 proxy。

package mainimport ("os""fmt""net""io""sync""time")type proxy struct{Host stringPort stringLocal string}func runProxy(list []proxy) {wg := sync.WaitGroup{}for _, v := range list {wg.Add(1)go func() {eachServer(v.Host, v.Port, v.Local)wg.Done()}()}wg.Wait()}func eachServer(host string, port string, local string) {l,err := net.Listen("tcp", fmt.Sprintf("%s:%s", host, port))if err != nil {fmt.Print("listen tcp :%s",err.Error())os.Exit(1)}defer l.Close()for {tc,err := l.Accept()if err != nil {fmt.Printf("accept tcp conn :%s",err.Error())tc.Close()continue}go eachConn(local, tc)}}func eachConn(local string, tc net.Conn) {uc,err := net.Dial("unix", local)if err != nil {fmt.Printf("get unix conn :%s",err.Error())uc.Close()return}go io.Copy(tc, uc)go io.Copy(uc, tc)}func main() {list := []proxy{proxy{Host:"192.168.8.101",Port:"3306",Local:"/home/work/data/tmp/mysql3306.sock"}}runProxy(list)}

3 0