winsocket main.cpp
#include "main.h"
/* Declare Dialog procedure */
BOOL CALLBACK DlgProc(HWND, UINT, WPARAM, LPARAM);
/* Make the class name into a global variable */
HINSTANCE hinst;
HMENU menu;
BOOL server;
int WINAPI WinMain (HINSTANCE hThisInstance,
HINSTANCE hPrevInstance,
LPSTR lpszArgument,
int nCmdShow)
{
HWND hwnd; /* This is the handle for our window */
MSG messages; /* Here messages to the application are saved */
hinst=hThisInstance;
hwnd = CreateDialog(hThisInstance,
MAKEINTRESOURCE(IDD_DEMO),//identifies dialog box template name
0,//HWND_DESKTOP,// handle to owner window
(DLGPROC)DlgProc// pointer to dialog box procedure
);
if (!hwnd)
{
char buf[100];
//formats and stores a series of characters and values in a buffer
wsprintf (buf,"Error x%x", GetLastError());
MessageBox(0,buf,szDlgTitle,MB_ICONEXCLAMATION | MB_OK);
return 1;
}
/* Run the message loop. It will run until GetMessage() returns 0 */
int status;
while ((status=GetMessage (&messages, NULL, 0, 0))!=0)
{
if(!IsDialogMessage(hwnd,&messages))
{
if (status==-1)
return -1;
/* Translate virtual-key messages into character messages */
TranslateMessage(&messages);
/* Send message to WindowProcedure */
DispatchMessage(&messages);
}
}
/* The program return-value is 0 - The value that PostQuitMessage() gave */
return messages.wParam;
}
/* This function is called by the Windows function DispatchMessage() */
BOOL CALLBACK DlgProc(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
{
HWND hSend;
switch (message) /* handle the messages */
{
case WM_INITDIALOG:
menu = LoadMenu(hinst, MAKEINTRESOURCE(ID_MENU));
SetMenu(hwnd, menu);
hSend=GetDlgItem(hwnd,IDB_SEND);
//SetWindowText(hSend,"dsds" );
SetDlgItemText(hwnd,IDB_LC,"LC" );
EnableWindow(GetDlgItem(hwnd,IDB_LC),FALSE);
EnableWindow(GetDlgItem(hwnd,IDC_IP),FALSE);
EnableWindow(GetDlgItem(hwnd,IDC_PORT),FALSE);
SetDlgItemText(hwnd,IDC_IP,"192.168.10.4");
SetDlgItemText(hwnd,IDC_PORT,"7007" );
//创建一个计时器
SetTimer(hwnd,1,1000,NULL);
//SetFocus(GetDlgItem(hwnd,IDC_SENDTXT));
break;
case WM_COMMAND:
switch(wParam)
{
case IDM_FILENEW:
HttpClient();
break;
case IDM_FILEOPEN:
case IDM_FILELISTEN:
server=TRUE;
SetDlgItemText(hwnd,IDB_LC,"监听(&L)" );
SetWindowText(hwnd,"服务器端" );
EnableWindow(GetDlgItem(hwnd,IDB_LC),TRUE);
EnableWindow(GetDlgItem(hwnd,IDC_IP),FALSE);
EnableWindow(GetDlgItem(hwnd,IDC_PORT),FALSE);
EnableMenuItem(menu,IDM_FILECONNECT,MF_GRAYED);
break;
case IDM_FILECONNECT:
server=FALSE;
SetDlgItemText(hwnd,IDB_LC,"连接服务器(&C)" );
SetWindowText(hwnd,"客户端" );
EnableWindow(GetDlgItem(hwnd,IDB_LC),TRUE);
EnableWindow(GetDlgItem(hwnd,IDC_IP),TRUE);
EnableWindow(GetDlgItem(hwnd,IDC_PORT),TRUE);
EnableMenuItem(menu,IDM_FILELISTEN,MF_GRAYED);
break;
case IDM_EDITUNDO:
case IDM_EDITCUT:
case IDM_EDITCOPY:
case IDM_EDITPASTE:
case IDM_EDITDELETE:
case IDM_TRANSPORTFILE:
MessageBox( hwnd, (LPSTR) "功能没有实现",
(LPSTR)szDlgTitle,
MB_ICONINFORMATION | MB_OK );
break;
case IDM_HELPHELP:
WinHelp( hwnd, (LPSTR) "HELPFILE.HLP",
HELP_HELPONHELP, 0L );
break;
case IDM_FILEEXIT:
SendMessage( hwnd, WM_CLOSE, 0, 0L );
break;
case IDM_HELPABOUT:
MessageBox(hwnd,"欢迎使用聊天程序1.0版/n/n**科技大学胡子/n2004(C)版权所有 " ,
(LPSTR)szDlgTitle,
MB_ICONINFORMATION | MB_OK );
break;
case IDM_HELPMYBLOG:
ShellExecute(hwnd,"open","http://blog.csdn.net/huyoo",NULL,NULL,SW_NORMAL);
break;
case IDCANCEL:
//MessageBox(hwnd,"你单击了退出, 对话框将会关闭./n/n再见!",szDlgTitle,0);
EndDialog(hwnd,TRUE);
SendMessage( hwnd, WM_CLOSE, 0, 0L );
break;
case IDB_SEND:
char szText[128],buf[256];
GetDlgItemText(hwnd,IDC_SENDTXT,szText,sizeof(szText));
wsprintf(buf,"你单击了发送消息./n/n你输入的文本是:%s",szText);
if(server)
{
if( send(connected_skt,buf,strlen(buf),0) == SOCKET_ERROR ) //客户端是阻塞式的
{
MessageBox(hwnd, "发送消息出错!",szDlgTitle,MB_OK);
break;
}
}
else
{
if( send(skt,buf,strlen(buf),0) == SOCKET_ERROR ) //客户端是阻塞式的
{
MessageBox(hwnd, "发送消息出错!",szDlgTitle,MB_OK);
break;
}
//recv(skt,buf,strlen(buf),0); //它将会停在这里,直到接到数据为止
}
SetDlgItemText(hwnd,IDC_REVTXT,szText);
break;
case IDB_QQ:
ShellExecute(hwnd,"open","qq.exe",NULL,NULL,SW_NORMAL);
break;
case IDB_LC:
if(server)
{
//MessageBox(hwnd,"listen",szDlgTitle,MB_OK);
if(!CreateServer(hwnd))
{
MessageBox(hwnd,"服务器创建失败",(LPSTR)szDlgTitle,MB_ICONINFORMATION | MB_OK );
break;
}
}
else
{
//MessageBox(hwnd,"connect",szDlgTitle,MB_OK);
if(!CreateClient(hwnd))
{
MessageBox(hwnd,"客户端创建失败",(LPSTR)szDlgTitle,MB_ICONINFORMATION | MB_OK );
break;
}
char szIP[127],szPORT[127],buf[256];
GetDlgItemText(hwnd,IDC_IP,szIP,sizeof(szIP));
GetDlgItemText(hwnd,IDC_PORT,szPORT,sizeof(szPORT));
wsprintf(buf,">>>>>>连接到服务器:%s端口:%s",szIP,szPORT);
SetDlgItemText(hwnd,IDC_REVTXT,buf);
}
break;
}
break;
case WM_TIMER:
SYSTEMTIME ti;
char buf[256];
GetLocalTime(&ti);
wsprintf(buf,"现在时间是%04d年%02d月%02d日%02d:%02d:%02d,周%d",
ti.wYear,ti.wMonth,ti.wDay,ti.wHour,ti.wMinute,ti.wSecond,ti.wDayOfWeek);
SetWindowText(GetDlgItem(hwnd,IDC_TIME),buf);
break;
case SOCKETMSG:
//if(WSAGETSELECTERROR(lParam))
//break;
switch(WSAGETSELECTEVENT(lParam))
{
case FD_READ:
char buf[256];
recv(connected_skt,buf,strlen(buf),0);
SetDlgItemText(hwnd,IDC_REVTXT,buf);//显示接受到的消息
break;
case FD_WRITE:
break;
case FD_CLOSE:
closesocket(connected_skt);
break;
case FD_CONNECT:
SetDlgItemText(hwnd,IDC_REVTXT,"已经连接上");//显示接受到的消息
break;
case FD_ACCEPT:
int Len = sizeof(clientaddr);
//建立连接,得到的地址存入客户端clientaddr地址中去
if((connected_skt=accept(skt,(struct sockaddr *)&clientaddr,&Len))== INVALID_SOCKET )
{
MessageBox(hwnd,"接受客户端的Socket连接失败", szDlgTitle,MB_OK);
return FALSE;
}
//连接上了,然后监听客户端的FD_READ和关闭
WSAAsyncSelect(connected_skt, hwnd,SOCKETMSG,FD_READ|FD_CLOSE);
char *welcome="欢迎来到聊天室!!";
send(wParam, welcome, strlen(welcome), 0);
break;
}
SetDlgItemText(hwnd,IDC_REVTXT,"客户端发送消息过来了");//显示接受到的消息
break;
case WM_CLOSE:
if(connected_skt != INVALID_SOCKET)
{
closesocket(connected_skt);
}
if( skt != INVALID_SOCKET )
{
closesocket(skt);
}
if( WSACleanup() != 0 )
{
MessageBox(hwnd, "不能释放Winsocket",szDlgTitle,MB_OK );
}
KillTimer(hwnd,1);
DestroyWindow(hwnd);
break;
case WM_DESTROY:
PostQuitMessage (0); /* send a WM_QUIT to the message queue */
break;
default:
return FALSE;
}
return TRUE;
}