#4829

Committed 30 Oct 2025 09:25AM UTC coverage: 49.734% (-11.3%) from 61.071%

Build # #4829

Build Type

push

travis-ci

Committed by

web-flow

Commit Message

Merge pull request #33435 from taosdata/3.0

merge 3.0

Run Details

123072 of 323930 branches covered (37.99%)

Branch coverage included in aggregate %.

7 of 25 new or added lines in 3 files covered. (28.0%)

35232 existing lines in 327 files now uncovered.

172062 of 269495 relevant lines covered (63.85%)

70709785.06 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

53.66

/tools/shell/src/shellNettest.c

/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#define _GNU_SOURCE
#include "shellInt.h"

static void shellWorkAsClient() {
  SShellArgs *pArgs = &shell.args;
  SRpcInit    rpcInit = {0};
  SEpSet      epSet = {.inUse = 0, .numOfEps = 1};
  SRpcMsg     rpcRsp = {0};
  void       *clientRpc = NULL;
  char        pass[TSDB_PASSWORD_LEN + 1] = {0};

  taosEncryptPass_c((uint8_t *)("_pwd"), strlen("_pwd"), pass);
  rpcInit.label = "CHK";
  rpcInit.numOfThreads = 1;
  rpcInit.sessions = 16;
  rpcInit.connType = TAOS_CONN_CLIENT;
  rpcInit.idleTime = 3000;
  rpcInit.user = "_dnd";
  rpcInit.timeToGetConn = 500000;

  memcpy(rpcInit.caPath, tsTLSCaPath, strlen(tsTLSCaPath));
  memcpy(rpcInit.certPath, tsTLSSvrCertPath, strlen(tsTLSSvrCertPath));
  memcpy(rpcInit.keyPath, tsTLSSvrKeyPath, strlen(tsTLSSvrKeyPath));
  memcpy(rpcInit.cliCertPath, tsTLSCliCertPath, strlen(tsTLSCliCertPath));
  memcpy(rpcInit.cliKeyPath, tsTLSCliKeyPath, strlen(tsTLSCliKeyPath));

  taosVersionStrToInt(td_version, &rpcInit.compatibilityVer);
  clientRpc = rpcOpen(&rpcInit);
  if (clientRpc == NULL) {
    printf("failed to init net test client since %s\r\n", terrstr());
    goto _OVER;
  }

  if (pArgs->port == 0) {
    pArgs->port = 6030;
  }
  if (pArgs->host == NULL) {
    pArgs->host = "localhost";
  }
  char fqdn[TSDB_FQDN_LEN] = {0};
  tstrncpy(fqdn, pArgs->host, TSDB_FQDN_LEN);
  strtok(fqdn, ":");

  printf("network test client is initialized, the server is %s:%u\r\n", fqdn, pArgs->port);

  tstrncpy(epSet.eps[0].fqdn, fqdn, TSDB_FQDN_LEN);
  epSet.eps[0].port = (uint16_t)pArgs->port;

  int32_t  totalSucc = 0;
  uint64_t startTime = taosGetTimestampUs();

  for (int32_t i = 0; i < pArgs->pktNum; ++i) {
    SRpcMsg rpcMsg = {.info.ahandle = (void *)0x9525, .info.notFreeAhandle = 1, .msgType = TDMT_DND_NET_TEST};
    rpcMsg.pCont = rpcMallocCont(pArgs->pktLen);
    rpcMsg.contLen = pArgs->pktLen;

    printf("request is sent, size:%d\r\n", rpcMsg.contLen);
    rpcSendRecv(clientRpc, &epSet, &rpcMsg, &rpcRsp);
    if (rpcRsp.code == 0 && rpcRsp.contLen == rpcMsg.contLen) {
      printf("response is received, size:%d\r\n", rpcMsg.contLen);
      if (rpcRsp.code == 0) totalSucc++;
    } else {
      printf("response not received since %s\r\n", tstrerror(rpcRsp.code));
    }

    rpcFreeCont(rpcRsp.pCont);
    rpcRsp.pCont = NULL;
  }

  uint64_t endTime = taosGetTimestampUs();
  uint64_t elT = endTime - startTime;

  printf("\r\ntotal succ:%5d/%d\tcost:%8.2lf ms\tspeed:%8.2lf MB/s\r\n", totalSucc, pArgs->pktNum, elT / 1000.0,
         pArgs->pktLen / (elT / 1000000.0) / 1024.0 / 1024.0 * totalSucc);

_OVER:
  if (clientRpc != NULL) {
    rpcClose(clientRpc);
  }
  if (rpcRsp.pCont != NULL) {
    rpcFreeCont(rpcRsp.pCont);
  }
}

static void shellProcessMsg(void *p, SRpcMsg *pRpc, SEpSet *pEpSet) {
  printf("request is received, size:%d\r\n", pRpc->contLen);
  fflush(stdout);
  SRpcMsg rsp = {.info = pRpc->info, .code = 0};
  rsp.pCont = rpcMallocCont(pRpc->contLen);
  if (rsp.pCont == NULL) {
    rsp.code = TSDB_CODE_OUT_OF_MEMORY;
  } else {
    rsp.contLen = pRpc->contLen;
  }
  rpcSendResponse(&rsp);
}

void shellNettestHandler(int32_t signum, void *sigInfo, void *context) { shellExit(); }

static void shellWorkAsServer() {
  SShellArgs *pArgs = &shell.args;

  if (pArgs->port == 0) {
    pArgs->port = 6030;
  }
  if (pArgs->host == NULL) {
    pArgs->host = "127.0.0.1";
  }

  SRpcInit rpcInit = {0};
  memcpy(rpcInit.localFqdn, pArgs->host, strlen(pArgs->host));
  rpcInit.localPort = pArgs->port;
  rpcInit.label = "CHK";
  rpcInit.numOfThreads = 2;
  rpcInit.cfp = (RpcCfp)shellProcessMsg;
  rpcInit.sessions = 10;
  rpcInit.connType = TAOS_CONN_SERVER;
  rpcInit.idleTime = 3000;

  taosVersionStrToInt(td_version, &rpcInit.compatibilityVer);

  void *serverRpc = rpcOpen(&rpcInit);
  if (serverRpc == NULL) {
    printf("failed to init net test server since %s\r\n", terrstr());
  } else {
    printf("network test server is initialized, %s:%u\r\n", pArgs->host, pArgs->port);
    taosSetSignal(SIGTERM, shellNettestHandler);
    while (1) taosMsleep(10);
  }
}

void shellTestNetWork() {
  (void)osDefaultInit();
  (void)rpcInit();

  if (strcmp(shell.args.netrole, "client") == 0) {
    shellWorkAsClient();
  }

  if (strcmp(shell.args.netrole, "server") == 0) {
    shellWorkAsServer();
  }
}

1	/*
2	* Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
3	*
4	* This program is free software: you can use, redistribute, and/or modify
5	* it under the terms of the GNU Affero General Public License, version 3
6	* or later ("AGPL"), as published by the Free Software Foundation.
7	*
8	* This program is distributed in the hope that it will be useful, but WITHOUT
9	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10	* FITNESS FOR A PARTICULAR PURPOSE.
11	*
12	* You should have received a copy of the GNU Affero General Public License
13	* along with this program. If not, see <http://www.gnu.org/licenses/>.
14	*/
15
16	#define _GNU_SOURCE
17	#include "shellInt.h"
18
19	static void shellWorkAsClient() {	252✔
20	SShellArgs *pArgs = &shell.args;	252✔
21	SRpcInit rpcInit = {0};	252✔
22	SEpSet epSet = {.inUse = 0, .numOfEps = 1};	252✔
23	SRpcMsg rpcRsp = {0};	252✔
24	void *clientRpc = NULL;	252✔
25	char pass[TSDB_PASSWORD_LEN + 1] = {0};	252✔
26
27	taosEncryptPass_c((uint8_t *)("_pwd"), strlen("_pwd"), pass);
28	rpcInit.label = "CHK";	252✔
29	rpcInit.numOfThreads = 1;	252✔
30	rpcInit.sessions = 16;	252✔
31	rpcInit.connType = TAOS_CONN_CLIENT;	252✔
32	rpcInit.idleTime = 3000;	252✔
33	rpcInit.user = "_dnd";	252✔
34	rpcInit.timeToGetConn = 500000;	252✔
35
36	memcpy(rpcInit.caPath, tsTLSCaPath, strlen(tsTLSCaPath));	252!
37	memcpy(rpcInit.certPath, tsTLSSvrCertPath, strlen(tsTLSSvrCertPath));	252!
38	memcpy(rpcInit.keyPath, tsTLSSvrKeyPath, strlen(tsTLSSvrKeyPath));	252!
39	memcpy(rpcInit.cliCertPath, tsTLSCliCertPath, strlen(tsTLSCliCertPath));	252!
40	memcpy(rpcInit.cliKeyPath, tsTLSCliKeyPath, strlen(tsTLSCliKeyPath));	252!
41
42	taosVersionStrToInt(td_version, &rpcInit.compatibilityVer);	252✔
43	clientRpc = rpcOpen(&rpcInit);	252✔
44	if (clientRpc == NULL) {	252!
45	printf("failed to init net test client since %s\r\n", terrstr());	×
46	goto _OVER;	×
47	}
48
49	if (pArgs->port == 0) {	252!
50	pArgs->port = 6030;	252✔
51	}
52	if (pArgs->host == NULL) {	252!
53	pArgs->host = "localhost";	252✔
54	}
55	char fqdn[TSDB_FQDN_LEN] = {0};	252✔
56	tstrncpy(fqdn, pArgs->host, TSDB_FQDN_LEN);	252!
57	strtok(fqdn, ":");	252✔
58
59	printf("network test client is initialized, the server is %s:%u\r\n", fqdn, pArgs->port);	252✔
60
61	tstrncpy(epSet.eps[0].fqdn, fqdn, TSDB_FQDN_LEN);	252✔
62	epSet.eps[0].port = (uint16_t)pArgs->port;	252✔
63
64	int32_t totalSucc = 0;	252✔
65	uint64_t startTime = taosGetTimestampUs();	252✔
66
67	for (int32_t i = 0; i < pArgs->pktNum; ++i) {	5,202✔
68	SRpcMsg rpcMsg = {.info.ahandle = (void *)0x9525, .info.notFreeAhandle = 1, .msgType = TDMT_DND_NET_TEST};	4,950✔
69	rpcMsg.pCont = rpcMallocCont(pArgs->pktLen);	4,950✔
70	rpcMsg.contLen = pArgs->pktLen;	4,950✔
71
72	printf("request is sent, size:%d\r\n", rpcMsg.contLen);	4,950✔
73	rpcSendRecv(clientRpc, &epSet, &rpcMsg, &rpcRsp);	4,950✔
74	if (rpcRsp.code == 0 && rpcRsp.contLen == rpcMsg.contLen) {	4,950!
75	printf("response is received, size:%d\r\n", rpcMsg.contLen);	4,950✔
76	if (rpcRsp.code == 0) totalSucc++;	4,950!
77	} else {
78	printf("response not received since %s\r\n", tstrerror(rpcRsp.code));	×
79	}
80
81	rpcFreeCont(rpcRsp.pCont);	4,950✔
82	rpcRsp.pCont = NULL;	4,950✔
83	}
84
85	uint64_t endTime = taosGetTimestampUs();	252✔
86	uint64_t elT = endTime - startTime;	252✔
87
88	printf("\r\ntotal succ:%5d/%d\tcost:%8.2lf ms\tspeed:%8.2lf MB/s\r\n", totalSucc, pArgs->pktNum, elT / 1000.0,	252✔
89	pArgs->pktLen / (elT / 1000000.0) / 1024.0 / 1024.0 * totalSucc);	252!
90
91	_OVER:	252✔
92	if (clientRpc != NULL) {	252!
93	rpcClose(clientRpc);	252✔
94	}
95	if (rpcRsp.pCont != NULL) {	252!
96	rpcFreeCont(rpcRsp.pCont);	×
97	}
98	}	252✔
99
100	static void shellProcessMsg(void p, SRpcMsg pRpc, SEpSet *pEpSet) {	×
101	printf("request is received, size:%d\r\n", pRpc->contLen);	×
102	fflush(stdout);	×
103	SRpcMsg rsp = {.info = pRpc->info, .code = 0};	×
104	rsp.pCont = rpcMallocCont(pRpc->contLen);	×
105	if (rsp.pCont == NULL) {	×
106	rsp.code = TSDB_CODE_OUT_OF_MEMORY;	×
107	} else {
108	rsp.contLen = pRpc->contLen;	×
109	}
110	rpcSendResponse(&rsp);	×
111	}	×
112
113	void shellNettestHandler(int32_t signum, void sigInfo, void context) { shellExit(); }	×
114
UNCOV 115	static void shellWorkAsServer() {	×
UNCOV 116	SShellArgs *pArgs = &shell.args;	×
117
UNCOV 118	if (pArgs->port == 0) {	×
UNCOV 119	pArgs->port = 6030;	×
120	}
UNCOV 121	if (pArgs->host == NULL) {	×
UNCOV 122	pArgs->host = "127.0.0.1";	×
123	}
124
UNCOV 125	SRpcInit rpcInit = {0};	×
UNCOV 126	memcpy(rpcInit.localFqdn, pArgs->host, strlen(pArgs->host));	×
UNCOV 127	rpcInit.localPort = pArgs->port;	×
UNCOV 128	rpcInit.label = "CHK";	×
UNCOV 129	rpcInit.numOfThreads = 2;	×
UNCOV 130	rpcInit.cfp = (RpcCfp)shellProcessMsg;	×
UNCOV 131	rpcInit.sessions = 10;	×
UNCOV 132	rpcInit.connType = TAOS_CONN_SERVER;	×
UNCOV 133	rpcInit.idleTime = 3000;	×
134
UNCOV 135	taosVersionStrToInt(td_version, &rpcInit.compatibilityVer);	×
136
UNCOV 137	void *serverRpc = rpcOpen(&rpcInit);	×
UNCOV 138	if (serverRpc == NULL) {	×
UNCOV 139	printf("failed to init net test server since %s\r\n", terrstr());	×
140	} else {
141	printf("network test server is initialized, %s:%u\r\n", pArgs->host, pArgs->port);	×
142	taosSetSignal(SIGTERM, shellNettestHandler);	×
143	while (1) taosMsleep(10);	×
144	}
UNCOV 145	}	×
146
147	void shellTestNetWork() {	252✔
148	(void)osDefaultInit();	252✔
149	(void)rpcInit();	252✔
150
151	if (strcmp(shell.args.netrole, "client") == 0) {	252!
152	shellWorkAsClient();	252✔
153	}
154
155	if (strcmp(shell.args.netrole, "server") == 0) {	252!
UNCOV 156	shellWorkAsServer();	×
157	}
158	}	252✔

taosdata / TDengine / #4829

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous