/* * * Copyright (c) 2018 Huawei Technologies Co.,Ltd. * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at: * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #define _GNU_SOURCE #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #define _BIND bind #define _LISTEN listen #define _SOCKET socket #define _ACCEPT accept #define _SEND send #define _RECV recv #define _CLOSE close #define _CONNECT connect #define _PROTOCOL IPPROTO_TCP #define MAX_TEST_TIME 1000 #define MSG_LENGTH 256 #define CORE_NUM 8 #define START_CPU_ID 2 #define MAX_CONN_LIMIT 256 #define MAX_PORT_NUM 65535 //1:A-B ;0:A-B-A #define SEND_RECV_MODE 1 #define MAX_EVENTS 1000 #define Default_PortID 12345 #define Default_SleepCnt 10000 #define Default_SleepTime 1000000 #define Flag_Print 1 #define Fd_Number 1 static struct sockaddr_in g_dest; static struct sockaddr_in g_src; static struct sockaddr_in g_recv; int times = MAX_TEST_TIME; int msg_length = MSG_LENGTH; int coreNum = CORE_NUM; int startCPUId = START_CPU_ID; int connectNum = MAX_CONN_LIMIT; int msgMode = SEND_RECV_MODE; int sleepCnt = Default_SleepCnt; int sleepTime = Default_SleepTime; //us int fdNumber = Fd_Number; int flagPrint = Flag_Print; int unitPrint = 0; int waitTime = 0; int srcPort = Default_PortID; int destPort = 0; int recvPort = 0; char sendarr[256] = ""; char recvarr[256] = ""; char destarr[256] = ""; static void setArgsDefault () { memset (&g_dest, 0, sizeof (g_dest)); g_dest.sin_family = AF_INET; g_dest.sin_addr.s_addr = inet_addr ("127.0.0.1"); g_dest.sin_port = htons (12345); bzero (&(g_dest.sin_zero), 8); memset (&g_src, 0, sizeof (g_src)); g_src.sin_family = AF_INET; g_src.sin_addr.s_addr = inet_addr ("0.0.0.0"); g_src.sin_port = htons (7895); bzero (&(g_src.sin_zero), 8); memset (&g_recv, 0, sizeof (g_recv)); g_recv.sin_family = AF_INET; g_recv.sin_addr.s_addr = inet_addr ("0.0.0.0"); g_recv.sin_port = htons (7895); bzero (&(g_recv.sin_zero), 8); } static int process_arg (int argc, char *argv[]) { int opt = 0; int error = 0; const char *optstring = "p:d:s:a:l:t:e:i:f:r:n:w:u:x:"; int rw_mark = 0; if (argc < 4) { printf ("Param info :-p dest_portid; -d dest_serverIP; -a src_portid; -s src_clientIP; -l msg_length; \n"); printf ("\t-t MAX_TEST_TIME; -i msg_interval ; -f client fd number ; -r receive port; -n connectNum(one server vs n client)\n"); return 0; } while ((opt = getopt (argc, argv, optstring)) != -1) { switch (opt) { case 'p': destPort = atoi (optarg); g_dest.sin_port = htons (atoi (optarg)); break; case 'd': stpcpy (destarr, optarg); g_dest.sin_addr.s_addr = inet_addr (optarg); break; case 's': stpcpy (sendarr, optarg); g_src.sin_addr.s_addr = inet_addr (optarg); g_recv.sin_addr.s_addr = inet_addr (optarg); break; case 'a': //g_src.sin_port = htons(atoi(optarg)); srcPort = atoi (optarg); break; case 'l': msg_length = atoi (optarg); break; case 't': times = atoi (optarg); break; case 'e': sleepCnt = atoi (optarg); break; case 'i': sleepTime = atoi (optarg); break; case 'f': fdNumber = atoi (optarg); break; case 'r': recvPort = atoi (optarg); g_recv.sin_port = htons (atoi (optarg)); break; case 'n': connectNum = atoi (optarg); break; case 'w': waitTime = atoi (optarg); break; case 'u': unitPrint = atoi (optarg); break; case 'x': flagPrint = atoi (optarg); break; } } return 1; } void process_client (void) { int sendLen = 0; int i = 0, t = 0, p = 0, optval = 0, ret = 0; char send_buf[1000] = ""; char recv_buf[1000] = ""; int c_socketfd[100] = { 0 }; int errbind[1000] = { 0 }; int errconn[1000] = { 0 }; int send_count[1000] = { 0 }; int pps = 0; long pps_time = 0; struct timespec startTime, endTime; struct timespec countStart; struct timespec countEnd; memset (&countStart, 0, sizeof (countStart)); memset (&countEnd, 0, sizeof (countEnd)); for (i = 0; i < fdNumber; i++) { c_socketfd[i] = _SOCKET (PF_INET, SOCK_STREAM, _PROTOCOL); if (0 > c_socketfd[i]) { printf ("client %d failed,err %d\n", i, errno); } else { printf ("client %d created success\n", i); } } for (i = 0; i < fdNumber; i++) { optval = 1; ret = setsockopt (c_socketfd[i], SOL_SOCKET, SO_REUSEADDR, (void *) &optval, sizeof (optval)); if (ret == -1) { printf ("Couldn't setsockopt(SO_REUSEADDR)\n"); break; } } for (i = 0; i < fdNumber; i++) { g_src.sin_port = htons (srcPort + i); errbind[i] = _BIND (c_socketfd[i], (struct sockaddr *) &g_src, sizeof (g_src)); if (errbind[i] < 0) { printf ("client %d bind Failed %d\n", i, errno); _CLOSE (c_socketfd[i]); c_socketfd[i] = -1; continue; } else { printf ("client %d bind Success port:%d IP:%s\n", i, ntohs (g_src.sin_port), inet_ntoa (*((struct in_addr *) &(g_src.sin_addr.s_addr)))); } } for (i = 0; i < fdNumber; i++) { if (errbind[i] >= 0) { errconn[i] = _CONNECT (c_socketfd[i], (struct sockaddr *) &g_dest, sizeof (g_dest)); if (errconn[i] < 0) { printf ("client %d Connect Failed %s\n", i, strerror(errno)); _CLOSE (c_socketfd[i]); c_socketfd[i] = -1; continue; } else { printf ("client %d Connect Success port:%d, IP:%s\n", i, ntohs (g_dest.sin_port), inet_ntoa (* ((struct in_addr *) &(g_dest.sin_addr.s_addr)))); } } } sleep (1); clock_gettime (CLOCK_MONOTONIC, &startTime); clock_gettime (CLOCK_MONOTONIC, &countStart); int recvLen2, recvLen; for (t = 0; t < times; t++) { for (i = 0; i < fdNumber; i++) { if (c_socketfd[i] < 0) { continue; } do { sendLen = _SEND (c_socketfd[i], send_buf, msg_length, 0); } while (sendLen <= 0); send_count[i]++; recvLen = 0; do { recvLen2 = recv (c_socketfd[i], recv_buf + recvLen, msg_length - recvLen, 0); if (recvLen2 <= 0) { continue; } else if (recvLen2 == msg_length - recvLen) { recvLen = 0; break; } else if (recvLen2 < msg_length - recvLen) { recvLen += recvLen2; } } while (1); } if (0 != sleepTime) { if ((t % sleepCnt) == 0) { usleep (sleepTime); } } if ((send_count[0] % unitPrint) == 0 && send_count[0] > 0) { clock_gettime (CLOCK_MONOTONIC, &countEnd); pps_time = (countEnd.tv_sec - countStart.tv_sec) * 1000000000 + countEnd.tv_nsec - countStart.tv_nsec; pps = ((float) 1000000000 / pps_time) * unitPrint * fdNumber; if ((flagPrint != 0)) { printf (" sendcount %d, time: %ld ns\n", send_count[0] * fdNumber, pps_time); } printf ("sendcount %d , pps=%d\n", send_count[0] * fdNumber, pps); clock_gettime (CLOCK_MONOTONIC, &countStart); } } clock_gettime (CLOCK_MONOTONIC, &endTime); for (i = 0; i < fdNumber; i++) { printf ("client %d send %d , sendtime :%ld s %ld ns\n", i, send_count[i], endTime.tv_sec - startTime.tv_sec, endTime.tv_nsec - startTime.tv_nsec); } for (i = 0; i < fdNumber; i++) { printf ("client %d close!\n", i); if (c_socketfd[i] > 0) _CLOSE (c_socketfd[i]); } pthread_exit (NULL); } void main (int argc, char *argv[]) { socklen_t addrlen = sizeof (struct sockaddr); int err = 0, result = 0, ret = 0; int i = 0, j = 0, optval = 0, z = 0, x, listenfd; cpu_set_t mask; setArgsDefault (); ret = process_arg (argc, argv); if (ret != 1) { printf ("The param error\n"); return; } pthread_t client_thread_id; if (pthread_create (&client_thread_id, NULL, (void *) (&process_client), NULL) == -1) { printf ("create client thread fail\n"); return; } printf ("create client thread success\n"); if (client_thread_id != 0) { printf ("Client Thread join\n"); pthread_join (client_thread_id, NULL); } return; }