c/s模型之tcp群聊

/服务端
// WSASever.cpp : 定义控制台应用程序的入口点。
//
#include "stdafx.h"
#include <WinSock2.h>
#include <Windows.h>
#include <vector>
#pragma comment (lib,"wSock32.lib")

SOCKET sockLink;
SOCKET g_psockSockLink[1024] = {0};    //存放客户端的sock
int g_nSocketNum=0;            //记录客户端sock的数目


//多线程进行接受和转发
DWORD WINAPI SeverThread(LPVOID lParam)
{
    int nErr = 0;
    char pSeverBuff[MAXBYTE] = { 0 };    //接受客户端的数据
    char pSendBuff[MAXBYTE] = { 0 };    //显示在窗口，包括来自哪个IP地址，端口号，数据
    
    SOCKET sockLink = (SOCKET)lParam;    //当前的客户端sock

    SOCKADDR_IN sockAddr;
    int len = sizeof(SOCKADDR_IN);

    while (TRUE)
    {
        //接受客户端
        nErr = recv(sockLink,pSeverBuff, MAXBYTE, 0);
        if (nErr == SOCKET_ERROR)
        {
            break;
            return -1;
        }
        
        //根据sock获取sock地址
        getpeername(sockLink, (sockaddr*)&sockAddr, &len);
        
        //将Ip、端口号、数据存入pSendBuff
        sprintf_s(pSendBuff,"%s(%d):%s
", inet_ntoa(sockAddr.sin_addr), ntohs(sockAddr.sin_port), pSeverBuff);
        
        //显示在窗口
        printf("%s
", pSendBuff);
        
        //转发
        for (int i = 0;i<g_nSocketNum;++i)
        {
            //不为当前发送方的sock
            if (g_psockSockLink[i] != sockLink)
            {
                send(g_psockSockLink[i], pSendBuff, MAXBYTE, 0);
            }
        }
            
    }
    //当客户端关闭时，服务端也随之关闭
    //if (nErr == INVALID_SOCKET)
        //return-1;
    return 0;
}



int _tmain(int argc, _TCHAR* argv[])
{
    //版本检测
    WORD wVersionRequested;
    WSADATA wsaData;
    int err;

    wVersionRequested = MAKEWORD(2, 2);

    err = WSAStartup(wVersionRequested, &wsaData);
    if (err != 0) {

        printf("WSAStartup failed with error: %d
", err);
        return 1;
    }

    if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2)
    {
        printf("Could not find a usable version of Winsock.dll
");
        WSACleanup();
        return 1;
    }
    else
        printf("The Winsock 2.2 dll was found okay
");

    //程序开始
    
    //创建socket->bind-》listen->accept->recv->send->closesocket

    SOCKET severSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    if (severSocket == INVALID_SOCKET)
    {
        printf("new socket error!");
    }

    //设置端口号和IP地址、协议。
    SOCKADDR_IN sockAddr;
    sockAddr.sin_port = htons(10086);
    sockAddr.sin_family = AF_INET;
    sockAddr.sin_addr.s_addr = htonl (INADDR_ANY);
        

    //IP地址表示方法
    /*方法1：m_addr.sin_addr.S_un.S_un_b.s_b1 = 192; 
           m_addr.sin_addr.S_un.S_un_b.s_b2 = 168; 
         m_addr.sin_addr.S_un.S_un_b.s_b3 = 0;
         m_addr.sin_addr.S_un.S_un_b.s_b4 = 1;
    方法2：  m_addr.sin_addr.S_un.S_un_w.s_w1 = (168 << 8) | 192; m_addr.sin_addr.S_un.S_un_w.s_w2 = (1 << 8) | 0;
    方法3：  m_addr.sin_addr.S_un.S_addr = (1 << 24) | (0 << 16) | (168 << 8) | 192
    方法4;    service.sin_addr.s_addr = inet_addr("127.0.0.1");
    */
    
    /*sockAddr.sin_addr.S_un.S_un_b.s_b1 = 127;
    sockAddr.sin_addr.S_un.S_un_b.s_b2 = 0;
    sockAddr.sin_addr.S_un.S_un_b.s_b3 = 0;
    sockAddr.sin_addr.S_un.S_un_b.s_b4 = 1;*/
    
    //绑定
    if (bind(severSocket, (sockaddr*)&sockAddr, sizeof(SOCKADDR_IN)) == SOCKET_ERROR)
    {
        printf("bind error! %d
", WSAGetLastError());
    }

    //监听5个
    if (listen(severSocket, 5) == SOCKET_ERROR)
    {
        printf("listen error!%d
", WSAGetLastError());
    }

    //创建一个一客户端连接的socket
    while (true)
    {    
        //接受来自客户端的sock，并存入客户端的数组中
        SOCKET sockLink = accept(severSocket, NULL, NULL);
        if (sockLink != INVALID_SOCKET)
        {
            printf("communication sucess!
");
        }
        
        g_psockSockLink[g_nSocketNum++] = sockLink;

        //每启动一个客户端，启动一条线程
        HANDLE hThread = CreateThread(NULL, 0, SeverThread, (LPVOID)sockLink, 0, NULL);
        //CloseHandle(hThread);
        if (hThread == NULL)
            continue;
    }

    closesocket(severSocket);
    closesocket(sockLink);
    WSACleanup();
    return 0;

}

//客户端
// WASClient.cpp : 定义控制台应用程序的入口点。
//

//#include <WinSock2.h>一定要在#include <Windows.h>前面

#include "stdafx.h"
#include <WinSock2.h>
#include <Windows.h>
#pragma comment (lib,"wSock32.lib")


//用于来自接受服务器的数据，避免的send中造成阻塞。
DWORD WINAPI RectThread(LPVOID lParam)
{
    SOCKET sockLink = (SOCKET)lParam;
    char pReturnValue[MAXBYTE] = { 0 };
    while (true)
    {
        recv(sockLink, pReturnValue, MAXBYTE, 0);
        printf("%s
", pReturnValue);
    }
    return 0;
}


int _tmain(int argc, _TCHAR* argv[])
{
    WORD wVersionRequested;
    WSADATA wsaData;
    int err;
    
    //版本检测
    wVersionRequested = MAKEWORD(2, 2);

    err = WSAStartup(wVersionRequested, &wsaData);
    if (err != 0) {

        printf("WSAStartup failed with error: %d
", err);
        return 1;
    }

    if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2) 
    {
        printf("Could not find a usable version of Winsock.dll
");
        WSACleanup();
        return 1;
    }
    else
        printf("The Winsock 2.2 dll was found okay
");


    //程序开始
    //创建socket-》连接connect-》发送send-》接受recv-》释放closesocke

    SOCKET clientSocket=socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    if (clientSocket == INVALID_SOCKET)
    {
        printf("new socket error!");
    }

    SOCKADDR_IN sockAddr;
    //一定要把主机字节序换成网络字节序 并是short类型   htons（）
    sockAddr.sin_port = htons(10086);
    sockAddr.sin_family = AF_INET;
    sockAddr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
    

    //IP地址表示方法
    /*方法1：  m_addr.sin_addr.S_un.S_un_b.s_b1 = 192; m_addr.sin_addr.S_un.S_un_b.s_b2 = 168; m_addr.sin_addr.S_un.S_un_b.s_b3 = 0; m_addr.sin_addr.S_un.S_un_b.s_b4 = 1; 
    方法2：  m_addr.sin_addr.S_un.S_un_w.s_w1 = (168 << 8) | 192; m_addr.sin_addr.S_un.S_un_w.s_w2 = (1 << 8) | 0; 
    方法3：  m_addr.sin_addr.S_un.S_addr = (1 << 24) | (0 << 16) | (168 << 8) | 192*/

    /*sockAddr.sin_addr.S_un.S_un_b.s_b1 = 127;
    sockAddr.sin_addr.S_un.S_un_b.s_b2 = 0;
    sockAddr.sin_addr.S_un.S_un_b.s_b3 = 0;
    sockAddr.sin_addr.S_un.S_un_b.s_b4 = 1;
    */
    
    //连接
    if (connect(clientSocket, (sockaddr*)&sockAddr, sizeof(SOCKADDR_IN)) != SOCKET_ERROR)
    {
        printf("communication sucess!
");
    }

    char pClientBuf[MAXBYTE] = { 0 };        //存放输入数据

    //启动线程
    HANDLE hThread = CreateThread(NULL, 0, RectThread, (LPVOID)clientSocket, 0, NULL);
    if (hThread == NULL)
    {
        printf("CreateThread Error num:%d", GetLastError());
        CloseHandle(hThread);
    }
    CloseHandle(hThread);


    //请求连接，发送数据
    while (TRUE)
    {
        gets_s(pClientBuf);
        int  nSendErr=send(clientSocket, pClientBuf, MAXBYTE, 0);
        if (nSendErr== SOCKET_ERROR)
        {
            break;
        }
    }

    WSACleanup();
    closesocket(clientSocket);

    return 0;
}