/****************************************************

【服务端】基于TCP,多线程的聊天框架代码
评注:非常完整
*****************************************************
*/

#include 
"stdafx.h"
#include 
<stdio.h>
#include 
<winsock2.h>
#pragma comment(lib,
"ws2_32.lib")

/*****************定义客户端连接上后的聊天线程函数************/

DWORD WINAPI ClientThread(LPVOID lpParam)
{
    SOCKET  sock 
= (SOCKET)lpParam;//定义连接客户端的套接字
    char  szBuff[2048];//定义接收缓冲区
    char  szMessage[2048];//定义发送的消息
    int  ret,
        nLeft,
        idx;
//nLeft,idx用以控制写缓冲的数据,以保证数据写入正确
    
//提示输入exit退出聊天
    puts("输入\"exit\"可退出聊天\n");
    
//进入数据传输循环,即聊天
    
//缺陷是只能一人一句来回对话
    while(1)
    
{
        
        
///////////////////////// 接收  /////////////////
        ret = recv(sock,szBuff,2048,0);
        
if(ret == 0
            
break;
        
else if(ret == SOCKET_ERROR)
        
{
            printf(
"recv() failed:%d\n",WSAGetLastError());
            
break;
        }

        
        szBuff[ret] 
= '\0';
        
//判断对方发过来的是否为exit退出命令,若是则退出聊天继续监听
        if(!strcmp(szBuff,"exit")) 
        
{
            printf(
"对方已经停止聊天!\n");
            printf(
"服务器正在监听");
            
break;
        }

        printf(
"客户:%s\n",szBuff);//在控制台打印客户的聊天语句
        ////////////////////////// 回复  //////////////////
        printf("发送消息:"); 
        
//服务器输入数据回答客户
        gets(szMessage);
        
//若发送为空,则传送‘不说话’三字,并提示
        if(strlen(szMessage)==0)
        
{
            printf(
"发送不能为空哦\n");
            strcpy(szMessage,
"不说话!");
        }

        
//传送数据
        nLeft = strlen(szMessage);
        idx 
= 0;
        
//确保写进所有数据
        while(nLeft>0)
        
{
            ret 
= send(sock,&szMessage[idx],nLeft,0);
            
if(ret == 0)
                
break;
            
else if(ret == SOCKET_ERROR)
            
{
                printf(
"send error!%d\n",WSAGetLastError());
                
break;
            }

            nLeft
-=ret;
            idx 
+=ret;
        }

        
//判断szMessage是否为exit命令,若是则退出聊天继续监听
        if(!strcmp(szMessage,"exit")) 
        
{
            printf(
"连接正在断开!\n");
            printf(
"服务器继续监听\n");
            
break;
        }

    }

    
return 0;
}


//主函数
int main(int argc, char* argv[])
{
    WSADATA  wsd;
//定义WINSOCK32消息结构体
    SOCKET  sServSock;//服务器端的套接字
    SOCKET  sConns;//服务器的各连接
    HANDLE  hThread;//定义处理客户连接的县城
    DWORD  dwThreadId;//定义线程ID
    char  szAddress[128];//监听的地址
    struct hostent *host = NULL;//定义本地地址指针
    sockaddr_in local,
        client;
//分别定义本地,客户端的地址结构
    int nSockErr;//定义出错信息
    printf("请你输入监听地址(格式如202.204.118.138):");
    gets(szAddress);
    
int nAddrLen = sizeof(client);//得到地址结构长度
    
    
    
//初始化Winsock32库
    if(WSAStartup(MAKEWORD(2,2),&wsd) != 0)
    
{
        printf(
"failed to load winsock!\n");
        
return 1;
    }

    
//建立socket对象
    sServSock = socket(AF_INET,SOCK_STREAM,IPPROTO_IP);//流套接字,面向连接
    
    
//为socket分配端口地址监听
    local.sin_family = AF_INET;
    local.sin_port 
= htons(5150);//监听端口
    
    
//若地址出错则监听本机地址
    if((local.sin_addr.s_addr = inet_addr(szAddress))
        
==INADDR_NONE)
    
{
        puts(
"所输入的地址不正确,本服务将使用本机地址!");
        
//得到主机名
        if(gethostname(szAddress,sizeof(szAddress))==SOCKET_ERROR)//得到本机的域名,名称
        {   
            puts(
"Can't getting local host name.");
            
        }

        
//通过主机名得到主机IP地址
        host = gethostbyname(szAddress);//得到本地ip
        if(host)
            CopyMemory(
&local.sin_addr,host->h_addr_list[0],
            host
->h_length);
        
else
        
{
            printf(
"gethostbyname() failed:%d\n",WSAGetLastError());
            Sleep(
5000);
            
return 1;
        }

    }

    
    
//将套接字绑定到本机地址local上
    if(bind(sServSock,(LPSOCKADDR)&local,sizeof(local))==SOCKET_ERROR)
    
{
        nSockErr 
= WSAGetLastError();
        printf(
"bind error:%d!\n", nSockErr);
        
return 1;
    }

    
//监听客户连接请求
    if(listen(sServSock,5)==SOCKET_ERROR)
    
{
        nSockErr 
=WSAGetLastError();
        printf(
"listen error:%d\n", nSockErr);
        
return 1;
    }

    
//提示状态
    printf("服务器启动成功!\n");
    printf(
"服务器正在监听\n");
    
//进入处理连接循环
    while(1)
    
{
        
//若有客户连接,则接受连接
        sConns = accept(sServSock,(struct sockaddr *)&client,&nAddrLen);
        
if(sConns == INVALID_SOCKET)
        
{
            nSockErr 
= WSAGetLastError();
            printf(
"accept error %d\n",nSockErr);
            
break;
        }

        
//连接正确则提示可以开始聊天
        printf("%s:%d连接到了本服务,现在可以聊天了.\n",inet_ntoa(client.sin_addr),ntohs(client.sin_port));
        
//创建一个线程用于聊天
        hThread = CreateThread(NULL,0,ClientThread,    (LPVOID)sConns,0,&dwThreadId);
        
if(hThread == NULL)
        
{
            printf(
"CreateThread() failed %d\n",GetLastError());
            
break;
        }

        
//聊天结束关闭聊天线程,继续监听
        CloseHandle(hThread);
    }

    closesocket(sServSock);
    WSACleanup();
    
return 0;
}









/***********************************************************
客户端
***********************************************************
*/

#include 
"stdafx.h"
#include 
<stdio.h>
#include 
<winsock2.h>
#pragma comment(lib,
"ws2_32.lib")
int main(int argc,char **argv)
{
    WSADATA  wsd;
//定义winsock32信息结构
    SOCKET  sClient;//定义本地套接字
    char  szBuffer[2048];//定义接受的缓冲
    char  szMessage[2048];//发送的消息
    char  szServer[128];//连接的服务器地址,IP地址
    int   ret;
    
    
struct  sockaddr_in server;//定义连接的服务器地址
    struct  hostent    *host =NULL;//定义地址
    
    
//提示输入连接的服务器地址
    printf("请输入连接的服务器IP地址(如:202.204.118.138):");
    gets(szServer);
    
//初始winsock库
    if(WSAStartup(MAKEWORD(2,2),&wsd)!=0)
    
{
        printf(
"Failed to load Winsock library!\n");Sleep(5000);
        
return 1;
    }

    
    
// strcpy(szMessage,"我是***");
    
    
    
//建立socket对象
    sClient =  socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
    
if(sClient == INVALID_SOCKET)
    
{
        printf(
"socket() failed :%d\n",WSAGetLastError());Sleep(5000);
        
return 1;
    }

    
    
//定义服务器地址以发送信息
    server.sin_family = AF_INET;
    server.sin_port 
= htons(5150);//端口
    
    server.sin_addr.s_addr 
= inet_addr(szServer);
    
    
//若没有地址,则将地址设置为本机地址
    if(server.sin_addr.s_addr == INADDR_NONE)
    
{
        puts(
"所输入服务器地址不正确,本将使用本机地址!");
        
//得到主机名
        if(gethostname(szServer,sizeof(szServer))==SOCKET_ERROR)//得到本机的域名,名称
        {   
            puts(
"Can't getting local host name.");
            
        }

        
//通过主机名得到IP
        host = gethostbyname(szServer);
        
if(host == NULL)
        
{
            printf(
"Unable to resolve server:%s\n",szServer);Sleep(5000);
            
return 1;
        }

        CopyMemory(
&server.sin_addr,host->h_addr_list[0],host->h_length);
    }

    
//建立连接
    if(connect(sClient,(struct sockaddr *)&server,sizeof(server))==SOCKET_ERROR)
    
{
        printf(
"connect() failed:%d\n",WSAGetLastError());
        Sleep(
5000);
        
return 1;
    }

    
//提示当前状态
    puts("连接成功,现在可以聊天了!\n");
    puts(
"输入\"exit\"可退出聊天\n");
    
    
    
//进入聊天状态
    while(1)
    
{
        
////////发送
        printf("发送消息:");
        
//写入发送信息
        gets(szMessage);
        
if(strlen(szMessage)==0)
        
{
            printf(
"发送不能为空哦\n");
            strcpy(szMessage,
"不说话!");
        }

        
//发送信息
        ret = send(sClient,szMessage,strlen(szMessage),0);
        
if(ret == 0)
        
{
        }

        
else if(ret == SOCKET_ERROR)
        
{
            printf(
"send() failed: %d\n",WSAGetLastError());
            Sleep(
5000);
        }

        
//判断输入信息是否为exit命令,若是则退出
        if(!strcmp(szMessage,"exit"))
        
{
            printf(
"你已经退出了聊天!");
            
break;
        }

        
// printf("send %d byte\n",ret);
        
        
/////接收回复的信息
        ret = recv(sClient,szBuffer,2048,0);
        
if(ret == 0
            ;
        
else if(ret == SOCKET_ERROR)
        
{
            printf(
"recv()failed:%d\n",WSAGetLastError());
            
        }

        
        
//设置接收得到的字符串,并打印
        szBuffer[ret] = '\0';
        
if(!strcmp(szBuffer,"exit"))
        
{
            printf(
"服务器已经停止聊天!");
            
break;
        }

        printf(
"服务器:%s\n",szBuffer);
    }
 
    
    
//关闭套接字
    closesocket(sClient);
    
//清空winsock环境
    WSACleanup();
    
return 0;
    
}