开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > Ftp客户端 > 查看源码
PageGeneral.cpp
资源名称：Netmanag.zip　[点击查看]
上传用户：geanq888
上传日期：2007-01-03
资源大小：316k
文件大小：14k
源码类别：

Ftp客户端

开发平台：

Visual C++

PageGeneral.cpp：源码内容
  1. // PageGeneral.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "NetManager.h"
  6. #include "PageGeneral.h"
  7. #include "GlobalsExtern.h"
  9. #ifdef _DEBUG
  10. #define new DEBUG_NEW
  11. #undef THIS_FILE
  12. static char THIS_FILE[] = __FILE__;
  13. #endif
  15. /////////////////////////////////////////////////////////////////////////////
  16. // CPageGeneral property page
  18. IMPLEMENT_DYNCREATE(CPageGeneral, CPropertyPage)
  20. CPageGeneral::CPageGeneral() : CPropertyPage(CPageGeneral::IDD)
  21. {
  22. //{{AFX_DATA_INIT(CPageGeneral)
  23. m_sDisplayType = _T("");
  24. m_sLocalName = _T("");
  25. m_sProvider = _T("");
  26. m_sRemoteName = _T("");
  27. m_sScope = _T("");
  28. m_sType = _T("");
  29. m_sUsage = _T("");
  30. m_sComment = _T("");
  31. m_dwPingTimeout = 1000;
  32. m_bPingPacket = 32;
  33. //}}AFX_DATA_INIT
  34. }
  36. CPageGeneral::~CPageGeneral()
  37. {
  38. }
  40. void CPageGeneral::DoDataExchange(CDataExchange* pDX)
  41. {
  42. CPropertyPage::DoDataExchange(pDX);
  43. //{{AFX_DATA_MAP(CPageGeneral)
  44. DDX_Control(pDX, IDC_RESOLVE, m_Resolve);
  45. DDX_Control(pDX, IDC_PING, m_Ping);
  46. DDX_Control(pDX, IDC_ENUMERATE, m_Enumerate);
  47. DDX_Control(pDX, IDC_ADDRESS, m_Address);
  48. DDX_Control(pDX, IDC_TREE, m_Tree);
  49. DDX_Text(pDX, IDC_DISPLAY_TYPE, m_sDisplayType);
  50. DDX_Text(pDX, IDC_LOCAL_NAME, m_sLocalName);
  51. DDX_Text(pDX, IDC_PROVIDER, m_sProvider);
  52. DDX_Text(pDX, IDC_REMOTE_NAME, m_sRemoteName);
  53. DDX_Text(pDX, IDC_SCOPE, m_sScope);
  54. DDX_Text(pDX, IDC_TYPE, m_sType);
  55. DDX_Text(pDX, IDC_USAGE, m_sUsage);
  56. DDX_Text(pDX, IDC_COMMENT, m_sComment);
  57. DDX_Text(pDX, IDC_PING_TIMEOUT, m_dwPingTimeout);
  58. DDV_MinMaxDWord(pDX, m_dwPingTimeout, 0, 100000);
  59. DDX_Text(pDX, IDC_PING_PACKET, m_bPingPacket);
  60. //}}AFX_DATA_MAP
  61. }
  64. BEGIN_MESSAGE_MAP(CPageGeneral, CPropertyPage)
  65. //{{AFX_MSG_MAP(CPageGeneral)
  66. ON_BN_CLICKED(IDC_RESOLVE, OnResolve)
  67. ON_NOTIFY(TVN_ITEMEXPANDING, IDC_TREE, OnItemexpandingTree)
  68. ON_NOTIFY(TVN_SELCHANGED, IDC_TREE, OnSelchangedTree)
  69. ON_BN_CLICKED(IDC_ENUMERATE, OnEnumerate)
  70. ON_BN_CLICKED(IDC_PING, OnPing)
  71. ON_EN_CHANGE(IDC_ADDRESS, OnChangeAddress)
  72. //}}AFX_MSG_MAP
  73. END_MESSAGE_MAP()
  75. /////////////////////////////////////////////////////////////////////////////
  76. // CPageGeneral message handlers
  78. BOOL CPageGeneral::OnInitDialog() 
  79. {
  80. CPropertyPage::OnInitDialog();
  82.   m_ToolTip.Create(this);
  83.   m_ToolTip.Activate(TRUE);
  85.   CWnd* pWnd = GetWindow(GW_CHILD);
  86.   while(pWnd)
  87.   {
  88.     int nID = pWnd->GetDlgCtrlID();
  89.     if (nID != -1)
  90.       m_ToolTip.AddTool(pWnd, pWnd->GetDlgCtrlID());
  91.     pWnd = pWnd->GetWindow(GW_HWNDNEXT);
  92.   }
  94. return TRUE;  // return TRUE unless you set the focus to a control
  95.               // EXCEPTION: OCX Property Pages should return FALSE
  96. }
  98. /////////////////////////////////////////////////////////////////////////////
  100. UINT g_ResolveAddress(LPVOID pParam)
  101. {
  102.   CString sAddress = *((CString*)pParam);
  103.   delete ((CString*)pParam);
  105.   CString sResolved;
  106.   unsigned long ulIP;
  107.   hostent* pHostent;
  108.   sockaddr_in sin;
  110.   ulIP = inet_addr(sAddress);
  112.   if(ulIP != INADDR_NONE)
  113.   {
  114.     sin.sin_family = AF_INET;
  115.     sin.sin_addr.S_un.S_addr = ulIP;
  116.     pHostent = gethostbyaddr((char*)&sin.sin_addr, 4, PF_INET);
  117.     if(pHostent)
  118.       sResolved = pHostent->h_name;
  119.     else
  120.     {
  121.       g_WriteToOutput(TRUE, "[Resolve] Cannot resolve " + sAddress);
  122.       return 0;
  123.     }
  124.   }
  125.   else
  126.   {
  127.     in_addr inetAddr;
  129.     pHostent = gethostbyname(sAddress);
  130.     if(pHostent)
  131.     {
  132.       LPSTR szAddr;
  133.       ulIP = *(DWORD*)(*pHostent->h_addr_list);
  134.       inetAddr.s_addr = ulIP;
  135.       szAddr = inet_ntoa(inetAddr);
  136.       sResolved = szAddr;
  137.     }
  138.     else
  139.     {
  140.       g_WriteToOutput(TRUE, "[Resolve] Cannot resolve " + sAddress);
  141.       return 0;
  142.     }
  143.   }
  145.   g_WriteToOutput(TRUE, "[Resolve] " + sAddress + " -> " + sResolved);
  147.   return 0;
  148. }
  150. /////////////////////////////////////////////////////////////////////////////
  152. void CPageGeneral::OnResolve() 
  153. {
  154.   g_AnimateWait->Play(0, -1, -1);
  156.   CString sResolveAddress;
  157.   m_Address.GetWindowText(sResolveAddress);
  159.   CString* psAddress = new CString(sResolveAddress);
  160.   AfxBeginThread(g_ResolveAddress, psAddress);
  161. }
  163. /////////////////////////////////////////////////////////////////////////////
  165. void CPageGeneral::OnChangeAddress() 
  166. {
  167.   CString sText;
  168.   m_Address.GetWindowText(sText);
  169.   if(sText != "")
  170.   {
  171.     m_Resolve.EnableWindow(TRUE);
  172.     m_Enumerate.EnableWindow(TRUE);
  173.     m_Ping.EnableWindow(TRUE);
  174.     GetDlgItem(IDC_PING_PACKET)->EnableWindow(TRUE);
  175.     GetDlgItem(IDC_PING_TIMEOUT)->EnableWindow(TRUE);
  176.   }
  177.   else
  178.   {
  179.     m_Resolve.EnableWindow(FALSE);
  180.     m_Enumerate.EnableWindow(FALSE);
  181.     m_Ping.EnableWindow(FALSE);
  182.     GetDlgItem(IDC_PING_PACKET)->EnableWindow(FALSE);
  183.     GetDlgItem(IDC_PING_TIMEOUT)->EnableWindow(FALSE);
  184.   }
  185. }
  187. /////////////////////////////////////////////////////////////////////////////
  189. void CPageGeneral::CreateRootNode()
  190. {
  191.   m_pRootNode = new CNetNode;   // Create the root network node (use default constructor)
  192.   AddNodeToTree(m_pRootNode, TVI_ROOT);   // Add root node to tree
  193.   UpdateDisplay(NULL);
  194. }
  196. /////////////////////////////////////////////////////////////////////////////
  198. void CPageGeneral::AddNodeToTree(CNetNode* pNode, HTREEITEM parent)
  199. {
  200.   TV_INSERTSTRUCT    tvIns = { 0 };
  202.   tvIns.hParent = parent;
  203.   tvIns.hInsertAfter = TVI_SORT;
  204.   tvIns.item.mask = TVIF_TEXT | TVIF_CHILDREN | TVIF_PARAM;
  206.   // It is safe to cast away the const-ness here because
  207.   // we do not allow the tree labels to be edited
  208.   tvIns.item.pszText =
  209.     (char *) ((const char *)pNode->GetText());
  211.   if (pNode->IsContainer())
  212.     tvIns.item.cChildren = 1;
  213.   tvIns.item.lParam = (LPARAM) pNode;
  215.   HTREEITEM hItem = m_Tree.InsertItem(&tvIns);
  216. }
  218. /////////////////////////////////////////////////////////////////////////////
  220. void CPageGeneral::OnItemexpandingTree(NMHDR* pNMHDR, LRESULT* pResult) 
  221. {
  222.   NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
  223.   
  224.   LPTV_ITEM pParent = &pNMTreeView->itemNew;
  226.   // Assume we will return FALSE,
  227.   // allowing the expansion to occur
  228.   *pResult = FALSE;
  230.   // If the action is NOT expand, or if we have
  231.   // already expanded at least once, we don't need
  232.   // to do anything here
  233.   if ( pNMTreeView->action != TVE_EXPAND ||
  234.        (pParent->state & TVIS_EXPANDEDONCE) )
  235.     return;
  237.   // Get a handle to the parent item
  238.   TV_ITEM    tvItem;
  239.   tvItem.hItem = pParent->hItem;
  240.   tvItem.mask = TVIF_HANDLE;
  241.   m_Tree.GetItem(&tvItem);
  243.   // We are expanding an item for the first time
  244.   // Enumerate its children and add them to the tree
  245.   DWORD   numChildren = 0;
  246.   DWORD   ret = 0;
  248.   CNetNode * pParentNode = (CNetNode *) pParent->lParam;
  250.   {
  251.     CWaitCursor  wc;  // display hourglass cursor
  252.     ret = pParentNode->EnumerateNetwork();
  253.   }
  255.   if (ret)
  256.   {
  257.     static const DWORD maxMsgLen = 1024;
  258.     CString errMsg;
  259.     char * pMsgBuf = errMsg.GetBuffer(maxMsgLen);
  261.     DWORD formatRet = FormatMessage(
  262.       FORMAT_MESSAGE_IGNORE_INSERTS |
  263.       FORMAT_MESSAGE_FROM_SYSTEM,
  264.       NULL, ret, 0, pMsgBuf, maxMsgLen, NULL);
  266.     errMsg.ReleaseBuffer();
  268.     if (formatRet == 0)
  269.       errMsg.Format("Network enumeration failed.  nn"
  270.       "Error code: %lu.", ret);
  272.     MessageBox(errMsg, "Error",
  273.      MB_ICONEXCLAMATION | MB_OK);
  274.   }
  275.   else
  276.   {
  277.     CNetNode * pChildNode;
  278.     numChildren = pParentNode->GetChildCount();
  280.     for (DWORD i = 0; i < numChildren; i++)
  281.     {
  282.       pChildNode = pParentNode->GetChildNode(i);
  283.       AddNodeToTree(pChildNode, tvItem.hItem);
  284.     }
  285.   }
  287.   // If there are no child items, or an error occurred,
  288.   // remove the "+" from the parent item by indicating
  289.   // it has no child items
  290.   if (ret || numChildren == 0)
  291.   {
  292.     tvItem.mask |= TVIF_CHILDREN;
  293.     tvItem.cChildren = 0;
  294.     m_Tree.SetItem(&tvItem);
  296.     // indicate that tree expansion should NOT occur
  297.     *pResult = TRUE;
  298.   }
  299. }
  301. /////////////////////////////////////////////////////////////////////////////
  303. void CPageGeneral::OnSelchangedTree(NMHDR* pNMHDR, LRESULT* pResult) 
  304. {
  305.   NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
  307.   LPTV_ITEM pNewItem = &pNMTreeView->itemNew;
  308.   UpdateDisplay(pNewItem);
  309.   *pResult = 0;
  310. }
  312. /////////////////////////////////////////////////////////////////////////////
  314. void CPageGeneral::UpdateDisplay(LPTV_ITEM pItem)
  315. {
  316.   CNetNode* pNode;
  318.   if (pItem == NULL)
  319.     pNode = m_pRootNode;
  320.   else
  321.     pNode = (CNetNode*)pItem->lParam;
  323.   // Get text items from net node
  325.   m_sLocalName = pNode->GetLocalName();
  326.   m_sRemoteName = pNode->GetRemoteName();
  327.   m_sComment = pNode->GetComment();
  328.   m_sProvider = pNode->GetProvider();
  330.   // Get numeric items from net node
  331.   DWORD scope = pNode->GetScope();
  332.   DWORD type = pNode->GetType();
  333.   DWORD displayType = pNode->GetDisplayType();
  334.   DWORD usage = pNode->GetUsage();
  336.   // Convert numeric codes to text for display
  337.   if (scope == RESOURCE_CONNECTED)
  338.       m_sScope = "CONNECTED";
  339.   else if (scope == RESOURCE_GLOBALNET)
  340.       m_sScope = "GLOBALNET";
  341.   else if (scope == RESOURCE_REMEMBERED)
  342.       m_sScope = "REMEMBERED";
  343. #if(WINVER >= 0x0400)
  344.   else if (scope == RESOURCE_RECENT)
  345.       m_sScope = "RECENT";
  346.   else if (scope == RESOURCE_CONTEXT)
  347.       m_sScope = "CONTEXT";
  348. #endif // WINVER >= 0x0400
  350.   if (type == RESOURCETYPE_UNKNOWN)
  351.     m_sType = "UNKNOWN";
  352.   else
  353.   {
  354.     m_sType = "";
  355.     if (type & RESOURCETYPE_ANY)
  356.       m_sType += "ANY ";
  357.     if (type & RESOURCETYPE_DISK)
  358.       m_sType += "DISK ";
  359.     if (type & RESOURCETYPE_PRINT)
  360.       m_sType += "PRINT ";
  361.   }
  362.     
  363.   if (displayType == RESOURCEDISPLAYTYPE_GENERIC)
  364.     m_sDisplayType = "GENERIC";
  365.   else if (displayType == RESOURCEDISPLAYTYPE_DOMAIN)
  366.     m_sDisplayType = "DOMAIN";
  367.   else if (displayType == RESOURCEDISPLAYTYPE_SERVER)
  368.     m_sDisplayType = "SERVER";
  369.   else if (displayType == RESOURCEDISPLAYTYPE_SHARE)
  370.     m_sDisplayType = "SHARE";
  371.   else if (displayType == RESOURCEDISPLAYTYPE_FILE)
  372.     m_sDisplayType = "FILE";
  373.   else if (displayType == RESOURCEDISPLAYTYPE_GROUP)
  374.     m_sDisplayType = "GROUP";
  375. #if(WINVER >= 0x0400)
  376.   else if (displayType == RESOURCEDISPLAYTYPE_NETWORK)
  377.     m_sDisplayType = "NETWORK";
  378.   else if (displayType == RESOURCEDISPLAYTYPE_ROOT)
  379.     m_sDisplayType = "ROOT";
  380.   else if (displayType == RESOURCEDISPLAYTYPE_SHAREADMIN)
  381.     m_sDisplayType = "SHAREADMIN";
  382.   else if (displayType == RESOURCEDISPLAYTYPE_DIRECTORY)
  383.     m_sDisplayType = "DIRECTORY";
  384. #endif /* WINVER >= 0x0400 */
  385.   else if (displayType == RESOURCEDISPLAYTYPE_TREE)
  386.     m_sDisplayType = "TREE";
  388.   m_sUsage = "";
  389. #if(WINVER >= 0x0400)
  390.   if ((usage & RESOURCEUSAGE_ALL) == RESOURCEUSAGE_ALL)
  391.     m_sUsage += "ALL ";
  392.   else
  393. #endif
  394.   {
  395.     if (usage & RESOURCEUSAGE_CONNECTABLE)
  396.       m_sUsage += "CONNECTABLE ";
  397.     if (usage & RESOURCEUSAGE_CONTAINER)
  398.       m_sUsage += "CONTAINER ";
  399.   }
  401.   if (usage & RESOURCEUSAGE_NOLOCALDEVICE)
  402.     m_sUsage += "NOLOCALDEVICE ";
  403.   if (usage & RESOURCEUSAGE_SIBLING)
  404.     m_sUsage += "SIBLING ";
  406.   UpdateData(FALSE);  // copy member variables to screen
  407. }
  409. /////////////////////////////////////////////////////////////////////////////
  411. UINT g_Enumerate(LPVOID pParam)
  412. {
  413.   CString sAddress = *((CString*)pParam);
  414.   delete ((CString*)pParam);
  416.   g_WriteToOutput(TRUE, "[Enumerate] Started: " + sAddress);
  417.   g_AnimateWait->Play(0, -1, -1);
  419.   CString sOutput;
  420.   hostent* pHostent;
  422.   unsigned long ulIpAddress = inet_addr(sAddress);
  424.   if(ulIpAddress == INADDR_NONE)
  425.   {
  426.     pHostent = gethostbyname(sAddress);
  427.     if(pHostent)
  428.       ulIpAddress = *(DWORD*)(*pHostent->h_addr_list);
  429.   }
  431.   ulIpAddress &= 0x00ffffff;
  433.   for(int i = 0; i < 255; i++)
  434.   {
  435.     if(pHostent = gethostbyaddr((LPCSTR)&ulIpAddress, 4, PF_INET))
  436.     {
  437.       sOutput = pHostent->h_name;
  438.       in_addr inaddr;
  439.       inaddr.s_addr = ulIpAddress;
  440.       CString sIpAddress = inet_ntoa(inaddr);
  441.       g_WriteToOutput(TRUE, sOutput + " - " + sIpAddress);
  442.     }
  443.     ulIpAddress += 0x01000000;
  444.   }
  446.   g_WriteToOutput(TRUE, "[Enumerate] Finished...");
  448.   return 0;
  449. }
  451. /////////////////////////////////////////////////////////////////////////////
  453. void CPageGeneral::OnEnumerate() 
  454. {
  455.   CString sAddress;
  456.   m_Address.GetWindowText(sAddress);
  458.   CString* psAddress = new CString(sAddress);
  459.   AfxBeginThread(g_Enumerate, psAddress);
  460. }
  462. /////////////////////////////////////////////////////////////////////////////
  464. void CPageGeneral::OnPing() 
  465. {
  466.   if(UpdateData())
  467.   {
  468.     CString sAddress;
  469.     m_Address.GetWindowText(sAddress);
  471.     char* psOutput = new char[1000];
  472.     if(m_p.Ping(sAddress, m_pr, 10, m_dwPingTimeout, m_bPingPacket))
  473.     {
  474.       hostent* phostent = gethostbyaddr((char*)&m_pr.Address.S_un.S_addr, 4, PF_INET);
  475.       sprintf(psOutput, "[Ping] %d.%d.%d.%d [%s], replied in RTT:%d ms", 
  476.            m_pr.Address.S_un.S_un_b.s_b1, m_pr.Address.S_un.S_un_b.s_b2, m_pr.Address.S_un.S_un_b.s_b3, 
  477.            m_pr.Address.S_un.S_un_b.s_b4, phostent->h_name, m_pr.RTT);
  478.     }
  479.     else
  480.       sprintf(psOutput, "[Ping] Error %d", GetLastError());
  481.   
  482.     g_WriteToOutput(TRUE, psOutput);
  483.     delete psOutput;
  484.   }
  485. }
  487. /////////////////////////////////////////////////////////////////////////////
  489. BOOL CPageGeneral::PreTranslateMessage(MSG* pMsg) 
  490. {
  491.   // transate the message based on TTM_WINDOWFROMPOINT
  492.   MSG msg = *pMsg;
  493.   msg.hwnd = (HWND)m_ToolTip.SendMessage(TTM_WINDOWFROMPOINT, 0, (LPARAM)&msg.pt);
  494.   CPoint pt = pMsg->pt;
  495.   if (msg.message >= WM_MOUSEFIRST && msg.message <= WM_MOUSELAST)
  496.           ::ScreenToClient(msg.hwnd, &pt);
  497.   msg.lParam = MAKELONG(pt.x, pt.y);
  499.   // Let the ToolTip process this message.
  500.   m_ToolTip.RelayEvent(&msg);
  502. return CPropertyPage::PreTranslateMessage(pMsg);
  503. }
  505. /////////////////////////////////////////////////////////////////////////////