コード例 #1
0
ファイル: portmap.go プロジェクト: zengkid/goproxy
func TcpPortmap(pm PortMap, dialer sutils.Dialer) (err error) {
	lsock, err := net.Listen(pm.Net, pm.Src)
	if err != nil {
		return
	}
	log.Info("tcp listening in %s", pm.Src)

	for {
		var sconn, dconn net.Conn

		sconn, err = lsock.Accept()
		if err != nil {
			continue
		}
		log.Info("accept in %s:%s, try to dial %s.", pm.Net, pm.Src, pm.Dst)

		dconn, err = dialer.Dial(pm.Net, pm.Dst)
		if err != nil {
			sconn.Close()
			continue
		}

		go sutils.CopyLink(dconn, sconn)
	}
}
コード例 #2
0
ファイル: portmap.go プロジェクト: zengkid/goproxy
func (upm *UdpPortMapper) UdpPortmap(pm PortMap, dialer sutils.Dialer) (err error) {
	laddr, err := net.ResolveUDPAddr(pm.Net, pm.Src)
	if err != nil {
		return
	}
	sconn, err := net.ListenUDP(pm.Net, laddr)
	if err != nil {
		return
	}
	defer sconn.Close()
	sconn.SetReadBuffer(UDP_READBUFFER)
	log.Info("udp listening in %s", pm.Src)

	for {
		up := NewUdpPackage()
		nr, addr, err := sconn.ReadFrom(up.buf)
		switch err {
		case nil:
		case io.EOF:
			return nil
		default:
			log.Error("%s", err.Error())
			continue
		}
		up.nr = nr

		upm.lock.Lock()
		umc, ok := upm.ports[addr]
		if !ok {
			log.Info("udp forward got new addr %s.", addr)
			dconn, err := dialer.Dial(pm.Net, pm.Dst)
			if err != nil {
				upm.lock.Unlock()
				log.Error("%s", err.Error())
				continue
			}
			umc = NewUdpMapperConn(upm, sconn, dconn, addr, pm.Dst)
			upm.ports[addr] = umc
			umc.Run()
		}
		upm.lock.Unlock()

		umc.ch <- up
	}
}