1,658
社区成员




#include <Winsock2.h>
#include <windows.h>
#include <stdio.h>
#include <conio.h>
#include <vector>
using namespace std;
#pragma comment(lib, "Ws2_32.lib")
BOOL WINAPI HandlerRoutine(DWORD Event)
{
switch(Event)
{
case CTRL_C_EVENT:
case CTRL_BREAK_EVENT:
case CTRL_CLOSE_EVENT:
case CTRL_LOGOFF_EVENT:
case CTRL_SHUTDOWN_EVENT:
exit(0);
}
return TRUE;
}
void main()
{
WSADATA data;
WSAStartup(MAKEWORD( 2, 2 ), &data);
SOCKET sock = socket(AF_INET, SOCK_STREAM, 0);
if( sock == INVALID_SOCKET )
{
printf("Failed to create SOCKET!\n");
return;
}
sockaddr_in addr;
addr.sin_port = htons(80);
addr.sin_family = AF_INET;
addr.sin_addr.s_addr = htonl(INADDR_ANY);
if( SOCKET_ERROR == bind(sock, (const sockaddr*)&addr, sizeof(sockaddr_in)) )
{
printf("The socket binding port failed!\n");
return;
}
if( SOCKET_ERROR == listen(sock, SOMAXCONN) )
{
printf("failed to listen socket!\n");
return;
}
printf("http server is running!\n");
while(1)
{
sockaddr_in addr_client;
int nSize = sizeof(sockaddr_in);
SOCKET socket_client = accept(sock, (sockaddr*)&addr_client, &nSize);
if( INVALID_SOCKET == socket_client )
{
Sleep(100);
continue;
}
vector<char> vecBuffer;
while(1)
{
bool bIsRead = false;
char szBufer[128] = {0};
int nRead = recv(socket_client, szBufer, sizeof(szBufer), 0);
if( SOCKET_ERROR == nRead )
{
printf("Socket failed to receive data! error code = %d\n", WSAGetLastError());
}
if( strlen(szBufer) > 0 )
{
vecBuffer.reserve(vecBuffer.size() + strlen(szBufer));
vecBuffer.insert(vecBuffer.end(), &szBufer[0], &szBufer[0] + nRead);
}
if( strstr(szBufer, "\r\n\r\n") )
{
printf("Has received the HTTP head!\n");
break;
}
}
vecBuffer.reserve(vecBuffer.size()+1);
vecBuffer.insert(vecBuffer.end(), '\0', 1);
char* pHttpBuffer = &vecBuffer[0];
printf("recv data -----------------------------\n%s\n", &vecBuffer[0]);
if( strnicmp(pHttpBuffer, "GET ", 4) == 0 )
{
char szResponse[512] = {0};
sprintf(szResponse, "HTTP/1.1 200 OK\r\nContent-Type: text/plain\r\nContent-Length: %d\r\n\r\nhello world!",
strlen("hello world!"));
if( SOCKET_ERROR == send(socket_client, szResponse, strlen(szResponse), 0) )
{
printf("failed to response data! error code = %d\n", WSAGetLastError());
}
}
shutdown(socket_client, SD_BOTH);
closesocket(socket_client);
}
WSACleanup();
}