|
|
@@ -0,0 +1,338 @@
|
|
|
+#include <WinSock2.h>
|
|
|
+#include <stdio.h>
|
|
|
+#include <stdarg.h>
|
|
|
+#include <time.h>
|
|
|
+
|
|
|
+char * HOST = "127.0.0.1";
|
|
|
+unsigned short PORT = 80;
|
|
|
+static const char * RESOURCELIST[] = {
|
|
|
+ "/ajax/echo.cgi",
|
|
|
+ "/imagetest/00.png",
|
|
|
+ "/args.cgi",
|
|
|
+ "/_stat",
|
|
|
+ "/_echo"
|
|
|
+};
|
|
|
+static const char * METHODLIST[] = {
|
|
|
+ "GET",
|
|
|
+ "POST"
|
|
|
+};
|
|
|
+
|
|
|
+
|
|
|
+static int CLIENTCOUNT = 20;
|
|
|
+static int TESTCYCLES = 50;
|
|
|
+static int RESOURCEINDEX = 0;
|
|
|
+static int METHODINDEX = 0;
|
|
|
+
|
|
|
+
|
|
|
+int sockvprintf(SOCKET soc, const char * fmt, va_list vl) {
|
|
|
+
|
|
|
+ char buf[1024*8];
|
|
|
+ int len = vsprintf_s(buf, sizeof(buf), fmt, vl);
|
|
|
+ int ret = send(soc, buf, len, 0);
|
|
|
+ return ret;
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+int sockprintf(SOCKET soc, const char * fmt, ...) {
|
|
|
+
|
|
|
+ int ret = -1;
|
|
|
+ va_list vl;
|
|
|
+ va_start(vl, fmt);
|
|
|
+ ret = sockvprintf(soc, fmt, vl);
|
|
|
+ va_end(vl);
|
|
|
+ return ret;
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+static struct sockaddr_in target = {0};
|
|
|
+static CRITICAL_SECTION cs = {0};
|
|
|
+static size_t expectedData = 0;
|
|
|
+static DWORD_PTR availableCPUs = 1;
|
|
|
+static DWORD_PTR totalCPUs = 1;
|
|
|
+
|
|
|
+static unsigned good = 0;
|
|
|
+static unsigned bad = 0;
|
|
|
+unsigned long postSize = 0;
|
|
|
+unsigned long extraHeadSize = 0;
|
|
|
+unsigned long queryStringSize = 0;
|
|
|
+
|
|
|
+
|
|
|
+int WINAPI ClientMain(void * clientNo) {
|
|
|
+
|
|
|
+ SOCKET soc;
|
|
|
+ time_t lastData;
|
|
|
+ size_t totalData = 0;
|
|
|
+ int isBody = 0;
|
|
|
+ int isTest = (clientNo == 0);
|
|
|
+ int cpu = ((int)clientNo) % 1000;
|
|
|
+ int timeOut = 10;
|
|
|
+ const char * resource = 0;
|
|
|
+ const char * method = 0;
|
|
|
+ unsigned long i;
|
|
|
+
|
|
|
+ // Method: PUT or GET
|
|
|
+ if (METHODINDEX < sizeof(METHODLIST)/sizeof(METHODLIST[0])) {
|
|
|
+ method = METHODLIST[METHODINDEX];
|
|
|
+ }
|
|
|
+ if (method == 0) {
|
|
|
+ EnterCriticalSection(&cs);
|
|
|
+ printf("\r\nClient %u: bad method\a\r\n", (int)clientNo);
|
|
|
+ LeaveCriticalSection(&cs);
|
|
|
+ return 1;
|
|
|
+ }
|
|
|
+
|
|
|
+ // Resource
|
|
|
+ if (RESOURCEINDEX < sizeof(RESOURCELIST)/sizeof(RESOURCELIST[0])) {
|
|
|
+ resource = RESOURCELIST[RESOURCEINDEX];
|
|
|
+ }
|
|
|
+ if (resource == 0) {
|
|
|
+ EnterCriticalSection(&cs);
|
|
|
+ printf("\r\nClient %u: bad resource\a\r\n", (int)clientNo);
|
|
|
+ LeaveCriticalSection(&cs);
|
|
|
+ return 2;
|
|
|
+ }
|
|
|
+
|
|
|
+ // CPU
|
|
|
+ if ((!isTest) && (((1ULL<<cpu) & availableCPUs)!=0)) {
|
|
|
+ SetThreadAffinityMask(GetCurrentThread(), 1ULL<<cpu);
|
|
|
+ }
|
|
|
+
|
|
|
+ // TCP
|
|
|
+ soc = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
|
|
|
+ if (soc==INVALID_SOCKET) {
|
|
|
+ EnterCriticalSection(&cs);
|
|
|
+ printf("\r\nClient %u: cannot create socket\a\r\n", (int)clientNo);
|
|
|
+ LeaveCriticalSection(&cs);
|
|
|
+ return 3;
|
|
|
+ }
|
|
|
+
|
|
|
+ // comment in to disable Nagle:
|
|
|
+ {int disable_Nagle = 1; setsockopt(soc, IPPROTO_TCP, TCP_NODELAY, (char *) &disable_Nagle, sizeof(disable_Nagle));}
|
|
|
+
|
|
|
+ if (connect(soc, (SOCKADDR*)&target, sizeof(target))) {
|
|
|
+ EnterCriticalSection(&cs);
|
|
|
+ printf("\r\nClient %u: cannot connect to server %s:%u\a\r\n", (int)clientNo, HOST, PORT);
|
|
|
+ LeaveCriticalSection(&cs);
|
|
|
+ return 4;
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ // HTTP request
|
|
|
+ if (queryStringSize>0) {
|
|
|
+ sockprintf(soc, "%s %s?", method, resource);
|
|
|
+
|
|
|
+ for (i=0;i<(queryStringSize/10);i++) {sockprintf(soc, "1234567890");}
|
|
|
+ for (i=0;i<(queryStringSize%10);i++) {sockprintf(soc, "_");}
|
|
|
+
|
|
|
+ sockprintf(soc, " HTTP/1.1\r\nHost: %s\r\nConnection: Close\r\n", HOST);
|
|
|
+ } else {
|
|
|
+ sockprintf(soc, "%s %s HTTP/1.1\r\nHost: %s\r\nConnection: Close\r\n", method, resource, HOST);
|
|
|
+ }
|
|
|
+
|
|
|
+ for (i=0;i<(extraHeadSize/25);i++) {sockprintf(soc, "Comment%04u: 1234567890\r\n", i % 10000);} /* omit (extraHeadSize%25) */
|
|
|
+
|
|
|
+ if (!strcmp(method,"GET")) {
|
|
|
+ sockprintf(soc, "\r\n");
|
|
|
+ } else {
|
|
|
+ // not GET
|
|
|
+ sockprintf(soc, "Content-Length: %u\r\n\r\n", postSize);
|
|
|
+
|
|
|
+ for (i=0;i<postSize/10;i++) {sockprintf(soc, "1234567890");}
|
|
|
+ for (i=0;i<postSize%10;i++) {sockprintf(soc, ".");}
|
|
|
+
|
|
|
+ timeOut += postSize/10000;
|
|
|
+ }
|
|
|
+
|
|
|
+ shutdown(soc, SD_SEND);
|
|
|
+
|
|
|
+ // wait for response from the server
|
|
|
+ lastData = time(0);
|
|
|
+ for (;;) {
|
|
|
+ char buf[2048];
|
|
|
+ int chunkSize = 0;
|
|
|
+ unsigned long dataReady = 0;
|
|
|
+
|
|
|
+ Sleep(1);
|
|
|
+
|
|
|
+ if (ioctlsocket(soc, FIONREAD, &dataReady) < 0) break;
|
|
|
+ if (dataReady) {
|
|
|
+ chunkSize = recv(soc, buf, sizeof(buf), 0);
|
|
|
+ if (chunkSize<0) {
|
|
|
+ printf("Error: recv failed for client %i\r\n", (int)clientNo);
|
|
|
+ break;
|
|
|
+ } else if (!isBody) {
|
|
|
+ char * headEnd = strstr(buf,"\xD\xA\xD\xA");
|
|
|
+ if (headEnd) {
|
|
|
+ headEnd+=4;
|
|
|
+ chunkSize -= ((int)headEnd - (int)buf);
|
|
|
+ if (chunkSize>0) {
|
|
|
+ totalData += chunkSize;
|
|
|
+ lastData = time(0);
|
|
|
+ //fwrite(headEnd,1,got,STORE);
|
|
|
+ }
|
|
|
+ isBody=1;
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ totalData += chunkSize;
|
|
|
+ lastData = time(0);
|
|
|
+ //fwrite(buf,1,got,STORE);
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ time_t current = time(0);
|
|
|
+ if (difftime(current, lastData) > timeOut) break;
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ shutdown(soc, SD_BOTH);
|
|
|
+
|
|
|
+ closesocket(soc);
|
|
|
+
|
|
|
+ EnterCriticalSection(&cs);
|
|
|
+ if (isTest) {
|
|
|
+ expectedData = totalData;
|
|
|
+ } else if (totalData != expectedData) {
|
|
|
+ printf("Error: Client %u got %u bytes instead of %u\r\n", (int)clientNo, totalData, expectedData);
|
|
|
+ bad++;
|
|
|
+ } else {
|
|
|
+ good++;
|
|
|
+ }
|
|
|
+ LeaveCriticalSection(&cs);
|
|
|
+
|
|
|
+ return 0;
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+void RunMultiClientTest(int loop) {
|
|
|
+
|
|
|
+ HANDLE *hThread = calloc(CLIENTCOUNT, sizeof(hThread[0]));
|
|
|
+ int i;
|
|
|
+ DWORD res;
|
|
|
+
|
|
|
+ for (i=0;i<CLIENTCOUNT;i++) {
|
|
|
+ DWORD dummy;
|
|
|
+ hThread[i] = CreateThread(NULL, 1024*32, (LPTHREAD_START_ROUTINE)ClientMain, (void*)(1000*loop+i), 0, &dummy);
|
|
|
+ }
|
|
|
+
|
|
|
+ WaitForMultipleObjects(CLIENTCOUNT, hThread, TRUE, 15000);
|
|
|
+ for (i=0;i<CLIENTCOUNT;i++) {
|
|
|
+ res = WaitForSingleObject(hThread[i], 0);
|
|
|
+ if (res == WAIT_OBJECT_0) {
|
|
|
+ CloseHandle(hThread[i]);
|
|
|
+ hThread[i]=0;
|
|
|
+ }
|
|
|
+ }
|
|
|
+ for (i=0;i<CLIENTCOUNT;i++) {
|
|
|
+ if (hThread[i]) {
|
|
|
+ EnterCriticalSection(&cs);
|
|
|
+ SuspendThread(hThread[i]); // -> check this thread in the debugger
|
|
|
+ printf("Thread %i did not finish!\r\n", (int)(1000*loop+i));
|
|
|
+ LeaveCriticalSection(&cs);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ EnterCriticalSection(&cs);
|
|
|
+ printf("Test cylce %u completed\r\n\r\n", loop);
|
|
|
+ LeaveCriticalSection(&cs);
|
|
|
+
|
|
|
+ free(hThread);
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+int MultiClientTestAutomatic(unsigned long initialPostSize) {
|
|
|
+
|
|
|
+ FILE * log;
|
|
|
+ int cycle;
|
|
|
+
|
|
|
+ postSize = initialPostSize;
|
|
|
+
|
|
|
+ do {
|
|
|
+ printf("Preparing test with %u bytes of data ...", postSize);
|
|
|
+ ClientMain(0);
|
|
|
+ if (expectedData==0) {
|
|
|
+ printf(" Error: Could not read any data\a\r\n");
|
|
|
+ return 1;
|
|
|
+ }
|
|
|
+ printf(" OK: %u bytes of data\r\n", expectedData);
|
|
|
+ printf("Starting multi client test: %i cycles, %i clients each\r\n\r\n", (int)TESTCYCLES, (int)CLIENTCOUNT);
|
|
|
+ good=bad=0;
|
|
|
+
|
|
|
+ for (cycle=1;cycle<=TESTCYCLES;cycle++) {
|
|
|
+ RunMultiClientTest(cycle);
|
|
|
+ }
|
|
|
+
|
|
|
+ printf("\r\n--------\r\n%u errors\r\n%u OK\r\n--------\r\n\r\n", bad, good);
|
|
|
+ log = fopen("testclient.log", "at");
|
|
|
+ if (log) {
|
|
|
+ fprintf(log, "%u\t%u\t%u\r\n", postSize, good, bad);
|
|
|
+ fclose(log);
|
|
|
+ }
|
|
|
+
|
|
|
+ postSize = (postSize!=0) ? (postSize<<1) : 1;
|
|
|
+
|
|
|
+ } while (postSize!=0);
|
|
|
+
|
|
|
+ return 0;
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+int SingleClientTestAutomatic(unsigned long initialPostSize) {
|
|
|
+
|
|
|
+ FILE * log;
|
|
|
+ int cycle;
|
|
|
+ int i;
|
|
|
+
|
|
|
+ postSize = initialPostSize;
|
|
|
+ for (cycle=0;;cycle++) {
|
|
|
+ good=bad=0;
|
|
|
+ for (i=0;i<1000;i++) {
|
|
|
+ expectedData=3;
|
|
|
+ ClientMain((void*)1);
|
|
|
+ }
|
|
|
+ log = fopen("testclient.log", "at");
|
|
|
+ if (log) {
|
|
|
+ fprintf(log, "Cylce<%u>\t%u\t%u\r\n", cycle, good, bad);
|
|
|
+ fclose(log);
|
|
|
+ }
|
|
|
+ printf("test cycle %u: %u good, %u bad\r\n", cycle, good, bad);
|
|
|
+ }
|
|
|
+
|
|
|
+ return 0;
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+int main(int argc, char * argv[]) {
|
|
|
+
|
|
|
+ WSADATA wsaData = {0};
|
|
|
+ HOSTENT * lpHost = 0;
|
|
|
+
|
|
|
+ if (WSAStartup(MAKEWORD(2,2), &wsaData) != NO_ERROR) {
|
|
|
+ printf("\r\nCannot init WinSock\a\r\n");
|
|
|
+ return 1;
|
|
|
+ }
|
|
|
+
|
|
|
+ lpHost = gethostbyname(HOST);
|
|
|
+ if (lpHost == NULL) {
|
|
|
+ printf("\r\nCannot find host %s\a\r\n",HOST);
|
|
|
+ return 2;
|
|
|
+ }
|
|
|
+
|
|
|
+ target.sin_family = AF_INET;
|
|
|
+ target.sin_addr.s_addr = *((u_long FAR *) (lpHost->h_addr));
|
|
|
+ target.sin_port = htons(PORT);
|
|
|
+
|
|
|
+ GetProcessAffinityMask(GetCurrentProcess(), &availableCPUs, &totalCPUs);
|
|
|
+ printf("CPUs (bit masks): process=%x, system=%x\r\n", availableCPUs, totalCPUs);
|
|
|
+
|
|
|
+ InitializeCriticalSectionAndSpinCount(&cs, 100);
|
|
|
+
|
|
|
+ /* Do the actual test here */
|
|
|
+ if (CLIENTCOUNT>0) {
|
|
|
+ MultiClientTestAutomatic(2000);
|
|
|
+ } else {
|
|
|
+ SingleClientTestAutomatic(2000);
|
|
|
+ }
|
|
|
+
|
|
|
+ /* Cleanup */
|
|
|
+ DeleteCriticalSection(&cs);
|
|
|
+ WSACleanup();
|
|
|
+ return 0;
|
|
|
+}
|
bel