1040 lines
26 KiB
C
1040 lines
26 KiB
C
|
#include "ctlspriv.h"
|
||
|
#include <memory.h>
|
||
|
|
||
|
#define MRU_ORDERDIRTY 0x1000
|
||
|
|
||
|
#define DM_MRULAZY DM_TRACE
|
||
|
|
||
|
#define MAX_CHAR 126
|
||
|
#define BASE_CHAR TEXT('a')
|
||
|
|
||
|
typedef struct tagMRUDATA
|
||
|
{
|
||
|
UINT fFlags;
|
||
|
UINT uMax;
|
||
|
LPVOID lpfnCompare;
|
||
|
HKEY hKey;
|
||
|
#ifdef DEBUG
|
||
|
TCHAR szSubKey[32];
|
||
|
#endif
|
||
|
LPTSTR cOrder;
|
||
|
} MRUDATA, *PMRUDATA;
|
||
|
|
||
|
#define c_szMRU TEXT("MRUList")
|
||
|
|
||
|
#define NTHSTRING(p, n) (*((LPTSTR *)((LPBYTE)p+sizeof(MRUDATA))+n))
|
||
|
#define NTHDATA(p, n) (*((LPBYTE *)((LPBYTE)p+sizeof(MRUDATA))+n))
|
||
|
#define NUM_OVERHEAD 3
|
||
|
|
||
|
|
||
|
#ifdef VSTF
|
||
|
|
||
|
/*----------------------------------------------------------
|
||
|
Purpose: Validate the MRU structure
|
||
|
*/
|
||
|
BOOL IsValidPMRUDATA(PMRUDATA pmru)
|
||
|
{
|
||
|
return (IS_VALID_WRITE_PTR(pmru, MRUDATA) &&
|
||
|
(NULL == pmru->lpfnCompare || IS_VALID_CODE_PTR(pmru->lpfnCompare, void)));
|
||
|
}
|
||
|
|
||
|
#endif // VSTF
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
// Internal memcmp - saves loading crt's, cdecl so we can use
|
||
|
// as MRUCMPDATAPROC
|
||
|
|
||
|
int CDECL _mymemcmp(const void *pBuf1, const void *pBuf2, size_t cb)
|
||
|
{
|
||
|
// Take advantage of the intrinsic version from crtfree.h
|
||
|
return memcmp(pBuf1, pBuf2, cb);
|
||
|
}
|
||
|
|
||
|
|
||
|
// Use this macro to get the original size of the data.
|
||
|
#define DATASIZE(p) (*((LPDWORD)p))
|
||
|
// And this to get a pointer to the original data.
|
||
|
#define DATAPDATA(p) (p+sizeof(DWORD))
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
// For binary data we stick the size of the data at the begining and store the
|
||
|
// whole thing in one go.
|
||
|
BOOL MRUIsSameData(PMRUDATA pMRU, BYTE* pVal, const void *lpData, UINT cbData)
|
||
|
{
|
||
|
int cbUseSize;
|
||
|
MRUCMPDATAPROC lpfnCompare;
|
||
|
|
||
|
ASSERT(IS_VALID_STRUCT_PTR(pMRU, MRUDATA));
|
||
|
|
||
|
lpfnCompare = pMRU->lpfnCompare;
|
||
|
|
||
|
ASSERT(IS_VALID_CODE_PTR(lpfnCompare, MRUCMPDATAPROC));
|
||
|
|
||
|
// if there's something other than a mem compare,
|
||
|
// don't require the sizes to be equal in order for the
|
||
|
// data to be equivalent.
|
||
|
|
||
|
if (pMRU->lpfnCompare == _mymemcmp)
|
||
|
{
|
||
|
if (DATASIZE(pVal) != cbData)
|
||
|
return FALSE;
|
||
|
|
||
|
cbUseSize = cbData;
|
||
|
}
|
||
|
else
|
||
|
cbUseSize = min(DATASIZE(pVal), cbData);
|
||
|
|
||
|
return ((*lpfnCompare)(lpData, DATAPDATA(pVal), cbUseSize) == 0);
|
||
|
}
|
||
|
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
HANDLE WINAPI CreateMRUListLazy(LPMRUINFO lpmi, const void *lpData, UINT cbData, LPINT lpiSlot)
|
||
|
{
|
||
|
PTSTR pOrder, pNewOrder, pTemp;
|
||
|
LPBYTE pVal;
|
||
|
LONG cbVal;
|
||
|
DWORD dwDisposition;
|
||
|
DWORD dwType;
|
||
|
PMRUDATA pMRU = NULL;
|
||
|
HKEY hkeySubKey = NULL;
|
||
|
TCHAR szTemp[2];
|
||
|
UINT uMax = lpmi->uMax;
|
||
|
HKEY hKey = lpmi->hKey;
|
||
|
LPCTSTR lpszSubKey = lpmi->lpszSubKey;
|
||
|
MRUCMPPROC lpfnCompare = lpmi->lpfnCompare;
|
||
|
int cb;
|
||
|
|
||
|
#ifdef DEBUG
|
||
|
DWORD dwStart = GetTickCount();
|
||
|
#endif
|
||
|
if (!lpfnCompare)
|
||
|
{
|
||
|
lpfnCompare = (lpmi->fFlags & MRU_BINARY) ? (MRUCMPPROC)_mymemcmp :
|
||
|
((lpmi->fFlags & MRU_ANSI) ? (MRUCMPPROC)lstrcmpiA :
|
||
|
(MRUCMPPROC)lstrcmpi);
|
||
|
}
|
||
|
|
||
|
// limit to 126 so that we don't use extended chars
|
||
|
if (uMax > MAX_CHAR-BASE_CHAR)
|
||
|
uMax = MAX_CHAR-BASE_CHAR;
|
||
|
|
||
|
if (RegCreateKeyEx(hKey, lpszSubKey, 0L, (LPTSTR)c_szShell, REG_OPTION_NON_VOLATILE,
|
||
|
KEY_READ | KEY_WRITE, NULL, &hkeySubKey, &dwDisposition) != ERROR_SUCCESS)
|
||
|
{
|
||
|
goto Error1;
|
||
|
}
|
||
|
|
||
|
pOrder = (PTSTR)Alloc((uMax + 1) * sizeof(TCHAR));
|
||
|
if (!pOrder)
|
||
|
{
|
||
|
goto Error1;
|
||
|
}
|
||
|
|
||
|
cbVal = ((LONG)uMax + 1) * sizeof(TCHAR);
|
||
|
|
||
|
if (RegQueryValueEx(hkeySubKey, (LPTSTR)c_szMRU, NULL, &dwType, (LPBYTE)pOrder, &cbVal) != ERROR_SUCCESS)
|
||
|
{
|
||
|
// if not already in the registry, then start fresh
|
||
|
*pOrder = 0;
|
||
|
}
|
||
|
|
||
|
// Uppercase is not allowed
|
||
|
CharLower(pOrder);
|
||
|
|
||
|
// We allocate room for the MRUDATA structure, plus the order list,
|
||
|
// and the list of strings.
|
||
|
cb = (lpmi->fFlags & MRU_BINARY) ? sizeof(LPBYTE) : sizeof(LPTSTR);
|
||
|
pMRU = (PMRUDATA)Alloc(sizeof(MRUDATA)+(uMax*cb));
|
||
|
if (!pMRU)
|
||
|
{
|
||
|
goto Error2;
|
||
|
}
|
||
|
|
||
|
// Allocate space for the order list
|
||
|
pMRU->cOrder = (LPTSTR)Alloc((uMax+1)*sizeof(TCHAR));
|
||
|
if (!pMRU->cOrder)
|
||
|
{
|
||
|
Free(pMRU);
|
||
|
pMRU = NULL;
|
||
|
goto Error2;
|
||
|
}
|
||
|
|
||
|
pMRU->fFlags = lpmi->fFlags;
|
||
|
pMRU->uMax = uMax;
|
||
|
pMRU->lpfnCompare = lpfnCompare;
|
||
|
pMRU->hKey = hkeySubKey;
|
||
|
#ifdef DEBUG
|
||
|
lstrcpyn(pMRU->szSubKey, lpszSubKey, ARRAYSIZE(pMRU->szSubKey));
|
||
|
#endif
|
||
|
|
||
|
// Traverse through the MRU list, adding strings to the end of the
|
||
|
// list.
|
||
|
szTemp[1] = TEXT('\0');
|
||
|
for (pTemp = pOrder, pNewOrder = pMRU->cOrder; ; ++pTemp)
|
||
|
{
|
||
|
// Stop when we get to the end of the list.
|
||
|
szTemp[0] = *pTemp;
|
||
|
if (!szTemp[0])
|
||
|
{
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
if (lpmi->fFlags & MRU_BINARY)
|
||
|
{
|
||
|
// Check if in range and if we have already used this letter.
|
||
|
if ((UINT)(szTemp[0]-BASE_CHAR)>=uMax || NTHDATA(pMRU, szTemp[0]-BASE_CHAR))
|
||
|
{
|
||
|
continue;
|
||
|
}
|
||
|
// Get the value from the registry
|
||
|
cbVal = 0;
|
||
|
// first find the size
|
||
|
if ((RegQueryValueEx(hkeySubKey, szTemp, NULL, &dwType, NULL, &cbVal)
|
||
|
!= ERROR_SUCCESS) || (dwType != REG_BINARY))
|
||
|
{
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
// Binary data has the size at the begining so we'll need a little extra room.
|
||
|
pVal = (LPBYTE)Alloc(cbVal + sizeof(DWORD));
|
||
|
|
||
|
if (!pVal)
|
||
|
{
|
||
|
// REARCHITECT perhaps sort of error is in order.
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
// now really get it
|
||
|
DATASIZE(pVal) = cbVal;
|
||
|
if (RegQueryValueEx(hkeySubKey, szTemp, NULL, &dwType, pVal+sizeof(DWORD),
|
||
|
(LPDWORD)pVal) != ERROR_SUCCESS)
|
||
|
{
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
// Note that blank elements ARE allowed in the list.
|
||
|
NTHDATA(pMRU, szTemp[0]-BASE_CHAR) = pVal;
|
||
|
*pNewOrder++ = szTemp[0];
|
||
|
|
||
|
//
|
||
|
// OPTIMIZATION
|
||
|
// If lpData and lpiSlot are specified, we stop the enumeratation
|
||
|
// when we find the item.
|
||
|
//
|
||
|
if (lpData && lpiSlot)
|
||
|
{
|
||
|
// Check if we have the specified one or not.
|
||
|
if (MRUIsSameData(pMRU, pVal, lpData, cbData))
|
||
|
{
|
||
|
// Found it.
|
||
|
*lpiSlot = (INT) (pNewOrder - pMRU->cOrder);
|
||
|
|
||
|
TraceMsg(DM_MRULAZY, "CreateMRUListLazy found it. Copying %s", pTemp);
|
||
|
|
||
|
pMRU->fFlags |= MRU_LAZY;
|
||
|
//
|
||
|
// Copy the rest of slot. Notice that we don't load
|
||
|
// data for those slot.
|
||
|
//
|
||
|
for (pTemp++; *pTemp; pTemp++)
|
||
|
{
|
||
|
*pNewOrder++ = *pTemp;
|
||
|
}
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
|
||
|
// Check if in range and if we have already used this letter.
|
||
|
if ((UINT)(szTemp[0]-BASE_CHAR)>=uMax || NTHSTRING(pMRU, szTemp[0]-BASE_CHAR))
|
||
|
{
|
||
|
continue;
|
||
|
}
|
||
|
// Get the value from the registry
|
||
|
cbVal = 0;
|
||
|
// first find the size
|
||
|
if ((RegQueryValueEx(hkeySubKey, szTemp, NULL, &dwType, NULL, &cbVal)
|
||
|
!= ERROR_SUCCESS) || (dwType != REG_SZ))
|
||
|
{
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
cbVal *= sizeof(TCHAR);
|
||
|
pVal = (LPBYTE)Alloc(cbVal);
|
||
|
|
||
|
if (!pVal)
|
||
|
{
|
||
|
// REARCHITECT perhaps sort of error is in order.
|
||
|
continue;
|
||
|
}
|
||
|
// now really get it
|
||
|
if (RegQueryValueEx(hkeySubKey, szTemp, NULL, &dwType, (LPBYTE)pVal, &cbVal) != ERROR_SUCCESS)
|
||
|
{
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
// Note that blank elements are not allowed in the list.
|
||
|
if (*((LPTSTR)pVal))
|
||
|
{
|
||
|
NTHSTRING(pMRU, szTemp[0]-BASE_CHAR) = (LPTSTR)pVal;
|
||
|
*pNewOrder++ = szTemp[0];
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
Free(pVal);
|
||
|
pVal = NULL;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
/* NULL terminate the order list so we can tell how many strings there
|
||
|
* are.
|
||
|
*/
|
||
|
*pNewOrder = TEXT('\0');
|
||
|
|
||
|
if (lpData && lpiSlot)
|
||
|
{
|
||
|
TraceMsg(DM_MRULAZY, "CreateMRUListLazy. End of loop. %s", pMRU->cOrder);
|
||
|
// If we failed to find, put -1 in it.
|
||
|
if (!(pMRU->fFlags & MRU_LAZY))
|
||
|
{
|
||
|
*lpiSlot = -1;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/* Actually, this is success rather than an error.
|
||
|
*/
|
||
|
goto Error2;
|
||
|
|
||
|
Error2:
|
||
|
if (pOrder)
|
||
|
{
|
||
|
Free((HLOCAL)pOrder);
|
||
|
pOrder = NULL;
|
||
|
}
|
||
|
|
||
|
Error1:
|
||
|
if (!pMRU && hkeySubKey)
|
||
|
RegCloseKey(hkeySubKey);
|
||
|
|
||
|
return((HANDLE)pMRU);
|
||
|
}
|
||
|
|
||
|
HANDLE WINAPI CreateMRUList(LPMRUINFO lpmi)
|
||
|
{
|
||
|
return CreateMRUListLazy(lpmi, NULL, 0, NULL);
|
||
|
}
|
||
|
|
||
|
#ifdef UNICODE
|
||
|
|
||
|
//
|
||
|
// ANSI thunk
|
||
|
//
|
||
|
|
||
|
HANDLE WINAPI CreateMRUListLazyA(LPMRUINFOA lpmi, const void *lpData, UINT cbData, LPINT lpiSlot)
|
||
|
{
|
||
|
MRUINFOW MRUInfoW;
|
||
|
HANDLE hMRU;
|
||
|
|
||
|
MRUInfoW.cbSize = sizeof (MRUINFOW);
|
||
|
MRUInfoW.uMax = lpmi->uMax;
|
||
|
MRUInfoW.fFlags = lpmi->fFlags;
|
||
|
MRUInfoW.hKey = lpmi->hKey;
|
||
|
MRUInfoW.lpszSubKey = ProduceWFromA(CP_ACP, lpmi->lpszSubKey);
|
||
|
MRUInfoW.lpfnCompare = (MRUCMPPROCW)lpmi->lpfnCompare;
|
||
|
|
||
|
MRUInfoW.fFlags |= MRU_ANSI;
|
||
|
|
||
|
hMRU = CreateMRUListLazy(&MRUInfoW, lpData, cbData, lpiSlot);
|
||
|
|
||
|
FreeProducedString((LPWSTR)MRUInfoW.lpszSubKey);
|
||
|
|
||
|
return hMRU;
|
||
|
}
|
||
|
|
||
|
HANDLE WINAPI CreateMRUListA(LPMRUINFOA lpmi)
|
||
|
{
|
||
|
return CreateMRUListLazyA(lpmi, NULL, 0, NULL);
|
||
|
}
|
||
|
#else
|
||
|
|
||
|
//
|
||
|
// Unicode stub when this code is built ANSI
|
||
|
//
|
||
|
|
||
|
HANDLE WINAPI CreateMRUListW(LPMRUINFOW lpmi)
|
||
|
{
|
||
|
SetLastErrorEx(ERROR_CALL_NOT_IMPLEMENTED, SLE_WARNING);
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
HANDLE WINAPI CreateMRUListLazyW(LPMRUINFOW lpmi, const void *lpData, UINT cbData, LPINT lpiSlot)
|
||
|
{
|
||
|
SetLastErrorEx(ERROR_CALL_NOT_IMPLEMENTED, SLE_WARNING);
|
||
|
return NULL;
|
||
|
}
|
||
|
#endif
|
||
|
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
STDAPI_(void) FreeMRUList(HANDLE hMRU)
|
||
|
{
|
||
|
int i;
|
||
|
LPVOID *pTemp;
|
||
|
PMRUDATA pMRU = (PMRUDATA)hMRU;
|
||
|
|
||
|
ASSERT(IS_VALID_STRUCT_PTR(pMRU, MRUDATA));
|
||
|
|
||
|
if (pMRU)
|
||
|
{
|
||
|
pTemp = (pMRU->fFlags & MRU_BINARY) ?
|
||
|
&NTHDATA(pMRU, 0) : (LPBYTE *)&NTHSTRING(pMRU, 0);
|
||
|
|
||
|
if (pMRU->fFlags & MRU_ORDERDIRTY)
|
||
|
{
|
||
|
RegSetValueEx(pMRU->hKey, c_szMRU, 0L, REG_SZ, (CONST BYTE *)pMRU->cOrder,
|
||
|
sizeof(TCHAR) * (lstrlen(pMRU->cOrder) + 1));
|
||
|
}
|
||
|
|
||
|
for (i=pMRU->uMax-1; i>=0; --i, ++pTemp)
|
||
|
{
|
||
|
if (*pTemp)
|
||
|
{
|
||
|
if (pMRU->fFlags & MRU_BINARY)
|
||
|
{
|
||
|
Free((LPBYTE)*pTemp);
|
||
|
*pTemp = NULL;
|
||
|
}
|
||
|
else
|
||
|
Str_SetPtr((LPTSTR *)pTemp, NULL);
|
||
|
}
|
||
|
}
|
||
|
RegCloseKey(pMRU->hKey);
|
||
|
Free(pMRU->cOrder);
|
||
|
Free((HLOCAL)pMRU);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
/* Add a string to an MRU list.
|
||
|
*/
|
||
|
STDAPI_(int) AddMRUString(HANDLE hMRU, LPCTSTR szString)
|
||
|
{
|
||
|
/* The extra +1 is so that the list is NULL terminated.
|
||
|
*/
|
||
|
TCHAR cFirst;
|
||
|
int iSlot = -1;
|
||
|
LPTSTR lpTemp;
|
||
|
LPTSTR * pTemp;
|
||
|
int i;
|
||
|
UINT uMax;
|
||
|
MRUCMPPROC lpfnCompare;
|
||
|
BOOL fShouldWrite;
|
||
|
PMRUDATA pMRU = (PMRUDATA)hMRU;
|
||
|
|
||
|
#ifdef DEBUG
|
||
|
DWORD dwStart = GetTickCount();
|
||
|
#endif
|
||
|
|
||
|
ASSERT(IS_VALID_STRUCT_PTR(pMRU, MRUDATA));
|
||
|
|
||
|
if (hMRU == NULL)
|
||
|
return(-1); // Error
|
||
|
|
||
|
fShouldWrite = !(pMRU->fFlags & MRU_CACHEWRITE);
|
||
|
uMax = pMRU->uMax;
|
||
|
lpfnCompare = (MRUCMPPROC)pMRU->lpfnCompare;
|
||
|
|
||
|
/* Check if the string already exists in the list.
|
||
|
*/
|
||
|
for (i=0, pTemp=&NTHSTRING(pMRU, 0); (UINT)i<uMax; ++i, ++pTemp)
|
||
|
{
|
||
|
if (*pTemp)
|
||
|
{
|
||
|
int iResult;
|
||
|
|
||
|
#ifdef UNICODE
|
||
|
if (pMRU->fFlags & MRU_ANSI)
|
||
|
{
|
||
|
LPSTR lpStringA, lpTempA;
|
||
|
|
||
|
lpStringA = ProduceAFromW (CP_ACP, szString);
|
||
|
lpTempA = ProduceAFromW (CP_ACP, (LPWSTR)*pTemp);
|
||
|
|
||
|
iResult = (*lpfnCompare)((const void *)lpStringA, (const void *)lpTempA);
|
||
|
|
||
|
FreeProducedString (lpStringA);
|
||
|
FreeProducedString (lpTempA);
|
||
|
}
|
||
|
else
|
||
|
#endif
|
||
|
{
|
||
|
iResult = (*lpfnCompare)((const void *)szString, (const void *)*pTemp);
|
||
|
}
|
||
|
|
||
|
if (!iResult)
|
||
|
{
|
||
|
// found it, so don't do the write out
|
||
|
cFirst = i + BASE_CHAR;
|
||
|
iSlot = i;
|
||
|
goto FoundEntry;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/* Attempt to find an unused entry. Count up the used entries at the
|
||
|
* same time.
|
||
|
*/
|
||
|
for (i=0, pTemp=&NTHSTRING(pMRU, 0); ; ++i, ++pTemp)
|
||
|
{
|
||
|
if ((UINT)i >= uMax) // If we got to the end of the list.
|
||
|
{
|
||
|
// use the entry at the end of the cOrder list
|
||
|
cFirst = pMRU->cOrder[uMax-1];
|
||
|
pTemp = &NTHSTRING(pMRU, cFirst-BASE_CHAR);
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
// Is the entry not used?
|
||
|
if (!*pTemp)
|
||
|
{
|
||
|
// yes
|
||
|
cFirst = i+BASE_CHAR;
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (Str_SetPtr(pTemp, szString))
|
||
|
{
|
||
|
TCHAR szTemp[2];
|
||
|
|
||
|
iSlot = (int)(cFirst-BASE_CHAR);
|
||
|
|
||
|
szTemp[0] = cFirst;
|
||
|
szTemp[1] = TEXT('\0');
|
||
|
|
||
|
RegSetValueEx(pMRU->hKey, szTemp, 0L, REG_SZ, (CONST BYTE *)szString,
|
||
|
sizeof(TCHAR) * (lstrlen(szString) + 1));
|
||
|
|
||
|
fShouldWrite = TRUE;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
/* Since iSlot == -1, we will remove the reference to cFirst
|
||
|
* below.
|
||
|
*/
|
||
|
}
|
||
|
|
||
|
FoundEntry:
|
||
|
/* Remove any previous reference to cFirst.
|
||
|
*/
|
||
|
lpTemp = StrChr(pMRU->cOrder, cFirst);
|
||
|
if (lpTemp)
|
||
|
{
|
||
|
lstrcpy(lpTemp, lpTemp+1);
|
||
|
}
|
||
|
|
||
|
if (iSlot != -1)
|
||
|
{
|
||
|
// shift everything over and put cFirst at the front
|
||
|
hmemcpy(pMRU->cOrder+1, pMRU->cOrder, pMRU->uMax*sizeof(TCHAR));
|
||
|
pMRU->cOrder[0] = cFirst;
|
||
|
}
|
||
|
|
||
|
if (fShouldWrite)
|
||
|
{
|
||
|
RegSetValueEx(pMRU->hKey, c_szMRU, 0L, REG_SZ, (CONST BYTE *)pMRU->cOrder,
|
||
|
sizeof(TCHAR) * (lstrlen(pMRU->cOrder) + 1));
|
||
|
pMRU->fFlags &= ~MRU_ORDERDIRTY;
|
||
|
} else
|
||
|
pMRU->fFlags |= MRU_ORDERDIRTY;
|
||
|
|
||
|
#ifdef DEBUG
|
||
|
// DebugMsg(DM_TRACE, TEXT("AddMRU: %d msec"), LOWORD(GetTickCount()-dwStart));
|
||
|
#endif
|
||
|
return(iSlot);
|
||
|
}
|
||
|
|
||
|
|
||
|
#ifdef UNICODE
|
||
|
//
|
||
|
// ANSI thunk
|
||
|
//
|
||
|
|
||
|
STDAPI_(int) AddMRUStringA(HANDLE hMRU, LPCSTR szString)
|
||
|
{
|
||
|
LPWSTR lpStringW;
|
||
|
INT iResult;
|
||
|
|
||
|
lpStringW = ProduceWFromA(CP_ACP, szString);
|
||
|
|
||
|
iResult = AddMRUString(hMRU, lpStringW);
|
||
|
|
||
|
FreeProducedString (lpStringW);
|
||
|
|
||
|
return iResult;
|
||
|
}
|
||
|
|
||
|
#else
|
||
|
|
||
|
//
|
||
|
// Unicode stub when this code is build ANSI
|
||
|
//
|
||
|
|
||
|
STDAPI_(int) AddMRUStringW(HANDLE hMRU, LPCWSTR szString)
|
||
|
{
|
||
|
SetLastErrorEx(ERROR_CALL_NOT_IMPLEMENTED, SLE_WARNING);
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
#endif
|
||
|
|
||
|
|
||
|
/* Remove a string from an MRU list.
|
||
|
*/
|
||
|
STDAPI_(int) DelMRUString(HANDLE hMRU, int nItem)
|
||
|
{
|
||
|
BOOL bRet = FALSE;
|
||
|
LPTSTR lpTemp;
|
||
|
PMRUDATA pMRU = (PMRUDATA)hMRU;
|
||
|
|
||
|
ASSERT(IS_VALID_STRUCT_PTR(pMRU, MRUDATA));
|
||
|
|
||
|
if (pMRU)
|
||
|
{
|
||
|
//
|
||
|
// Make sure the index value is within the length of
|
||
|
// the string so we don't pick up some random value.
|
||
|
//
|
||
|
if (!InRange(nItem, 0, pMRU->uMax) || nItem >= lstrlen(pMRU->cOrder))
|
||
|
return FALSE;
|
||
|
|
||
|
// Be easy -- just remove the entry from the cOrder list
|
||
|
lpTemp = &pMRU->cOrder[nItem];
|
||
|
if (lpTemp)
|
||
|
{
|
||
|
int iSlot = *lpTemp - BASE_CHAR;
|
||
|
if (iSlot >= 0 && iSlot < MAX_CHAR - BASE_CHAR)
|
||
|
Str_SetPtr(&NTHSTRING(pMRU, iSlot), NULL);
|
||
|
lstrcpy(lpTemp, lpTemp+1);
|
||
|
|
||
|
if (!(pMRU->fFlags & MRU_CACHEWRITE))
|
||
|
{
|
||
|
RegSetValueEx(pMRU->hKey, c_szMRU, 0L, REG_SZ, (CONST BYTE *)pMRU->cOrder,
|
||
|
sizeof(TCHAR) * (lstrlen(pMRU->cOrder) + 1));
|
||
|
pMRU->fFlags &= ~MRU_ORDERDIRTY;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
pMRU->fFlags |= MRU_ORDERDIRTY;
|
||
|
}
|
||
|
|
||
|
bRet = TRUE;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return bRet;
|
||
|
}
|
||
|
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
// Add data to an MRU list.
|
||
|
STDAPI_(int) AddMRUData(HANDLE hMRU, const void *lpData, UINT cbData)
|
||
|
{
|
||
|
TCHAR cFirst;
|
||
|
int iSlot = -1;
|
||
|
LPTSTR lpTemp;
|
||
|
LPBYTE *ppData;
|
||
|
int i;
|
||
|
UINT uMax;
|
||
|
MRUCMPDATAPROC lpfnCompare;
|
||
|
PMRUDATA pMRU = (PMRUDATA)hMRU;
|
||
|
BOOL fShouldWrite;
|
||
|
|
||
|
#ifdef DEBUG
|
||
|
DWORD dwStart = GetTickCount();
|
||
|
#endif
|
||
|
|
||
|
ASSERT(IS_VALID_STRUCT_PTR(pMRU, MRUDATA));
|
||
|
|
||
|
if (hMRU == NULL)
|
||
|
return(-1); // Error
|
||
|
|
||
|
fShouldWrite = !(pMRU->fFlags & MRU_CACHEWRITE);
|
||
|
|
||
|
uMax = pMRU->uMax;
|
||
|
lpfnCompare = (MRUCMPDATAPROC)pMRU->lpfnCompare;
|
||
|
|
||
|
// Check if the data already exists in the list.
|
||
|
for (i=0, ppData=&NTHDATA(pMRU, 0); (UINT)i<uMax; ++i, ++ppData)
|
||
|
{
|
||
|
if (*ppData && MRUIsSameData(pMRU, *ppData, lpData, cbData))
|
||
|
{
|
||
|
// found it, so don't do the write out
|
||
|
cFirst = i + BASE_CHAR;
|
||
|
iSlot = i;
|
||
|
goto FoundEntry;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// When created "lazy", we are not supposed to add a new item.
|
||
|
//
|
||
|
if (pMRU->fFlags & MRU_LAZY)
|
||
|
{
|
||
|
ASSERT(0);
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
// Attempt to find an unused entry. Count up the used entries at the
|
||
|
// same time.
|
||
|
for (i=0, ppData=&NTHDATA(pMRU, 0); ; ++i, ++ppData)
|
||
|
{
|
||
|
if ((UINT)i >= uMax)
|
||
|
// If we got to the end of the list.
|
||
|
{
|
||
|
// use the entry at the end of the cOrder list
|
||
|
cFirst = pMRU->cOrder[uMax-1];
|
||
|
ppData = &NTHDATA(pMRU, cFirst-BASE_CHAR);
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
if (!*ppData)
|
||
|
// If the entry is not used.
|
||
|
{
|
||
|
cFirst = i+BASE_CHAR;
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
*ppData = ReAlloc(*ppData, cbData+sizeof(DWORD));
|
||
|
if (*ppData)
|
||
|
{
|
||
|
TCHAR szTemp[2];
|
||
|
|
||
|
*((LPDWORD)(*ppData)) = cbData;
|
||
|
hmemcpy(DATAPDATA(*ppData), lpData, cbData);
|
||
|
|
||
|
iSlot = (int)(cFirst-BASE_CHAR);
|
||
|
|
||
|
szTemp[0] = cFirst;
|
||
|
szTemp[1] = TEXT('\0');
|
||
|
|
||
|
RegSetValueEx(pMRU->hKey, szTemp, 0L, REG_BINARY, (LPVOID)lpData, cbData);
|
||
|
fShouldWrite = TRUE;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
// Since iSlot == -1, we will remove the reference to cFirst
|
||
|
// below.
|
||
|
}
|
||
|
|
||
|
FoundEntry:
|
||
|
// Remove any previous reference to cFirst.
|
||
|
lpTemp = StrChr(pMRU->cOrder, cFirst);
|
||
|
if (lpTemp)
|
||
|
{
|
||
|
lstrcpy(lpTemp, lpTemp+1);
|
||
|
}
|
||
|
|
||
|
if (iSlot != -1)
|
||
|
{
|
||
|
// shift everything over and put cFirst at the front
|
||
|
hmemcpy(pMRU->cOrder+1, pMRU->cOrder, pMRU->uMax*sizeof(TCHAR));
|
||
|
pMRU->cOrder[0] = cFirst;
|
||
|
}
|
||
|
|
||
|
if (fShouldWrite)
|
||
|
{
|
||
|
RegSetValueEx(pMRU->hKey, c_szMRU, 0L, REG_SZ, (CONST BYTE *)pMRU->cOrder,
|
||
|
sizeof(TCHAR) * (lstrlen(pMRU->cOrder) + 1));
|
||
|
pMRU->fFlags &= ~MRU_ORDERDIRTY;
|
||
|
} else
|
||
|
pMRU->fFlags |= MRU_ORDERDIRTY;
|
||
|
|
||
|
#ifdef DEBUG
|
||
|
// DebugMsg(DM_TRACE, TEXT("AddMRU: %d msec"), LOWORD(GetTickCount()-dwStart));
|
||
|
#endif
|
||
|
return(iSlot);
|
||
|
}
|
||
|
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
// Find data in an MRU list.
|
||
|
// Returns the slot number.
|
||
|
STDAPI_(int) FindMRUData(HANDLE hMRU, const void *lpData, UINT cbData, LPINT lpiSlot)
|
||
|
{
|
||
|
TCHAR cFirst;
|
||
|
int iSlot = -1;
|
||
|
LPTSTR lpTemp;
|
||
|
LPBYTE *ppData;
|
||
|
int i;
|
||
|
UINT uMax;
|
||
|
PMRUDATA pMRU = (PMRUDATA)hMRU;
|
||
|
|
||
|
#ifdef DEBUG
|
||
|
DWORD dwStart = GetTickCount();
|
||
|
#endif
|
||
|
|
||
|
ASSERT(IS_VALID_STRUCT_PTR(pMRU, MRUDATA));
|
||
|
|
||
|
if (hMRU == NULL)
|
||
|
return(-1); // Error state.
|
||
|
|
||
|
// Can't call this API when it's created lazily.
|
||
|
if (pMRU->fFlags & MRU_LAZY)
|
||
|
{
|
||
|
ASSERT(0);
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
uMax = pMRU->uMax;
|
||
|
|
||
|
/* Find the item in the list.
|
||
|
*/
|
||
|
for (i=0, ppData=&NTHDATA(pMRU, 0); (UINT)i<uMax; ++i, ++ppData)
|
||
|
{
|
||
|
if (!*ppData)
|
||
|
continue;
|
||
|
|
||
|
if (MRUIsSameData(pMRU, *ppData, lpData, cbData))
|
||
|
{
|
||
|
// So i now has the slot number in it.
|
||
|
if (lpiSlot != NULL)
|
||
|
*lpiSlot = i;
|
||
|
|
||
|
// Now convert the slot number into an index number
|
||
|
cFirst = i + BASE_CHAR;
|
||
|
lpTemp = StrChr(pMRU->cOrder, cFirst);
|
||
|
ASSERT(lpTemp);
|
||
|
return((lpTemp == NULL)? -1 : (int)(lpTemp - (LPTSTR)pMRU->cOrder));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
|
||
|
/* Find a string in an MRU list.
|
||
|
*/
|
||
|
STDAPI_(int) FindMRUString(HANDLE hMRU, LPCTSTR szString, LPINT lpiSlot)
|
||
|
{
|
||
|
/* The extra +1 is so that the list is NULL terminated.
|
||
|
*/
|
||
|
TCHAR cFirst;
|
||
|
int iSlot = -1;
|
||
|
LPTSTR lpTemp;
|
||
|
LPTSTR *pTemp;
|
||
|
int i;
|
||
|
UINT uMax;
|
||
|
MRUCMPPROC lpfnCompare;
|
||
|
PMRUDATA pMRU = (PMRUDATA)hMRU;
|
||
|
|
||
|
#ifdef DEBUG
|
||
|
DWORD dwStart = GetTickCount();
|
||
|
#endif
|
||
|
|
||
|
ASSERT(IS_VALID_STRUCT_PTR(pMRU, MRUDATA));
|
||
|
|
||
|
if (hMRU == NULL)
|
||
|
return(-1); // Error state.
|
||
|
|
||
|
uMax = pMRU->uMax;
|
||
|
lpfnCompare = (MRUCMPPROC)pMRU->lpfnCompare;
|
||
|
|
||
|
/* Find the item in the list.
|
||
|
*/
|
||
|
for (i=0, pTemp=&NTHSTRING(pMRU, 0); (UINT)i<uMax; ++i, ++pTemp)
|
||
|
{
|
||
|
if (*pTemp)
|
||
|
{
|
||
|
int iResult;
|
||
|
|
||
|
#ifdef UNICODE
|
||
|
if (pMRU->fFlags & MRU_ANSI)
|
||
|
{
|
||
|
LPSTR lpStringA, lpTempA;
|
||
|
|
||
|
lpStringA = ProduceAFromW (CP_ACP, szString);
|
||
|
lpTempA = ProduceAFromW (CP_ACP, (LPWSTR)*pTemp);
|
||
|
|
||
|
iResult = (*lpfnCompare)((const void *)lpStringA, (const void *)lpTempA);
|
||
|
|
||
|
FreeProducedString (lpStringA);
|
||
|
FreeProducedString (lpTempA);
|
||
|
}
|
||
|
else
|
||
|
#endif
|
||
|
{
|
||
|
iResult = (*lpfnCompare)((CONST VOID *)szString, (CONST VOID *)*pTemp);
|
||
|
}
|
||
|
|
||
|
if (!iResult)
|
||
|
{
|
||
|
// So i now has the slot number in it.
|
||
|
if (lpiSlot != NULL)
|
||
|
*lpiSlot = i;
|
||
|
|
||
|
// Now convert the slot number into an index number
|
||
|
cFirst = i + BASE_CHAR;
|
||
|
lpTemp = StrChr(pMRU->cOrder, cFirst);
|
||
|
return((lpTemp == NULL)? -1 : (int)(lpTemp - (LPTSTR)pMRU->cOrder));
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return(-1);
|
||
|
}
|
||
|
|
||
|
|
||
|
#ifdef UNICODE
|
||
|
//
|
||
|
// ANSI thunk
|
||
|
//
|
||
|
|
||
|
int WINAPI FindMRUStringA(HANDLE hMRU, LPCSTR szString, LPINT lpiSlot)
|
||
|
{
|
||
|
LPWSTR lpStringW;
|
||
|
INT iResult;
|
||
|
|
||
|
lpStringW = ProduceWFromA(CP_ACP, szString);
|
||
|
|
||
|
iResult = FindMRUString(hMRU, lpStringW, lpiSlot);
|
||
|
|
||
|
FreeProducedString (lpStringW);
|
||
|
|
||
|
return iResult;
|
||
|
}
|
||
|
|
||
|
#else
|
||
|
|
||
|
//
|
||
|
// Unicode stub when build ANSI
|
||
|
//
|
||
|
|
||
|
int WINAPI FindMRUStringW(HANDLE hMRU, LPCWSTR szString, LPINT lpiSlot)
|
||
|
{
|
||
|
SetLastErrorEx(ERROR_CALL_NOT_IMPLEMENTED, SLE_WARNING);
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
#endif
|
||
|
|
||
|
/* If lpszString is NULL, then this returns the number of MRU items or less than
|
||
|
* 0 on error.
|
||
|
* if nItem < 0, we'll return the number of items currently in the MRU.
|
||
|
* Otherwise, fill in as much of the buffer as possible (uLen includes the
|
||
|
* terminating NULL) and return the actual length of the string (including the
|
||
|
* terminating NULL) or less than 0 on error.
|
||
|
*/
|
||
|
STDAPI_(int) EnumMRUList(HANDLE hMRU, int nItem, LPVOID lpData, UINT uLen)
|
||
|
{
|
||
|
PMRUDATA pMRU = (PMRUDATA)hMRU;
|
||
|
int nItems = -1;
|
||
|
LPTSTR pTemp;
|
||
|
LPBYTE pData;
|
||
|
|
||
|
ASSERT(IS_VALID_STRUCT_PTR(pMRU, MRUDATA));
|
||
|
|
||
|
if (pMRU)
|
||
|
{
|
||
|
nItems = lstrlen(pMRU->cOrder);
|
||
|
|
||
|
if (nItem < 0 || !lpData)
|
||
|
return nItems;
|
||
|
|
||
|
if (nItem < nItems)
|
||
|
{
|
||
|
if (pMRU->fFlags & MRU_BINARY)
|
||
|
{
|
||
|
pData = NTHDATA(pMRU, pMRU->cOrder[nItem]-BASE_CHAR);
|
||
|
if (!pData)
|
||
|
return -1;
|
||
|
|
||
|
uLen = min((UINT)DATASIZE(pData), uLen);
|
||
|
hmemcpy(lpData, DATAPDATA(pData), uLen);
|
||
|
|
||
|
nItems = uLen;
|
||
|
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
pTemp = NTHSTRING(pMRU, pMRU->cOrder[nItem]-BASE_CHAR);
|
||
|
if (!pTemp)
|
||
|
return -1;
|
||
|
|
||
|
lstrcpyn((LPTSTR)lpData, pTemp, uLen);
|
||
|
|
||
|
nItems = lstrlen(pTemp);
|
||
|
}
|
||
|
}
|
||
|
else // revert to error condition
|
||
|
nItems = -1;
|
||
|
}
|
||
|
|
||
|
return nItems;
|
||
|
}
|
||
|
|
||
|
|
||
|
#ifdef UNICODE
|
||
|
|
||
|
STDAPI_(int) EnumMRUListA(HANDLE hMRU, int nItem, LPVOID lpData, UINT uLen)
|
||
|
{
|
||
|
int iResult = -1;
|
||
|
PMRUDATA pMRU = (PMRUDATA)hMRU;
|
||
|
|
||
|
ASSERT(IS_VALID_STRUCT_PTR(pMRU, MRUDATA));
|
||
|
|
||
|
if (pMRU)
|
||
|
{
|
||
|
LPVOID lpDataW;
|
||
|
BOOL bAllocatedMemory = FALSE;
|
||
|
|
||
|
//
|
||
|
// we need a temp buffer if the data is a string.
|
||
|
// but if it is binary, then we trust the callers buffer.
|
||
|
//
|
||
|
if (!(pMRU->fFlags & MRU_BINARY) && uLen && lpData)
|
||
|
{
|
||
|
lpDataW = LocalAlloc(LPTR, uLen * sizeof(TCHAR));
|
||
|
|
||
|
if (!lpDataW)
|
||
|
return -1;
|
||
|
|
||
|
bAllocatedMemory = TRUE;
|
||
|
}
|
||
|
else
|
||
|
lpDataW = lpData;
|
||
|
|
||
|
// call the real thing
|
||
|
iResult = EnumMRUList(hMRU, nItem, lpDataW, uLen);
|
||
|
|
||
|
//
|
||
|
// if the buffer was a string that we allocated
|
||
|
// then we need to thunk the string into the callers buffer
|
||
|
//
|
||
|
if (!(pMRU->fFlags & MRU_BINARY) && lpData && uLen && (iResult != -1))
|
||
|
{
|
||
|
WideCharToMultiByte(CP_ACP, 0, (LPWSTR)lpDataW, -1,
|
||
|
(LPSTR)lpData, uLen, NULL, NULL);
|
||
|
}
|
||
|
|
||
|
if (bAllocatedMemory)
|
||
|
LocalFree(lpDataW);
|
||
|
}
|
||
|
|
||
|
return iResult;
|
||
|
}
|
||
|
|
||
|
#else
|
||
|
|
||
|
STDAPI_(int) EnumMRUListW(HANDLE hMRU, int nItem, LPVOID lpData, UINT uLen)
|
||
|
{
|
||
|
SetLastErrorEx(ERROR_CALL_NOT_IMPLEMENTED, SLE_WARNING);
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
#endif // UNICODE
|