void FreeProws(LPWABOBJECT pWABObject, LPSRowSet prows)
{
ULONG irow;
if (!prows)
return;
for (irow = 0; irow < prows->cRows; ++irow)
pWABObject->FreeBuffer(prows->aRow[irow].lpProps);
pWABObject->FreeBuffer(prows);
}
void FreeWAB(HINSTANCE hWAB,
LPADRBOOK pAdrBook,
LPWABOBJECT pWABObject,
AdrBookTable *pTable,
UINT nNumEntries)
{
if (pTable && nNumEntries)
{
for (UINT i=0; i<nNumEntries; i++)
{
if (pTable[i].szName)
free(pTable[i].szName);
if (pTable[i].szEmail)
free(pTable[i].szEmail);
}
free(pTable);
}
if(pAdrBook)
pAdrBook->Release();
if(pWABObject)
pWABObject->Release();
if(hWAB)
FreeLibrary(hWAB);
return;
}
void CProcessor::ImportWAB()
{
HINSTANCE hinstLib;
HRESULT hRes;
LPADRBOOK lpAdrBook;
LPWABOBJECT lpWABObject;
DWORD Reserved2 = NULL;
fWABOpen procWABOpen;
{
TCHAR szWABDllPath[MAX_PATH];
DWORD dwType = 0;
ULONG cbData = sizeof(szWABDllPath);
HKEY hKey = NULL;
*szWABDllPath = '\0';
// First we look under the default WAB DLL path location in the
// Registry.
// WAB_DLL_PATH_KEY is defined in wabapi.h
//
if (ERROR_SUCCESS == RegOpenKeyEx(HKEY_LOCAL_MACHINE, WAB_DLL_PATH_KEY, 0, KEY_READ, &hKey))
RegQueryValueEx( hKey, "", NULL, &dwType, (LPBYTE) szWABDllPath, &cbData);
if(hKey) RegCloseKey(hKey);
// if the Registry came up blank, we do a loadlibrary on the wab32.dll
// WAB_DLL_NAME is defined in wabapi.h
//
hinstLib = LoadLibrary( (lstrlen(szWABDllPath)) ? szWABDllPath : WAB_DLL_NAME );
}
if (hinstLib != NULL)
{
procWABOpen = (fWABOpen) GetProcAddress(hinstLib, "WABOpen");
if (procWABOpen != NULL)
{
hRes = (procWABOpen)(&lpAdrBook,&lpWABObject,NULL,Reserved2); // WABOpen
_ASSERTE(hRes == S_OK);
if (hRes != S_OK) return;
ULONG lpcbEntryID;
ENTRYID *lpEntryID;
hRes = lpAdrBook->GetPAB(
&lpcbEntryID,
&lpEntryID
);
_ASSERTE(hRes == S_OK);
if (hRes != S_OK) return;
ULONG ulFlags = MAPI_BEST_ACCESS;
ULONG ulObjType = NULL;
LPUNKNOWN lpUnk = NULL;
hRes = lpAdrBook->OpenEntry(
lpcbEntryID,
lpEntryID,
NULL,
ulFlags,
&ulObjType,
&lpUnk
);
ulFlags = NULL;
//IABTable *lpTable;
if (ulObjType == MAPI_ABCONT)
{
IABContainer *lpContainer = static_cast <IABContainer *>(lpUnk);
LPMAPITABLE lpTable = NULL;
hRes = lpContainer->GetContentsTable(
ulFlags,
&lpTable
);
_ASSERT(lpTable);
ULONG ulRows, ulFound = 0, ulExisted = 0;
hRes = lpTable->GetRowCount(0,&ulRows);
_ASSERTE(hRes == S_OK);
SRowSet *lpRows;
hRes = lpTable->SetColumns( (LPSPropTagArray)&ptaEid, 0 );
hRes = lpTable->QueryRows(
ulRows, // Get all Rows
0,
&lpRows
);
for(ULONG i=0;i<lpRows->cRows;i++)
{
bool bBirthdayProcessed = false;
CBirthday* pbd = new CBirthday;
SRow *lpRow = &lpRows->aRow[i];
for(ULONG j=0;j<lpRow->cValues;j++)
{
SPropValue *lpProp = &lpRow->lpProps[j];
if(lpProp->ulPropTag == PR_BIRTHDAY)
{
SYSTEMTIME st;
FileTimeToSystemTime(&lpProp->Value.ft,&st);
pbd->m_date = CDate((BYTE)st.wDay, (BYTE)st.wMonth, (UINT)st.wYear);
bBirthdayProcessed = true;
}
if(lpProp->ulPropTag == PR_DISPLAY_NAME_A)
{
pbd->m_name = lpProp->Value.lpszA;
}
}
if( bBirthdayProcessed && pbd->m_date.ValidDate() )
{
ulFound++;
if( !WABExists(pbd) )
{
pbd->m_medium = MEDIUM_WAB;
Add( pbd );
}
else
{
delete pbd;
ulExisted++;
}
}
else
delete pbd;
lpWABObject->FreeBuffer(lpRow);
}
CString c;
c.Format( "Processed %d contacts\r\n\r\nFound %d birthdays of which\r\n%d already existed",
ulRows, ulFound, ulExisted );
AfxMessageBox( c, MB_OK|MB_SYSTEMMODAL, 0 );
lpWABObject->FreeBuffer(lpRows);
}
if(lpAdrBook)
lpAdrBook->Release();
if(lpWABObject)
lpWABObject->Release();
}
// This would be nice but leads to crashing Trillian
// FreeLibrary(hinstLib);
}
}
BOOL ReadWAB(LPADRBOOK pAdrBook,
LPWABOBJECT pWABObject,
AdrBookTable **ppTable,
UINT *pNumEntries)
{
ULONG ulObjType = 0;
LPMAPITABLE lpAB = NULL;
LPTSTR * lppszArray=NULL;
ULONG cRows = 0;
LPSRowSet lpRow = NULL;
LPSRowSet lpRowAB = NULL;
LPABCONT lpContainer = NULL;
int cNumRows = 0;
int nRows=0;
int nCount = 0;
HRESULT hr = E_FAIL;
ULONG lpcbEID;
LPENTRYID lpEID = NULL;
// Get the entryid of the root PAB container
//
hr = pAdrBook->GetPAB( &lpcbEID, &lpEID);
ulObjType = 0;
// Open the root PAB container
// This is where all the WAB contents reside
//
hr = pAdrBook->OpenEntry(lpcbEID,
(LPENTRYID)lpEID,
NULL,
0,
&ulObjType,
(LPUNKNOWN *)&lpContainer);
pWABObject->FreeBuffer(lpEID);
lpEID = NULL;
if(HR_FAILED(hr))
goto exit;
// Get a contents table of all the contents in the
// WABs root container
//
hr = lpContainer->GetContentsTable( 0,
&lpAB);
if(HR_FAILED(hr))
goto exit;
// Order the columns in the ContentsTable to conform to the
// ones we want - which are mainly DisplayName, EntryID and
// ObjectType
// The table is gauranteed to set the columns in the order
// requested
//
hr =lpAB->SetColumns( (LPSPropTagArray)&ptaEid, 0 );
if(HR_FAILED(hr))
goto exit;
// Reset to the beginning of the table
//
hr = lpAB->SeekRow( BOOKMARK_BEGINNING, 0, NULL );
if(HR_FAILED(hr))
goto exit;
// Read all the rows of the table one by one
//
do {
hr = lpAB->QueryRows(1, 0, &lpRowAB);
if(HR_FAILED(hr))
break;
if(lpRowAB)
{
cNumRows = lpRowAB->cRows;
if (cNumRows)
{
LPTSTR szName = lpRowAB->aRow[0].lpProps[ieidPR_DISPLAY_NAME].Value.lpszA;
LPTSTR szEmail = NULL;
LPENTRYID lpEID = (LPENTRYID) lpRowAB->aRow[0].lpProps[ieidPR_ENTRYID].Value.bin.lpb;
ULONG cbEID = lpRowAB->aRow[0].lpProps[ieidPR_ENTRYID].Value.bin.cb;
LPMAILUSER pMailUser = NULL;
ULONG ulObjType = 0;
ULONG ulValues;
LPSPropValue lpPropArray;
*ppTable = (AdrBookTable *) realloc(*ppTable,
sizeof(AdrBookTable) * (nCount+1));
(*ppTable)[nCount].szName =
(char *) calloc(strlen(szName)+1, 1);
strcpy((*ppTable)[nCount].szName, szName);
if (lpRowAB->aRow[0].lpProps[ieidPR_OBJECT_TYPE].Value.l ==
MAPI_MAILUSER)
{
pAdrBook->OpenEntry(cbEID,
lpEID,
NULL, // interface
0, // flags
&ulObjType,
(LPUNKNOWN *)&pMailUser);
if(pMailUser)
{
pMailUser->GetProps((LPSPropTagArray) &ptaEmail,
0,
&ulValues,
&lpPropArray);
pMailUser->Release();
}
if ((lpPropArray[iemailPR_EMAIL_ADDRESS].ulPropTag &
0xffff) == PT_ERROR)
{
szEmail =
lpPropArray[iemailPR_DISPLAY_NAME].Value.lpszA;
}
else
szEmail =
lpPropArray[iemailPR_EMAIL_ADDRESS].Value.lpszA;
if (szEmail != NULL)
{
(*ppTable)[nCount].szEmail =
(char *) calloc(strlen(szEmail)+1, 1);
strcpy((*ppTable)[nCount].szEmail, szEmail);
}
else
{
(*ppTable)[nCount].szEmail =
(char *) calloc(strlen("")+1, 1);
strcpy((*ppTable)[nCount].szEmail, "");
}
pWABObject->FreeBuffer(lpPropArray);
}
else
(*ppTable)[nCount].szEmail = NULL;
nCount++;
}
FreeProws(pWABObject, lpRowAB);
}
}while ( SUCCEEDED(hr) && cNumRows && lpRowAB) ;
exit:
if ( lpContainer )
lpContainer->Release();
if ( lpAB )
lpAB->Release();
*pNumEntries = nCount;
if (SUCCEEDED(hr))
return TRUE;
else
return FALSE;
}
