feat(db): change log

Author: 317787106

framework/src/test/java/org/tron/core/net/messagehandler/MessageHandlerTest.java

@@ -2,6 +2,7 @@
 
 
 import static org.mockito.Mockito.mock;
+import static org.tron.core.net.message.handshake.HelloMessage.getEndpointFromNode;
 
 import com.google.protobuf.ByteString;
 import java.lang.reflect.Field;
@@ -36,7 +37,9 @@
 import org.tron.p2p.connection.Channel;
 import org.tron.p2p.discover.Node;
 import org.tron.p2p.utils.NetUtil;
+import org.tron.protos.Discover.Endpoint;
 import org.tron.protos.Protocol;
+import org.tron.protos.Protocol.HelloMessage.Builder;
 import org.tron.protos.Protocol.ReasonCode;
 
 public class MessageHandlerTest {
@@ -63,10 +66,14 @@ public static void init() throws Exception {
 
 
   @Before
-  public void clearPeers() throws NoSuchFieldException, IllegalAccessException {
-    Field field = PeerManager.class.getDeclaredField("peers");
-    field.setAccessible(true);
-    field.set(PeerManager.class, Collections.synchronizedList(new ArrayList<>()));
+  public void clearPeers() {
+    try {
+      Field field = PeerManager.class.getDeclaredField("peers");
+      field.setAccessible(true);
+      field.set(PeerManager.class, Collections.synchronizedList(new ArrayList<>()));
+    } catch (NoSuchFieldException | IllegalAccessException e) {
+      //ignore
+    }
   }
 
   @Test
@@ -115,30 +122,95 @@ public void testHelloMessage()
     Mockito.when(c1.getInetSocketAddress()).thenReturn(a1);
     Mockito.when(c1.getInetAddress()).thenReturn(a1.getAddress());
     PeerManager.add(ctx, c1);
+    peer = PeerManager.getPeers().get(0);
 
     Method method = p2pEventHandler.getClass()
-        .getDeclaredMethod("onMessage", Channel.class, byte[].class);
+        .getDeclaredMethod("processMessage", PeerConnection.class, byte[].class);
     method.setAccessible(true);
 
     //ok
-    peer = PeerManager.getPeers().get(0);
     Node node = new Node(NetUtil.getNodeId(),
         a1.getAddress().getHostAddress(),
         null,
         a1.getPort());
     HelloMessage helloMessage = new HelloMessage(node, System.currentTimeMillis(),
         ChainBaseManager.getChainBaseManager());
-    method.invoke(p2pEventHandler, c1, helloMessage.getSendBytes());
+    method.invoke(p2pEventHandler, peer, helloMessage.getSendBytes());
 
     //dup hello message
     peer.setHelloMessageReceive(helloMessage);
-    method.invoke(p2pEventHandler, c1, helloMessage.getSendBytes());
+    method.invoke(p2pEventHandler, peer, helloMessage.getSendBytes());
 
     //dup peer
     peer.setHelloMessageReceive(null);
     Mockito.when(c1.isDisconnect()).thenReturn(true);
-    method.invoke(p2pEventHandler, c1, helloMessage.getSendBytes());
+    method.invoke(p2pEventHandler, peer, helloMessage.getSendBytes());
+
+    //invalid hello message
+    try {
+      Protocol.HelloMessage.Builder builder =
+          getHelloMessageBuilder(node, System.currentTimeMillis(),
+              ChainBaseManager.getChainBaseManager());
+
+      BlockCapsule.BlockId hid = ChainBaseManager.getChainBaseManager().getHeadBlockId();
+      Protocol.HelloMessage.BlockId hBlockId = Protocol.HelloMessage.BlockId.newBuilder()
+          .setHash(ByteString.copyFrom(new byte[0]))
+          .setNumber(hid.getNum())
+          .build();
+      builder.setHeadBlockId(hBlockId);
+      helloMessage = new HelloMessage(builder.build().toByteArray());
+      Assert.assertTrue(!helloMessage.valid());
+    } catch (Exception e) {
+      Assert.fail();
+    }
+
+    //relay check service check failed
+    Args.getInstance().fastForward = true;
+    clearPeers();
+    try {
+      Protocol.HelloMessage.Builder builder =
+          getHelloMessageBuilder(node, System.currentTimeMillis(),
+              ChainBaseManager.getChainBaseManager());
+      helloMessage = new HelloMessage(builder.build().toByteArray());
+      method.invoke(p2pEventHandler, peer, helloMessage.getSendBytes());
+    } catch (Exception e) {
+      Assert.fail();
+    }
+
+  }
 
-//    Assert.assertEquals(1, PeerManager.getPeers().size());
+  private Protocol.HelloMessage.Builder getHelloMessageBuilder(Node from, long timestamp,
+      ChainBaseManager chainBaseManager) throws Exception {
+    Endpoint fromEndpoint = getEndpointFromNode(from);
+
+    BlockCapsule.BlockId gid = chainBaseManager.getGenesisBlockId();
+    Protocol.HelloMessage.BlockId gBlockId = Protocol.HelloMessage.BlockId.newBuilder()
+        .setHash(gid.getByteString())
+        .setNumber(gid.getNum())
+        .build();
+
+    BlockCapsule.BlockId sid = chainBaseManager.getSolidBlockId();
+    Protocol.HelloMessage.BlockId sBlockId = Protocol.HelloMessage.BlockId.newBuilder()
+        .setHash(sid.getByteString())
+        .setNumber(sid.getNum())
+        .build();
+
+    BlockCapsule.BlockId hid = chainBaseManager.getHeadBlockId();
+    Protocol.HelloMessage.BlockId hBlockId = Protocol.HelloMessage.BlockId.newBuilder()
+        .setHash(hid.getByteString())
+        .setNumber(hid.getNum())
+        .build();
+    Builder builder = Protocol.HelloMessage.newBuilder();
+    builder.setFrom(fromEndpoint);
+    builder.setVersion(Args.getInstance().getNodeP2pVersion());
+    builder.setTimestamp(timestamp);
+    builder.setGenesisBlockId(gBlockId);
+    builder.setSolidBlockId(sBlockId);
+    builder.setHeadBlockId(hBlockId);
+    builder.setNodeType(chainBaseManager.getNodeType().getType());
+    builder.setLowestBlockNum(chainBaseManager.isLiteNode()
+        ? chainBaseManager.getLowestBlockNum() : 0);
+
+    return builder;
   }
 }