civetweb/testutils/testclient/testclient.c

#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;
}