Browse Source

SetupConn未被调用

skyfffire 2 năm trước cách đây
mục cha
commit
4d5dda2998
1 tập tin đã thay đổi với 15 bổ sung9 xóa
  1. 15 9
      p2p/server.go

+ 15 - 9
p2p/server.go

@@ -106,15 +106,6 @@ func (server *Server) Start() (err error) {
 	return nil
 }
 
-func (server *Server) StopListener() {
-	<-server.sigs
-
-	close(server.quit)
-	server.loopWG.Done()
-	server.discmix.Close()
-	server.dialsched.stop()
-}
-
 func (server *Server) run() {
 	server.loopWG.Add(1)
 
@@ -131,10 +122,23 @@ running:
 			fmt.Printf("exit signal by user.\n")
 
 			break running
+		case c := <-server.checkpointPostHandshake:
+			fmt.Printf("checkpointPostHandshake: %v\n", c.name)
+		case c := <-server.checkpointAddPeer:
+			fmt.Printf("checkpointAddPeer: %v.\n", c.name)
 		}
 	}
 }
 
+func (server *Server) StopListener() {
+	<-server.sigs
+
+	close(server.quit)
+	server.loopWG.Done()
+	server.discmix.Close()
+	server.dialsched.stop()
+}
+
 // 本地协议
 func (server *Server) setupCaps() (err error) {
 	// 创建握手器
@@ -385,6 +389,8 @@ func (server *Server) newRLPX(conn net.Conn, dialDest *ecdsa.PublicKey) transpor
 }
 
 func (server *Server) SetupConn(fd net.Conn, flags connFlag, dialDest *enode.Node) error {
+	fmt.Printf("setup conn %v.", fd.RemoteAddr())
+
 	c := &conn{fd: fd, flags: flags, cont: make(chan error)}
 	if dialDest == nil {
 		c.transport = server.newRLPX(fd, nil)