C++简易TCP服务器

在本机上使用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;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值