用一个死循环来监视就可以了,不用窗口消息来触发.
///////////////////////////////////////////////////////////////////
//Name:Server.cpp
//Description:Chatting server implement
//
//Author:
//Last Update:2004/6/28
///////////////////////////////////////////////////////////////////
#include "stdafx.h"
#include "winsock.h"
#include "windows.h"
#pragma comment(lib,"ws2_32")
#define MAX_USER 20
#define MAX_MESSAGE_LENGTH 256
#define ERR -1
#define OK 0
//User information
struct _USERINFO
{
char ip[32]; //IP address
char user[16]; //User name
bool IsAlive; //online signature true:online, false :not online
int port; //listening port of the user
}USERINFO[MAX_USER];
typedef struct _MESSAGE
{
char ip[32]; //User's IP address
int port; //listing port of the source user
char user[32]; //Source user's name
char dest[32]; //Destination user's name
char content[256]; //The content of the message
char attribute[32]; //The attribute of the message
}MESSAGE;
MESSAGE message;
int nUserCounter=0;
int ListenConnect();
int SendMsg(char* ip,int port,char* attribute,char* content);
unsigned long _stdcall ListenThread(void* pParam);
int QueryUser(char* user);
int AddUser(char* user,char* ip,int port,bool IsAlive);
int DelUser(char* user,char* ip,bool IsAlive);
char* QueryIP(char* user);
int QueryPort(char* user);
void BroadCastMsg(char* content);
int main(int argc, char* argv[])
{
printf("/////////////////////////////////////////////////////\r\n");
printf("// Welcome to use howtotell chatting server. //\r\n");
printf("// Copyright 2004 All right reserved. //\r\n");
printf("/////////////////////////////////////////////////////\r\n\r\n");
printf("The server started successfully.\r\nNow listening...\r\n");
CreateThread(NULL,0,&ListenThread,0,0,0);
Sleep(1000);
char cmd[32];
memset(cmd,0,32);
while(strcmp(cmd,"stop")!=0)
{
gets(cmd);
}
return OK;
}
int ListenConnect()
{
struct sockaddr_in sin;
WSADATA wsaData;
SOCKET sockSrvr,sockRecv;
if(WSAStartup(MAKEWORD(1, 1), &wsaData ))
{
printf("TCP/IP stack initialise error!\r\n");
return ERR;
}
if(wsaData.wVersion != MAKEWORD(1, 1))
{
printf("Winsock version is incorrect!\r\n");
WSACleanup();
return ERR;
}
sin.sin_addr.S_un.S_addr=INADDR_ANY;
sin.sin_family=AF_INET;
sin.sin_port=htons(8000);
if((sockSrvr = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP))==SOCKET_ERROR)
{
printf("sockSrvr error\r\n");
return ERR;
}
if((sockRecv = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP))==SOCKET_ERROR)
{
printf("sockRecv error\r\n");
return ERR;
}
if(sockSrvr==SOCKET_ERROR||sockRecv==SOCKET_ERROR)
{
printf("socket create error\r\n");
return ERR;
}
if(bind(sockSrvr,(SOCKADDR *)&sin,sizeof(sin))<0)
{
printf("bind error\r\n");
return ERR;
}
listen(sockSrvr,5);
int length=sizeof(sin);
sockRecv=accept(sockSrvr,(struct sockaddr*)&sin,&length);
if(sockRecv==INVALID_SOCKET)
{
printf("accept error \r\n");
return ERR;
}
memset(&message,0,sizeof(MESSAGE));
recv(sockRecv,(char*)&message,sizeof(MESSAGE),0);
//Broadcast message
if(strcmp(message.attribute,"public")==0)
{
BroadCastMsg(message.content);
closesocket(sockRecv);
closesocket(sockSrvr);
WSACleanup();
return OK;
}
//Specified user is existent and is on line
if(QueryUser(message.dest)!=-1)
{
printf("%s (%s)> %s \r\n",message.user,message.ip,message.content);
strcat(message.content,"<Server delivered>");
SendMsg(QueryIP(message.dest),QueryPort(message.dest),message.attribute,message.content);
}
else
{
if(strcmp(message.dest,"@HOWTOTELL_SERVER")==0)//Reserved user name,this is the handshake signal
{
if(strcmp(message.content,"HAND_SHAKE")==0)
{
AddUser(message.user,message.ip,message.port,true);
strcpy(message.content,"Welcome to howtotell chatting server!\r\n");
strcat(message.content,"////////////User List ////////////////\r\n");
for(int i=0;i<MAX_USER;i++)
{
if(strlen(USERINFO[i].user)>0)
{
strcat(message.content,USERINFO[i].user);
strcat(message.content,"\r\n");
}
}
strcat(message.content,"//////////////////////////////////////\r\n");
printf("%s (%s)> %s \r\n",message.user,message.ip,"Logged in");
BroadCastMsg(message.content);
}
else
{
DelUser(message.user,message.ip,false);
sprintf(message.content,
"Chatting server system message:\r\nUser leaved:\r\n%s %s\r\n",
message.user,message.ip);
printf("%s (%s)> %s \r\n",message.user,message.ip,"Logged Out");
BroadCastMsg(message.content);
}
}
else //user is not existent
{
printf("user %s is not on line now\r\n",message.dest);
SendMsg(message.ip,message.port,message.attribute,"User is not on line\r\n");
}
}
closesocket(sockRecv);
closesocket(sockSrvr);
WSACleanup();
return OK;
}
int SendMsg(char* ip,int port,char* attribute,char* content)
{
struct sockaddr_in sin;
WSADATA wsaData;
SOCKET sockClient;
char *IpAddr;
IpAddr=new char[64];
memset(IpAddr,0,64);
strcpy(IpAddr,ip);
if(WSAStartup(0x0101, &wsaData ))
{
printf("Init TCP/IP stack error ");
return ERR;
}
if(wsaData.wVersion != 0x0101)
{
printf("Winsock version error");
WSACleanup();
return ERR;
}
if ((sockClient = socket (PF_INET, SOCK_STREAM, IPPROTO_IP)) == INVALID_SOCKET)
{
printf("create socket error!");
return ERR;
}
sin.sin_family=AF_INET;
sin.sin_addr.s_addr=inet_addr(IpAddr);
sin.sin_port=htons((short)port);
if(connect (sockClient, (struct sockaddr *)&sin, sizeof (sin)) == SOCKET_ERROR)
{
printf("connect error");
closesocket(sockClient);
WSACleanup();
return ERR;
}
else
{
strcpy(message.content, content);
send(sockClient,(char*)&message,sizeof(MESSAGE),0);
}
closesocket (sockClient);
WSACleanup();
return OK;
}
unsigned long _stdcall ListenThread(void* pParam)
{
while(1)
{
if(ListenConnect()==ERR)
{
break;
}
}
return OK;
}
int QueryUser(char* user)
{
for(int i=0;i<MAX_USER;i++)
{
if(strcmp(user,USERINFO[i].user)==0&&USERINFO[i].IsAlive)
{
return i; //Specified user is existential and is on line
}
}
return ERR;
}
int AddUser(char* user,char* ip,int port,bool IsAlive)
{
if(QueryUser(user)!=-1)
{
return ERR;
}
//Specified user is not existential
for(int i=0;i<MAX_USER;i++)
{
if(strlen(USERINFO[i].user)==0||!USERINFO[i].IsAlive)
{
strcpy(USERINFO[i].user,user);
strcpy(USERINFO[i].ip,ip);
USERINFO[i].IsAlive=IsAlive;
USERINFO[i].port=port;
break;
}
}
return OK;
}
int DelUser(char* user,char* ip,bool IsAlive)
{
for(int i=0;i<MAX_USER;i++)
{
if(strcmp(USERINFO[i].user,user)==0)
{
strcpy(USERINFO[i].user,"");
USERINFO[i].IsAlive=false;
}
}
return OK;
}
char* QueryIP(char* user)
{
for(int i=0;i<MAX_USER;i++)
{
if(strcmp(USERINFO[i].user,user)==0)
{
return USERINFO[i].ip;
}
}
return "";
}
int QueryPort(char* user)
{
for(int i=0;i<MAX_USER;i++)
{
if(strcmp(USERINFO[i].user,user)==0)
{
return USERINFO[i].port;
}
}
return ERR;
}
void BroadCastMsg(char* content)
{
for(int i=0;i<MAX_USER;i++)
{
if(strlen(USERINFO[i].user)>0&&USERINFO[i].IsAlive)
{
SendMsg(USERINFO[i].ip,USERINFO[i].port,"public",content);
}
}
printf("Broadcasting Message %s(%s)>%s\r\n",message.user,message.ip,content);
}
