TCP.普通socket(阻塞/非阻塞)模型

在此介绍window下的TCP普通socket编程.

1.服务器端

// CommonSocket.cpp : Defines the entry point for the console application.

//

#include "stdafx.h"

#include "winsock2.h"

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

int _tmain(int argc, _TCHAR* argv[])

{

// 初始化Windows Socket 2.2

WSADATA wsaData;

int iResult = WSAStartup(MAKEWORD(2,2), &wsaData);

if (iResult != NO_ERROR) 

{

wprintf(L"WSAStartup failed with error: %ld\n", iResult);

getchar();

return 1;

}

// 创建一个新的Socket来响应客户端的连接请求

SOCKET ListeningSocket;

ListeningSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

if (ListeningSocket == INVALID_SOCKET) {

wprintf(L"Socket failed with error: %ld\n", WSAGetLastError());

WSACleanup();

getchar();

return 1;

}

//设置阻塞方式连接

    unsigned long ul = 0;//1为非阻塞,0为阻塞

    iResult = ioctlsocket(ListeningSocket, FIONBIO, (unsigned long*)&ul);

    if (iResult == SOCKET_ERROR)

{

wprintf(L"Set block or not failed with error: %ld\n", WSAGetLastError());

WSACleanup();

getchar();

return 1;

}

// 填写服务器地址信息

SOCKADDR_IN ServerAddr;

int Port = 5150;// 端口为5150

ServerAddr.sin_family = AF_INET;//协议族TCP/IP

ServerAddr.sin_addr.s_addr = inet_addr("127.0.0.1");

ServerAddr.sin_port = htons(Port);//本机监听端口

ServerAddr.sin_addr.s_addr = htonl(INADDR_ANY);//客户端口

// 绑定监听端口 

iResult = bind(ListeningSocket, (SOCKADDR *)&ServerAddr, sizeof(ServerAddr)); 

if (iResult == SOCKET_ERROR) 

{

wprintf(L"bind failed with error: %ld\n", WSAGetLastError());

closesocket(ListeningSocket);

WSACleanup();

getchar();

return 1;

}

// 开始监听,指定最大同时连接数为5

iResult = listen(ListeningSocket, 5);

if (iResult == SOCKET_ERROR) 

{

wprintf(L"listen failed with error: %ld\n", WSAGetLastError());

closesocket(ListeningSocket);

closesocket(ListeningSocket);

WSACleanup();

getchar();

return 1;

}

// 等待客户端连金额

wprintf(L"Waiting for client to connect...\n");

// 接受新的连接

SOCKET NewConnection;

NewConnection = accept(ListeningSocket, NULL, NULL);// 阻塞模式下，程序阻塞在此,直到有连接

if (NewConnection == INVALID_SOCKET)// 为非阻塞的时候，不能如此检测是否有效,要用:WSAEWOULDBLOCK

{

wprintf(L"accept failed with error: %ld\n", WSAGetLastError());

closesocket(NewConnection);

closesocket(ListeningSocket);

WSACleanup();

getchar();

return 1;

} else

{

wprintf(L"Client connected.\n");

}

//获取消息

int i = 0;

const int bufsize = 100;

char buffer[bufsize + 1] = {0};

while(1)  

    {  

        i = 0;

        if ((i = recv(NewConnection,buffer,100,0)) == SOCKET_ERROR)  

        {  

            wprintf(L"Get client message error\n");  

            closesocket(NewConnection);  

closesocket(ListeningSocket);

WSACleanup();

            getchar();  

            return 0;  

        }  

if(i == 0)

{

break;

}

        buffer[i] = '\0';  

        printf("%s\n",buffer);  

    }

//发送消息

const char* stranswer = "Hi, I have received your message.\n";

iResult = send(NewConnection, stranswer, strlen(stranswer), 0);

if(iResult == SOCKET_ERROR)

{

wprintf(L"Answer message error\n");  

closesocket(NewConnection);  

closesocket(ListeningSocket);

WSACleanup();

getchar();  

return 0;  

}

// 并关闭监听Socket,然后退出应用程序

closesocket(NewConnection);

closesocket(ListeningSocket);

// 释放Windows Socket DLL的相关资源

WSACleanup();

getchar();

return 0;

}

2.客户端

// CommonSocketClient.cpp : Defines the entry point for the console application.

//

#include "stdafx.h"

#include "winsock2.h"

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

int _tmain(int argc, _TCHAR* argv[])

{

// 初始化Windows Socket 2.2

WSADATA wsaData;

int iResult = WSAStartup(MAKEWORD(2,2), &wsaData);

if (iResult != NO_ERROR) 

{

wprintf(L"WSAStartup failed with error: %ld\n", iResult);

getchar();

return 1;

}

// 创建一个新的Socket来连接服务器

SOCKET s;

s = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

if (s == INVALID_SOCKET) {

wprintf(L"Socket failed with error: %ld\n", WSAGetLastError());

closesocket(s);

WSACleanup();

getchar();

return 1;

}

// 填写服务器地址信息

SOCKADDR_IN ServerAddr;

int Port = 5150;

ServerAddr.sin_family = AF_INET;

ServerAddr.sin_port = htons(Port);// 端口为5150

ServerAddr.sin_addr.s_addr = inet_addr("127.0.0.1");

// 向服务器发出连接请求

iResult = connect(s, (SOCKADDR *) &ServerAddr, sizeof(ServerAddr));

if (iResult != NO_ERROR) 

{

wprintf(L"Connect server error: %ld\n", iResult);

closesocket(s);

WSACleanup();

getchar();

return 1;

}

// 发送消息C

const char* stranswer = "Hi, server.\n";

iResult = send(s, stranswer, strlen(stranswer), 0);

if(iResult == SOCKET_ERROR)

{

wprintf(L"Answer message error\n");  

closesocket(s);

WSACleanup();

getchar();  

return 0;  

}

wprintf(L"Send message OK...\n");

// 关闭客户端Socket,然后退出应用程序

closesocket(s);

// 释放Windows Socket DLL的相关资源

WSACleanup();

getchar();

return 0;

}