将服务端移植到Linux和MAC OS
生活随笔
收集整理的這篇文章主要介紹了
将服务端移植到Linux和MAC OS
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
文章目錄
- 1 將服務端移植到Linux和MAC OS
- 1.1 概述
- 1.2 服務端代碼實現
1 將服務端移植到Linux和MAC OS
1.1 概述
這里我們將服務端移植到Linux和MAC OS,差異很小,直接上代碼即可。
1.2 服務端代碼實現
#ifdef _WIN32#define WIN32_LEAN_AND_MEAN#define _WINSOCK_DEPRECATED_NO_WARNINGS#include<windows.h>#include<WinSock2.h>#pragma comment(lib,"ws2_32.lib") #else#include<unistd.h> //uni std#include<arpa/inet.h>#include<string.h>#define SOCKET int#define INVALID_SOCKET (SOCKET)(~0)#define SOCKET_ERROR (-1) #endif#include<stdio.h> #include<thread> #include<vector>enum CMD {CMD_LOGIN,CMD_LOGIN_RESULT,CMD_LOGOUT,CMD_LOGOUT_RESULT,CMD_NEW_USER_JOIN,CMD_ERROR };struct DataHeader {short dataLength;short cmd; };//DataPackage struct Login : public DataHeader {Login(){dataLength = sizeof(Login);cmd = CMD_LOGIN;}char userName[32];char PassWord[32]; };struct LoginResult : public DataHeader {LoginResult(){dataLength = sizeof(LoginResult);cmd = CMD_LOGIN_RESULT;result = 0;}int result; };struct Logout : public DataHeader {Logout(){dataLength = sizeof(Logout);cmd = CMD_LOGOUT;}char userName[32]; };struct LogoutResult : public DataHeader {LogoutResult(){dataLength = sizeof(LogoutResult);cmd = CMD_LOGOUT_RESULT;result = 0;}int result; };struct NewUserJoin : public DataHeader {NewUserJoin(){dataLength = sizeof(NewUserJoin);cmd = CMD_NEW_USER_JOIN;scok = 0;}int scok; };std::vector<SOCKET> g_clients;int processor(SOCKET _cSock) {//緩沖區char szRecv[4096] = {};// 5 接收客戶端數據int nLen = (int)recv(_cSock, szRecv, sizeof(DataHeader), 0);DataHeader* header = (DataHeader*)szRecv;if (nLen <= 0){printf("客戶端<Socket=%d>已退出,任務結束。\n", _cSock);return -1;}switch (header->cmd){case CMD_LOGIN:{recv(_cSock, szRecv + sizeof(DataHeader), header->dataLength - sizeof(DataHeader), 0);Login* login = (Login*)szRecv;printf("收到客戶端<Socket=%d>請求:CMD_LOGIN,數據長度:%d,userName=%s PassWord=%s\n", _cSock, login->dataLength, login->userName, login->PassWord);//忽略判斷用戶密碼是否正確的過程LoginResult ret;send(_cSock, (char*)&ret, sizeof(LoginResult), 0);}break;case CMD_LOGOUT:{recv(_cSock, szRecv + sizeof(DataHeader), header->dataLength - sizeof(DataHeader), 0);Logout* logout = (Logout*)szRecv;printf("收到客戶端<Socket=%d>請求:CMD_LOGOUT,數據長度:%d,userName=%s \n", _cSock, logout->dataLength, logout->userName);//忽略判斷用戶密碼是否正確的過程LogoutResult ret;send(_cSock, (char*)&ret, sizeof(ret), 0);}break;default:{DataHeader header = { 0,CMD_ERROR };send(_cSock, (char*)&header, sizeof(header), 0);}break;}return 0; }int main() { #ifdef _WIN32//啟動Windows socket 2.x環境WORD ver = MAKEWORD(2, 2);WSADATA dat;WSAStartup(ver, &dat);//------------ #endif//-- 用Socket API建立簡易TCP服務端// 1 建立一個socket 套接字SOCKET _sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);// 2 bind 綁定用于接受客戶端連接的網絡端口sockaddr_in _sin = {};_sin.sin_family = AF_INET;_sin.sin_port = htons(4567);//host to net unsigned short #ifdef _WIN32_sin.sin_addr.S_un.S_addr = INADDR_ANY;//inet_addr("127.0.0.1"); #else_sin.sin_addr.s_addr = INADDR_ANY; #endifif (SOCKET_ERROR == bind(_sock, (sockaddr*)&_sin, sizeof(_sin))){printf("錯誤,綁定網絡端口失敗...\n");}else {printf("綁定網絡端口成功...\n");}// 3 listen 監聽網絡端口if (SOCKET_ERROR == listen(_sock, 5)){printf("錯誤,監聽網絡端口失敗...\n");}else {printf("監聽網絡端口成功...\n");}while (true){//伯克利套接字 BSD socketfd_set fdRead;//描述符(socket) 集合fd_set fdWrite;fd_set fdExp;//清理集合FD_ZERO(&fdRead);FD_ZERO(&fdWrite);FD_ZERO(&fdExp);//將描述符(socket)加入集合FD_SET(_sock, &fdRead);FD_SET(_sock, &fdWrite);FD_SET(_sock, &fdExp);SOCKET maxSock = _sock;for (int n = (int)g_clients.size() - 1; n >= 0; n--){FD_SET(g_clients[n], &fdRead);if (maxSock < g_clients[n]){maxSock = g_clients[n];}}///nfds 是一個整數值 是指fd_set集合中所有描述符(socket)的范圍,而不是數量///既是所有文件描述符最大值+1 在Windows中這個參數可以寫0timeval t = { 1,0 };int ret = select(maxSock + 1, &fdRead, &fdWrite, &fdExp, &t);if (ret < 0){printf("select任務結束。\n");break;}//判斷描述符(socket)是否在集合中if (FD_ISSET(_sock, &fdRead)){FD_CLR(_sock, &fdRead);// 4 accept 等待接受客戶端連接sockaddr_in clientAddr = {};int nAddrLen = sizeof(sockaddr_in);SOCKET _cSock = INVALID_SOCKET; #ifdef _WIN32_cSock = accept(_sock, (sockaddr*)&clientAddr, &nAddrLen); #else_cSock = accept(_sock, (sockaddr*)&clientAddr, (socklen_t *)&nAddrLen); #endifif (INVALID_SOCKET == _cSock){printf("錯誤,接受到無效客戶端SOCKET...\n");}else{for (int n = (int)g_clients.size() - 1; n >= 0; n--){NewUserJoin userJoin;send(g_clients[n], (const char*)&userJoin, sizeof(NewUserJoin), 0);}g_clients.push_back(_cSock);printf("新客戶端加入:socket = %d,IP = %s \n", (int)_cSock, inet_ntoa(clientAddr.sin_addr));}}for (int n = (int)g_clients.size() - 1; n >= 0; n--){if (FD_ISSET(g_clients[n], &fdRead)){if (-1 == processor(g_clients[n])){auto iter = g_clients.begin() + n;//std::vector<SOCKET>::iteratorif (iter != g_clients.end()){g_clients.erase(iter);}}}}//printf("空閑時間處理其它業務..\n");} #ifdef _WIN32for (int n = (int)g_clients.size() - 1; n >= 0; n--){closesocket(g_clients[n]);}// 8 關閉套節字closesocketclosesocket(_sock);//------------//清除Windows socket環境WSACleanup(); #elsefor (int n = (int)g_clients.size() - 1; n >= 0; n--){close(g_clients[n]);}// 8 關閉套節字closesocketclose(_sock); #endifprintf("已退出。\n");getchar();return 0; }參考資料:
總結
以上是生活随笔為你收集整理的将服务端移植到Linux和MAC OS的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 淘宝文件系统文件映射原理及实现
- 下一篇: xp系统怎么从做系统教程视频教程 学习如