RecvDlg.cpp
上传用户:fengliqin8
上传日期:2007-04-13
资源大小:28k
文件大小:8k
源码类别:

进程与线程

开发平台:

Visual C++

  1. // RecvDlg.cpp : implementation file
  2. //
  3. #include "stdafx.h"
  4. #include "Recv.h"
  5. #include "RecvDlg.h"
  6. #include "SysUtils.h"
  7. #ifdef _DEBUG
  8. #define new DEBUG_NEW
  9. #undef THIS_FILE
  10. static char THIS_FILE[] = __FILE__;
  11. #endif
  12. /////////////////////////////////////////////////////////////////////////////
  13. // CAboutDlg dialog used for App About
  14. class CAboutDlg : public CDialog
  15. {
  16. public:
  17. CAboutDlg();
  18. // Dialog Data
  19. //{{AFX_DATA(CAboutDlg)
  20. enum { IDD = IDD_ABOUTBOX };
  21. //}}AFX_DATA
  22. // ClassWizard generated virtual function overrides
  23. //{{AFX_VIRTUAL(CAboutDlg)
  24. protected:
  25. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  26. //}}AFX_VIRTUAL
  27. // Implementation
  28. protected:
  29. //{{AFX_MSG(CAboutDlg)
  30. //}}AFX_MSG
  31. DECLARE_MESSAGE_MAP()
  32. };
  33. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  34. {
  35. //{{AFX_DATA_INIT(CAboutDlg)
  36. //}}AFX_DATA_INIT
  37. }
  38. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  39. {
  40. CDialog::DoDataExchange(pDX);
  41. //{{AFX_DATA_MAP(CAboutDlg)
  42. //}}AFX_DATA_MAP
  43. }
  44. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  45. //{{AFX_MSG_MAP(CAboutDlg)
  46. // No message handlers
  47. //}}AFX_MSG_MAP
  48. END_MESSAGE_MAP()
  49. /////////////////////////////////////////////////////////////////////////////
  50. // CRecvDlg dialog
  51. CRecvDlg::CRecvDlg(CWnd* pParent /*=NULL*/)
  52. : CDialog(CRecvDlg::IDD, pParent)
  53. {
  54. //{{AFX_DATA_INIT(CRecvDlg)
  55. // NOTE: the ClassWizard will add member initialization here
  56. //}}AFX_DATA_INIT
  57. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  58. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  59. m_pTcpServer1 = new CTcpServer(this);
  60. }
  61. CRecvDlg::~CRecvDlg(void)
  62. {
  63. delete m_pTcpServer1;
  64. m_pTcpServer1 = NULL;
  65. }
  66. void CRecvDlg::DoDataExchange(CDataExchange* pDX)
  67. {
  68. CDialog::DoDataExchange(pDX);
  69. //{{AFX_DATA_MAP(CRecvDlg)
  70. DDX_Control(pDX, IDC_CONN_COUNT, m_ctlCnnCount);
  71. DDX_Control(pDX, IDC_INFO1, m_ctlInfo1);
  72. //}}AFX_DATA_MAP
  73. }
  74. BEGIN_MESSAGE_MAP(CRecvDlg, CDialog)
  75. //{{AFX_MSG_MAP(CRecvDlg)
  76. ON_WM_SYSCOMMAND()
  77. ON_WM_PAINT()
  78. ON_WM_QUERYDRAGICON()
  79. ON_BN_CLICKED(IDC_CANCEL_RECV, OnCancelRecv)
  80. ON_BN_CLICKED(IDC_CLOSE_CNN, OnCloseCnn)
  81. ON_WM_DESTROY()
  82. //}}AFX_MSG_MAP
  83. END_MESSAGE_MAP()
  84. /////////////////////////////////////////////////////////////////////////////
  85. // CRecvDlg message handlers
  86. BOOL CRecvDlg::OnInitDialog()
  87. {
  88. CDialog::OnInitDialog();
  89. // Add "About..." menu item to system menu.
  90. // IDM_ABOUTBOX must be in the system command range.
  91. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  92. ASSERT(IDM_ABOUTBOX < 0xF000);
  93. CMenu* pSysMenu = GetSystemMenu(FALSE);
  94. if (pSysMenu != NULL)
  95. {
  96. CString strAboutMenu;
  97. strAboutMenu.LoadString(IDS_ABOUTBOX);
  98. if (!strAboutMenu.IsEmpty())
  99. {
  100. pSysMenu->AppendMenu(MF_SEPARATOR);
  101. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  102. }
  103. }
  104. // Set the icon for this dialog.  The framework does this automatically
  105. //  when the application's main window is not a dialog
  106. SetIcon(m_hIcon, TRUE); // Set big icon
  107. SetIcon(m_hIcon, FALSE); // Set small icon
  108. m_pTcpServer1->SetBindAddr("");
  109. m_pTcpServer1->SetPort(8000);
  110. m_pTcpServer1->SetOnAccept(OnAccept);
  111. m_pTcpServer1->SetOnAcceptErr(OnAcceptErr);
  112. m_pTcpServer1->SetOnSocketConnect(OnSocketConnect);
  113. m_pTcpServer1->SetOnSocketDisconnect(OnSocketDisconnect);
  114. m_pTcpServer1->SetOnSocketSendErr(OnSocketSendErr);
  115. m_pTcpServer1->SetOnSocketRecvErr(OnSocketRecvErr);
  116. m_pTcpServer1->SetOnOneNetMsg(OnOneNetMsg);
  117. m_pTcpServer1->SetOnRecvFileStart(OnRecvFileStart);
  118. m_pTcpServer1->SetOnRecvFileProgress(OnRecvFileProgress);
  119. m_pTcpServer1->SetOnRecvFileFail(OnRecvFileFail);
  120. m_pTcpServer1->SetOnRecvFileSucc(OnRecvFileSucc);
  121. m_pTcpServer1->SetOnRecvFileCancel(OnRecvFileCancel);
  122. if(!m_pTcpServer1->StartAccept())
  123. {
  124. AfxMessageBox("开始服务失败");
  125. return FALSE;
  126. }
  127. return TRUE;  // return TRUE  unless you set the focus to a control
  128. }
  129. void CRecvDlg::OnSysCommand(UINT nID, LPARAM lParam)
  130. {
  131. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  132. {
  133. CAboutDlg dlgAbout;
  134. dlgAbout.DoModal();
  135. }
  136. else
  137. {
  138. CDialog::OnSysCommand(nID, lParam);
  139. }
  140. }
  141. // If you add a minimize button to your dialog, you wil need the code below
  142. //  to draw the icon.  For MFC applications using the document/view model,
  143. //  this is automatically done for you by the framework.
  144. void CRecvDlg::OnPaint() 
  145. {
  146. if (IsIconic())
  147. {
  148. CPaintDC dc(this); // device context for painting
  149. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  150. // Center icon in client rectangle
  151. int cxIcon = GetSystemMetrics(SM_CXICON);
  152. int cyIcon = GetSystemMetrics(SM_CYICON);
  153. CRect rect;
  154. GetClientRect(&rect);
  155. int x = (rect.Width() - cxIcon + 1) / 2;
  156. int y = (rect.Height() - cyIcon + 1) / 2;
  157. // Draw the icon
  158. dc.DrawIcon(x, y, m_hIcon);
  159. }
  160. else
  161. {
  162. CDialog::OnPaint();
  163. }
  164. }
  165. // The system calls this to obtain the cursor to display while the user drags
  166. //  the minimized window.
  167. HCURSOR CRecvDlg::OnQueryDragIcon()
  168. {
  169. return (HCURSOR) m_hIcon;
  170. }
  171. void CRecvDlg::OnAccept(void *pNotifyObj, SOCKET hSocket, BOOL &bAccept)
  172. {
  173. CRecvDlg *pRecvDlg = (CRecvDlg *)pNotifyObj;
  174. CString strInfo;
  175. strInfo.Format("OnAccept-%d", hSocket);
  176. pRecvDlg->DispInfo(strInfo);
  177. }
  178. void CRecvDlg::OnAcceptErr(void *pNotifyObj, SOCKET hAccept)
  179. {
  180. CRecvDlg *pRecvDlg = (CRecvDlg *)pNotifyObj;
  181. CString strInfo;
  182. strInfo.Format("OnAcceptErr-%d", hAccept);
  183. pRecvDlg->DispInfo(strInfo);
  184. }
  185. void CRecvDlg::OnOneNetMsg(void *pNotifyObj, char *Msg, int nMsgLen)
  186. {
  187. CRecvDlg *pRecvDlg = (CRecvDlg *)pNotifyObj;
  188. CString strInfo;
  189. char s[10240];
  190. memcpy(s, Msg, nMsgLen);
  191. s[nMsgLen] = 0;
  192. strInfo = (LPCTSTR)s;
  193. pRecvDlg->DispInfo(strInfo);
  194. }
  195. void CRecvDlg::OnRecvFileStart(void *pNotifyObj, char *szPathName, BOOL &bRecv)
  196. {
  197. }
  198. void CRecvDlg::OnRecvFileProgress(void *pNotifyObj, DWORD dwRecvedBytes, DWORD dwFileSize)
  199. {
  200. CRecvDlg *pRecvDlg = (CRecvDlg *)pNotifyObj;
  201. CString strInfo;
  202. strInfo.Format("%d / %d", dwRecvedBytes, dwFileSize);
  203. pRecvDlg->DispInfo(strInfo);
  204. }
  205. void CRecvDlg::OnRecvFileSucc(void *pNotifyObj, char *szPathName)
  206. {
  207. CRecvDlg *pRecvDlg = (CRecvDlg *)pNotifyObj;
  208. CString strInfo = "OnRecvFileSucc";
  209. pRecvDlg->DispInfo(strInfo);
  210. }
  211. void CRecvDlg::OnRecvFileFail(void *pNotifyObj, char *szPathName)
  212. {
  213. CRecvDlg *pRecvDlg = (CRecvDlg *)pNotifyObj;
  214. CString strInfo = "OnRecvFileFail";
  215. pRecvDlg->DispInfo(strInfo);
  216. }
  217. void CRecvDlg::OnRecvFileCancel(void *pNotifyObj, char *szPathName)
  218. {
  219. CRecvDlg *pRecvDlg = (CRecvDlg *)pNotifyObj;
  220. CString strInfo = "OnRecvFileCancel";
  221. pRecvDlg->DispInfo(strInfo);
  222. }
  223. void CRecvDlg::DispInfo(CString &strInfo)
  224. {
  225. m_ctlInfo1.SetWindowText(strInfo);
  226. }
  227. void CRecvDlg::DispCnnCount(void)
  228. {
  229. CString strCnnCount;
  230. strCnnCount.Format("%d", m_pTcpServer1->GetClientCount());
  231. m_ctlCnnCount.SetWindowText(strCnnCount);
  232. }
  233. void CRecvDlg::OnCancelRecv() 
  234. {
  235. m_pTcpServer1->CancelAllRecvFile();
  236. }
  237. void CRecvDlg::OnSocketConnect(void *pNotifyObj, SOCKET hSocket)
  238. {
  239. CRecvDlg *pRecvDlg = (CRecvDlg *)pNotifyObj;
  240. CString strInfo;
  241. strInfo.Format("OnSocketConnect-%d", hSocket);
  242. pRecvDlg->DispInfo(strInfo);
  243. pRecvDlg->DispCnnCount();
  244. }
  245. void CRecvDlg::OnSocketDisconnect(void *pNotifyObj, SOCKET hSocket)
  246. {
  247. CRecvDlg *pRecvDlg = (CRecvDlg *)pNotifyObj;
  248. CString strInfo;
  249. strInfo.Format("OnSocketDisconnect-%d", hSocket);
  250. pRecvDlg->DispInfo(strInfo);
  251. pRecvDlg->DispCnnCount();
  252. }
  253. void CRecvDlg::OnSocketSendErr(void *pNotifyObj, CServerClientSocket *pServerClientSocket)
  254. {
  255. CRecvDlg *pRecvDlg = (CRecvDlg *)pNotifyObj;
  256. CString strInfo = "OnSocketSendErr";
  257. pRecvDlg->DispInfo(strInfo);
  258. }
  259. void CRecvDlg::OnSocketRecvErr(void *pNotifyObj, CServerClientSocket *pServerClientSocket)
  260. {
  261. CRecvDlg *pRecvDlg = (CRecvDlg *)pNotifyObj;
  262. CString strInfo = "OnSocketRecvErr";
  263. pRecvDlg->DispInfo(strInfo);
  264. }
  265. void CRecvDlg::OnCloseCnn() 
  266. {
  267. }
  268. void CRecvDlg::OnDestroy() 
  269. {
  270. CDialog::OnDestroy();
  271. m_pTcpServer1->CloseAllServerClientSocket();
  272. }