在本机上使用TCP实现简易的进程通信。
服务器端:
#define WIN32_LEAN_AND_MEAN
#define _WINSOCK_DEPRECATED_NO_WARNINGS
#define _CRT_SECURE_NO_WARNINGS
#include <windows.h>
#include <WinSock2.h>
#include <stdio.h>
#include <map>
#include <string>
#include <vector>
using namespace std;
string user[2] = { "MorganStark", "Thor" };
string pwd[2] = { "123456", "sonofodin" };
string chat[2] = {"Sir, I am here.", "Mr.Stark is gone." };
vector<string> loginusr;
enum cmd
{
cmd_login,
cmd_logout,
cmd_chat,
cmd_error
};
//数据包头
struct DataHeader
{
short datalength;
short cmd;
};
struct Login
{
char username[32];
char password[32];
};
struct Loginresult
{
int result;
};
struct Logout
{
char username[32];
};
struct Logoutresult
{
int result;
};
struct Message
{
char message[256];
};
struct Messageret
{
char messageret[256];
};
int main()
{
WORD ver = MAKEWORD(2, 2);
WSADATA dat;
WSAStartup(ver, &dat);
//用socket API建立简易的服务器端
//1.建立一个socket
SOCKET _sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
//2.bind 绑定用于接受客户端连接的网络端口号
sockaddr_in _sin = {};
_sin.sin_family = AF_INET; //用于网络连接的类型,IPv4,与上面定义的一样
_sin.sin_port = htons(4567); //host to net unsigned short,这里要转换一下端口号
_sin.sin_addr.S_un.S_addr = inet_addr("127.0.0.1"); //服务器程序绑定的IP地址,纯内网的应用程序可以用127.0.0.1
//_sin.sin_addr.S_un.S_addr = INADDR_ANY; //随意什么地址都可以访问
//bind是有可能失败的,端口被占用的时候,所以要判断是否成功
int bindret = bind(_sock, (sockaddr*)&_sin, sizeof(sockaddr_in));
if (SOCKET_ERROR == bindret)
{
printf("ERROR! bind failed !\n");
getchar();
return 1;
}
//3.listen 监听网络端口
//同样的监听也可能失败的,需要加以判断
int listenret = listen(_sock, 3);
if (SOCKET_ERROR == listenret)
{
printf("ERROR! listen failed !\n");
getchar();
return 1;
}
printf("* * * * * * * * * * * * * * * * * * * * * * * * * * * *\n");
printf("* *\n");
printf("* Welcome to the first Socket Sever *\n");
printf("* *\n");
printf("* * * * * * * * * * * * * * * * * * * * * * * * * * * *\n");
//4.accept 等待接受客户端连接
sockaddr_in clientAddr = {}; //远程客户端的地址
int nAddrlen = sizeof(sockaddr_in);
SOCKET _clientSock = INVALID_SOCKET; //接受的客户端是无效的,大规模程序的时候有可能
_clientSock = accept(_sock, (sockaddr*)&clientAddr, &nAddrlen);
if (INVALID_SOCKET == _clientSock)
{
printf("ERROR!, INVALID CLIENT!\n");
}
printf("new Client connected :socket = %d IP = %s \n",(int)_clientSock, inet_ntoa(clientAddr.sin_addr));
//5. send 向客户端发送数据
while (1)
{
//char _recvbuf[1024] = {}; 不是用简单的char了,用结构体
DataHeader header = {};
//5.1. recv 接收客户端发送的数据
int nlen = recv(_clientSock, (char*)&header, sizeof(DataHeader), 0);
if (nlen <= 0)
{
continue;
}
printf("receive cmd : %d , datalength is : %d .\n", header.cmd, header.datalength);
//5.2 send 处理请求, 向客户端发送数据, 用switch判断命令
switch (header.cmd)
{
case cmd_login:
{
Login login = {};
//接受客户端输入的账号密码
recv(_clientSock, (char*)&login, sizeof(Login), 0);
Loginresult result;
result.result = 0;
//防止重复登录
if (0 != loginusr.size())
{
for (int i = 0; i < (int)loginusr.size(); i++)
{
if (loginusr[i] == string(login.username))
{
result.result = 2;
}
}
if (2 == result.result)
{
//先发消息头
send(_clientSock, (char*)&header, sizeof(DataHeader), 0);
//再发消息体
send(_clientSock, (char*)&result, sizeof(Loginresult), 0);
break;
}
}
//判断用户名与密码是否正确的过程
for (int i = 0; i < 2; i++)
{
if (user[i] == string(login.username))
{
if (pwd[i] == string(login.password))
{
loginusr.push_back(user[i]);
printf(" it is %d users connect now .\n", loginusr.size());
result.result = 1;
//先发消息头
send(_clientSock, (char*)&header, sizeof(DataHeader), 0);
//再发消息体
send(_clientSock, (char*)&result, sizeof(Loginresult), 0);
break;
}
else
{
result.result = 0;
//先发消息头
send(_clientSock, (char*)&header, sizeof(DataHeader), 0);
//再发消息体
send(_clientSock, (char*)&result, sizeof(Loginresult), 0);
break;
}
}
}
}
break;
case cmd_logout:
{
Logout logout = {};
recv(_clientSock, (char*)&logout, sizeof(Logout), 0);
Logoutresult result;
result.result = 0;
//判断用户名是否正确,且用户在已登录的名单中
for (int i = 0; i < (int)loginusr.size(); i++)
{
if (loginusr[i] == string(logout.username))
{
result.result = 1;
printf("%s logout !\n", logout.username);
loginusr.erase(loginusr.begin() + i);
printf(" it is %d users connect now .\n", loginusr.size());
break;
}
}
//先发消息头
send(_clientSock, (char*)&header, sizeof(DataHeader), 0);
//再发消息体
send(_clientSock, (char*)&result, sizeof(Logoutresult), 0);
}
break;
case cmd_chat:
{
Message message = {};
recv(_clientSock, (char*)&message, sizeof(Message), 0);
Messageret messageret;
if (0 == strcmp(message.message, "Friday"))
{
strcpy(messageret.messageret, chat[0].c_str());
//先发消息头
send(_clientSock, (char*)&header, sizeof(DataHeader), 0);
//再发消息体
send(_clientSock, (char*)&messageret, sizeof(Messageret), 0);
break;
}
else if(0 == strcmp(message.message, "Whereismydad?"))
{
strcpy(messageret.messageret, chat[1].c_str());
//先发消息头
send(_clientSock, (char*)&header, sizeof(DataHeader), 0);
//再发消息体
send(_clientSock, (char*)&messageret, sizeof(Messageret), 0);
break;
}
else
{
strcpy(messageret.messageret, "May I help you?");
//先发消息头
send(_clientSock, (char*)&header, sizeof(DataHeader), 0);
//再发消息体
send(_clientSock, (char*)&messageret, sizeof(Messageret), 0);
break;
}
}
break;
default:
{
header.cmd = cmd_error;
header.datalength = 0;
send(_clientSock, (char*)&header, sizeof(DataHeader), 0);
}
break;
}
}
//6.close 关闭socket
closesocket(_sock);
WSACleanup();
//printf("Client exit !, Sever exit ! \n");
getchar();
return 0;
}
客户端:
#define WIN32_LEAN_AND_MEAN
#define _WINSOCK_DEPRECATED_NO_WARNINGS
#define _CRT_SECURE_NO_WARNINGS
//或者把winsock2放在Windows前面
#include <Windows.h>
#include <WinSock2.h>
#include <stdio.h>
//#pragma comment(lib, "ws2_32.lib") //调用了Windows的库,要加载这个静态链接库才能使用
//Windows下才可以这样写,Linux下不可以这样写
enum cmd
{
cmd_login,
cmd_logout,
cmd_chat,
cmd_error
};
//数据包头
struct DataHeader
{
short datalength;
short cmd;
};
struct Login
{
char username[32];
char password[32];
};
struct Loginresult
{
int result;
};
struct Logout
{
char username[32];
};
struct Logoutresult
{
int result;
};
struct Message
{
char message[256];
};
struct Messageret
{
char messageret[256];
};
int main()
{
WORD ver = MAKEWORD(2, 2); //版本号
WSADATA dat;
WSAStartup(ver, &dat);
//1. socket 建立一个socket
SOCKET _sock = socket(AF_INET, SOCK_STREAM, 0);
if (INVALID_SOCKET == _sock)
{
printf("ERROR! build socket failed !\n");
getchar();
return 1;
}
//2.connect 连接服务器
sockaddr_in _sin = {};
_sin.sin_family = AF_INET;
_sin.sin_port = htons(4567);
_sin.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
int ret = connect(_sock, (sockaddr*)&_sin, sizeof(sockaddr_in));
if (SOCKET_ERROR == ret)
{
printf("ERROR! connect sever failed !\n");
getchar();
return 1;
}
printf("Successfully connect to the Sever, Please input your cmd.\n");
while (1)
{
//3. send 发送数据到服务器
char msgbuf[1024] = {};
scanf("%s", msgbuf);
if (0 == strcmp(msgbuf, "exit"))
{
printf("exit client !\n");
break;
}
else if (0 == strcmp(msgbuf, "login"))
{
char username[32] = {};
char password[32] = {};
printf("input your username:\n");
scanf("%s", username);
printf("input the password:\n");
scanf("%s", password);
Login login;
strcpy(login.username, username);
strcpy(login.password, password);
DataHeader head = { sizeof(login), cmd_login };
//向服务器发送包头
send(_sock, (const char*)&head, sizeof(DataHeader), 0);
//向服务器发送包体
send(_sock, (const char*)&login, sizeof(Login), 0);
//接受服务器返回数据
DataHeader rethead = {};
Loginresult loginret = {};
recv(_sock, (char*)&rethead, sizeof(DataHeader), 0);
recv(_sock, (char*)&loginret, sizeof(Loginresult), 0);
//判断返回的结果
//printf("loginresult is : %d \n", loginret.result);
if (loginret.result == 1)
{
printf(" %s login successfully....\n", username);
printf("Can I help you? Sir.\n");
//登录之后才能进行的操作
while (1)
{
char msg[256] = {};
scanf("%s", msg);
if (0 == strcmp(msg, "exit"))
{
printf("you are welcome.\n");
break;
}
Message message;
strcpy(message.message, msg);
DataHeader hd = { sizeof(Message), cmd_chat };
//向服务器发送包头
send(_sock, (const char*)&hd, sizeof(DataHeader), 0);
//向服务器发送包体
send(_sock, (const char*)&message, sizeof(Message), 0);
DataHeader rethead = {};
Messageret messageret = {};
recv(_sock, (char*)&rethead, sizeof(DataHeader), 0);
recv(_sock, (char*)&messageret, sizeof(Messageret), 0);
//打印收到的消息
printf("Friday : %s\n", messageret.messageret);
}
}
else if(loginret.result == 2)
printf("the %s has already login !\n", username);
else
printf("username or password error, please check.\n");
}
else if (0 == strcmp(msgbuf, "logout"))
{
char username[32] = {};
printf("input the username you want to logout:\n");
scanf("%s", username);
Logout logout;
strcpy(logout.username, username);
DataHeader head = { sizeof(Logout), cmd_logout };
//向服务器发送包头
send(_sock, (const char*)&head, sizeof(DataHeader), 0);
//向服务器发送包体
send(_sock, (const char*)&logout, sizeof(Logout), 0);
//接受服务器返回数据
DataHeader rethead = {};
Logoutresult logoutret = {};
recv(_sock, (char*)&rethead, sizeof(DataHeader), 0);
recv(_sock, (char*)&logoutret, sizeof(Logoutresult), 0);
//判断返回的结果
//printf("%d\n", logoutret.result);
if(logoutret.result == 1)
printf("%s logout successful !\n", username);
else
printf("%s logout failed ! %s may not login or it is not the legal user.\n", username, username);
}
else
{
printf("Unsupportable cmd, request end.\n");
}
//send(_sock, msgbuf, strlen(msgbuf) + 1, 0);
4.recv 接受服务器信息
//char recvBuf[1024] = {};
//int nlen = recv(_sock, recvBuf, 1024, 0);
//if (nlen > 0)
//{
// printf("receive message from Sever.....\n");
// printf("%s\n", recvBuf);
//}
}
//5.close 关闭socket
closesocket(_sock);
WSACleanup();
printf("Client exit !\n");
getchar();
return 0;
}