Windows Socket 编程：支持多线程（TCP）（环境：VS2010）-白红宇

Windows Socket 编程：支持多线程（TCP）（环境：VS2010）

阅读量：5113 次

发布时间：2019-06-13

本文共 10722 字，大约阅读时间需要 35 分钟。

几点说明：

（1）发送的数据不能有空格；

（2）这是控制台应用程序；

1、UserMesInfo.h 公共文件，客户端和服务器端公用；

1 #ifndef  User_Mes_Sock 2 #define  User_Mes_Sock 3 #include 
     
       4 #endif 5  6 const short  MesLogin   =  1001; 7 const short  MesError   =  1002; 8 const short  MesNormal  =  1003; 9 const short  LoginSuc   =  1004;10 const short  MesToAll   =  1005;11 const short  OnLineUser =  1010;12 13 14 //User Struct15 struct UserInfo16 {17     short        userId;18     char         userName[32];19     SOCKET       userSock;20     LONG         userAddr;21     UserInfo     *pNext;22 };23 24 //Message Struct25 struct MesInfo26 {27     short        MesType;28     char         fromUserName[32];29     char         toUserName[32];30     char         MesContent[1024];31 };

2、TCP：客户端程序

1 #include 
     
        2 #include 
      
         3 #include 
       
          4   5 #include "UserMesInfo.h"  6   7 #pragma comment(lib,"ws2_32.lib")  8   9 using namespace std; 10  11 BOOL isLogin = false; 12  13  14 string charToString(char temp[]) 15 { 16     string tempStr = ""; 17     for(int i = 0; i < sizeof(temp); i++) 18         tempStr += temp[i]; 19     return tempStr; 20 } 21  22 BOOL SendMes(SOCKET servSocket) 23 { 24     SOCKET servSock = servSocket; 25     while(TRUE) 26     { 27         int tempWho; 28         char sendBuff[1024] = {
   0}; 29         if(isLogin) 30         { 31             printf(" '1' : speak to ONE ;\n '2' : speak to ALL ;\n '3' : OnLine Users ; \n '4' : Exit ;\n"); 32  33             cin>>tempWho; 34         } 35         if( isLogin && tempWho == 1) 36         { 37             //To One 38             printf("Please input NAME you want to CHAT : "); 39             char toName[32] = {
   0}; 40             scanf("%s", toName); 41             memcpy(sendBuff, &MesNormal, 2); 42             memcpy(sendBuff+2, toName, 32); 43             printf("Please input what do you want to say : "); 44             char Message[1000] = {
   0}; 45             scanf("%s", Message); 46             memcpy(sendBuff+34, Message, strlen(Message)); 47             send(servSock, sendBuff, sizeof(sendBuff)+1, 0); 48         } else if(isLogin && tempWho == 2){ 49             //To ALL 50             printf("Please input DATA : "); 51             char Message[1000] = {
   0}; 52             scanf("%s", Message); 53  54             memcpy(sendBuff, &MesToAll, 2); 55             memcpy(sendBuff+2, Message, strlen(Message)); 56             send(servSock, sendBuff, sizeof(sendBuff)+1, 0); 57  58         } else if(isLogin && tempWho == 3){ 59             char getOnLineUsers[4]; 60             memcpy(getOnLineUsers, &OnLineUser, 2); 61             send(servSock, getOnLineUsers, sizeof(getOnLineUsers), 0); 62         } else if(isLogin && tempWho == 4){ 63             break; 64         } else if(isLogin){ 65             printf("You input Error! Please Again !"); 66         } 67          68     } 69     return TRUE; 70 } 71  72  73 DWORD WINAPI ProcThread(LPVOID lpParameter) 74 { 75     SOCKET cliSocket = *(SOCKET*)lpParameter; 76  77     //SOCKADDR_IN sockFrom ; 78     //int length = sizeof(SOCKADDR); 79  80     char recvBuf[1024] = {
   0}; 81  82     while(true) 83     { 84         memset(recvBuf, 0x00, 1024); 85         int recvLen = recv(cliSocket, recvBuf, 1024, 0); 86         if(recvLen > 0) 87         { 88             short megId = *(short*)recvBuf; 89             switch(megId) 90             { 91             case LoginSuc: 92                 { 93                     //char userName[32] = {0}; 94                     //memcpy(userName, recvBuf+2, 4); 95                     printf("Login Success !\n"); 96                     isLogin = true; 97                 } 98                 break; 99             case MesNormal:100                 {101                     char Mess[1000];102                     memcpy(Mess, recvBuf+2, sizeof(recvBuf)-2);103                     printf(" ==> Me :%s\n", Mess);104                     105                 }106                 break;107             case MesToAll:108                 {109                     char Mess[1000];110                     memcpy(Mess, recvBuf+2, sizeof(recvBuf)-2);111                     printf(" ==> All :%s\n", Mess);112                 }113                 break;114             case OnLineUser:115                 {116                     char onLineUser[100] = {
   0};117                     118                     memcpy(onLineUser, recvBuf+2, sizeof(recvBuf)-2);119 120                     //string userStr = charToString(onLineUser);121                     122                     for(int i =0; i < sizeof(onLineUser); i++)123                         cout<
        
         userSock = clientSock;201     strcpy_s(user->userName, userName);202     user->userAddr = addrClient.sin_addr.S_un.S_addr;203     */204     205     char loginInfo[64] = {
   0};206     memcpy(loginInfo, &MesLogin, 2);207     memcpy(loginInfo+2, userName, strlen(userName));208 209     210     //int a = send(clientSock, loginInfo, strlen(loginInfo), 0);211     int a = send(clientSock, loginInfo, strlen(loginInfo), 0);212     213     SendMes(clientSock);214 215     //Sleep(20000);216     //closesocket(clientSock);217     CloseHandle(thHandle);218     WSACleanup();219 220     221     system("PAUSE");222     return 0;223 }

3、TCP：服务器端程序

1 #include 
     
        2 #include 
      
         3 #include 
       
          4 #include 
        
           5 #include 
         
            6 #include "UserMesInfo.h"  7   8 #pragma comment(lib,"ws2_32.lib")  9  10 using namespace std; 11  12 UserInfo *gUserInfo ; 13  14 string charToString(char temp[]) 15 { 16     string tempStr = ""; 17     for(int i = 0; i < sizeof(temp); i++) 18         tempStr += temp[i]; 19     return tempStr; 20 } 21  22  23 DWORD WINAPI ProcMes(LPVOID lpParameter) 24 { 25     SOCKET cliSocket = ((UserInfo*)lpParameter)->userSock; 26     //SOCKADDR_IN sockFrom ; 27  28     char recvBuf[1024]; 29     //int length = sizeof(SOCKADDR); 30  31     while(true) 32     { 33         memset(recvBuf, 0x00, 1024); 34         recv(cliSocket, recvBuf, 1024, 0); 35          36         short MesType = *(short*)recvBuf; 37  38         switch(MesType) 39         { 40         case MesLogin: 41             { 42                 static short USERID = 1001; 43                 UserInfo* userInfo = new UserInfo; 44                 char* userName = (char*)(recvBuf + 2); 45                 strcpy_s(userInfo->userName, userName); 46                 userInfo->userId = USERID++; 47                 userInfo->userSock = cliSocket; 48                 userInfo->userAddr = ((UserInfo*)lpParameter)->userAddr; 49                 userInfo->pNext = NULL; 50  51                 if(!gUserInfo) 52                 { 53                     gUserInfo = userInfo; 54                 }else  55                 { 56                     userInfo->pNext = gUserInfo; 57                     gUserInfo = userInfo; 58                 } 59                 printf("Login  :  %s ==> %d !\n", userInfo->userName, userInfo->userId); 60                  61                 char loginInfo[4] = {
   0}; 62                 memcpy(loginInfo, &LoginSuc, 2); 63                  64                 int i = send(cliSocket, loginInfo, strlen(loginInfo), 0); 65                  66             } 67             break; 68         case MesNormal: 69             { 70                 //printf("The Normal Message !"); 71                 char toName[32] = {
   0}; 72                 memcpy(toName, recvBuf+2, 32); 73                 cout<
          
           <
           
            userName, strlen(toName))) 89 { 90 SOCKET toSock = tempUser->userSock; 91 send(toSock, sendMes, strlen(sendMes)+1, 0); 92 sendFlag = true; 93 } 94 tempUser = tempUser->pNext; 95 } 96 if(!sendFlag) 97 { 98 printf("Send Failed !\n"); 99 }100 101 }102 break;103 case MesToAll:104 {105 char Message[1000] = { 0};106 memcpy(Message, recvBuf+2, sizeof(recvBuf)-2);107 UserInfo* tempUser = gUserInfo;108 char sendMes[1002];109 memcpy(sendMes, &MesToAll, 2);110 memcpy(sendMes+2, Message, sizeof(Message));111 while(tempUser)112 {113 SOCKET toSock = tempUser->userSock;114 int i = send(toSock, sendMes, strlen(sendMes)+1, 0);115 if(i > 0)116 {117 printf("Send To %s Success !\n", tempUser->userName);118 }119 tempUser = tempUser->pNext;120 }121 }122 break;123 case MesError:124 //printf("The Error Message !");125 break; 126 case OnLineUser:127 {128 129 char onLineUserBuff[1000] = { 0};130 UserInfo* tempUser = gUserInfo;131 132 memcpy(onLineUserBuff, &OnLineUser, 2);133 134 int i = 0;135 string userStr = "";136 while(tempUser)137 {138 userStr += charToString(tempUser->userName);139 //处理方式有问题140 //memcpy(onLineUserBuff+2, tempUser->userName, 32);141 //send(cliSocket, onLineUserBuff, sizeof(onLineUserBuff), 0);142 tempUser = tempUser->pNext;143 }144 memcpy(onLineUserBuff+2, userStr.c_str(), userStr.length());145 send(cliSocket, onLineUserBuff, sizeof(onLineUserBuff), 0);146 147 }148 break;149 default:150 //printf("The Default Process !");151 break;152 }153 154 }155 156 157 return 0;158 }159 160 int main()161 {162 WORD wVersionRequested;163 WSADATA wsaData;164 int err;165 166 wVersionRequested = MAKEWORD(2, 2);167 168 err = WSAStartup(wVersionRequested, &wsaData);169 if (err != 0) {170 printf("WSAStartup failed with error: %d\n", err);171 return 1;172 }173 174 if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2) {175 /* Tell the user that we could not find a usable */176 /* WinSock DLL. */177 printf("Could not find a usable version of Winsock.dll\n");178 WSACleanup();179 return 1;180 }181 else182 printf("The Winsock 2.2 dll was found okay\n");183 184 185 SOCKET servSock = socket(AF_INET, SOCK_STREAM, 0);186 187 SOCKADDR_IN addrServ ;188 addrServ.sin_family = AF_INET;189 addrServ.sin_port = htons(6000);190 addrServ.sin_addr.S_un.S_addr = htonl(INADDR_ANY);191 192 bind(servSock, (SOCKADDR*)&addrServ, sizeof(SOCKADDR));193 194 if(listen(servSock, 20) == SOCKET_ERROR)195 {196 printf("listen failed with error: %ld\n", WSAGetLastError());197 closesocket(servSock);198 WSACleanup();199 return -1;200 }201 202 203 while(true)204 {205 //wait for users Login, and create thread for it206 SOCKADDR_IN cliAddr;207 int length = sizeof(SOCKADDR);208 209 SOCKET cliSock = accept(servSock, (SOCKADDR*)&cliAddr, &length);210 211 if(INVALID_SOCKET == cliSock)212 {213 printf("listen failed with error: %ld\n", WSAGetLastError());214 closesocket(cliSock);215 WSACleanup();216 return -1;217 }218 219 UserInfo* user = new UserInfo;220 221 user->userSock = cliSock;222 user->userAddr = cliAddr.sin_addr.S_un.S_addr;223 224 //cout<
            
             userName)<