diff options
Diffstat (limited to 'NetCmpts/ClearInetCache.cpp')
| -rw-r--r-- | NetCmpts/ClearInetCache.cpp | 579 |
1 files changed, 579 insertions, 0 deletions
diff --git a/NetCmpts/ClearInetCache.cpp b/NetCmpts/ClearInetCache.cpp new file mode 100644 index 0000000..b4d5b4d --- /dev/null +++ b/NetCmpts/ClearInetCache.cpp @@ -0,0 +1,579 @@ +// ClearInetCache.cpp : Implementation of CClearInetCache +#include "stdafx.h" +#include "NetCmpts.h" +#include "ClearInetCache.h" +#include "ProgressDlg.h" +#include "..\common\defines.h" +#include <appmisc.h> +#include "parseurl.h" + +#include <algorithm> +using std::find; + +///////////////////////////////////////////////////////////////////////////// +// CClearInetCache + +STDMETHODIMP CClearInetCache::InterfaceSupportsErrorInfo(REFIID riid) +{ + static const IID* arr[] = + { + &IID_ISecureShutdownWin, + }; + for (int i=0;i<sizeof(arr)/sizeof(arr[0]);i++) + { + if (InlineIsEqualGUID(*arr[i],riid)) + return S_OK; + } + return S_FALSE; +} + +////////////////////////////////////////////////////////////////// +// + +#ifndef MAX_CACHE_ENTRY_INFO_SIZE +#define MAX_CACHE_ENTRY_INFO_SIZE 4096 +#endif + +STDMETHODIMP CClearInetCache::DoShutdown(long hParent, long lMode) +{ + HWND hwndParent = (HWND)hParent; + + bool bPrompt = lMode & nsNoPrompt ? false : true; + bool bSilent = lMode & nsQuiet ? true : false; + + // Return Value + HRESULT hRet = S_OK; + + MSG msg; + + // Handle for Searching + HANDLE hFindURL; + // Structure for Cache Info + LPINTERNET_CACHE_ENTRY_INFO lpICEInfo; + // We use MAX_CACHE_ENTRY_INFO_SIZE so we don't get ERROR_INSUFFICIENT_BUFFER's + unsigned long nBufferSize = MAX_CACHE_ENTRY_INFO_SIZE; + + // Alloc Memory (Use malloc for readability) + lpICEInfo = (LPINTERNET_CACHE_ENTRY_INFO)malloc(sizeof(INTERNET_CACHE_ENTRY_INFO) + MAX_CACHE_ENTRY_INFO_SIZE); + if(lpICEInfo == NULL) + return E_OUTOFMEMORY; + + + lpICEInfo->dwStructSize = sizeof(INTERNET_CACHE_ENTRY_INFO); + lpICEInfo->dwReserved = 0; // Always has to be zero + + // Find First + hFindURL = FindFirstUrlCacheEntry (_T("*.*"), lpICEInfo, &nBufferSize); + + if(hFindURL != NULL) + { + // Create Progress dialog + CProgressDlg dlgProg; + dlgProg.Create(hwndParent); + dlgProg.DispEventAdvise(m_spUnkSite); // Pointer from IObjectWithSite + dlgProg.m_ctlProgress.SetRange(0, 2); + dlgProg.m_ctlProgress.SetStep(1); + + bool bLoop = true; + + while(bLoop) + { + + // Update Progress Dialog + dlgProg.SetDlgItemText(IDC_FILENAME, lpICEInfo->lpszSourceUrlName); + dlgProg.m_ctlProgress.SetPos(0); + + // Check if we should delete this URL + if(IsDeleteableURL(lpICEInfo->lpszSourceUrlName)) + { + // Keep going even if error + DeleteUrlCacheEntry(lpICEInfo->lpszSourceUrlName); + + // Update Progress Dialog + dlgProg.m_ctlProgress.SetPos(2); + } + + // nBufferSize Gets changed in FindNext... funcs + // so have to reset + nBufferSize = MAX_CACHE_ENTRY_INFO_SIZE; + + // Get Next One + if(!((bLoop = FindNextUrlCacheEntry(hFindURL, lpICEInfo, &nBufferSize)) ? true : false)) + { + DWORD dwError = ::GetLastError(); + switch(::GetLastError()) + { + case ERROR_NO_MORE_FILES: + case ERROR_NO_MORE_ITEMS: + hRet = S_OK; + break; + // For Debugging + case ERROR_INSUFFICIENT_BUFFER: + hRet = E_OUTOFMEMORY; + break; + default: + hRet = HRESULT_FROM_WIN32(::GetLastError()); + break; + } + } + + PEEK_ALL_MESSAGES(msg); + + // Check for Cancel + if(dlgProg.IsCancelled()) + { + bLoop = false; + hRet = E_ABORT; + } + + } // while bLoop + + // Clean up time + FindCloseUrlCache(hFindURL); + free(lpICEInfo); + + if(dlgProg.IsWindow()) + dlgProg.DestroyWindow(); + + dlgProg.DispEventUnadvise(m_spUnkSite); // Pointer from IObjectWithSite + + return hRet; + + } // if handle is null + + // Clean up + free(lpICEInfo); + + switch(::GetLastError()) + { + case ERROR_NO_MORE_ITEMS: + case ERROR_NO_MORE_FILES: + return S_OK; + break; + default: + return HRESULT_FROM_WIN32(::GetLastError()); + break; + } + +} + +////////////////////////////////////////////////////////////////// +// Load Server List from Registry + +int CClearInetCache::GetServersToDelete() +{ + USES_CONVERSION; + + LoadFlags(); + + int nCnt = 0; // Number of URLs from Registry + int nReadServers = 0; // Number of 'sane' URLs + string sURL = ""; + string sKeyName = ""; + + // Format Key Name + sKeyName.format("Server%.4d", nCnt); + + // Used in URL Parsing + DWORD dwProtocol = 0; + string sServer = ""; + string sObject = ""; + INTERNET_PORT inetPort; + + // Add URLs to List Box + while((sURL = m_Data.GetString(sKeyName, NO_KEY)) != NO_KEY) + { + // Get the Host/Server Part + if(ParseURL(sURL, dwProtocol, sServer, sObject, inetPort)) + { + // Add to Array + m_asServerURLs.push_back(sServer); + nReadServers++; + } + + nCnt++; + + // Format Key Name + sKeyName.format("Server%.4d", nCnt); + + } + + m_bInitialized = true; + + return nReadServers; +} + +////////////////////////////////////////////////////////////////// +// Checks a URL against the list + +bool CClearInetCache::IsDeleteableURL(const string& sURL) +{ + // Sanity Check + if(sURL.empty()) + return false; + + + // Read Servers from Registry + if(!m_bInitialized) + + // Only continue of more than one server + GetServersToDelete(); + + + + // Used in URL Parsing + DWORD dwProtocol = 0; + string sServer = ""; + string sObject = ""; + INTERNET_PORT inetPort; + + // Get the Host/Server Part + if(ParseURL(sURL, dwProtocol, sServer, sObject, inetPort)) + { + if(dwProtocol == INTERNET_SCHEME_UNKNOWN) + return m_bCookies; + + if(m_bClearAll) + return true; + + // Check it against our array + return find(m_asServerURLs.begin(), m_asServerURLs.end(), sServer) + != m_asServerURLs.end(); + + } + + return false; + +} + +STDMETHODIMP CClearInetCache::get_Info(NightSecInfo nsItem, VARIANT* pvVal) +{ + ::VariantClear(pvVal); + + CComBSTR bsRetVal; + + switch(nsItem) + { + case nsName: + pvVal->vt = VT_BSTR; + bsRetVal.LoadString(IDS_INETNAME); + pvVal->bstrVal = bsRetVal.Detach(); + return S_OK; + case nsHelpText: + pvVal->vt = VT_BSTR; + bsRetVal.LoadString(IDS_INETDESC); + pvVal->bstrVal = bsRetVal.Detach(); + return S_OK; + case nsCmdLine: + pvVal->vt = VT_BSTR; + bsRetVal.LoadString(IDS_INETPARAM); + pvVal->bstrVal = bsRetVal.Detach(); + return S_OK; + case nsHideNormal: + pvVal->vt = VT_BOOL; + pvVal->bVal = TRUE; + return S_OK; + } + + ::VariantClear(pvVal); + return S_FALSE; +} + + +STDMETHODIMP CClearInetCache::SetData(IUnknown* pUnk) +{ + return m_Data.Initialize(pUnk); +} + + + + + + + + +///////////////////////////////////////////////////////////////////// +// +// Property Sheet +// +///////////////////////////////////////////////////////////////////// + +HRESULT CClearInetCache::SaveChanges() +{ + int nServer = 0; + string sKeyName = ""; + string sURL; + int iItem = -1; + + // Setup Structures + LV_ITEM lv; + + lv.mask = LVIF_TEXT; + lv.iSubItem = 0; + lv.cchTextMax = MAX_PATH * 2; + + HRESULT hr; + HRESULT hrRet = S_OK; + + // Loop through selected items + while((iItem = m_ctlServerList.GetNextItem(iItem, LVNI_ALL)) != -1) + { + lv.iItem = iItem; + lv.iSubItem = 0; + lv.pszText = sURL.get_buffer(MAX_PATH * 2); + + if(m_ctlServerList.GetItem(&lv)) + { + // Format Registry Key + sKeyName.format("Server%.4d", nServer); + + sURL.release_buffer(); + + hr = m_Data.WriteString(sKeyName, sURL); + if(FAILED(hr)) hrRet = hr; + + } + + nServer++; + + } + + for(nServer = nServer; nServer < m_nLoadedURLs; nServer++) + { + // Format Registry Key + sKeyName.format("Server%.4d", nServer); + + m_Data.DeleteProperty(sKeyName); + } + + + // Save Clear All State + hr = m_Data.WriteInt("Clear All", IsDlgButtonChecked(IDC_CLEARALLSERVERS)); + if(FAILED(hr)) hrRet = hr; + + // Save Cookie State + hr = m_Data.WriteInt("Clear Cookies", IsDlgButtonChecked(IDC_COOKIES)); + if(FAILED(hr)) hrRet = hr; + + return hrRet; +} + +LRESULT CClearInetCache::OnInitDialog(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled) +{ + LoadFlags(); + + + m_ctlServerList = GetDlgItem(IDC_SERVERURLLIST); + + // Create the small icon image list. + m_ImageListSmall.Create(16, + 16, + ILC_MASK, // list does not include masks + 1, + 0 ); // list won't grow + + + // Set Mask color for Image Lists + m_ImageListSmall.SetBkColor(::GetSysColor(COLOR_WINDOW)); + + + // Load the icon and add to the image list. + if(m_ImageListSmall.AddIcon(::LoadIcon(_Module.m_hInstResource, + MAKEINTRESOURCE(IDI_INETSHORTCUT))) == -1) + return false; + + + // Associate the image lists with the list view control. + m_ctlServerList.SetImageList(m_ImageListSmall, LVSIL_SMALL); + + + // Fill List Control + string sURL; + string sKeyName = ""; + int nCnt = 0; + + // Format Key Name + sKeyName.format("Server%.4d", nCnt); + + while((sURL = m_Data.GetString(sKeyName, NO_KEY)) != NO_KEY) + { + m_ctlServerList.InsertItem(nCnt, sURL, 0); + + // Make sure each one has a unique lParam + m_dwIDCounter++; + nCnt++; + + // Format Key Name + sKeyName.format("Server%.4d", nCnt); + } + + m_nLoadedURLs = nCnt + 1; + + CheckDlgButton(IDC_CLEARALLSERVERS, m_bClearAll); + CheckDlgButton(IDC_COOKIES, m_bCookies); + UpdateClearAll(); + + m_hIconNew = ::LoadIcon(_Module.m_hInstResource, MAKEINTRESOURCE(IDI_NEW)); + m_hIconDel = ::LoadIcon(_Module.m_hInstResource, MAKEINTRESOURCE(IDI_DEL)); + + CButton btn = GetDlgItem(IDC_NEWURL); + btn.SetIcon(m_hIconNew); + + btn = GetDlgItem(IDC_DELETE); + btn.SetIcon(m_hIconDel); + + return false; // return TRUE unless you set the focus to a control + // EXCEPTION: OCX Property Pages should return FALSE + +} + + + +LRESULT CClearInetCache::OnURLListEndLabelEdit(int idCtrl, LPNMHDR pnmh, BOOL& bHandled) +{ + LV_DISPINFO* pDispInfo = (LV_DISPINFO*)pnmh; + + // Only Change if there is Text + if(pDispInfo->item.pszText != NULL) + { + // Used in Checking URL + DWORD dwProtocol = 0; + string sTemp1 = ""; + string sTemp2 = ""; + INTERNET_PORT inetPort; + + + // Do a Check on URL + if(ParseURL(pDispInfo->item.pszText, dwProtocol, sTemp1, sTemp2, inetPort)) + { + // Set Item text + m_ctlServerList.SetItem(&(pDispInfo->item)); + SetDirty(true); + } + else + { + // Try adding http:// + string sNewURL(_T("http://")); + sNewURL += pDispInfo->item.pszText; + if(ParseURL(sNewURL, dwProtocol, sTemp1, sTemp2, inetPort)) + { + // Set Item text + pDispInfo->item.pszText = sNewURL.get_buffer(); + m_ctlServerList.SetItem(&(pDispInfo->item)); + SetDirty(true); + } + else + { + // Give an Error Box + string sMsg; + sMsg.load_string(IDS_NOTVALIDURL); + MessageBox(sMsg, "Night Security", MB_ICONSTOP | MB_OK); + } + + } + } + + return 0; + +} + + +LRESULT CClearInetCache::OnURLListKeyDown(int idCtrl, LPNMHDR pnmh, BOOL& bHandled) +{ + LV_KEYDOWN* pLVKeyDow = (LV_KEYDOWN*)pnmh; + + // if F2 then Edit + if(pLVKeyDow->wVKey == VK_F2) + { + m_ctlServerList.SetFocus(); // Needs to have the Focus in order to edit + m_ctlServerList.EditLabel(m_ctlServerList.GetNextItem(-1, LVNI_ALL | LVNI_FOCUSED)); + } + + // If Delete then call Delete Handler + if(pLVKeyDow->wVKey == VK_DELETE) + OnDelete(0, 0, 0, bHandled); + + return 0; + +} + +LRESULT CClearInetCache::OnListSetFocus(int idCtrl, LPNMHDR pnmh, BOOL& bHandled) +{ + // Disclaimer: This is a kludge fix + // SysListView32 wouldn't redraw itself when it had focus + // and app was activated + m_ctlServerList.RedrawWindow(NULL, NULL, RDW_FRAME | RDW_INVALIDATE); + return 0; +} + + +LRESULT CClearInetCache::OnNewURL(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled) +{ + int iItem = -1; + string sNewURL; + sNewURL.load_string(IDS_NEWURL); + + // Insert Item + if((iItem = m_ctlServerList.InsertItem(0, sNewURL, 0)) != -1) + { + m_dwIDCounter++; + + // Open for Editing + m_ctlServerList.SetFocus(); // Needs to have the Focus in order to edit + m_ctlServerList.EditLabel(iItem); + + SetDirty(true); + + } + + return 0; + +} + +LRESULT CClearInetCache::OnDelete(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled) +{ + int iItem = -1; + + LV_ITEM lv; + + lv.mask = LVIF_TEXT; + lv.iSubItem = 0; + + // Loop through selected items + while((iItem = m_ctlServerList.GetNextItem(-1, LVNI_ALL | LVNI_SELECTED)) != -1) + { + // Take it out of List Box + m_ctlServerList.DeleteItem(iItem); + SetDirty(true); + } + + return 0; + +} + + +LRESULT CClearInetCache::OnClearAll(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled) +{ + UpdateClearAll(); + + SetDirty(true); + + return 0; +} + +void CClearInetCache::UpdateClearAll() +{ + bool bEnable = !IsDlgButtonChecked(IDC_CLEARALLSERVERS); + + ::EnableWindow(GetDlgItem(IDC_SERVERURLLIST), bEnable); + ::EnableWindow(GetDlgItem(IDC_NEWURL), bEnable); + ::EnableWindow(GetDlgItem(IDC_DELETE), bEnable); + + SetDirty(true); +} + +void CClearInetCache::LoadFlags() +{ + m_bClearAll = m_Data.GetInt("Clear All", true) ? true : false; + m_bCookies = m_Data.GetInt("Clear Cookies", false) ? true : false; +} |