|
] = { 0 }; char chatName[16] = { 0 }; int iStatus = RECV_YET; //接受数据 unsigned __stdcall ThreadRecv(void* param) { char buf[128] = { 0 }; while (1) { int ret = recv(*(SOCKET*)param, buf, sizeof(buf), 0); if (ret == SOCKET_ERROR) { Sleep(500); continue; } if (strlen(buf) != 0) { printf("%s\n", buf); iStatus = RECV_OVER; } else Sleep(100); } return 0; } //发送数据 unsigned __stdcall ThreadSend(void* param) { char buf[128] = { 0 }; int ret = 0; while (1) { int c = getch(); if (c == 27) //ESC ASCII是27 { memset(buf, 0, sizeof(buf)); printf("Please input the chat name:"); gets_s(buf); char b[17] = { 0 }; sprintf(b, "#%s", buf); ret = send(*(SOCKET*)param,b , sizeof(b), 0); if (ret == SOCKET_ERROR) return 1; continue; } if(c == 72 || c == 0 || c == 68)//为了显示美观，加一个无回显的读取字符函数 continue; //getch返回值我是经过实验得出如果是返回这几个值，则getch就会自动跳过，具体我也不懂。 printf("%s: ", userName); gets_s(buf); ret = send(*(SOCKET*)param, buf, sizeof(buf), 0); if (ret == SOCKET_ERROR) return 1; } return 0; } //连接服务器 int ConnectServer() { WSADATA wsaData = { 0 };//存放套接字信息 SOCKET ClientSocket = INVALID_SOCKET;//客户端套接字 SOCKADDR_IN ServerAddr = { 0 };//服务端地址 USHORT uPort = 18000;//服务端端口 //初始化套接字 if (WSAStartup(MAKEWORD(2, 2), &wsaData)) { printf("WSAStartup failed with error code: %d\n", WSAGetLastError()); return -1; } //判断套接字版本 if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2) { printf("wVersion was not 2.2\n"); return -1; } //创建套接字 ClientSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); if (ClientSocket == INVALID_SOCKET) { printf("socket failed with error code: %d\n", WSAGetLastError()); return -1; } //输入服务器IP printf("Please input server IP:"); char IP[32] = { 0 }; gets_s(IP); //设置服务器地址 ServerAddr.sin_family = AF_INET; ServerAddr.sin_port = htons(uPort);//服务器端口 ServerAddr.sin_addr.S_un.S_addr = inet_addr(IP);//服务器地址 printf("connecting......\n"); //连接服务器 if (SOCKET_ERROR == connect(ClientSocket, (SOCKADDR*)&ServerAddr, sizeof(ServerAddr))) { printf("connect failed with error code: %d\n", WSAGetLastError()); closesocket(ClientSocket); WSACleanup(); return -1; } printf("Connecting server successfully IP:%s Port:%d\n", IP, htons(ServerAddr.sin_port)); printf("Please input your UserName: "); gets_s(userName); send(ClientSocket, userName, sizeof(userName), 0); printf("Please input the ChatName: "); gets_s(chatName); send(ClientSocket, chatName, sizeof(chatName), 0); printf("\n\n"); _beginthreadex(NULL, 0, ThreadRecv, &ClientSocket, 0, NULL); //启动接收和发送消息线程 _beginthreadex(NULL, 0, ThreadSend, &ClientSocket, 0, NULL); for (int k = 0;k < 1000;k++) Sleep(10000000); closesocket(ClientSocket); WSACleanup(); return 0; } int main() { ConnectServer(); //连接服务器 return 0; }
最后，需要改进的有以下几点：
1.没有消息记录，所以最好用文件或者数据库的方式记录，个人推荐数据库。
2.没有用户注册，登陆的操作，也是用文件或者数据库来弄。程序一运行就读取数据库信息就行。
3.群聊功能没有弄，这个其实很简单，就是服务器不管3721，把接收到的消息转发给所有在线用户。
4.没有离线消息，这个就用数据库存储离线消息，然后用户上线后立即发送过去就行。
最后总结一下，没有数据库的聊天程序果然功能简陋~，C语言写的程序要注意对内存的操作。还有TCP方式的连接太费时费内存(用户量达的时候)。
|