dubey/WindowsXP
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
WindowsXP/inetcore/outlookexpress/msoeacct/agntnews.cpp
Tanishq Dubey cb60ae51e5
Initial Commit
2025-04-27 07:49:33 -04:00

691 lines
19 KiB
C++
Raw Permalink Blame History

#include "pch.hxx"
#include <imnact.h>
#include <acctimp.h>
#include <dllmain.h>
#include <resource.h>
#include <AgntNews.h> // Forte Agent
#include "newimp.h"
ASSERTDATA
#define MEMCHUNK 512
CAgentAcctImport::CAgentAcctImport()
{
m_cRef = 1;
m_fIni = FALSE;
*m_szIni = 0;
m_cInfo = 0;
m_rgInfo = NULL;
}
CAgentAcctImport::~CAgentAcctImport()
{
if (m_rgInfo != NULL)
MemFree(m_rgInfo);
}
STDMETHODIMP CAgentAcctImport::QueryInterface(REFIID riid, LPVOID *ppv)
{
if (ppv == NULL)
return(E_INVALIDARG);
*ppv = NULL;
if ((IID_IUnknown == riid) || (IID_IAccountImport == riid))
*ppv = (IAccountImport *)this;
else if (IID_IAccountImport2 == riid)
*ppv = (IAccountImport2 *)this;
else
return(E_NOINTERFACE);
((LPUNKNOWN)*ppv)->AddRef();
return(S_OK);
}
STDMETHODIMP_(ULONG) CAgentAcctImport::AddRef()
{
return(++m_cRef);
}
STDMETHODIMP_(ULONG) CAgentAcctImport::Release()
{
if (--m_cRef == 0)
{
delete this;
return(0);
}
return(m_cRef);
}
const static char c_szRegAgnt[] = "Software\\Forte\\Agent\\Paths";
const static char c_szDefPath[] = "c:\\Agent\\Data\\Agent.ini";
const static char c_szRegIni[] = "IniFile";
const static char c_szRegUninstall[] = "Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\Forte Agent";
const static char c_szRegString[] = "UninstallString";
HRESULT STDMETHODCALLTYPE CAgentAcctImport::AutoDetect(DWORD *pcAcct, DWORD dwFlags)
{
HRESULT hr = S_OK;
DWORD cb = MAX_PATH, dwType;
char szUserName[MAX_PATH];
char szUserIniPath[MAX_PATH];
char szExpanded[MAX_PATH];
char szNewsServer[MAX_PATH];
char szIniPath[] = "Data\\Agent.ini";
char *psz;
int nCount = 0;
HKEY hkey;
Assert(m_cInfo == 0);
if (pcAcct == NULL)
return(E_INVALIDARG);
*pcAcct = 0;
// FIRST CHECK FOR FORTE AGENT
if (ERROR_SUCCESS == RegOpenKeyEx(HKEY_CURRENT_USER, c_szRegAgnt, 0, KEY_ALL_ACCESS, &hkey))
{
cb = sizeof(szUserIniPath);
if(ERROR_SUCCESS == RegQueryValueEx(hkey, c_szRegIni, NULL, &dwType, (LPBYTE)szUserIniPath, &cb ))
{
if (REG_EXPAND_SZ == dwType)
{
ExpandEnvironmentStrings(szUserIniPath, szExpanded, ARRAYSIZE(szExpanded));
psz = szExpanded;
}
else
psz = szUserIniPath;
GetPrivateProfileString("Profile", "FullName", "Default User", szUserName, MAX_PATH, psz);
if(GetPrivateProfileString("Servers", "NewsServer", "", szNewsServer, MAX_PATH, psz))
{
if (!MemAlloc((void **)&m_rgInfo, 1*sizeof(AGNTNEWSACCTINFO)))
{
hr = E_OUTOFMEMORY;
goto done;
}
m_rgInfo[m_cInfo].dwCookie = m_cInfo;
lstrcpyn(m_rgInfo[m_cInfo].szUserPath, psz, ARRAYSIZE(m_rgInfo[m_cInfo].szUserPath));
lstrcpyn(m_rgInfo[m_cInfo].szDisplay, szUserName, ARRAYSIZE(m_rgInfo[m_cInfo].szDisplay));
m_cInfo++;
}
}
}
// IF FORTE AGENT IS NOT FOUND, CHECK FOR FREE AGENT.
// CHECK THE DEFAULT PATH "C:\AGENT\DATA " FOR THE "AGENT.INI" FILE
else if(GetPrivateProfileString("Servers", "NewsServer", "", szNewsServer, MAX_PATH, c_szDefPath))
{
if (!MemAlloc((void **)&m_rgInfo, 1*sizeof(AGNTNEWSACCTINFO)))
{
hr = E_OUTOFMEMORY;
goto done;
}
GetPrivateProfileString("Profile", "FullName", "Default User", szUserName, MAX_PATH, c_szDefPath);
m_rgInfo[m_cInfo].dwCookie = m_cInfo;
lstrcpyn(m_rgInfo[m_cInfo].szUserPath, c_szDefPath, ARRAYSIZE(m_rgInfo[m_cInfo].szUserPath));
lstrcpyn(m_rgInfo[m_cInfo].szDisplay, szUserName, ARRAYSIZE(m_rgInfo[m_cInfo].szDisplay));
m_cInfo++;
}
// ELSE THE WORKAROUND FOR GETTING THE FREE AGENT INSTALL PATH i.e. RETRIEVE THE (UN)INSTALLATION PATH FOR (FREE?)-AGENT FROM THE REGISTRY.
else
{
if (ERROR_SUCCESS == RegOpenKeyEx(HKEY_LOCAL_MACHINE, c_szRegUninstall, 0, KEY_ALL_ACCESS, &hkey))
{
cb = sizeof(szUserIniPath);
if(ERROR_SUCCESS == RegQueryValueEx(hkey, c_szRegString, NULL, &dwType, (LPBYTE)szUserIniPath, &cb ))
{
// $$$Review: [NAB] Seems like this would break if there were a space in the the dir name!
while(szUserIniPath[nCount] != ' ')
nCount++;
// Come back now.
while(szUserIniPath[nCount] != '\\')
nCount--;
nCount++;
szUserIniPath[nCount] = '\0';
lstrcat(szUserIniPath, szIniPath);
if (REG_EXPAND_SZ == dwType)
{
ExpandEnvironmentStrings(szUserIniPath, szExpanded, ARRAYSIZE(szExpanded));
psz = szExpanded;
}
else
psz = szUserIniPath;
GetPrivateProfileString("Profile", "FullName", "Default User", szUserName, MAX_PATH, psz);
if(GetPrivateProfileString("Servers", "NewsServer", "", szNewsServer, MAX_PATH, psz))
{
if (!MemAlloc((void **)&m_rgInfo, 1*sizeof(AGNTNEWSACCTINFO)))
{
hr = E_OUTOFMEMORY;
goto done;
}
m_rgInfo[m_cInfo].dwCookie = m_cInfo;
lstrcpyn(m_rgInfo[m_cInfo].szUserPath, psz, ARRAYSIZE(m_rgInfo[m_cInfo].szUserPath));
lstrcpyn(m_rgInfo[m_cInfo].szDisplay, szUserName, ARRAYSIZE(m_rgInfo[m_cInfo].szDisplay));
m_cInfo++;
}
}
}
}
if (hr == S_OK)
{
*pcAcct = m_cInfo;
}
done:
RegCloseKey(hkey);
return(hr);
}
HRESULT STDMETHODCALLTYPE CAgentAcctImport::InitializeImport(HWND hwnd, DWORD_PTR dwCookie)
{
return S_OK;
}
HRESULT STDMETHODCALLTYPE CAgentAcctImport::EnumerateAccounts(IEnumIMPACCOUNTS **ppEnum)
{
CEnumAGNTACCT *penum;
HRESULT hr;
if (ppEnum == NULL)
return(E_INVALIDARG);
*ppEnum = NULL;
if (m_cInfo == 0)
return(S_FALSE);
Assert(m_rgInfo != NULL);
penum = new CEnumAGNTACCT;
if (penum == NULL)
return(E_OUTOFMEMORY);
hr = penum->Init(m_rgInfo, m_cInfo);
if (FAILED(hr))
{
penum->Release();
penum = NULL;
}
*ppEnum = penum;
return(hr);
}
HRESULT STDMETHODCALLTYPE CAgentAcctImport::GetSettings(DWORD_PTR dwCookie, IImnAccount *pAcct)
{
HRESULT hr;
if (pAcct == NULL)
return(E_INVALIDARG);
hr = IGetSettings(dwCookie, pAcct, NULL);
return(hr);
}
HRESULT CAgentAcctImport::IGetSettings(DWORD_PTR dwCookie, IImnAccount *pAcct, IMPCONNINFO *pInfo)
{
HRESULT hr;
AGNTNEWSACCTINFO *pinfo;
char szUserPrefs[AGNTSUSERROWS][AGNTSUSERCOLS];
char sz[512];
DWORD cb, type;
DWORD dwNewsPort = 119;
ZeroMemory((void*)&szUserPrefs[0], AGNTSUSERCOLS*AGNTSUSERROWS*sizeof(char));
Assert(((int) dwCookie) >= 0 && dwCookie < (DWORD_PTR)m_cInfo);
pinfo = &m_rgInfo[dwCookie];
Assert(pinfo->dwCookie == dwCookie);
hr = GetUserPrefs(pinfo->szUserPath, szUserPrefs);
Assert(0 < lstrlen(szUserPrefs[0]));
hr = pAcct->SetPropSz(AP_ACCOUNT_NAME, szUserPrefs[0]);
if (FAILED(hr))
return(hr);
hr = pAcct->SetPropSz(AP_NNTP_SERVER, szUserPrefs[0]);
Assert(!FAILED(hr));
if(lstrcmp(szUserPrefs[1], "119"))
{
int Len = lstrlen(szUserPrefs[1]);
if(Len)
{
// Convert the string to a dw.
DWORD dwMult = 1;
dwNewsPort = 0;
while(Len)
{
Len--;
dwNewsPort += ((int)szUserPrefs[1][Len] - 48)*dwMult;
dwMult *= 10;
}
}
}
hr = pAcct->SetPropDw(AP_NNTP_PORT, dwNewsPort);
Assert(!FAILED(hr));
if(lstrlen(szUserPrefs[2]))
{
hr = pAcct->SetPropSz(AP_NNTP_DISPLAY_NAME, szUserPrefs[2]);
Assert(!FAILED(hr));
}
if(lstrlen(szUserPrefs[3]))
{
hr = pAcct->SetPropSz(AP_NNTP_EMAIL_ADDRESS, szUserPrefs[3]);
Assert(!FAILED(hr));
}
if (pInfo != NULL)
{
// TODO: can we do any better than this???
pInfo->dwConnect = CONN_USE_DEFAULT;
}
return(S_OK);
}
HRESULT CAgentAcctImport::GetUserPrefs(char *szUserPath, char szUserPrefs[][AGNTSUSERCOLS])
{
HRESULT hr = E_FAIL;
DWORD dwResult = 0;
if(!GetPrivateProfileString("Servers", "NewsServer", "News", szUserPrefs[0], AGNTSUSERCOLS, szUserPath))
hr = S_FALSE;
if(!GetPrivateProfileString("Servers", "NNTPPort", "119", szUserPrefs[1], AGNTSUSERCOLS, szUserPath))
hr = S_FALSE;
if(!GetPrivateProfileString("Profile", "FullName", "Default User", szUserPrefs[2], AGNTSUSERCOLS, szUserPath))
hr = S_FALSE;
if(!GetPrivateProfileString("Profile", "EMailAddress", "", szUserPrefs[3], AGNTSUSERCOLS, szUserPath))
hr = S_FALSE;
return hr;
}
HRESULT CAgentAcctImport::GetNewsGroup(INewsGroupImport *pImp, DWORD dwReserved)
{
// We can ignore the first parameter as we have only one server.
HRESULT hr = S_OK;
char *pListGroups = NULL;
char szFilePath[AGNTSUSERCOLS];
int nCounter;
Assert(pImp != NULL);
lstrcpy(szFilePath, m_rgInfo[0].szUserPath);
nCounter = lstrlen(szFilePath);
if (nCounter)
{
while (nCounter)
{
if (szFilePath[nCounter] == '\\')
{
szFilePath[nCounter] = '\0';
break;
}
nCounter--;
}
}
else
{
return S_FALSE;
}
if (!FAILED(GetSubListGroups(szFilePath, &pListGroups)))
{
if (!SUCCEEDED(pImp->ImportSubList(pListGroups)))
hr = S_FALSE;
}
if (pListGroups != NULL)
MemFree(pListGroups);
return hr;
}
const static char c_szGroupFile[] = "Groups.dat";
const static char c_szBakupFile[] = "Grpdat.bak";
HRESULT CAgentAcctImport::GetSubListGroups(char *szHomeDir, char **ppListGroups)
{
return(E_FAIL);
#ifdef DEAD
HRESULT hr = E_FAIL;
HANDLE hRCHandle = NULL;
HANDLE hRCFile = NULL;
ULONG cbRCFile = 0;
BYTE *pBegin = NULL,
*pCurr = NULL,
*pEnd = NULL;
int nBalMem = MEMCHUNK;
int nLine = 0,
nCount = 0;
char cPlaceHolder = 1;
char szLineHolder[MEMCHUNK];
char *pListGroups = NULL;
char szWildSearch[MEMCHUNK];
char szDataFile[MEMCHUNK];
char szEnumFile[MEMCHUNK];
WIN32_FIND_DATA rgFileData;
HANDLE hDatFile = NULL;
BOOL bFileFound = TRUE;
DWORD dwNumLines = 0;
char szSubList[MEMCHUNK];
int nTempStore = 0;
DWORD dwFactor = 1;
Assert(szHomeDir);
Assert(ppListGroups);
*ppListGroups = NULL;
lstrcpy(szWildSearch, szHomeDir);
lstrcat(szWildSearch, "\\*.dat");
lstrcpy(szDataFile, szHomeDir);
lstrcat(szDataFile, "\\");
lstrcat(szDataFile, c_szGroupFile);
hRCHandle = CreateFile( szDataFile, GENERIC_READ, FILE_SHARE_READ, NULL,
OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN, NULL);
if(hRCHandle == INVALID_HANDLE_VALUE)
{
// If we failed to open the Groups.dat file, try opening the backup file...
lstrcpy(szDataFile, szHomeDir);
lstrcat(szDataFile, "\\");
lstrcat(szDataFile, c_szBakupFile);
hRCHandle = CreateFile( szDataFile, GENERIC_READ, FILE_SHARE_READ, NULL,
OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN, NULL);
if(hRCHandle == INVALID_HANDLE_VALUE)
return hr;
}
cbRCFile = GetFileSize(hRCHandle, NULL);
if(!cbRCFile) // Empty File.
goto Done;
hRCFile = CreateFileMapping(hRCHandle, NULL, PAGE_READONLY, 0, 0, NULL);
if(hRCFile == NULL)
{
CloseHandle(hRCHandle);
return hr;
}
pBegin = (BYTE *)MapViewOfFile( hRCFile, FILE_MAP_READ, 0, 0, 0);
if(pBegin == NULL)
{
CloseHandle(hRCHandle);
CloseHandle(hRCFile);
return hr;
}
pCurr = pBegin;
pEnd = pCurr + cbRCFile;
if (!MemAlloc((void **)&pListGroups, MEMCHUNK))
{
hr = E_OUTOFMEMORY;
goto Done;
}
ZeroMemory((void*)pListGroups, MEMCHUNK);
hDatFile = FindFirstFile(szWildSearch, &rgFileData);
if(INVALID_HANDLE_VALUE != hDatFile)
{
while(bFileFound)
{
dwNumLines = 0;
pCurr = pBegin;
lstrcpy(szEnumFile, rgFileData.cFileName);
int nCounter = lstrlen(szEnumFile);
if(nCounter)
{
while(nCounter)
{
if(szEnumFile[nCounter] == '.')
{
szEnumFile[nCounter] = '\0';
nCounter--;
break;
}
nCounter--;
}
}
else
goto NextFile;
if(!nCounter)
goto NextFile;
// The following converts a given string into a number. It assumes that the string represents a hex value.
for(nCount = nCounter; nCount >= 0; nCount--)
{
nTempStore = (int)szEnumFile[nCount];
if((48 <= nTempStore && nTempStore <= 57) || (65 <= nTempStore && nTempStore <= 70) || (97 <= nTempStore && nTempStore <= 102))
{
if(48 <= nTempStore && nTempStore <= 57)
nTempStore -= 48;
if(65 <= nTempStore && nTempStore <= 70)
nTempStore -= 55;
if(97 <= nTempStore && nTempStore <= 102)
nTempStore -= 87;
dwFactor = 1;
int nLoop = nCounter - nCount;
while(nLoop)
{
dwFactor *= 16;
nLoop--;
}
dwNumLines += nTempStore*dwFactor;
}
else
goto NextFile;
}
while ((pCurr < pEnd) && (dwNumLines - 1))
{
nLine = 0;
while(!((pCurr[nLine] == 0x0D) && (pCurr[nLine + 1] == 0x0A)) && (pCurr + nLine < pEnd))
nLine++;
pCurr += (nLine + 2);
dwNumLines--;
}
nLine = 0;
while(!((pCurr[nLine] == 0x0D) && (pCurr[nLine + 1] == 0x0A)) && (pCurr + nLine < pEnd))
nLine++;
nLine++;
lstrcpyn(szSubList, (char*)pCurr, nLine);
if(nLine + 2 < nBalMem)
{
lstrcat(pListGroups, szSubList);
lstrcat(pListGroups, "\1");
nBalMem -= (nLine + 2);
}
else
{
int nPresent = lstrlen(pListGroups) + 1;
if(!MemRealloc((void **)&pListGroups, nPresent + MEMCHUNK))
{
hr = E_OUTOFMEMORY;
goto Done;
}
nBalMem += MEMCHUNK;
lstrcat(pListGroups, szSubList);
lstrcat(pListGroups, "\1");
nBalMem -= (nLine + 2);
}
NextFile:
bFileFound = FindNextFile(hDatFile, &rgFileData);
}
}
if(lstrlen(pListGroups))
{
nCount = 0;
while(pListGroups[nCount] != '\0')
{
if(pListGroups[nCount] == cPlaceHolder)
{
pListGroups[nCount] = '\0';
}
nCount++;
}
*ppListGroups = pListGroups;
hr = S_OK;
}
Done:
if(hRCHandle != INVALID_HANDLE_VALUE)
CloseHandle(hRCHandle);
if(pBegin)
UnmapViewOfFile(pBegin);
if(hRCFile)
CloseHandle(hRCFile);
return hr;
#endif // DEAD
}
STDMETHODIMP CAgentAcctImport::GetSettings2(DWORD_PTR dwCookie, IImnAccount *pAcct, IMPCONNINFO *pInfo)
{
if (pAcct == NULL ||
pInfo == NULL)
return(E_INVALIDARG);
return(IGetSettings(dwCookie, pAcct, pInfo));
}
CEnumAGNTACCT::CEnumAGNTACCT()
{
m_cRef = 1;
// m_iInfo
m_cInfo = 0;
m_rgInfo = NULL;
}
CEnumAGNTACCT::~CEnumAGNTACCT()
{
if (m_rgInfo != NULL)
MemFree(m_rgInfo);
}
STDMETHODIMP CEnumAGNTACCT::QueryInterface(REFIID riid, LPVOID *ppv)
{
if (ppv == NULL)
return(E_INVALIDARG);
*ppv = NULL;
if (IID_IUnknown == riid)
*ppv = (IUnknown *)this;
else if (IID_IEnumIMPACCOUNTS == riid)
*ppv = (IEnumIMPACCOUNTS *)this;
if (*ppv != NULL)
((LPUNKNOWN)*ppv)->AddRef();
else
return(E_NOINTERFACE);
return(S_OK);
}
STDMETHODIMP_(ULONG) CEnumAGNTACCT::AddRef()
{
return(++m_cRef);
}
STDMETHODIMP_(ULONG) CEnumAGNTACCT::Release()
{
if (--m_cRef == 0)
{
delete this;
return(0);
}
return(m_cRef);
}
HRESULT STDMETHODCALLTYPE CEnumAGNTACCT::Next(IMPACCOUNTINFO *pinfo)
{
if (pinfo == NULL)
return(E_INVALIDARG);
m_iInfo++;
if ((UINT)m_iInfo >= m_cInfo)
return(S_FALSE);
Assert(m_rgInfo != NULL);
pinfo->dwCookie = m_rgInfo[m_iInfo].dwCookie;
pinfo->dwReserved = 0;
lstrcpy(pinfo->szDisplay, m_rgInfo[m_iInfo].szDisplay);
return(S_OK);
}
HRESULT STDMETHODCALLTYPE CEnumAGNTACCT::Reset()
{
m_iInfo = -1;
return(S_OK);
}
HRESULT CEnumAGNTACCT::Init(AGNTNEWSACCTINFO *pinfo, int cinfo)
{
DWORD cb;
Assert(pinfo != NULL);
Assert(cinfo > 0);
cb = cinfo * sizeof(AGNTNEWSACCTINFO);
if (!MemAlloc((void **)&m_rgInfo, cb))
return(E_OUTOFMEMORY);
m_iInfo = -1;
m_cInfo = cinfo;
CopyMemory(m_rgInfo, pinfo, cb);
return(S_OK);
}
Reference in New Issue View Git Blame Copy Permalink