Example #1
0
func (this *Peer) connect() (err error) {
	//bind local addr since we will use this for peer addr on remote server
	laddr, err := net.ResolveTCPAddr("tcp", config.PushdConf.TcpListenAddr)
	if err != nil {
		panic(err)
	}
	laddr.Port = 0
	dialer := &net.Dialer{LocalAddr: laddr}

	this.Conn, err = dialer.Dial("tcp", this.addr)
	if err != nil {
		log.Warn("s2s connect to %s error: %s", this.addr, err.Error())
	}
	if this.Conn != nil {
		// just wait for s2s server close the connection
		go func() {
			for {
				input := make([]byte, 1460)
				_, err = this.Conn.Read(input)
				if err != nil {
					this.Conn.Close()
					return
				}
			}
		}()
	}
	return
}
Example #2
0
func (this Parser) match(txt, tp string) bool {
	re, exists := this.Regexps[tp]
	if !exists {
		log.Warn("regexp not found for %s", tp)
	}
	return re.MatchString(txt)
}
Example #3
0
func (this *Parser) parse(txt, tp string) []string {
	re, exists := this.Regexps[tp]
	if !exists {
		log.Warn("regexp not found for %s", tp)
	}
	match := re.FindAllStringSubmatch(txt, -1)
	for _, r := range match {
		for _, rr := range r {
			log.Debug(rr)
		}
	}

	if len(match) < 1 || len(match[0]) < 2 {
		return []string{}
	}
	return match[0][1:]
}
Example #4
0
func (this *Peer) writeMsg(msg string) {
	var err error

	if this.Conn != nil {
		_, err = this.Write([]byte(msg))
	}
	if err != nil || this.Conn == nil {
		// retry
		for i := 0; i < RETRY_CNT; i++ {
			err = this.connect()
			if err != nil {
				log.Warn("write to peer %s error: %s", this.addr, err.Error())
			} else {
				if this.Conn != nil {
					_, err = this.Write([]byte(msg))
					if err == nil {
						break
					}
				}
			}
		}
	}
}
Example #5
0
func (this *Forwarder) reconnect() {
	if this.Conn != nil {
		this.Conn.Close()
	}
	conn, err := net.Dial("tcp", this.config.ToAddr)
	if err != nil {
		log.Error(err)
	}
	this.Conn = conn

	if conn != nil {
		go func() {
			for {
				_, err := this.Conn.Read(make([]byte, 1000))
				if err != nil {
					log.Warn(err)
				}
				this.Conn.Close()
				break
			}
		}()
	}
}