Windows IO 性能简单测试

#include <windows.h>

#include <process.h>

#include<memory>

#pragma comment(lib,"ws2_32.lib")

ULONGLONG g_nReadCounts=0,g_nWriteCounts=0,g_nOtherCounts=0,g_nReadsBytes=0,g_nWriteBytes=0,g_nOtherBytes=0;

void PrintIODetails()

{

static DWORD lastTickCount;

static IO_COUNTERS lastIoCounters;

DWORD tickCount;

IO_COUNTERS ioCounters;

static int firstTime = 1;

tickCount = GetTickCount();

if (!GetProcessIoCounters(GetCurrentProcess(), &ioCounters)) {

return;

}

if (firstTime)

{

firstTime = 0;

printf("    Time    Reads   Writes   Others     Read(Bytes)    Write(Bytes)    Other(Bytes)\n");

printf("--------------------------------------------------------------------------------\n");

}

else

{

printf("%8lu%8llu%8llu%8llu%16llu%16llu%16llu\n",

tickCount - lastTickCount,

ioCounters.ReadOperationCount - lastIoCounters.ReadOperationCount,

ioCounters.WriteOperationCount - lastIoCounters.WriteOperationCount,

ioCounters.OtherOperationCount - lastIoCounters.OtherOperationCount,

ioCounters.ReadTransferCount - lastIoCounters.ReadTransferCount,

ioCounters.WriteTransferCount - lastIoCounters.WriteTransferCount,

ioCounters.OtherTransferCount - lastIoCounters.OtherTransferCount

);

g_nReadCounts += (ioCounters.ReadOperationCount - lastIoCounters.ReadOperationCount);

g_nWriteCounts += (ioCounters.WriteOperationCount - lastIoCounters.WriteOperationCount);

g_nOtherCounts += (ioCounters.OtherOperationCount - lastIoCounters.OtherOperationCount);

g_nReadsBytes += (ioCounters.ReadTransferCount - lastIoCounters.ReadTransferCount);

g_nWriteBytes += (ioCounters.WriteTransferCount - lastIoCounters.WriteTransferCount);

g_nOtherBytes += (ioCounters.OtherTransferCount - lastIoCounters.OtherTransferCount);

}

lastTickCount = tickCount;

memcpy(&lastIoCounters, &ioCounters, sizeof(IO_COUNTERS));

}

#define PACKCOUNT 10

#define BUFFSIEZ 50000

volatile bool bWorkThreadFinish = false;

void workerThread(void * dummy)

{

int i;

FILE * fp;

char buff[BUFFSIEZ];

WSADATA wsaData;

SOCKET s;

struct sockaddr_in localAddr;

struct sockaddr_in peerAddr;

memset(buff, 'X', sizeof(buff));

Sleep(2000);

// 写文件

fp = fopen("io.txt", "w");

if (fp) {

printf("fwrite = %d\n", fwrite(buff, 1, sizeof(buff), fp));

Sleep(10000);

// 关闭文件，测试Flush

printf("fclose\n");

fclose(fp);

Sleep(2000);

}

// 读文件

fp = fopen("io.txt", "r");

if (fp) {

printf("fread = %d\n", fread(buff, 1, sizeof(buff), fp));

fclose(fp);

Sleep(2000);

}

// 网络发

memset((void *)&localAddr, 0, sizeof(localAddr));

localAddr.sin_family = AF_INET;

localAddr.sin_addr.s_addr = htonl(INADDR_ANY);

localAddr.sin_port = htons(0);

memset((void *)&peerAddr, 0, sizeof(peerAddr));

peerAddr.sin_family = AF_INET;

peerAddr.sin_addr.s_addr = inet_addr("10.19.85.1");

peerAddr.sin_port = htons(3389);

printf("WSAStartup\n");

WSAStartup(MAKEWORD(2, 2), &wsaData);

Sleep(2000);

// UDP

printf("UDP Test:\n");

printf("socket\n");

s = socket(AF_INET, SOCK_DGRAM, 0);

Sleep(2000);

printf("bind\n");

bind(s, (struct sockaddr *)&localAddr, sizeof(localAddr));

Sleep(2000);

for (i = 0; i < PACKCOUNT; i++) {

printf("sendto = %d\n", sendto(s, buff, i * 100, 0, (SOCKADDR *) &peerAddr, sizeof(peerAddr)));

Sleep(2000);

}

printf("closesocket\n");

closesocket(s);

Sleep(2000);

// TCP

printf("TCP Test:\n");

printf("socket\n");

s = socket(AF_INET, SOCK_STREAM, 0);

Sleep(2000);

printf("bind\n");

bind(s, (struct sockaddr *)&localAddr, sizeof(localAddr));

Sleep(2000);

printf("connect\n");

connect(s, (SOCKADDR *) &peerAddr, sizeof(peerAddr));

Sleep(2000);

printf("send = %d\n", send(s, buff, sizeof(buff), 0));

Sleep(2000);

printf("closesocket\n");

closesocket(s);

Sleep(2000);

printf("WSACleanup\n");

WSACleanup();

bWorkThreadFinish = true;

}

UINT _stdcall ThreadProc(PVOID pv)

{

workerThread(pv);

return 1;

}

//HANDLE g_hEvt = NULL;

int main(int argc, char ** argv)

{

// 启动工作线程

::_beginthreadex(NULL,0,ThreadProc,NULL,0,0);

//g_hEvt = ::CreateEventW(NULL,FALSE,FALSE,NULL);

printf("Ctrl-C to exit\n\n");

for (;;) 

{

PrintIODetails();

Sleep(1000);

if (bWorkThreadFinish)

{

break;

}

}

printf("*********************************Total*************************************\n");

printf("    Reads   Writes   Others     Read(Bytes)    Write(Bytes)    Other(Bytes)\n");

printf("%8llu%8llu%8llu%16llu%16llu%16llu\n",

g_nReadCounts,

g_nWriteCounts,

g_nOtherCounts,

g_nReadsBytes,

g_nWriteBytes,

g_nOtherBytes

);

return 0;

}