Socket编程-1V1聊天
==> 学习汇总（持续更新）
==> 从零搭建后端基础设施系列（一）-- 背景介绍
提供一个1V1聊天的小例子供大家参考.直接贴代码,废话不多说.
注:在VS2015代码测试成功,其它编译器如有问题自行解决.
先来一张效果图:
服务端:
#include <WinSock2.h>
#include <stdio.h>
#include <stdlib.h>
#pragma comment(lib,"ws2_32.lib")
int main()
{
//存放套接字信息的结构
WSADATA wsaData = { 0 };
SOCKET ServerSocket = INVALID_SOCKET;//服务端套接字
SOCKET ClientSocket = INVALID_SOCKET;//客户端套接字
SOCKADDR_IN ServerAddr = { 0 };//服务端地址
SOCKADDR_IN ClientAddr = { 0 };//客户端地址
int iClientAddrLen = sizeof(ClientAddr);
USHORT uPort = 18000;//服务器监听端口
//初始化套接字
if (WSAStartup(MAKEWORD(2, 2), &wsaData))
{
printf("WSAStartup failed with error code: %d\n", WSAGetLastError());
return -1;
}
//判断版本
if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2)
{
printf("wVersion was not 2.2\n");
return -1;
}
//创建套接字
ServerSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (ServerSocket == INVALID_SOCKET)
{
printf("socket failed with error code: %d\n", WSAGetLastError());
return -1;
}
//设置服务器地址
ServerAddr.sin_family = AF_INET;//连接方式
ServerAddr.sin_port = htons(uPort);//服务器监听端口
ServerAddr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);//任何客户端都能连接这个服务器
//绑定服务器
if (SOCKET_ERROR == bind(ServerSocket, (SOCKADDR*)&ServerAddr, sizeof(ServerAddr)))
{
printf("bind failed with error code: %d\n", WSAGetLastError());
closesocket(ServerSocket);
return -1;
}
//监听有无客户端连接
if (SOCKET_ERROR == listen(ServerSocket, 1))
{
printf("listen failed with error code: %d\n", WSAGetLastError());
closesocket(ServerSocket);
WSACleanup();
return -1;
}
//输入用户名
printf("Please input your name:");
char name[32] = { 0 };
gets_s(name);
printf("waiting connect.....\n");
//如果有客户端申请连接就接受连接
ClientSocket = accept(ServerSocket, (SOCKADDR*)&ClientAddr, &iClientAddrLen);
if (ClientSocket == INVALID_SOCKET)
{
printf("accept failed with error code: %d\n", WSAGetLastError());
closesocket(ServerSocket);
WSACleanup();
return -1;
}
printf("Successfuuly got a connection from IP:%s Port:%d\n\n\n\n",
inet_ntoa(ClientAddr.sin_addr),htons(ClientAddr.sin_port));
char buffer[4096] = {0};
int iRecvLen = 0;
int iSendLen = 0;
//发送用户名给对方
iSendLen = send(ClientSocket, name, strlen(name), 0);
if (SOCKET_ERROR == iSendLen)
{
printf("send failed with error code: %d\n", WSAGetLastError());
closesocket(ServerSocket);
closesocket(ClientSocket);
WSACleanup();
return -1;
}
//接收对方的用户名
char nameOther[32] = { 0 };
iRecvLen = recv(ClientSocket, nameOther, sizeof(nameOther), 0);
if (SOCKET_ERROR == iRecvLen)
{
printf("recv failed with error code: %d\n", WSAGetLastError());
closesocket(ServerSocket);
closesocket(ClientSocket);
WSACleanup();
return -1;
}
strcat(nameOther, "\0");
//一直接收客户端发来的消息 并且把消息发送回去
while (1)
{
memset(buffer, 0, sizeof(buffer));
//接收客户端消息
iRecvLen = recv(ClientSocket, buffer, sizeof(buffer), 0);
if (SOCKET_ERROR == iRecvLen)
{
printf("recv failed with error code: %d\n", WSAGetLastError());
closesocket(ServerSocket);
closesocket(ClientSocket);
WSACleanup();
return -1;
}
//printf("recv %d bytes from %s: ", iRecvLen, nameOther);
strcat(buffer, "\0"); //加一个截止符,因为接收到的数据没有这个
printf("%s: %s\n", nameOther, buffer);
//发送消息给客户端
memset(buffer, 0, sizeof(buffer));
printf("%s: ",name);
gets_s(buffer);
if (strcmp(buffer, "bye") == 0) break;
iSendLen = send(ClientSocket, buffer, strlen(buffer), 0);
if (SOCKET_ERROR == iSendLen)
{
printf("send failed with error code: %d\n", WSAGetLastError());
closesocket(ServerSocket);
closesocket(ClientSocket);
WSACleanup();
return -1;
}
//printf("send %d bytes to %s\n", iSendLen, nameOther);//为了美观最好不要打印这个了
}
closesocket(ClientSocket);
closesocket(ServerSocket);
WSACleanup();
system("pause");
return 0;
}
客户端:
#include <WinSock2.h>
#include <stdio.h>
#include <stdlib.h>
#pragma comment(lib,"ws2_32.lib")
int main()
{
WSADATA wsaData = { 0 };//存放套接字信息
SOCKET ClientSocket = INVALID_SOCKET;//客户端套接字
SOCKADDR_IN ServerAddr = { 0 };//服务端地址
USHORT uPort = 18000;//服务端端口
//初始化套接字
if (WSAStartup(MAKEWORD(2, 2), &wsaData))
{
printf("WSAStartup failed with error code: %d\n", WSAGetLastError());
return -1;
}
//判断套接字版本
if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2)
{
printf("wVersion was not 2.2\n");
return -1;
}
//创建套接字
ClientSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (ClientSocket == INVALID_SOCKET)
{
printf("socket failed with error code: %d\n", WSAGetLastError());
return -1;
}
//输入服务器IP
printf("Please input server IP:");
char IP[32] = { 0 };
gets_s(IP);
//输入聊天的用户名
printf("Please input your name:");
char name[32] = { 0 };
gets_s(name);
//设置服务器地址
ServerAddr.sin_family = AF_INET;
ServerAddr.sin_port = htons(uPort);//服务器端口
ServerAddr.sin_addr.S_un.S_addr = inet_addr(IP);//服务器地址
printf("connecting......\n");
//连接服务器
if (SOCKET_ERROR == connect(ClientSocket, (SOCKADDR*)&ServerAddr, sizeof(ServerAddr)))
{
printf("connect failed with error code: %d\n", WSAGetLastError());
closesocket(ClientSocket);
WSACleanup();
return -1;
}
printf("connecting server successfully IP:%s Port:%d\n\n\n\n",
inet_ntoa(ServerAddr.sin_addr),htons(ServerAddr.sin_port));
char buffer[4096] = {0};
int iRecvLen = 0;
int iSnedLen = 0;
//发送用户名给对方
iSnedLen = send(ClientSocket, name, strlen(name), 0);
if (SOCKET_ERROR == iSnedLen)
{
printf("send failed with error code: %d\n", WSAGetLastError());
closesocket(ClientSocket);
WSACleanup();
return -1;
}
//接收对方的用户名
char nameOther[32] = { 0 };
iRecvLen = recv(ClientSocket, nameOther, sizeof(nameOther), 0);
if (SOCKET_ERROR == iRecvLen)
{
printf("send failed with error code: %d\n", WSAGetLastError());
closesocket(ClientSocket);
WSACleanup();
return -1;
}
strcat(nameOther, "\0");
//发送并接收消息
while (1)
{
memset(buffer, 0, sizeof(buffer));
//发送消息
printf("%s: ", name);
gets_s(buffer);
if(strcmp(buffer,"bye") == 0) break;
iSnedLen = send(ClientSocket, buffer, strlen(buffer), 0);
if (SOCKET_ERROR == iSnedLen)
{
printf("send failed with error code: %d\n", WSAGetLastError());
closesocket(ClientSocket);
WSACleanup();
return -1;
}
//printf("send %d bytes to %s\n", iSnedLen, nameOther);//为了美观最好不要打印这个了
memset(buffer, 0, sizeof(buffer));
//接收消息
iRecvLen = recv(ClientSocket, buffer, sizeof(buffer), 0);
if (SOCKET_ERROR == iRecvLen)
{
printf("send failed with error code: %d\n", WSAGetLastError());
closesocket(ClientSocket);
WSACleanup();
return -1;
}
//printf("recv %d bytes from %s: ", iRecvLen, nameOther);//为了美观最好不要打印这个了
strcat(buffer, "\0");
printf("%s: %s\n", nameOther,buffer);
}
closesocket(ClientSocket);
WSACleanup();
system("pause");
return 0;
}