Function TestAddrReqHandle_maskok

p2pserver/message/utils/msg_handler_test.go:293–348

create two neighbors, one masked, one un-masked send addr-req from un-mask peer, get itself in addr-rsp

(t *testing.T)

// send addr-req from un-mask peer, get itself in addr-rsp
//
func TestAddrReqHandle_maskok(t *testing.T) {
	network = NewMockP2p()

	// Simulate a remote peer to be added to the neighbor peers
	testID := uint64(123456)
	remotePeer := peer.NewPeer()
	remotePeer.UpdateInfo(time.Now(), 1, 12345678, 20336,
		testID, 0, 12345, "1.5.2")
	remotePeer.Link.SetAddr("1.2.3.4:5001")
	remotePeer.Link.SetPort(5001)
	network.AddNbrNode(remotePeer)
	remotePeer.SetState(msgCommon.ESTABLISH)

	testID2 := uint64(1234567)
	remotePeer2 := peer.NewPeer()
	remotePeer2.UpdateInfo(time.Now(), 1, 12345678, 20336,
		testID2, 0, 12345, "1.5.2")
	remotePeer2.Link.SetAddr("1.2.3.5:5002")
	remotePeer2.Link.SetPort(5002)
	network.AddNbrNode(remotePeer2)
	remotePeer2.SetState(msgCommon.ESTABLISH)

	// Construct an address request packet
	buf := msgpack.NewAddrReq()

	msg := &types.MsgPayload{
		Id:      testID2,
		Addr:    "test",
		Payload: buf,
	}

	config.DefConfig.P2PNode.ReservedPeersOnly = true
	config.DefConfig.P2PNode.ReservedCfg.MaskPeers = []string{"1.2.3.4"}

	// Invoke AddrReqHandle to handle the msg
	AddrReqHandle(msg, network, nil)

	// verify 1.2.3.4 is masked
	for _, msg := range network.SentMsgs {
		addrMsg, ok := msg.(*types.Addr)
		if !ok {
			t.Fatalf("invalid addr msg %s", msg.CmdType())
		}
		if len(addrMsg.NodeAddrs) != 1 {
			t.Fatalf("invalid addr count: %v", addrMsg.NodeAddrs)
		}
		var ip net.IP
		ip = addrMsg.NodeAddrs[0].IpAddr[:]
		addr := fmt.Sprintf("%v:%d", ip, addrMsg.NodeAddrs[0].Port)
		if addr != remotePeer2.Link.GetAddr() {
			t.Fatalf("invalid addr: %s vs %s", addr, remotePeer2.Link.GetAddr())
		}
	}

	network.DelNbrNode(testID)
}

//