www.pudn.com > 截屏.rar > MyClientDlg.cpp
// MyClientDlg.cpp : implementation file
//
#include "stdafx.h"
#include "MyClient.h"
#include "MyClientDlg.h"
#pragma comment (lib,"Ws2_32.lib")
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
#define CMD_SHOW 3
#define CMD_CAPTURE 4
#define CMD_MSG 5
#define CMD_GET 7
#define CMD_LOCK 9
#define CMD_UNLOCK 10
#define SERVER_PORT 10007
struct InfoData{
char Info[4];
DWORD dwSize;
WORD nCommand;
WORD nReserved;
char Msg[200];
};
struct InfoData *pInfo,SendInfo;
HANDLE hDib;
u_long GetLocal_in_addr(void);
UINT MY_SOCK_SIZE=1024;
//此函数将指定计算机名字或IP地址转换为网络地址(网络字节格式)
u_long name2inet(LPCSTR address);
///此函数获得本机的本地网络地址(网络字节格式)
u_long GetLocal_in_addr(void);
/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About
class CAboutDlg : public CDialog
{
public:
CAboutDlg();
// Dialog Data
//{{AFX_DATA(CAboutDlg)
enum { IDD = IDD_ABOUTBOX };
//}}AFX_DATA
// ClassWizard generated virtual function overrides
//{{AFX_VIRTUAL(CAboutDlg)
protected:
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV support
//}}AFX_VIRTUAL
// Implementation
protected:
//{{AFX_MSG(CAboutDlg)
//}}AFX_MSG
DECLARE_MESSAGE_MAP()
};
CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
//{{AFX_DATA_INIT(CAboutDlg)
//}}AFX_DATA_INIT
}
void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
//{{AFX_DATA_MAP(CAboutDlg)
//}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
//{{AFX_MSG_MAP(CAboutDlg)
// No message handlers
//}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CMyClientDlg dialog
CMyClientDlg::CMyClientDlg(CWnd* pParent /*=NULL*/)
: CDialog(CMyClientDlg::IDD, pParent)
{
//{{AFX_DATA_INIT(CMyClientDlg)
m_Addr = _T("");
m_Msg = _T("");
//}}AFX_DATA_INIT
// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}
void CMyClientDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
//{{AFX_DATA_MAP(CMyClientDlg)
DDX_Text(pDX, IDC_EDIT1, m_Addr);
DDX_Text(pDX, IDC_EDIT2, m_Msg);
//}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CMyClientDlg, CDialog)
//{{AFX_MSG_MAP(CMyClientDlg)
ON_WM_SYSCOMMAND()
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
ON_BN_CLICKED(IDC_Exit, OnExit)
ON_BN_CLICKED(IDC_Lock, OnLock)
ON_BN_CLICKED(IDC_Capture, OnCapture)
ON_BN_CLICKED(IDC_Unlock, OnUnlock)
ON_BN_CLICKED(IDC_Send, OnSend)
//}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CMyClientDlg message handlers
BOOL CMyClientDlg::OnInitDialog()
{
CDialog::OnInitDialog();
// Add "About..." menu item to system menu.
// IDM_ABOUTBOX must be in the system command range.
ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
ASSERT(IDM_ABOUTBOX < 0xF000);
CMenu* pSysMenu = GetSystemMenu(FALSE);
if (pSysMenu != NULL)
{
CString strAboutMenu;
strAboutMenu.LoadString(IDS_ABOUTBOX);
if (!strAboutMenu.IsEmpty())
{
pSysMenu->AppendMenu(MF_SEPARATOR);
pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
}
}
// Set the icon for this dialog. The framework does this automatically
// when the application's main window is not a dialog
SetIcon(m_hIcon, TRUE); // Set big icon
SetIcon(m_hIcon, FALSE); // Set small icon
// TODO: Add extra initialization here
WORD WINSOCK_VERSION=0x0101;
WSADATA wsaData;
SendInfo.Info[0]='Y';
SendInfo.Info[1]='e';
SendInfo.Info[2]='s';
SendInfo.Info[3]='?';
if(WSAStartup(WINSOCK_VERSION,&wsaData))
AfxMessageBox("Cound not load Windows Socket DLL.");
return TRUE; // return TRUE unless you set the focus to a control
}
void CMyClientDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
if ((nID & 0xFFF0) == IDM_ABOUTBOX)
{
CAboutDlg dlgAbout;
dlgAbout.DoModal();
}
else
{
CDialog::OnSysCommand(nID, lParam);
}
}
// If you add a minimize button to your dialog, you will need the code below
// to draw the icon. For MFC applications using the document/view model,
// this is automatically done for you by the framework.
void CMyClientDlg::OnPaint()
{
if (IsIconic())
{
CPaintDC dc(this); // device context for painting
SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
// Center icon in client rectangle
int cxIcon = GetSystemMetrics(SM_CXICON);
int cyIcon = GetSystemMetrics(SM_CYICON);
CRect rect;
GetClientRect(&rect);
int x = (rect.Width() - cxIcon + 1) / 2;
int y = (rect.Height() - cyIcon + 1) / 2;
// Draw the icon
dc.DrawIcon(x, y, m_hIcon);
}
else
{
CDialog::OnPaint();
}
}
// The system calls this to obtain the cursor to display while the user drags
// the minimized window.
HCURSOR CMyClientDlg::OnQueryDragIcon()
{
return (HCURSOR) m_hIcon;
}
void CMyClientDlg::OnExit()
{
// TODO: Add your control notification handler code here
WSACleanup();
EndDialog(1);
}
//此函数将指定计算机名字或IP地址转换为网络地址(网络字节格式)
u_long name2inet(LPCSTR address)
{
if(address==NULL||strlen(address)<1){
return htonl(INADDR_ANY);
}
BOOL name=FALSE;
for (int i=0;address[i];i++){
if((!isdigit(address[i]))&&address[i]!='.'){
name=TRUE;break;
}
}
if(!name)return ::inet_addr(address);
else {
struct hostent*host=gethostbyname(address);
if(host==NULL){
TRACE("SERVER %s Can't Find Code=%d",address,WSAGetLastError());
return htonl(INADDR_ANY);
}
return *(long*)host->h_addr;
//*(long*)host->h_addr;
}
}
u_long GetLocal_in_addr(void)
///此函数获得本机的本地网络地址(网络字节格式)
{
char szHostname[256];
if (gethostname(szHostname, sizeof(szHostname)))
{
TRACE(_T("Failed in call to gethostname, WSAGetLastError returns %d\n"), WSAGetLastError());
return htonl(INADDR_ANY) ;
}
//get host information from the host name
HOSTENT* pHostEnt = gethostbyname(szHostname);
if (pHostEnt == NULL)
{
TRACE(_T("Failed in call to gethostbyname, WSAGetLastError returns %d\n"), WSAGetLastError());
return htonl(INADDR_ANY);
}
//check the length of the IP adress
if (pHostEnt->h_length != 4)
{
TRACE(_T("IP address returned is not 32 bits !!\n"));
return htonl(INADDR_ANY);
}
return
((struct in_addr *)*pHostEnt->h_addr_list)->s_addr;
}
void CMyClientDlg::OnCapture()
{
// TODO: Add your control notification handler code here
char *buf;
buf=new char[MY_SOCK_SIZE];
HANDLE hDib=NULL;
SOCKET clientSocket;
SOCKADDR_IN serverSocketAddr,clientSocketAddr;
memset(&serverSocketAddr,0,sizeof(SOCKADDR_IN));
memset(&clientSocketAddr,0,sizeof(SOCKADDR_IN));
UpdateData();//DDX/DDV 更新变量
serverSocketAddr.sin_addr.s_addr=name2inet(m_Addr);
serverSocketAddr.sin_family=AF_INET;
serverSocketAddr.sin_port=htons(SERVER_PORT);
clientSocketAddr.sin_family=AF_INET;
clientSocketAddr.sin_addr.s_addr=GetLocal_in_addr();
clientSocketAddr.sin_port=htons(1087);
int addrinSize=sizeof(SOCKADDR_IN);
if((clientSocket=socket(PF_INET,SOCK_DGRAM,IPPROTO_UDP))!=INVALID_SOCKET)
{
if(bind(clientSocket,(SOCKADDR*)&clientSocketAddr,sizeof(SOCKADDR_IN))!=SOCKET_ERROR)
{
BOOL fBroadCastEnable=1;
if(SOCKET_ERROR ==setsockopt(clientSocket,SOL_SOCKET,SO_BROADCAST ,(char FAR * )&fBroadCastEnable,sizeof(BOOL)))AfxMessageBox("Can't enable broadcast");
SendInfo.nCommand=CMD_CAPTURE;
BYTE *p,*p2;
sendto (clientSocket,(char*)&SendInfo,sizeof(SendInfo), 0, (SOCKADDR*)&serverSocketAddr, sizeof(SOCKADDR_IN) );
int len=recvfrom ( clientSocket,buf,MY_SOCK_SIZE, 0, (SOCKADDR*)&serverSocketAddr, &addrinSize );
//AfxMessageBox(inet_ntoa(clientSocketAddr.sin_addr));// 此语句用来测试能否自动填充clientSocketAddr结构，经过测试，是可以的
pInfo=(struct InfoData*)buf;
if(len>0&&pInfo->Info[0]=='Y'&&pInfo->Info[3]=='?')
{ DWORD dwImageSize=pInfo->dwSize;
if(hDib!=NULL){
GlobalFree (hDib);
}
if (hDib = GlobalAlloc( GMEM_MOVEABLE,dwImageSize))
{
p=(LPBYTE) GlobalLock(hDib);
p2=p;
SendInfo.nCommand=CMD_GET;
//发送命令，要求数据传送
sendto (clientSocket,(char*)&SendInfo,sizeof(SendInfo), 0, (SOCKADDR*)&serverSocketAddr, sizeof(SOCKADDR_IN) );
for(WORD i=0; iGetDC();
int ncolors = (1 <<((LPBITMAPINFOHEADER)p)->biBitCount); if( ncolors> 256 )
ncolors = 0;
GetDlgItem(IDC_PIC)->GetClientRect(rc);
::StretchDIBits(pDC->m_hDC ,
0,0,rc.Width(),rc.Height(),
0,0,((LPBITMAPINFOHEADER)p)->biWidth,((LPBITMAPINFOHEADER)p)->biHeight,
(LPBYTE)p+ (sizeof(BITMAPINFOHEADER) + ncolors * sizeof(RGBQUAD)),
(LPBITMAPINFO)p,
DIB_RGB_COLORS,
SRCCOPY);
///显示完毕，释放掉内存
GlobalUnlock(hDib);
GlobalFree(hDib);
}//end if allocmemory OK
}//end if (len>0
}//endif bind()
}//endif socket()
closesocket(clientSocket);//关闭套插口
delete buf;
}
void CMyClientDlg::OnLock()
{
// TODO: Add your control notification handler code here
char *buf;
buf=new char[MY_SOCK_SIZE];
SOCKET clientSocket;
SOCKADDR_IN serverSocketAddr,clientSocketAddr;
memset(&serverSocketAddr,0,sizeof(SOCKADDR_IN));
memset(&clientSocketAddr,0,sizeof(SOCKADDR_IN));
UpdateData();//DDX/DDV
serverSocketAddr.sin_addr.s_addr=name2inet(m_Addr);
serverSocketAddr.sin_family=AF_INET;
serverSocketAddr.sin_port=htons(SERVER_PORT);
clientSocketAddr.sin_family=AF_INET;
clientSocketAddr.sin_addr.s_addr=GetLocal_in_addr();
clientSocketAddr.sin_port=htons(1087);
int addrinSize=sizeof(SOCKADDR_IN);
if((clientSocket=socket(PF_INET,SOCK_DGRAM,IPPROTO_UDP))!=INVALID_SOCKET)
{
if(bind(clientSocket,(SOCKADDR*)&clientSocketAddr,sizeof(SOCKADDR_IN))!=SOCKET_ERROR)
{
BOOL fBroadCastEnable=1;
if(SOCKET_ERROR ==setsockopt(clientSocket,SOL_SOCKET,SO_BROADCAST ,(char FAR * )&fBroadCastEnable,sizeof(BOOL)))AfxMessageBox("Can't enable broadcast");
SendInfo.nCommand=CMD_LOCK;
//发送锁定计算机命令
sendto (clientSocket,(char*)&SendInfo,sizeof(SendInfo), 0, (SOCKADDR*)&serverSocketAddr, sizeof(SOCKADDR_IN) );
}//endif bind()
}//endif socket()
closesocket(clientSocket);
}
void CMyClientDlg::OnUnlock()
{
// TODO: Add your control notification handler code here
char *buf;
buf=new char[MY_SOCK_SIZE];
SOCKET clientSocket;
SOCKADDR_IN serverSocketAddr,clientSocketAddr;
memset(&serverSocketAddr,0,sizeof(SOCKADDR_IN));
memset(&clientSocketAddr,0,sizeof(SOCKADDR_IN));
UpdateData();//DDX/DDV
serverSocketAddr.sin_addr.s_addr=name2inet(m_Addr);
serverSocketAddr.sin_family=AF_INET;
serverSocketAddr.sin_port=htons(SERVER_PORT);
clientSocketAddr.sin_family=AF_INET;
clientSocketAddr.sin_addr.s_addr=GetLocal_in_addr();
clientSocketAddr.sin_port=htons(1087);
int addrinSize=sizeof(SOCKADDR_IN);
if((clientSocket=socket(PF_INET,SOCK_DGRAM,IPPROTO_UDP))!=INVALID_SOCKET)
{
if(bind(clientSocket,(SOCKADDR*)&clientSocketAddr,sizeof(SOCKADDR_IN))!=SOCKET_ERROR)
{
BOOL fBroadCastEnable=1;
if(SOCKET_ERROR ==setsockopt(clientSocket,SOL_SOCKET,SO_BROADCAST ,(char FAR * )&fBroadCastEnable,sizeof(BOOL)))AfxMessageBox("Can't enable broadcast");
SendInfo.nCommand=CMD_UNLOCK;
//发送解除锁定计算机命令
sendto (clientSocket,(char*)&SendInfo,sizeof(SendInfo), 0, (SOCKADDR*)&serverSocketAddr, sizeof(SOCKADDR_IN) );
}//endif bind()
}//endif socket()
closesocket(clientSocket);
}
void CMyClientDlg::OnSend()
{
// TODO: Add your control notification handler code here
char *buf;
buf=new char[MY_SOCK_SIZE];
SOCKET clientSocket;
SOCKADDR_IN serverSocketAddr,clientSocketAddr;
memset(&serverSocketAddr,0,sizeof(SOCKADDR_IN));
memset(&clientSocketAddr,0,sizeof(SOCKADDR_IN));
UpdateData();//DDX/DDV
serverSocketAddr.sin_addr.s_addr=name2inet(m_Addr);
serverSocketAddr.sin_family=AF_INET;
serverSocketAddr.sin_port=htons(SERVER_PORT);
clientSocketAddr.sin_family=AF_INET;
clientSocketAddr.sin_addr.s_addr=GetLocal_in_addr();
clientSocketAddr.sin_port=htons(1087);
int addrinSize=sizeof(SOCKADDR_IN);
if((clientSocket=socket(PF_INET,SOCK_DGRAM,IPPROTO_UDP))!=INVALID_SOCKET)
{
if(bind(clientSocket,(SOCKADDR*)&clientSocketAddr,sizeof(SOCKADDR_IN))!=SOCKET_ERROR)
{
BOOL fBroadCastEnable=1;
if(SOCKET_ERROR ==setsockopt(clientSocket,SOL_SOCKET,SO_BROADCAST ,(char FAR * )&fBroadCastEnable,sizeof(BOOL)))AfxMessageBox("Can't enable broadcast");
SendInfo.nCommand=CMD_MSG;
strcpy(SendInfo.Msg,m_Msg);
//发送消息
sendto (clientSocket,(char*)&SendInfo,sizeof(SendInfo), 0, (SOCKADDR*)&serverSocketAddr, sizeof(SOCKADDR_IN) );
}//endif bind()
}//endif socket()
closesocket(clientSocket);
}