mirror of
https://github.com/lammertb/libhttp.git
synced 2025-08-17 22:21:06 +03:00
353 lines
8.7 KiB
C
353 lines
8.7 KiB
C
#include <WinSock2.h>
|
|
#include <stdio.h>
|
|
#include <stdarg.h>
|
|
#include <time.h>
|
|
|
|
char * HOST = "127.0.0.1";
|
|
unsigned short PORT = 8080;
|
|
static const char * RESOURCELIST[] = {
|
|
"/hello.txt",
|
|
"/imagetest/00.png",
|
|
"/"
|
|
};
|
|
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;
|
|
unsigned long keep_alive = 1;
|
|
|
|
|
|
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;
|
|
}
|
|
|
|
for (i=0; i<((keep_alive>0)?keep_alive:1); i++) {
|
|
// 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\n", HOST);
|
|
} else {
|
|
sockprintf(soc, "%s %s HTTP/1.1\r\nHost: %s\r\nConnection: Close\r\n", method, resource, HOST);
|
|
}
|
|
if (keep_alive) {
|
|
sockprintf(soc, "Connection: Keep-Alive\r\n");
|
|
} else {
|
|
sockprintf(soc, "Connection: Close\r\n");
|
|
}
|
|
|
|
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;
|
|
}
|
|
|
|
if (!keep_alive) {
|
|
shutdown(soc, SD_SEND);
|
|
} else {
|
|
timeOut = 2;
|
|
}
|
|
|
|
// wait for response from the server
|
|
totalData = 0;
|
|
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;
|
|
}
|
|
}
|
|
|
|
if (keep_alive) {
|
|
Sleep(1000);
|
|
}
|
|
|
|
}
|
|
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;
|
|
}
|