本應用是一款簡單的模擬qq聊天應用.主要分為伺服器端與客戶端

伺服器select端：

客戶端client:

伺服器端程式碼如下:

select.cpp:

// select.cpp : 定義控制檯應用程式的入口點。//#include "stdafx.h"#include "dataHandle.h"#include#pragma comment(lib, "ws2_32")

void SockInit();

int _tmain(int argc, _TCHAR* argv[])

{

SockInit();

SOCKET lisSock = socket(AF_INET, SOCK_STREAM, 0);

sockaddr_in addr;

addr.sin_family = AF_INET;

addr.sin_port = htons(8090);

addr.sin_addr.S_un.S_addr = ADDR_ANY;

bind(lisSock, (sockaddr*)&addr, sizeof(sockaddr));

listen(lisSock, 5);

fd_set fds;

FD_ZERO(&fds);

FD_SET(lisSock, &fds);  //將監聽套接字加入陣列中

timeval timeout;

timeout.tv_sec = 5;

timeout.tv_usec = 0;

while (1)

{

fd_set tmSet = fds;  //臨時陣列

//移除掉沒有事件發生的套接字

// SELECT 在程式設計的過程中，經常會遇到許多阻塞的函式，

//好像read和網路程式設計時使用的recv, recvfrom函式都是阻塞的函式，

//當函式不能成功執行的時候，程式就會一直阻塞在這裡，無法執行下面的程式碼。

//這是就需要用到非阻塞的程式設計方式，使用select函式就可以實現非阻塞程式設計。

select(0, &tmSet, 0, 0, &timeout);

SOCKET s[5];

int j = 0;

//剩下的都是有事件發生的套接字

for (int i = 0; i < tmSet.fd_count; i++)

{

if (tmSet.fd_array[i] == lisSock) //如果是監聽監聽套接字

{

SOCKET cliSock = accept(lisSock, 0, 0);

printf("新連線.\n");

FD_SET(cliSock, &fds);  //將已連線的套接字加入陣列中

}

else

{

char recvBuf[1024] = { 0 };

int recvLen = recv(tmSet.fd_array[i], recvBuf, 1024, 0);

if (recvLen > 0)

{

//處理資料

HandleData(recvBuf,recvLen);

printf("%s\n",recvBuf);

}

else  //錯誤發生，或者客戶端斷開連線

{

printf("客戶已斷開\n");

FD_CLR(tmSet.fd_array[i], &fds);  //從陣列中移除已埠的客戶端

}

}

}

}

return 0;

}

void SockInit()

{

WORD wVersionRequested;

WSADATA wsaData;

int err;

wVersionRequested = MAKEWORD(2, 2);

err = WSAStartup(wVersionRequested, &wsaData);

if (err != 0) {

return;

}

if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2)

{

WSACleanup();

return;

}

}

dataHandle.h 用於伺服器接收到不同型別訊息處理

#pragma once#include//處理收到的資料

bool HandleData(const char* recvData,int len);

dataHandle.cpp

#include "stdafx.h"

#include "dataHandle.h"

#include "netStruct.h"

//處理收到的資料

bool HandleData(const char* recvData,int len)

{

//printf("%s : %d", recvData, len);

short msgID = *(short*)recvData;

switch (msgID)

{

case TALKALL_MSGID:

{

//...給所有人的訊息

const MSG_TALKALL* takAll = (MSG_TALKALL*)recvData;

printf("群聊訊息： %s\n", takAll->Content);

}

break;

case TALKONE_MSGID:

{

//給某個人的訊息

const MSG_TALKONE* takOne = (MSG_TALKONE*)recvData;

printf("私聊訊息： %s : %s\n", takOne->userName, takOne->Content);

//takOne.useName

}

break;

case LOGIN_MSGID:

//登陸訊息

break;

//.........

case FILEINFO_MSGID:

{

const MSG_SENDFILEINFO *fileInfo = (MSG_SENDFILEINFO*)recvData;

printf("檔案路徑:%s ,檔案大小:%d\n", fileInfo->fileName, fileInfo->fileSize);

}

break;

case FILE_MSGID:

{

const MSG_SENDFILE *file = (MSG_SENDFILE*)recvData;

printf("檔案內容:%s\n", file->fileBuf);

}

break;

default:

break;

}

return true;

}

netStruct.h 傳輸的資訊結構體

#pragma once

#define TALKALL_MSGID 10001

#define TALKONE_MSGID 10002

#define LOGIN_MSGID 10003

#define SENDPIC_MSGID 10004

#define FILEINFO_MSGID 10005

#define FILE_MSGID 10006

//給所有人發的訊息

struct MSG_TALKALL

{

short msgID;

char Content[1024];

};

//給某人發的訊息

struct MSG_TALKONE

{

short msgID;

char userName[32];

char Content[1024];

};

struct MSG_SENDFILEINFO {

short msgID;

char fileName[256];

int fileSize;

};

struct MSG_SENDFILE {

short msgID;

int bufID;

short bufSize;

char fileBuf[1024];

};

客戶端中

client.cpp

BOOL CclientApp::InitInstance(){

// 如果一個執行在 Windows XP 上的應用程式清單指定要

// 使用 ComCtl32.dll 版本 6 或更高版本來啟用視覺化方式，

//則需要 InitCommonControlsEx()。  否則，將無法建立視窗。

INITCOMMONCONTROLSEX InitCtrls;

InitCtrls.dwSize = sizeof(InitCtrls);

// 將它設定為包括所有要在應用程式中使用的

// 公共控制元件類。

InitCtrls.dwICC = ICC_WIN95_CLASSES;

InitCommonControlsEx(&InitCtrls);

CWinApp::InitInstance();

AfxEnableControlContainer();

// 建立 shell 管理器，以防對話方塊包含

// 任何 shell 樹檢視控制元件或 shell 列表檢視控制元件。

CShellManager *pShellManager = new CShellManager;

// 啟用“Windows Native”視覺管理器，以便在 MFC 控制元件中啟用主題

CMFCVisualManager::SetDefaultManager(RUNTIME_CLASS(CMFCVisualManagerWindows));

// 標準初始化

// 如果未使用這些功能並希望減小

// 最終可執行檔案的大小，則應移除下列

// 不需要的特定初始化例程

// 更改用於儲存設定的登錄檔項

// TODO: 應適當修改該字串，

// 例如修改為公司或組織名

SetRegistryKey(_T("應用程式嚮導生成的本地應用程式"));

AfxSocketInit();//網路初始化

clientSock sock;

sock.Create();

bool conRet = sock.Connect(_T("127.0.0.1"), 8090);

if (conRet == false) {

int errCode = GetLastError();

CString err;

err.Format(_T("連線伺服器失敗 : %d，請重試!"), errCode);

MessageBox(0, err, _T("錯誤"), MB_OK);

return FALSE;

}

Login log;//登入介面

INT_PTR res = log.DoModal();

if (res == IDOK) {

CclientDlg dlg(sock,log.userName);//傳送資訊介面,將socket傳入對話方塊介面，以發信資訊

m_pMainWnd = &dlg;

INT_PTR nResponse = dlg.DoModal();

if (nResponse == IDOK)

{

// TODO: 在此放置處理何時用

//  “確定”來關閉對話方塊的程式碼

}

else if (nResponse == IDCANCEL)

{

// TODO: 在此放置處理何時用

//  “取消”來關閉對話方塊的程式碼

}

else if (nResponse == -1)

{

TRACE(traceAppMsg, 0, "警告: 對話方塊建立失敗，應用程式將意外終止。\n");

TRACE(traceAppMsg, 0, "警告: 如果您在對話方塊上使用 MFC 控制元件，則無法 #define _AFX_NO_MFC_CONTROLS_IN_DIALOGS。\n");

}

// 刪除上面建立的 shell 管理器。

if (pShellManager != NULL)

{

delete pShellManager;

}

#ifndef _AFXDLL

ControlBarCleanUp();

#endif

// 由於對話方塊已關閉，所以將返回 FALSE 以便退出應用程式，

//  而不是啟動應用程式的訊息泵。

}

客戶端登入介面:

在文字框輸入文字，併發送出去程式碼如下:

void CclientDlg::OnBnClickedOk()

{

// TODO: 在此新增控制元件通知處理程式程式碼

//CDialogEx::OnOK();

UpdateData();

if (content_send.IsEmpty()) {

MessageBox(_T("請填入訊息!"), _T("提示"), MB_OK);

return;

}

if (isSendFile == FALSE) {

/*

char *buff;

buff = (char *)content_send.GetBuffer(0);

m_sock.Send(buff, strlen(buff),0);//用多位元組來發送

content_send = _T("");//清空內容

*/

/*

MSG_TALKONE takone;

takone.msgID = TALKONE_MSGID;

strcpy_s(takone.userName, userName);

strcpy_s(takone.Content, content_send.GetBuffer(0));

m_sock.Send(&takone, sizeof(MSG_TALKONE));

*/

MSG_TALKALL takall;

takall.msgID = TALKALL_MSGID;

strcpy_s(takall.Content, content_send.GetBuffer(0));

m_sock.Send(&takall, sizeof(MSG_TALKALL));

content_send = _T("");//清空內容

UpdateData(0);

}

else//傳送檔案

{

CreateThread(0, 0,sendFileThread, this,0,0);//方法只能為全域性方法，不能為類方法

相關推薦