Socket多线程局域网点对点聊天

client端

#include <iostream>
#include <string>
#define _WIN32_WINNT 0x501
#include <winsock2.h>
#include <ws2tcpip.h>
#include <w32api.h>
#include <stdio.h>
#include <process.h>
#include <windows.h>
using namespace std;
int socketfd;
unsigned _stdcall Receiving(void*)
{
    while (true)
    {
        ssize_t bytes_received;
        char incoming_data_buffer[1000];
        bytes_received = recv(socketfd,incoming_data_buffer,1000,0);
        if (bytes_received == 0)
        {
            cout << "host shut down."<<endl;
            break;
        }

        if (bytes_received == -1)
            continue;
       incoming_data_buffer[bytes_received]='\0';
       cout << bytes_received <<" "<<"bytes received :"<<endl;
       cout << incoming_data_buffer <<endl;
       cout <<"---------------Received--------------"<<endl<<""<<" ";
    }

}
unsigned _stdcall Sending(void*)
{
    string str,code=">q";
    fflush(stdin);
    while (true)
    {
        cout << " ";
        getline(cin,str);
        const char *msg=str.data();
        const char *c_ode=code.data();
        if (strcmp(msg,c_ode)==0)
            break;
        int len;
        ssize_t bytes_sent;
        len=strlen(msg);
        bytes_sent = send(socketfd,msg,len,0);
        cout << "----------------Sent----------------"<<endl;
    }
}
int main()
{
    WSADATA wsadata;
    if (WSAStartup(MAKEWORD(2,2),&wsadata)!=0)
    {
        cout <<"WSAStartup failed to initialize!"<<endl;
        return 0;
    }
    int status;
    addrinfo host_info;
    addrinfo *host_info_list;
    memset(&host_info,0,sizeof(host_info));
    cout << "Setting up the struct..."<<endl;
    host_info.ai_family=AF_UNSPEC;
    host_info.ai_socktype = SOCK_STREAM;
    char ip[16],port[9999];
    cout <<"IP Address:";
    cin>>ip;
    cout<<"Port:";
    cin >>port;
    status=getaddrinfo(ip,port,&host_info,&host_info_list);
    if (status!=0)
        cout << "getaddrinfo error"<<gai_strerror(status);
    cout << "Creating a socket..."<<endl;
    socketfd = socket(host_info_list->ai_family,host_info_list->ai_socktype,host_info_list->ai_protocol);
    if (socketfd == -1)
    {
        cout <<"Socket error."<<endl;
        return 0;
    }
    cout <<"Connecting to "<<ip<<endl;
    status=connect(socketfd,host_info_list->ai_addr,host_info_list->ai_addrlen);
    if (status == -1)
    {
        cout << "connect error"<<endl;
        return 0;
    }
    cout << "Connection established."<<endl;
    unsigned int thID1,thID2;
    HANDLE hth1,hth2;
    hth2=(HANDLE)_beginthreadex(NULL,0,Receiving,NULL,0,&thID2);
    hth1=(HANDLE)_beginthreadex(NULL,0,Sending,NULL,0,&thID1);
    WaitForSingleObject(hth1,INFINITE);
    cout << "Closing threads: ID: "<<thID1<<" "<<thID2<<endl;
    CloseHandle(hth1);
    CloseHandle(hth2);
    cout << "Closing socket..."<<endl;
    freeaddrinfo(host_info_list);
    closesocket(socketfd);
    return 0;
}

服务端

#include <iostream>
#include <cstring>
#define _WIN32_WINNT 0x501
#include <winsock2.h>
#include <ws2tcpip.h>
#include <w32api.h>
#include <stdio.h>
#include <process.h>
#include <windows.h>
using namespace std;
int new_sd;
unsigned _stdcall Receiving(void*)
{
    while (true)
    {
        ssize_t bytes_received;
        char incoming_data_buffer[1000];
        bytes_received = recv(new_sd,incoming_data_buffer,1000,0);
        if (bytes_received == 0)
        {
            cout << "host shut down."<<endl;
            break;
        }
        if (bytes_received == -1)
            continue;
        incoming_data_buffer[bytes_received]='\0';//important.it will appear "\" at the end of letter without it.
        cout << bytes_received <<" "<<"bytes received :"<<endl;
        cout << incoming_data_buffer <<endl;
        cout <<"-------------------Received-------------------"<<endl<<" ";
    }
}
unsigned _stdcall Sending(void*)
{
    string str,code=">q";
    fflush(stdin);
    while (true)
    {
        cout << " ";
        getline(cin,str);
        const char *msg=str.data();
        const char *c_ode=code.data();
        if (strcmp(msg,c_ode)==0)
            break;
        int len;
        ssize_t bytes_sent;
        len=strlen(msg);
        bytes_sent = send(new_sd,msg,len,0);
        cout << "---------------------Sent--------------------"<<endl;
    }
}
int main()
{
    WSADATA wsadata;
    if (WSAStartup(MAKEWORD(2,2),&wsadata)!=0)
    {
        cout <<"WSAStartup failed to initialize!"<<endl;
        return 0;
    }
    int status;
    addrinfo host_info;
    addrinfo *host_info_list;
    memset(&host_info,0,sizeof(host_info));
    cout << "Setting up the struct..."<<endl;
    host_info.ai_family=AF_UNSPEC;
    host_info.ai_socktype = SOCK_STREAM;
    host_info.ai_flags = AI_PASSIVE;
    char *ip=NULL;
    char name[32]="";
    PHOSTENT hostinfo;
    if(gethostname (name, sizeof(name)) == 0)
    {
            //如果成功 将本地主机名存放入name缓冲区
        if((hostinfo = gethostbyname(name)) != NULL)
        {
          ip = inet_ntoa (*(struct in_addr *)*hostinfo->h_addr_list);//获取主机IP
        }
    }
    cout <<"Server IP: "<<ip<<endl;
    status=getaddrinfo(ip,"5555",&host_info,&host_info_list);
    if (status!=0)
        cout << "getaddrinfo error"<<gai_strerror(status);
    cout << "Creating a socket..."<<endl;
    int socketfd;
    socketfd = socket(host_info_list->ai_family,host_info_list->ai_socktype,host_info_list->ai_protocol);
    if (socketfd == -1)
        cout <<"socket error";
    cout <<"Binding socket..."<<endl;
    status=setsockopt(socketfd,SOL_SOCKET,SO_REUSEADDR,"1",sizeof(int));
    status=bind(socketfd,host_info_list->ai_addr,host_info_list->ai_addrlen);
    if (status == -1)
        cout << "bind error"<<endl;
    cout <<"Listening for connections..."<<endl;
    status=listen(socketfd,5);
    if (status == -1)
        cout << "listen error"<<endl;
    struct sockaddr_storage their_addr;
    socklen_t addr_size = sizeof(their_addr);
    new_sd=accept(socketfd,(struct sockaddr *)&their_addr,&addr_size);
    if (new_sd == -1)
    {
        cout <<"listen error"<<endl;
    }
    else
    {
        cout <<"Connections accepted.Using new socketfd : "<<new_sd<<endl;
    }
    unsigned int thID1,thID2;
    HANDLE hth1,hth2;
    hth1=(HANDLE)_beginthreadex(NULL,0,Receiving,NULL,0,&thID1);
    hth2=(HANDLE)_beginthreadex(NULL,0,Sending,NULL,0,&thID2);
    WaitForSingleObject(hth2,INFINITE);
    cout << "Closing threads: ID: "<<thID1<<" "<<thID2<<endl;
    CloseHandle(hth1);
    CloseHandle(hth2);
    cout << "Stopping Server..."<<endl;
    freeaddrinfo(host_info_list);
    closesocket(new_sd);
    closesocket(socketfd);
    return 0;
}

先贴代码，有空再解释。