#include "status.h" #include "properties.h" #include "util.h" CStatusDialog::CStatusDialog(IInternetGateway* pInternetGateway) { m_pInternetGateway = pInternetGateway; m_pInternetGateway->AddRef(); m_uTimerId = 0; m_bGettingStatistics = FALSE; m_bShowingBytes = TRUE; } CStatusDialog::~CStatusDialog() { m_pInternetGateway->Release(); } LRESULT CStatusDialog::OnInitDialog(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled) { m_uTimerId = SetTimer(1, 1000); TCHAR szCloseText[64]; if(0 != ::LoadString(_Module.GetResourceInstance(), IDS_CLOSE, szCloseText, sizeof(szCloseText) / sizeof(TCHAR))) { ::SetDlgItemText(GetParent(), IDCANCEL, szCloseText); // set the cancel to close } ::ShowWindow(::GetDlgItem(GetParent(), IDOK), SW_HIDE); // hide the original close ::EnableWindow(::GetDlgItem(GetParent(), IDCANCEL), TRUE); // and re-enable the cancel button IUPnPService* pWANConnectionService; HRESULT hr = GetWANConnectionService(m_pInternetGateway, &pWANConnectionService); if(SUCCEEDED(hr)) { NETCON_STATUS Status; hr = GetConnectionStatus(pWANConnectionService, &Status); if(SUCCEEDED(hr)) { UpdateButtons(Status); } pWANConnectionService->Release(); } return TRUE; } LRESULT CStatusDialog::OnDestroy(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled) { if(0 != m_uTimerId) { ::KillTimer(m_hWnd, m_uTimerId); } return 0; } LRESULT CStatusDialog::OnTimer(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled) { HRESULT hr = S_OK; if(FALSE == m_bGettingStatistics) // this functions pumps messages, so don't let us be re-entered { m_bGettingStatistics = TRUE; NETCON_STATUS Status = NCS_CONNECTED; ULONG ulTotalBytesSent = 0; ULONG ulTotalBytesReceived = 0; ULONG ulTotalPacketsSent = 0; ULONG ulTotalPacketsReceived = 0; ULONG ulSpeedbps = 0; ULONG ulUptime = 0; BOOL bStatisticsAvailable = TRUE; IUPnPService* pWANConnection; hr = GetWANConnectionService(m_pInternetGateway, &pWANConnection); if(SUCCEEDED(hr)) { IUPnPService* pWANCommonInterfaceConfig; hr = m_pInternetGateway->GetService(SAHOST_SERVICE_WANCOMMONINTERFACECONFIG, &pWANCommonInterfaceConfig); if(SUCCEEDED(hr)) { hr = GetConnectionStatus(pWANConnection, &Status); if(SUCCEEDED(hr)) { if(NCS_CONNECTED == Status) { VARIANT OutArgs; VariantInit(&OutArgs); hr = InvokeVoidAction(pWANCommonInterfaceConfig, L"X_GetICSStatistics", &OutArgs); if(SUCCEEDED(hr)) { SAFEARRAY* pArray = V_ARRAY(&OutArgs); LONG lIndex = 0; VARIANT Param; lIndex = 0; hr = SafeArrayGetElement(pArray, &lIndex, &Param); if(SUCCEEDED(hr)) { if(V_VT(&Param) == VT_UI4) { ulTotalBytesSent = V_UI4(&Param); } VariantClear(&Param); } lIndex = 1; hr = SafeArrayGetElement(pArray, &lIndex, &Param); if(SUCCEEDED(hr)) { if(V_VT(&Param) == VT_UI4) { ulTotalBytesReceived = V_UI4(&Param); } VariantClear(&Param); } lIndex = 2; hr = SafeArrayGetElement(pArray, &lIndex, &Param); if(SUCCEEDED(hr)) { if(V_VT(&Param) == VT_UI4) { ulTotalPacketsSent = V_UI4(&Param); } VariantClear(&Param); } lIndex = 3; hr = SafeArrayGetElement(pArray, &lIndex, &Param); if(SUCCEEDED(hr)) { if(V_VT(&Param) == VT_UI4) { ulTotalPacketsReceived = V_UI4(&Param); } VariantClear(&Param); } lIndex = 4; hr = SafeArrayGetElement(pArray, &lIndex, &Param); if(SUCCEEDED(hr)) { if(V_VT(&Param) == VT_UI4) { ulSpeedbps = V_UI4(&Param); } VariantClear(&Param); } lIndex = 5; hr = SafeArrayGetElement(pArray, &lIndex, &Param); if(SUCCEEDED(hr)) { if(V_VT(&Param) == VT_UI4) { ulUptime = V_UI4(&Param); } VariantClear(&Param); } VariantClear(&OutArgs); } else if(UPNP_E_INVALID_ACTION == hr) { VARIANT OutArgs; LONG lIndex = 0; VARIANT Param; hr = InvokeVoidAction(pWANCommonInterfaceConfig, L"GetTotalBytesSent", &OutArgs); if(SUCCEEDED(hr)) { lIndex = 0; hr = SafeArrayGetElement(V_ARRAY(&OutArgs), &lIndex, &Param); if(SUCCEEDED(hr)) { if(V_VT(&Param) == VT_UI4) { ulTotalBytesSent = V_UI4(&Param); } VariantClear(&Param); } VariantClear(&OutArgs); } if(SUCCEEDED(hr)) { lIndex = 0; hr = InvokeVoidAction(pWANCommonInterfaceConfig, L"GetTotalBytesReceived", &OutArgs); if(SUCCEEDED(hr)) { hr = SafeArrayGetElement(V_ARRAY(&OutArgs), &lIndex, &Param); if(SUCCEEDED(hr)) { if(V_VT(&Param) == VT_UI4) { ulTotalBytesReceived = V_UI4(&Param); } VariantClear(&Param); } VariantClear(&OutArgs); } } if(SUCCEEDED(hr)) { lIndex = 0; hr = InvokeVoidAction(pWANCommonInterfaceConfig, L"GetTotalPacketsSent", &OutArgs); if(SUCCEEDED(hr)) { hr = SafeArrayGetElement(V_ARRAY(&OutArgs), &lIndex, &Param); if(SUCCEEDED(hr)) { if(V_VT(&Param) == VT_UI4) { ulTotalPacketsSent = V_UI4(&Param); } VariantClear(&Param); } VariantClear(&OutArgs); } } if(SUCCEEDED(hr)) { hr = InvokeVoidAction(pWANCommonInterfaceConfig, L"GetTotalPacketsReceived", &OutArgs); if(SUCCEEDED(hr)) { lIndex = 0; hr = SafeArrayGetElement(V_ARRAY(&OutArgs), &lIndex, &Param); if(SUCCEEDED(hr)) { if(V_VT(&Param) == VT_UI4) { ulTotalPacketsReceived = V_UI4(&Param); } VariantClear(&Param); } VariantClear(&OutArgs); } } if(SUCCEEDED(hr)) { hr = InvokeVoidAction(pWANCommonInterfaceConfig, L"GetCommonLinkProperties", &OutArgs); if(SUCCEEDED(hr)) { lIndex = 2; hr = SafeArrayGetElement(V_ARRAY(&OutArgs), &lIndex, &Param); if(SUCCEEDED(hr)) { if(V_VT(&Param) == VT_UI4) { ulSpeedbps = V_UI4(&Param); } VariantClear(&Param); } VariantClear(&OutArgs); } } if(SUCCEEDED(hr)) { hr = InvokeVoidAction(pWANConnection, L"GetStatusInfo", &OutArgs); if(SUCCEEDED(hr)) { lIndex = 2; hr = SafeArrayGetElement(V_ARRAY(&OutArgs), &lIndex, &Param); if(SUCCEEDED(hr)) { if(V_VT(&Param) == VT_UI4) { ulUptime = V_UI4(&Param); } VariantClear(&Param); } VariantClear(&OutArgs); } } if(UPNP_E_INVALID_ACTION == hr) { hr = S_OK; // server does not support statistics bStatisticsAvailable = FALSE; } } } } pWANCommonInterfaceConfig->Release(); } pWANConnection->Release(); } if(SUCCEEDED(hr) && NCS_CONNECTED == Status) { if(TRUE == bStatisticsAvailable) { if(0 != ulTotalBytesSent && 0 != ulTotalBytesReceived) { if(FALSE == m_bShowingBytes) // switch labels { m_bShowingBytes = TRUE; ::ShowWindow(GetDlgItem(IDC_STATUS_BYTESLABEL), SW_SHOW); ::ShowWindow(GetDlgItem(IDC_STATUS_PACKETSLABEL), SW_HIDE); } SetDlgItemInt(IDC_STATUS_BYTESSENT, ulTotalBytesSent, FALSE); SetDlgItemInt(IDC_STATUS_BYTESRECEIVED, ulTotalBytesReceived, FALSE); } else { if(TRUE == m_bShowingBytes) // switch labels { m_bShowingBytes = FALSE; ::ShowWindow(GetDlgItem(IDC_STATUS_PACKETSLABEL), SW_SHOW); ::ShowWindow(GetDlgItem(IDC_STATUS_BYTESLABEL), SW_HIDE); } SetDlgItemInt(IDC_STATUS_BYTESSENT, ulTotalPacketsSent, FALSE); SetDlgItemInt(IDC_STATUS_BYTESRECEIVED, ulTotalPacketsReceived, FALSE); } TCHAR szTimeDuration[128]; hr = FormatTimeDuration(ulUptime, szTimeDuration, sizeof(szTimeDuration) / sizeof(TCHAR)); if(SUCCEEDED(hr)) { SetDlgItemText(IDC_STATUS_DURATION, szTimeDuration); } \ TCHAR szBytesPerSecond[128]; hr = FormatBytesPerSecond(ulSpeedbps, szBytesPerSecond, sizeof(szBytesPerSecond) / sizeof(TCHAR)); if(SUCCEEDED(hr)) { SetDlgItemText(IDC_STATUS_SPEED, szBytesPerSecond); } } else { TCHAR szNotAvailable[64]; if(0 == LoadString(_Module.GetResourceInstance(), IDS_NOTAVAILABLE, szNotAvailable, sizeof(szNotAvailable) / sizeof(TCHAR))) { szNotAvailable[0] = TEXT('\0'); } SetDlgItemText(IDC_STATUS_BYTESSENT, szNotAvailable); SetDlgItemText(IDC_STATUS_BYTESRECEIVED, szNotAvailable); SetDlgItemText(IDC_STATUS_DURATION, szNotAvailable); SetDlgItemText(IDC_STATUS_SPEED, szNotAvailable); } } else { SetDlgItemText(IDC_STATUS_BYTESSENT, TEXT("")); SetDlgItemText(IDC_STATUS_BYTESRECEIVED, TEXT("")); SetDlgItemText(IDC_STATUS_DURATION, TEXT("")); SetDlgItemText(IDC_STATUS_SPEED, TEXT("")); } if(SUCCEEDED(hr) || UPNP_E_ACTION_REQUEST_FAILED == hr) // if we disconnected after getting status this will fail { UpdateButtons(Status); TCHAR szConnectionStatus[64]; hr = ConnectionStatusToString(Status, szConnectionStatus, sizeof(szConnectionStatus) / sizeof(TCHAR)); if(SUCCEEDED(hr)) { SetDlgItemText(IDC_STATUS_STATUS, szConnectionStatus); } } if(FAILED(hr)) { ::PropSheet_PressButton(GetParent(), PSBTN_CANCEL); } m_bGettingStatistics = FALSE; } return 0; } LRESULT CStatusDialog::OnProperties(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled) { HWND hPropertySheet = GetParent(); if(NULL != hPropertySheet) { HWND hTopLevelWindow = ::GetParent(hPropertySheet); if(NULL != hTopLevelWindow) { ::PostMessage(hTopLevelWindow, WM_COMMAND, IDM_TRAYICON_PROPERTIES, (LPARAM)hPropertySheet); } } return 0; } LRESULT CStatusDialog::OnDisconnect(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled) { HRESULT hr = S_OK; IUPnPService* pWANConnectionService; hr = GetWANConnectionService(m_pInternetGateway, &pWANConnectionService); if(SUCCEEDED(hr)) { VARIANT OutArgs; hr = InvokeVoidAction(pWANConnectionService, L"ForceTermination", &OutArgs); if(SUCCEEDED(hr)) { VariantClear(&OutArgs); } else if(UPNP_ACTION_HRESULT(800) == hr) { ShowErrorDialog(m_hWnd, IDS_ACCESSDENIED); } pWANConnectionService->Release(); } return 0; } LRESULT CStatusDialog::OnConnect(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled) { HRESULT hr = S_OK; IUPnPService* pWANConnectionService; hr = GetWANConnectionService(m_pInternetGateway, &pWANConnectionService); if(SUCCEEDED(hr)) { VARIANT OutArgs; hr = InvokeVoidAction(pWANConnectionService, L"RequestConnection", &OutArgs); if(SUCCEEDED(hr)) { VariantClear(&OutArgs); } else if(UPNP_ACTION_HRESULT(800) == hr) { ShowErrorDialog(m_hWnd, IDS_ACCESSDENIED); } else if(UPNP_E_DEVICE_TIMEOUT != hr) { ShowErrorDialog(m_hWnd, IDS_CONNECTIONFAILED); } pWANConnectionService->Release(); } return 0; } HRESULT CStatusDialog::UpdateButtons(NETCON_STATUS Status) { NETCON_MEDIATYPE MediaType; HRESULT hr = m_pInternetGateway->GetMediaType(&MediaType); if(SUCCEEDED(hr)) { TCHAR szButtonText[64]; if(NCS_CONNECTED == Status) { if(NCM_SHAREDACCESSHOST_RAS == MediaType) { if(0 != LoadString(_Module.GetResourceInstance(), IDS_DISCONNECT, szButtonText, sizeof(szButtonText) / sizeof(TCHAR))) { SetDlgItemText(IDC_STATUS_DISCONNECT, szButtonText); } } else { if(0 != LoadString(_Module.GetResourceInstance(), IDS_DISABLE, szButtonText, sizeof(szButtonText) / sizeof(TCHAR))) { SetDlgItemText(IDC_STATUS_DISCONNECT, szButtonText); } } ::ShowWindow(GetDlgItem(IDC_STATUS_DISCONNECT), SW_SHOW); ::ShowWindow(GetDlgItem(IDC_STATUS_CONNECT), SW_HIDE); } else if(NCS_DISCONNECTED == Status) { if(NCM_SHAREDACCESSHOST_RAS == MediaType) { if(0 != LoadString(_Module.GetResourceInstance(), IDS_CONNECT, szButtonText, sizeof(szButtonText) / sizeof(TCHAR))) { SetDlgItemText(IDC_STATUS_CONNECT, szButtonText); } } else { if(0 != LoadString(_Module.GetResourceInstance(), IDS_ENABLE, szButtonText, sizeof(szButtonText) / sizeof(TCHAR))) { SetDlgItemText(IDC_STATUS_CONNECT, szButtonText); } } ::ShowWindow(GetDlgItem(IDC_STATUS_CONNECT), SW_SHOW); ::ShowWindow(GetDlgItem(IDC_STATUS_DISCONNECT), SW_HIDE); } } return 0; }