Merge pull request #9995 from taosdata/fix/TS-487

[TS-487]<fix>(rpc): fix udp partial message chaos

src/rpc/src/rpcUdp.c

@@ -186,9 +186,11 @@ static void *taosRecvUdpData(void *param) {
   SUdpConn          *pConn = param;
   struct sockaddr_in sourceAdd;
   ssize_t            dataLen;
+  int32_t            msgLen;
   unsigned int       addLen;
   uint16_t           port;
   SRecvInfo          recvInfo;
+  SRpcHead          *pHead;
 
   memset(&sourceAdd, 0, sizeof(sourceAdd));
   addLen = sizeof(sourceAdd);
@@ -216,6 +218,13 @@ static void *taosRecvUdpData(void *param) {
       continue;
     }
 
+    pHead = (SRpcHead *)msg;
+    msgLen = (int32_t)htonl((uint32_t)pHead->msgLen);
+    if (dataLen < msgLen) {
+      tError("%s recvfrom failed(%s): dataLen: %ld, msgLen: %d", pConn->label, strerror(errno), (long)dataLen, msgLen);
+      continue;
+    }
+
     int32_t size = dataLen + tsRpcOverhead;
     char *tmsg = malloc(size);
     if (NULL == tmsg) {

tests/script/unique/migrate/mn2_vn2_repl2_rmMnodeVnodeDir.sim

@@ -51,7 +51,7 @@ sleep 1000
 sql connect
 sleep 1000
 sql create dnode $hostname2
-sleep 1000
+sleep 2000
 
 print ============== step2: create database with replica 2, and create table, insert data
 $totalTableNum = 10
@@ -268,4 +268,4 @@ system sh/exec.sh -n dnode4 -s stop  -x SIGINT
 system sh/exec.sh -n dnode5 -s stop  -x SIGINT
 system sh/exec.sh -n dnode6 -s stop  -x SIGINT
 system sh/exec.sh -n dnode7 -s stop  -x SIGINT
-system sh/exec.sh -n dnode8 -s stop  -x SIGINT
+system sh/exec.sh -n dnode8 -s stop  -x SIGINT
\ No newline at end of file

tests/script/unique/migrate/mn2_vn2_repl2_rmVnodeDir.sim

@@ -51,7 +51,7 @@ sleep 1000
 sql connect
 sleep 1000
 sql create dnode $hostname2
-sleep 1000
+sleep 2000
 
 print ============== step2: create database with replica 2, and create table, insert data
 $totalTableNum = 10
@@ -266,4 +266,4 @@ system sh/exec.sh -n dnode4 -s stop  -x SIGINT
 system sh/exec.sh -n dnode5 -s stop  -x SIGINT
 system sh/exec.sh -n dnode6 -s stop  -x SIGINT
 system sh/exec.sh -n dnode7 -s stop  -x SIGINT
-system sh/exec.sh -n dnode8 -s stop  -x SIGINT
+system sh/exec.sh -n dnode8 -s stop  -x SIGINT
\ No newline at end of file