界面用的Qt做的，制作的是一個小工具，實在是沒時間，代碼只是初步代碼，很多Bug，估計還不能完美運行起來了。開始打算增加個內網穿透模塊的，一來找不到服務器，二來工作上上最近換新框架，真的抽不出時間來，每天下班都比較累，想聽下音樂放松下就洗洗睡了，各位不要見笑。

下面上代碼。初步模型呢，畫了個圖，方便理解，都是小東西，select是過時的東西了，并不適合真正的項目了。

?

?

?

代碼如下：

Server:

Server.h class ServerInit { public:ServerInit(void);~ServerInit(void);void Init();void CheckPackType(SOCKET *_socket);void AddFileinfo(DataInfo _datainfo,SOCKET *_sock);void SendInfo(SOCKET _sock,ServerInit *pServer);void ClientRegister(DataInfo *_pdata,SOCKET *_sock);FD_SET * GetFD(){return &m_fd;}SOCKET* GetSocket(){return &m_ServerSock;}std::map<char*,ServerInfo>* GetMap(){return &m_ClientData;}std::map<SOCKET,sockaddr_in> m_clientIP;private:SOCKET m_ServerSock;WSADATA m_ServerData;SOCKADDR_IN m_ServerADDR;SOCKET m_acceptClient;HANDLE m_handle;DWORD m_ThreadID;FD_SET m_fd;std::map<char*,ServerInfo>m_ClientData;};

　　

Server.cpp#include "stdafx.h" #include "ServerInit.h"timeval g_timeout = {1,0}; //2s輪詢ServerInit::ServerInit(void) { #ifdef WIN32int errorValue = WSAStartup(MAKEWORD(2,2),&m_ServerData);if (errorValue != NO_ERROR){printf("Init Error,Get Version Error");return;}if(LOBYTE(m_ServerData.wVersion)!=2 || HIBYTE(m_ServerData.wVersion)!=2){printf("Error,網絡初始化錯誤");return ;} #endifm_ServerSock = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);if(m_ServerSock == INVALID_SOCKET){return;}m_ServerADDR.sin_family = AF_INET;m_ServerADDR.sin_addr.S_un.S_addr =INADDR_ANY;m_ServerADDR.sin_port = htons(SERVER_PORT);if(bind(m_ServerSock,(sockaddr*)&m_ServerADDR,sizeof(m_ServerADDR))<0){ return ;}listen(m_ServerSock,20);FD_ZERO(&m_fd);FD_SET(m_ServerSock,&m_fd); }ServerInit::~ServerInit(void) {}//檢測包的類型 void ServerInit::CheckPackType(SOCKET *_socket) { DataInfo *buff = new DataInfo;recv(*_socket,(char*)buff,g_Packlength,NULL);switch (buff->id){case IsRegister: ClientRegister(buff,_socket);break;case SEARCHDATA: SendInfo(_sock,pServer);break;case sendFileInfo: AddFileinfo(pdata,_sock);default:printf("錯誤的數據類型== %d",pdata.id);break;} }void ServerInit::ClientRegister(DataInfo *_pdata,SOCKET *_sock) { DataInfo RegisterInfo;std::map<SOCKET,sockaddr_in>::iterator _clientinfo = m_clientIP.find(*_sock);assert(_clientinfo != m_clientIP.end());char* clientIP = inet_ntoa(_clientinfo->second.sin_addr);std::map<char*,ServerInfo>::iterator _iter = m_ClientData.find(clientIP);if (_iter == m_ClientData.end()){ printf("新的用戶,%s",clientIP);RegisterInfo.id = UnKnowClient;send(*_sock,(char*)&RegisterInfo,sizeof(DataInfo),NULL);}else{RegisterInfo.id = KnowClient;send(*_sock,(char*)&RegisterInfo,sizeof(DataInfo),NULL);}}void ServerInit::SendInfo(SOCKET _sock,ServerInit *pServer) { std::map<char*,ServerInfo>::iterator _iter = pServer->m_ClientData.begin();DataInfo _datainfo;for (;_iter != pServer->m_ClientData.end();_iter++){ auto _piter = _iter->second.m_info.begin();for (;_piter != _iter->second.m_info.end();_piter++ ){ _datainfo = *_piter;send(_sock,(char*)&_datainfo,sizeof(DataInfo),NULL);_datainfo.Reset();}} }void ServerInit::AddFileinfo(DataInfo _datainfo,SOCKET *_sock) { ServerInfo _Sinfo;std::map<SOCKET,sockaddr_in>::iterator itet = m_clientIP.find(*_sock);if (itet == m_clientIP.end()){assert(false);}_Sinfo.ip = inet_ntoa(itet->second.sin_addr);_Sinfo.port = ntohs(itet->second.sin_port);_Sinfo.m_info.push_back(_datainfo);m_ClientData.insert(std::make_pair(_Sinfo.ip,_Sinfo)); }

　　

main:int g_state; static void ThreadNew(LPVOID lparam) { ServerInit *pData =(ServerInit*)&lparam;sockaddr_in clientinfo;int len = sizeof(sockaddr_in);SOCKET NewClient = accept(*pData->GetSocket(),(sockaddr*)&clientinfo,&len);FD_SET(NewClient,pData->GetFD());pData->m_clientIP.insert(std::make_pair(NewClient,clientinfo));_endthread(); }int _tmain(int argc, _TCHAR* argv[]) { ServerInit app;while(true){ FD_SET fd_read = *app.GetFD();g_state =select(*app.GetSocket()+1,&fd_read,NULL,NULL,NULL);if(g_state > 0){ for (int i = 0; i<app.GetFD()->fd_count;i++){if (FD_ISSET(app.GetFD()->fd_array[i],&fd_read)){if (app.GetFD()->fd_array[i] == *app.GetSocket()){_beginthread(ThreadNew,NULL,(void*)&app);}else{ app.CheckPackType((&)app.GetFD()->fd_array[i]);}}}}if (g_state == -1){int erron = WSAGetLastError();return -1;}}return 0; }

　　

?

?

?

?

Client:

?

client.h

class SockGUIThread : public QMainWindow
{
Q_OBJECT

public:
SockGUIThread(QWidget *parent = 0);
~SockGUIThread();

int InitClient();
int Register();
public slots:
void Search();

private:
Ui::SockGUIThreadClass ui;
SOCKET m_sock;
WSADATA m_data;
int m_errorValue;
SOCKADDR_IN addclient;
char Type[1]; //請求搜索的類型
DataInfo m_recvinfo;

};

　　

client.cpp

SockGUIThread::SockGUIThread(QWidget *parent)
: QMainWindow(parent)
{
ui.setupUi(this);
ui.m_Search->setEnabled(false);
ui.m_progressBar->setMaximum(100);
ui.m_progressBar->setMinimum(0);
QObject::connect(ui.m_Search,SIGNAL(clicked()),this,SLOT(Search()));
QObject:;connect(ui.m_close,SIGNAL(clicked()),this,SLOT(close()));
Type[0]=1;
if ( InitClient() < 0 )
{
QMessageBox::about(this,"Error","初始化失敗，請嘗試用管理員權限運行此程序");
closesocket(m_sock);
WSACleanup();
// close();
}

}

SockGUIThread::~SockGUIThread()
{
closesocket(m_sock);
WSACleanup();


}

//第一次安裝客戶端，提交客戶端文件信息
int SockGUIThread::InitClient()
{
m_errorValue = WSAStartup(MAKEWORD(2,2),&m_data);
if (m_errorValue != NO_ERROR)
{
QMessageBox::about(this,"Error","初始化網絡錯誤13001");
return -1;
}

if(LOBYTE(m_data.wVersion)!=2 || HIBYTE(m_data.wVersion)!=2)
{
QMessageBox::about(this,"Error","網絡初始化錯誤13003");
return -1;
}

m_sock = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
if(m_sock == INVALID_SOCKET)
{
QMessageBox::about(this,"Error","初始化網絡出錯13002");
return -1;
}

addclient.sin_addr.S_un.S_addr = inet_addr(SERVER_IP);
addclient.sin_family = AF_INET;
addclient.sin_port = htons(SERVER_PORT);

if (::connect(m_sock,(LPSOCKADDR)&addclient,sizeof(addclient)) < 0 )
{
QMessageBox::about(this,"Error","連接到服務器失敗 初始化");
return -1;
}
DataInfo req;
req.id = IsRegister;
send(m_sock,(char*)&req,g_Packlength("test"),0);
req.Reset();
recv(m_sock,(char*)&req,g_Packlength("test"),0);
if(req.id == UnKnowClient)
{
Register();
}

if(req.id == KnowClient)
{
ui.m_Search->setEnabled(true);
}
return 1;
}

int SockGUIThread::Register()
{
DataInfo req;


QString CurPath;
QDir dir;
CurPath = dir.currentPath();
QStringList _StringList;
_StringList <<QString("*.jpeg")<<QString("*.png")<<QString("*.jpg")<<QString("*.bmp")<<QString("*.rmvb")<<QString("*.avi")<<QString("*.mp4");
dir.setFilter(QDir::Files | QDir::NoSymLinks);
dir.setNameFilters(_StringList);
int fileCount = dir.count();
if(dir.count() == 0)
{
return EmptyFileter;
}
CurPath = dir.currentPath();
QChar _char = QChar('\\');
QStringList filelist;
for (int i=0;i<dir.count();i++)
{
QString file_name = dir[i];
filelist.append(dir.currentPath() + _char + file_name);
}
_StringList.clear();
_StringList<<"Name"<<"Size";
ui.m_tableWidget->setColumnCount(2);
// ui.m_tableWidget->setRowCount(filelist.count());

FileInfo _info;
for (int i=0;i < filelist.count();i++)
{
/* ui.m_tableWidget->setItem(i,0,new QTableWidgetItem(filelist[i]));*/
QFile _file(filelist[i]);
req.id = sendFileInfo;
req.filesize = _file.size();
req.pName =(char*)&filelist[i];
send(m_sock,(char*)&req,g_Packlength(req.pName),NULL);
req.Reset();
/* ui.m_tableWidget->setItem(i,1,new QTableWidgetItem(QString::number((float)_file.size()/1000)+"kb"));*/
}
ui.m_Search->setEnabled(true);
return 1;
}

void SockGUIThread::Search()
{
DataInfo m_recvinfo;
m_recvinfo.Reset();
m_recvinfo.id =SEARCHDATA;
send(m_sock,(char *)&m_recvinfo,g_Packlength(m_recvinfo.pName),0);
m_recvinfo.Reset();
int GetLen = recv(m_sock,(char*)&m_recvinfo,4096,0);
int _Count = ui.m_tableWidget->rowCount();
ui.m_tableWidget->setItem(_Count+1,0,new QTableWidgetItem(m_recvinfo.pName));
ui.m_tableWidget->setItem(_Count+1,1,new QTableWidgetItem(QString::number((float)m_recvinfo.filesize/(10*6))+"Mb"));

}

　

DataInfo.h#include <list> #include <map> #include <memory.h> #define MAX_ARRAY 1024 #define SERVER_IP "127.0.0.1" #define SERVER_PORT 5656 #define KnowClient 1 #define UnKnowClient 3 #define IsRegister 2 #define RequestFileInfo 10 #define sendFileInfo 11 #define EmptyFileter 20 #define SEND_ERROR -5 #define SEARCHDATA 30struct FileInfo {long filesize;char *pName;};//連接服務端數據類型 struct DataInfo {int id; //id 請求類型long filesize;char *pName;void Reset(){id =NULL;filesize =NULL;pName =NULL;}};struct ServerInfo {char *ip;int port;std::list<DataInfo> m_info; }; #define g_Packlength(str) (sizeof(DataInfo)+strlen(str))

　

轉載于:https://www.cnblogs.com/dragonmoon/p/4189374.html

總結

以上是生活随笔為你收集整理的Socket的select制作多客户端传输（Qt）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。