  /**//****************************************************
【服務端】基于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;
}
|