網絡報文的數據格式定義及使用3

客戶端，包頭跟包體一起發送。

服務端：

//vc的第一套socket，第二套socket，可能存在衝突

#define WIN32_LEAN_AND_MEAN
#define  _WINSOCK_DEPRECATED_NO_WARNINGS //也可以放到工程屬性，預處理中

#include <windows.h>
#include <WinSock2.h>
#include <stdio.h>

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

enum CMD
{
    CMD_LOGIN,
    CMD_LOGIN_RET,
    CMD_LOGOUT,
    CMD_LOGOUT_RET,
    CMD_ERROR
};


struct DataHeader 
{
    int dataLength;
    CMD    cmd;
};

struct Login :public DataHeader
{
    Login()
    {
        dataLength = sizeof(Login);
        cmd = CMD_LOGIN;
    }
    char userName[32];
    char passWord[32];
};

struct LoginResult :public DataHeader
{
    LoginResult()
    {
        dataLength = sizeof(LoginResult);
        cmd = CMD_LOGIN_RET;
        ret = 0;
    }
    int ret;
};

struct LogOut :public DataHeader
{
    LogOut()
    {
        dataLength = sizeof(LogOut);
        cmd = CMD_LOGOUT;
    }
    char userName[32];
};

struct LogOutResult :public DataHeader
{
    LogOutResult()
    {
        dataLength = sizeof(LogOutResult);
        cmd = CMD_LOGOUT_RET;
        ret = 0;
    }
    int ret;
};

int main()
{
    //啓動Windows socket 2.x環境
    WORD ver = MAKEWORD(2, 2);
    WSADATA dat;
    WSAStartup(ver, &dat);

    //--用Socket API建立簡易TCP服務端
    //1.建立一個socket
    SOCKET _sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    //2 bind綁定用於接受客戶端連接的網絡端口
    sockaddr_in _sin = {};
    _sin.sin_family = AF_INET;
    _sin.sin_port = htons(4567);//host to net unsigned
    _sin.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
    //使用127.0.0.1可以防止外網訪問
    //啓用本機全部的ip地址可以使用，INADDR_ANY
    if (SOCKET_ERROR == bind(_sock, (sockaddr*)&_sin, sizeof(sockaddr_in)))
    {
        printf("錯誤，綁定網絡端口失敗...\n");
    }
    else
    {
        printf("綁定網絡端口成功...\n");
    }
    //3 listen 監聽網絡端口
    if (SOCKET_ERROR == listen(_sock, 5))
    {
        printf("錯誤，監聽網絡端口失敗...\n");
    }
    else
    {
        printf("監聽網絡端口成功...\n");
    }
    //4 accept等待接受客戶端連接
    sockaddr_in clientAddr = {};
    int nAddrLen = sizeof(clientAddr);
    SOCKET _cSock = INVALID_SOCKET;
    

    _cSock = accept(_sock, (sockaddr*)&clientAddr, &nAddrLen);
    if (INVALID_SOCKET == _cSock)
    {
        printf("錯誤，接受到無效客戶端socket...\n");
    }
    printf("新客戶端加入:IP=%s\n", inet_ntoa(clientAddr.sin_addr));


    while (true)
    {
        //5 接收客戶端數據
        DataHeader header = {};

        int nLen = recv(_cSock, (char*)&header, sizeof(DataHeader), 0);
        if (nLen < 0)
        {
            printf("客戶端已退出，任務結束。\n");
            break;
        }

        switch (header.cmd)
        {
        case CMD_LOGIN:
        {
            Login login = {};
            recv(_cSock, (char*)&login + sizeof(DataHeader), sizeof(Login) - sizeof(DataHeader), 0);
            printf("cmd logint,user:%s,password:%s\n", login.userName, login.passWord);
            //----------------------------
            //----------------------------
            LoginResult longinRet = {};
            send(_cSock, (const char*)&longinRet, sizeof(LoginResult), 0);
        }
        break;
        case CMD_LOGOUT:
        {
            LogOut log_out = {};
            recv(_cSock, (char*)&log_out+sizeof(DataHeader), sizeof(LogOut)- sizeof(DataHeader), 0);
            printf("cmd logout,user:%s\n", log_out.userName);
            //----------------------------
            LogOutResult logOutRet = {};
            send(_cSock, (const char*)&logOutRet, sizeof(LogOutResult), 0);
        }
        break;
        default:
        {
            //剩餘的垃圾數據是否要讀取出來？
            if (header.dataLength >0 )
            {
                char buf[1024];
                recv(_cSock, buf, header.dataLength, 0);
            }            

            DataHeader header = {};
            header.cmd = CMD_ERROR;
            header.dataLength = 0;
            send(_cSock, (const char*)&header, sizeof(DataHeader), 0);
        }
        break;    
        }

    }

    //6 關閉套接字closesocket
    closesocket(_sock);


    //清除Windows socket環境
    WSACleanup();
    printf("已退出。\n");
    getchar();
    return 0;
}

 

客戶端：

#define WIN32_LEAN_AND_MEAN
#define _WINSOCK_DEPRECATED_NO_WARNINGS //也可以放到工程屬性，預處理中
//#define _CRT_SECURE_NO_WARNINGS

#include <windows.h>
#include <WinSock2.h>
#include <stdio.h>


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

enum CMD
{
    CMD_LOGIN,
    CMD_LOGIN_RET,
    CMD_LOGOUT,
    CMD_LOGOUT_RET,
    CMD_ERROR
};

struct DataHeader
{
    int dataLength;
    CMD    cmd;
};

struct Login:public DataHeader
{
    Login()
    {
        dataLength = sizeof(Login);
        cmd = CMD_LOGIN;
    }
    char userName[32];
    char passWord[32];
};

struct LoginResult :public DataHeader
{
    LoginResult()
    {
        dataLength = sizeof(LoginResult);
        cmd = CMD_LOGIN_RET;
        ret = 1;
    }
    int ret;
};

struct LogOut :public DataHeader
{
    LogOut()
    {
        dataLength = sizeof(LogOut);
        cmd = CMD_LOGOUT;
    }
    char userName[32];
};

struct LogOutResult :public DataHeader
{
    LogOutResult()
    {
        dataLength = sizeof(LogOutResult);
        cmd = CMD_LOGOUT_RET;
        ret = 1;
    }
    int ret;
};

int main()
{

    //啓動Windows socket 2.x環境
    WORD ver = MAKEWORD(2, 2);
    WSADATA dat;
    WSAStartup(ver, &dat);
    //---------------------------
    //--用Socket API建立簡易TCP客戶端
    //1 建立一個socket
    SOCKET _sock = INVALID_SOCKET;
    _sock = socket(AF_INET, SOCK_STREAM, 0);
    if (INVALID_SOCKET == _sock)
    {
        printf("錯誤，建立Socket失敗...\n");
    }
    else
    {
        printf("建立Socket成功...\n");
    }
    //2 連接服務器connect
    sockaddr_in _sin = {};
    _sin.sin_family = AF_INET;
    _sin.sin_port = htons(4567);//host to net unsigned
    _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("錯誤，連接服務端失敗...\n");


        closesocket(_sock);

        //清除Windows socket環境
        WSACleanup();
        printf("已退出。");
        getchar();
        return 0;
    }
    else
    {
        printf("連接服務端成功...\n");
    }

    
    while (true)
    {
        //3輸入請求命令
        char cmdBuf[128] = {};
        scanf_s("%s", cmdBuf, 128);
        getchar();
        //4處理請求
        if (0 == strcmp(cmdBuf, "exit"))
        {
            printf("收到exit命令，任務結束。\n");
            break;
        }
        else if(0 == strcmp(cmdBuf, "login"))
        {

            Login login = {};
            strcpy_s(login.userName, "jj");
            strcpy_s(login.passWord, "ww");
            send(_sock, (const char*)&login, sizeof(Login), 0);

            LoginResult cmd_ret = {};
            recv(_sock, (char*)&cmd_ret, sizeof(LoginResult), 0);
            printf("Login result:%d\n", cmd_ret.ret);
        }
        else if (0 == strcmp(cmdBuf, "logout"))
        {
            LogOut log_out = {};
            strcpy_s(log_out.userName, "jj");
            send(_sock, (const char*)&log_out, sizeof(LogOut), 0);

            LogOutResult cmd_ret = {};
            recv(_sock, (char*)&cmd_ret, sizeof(LogOutResult), 0);
            printf("LogOut result:%d\n", cmd_ret.ret);
        }
        else
        {
            printf("wrong cmd, please input cmd again.\n");
        }

    }



    //7 關閉套接字closesocket
    closesocket(_sock);

    //清除Windows socket環境
    WSACleanup();
    printf("已退出");
    getchar();
    return 0;
}