//============================================================================ // // DBCS and UNICODE aware string routines // // //============================================================================ #include "priv.h" #include "ids.h" #include // Get private NORM_ flag for StrEqIntl() #include #include "varutil.h" BOOL UnicodeFromAnsi(LPWSTR *, LPCSTR, LPWSTR, int); #define IS_DIGITA(ch) InRange(ch, '0', '9') #define IS_DIGITW(ch) InRange(ch, L'0', L'9') #define DM_INTERVAL 0 #ifdef UNIX #ifdef BIG_ENDIAN #define READNATIVEWORD(x) MAKEWORD(*(char*)(x), *(char*)((char*)(x) + 1)) #else #define READNATIVEWORD(x) MAKEWORD(*(char*)((char*)(x) + 1), *(char*)(x)) #endif #else #define READNATIVEWORD(x) (*(UNALIGNED WORD *)x) #endif __inline BOOL IsAsciiA(char ch) { return !(ch & 0x80); } __inline BOOL IsAsciiW(WCHAR ch) { return ch < 128; } __inline char Ascii_ToLowerA(char ch) { return (ch >= 'A' && ch <= 'Z') ? (ch - 'A' + 'a') : ch; } __inline WCHAR Ascii_ToLowerW(WCHAR ch) { return (ch >= L'A' && ch <= L'Z') ? (ch - L'A' + L'a') : ch; } // WARNING: all of these APIs do not setup DS, so you can not access // any data in the default data seg of this DLL. // // do not create any global variables... talk to chrisg if you don't // understand thid /* * ChrCmp - Case sensitive character comparison for DBCS * Assumes w1, wMatch are characters to be compared * Return FALSE if they match, TRUE if no match */ __inline BOOL ChrCmpA_inline(WORD w1, WORD wMatch) { /* Most of the time this won't match, so test it first for speed. */ if (LOBYTE(w1) == LOBYTE(wMatch)) { if (IsDBCSLeadByte(LOBYTE(w1))) { return(w1 != wMatch); } return FALSE; } return TRUE; } __inline BOOL ChrCmpW_inline(WCHAR w1, WCHAR wMatch) { return(!(w1 == wMatch)); } /* * ChrCmpI - Case insensitive character comparison for DBCS * Assumes w1, wMatch are characters to be compared; * HIBYTE of wMatch is 0 if not a DBC * Return FALSE if match, TRUE if not */ BOOL ChrCmpIA(WORD w1, WORD wMatch) { char sz1[3], sz2[3]; if (IsDBCSLeadByte(sz1[0] = LOBYTE(w1))) { sz1[1] = HIBYTE(w1); sz1[2] = '\0'; } else sz1[1] = '\0'; #if defined(MWBIG_ENDIAN) sz2[0] = LOBYTE(wMatch); sz2[1] = HIBYTE(wMatch); #else *(WORD *)sz2 = wMatch; #endif sz2[2] = '\0'; return lstrcmpiA(sz1, sz2); } BOOL ChrCmpIW(WCHAR w1, WCHAR wMatch) { WCHAR sz1[2], sz2[2]; sz1[0] = w1; sz1[1] = '\0'; sz2[0] = wMatch; sz2[1] = '\0'; return StrCmpIW(sz1, sz2); } LPWSTR Shlwapi_StrCpyW(LPWSTR pszDst, LPCWSTR pszSrc) { LPWSTR psz = pszDst; RIPMSG(NULL!=pszDst, "StrCpyW: Caller passed invalid pszDst"); RIPMSG(pszSrc && IS_VALID_STRING_PTRW(pszSrc, -1), "StrCpyW: Caller passed invalid pszSrc"); if (pszDst && pszSrc) { while (*pszDst++ = *pszSrc++) ; } return psz; } //*** StrCpyNX[AW] -- just like StrCpyN[AW], but returns ptr to EOS // NOTES // do we really need 'A' version? (for now we do for shell32 on 'old' // platforms that we test on but don't ship) LPSTR StrCpyNXA(LPSTR pszDst, LPCSTR pszSrc, int cchMax) { RIPMSG(cchMax >= 0, "StrCpyNXA: Caller passed bad cchMax"); RIPMSG(cchMax < 0 || (pszDst && IS_VALID_WRITE_BUFFER(pszDst, char, cchMax)), "StrCpyNXA: Caller passed bad pszDst"); RIPMSG(pszSrc && IS_VALID_STRING_PTRA(pszSrc, -1), "StrCpyNXA: Caller passed bad pszSrc"); // NOTE: Cannot use DEBUGWhackPathBuffer before copying because src and // dest might overlap. Must delay whacking until after we're done. if (0 < cchMax) { if (!pszSrc) goto NullItOut; // Leave room for the null terminator while (0 < --cchMax) { if (!(*pszDst++ = *pszSrc++)) { --pszDst; break; } } cchMax++; // in the cchMax>1 case, pszDst already points at the NULL, but reassigning it doesn't hurt NullItOut: // Whack the unused part of the buffer DEBUGWhackPathBufferA(pszDst, cchMax); *pszDst = '\0'; } return pszDst; } LPWSTR StrCpyNXW(LPWSTR pszDst, LPCWSTR pszSrc, int cchMax) { RIPMSG(cchMax >= 0, "StrCpyNXW: Caller passed bad cchMax"); RIPMSG(cchMax < 0 || (pszDst && IS_VALID_WRITE_BUFFER(pszDst, WCHAR, cchMax)), "StrCpyNXW: Caller passed bad pszDst"); RIPMSG(pszSrc && IS_VALID_STRING_PTRW(pszSrc, -1), "StrCpyNXW: Caller passed bad pszSrc"); // NOTE: Cannot use DEBUGWhackPathBuffer before copying because src and // dest might overlap. Must delay whacking until after we're done. if (0 < cchMax) { if (!pszSrc) // a test app passed in a NULL src ptr and we faulted, let's not fault here. goto NullItOut; // Leave room for the null terminator while (0 < --cchMax) { if (!(*pszDst++ = *pszSrc++)) { --pszDst; break; } } cchMax++; // in the cchMax>1 case, pszDst already points at the NULL, but reassigning it doesn't hurt NullItOut: // Whack the unused part of the buffer DEBUGWhackPathBufferW(pszDst, cchMax); *pszDst = L'\0'; } return pszDst; } LPWSTR StrCpyNW(LPWSTR pszDst, LPCWSTR pszSrc, int cchMax) { StrCpyNXW(pszDst, pszSrc, cchMax); return pszDst; } LPWSTR Shlwapi_StrCatW(LPWSTR pszDst, LPCWSTR pszSrc) { LPWSTR psz = pszDst; RIPMSG(pszDst && IS_VALID_STRING_PTRW(pszDst, -1), "StrCatW: Caller passed invalid pszDst"); RIPMSG(pszSrc && IS_VALID_STRING_PTRW(pszSrc, -1), "StrCatW: Caller passed invalid pszSrc"); if (pszDst && pszSrc) { while (0 != *pszDst) pszDst++; while (*pszDst++ = *pszSrc++) ; } return psz; } LWSTDAPI_(LPWSTR) StrCatBuffW(LPWSTR pszDest, LPCWSTR pszSrc, int cchDestBuffSize) { RIPMSG(pszDest && IS_VALID_STRING_PTRW(pszDest, -1), "StrCatBuffW: Caller passed invalid pszDest"); RIPMSG(pszSrc && IS_VALID_STRING_PTRW(pszSrc, -1), "StrCatBuffW: Caller passed invalid pszSrc"); RIPMSG(cchDestBuffSize >= 0, "StrCatBuffW: Caller passed invalid cchDestBuffSize"); RIPMSG(!(pszDest && IS_VALID_STRING_PTRW(pszDest, -1)) || cchDestBuffSize<0 || lstrlenW(pszDest) 0) { // call the shlwapi function here because win95 does not have lstrcpynW StrCpyNW(psz, pszSrc, cchDestBuffSize); } } return pszDest; } LWSTDAPI_(LPSTR) StrCatBuffA(LPSTR pszDest, LPCSTR pszSrc, int cchDestBuffSize) { RIPMSG(pszDest && IS_VALID_STRING_PTRA(pszDest, -1), "StrCatBuffA: Caller passed invalid pszDest"); RIPMSG(pszSrc && IS_VALID_STRING_PTRA(pszSrc, -1), "StrCatBuffA: Caller passed invalid pszSrc"); RIPMSG(cchDestBuffSize >= 0, "StrCatBuffA: Caller passed invalid cchDestBuffSize"); RIPMSG(!(pszDest && IS_VALID_STRING_PTRA(pszDest, -1)) || cchDestBuffSize<0 || lstrlen(pszDest) 0) { // Let kernel do the work for us. // // WARNING: We might generate a truncated DBCS sting becuase kernel's lstrcpynA // dosent check for this. Ask me if I care. lstrcpynA(psz, pszSrc, cchDestBuffSize); } } return pszDest; } /* StrNCat(front, back, count) - append count chars of back onto front */ LPSTR Shlwapi_StrNCatA(LPSTR front, LPCSTR back, int cchMax) { LPSTR start = front; RIPMSG(front && IS_VALID_STRING_PTRA(front, -1), "StrNCatA: Caller passed invalid front"); RIPMSG(back && IS_VALID_STRING_PTRA(front, cchMax), "StrNCatA: Caller passed invalid back"); RIPMSG(cchMax >= 0, "StrNCatA: Caller passed invalid cchMax"); if (front && back) { while (*front++) ; front--; lstrcpyn(front, back, cchMax); } return(start); } LPWSTR Shlwapi_StrNCatW(LPWSTR front, LPCWSTR back, int cchMax) { LPWSTR start = front; RIPMSG(front && IS_VALID_STRING_PTRW(front, -1), "StrNCatW: Caller passed invalid front"); RIPMSG(back && IS_VALID_STRING_PTRW(front, cchMax), "StrNCatW: Caller passed invalid back"); RIPMSG(cchMax >= 0, "StrNCatW: Caller passed invalid cchMax"); if (front && back) { while (*front++) ; front--; StrCpyNW(front, back, cchMax); } return(start); } /* * StrChr - Find first occurrence of character in string * Assumes lpStart points to start of null terminated string * wMatch is the character to match * returns ptr to the first occurrence of ch in str, NULL if not found. */ LPSTR _StrChrA(LPCSTR lpStart, WORD wMatch, BOOL fMBCS) { if (fMBCS) { for ( ; *lpStart; lpStart = AnsiNext(lpStart)) { if (!ChrCmpA_inline(READNATIVEWORD(lpStart), wMatch)) return((LPSTR)lpStart); } } else { for ( ; *lpStart; lpStart++) { if ((BYTE)*lpStart == LOBYTE(wMatch)) { return((LPSTR)lpStart); } } } return (NULL); } LPSTR StrChrA(LPCSTR lpStart, WORD wMatch) { CPINFO cpinfo; RIPMSG(lpStart && IS_VALID_STRING_PTR(lpStart, -1), "StrChrA: caller passed bad lpStart"); if (!lpStart) return NULL; return _StrChrA(lpStart, wMatch, GetCPInfo(CP_ACP, &cpinfo) && cpinfo.LeadByte[0]); } #ifdef ALIGNMENT_SCENARIO LPWSTR StrChrSlowW(const UNALIGNED WCHAR *lpStart, WCHAR wMatch) { for ( ; *lpStart; lpStart++) { if (!ChrCmpW_inline(*lpStart, wMatch)) { return((LPWSTR)lpStart); } } return NULL; } #endif LPWSTR StrChrW(LPCWSTR lpStart, WCHAR wMatch) { RIPMSG(lpStart && IS_VALID_STRING_PTRW(lpStart, -1), "StrChrW: caller passed bad lpStart"); if (!lpStart) return NULL; // // raymondc // Apparently, somebody is passing unaligned strings to StrChrW. // Find out who and make them stop. // RIPMSG(!((ULONG_PTR)lpStart & 1), "StrChrW: caller passed UNALIGNED lpStart"); // Assert alignedness #ifdef ALIGNMENT_SCENARIO // // Since unaligned strings arrive so rarely, put the slow // version in a separate function so the common case stays // fast. Believe it or not, we call StrChrW so often that // it is now a performance-sensitive function! // if ((ULONG_PTR)lpStart & 1) return StrChrSlowW(lpStart, wMatch); #endif for ( ; *lpStart; lpStart++) { if (!ChrCmpW_inline(*lpStart, wMatch)) { return((LPWSTR)lpStart); } } return (NULL); } /* * StrChrN - Find first occurrence of character in string * Assumes lpStart points to start of null terminated string * wMatch is the character to match * returns ptr to the first occurrence of ch in str, NULL if not found. */ #ifdef ALIGNMENT_SCENARIO LPWSTR StrChrSlowNW(const UNALIGNED WCHAR *lpStart, WCHAR wMatch, UINT cchMax) { LPCWSTR lpSentinel = lpStart + cchMax; for ( ; *lpStart && lpStart < lpSentinel; lpStart++) { if (!ChrCmpW_inline(*lpStart, wMatch)) { return((LPWSTR)lpStart); } } } #endif LPWSTR StrChrNW(LPCWSTR lpStart, WCHAR wMatch, UINT cchMax) { LPCWSTR lpSentinel = lpStart + cchMax; RIPMSG(lpStart && IS_VALID_STRING_PTRW(lpStart, -1), "StrChrNW: caller passed bad lpStart"); if (!lpStart) return NULL; // // raymondc // Apparently, somebody is passing unaligned strings to StrChrW. // Find out who and make them stop. // RIPMSG(!((ULONG_PTR)lpStart & 1), "StrChrNW: caller passed UNALIGNED lpStart"); // Assert alignedness #ifdef ALIGNMENT_SCENARIO // // Since unaligned strings arrive so rarely, put the slow // version in a separate function so the common case stays // fast. Believe it or not, we call StrChrW so often that // it is now a performance-sensitive function! // if ((ULONG_PTR)lpStart & 1) return StrChrSlowNW(lpStart, wMatch, cchMax); #endif for ( ; *lpStart && lpStart non-const } } psz = CharNextA(psz); } } return NULL; } /*---------------------------------------------------------- Purpose: Returns a pointer to the first occurrence of a character in psz that belongs to the set of characters in pszSet. The search does not include the null terminator. Returns: see above Cond: -- */ LPWSTR WINAPI StrPBrkW(LPCWSTR psz, LPCWSTR pszSet) { RIPMSG(psz && IS_VALID_STRING_PTRW(psz, -1), "StrPBrkA: caller passed bad psz"); RIPMSG(pszSet && IS_VALID_STRING_PTRW(pszSet, -1), "StrPBrkA: caller passed bad pszSet"); if (psz && pszSet) { while (*psz) { LPCWSTR pszSetT; for (pszSetT = pszSet; *pszSetT; pszSetT++) { if (*psz == *pszSetT) { // Found first character that matches return (LPWSTR)psz; // Const -> non-const } } psz++; } } return NULL; } int WINAPI StrToIntA(LPCSTR lpSrc) { RIPMSG(lpSrc && IS_VALID_STRING_PTRA(lpSrc, -1), "StrToIntA: Caller passed bad lpSrc"); if (lpSrc) { int n = 0; BOOL bNeg = FALSE; if (*lpSrc == '-') { bNeg = TRUE; lpSrc++; } while (IS_DIGITA(*lpSrc)) { n *= 10; n += *lpSrc - '0'; lpSrc++; } return bNeg ? -n : n; } return 0; } int WINAPI StrToIntW(LPCWSTR lpSrc) { RIPMSG(lpSrc && IS_VALID_STRING_PTRW(lpSrc, -1), "StrToIntW: Caller passed bad lpSrc"); if (lpSrc) { int n = 0; BOOL bNeg = FALSE; if (*lpSrc == L'-') { bNeg = TRUE; lpSrc++; } while (IS_DIGITW(*lpSrc)) { n *= 10; n += *lpSrc - L'0'; lpSrc++; } return bNeg ? -n : n; } return 0; } /*---------------------------------------------------------- Purpose: Special verion of atoi. Supports hexadecimal too. If this function returns FALSE, *phRet is set to 0. Returns: TRUE if the string is a number, or contains a partial number FALSE if the string is not a number dwFlags are STIF_ bitfield Cond: -- */ BOOL WINAPI StrToInt64ExW(LPCWSTR pszString, DWORD dwFlags, LONGLONG *pllRet) { BOOL bRet; RIPMSG(pszString && IS_VALID_STRING_PTRW(pszString, -1), "StrToInt64ExW: caller passed bad pszString"); if (pszString) { LONGLONG n; BOOL bNeg = FALSE; LPCWSTR psz; LPCWSTR pszAdj; // Skip leading whitespace // for (psz = pszString; *psz == L' ' || *psz == L'\n' || *psz == L'\t'; psz++) ; // Determine possible explicit signage // if (*psz == L'+' || *psz == L'-') { bNeg = (*psz == L'+') ? FALSE : TRUE; psz++; } // Or is this hexadecimal? // pszAdj = psz+1; if ((STIF_SUPPORT_HEX & dwFlags) && *psz == L'0' && (*pszAdj == L'x' || *pszAdj == L'X')) { // Yes // (Never allow negative sign with hexadecimal numbers) bNeg = FALSE; psz = pszAdj+1; pszAdj = psz; // Do the conversion // for (n = 0; ; psz++) { if (IS_DIGITW(*psz)) n = 0x10 * n + *psz - L'0'; else { WCHAR ch = *psz; int n2; if (ch >= L'a') ch -= L'a' - L'A'; n2 = ch - L'A' + 0xA; if (n2 >= 0xA && n2 <= 0xF) n = 0x10 * n + n2; else break; } } // Return TRUE if there was at least one digit bRet = (psz != pszAdj); } else { // No pszAdj = psz; // Do the conversion for (n = 0; IS_DIGITW(*psz); psz++) n = 10 * n + *psz - L'0'; // Return TRUE if there was at least one digit bRet = (psz != pszAdj); } if (pllRet) { *pllRet = bNeg ? -n : n; } } else { bRet = FALSE; } return bRet; } /*---------------------------------------------------------- Purpose: ansi wrapper for StrToInt64ExW. Returns: see StrToInt64ExW Cond: -- */ BOOL WINAPI StrToInt64ExA( LPCSTR pszString, DWORD dwFlags, // STIF_ bitfield LONGLONG FAR * pllRet) { BOOL bRet; RIPMSG(pszString && IS_VALID_STRING_PTRA(pszString, -1), "StrToInt64ExA: caller passed bad pszString"); if (pszString) { // Most strings will simply use this temporary buffer, but UnicodeFromAnsi // will allocate a buffer if the supplied string is bigger. WCHAR szBuf[MAX_PATH]; LPWSTR pwszString; bRet = UnicodeFromAnsi(&pwszString, pszString, szBuf, SIZECHARS(szBuf)); if (bRet) { bRet = StrToInt64ExW(pwszString, dwFlags, pllRet); UnicodeFromAnsi(&pwszString, NULL, szBuf, 0); } } else { bRet = FALSE; } return bRet; } /*---------------------------------------------------------- Purpose: Calls StrToInt64ExA (the real work horse), and then casts down to an int. Returns: see StrToInt64ExA */ BOOL WINAPI StrToIntExA( LPCSTR pszString, DWORD dwFlags, int *piRet) { LONGLONG llVal; BOOL fReturn; RIPMSG(pszString && IS_VALID_STRING_PTRA(pszString, -1), "StrToIntExA: caller passed bad pszString"); fReturn = StrToInt64ExA(pszString, dwFlags, &llVal); *piRet = fReturn ? (int)llVal : 0; return(fReturn); } /*---------------------------------------------------------- Purpose: Calls StrToInt64ExW (the real work horse), and then casts down to an int. Returns: see StrToInt64ExW */ BOOL WINAPI StrToIntExW( LPCWSTR pwszString, DWORD dwFlags, // STIF_ bitfield int FAR * piRet) { LONGLONG llVal; BOOL fReturn; RIPMSG(pwszString && IS_VALID_STRING_PTRW(pwszString, -1), "StrToIntExW: caller passed bad pwszString"); fReturn = StrToInt64ExW(pwszString, dwFlags, &llVal); *piRet = fReturn ? (int)llVal : 0; return(fReturn); } /*---------------------------------------------------------- Purpose: Returns an integer value specifying the length of the substring in psz that consists entirely of characters in pszSet. If psz begins with a character not in pszSet, then this function returns 0. This is a DBCS-safe version of the CRT strspn(). Returns: see above Cond: -- */ int StrSpnA(LPCSTR psz, LPCSTR pszSet) { LPCSTR pszT = psz; RIPMSG(psz && IS_VALID_STRING_PTRA(psz, -1), "StrSpnA: caller passed bad psz"); RIPMSG(pszSet && IS_VALID_STRING_PTRA(pszSet, -1), "StrSpnA: caller passed bad pszSet"); if (psz && pszSet) { // Go thru the string to be inspected for ( ; *pszT; pszT = CharNextA(pszT)) { LPCSTR pszSetT; // Go thru the char set for (pszSetT = pszSet; *pszSetT; pszSetT = CharNextA(pszSetT)) { if (*pszSetT == *pszT) { if ( !IsDBCSLeadByte(*pszSetT) ) { break; // Chars match } else if (pszSetT[1] == pszT[1]) { break; // Chars match } } } // End of char set? if (0 == *pszSetT) { break; // Yes, no match on this inspected char } } } return (int)(pszT - psz); } /*---------------------------------------------------------- Purpose: Returns an integer value specifying the length of the substring in psz that consists entirely of characters in pszSet. If psz begins with a character not in pszSet, then this function returns 0. This is a DBCS-safe version of the CRT strspn(). Returns: see above Cond: -- */ STDAPI_(int) StrSpnW(LPCWSTR psz, LPCWSTR pszSet) { LPCWSTR pszT = psz; RIPMSG(psz && IS_VALID_STRING_PTRW(psz, -1), "StrSpnW: caller passed bad psz"); RIPMSG(pszSet && IS_VALID_STRING_PTRW(pszSet, -1), "StrSpnW: caller passed bad pszSet"); if (psz && pszSet) { // Go thru the string to be inspected for ( ; *pszT; pszT++) { LPCWSTR pszSetT; // Go thru the char set for (pszSetT = pszSet; *pszSetT != *pszT; pszSetT++) { if (0 == *pszSetT) { // Reached end of char set without finding a match return (int)(pszT - psz); } } } } return (int)(pszT - psz); } // StrCSpn: return index to first char of lpStr that is present in lpSet. // Includes the NUL in the comparison; if no lpSet chars are found, returns // the index to the NUL in lpStr. // Just like CRT strcspn. // int StrCSpnA(LPCSTR lpStr, LPCSTR lpSet) { LPCSTR lp = lpStr; RIPMSG(lpStr && IS_VALID_STRING_PTRA(lpStr, -1), "StrCSpnA: Caller passed bad lpStr"); RIPMSG(lpSet && IS_VALID_STRING_PTRA(lpSet, -1), "StrCSpnA: Caller passed bad lpSet"); if (lpStr && lpSet) { // nature of the beast: O(lpStr*lpSet) work while (*lp) { if (StrChrA(lpSet, READNATIVEWORD(lp))) return (int)(lp-lpStr); lp = AnsiNext(lp); } } return (int)(lp-lpStr); // ==lstrlen(lpStr) } int StrCSpnW(LPCWSTR lpStr, LPCWSTR lpSet) { LPCWSTR lp = lpStr; RIPMSG(lpStr && IS_VALID_STRING_PTRW(lpStr, -1), "StrCSpnW: Caller passed bad lpStr"); RIPMSG(lpSet && IS_VALID_STRING_PTRW(lpSet, -1), "StrCSpnW: Caller passed bad lpSet"); if (lpStr && lpSet) { // nature of the beast: O(lpStr*lpSet) work while (*lp) { if (StrChrW(lpSet, *lp)) return (int)(lp-lpStr); lp++; } } return (int)(lp-lpStr); // ==lstrlen(lpStr) } // StrCSpnI: case-insensitive version of StrCSpn. // int StrCSpnIA(LPCSTR lpStr, LPCSTR lpSet) { LPCSTR lp = lpStr; RIPMSG(lpStr && IS_VALID_STRING_PTRA(lpStr, -1), "StrCSpnIA: Caller passed bad lpStr"); RIPMSG(lpSet && IS_VALID_STRING_PTRA(lpSet, -1), "StrCSpnIA: Caller passed bad lpSet"); if (lpStr && lpSet) { // nature of the beast: O(lpStr*lpSet) work while (*lp) { if (StrChrIA(lpSet, READNATIVEWORD(lp))) return (int)(lp-lpStr); lp = AnsiNext(lp); } } return (int)(lp-lpStr); // ==lstrlen(lpStr) } int StrCSpnIW(LPCWSTR lpStr, LPCWSTR lpSet) { LPCWSTR lp = lpStr; RIPMSG(lpStr && IS_VALID_STRING_PTRW(lpStr, -1), "StrCSpnIW: Caller passed bad lpStr"); RIPMSG(lpSet && IS_VALID_STRING_PTRW(lpSet, -1), "StrCSpnIW: Caller passed bad lpSet"); if (lpStr && lpSet) { // nature of the beast: O(lpStr*lpSet) work while (*lp) { if (StrChrIW(lpSet, *lp)) return (int)(lp-lpStr); lp++; } } return (int)(lp-lpStr); // ==lstrlen(lpStr) } /* * StrCmpN - Compare n bytes * * returns See lstrcmp return values. */ int _StrCmpNA(LPCSTR lpStr1, LPCSTR lpStr2, int nChar, BOOL fMBCS) { if (lpStr1 && lpStr2) { LPCSTR lpszEnd = lpStr1 + nChar; char sz1[4]; char sz2[4]; if (fMBCS) { for ( ; (lpszEnd > lpStr1) && (*lpStr1 || *lpStr2); lpStr1 = AnsiNext(lpStr1), lpStr2 = AnsiNext(lpStr2)) { WORD w1; WORD w2; // If either pointer is at the null terminator already, // we want to copy just one byte to make sure we don't read // past the buffer (might be at a page boundary). w1 = (*lpStr1) ? READNATIVEWORD(lpStr1) : 0; w2 = (*lpStr2) ? READNATIVEWORD(lpStr2) : 0; // (ChrCmpA returns FALSE if the characters match) // Do the characters match? if (ChrCmpA_inline(w1, w2)) { // No; determine the lexical value of the comparison // (since ChrCmp just returns true/false). // Since the character may be a DBCS character; we // copy two bytes into each temporary buffer // (in preparation for the lstrcmp call). (*(WORD *)sz1) = w1; (*(WORD *)sz2) = w2; // Add null terminators to temp buffers *AnsiNext(sz1) = 0; *AnsiNext(sz2) = 0; return lstrcmpA(sz1, sz2); } } } else { for ( ; (lpszEnd > lpStr1) && (*lpStr1 || *lpStr2); lpStr1++, lpStr2++) { if (*lpStr1 != *lpStr2) { // No; determine the lexical value of the comparison // (since ChrCmp just returns true/false). sz1[0] = *lpStr1; sz2[0] = *lpStr2; sz1[1] = sz2[1] = '\0'; return lstrcmpA(sz1, sz2); } } } } return 0; } STDAPI_(int) StrCmpNA(LPCSTR psz1, LPCSTR psz2, int nChar) { CPINFO cpinfo; RIPMSG(nChar == 0 || (psz1 && IS_VALID_STRING_PTRA(psz1, nChar)), "StrCmpNA: Caller passed bad psz1"); RIPMSG(nChar == 0 || (psz2 && IS_VALID_STRING_PTRA(psz2, nChar)), "StrCmpNA: Caller passed bad psz2"); RIPMSG(nChar >= 0, "StrCmpNA: caller passed bad nChar"); return _StrCmpNA(psz1, psz2, nChar, GetCPInfo(CP_ACP, &cpinfo) && cpinfo.LeadByte[0]); } // cch1 and cch2 are the maximum # of chars to compare int _StrCmpLocaleW(DWORD dwFlags, LPCWSTR psz1, int cch1, LPCWSTR psz2, int cch2) { int i = CompareStringW(GetThreadLocale(), dwFlags, psz1, cch1, psz2, cch2); if (!i) { i = CompareStringW(LOCALE_SYSTEM_DEFAULT, dwFlags, psz1, cch1, psz2, cch2); } return i - CSTR_EQUAL; } int _StrCmpLocaleA(DWORD dwFlags, LPCSTR psz1, int cch1, LPCSTR psz2, int cch2) { int i = CompareStringA(GetThreadLocale(), dwFlags, psz1, cch1, psz2, cch2); if (!i) { i = CompareStringA(LOCALE_SYSTEM_DEFAULT, dwFlags, psz1, cch1, psz2, cch2); } return i - CSTR_EQUAL; } STDAPI_(int) StrCmpNW(LPCWSTR psz1, LPCWSTR psz2, int nChar) { RIPMSG(nChar==0 || (psz1 && IS_VALID_STRING_PTRW(psz1, nChar)), "StrCmpNW: Caller passed bad psz1"); RIPMSG(nChar==0 || (psz2 && IS_VALID_STRING_PTRW(psz2, nChar)), "StrCmpNW: Caller passed bad psz2"); RIPMSG(nChar>=0, "StrCmpNA: caller passed bad nChar"); return _StrCmpLocaleW(NORM_STOP_ON_NULL, psz1, nChar, psz2, nChar); } /* * Compare n bytes, case insensitive * * returns See lstrcmpi return values. */ int StrCmpNIA(LPCSTR psz1, LPCSTR psz2, int nChar) { int nChar1, nChar2; RIPMSG(nChar==0 || (psz1 && IS_VALID_STRING_PTRA(psz1, nChar)), "StrCmpNIA: Caller passed bad psz1"); RIPMSG(nChar==0 || (psz2 && IS_VALID_STRING_PTRA(psz2, nChar)), "StrCmpNIA: Caller passed bad psz2"); RIPMSG(nChar>=0, "StrCmpNIA: caller passed bad nChar"); // Include the (nChar && (!psz1 || !psz2)) cases here so we go through the // validation layer and return the appropriate invalid parameter error code // instead of faulting on Win95. // // NOTE! That this means that StrCmpNI(NULL, NULL, 0) on NT returns -2 // but StrCmpNI(NULL, NULL, 0) on Win9x returns 0. This has always been // the case -- changing it is too scary for app compat reasons. // // NORM_STOP_ON_NULL is not supported by the ANSI version on NT // so we have to emulate it if (nChar && (!psz1 || !psz2)) { // This is the error scenario we are forcing through nChar1 = nChar; nChar2 = nChar; } else { // nChar1 = min(nChar, lstrlen(psz1)) // except that the "for" loop will not read more than nChar // characters from psz1 because psz1 might not be NULL-terminated for (nChar1 = 0; nChar1 < nChar && psz1[nChar1]; nChar1++) { } // And similarly for nChar2 for (nChar2 = 0; nChar2 < nChar && psz2[nChar2]; nChar2++) { } } return _StrCmpLocaleA(NORM_IGNORECASE, psz1, nChar1, psz2, nChar2); } int StrCmpNIW(LPCWSTR psz1, LPCWSTR psz2, int nChar) { RIPMSG(nChar==0 || (psz1 && IS_VALID_STRING_PTRW(psz1, nChar)), "StrCmpNIW: Caller passed bad psz1"); RIPMSG(nChar==0 || (psz2 && IS_VALID_STRING_PTRW(psz2, nChar)), "StrCmpNIW: Caller passed bad psz2"); RIPMSG(nChar>=0, "StrCmpNW: caller passed bad nChar"); return _StrCmpLocaleW(NORM_IGNORECASE | NORM_STOP_ON_NULL, psz1, nChar, psz2, nChar); } /* * StrRStrI - Search for last occurrence of a substring * * Assumes lpSource points to the null terminated source string * lpLast points to where to search from in the source string * lpLast is not included in the search * lpSrch points to string to search for * returns last occurrence of string if successful; NULL otherwise */ LPSTR StrRStrIA(LPCSTR lpSource, LPCSTR lpLast, LPCSTR lpSrch) { LPCSTR lpFound = NULL; RIPMSG(lpSource && IS_VALID_STRING_PTRA(lpSource, -1), "StrRStrIA: Caller passed bad lpSource"); RIPMSG(!lpLast || (IS_VALID_STRING_PTRA(lpLast, -1) && lpLast>=lpSource && lpLast<=lpSource+lstrlenA(lpSource)), "StrRStrIA: Caller passed bad lpLast"); RIPMSG(lpSrch && IS_VALID_STRING_PTRA(lpSrch, -1) && *lpSrch, "StrRStrIA: Caller passed bad lpSrch"); if (!lpLast) lpLast = lpSource + lstrlenA(lpSource); if (lpSource && lpSrch && *lpSrch) { WORD wMatch; UINT uLen; LPCSTR lpStart; wMatch = READNATIVEWORD(lpSrch); wMatch = (UINT)(IsDBCSLeadByte(LOBYTE(wMatch)) ? wMatch : LOBYTE(wMatch)); uLen = lstrlenA(lpSrch); lpStart = lpSource; while (*lpStart && (lpStart < lpLast)) { if (!ChrCmpIA(READNATIVEWORD(lpStart), wMatch)) { if (StrCmpNIA(lpStart, lpSrch, uLen) == 0) lpFound = lpStart; } lpStart = AnsiNext(lpStart); } } return((LPSTR)lpFound); } LPWSTR StrRStrIW(LPCWSTR lpSource, LPCWSTR lpLast, LPCWSTR lpSrch) { LPCWSTR lpFound = NULL; RIPMSG(lpSource && IS_VALID_STRING_PTRW(lpSource, -1), "StrRStrIW: Caller passed bad lpSource"); RIPMSG(!lpLast || (IS_VALID_STRING_PTRW(lpLast, -1) && lpLast>=lpSource && lpLast<=lpSource+lstrlenW(lpSource)), "StrRStrIW: Caller passed bad lpLast"); RIPMSG(lpSrch && IS_VALID_STRING_PTRW(lpSrch, -1) && *lpSrch, "StrRStrIW: Caller passed bad lpSrch"); if (!lpLast) lpLast = lpSource + lstrlenW(lpSource); if (lpSource && lpSrch && *lpSrch) { WCHAR wMatch; UINT uLen; LPCWSTR lpStart; wMatch = *lpSrch; uLen = lstrlenW(lpSrch); lpStart = lpSource; while (*lpStart && (lpStart < lpLast)) { if (!ChrCmpIW(*lpStart, wMatch)) { if (StrCmpNIW(lpStart, lpSrch, uLen) == 0) lpFound = lpStart; } lpStart++; } } return((LPWSTR)lpFound); } /* * StrStr - Search for first occurrence of a substring * * Assumes lpSource points to source string * lpSrch points to string to search for * returns first occurrence of string if successful; NULL otherwise */ LPSTR StrStrA(LPCSTR lpFirst, LPCSTR lpSrch) { RIPMSG(lpFirst && IS_VALID_STRING_PTRA(lpFirst, -1), "StrStrA: Caller passed bad lpFirst"); RIPMSG(lpSrch && IS_VALID_STRING_PTRA(lpSrch, -1), "StrStrA: Caller passed bad lpSrch"); if (lpFirst && lpSrch) { UINT uLen; WORD wMatch; CPINFO cpinfo; BOOL fMBCS = GetCPInfo(CP_ACP, &cpinfo) && cpinfo.LeadByte[0]; uLen = (UINT)lstrlenA(lpSrch); wMatch = READNATIVEWORD(lpSrch); for ( ; (lpFirst=_StrChrA(lpFirst, wMatch, fMBCS))!=0 && _StrCmpNA(lpFirst, lpSrch, uLen, fMBCS); lpFirst=AnsiNext(lpFirst)) continue; /* continue until we hit the end of the string or get a match */ return((LPSTR)lpFirst); } return(NULL); } LPWSTR StrStrW(LPCWSTR lpFirst, LPCWSTR lpSrch) { RIPMSG(lpFirst && IS_VALID_STRING_PTRW(lpFirst, -1), "StrStrW: Caller passed bad lpFirst"); RIPMSG(lpSrch && IS_VALID_STRING_PTRW(lpSrch, -1), "StrStrW: Caller passed bad lpSrch"); if (lpFirst && lpSrch) { UINT uLen; WCHAR wMatch; uLen = (UINT)lstrlenW(lpSrch); wMatch = *lpSrch; for ( ; (lpFirst=StrChrW(lpFirst, wMatch))!=0 && StrCmpNW(lpFirst, lpSrch, uLen); lpFirst++) continue; /* continue until we hit the end of the string or get a match */ return (LPWSTR)lpFirst; } return NULL; } /* * StrStrN - Search for first occurrence of a substring * * Assumes lpSource points to source string * lpSrch points to string to search for * returns first occurrence of string if successful; NULL otherwise */ LPWSTR StrStrNW(LPCWSTR lpFirst, LPCWSTR lpSrch, UINT cchMax) { RIPMSG(lpFirst && IS_VALID_STRING_PTRW(lpFirst, cchMax), "StrStrW: Caller passed bad lpFirst"); RIPMSG(lpSrch && IS_VALID_STRING_PTRW(lpSrch, cchMax), "StrStrW: Caller passed bad lpSrch"); if (lpFirst && lpSrch) { UINT uLen; WCHAR wMatch; LPCWSTR lpSentinel = lpFirst+cchMax; uLen = (UINT)lstrlenW(lpSrch); wMatch = *lpSrch; // the first two conditions in this loop signify failure when they eval to false, // while the third condition signifies success. We need to special case the second // condition at the end of the function because it doesn't automatically cause the // right value to be returned while((lpFirst=StrChrNW(lpFirst, wMatch, cchMax))!=0 && cchMax>=uLen &&StrCmpNW(lpFirst, lpSrch, uLen)) { lpFirst++; cchMax=(UINT)(lpSentinel-lpFirst); }/* continue until we hit the end of the string or get a match */ if(cchMax= uLen && StrCmpNIW(lpFirst, lpSrch, uLen)) { lpFirst++; cchMax = (UINT)(lpSentinel - lpFirst); }/* continue until we hit the end of the string or get a match */ if(cchMax= -1, "StrIsIntlEqualA: Caller passed invalid nChar"); dwFlags |= NORM_STOP_ON_NULL; // only supported on NT return 0 == _StrCmpLocaleA(dwFlags, lpString1, nChar, lpString2, nChar); } BOOL StrIsIntlEqualW(BOOL fCaseSens, LPCWSTR psz1, LPCWSTR psz2, int nChar) { RIPMSG(psz1 && IS_VALID_STRING_PTRW(psz1, nChar), "StrIsIntlEqualW: Caller passed invalid psz1"); RIPMSG(psz2 && IS_VALID_STRING_PTRW(psz2, nChar), "StrIsIntlEqualW: Caller passed invalid psz2"); RIPMSG(nChar >= -1, "StrIsIntlEqualW: Caller passed invalid nChar"); return 0 == _StrCmpLocaleW(fCaseSens ? NORM_STOP_ON_NULL : NORM_IGNORECASE | NORM_STOP_ON_NULL, psz1, nChar, psz2, nChar); } // This is stolen from shell32 - util.c #define LODWORD(_qw) (DWORD)(_qw) const short c_aOrders[] = {IDS_BYTES, IDS_ORDERKB, IDS_ORDERMB, IDS_ORDERGB, IDS_ORDERTB, IDS_ORDERPB, IDS_ORDEREB}; void Int64ToStr(LONGLONG n, LPWSTR lpBuffer) { WCHAR szTemp[40]; LONGLONG iChr; iChr = 0; do { szTemp[iChr++] = L'0' + (WCHAR)(n % 10); n = n / 10; } while (n != 0); do { iChr--; *lpBuffer++ = szTemp[iChr]; } while (iChr != 0); *lpBuffer++ = L'\0'; } // // Obtain NLS info about how numbers should be grouped. // // The annoying thing is that LOCALE_SGROUPING and NUMBERFORMAT // have different ways of specifying number grouping. // // LOCALE NUMBERFMT Sample Country // // 3;0 3 1,234,567 United States // 3;2;0 32 12,34,567 India // 3 30 1234,567 ?? // // Not my idea. That's the way it works. // // Bonus treat - Win9x doesn't support complex number formats, // so we return only the first number. // UINT GetNLSGrouping(void) { UINT grouping; LPWSTR psz; WCHAR szGrouping[32]; // If no locale info, then assume Western style thousands if (!GetLocaleInfoW(LOCALE_USER_DEFAULT, LOCALE_SGROUPING, szGrouping, ARRAYSIZE(szGrouping))) return 3; grouping = 0; psz = szGrouping; for (;;) { if (*psz == L'0') break; // zero - stop else if ((UINT)(*psz - L'0') < 10) // digit - accumulate it grouping = grouping * 10 + (UINT)(*psz - L'0'); else if (*psz) // punctuation - ignore it { } else // end of string, no "0" found { grouping = grouping * 10; // put zero on end (see examples) break; // and finished } psz++; } return grouping; } // Sizes of various stringized numbers #define MAX_INT64_SIZE 30 // 2^64 is less than 30 chars long #define MAX_COMMA_NUMBER_SIZE (MAX_INT64_SIZE + 10) // takes a DWORD add commas etc to it and puts the result in the buffer LPWSTR CommifyString(LONGLONG n, LPWSTR pszBuf, UINT cchBuf) { WCHAR szNum[MAX_COMMA_NUMBER_SIZE], szSep[5]; NUMBERFMTW nfmt; nfmt.NumDigits = 0; nfmt.LeadingZero = 0; nfmt.Grouping = GetNLSGrouping(); GetLocaleInfoW(LOCALE_USER_DEFAULT, LOCALE_STHOUSAND, szSep, ARRAYSIZE(szSep)); nfmt.lpDecimalSep = nfmt.lpThousandSep = szSep; nfmt.NegativeOrder = 0; Int64ToStr(n, szNum); if (GetNumberFormatW(LOCALE_USER_DEFAULT, 0, szNum, &nfmt, pszBuf, cchBuf) == 0) StrCpyNW(pszBuf, szNum, cchBuf); return pszBuf; } /* converts numbers into sort formats * 532 -> 523 bytes * 1340 -> 1.3KB * 23506 -> 23.5KB * -> 2.4MB * -> 5.2GB */ LPWSTR StrFormatByteSizeW(LONGLONG n, LPWSTR pszBuf, UINT cchBuf) { RIPMSG(pszBuf && IS_VALID_WRITE_BUFFER(pszBuf, WCHAR, cchBuf), "StrFormatByteSizeW: Caller passed invalid pszBuf"); DEBUGWhackPathBufferW(pszBuf, cchBuf); if (pszBuf) { WCHAR szWholeNum[32], szOrder[32]; int iOrder; // If the size is less than 1024, then the order should be bytes we have nothing // more to figure out if (n < 1024) { wnsprintfW(szWholeNum, ARRAYSIZE(szWholeNum), L"%d", LODWORD(n)); iOrder = 0; } else { UINT uInt, uLen, uDec; WCHAR szFormat[8]; // Find the right order for (iOrder = 1; iOrder < ARRAYSIZE(c_aOrders) -1 && n >= 1000L * 1024L; n >>= 10, iOrder++); /* do nothing */ uInt = LODWORD(n >> 10); CommifyString(uInt, szWholeNum, ARRAYSIZE(szWholeNum)); uLen = lstrlenW(szWholeNum); if (uLen < 3) { uDec = LODWORD(n - (LONGLONG)uInt * 1024L) * 1000 / 1024; // At this point, uDec should be between 0 and 1000 // we want get the top one (or two) digits. uDec /= 10; if (uLen == 2) uDec /= 10; // Note that we need to set the format before getting the // intl char. StringCchCopyW(szFormat, ARRAYSIZE(szFormat), L"%02d"); szFormat[2] = TEXT('0') + 3 - uLen; GetLocaleInfoW(LOCALE_USER_DEFAULT, LOCALE_SDECIMAL, szWholeNum + uLen, ARRAYSIZE(szWholeNum) - uLen); uLen = lstrlenW(szWholeNum); wnsprintfW(szWholeNum + uLen, ARRAYSIZE(szWholeNum) - uLen, szFormat, uDec); } } MLLoadStringW(c_aOrders[iOrder], szOrder, ARRAYSIZE(szOrder)); wnsprintfW(pszBuf, cchBuf, szOrder, szWholeNum); } return pszBuf; } // dw - the nubmer to be converted // pszBuf - buffer for the resulting string // cchBuf - Max characters in Buffer LPSTR StrFormatByteSize64A(LONGLONG dw, LPSTR pszBuf, UINT cchBuf) { WCHAR szT[32]; DEBUGWhackPathBuffer(pszBuf, cchBuf); StrFormatByteSizeW(dw, szT, SIZECHARS(szT)); SHUnicodeToAnsi(szT, pszBuf, cchBuf); return pszBuf; } LPSTR StrFormatByteSizeA(DWORD dw, LPSTR pszBuf, UINT cchBuf) { return StrFormatByteSize64A((LONGLONG)dw, pszBuf, cchBuf); } LPWSTR StrFormatKBSizeW(LONGLONG n, LPWSTR pszBuf, UINT cchBuf) { RIPMSG(pszBuf && IS_VALID_WRITE_BUFFER(pszBuf, WCHAR, cchBuf), "StrFormatKBSizeW: Caller passed invalid pszBuf"); DEBUGWhackPathBufferW(pszBuf, cchBuf); if (pszBuf) { static WCHAR s_szOrder[16] = {0}; WCHAR szNum[64]; if (s_szOrder[0] == TEXT('\0')) LoadStringW(HINST_THISDLL, IDS_ORDERKB, s_szOrder, ARRAYSIZE(s_szOrder)); CommifyString((n + 1023) / 1024, szNum, ARRAYSIZE(szNum)); wnsprintfW(pszBuf, cchBuf, s_szOrder, szNum); } return pszBuf; } LPSTR StrFormatKBSizeA(LONGLONG n, LPSTR pszBuf, UINT cchBuf) { WCHAR szNum[64]; DEBUGWhackPathBufferA(pszBuf, cchBuf); StrFormatKBSizeW(n, szNum, ARRAYSIZE(szNum)); SHUnicodeToAnsi(szNum, pszBuf, cchBuf); return pszBuf; } // Win95 does not support the wide-char version of lstrcmp, lstrcmpi // Wrapper for lstrcmpW so it works on Win95 int StrCmpW(LPCWSTR pwsz1, LPCWSTR pwsz2) { RIPMSG(pwsz1 && IS_VALID_STRING_PTRW(pwsz1, -1), "StrCmpW: Caller passed invalid pwsz1"); RIPMSG(pwsz2 && IS_VALID_STRING_PTRW(pwsz2, -1), "StrCmpW: Caller passed invalid pwsz2"); return _StrCmpLocaleW(0, pwsz1, -1, pwsz2, -1); } // Wrapper for lstrcmpiW so it works on Win95 int StrCmpIW(LPCWSTR pwsz1, LPCWSTR pwsz2) { RIPMSG(pwsz1 && IS_VALID_STRING_PTRW(pwsz1, -1), "StrCmpIW: Caller passed invalid pwsz1"); RIPMSG(pwsz2 && IS_VALID_STRING_PTRW(pwsz2, -1), "StrCmpIW: Caller passed invalid pwsz2"); return _StrCmpLocaleW(NORM_IGNORECASE, pwsz1, -1, pwsz2, -1); } /*---------------------------------------------------------- Purpose: Trim the string pszTrimMe of any leading or trailing characters that are in pszTrimChars. Returns: TRUE if anything was stripped */ STDAPI_(BOOL) StrTrimA(IN OUT LPSTR pszTrimMe, LPCSTR pszTrimChars) { BOOL bRet = FALSE; RIPMSG(pszTrimMe && IS_VALID_STRING_PTRA(pszTrimMe, -1), "StrTrimA: Caller passed invalid pszTrimMe"); RIPMSG(pszTrimChars && IS_VALID_STRING_PTRA(pszTrimChars, -1), "StrTrimA: Caller passed invalid pszTrimChars"); if (pszTrimMe && pszTrimChars) { LPSTR psz; LPSTR pszStartMeat; LPSTR pszMark = NULL; /* Trim leading characters. */ psz = pszTrimMe; while (*psz && StrChrA(pszTrimChars, READNATIVEWORD(psz))) psz = CharNextA(psz); pszStartMeat = psz; /* Trim trailing characters. */ // (The old algorithm used to start from the end and go // backwards, but that is piggy because DBCS version of // CharPrev iterates from the beginning of the string // on every call.) while (*psz) { if (StrChrA(pszTrimChars, READNATIVEWORD(psz))) { if (!pszMark) { pszMark = psz; } } else { pszMark = NULL; } psz = CharNextA(psz); } // Any trailing characters to clip? if (pszMark) { // Yes *pszMark = '\0'; bRet = TRUE; } /* Relocate stripped string. */ if (pszStartMeat > pszTrimMe) { /* (+ 1) for null terminator. */ MoveMemory(pszTrimMe, pszStartMeat, CbFromCchA(lstrlenA(pszStartMeat) + 1)); bRet = TRUE; } else ASSERT(pszStartMeat == pszTrimMe); ASSERT(IS_VALID_STRING_PTRA(pszTrimMe, -1)); } return bRet; } /*---------------------------------------------------------- Purpose: Trim the string pszTrimMe of any leading or trailing characters that are in pszTrimChars. Returns: TRUE if anything was stripped */ STDAPI_(BOOL) StrTrimW(IN OUT LPWSTR pszTrimMe, LPCWSTR pszTrimChars) { BOOL bRet = FALSE; RIPMSG(pszTrimMe && IS_VALID_STRING_PTRW(pszTrimMe, -1), "StrTrimW: Caller passed invalid pszTrimMe"); RIPMSG(pszTrimChars && IS_VALID_STRING_PTRW(pszTrimChars, -1), "StrTrimW: Caller passed invalid pszTrimChars"); if (pszTrimMe && pszTrimChars) { LPWSTR psz; LPWSTR pszStartMeat; LPWSTR pszMark = NULL; /* Trim leading characters. */ psz = pszTrimMe; while (*psz && StrChrW(pszTrimChars, *psz)) psz++; pszStartMeat = psz; /* Trim trailing characters. */ // (The old algorithm used to start from the end and go // backwards, but that is piggy because DBCS version of // CharPrev iterates from the beginning of the string // on every call.) while (*psz) { if (StrChrW(pszTrimChars, *psz)) { if (!pszMark) { pszMark = psz; } } else { pszMark = NULL; } psz++; } // Any trailing characters to clip? if (pszMark) { // Yes *pszMark = '\0'; bRet = TRUE; } /* Relocate stripped string. */ if (pszStartMeat > pszTrimMe) { /* (+ 1) for null terminator. */ MoveMemory(pszTrimMe, pszStartMeat, CbFromCchW(lstrlenW(pszStartMeat) + 1)); bRet = TRUE; } else ASSERT(pszStartMeat == pszTrimMe); ASSERT(IS_VALID_STRING_PTRW(pszTrimMe, -1)); } return bRet; } /*---------------------------------------------------------- Purpose: Compare strings using C runtime (ASCII) collation rules. Returns: < 0 if pch1 < pch2 = 0 if pch1 == pch2 > 0 if pch1 > pch2 */ LWSTDAPI_(int) StrCmpNCA(LPCSTR pch1, LPCSTR pch2, int n) { if (n == 0) return 0; while (--n && *pch1 && *pch1 == *pch2) { pch1++; pch2++; } return *(unsigned char *)pch1 - *(unsigned char *)pch2; } /*---------------------------------------------------------- Purpose: Compare strings using C runtime (ASCII) collation rules. Returns: < 0 if pch1 < pch2 = 0 if pch1 == pch2 > 0 if pch1 > pch2 */ LWSTDAPI_(int) StrCmpNCW(LPCWSTR pch1, LPCWSTR pch2, int n) { if (n == 0) return 0; while (--n && *pch1 && *pch1 == *pch2) { pch1++; pch2++; } return *pch1 - *pch2; } /*---------------------------------------------------------- Purpose: Compare strings using C runtime (ASCII) collation rules. Returns: < 0 if pch1 < pch2 = 0 if pch1 == pch2 > 0 if pch1 > pch2 */ LWSTDAPI_(int) StrCmpNICA(LPCSTR pch1, LPCSTR pch2, int n) { int ch1, ch2; if (n != 0) { do { ch1 = *pch1++; if (ch1 >= 'A' && ch1 <= 'Z') ch1 += 'a' - 'A'; ch2 = *pch2++; if (ch2 >= 'A' && ch2 <= 'Z') ch2 += 'a' - 'A'; } while ( --n && ch1 && (ch1 == ch2) ); return ch1 - ch2; } else { return 0; } } /*---------------------------------------------------------- Purpose: Compare strings using C runtime (ASCII) collation rules. Returns: < 0 if pch1 < pch2 = 0 if pch1 == pch2 > 0 if pch1 > pch2 */ LWSTDAPI_(int) StrCmpNICW(LPCWSTR pch1, LPCWSTR pch2, int n) { int ch1, ch2; if (n != 0) { do { ch1 = *pch1++; if (ch1 >= L'A' && ch1 <= L'Z') ch1 += L'a' - L'A'; ch2 = *pch2++; if (ch2 >= L'A' && ch2 <= L'Z') ch2 += L'a' - L'A'; } while ( --n && ch1 && (ch1 == ch2) ); return ch1 - ch2; } else { return 0; } } /*---------------------------------------------------------- Purpose: Compare strings using C runtime (ASCII) collation rules. Returns: < 0 if pch1 < pch2 = 0 if pch1 == pch2 > 0 if pch1 > pch2 */ LWSTDAPI_(int) StrCmpCA(LPCSTR pch1, LPCSTR pch2) { while (*pch1 && (*pch1 == *pch2)) { ++pch1; ++pch2; } return *(unsigned char *)pch1 - *(unsigned char *)pch2; } /*---------------------------------------------------------- Purpose: Compare strings using C runtime (ASCII) collation rules. Returns: < 0 if pch1 < pch2 = 0 if pch1 == pch2 > 0 if pch1 > pch2 */ LWSTDAPI_(int) StrCmpCW(LPCWSTR pch1, LPCWSTR pch2) { while (*pch1 && (*pch1 == *pch2)) { ++pch1; ++pch2; } return *pch1 - *pch2; } /*---------------------------------------------------------- Purpose: Compare strings using C runtime (ASCII) collation rules. Returns: < 0 if pch1 < pch2 = 0 if pch1 == pch2 > 0 if pch1 > pch2 */ LWSTDAPI_(int) StrCmpICA(LPCSTR pch1, LPCSTR pch2) { int ch1, ch2; do { ch1 = *pch1++; if (ch1 >= 'A' && ch1 <= 'Z') ch1 += 'a' - 'A'; ch2 = *pch2++; if (ch2 >= 'A' && ch2 <= 'Z') ch2 += 'a' - 'A'; } while (ch1 && (ch1 == ch2)); return ch1 - ch2; } /*---------------------------------------------------------- Purpose: Compare strings using C runtime (ASCII) collation rules. Returns: < 0 if pch1 < pch2 = 0 if pch1 == pch2 > 0 if pch1 > pch2 */ LWSTDAPI_(int) StrCmpICW(LPCWSTR pch1, LPCWSTR pch2) { int ch1, ch2; do { ch1 = *pch1++; if (ch1 >= L'A' && ch1 <= L'Z') ch1 += L'a' - L'A'; ch2 = *pch2++; if (ch2 >= L'A' && ch2 <= L'Z') ch2 += L'a' - L'A'; } while (ch1 && (ch1 == ch2)); return ch1 - ch2; } LWSTDAPI StrRetToStrW(STRRET *psr, LPCITEMIDLIST pidl, WCHAR **ppsz) { HRESULT hres = S_OK; switch (psr->uType) { case STRRET_WSTR: *ppsz = psr->pOleStr; psr->pOleStr = NULL; // avoid alias hres = *ppsz ? S_OK : E_FAIL; break; case STRRET_OFFSET: hres = SHStrDupA(STRRET_OFFPTR(pidl, psr), ppsz); break; case STRRET_CSTR: hres = SHStrDupA(psr->cStr, ppsz); break; default: *ppsz = NULL; hres = E_FAIL; } return hres; } LWSTDAPI StrRetToBSTR(STRRET *psr, LPCITEMIDLIST pidl, BSTR *pbstr) { switch (psr->uType) { case STRRET_WSTR: { LPWSTR psz = psr->pOleStr; psr->pOleStr = NULL; // avoid alias *pbstr = SysAllocString(psz); CoTaskMemFree(psz); break; } case STRRET_OFFSET: *pbstr = SysAllocStringA(STRRET_OFFPTR(pidl, psr)); break; case STRRET_CSTR: *pbstr = SysAllocStringA(psr->cStr); break; default: *pbstr = NULL; return E_FAIL; } return (*pbstr) ? S_OK : E_OUTOFMEMORY; } HRESULT DupWideToAnsi(LPCWSTR pwsz, LPSTR *ppsz) { UINT cch = WideCharToMultiByte(CP_ACP, 0, pwsz, -1, NULL, 0, NULL, NULL) + 1; *ppsz = CoTaskMemAlloc(cch * sizeof(**ppsz)); if (*ppsz) { SHUnicodeToAnsi(pwsz, *ppsz, cch); return S_OK; } return E_OUTOFMEMORY; } HRESULT DupAnsiToAnsi(LPCSTR psz, LPSTR *ppsz) { int cch = lstrlenA(psz) + 1; *ppsz = (LPSTR)CoTaskMemAlloc(cch * sizeof(**ppsz)); if (*ppsz) { StringCchCopyA(*ppsz, cch, psz); return S_OK; } return E_OUTOFMEMORY; } LWSTDAPI StrRetToStrA(STRRET *psr, LPCITEMIDLIST pidl, CHAR **ppsz) { HRESULT hres; LPWSTR pwsz; switch (psr->uType) { case STRRET_WSTR: hres = DupWideToAnsi(psr->pOleStr, ppsz); pwsz = psr->pOleStr; psr->pOleStr = NULL; // avoid alias CoTaskMemFree(pwsz); break; case STRRET_OFFSET: hres = DupAnsiToAnsi(STRRET_OFFPTR(pidl, psr), ppsz); break; case STRRET_CSTR: hres = DupAnsiToAnsi(psr->cStr, ppsz); break; default: *ppsz = NULL; hres = E_FAIL; } return hres; } STDAPI StrRetToBufA(STRRET *psr, LPCITEMIDLIST pidl, LPSTR pszBuf, UINT cchBuf) { HRESULT hres = E_FAIL; switch (psr->uType) { case STRRET_WSTR: { LPWSTR pszStr = psr->pOleStr; // temp copy because SHUnicodeToAnsi may overwrite buffer if (pszStr) { SHUnicodeToAnsi(pszStr, pszBuf, cchBuf); CoTaskMemFree(pszStr); // Make sure no one thinks things are allocated still psr->uType = STRRET_CSTR; psr->cStr[0] = 0; hres = S_OK; } } break; case STRRET_CSTR: SHAnsiToAnsi(psr->cStr, pszBuf, cchBuf); hres = S_OK; break; case STRRET_OFFSET: if (pidl) { SHAnsiToAnsi(STRRET_OFFPTR(pidl, psr), pszBuf, cchBuf); hres = S_OK; } break; } if (FAILED(hres) && cchBuf) *pszBuf = 0; return hres; } STDAPI StrRetToBufW(STRRET *psr, LPCITEMIDLIST pidl, LPWSTR pszBuf, UINT cchBuf) { HRESULT hres = E_FAIL; switch (psr->uType) { case STRRET_WSTR: { LPWSTR pwszTmp = psr->pOleStr; if (pwszTmp) { StrCpyNW(pszBuf, pwszTmp, cchBuf); CoTaskMemFree(pwszTmp); // Make sure no one thinks things are allocated still psr->uType = STRRET_CSTR; psr->cStr[0] = 0; hres = S_OK; } } break; case STRRET_CSTR: SHAnsiToUnicode(psr->cStr, pszBuf, cchBuf); hres = S_OK; break; case STRRET_OFFSET: if (pidl) { SHAnsiToUnicode(STRRET_OFFPTR(pidl, psr), pszBuf, cchBuf); hres = S_OK; } break; } if (FAILED(hres) && cchBuf) *pszBuf = 0; return hres; } // dupe a string using the task allocator for returing from a COM interface // STDAPI SHStrDupA(LPCSTR psz, WCHAR **ppwsz) { WCHAR *pwsz; DWORD cch; RIPMSG(psz && IS_VALID_STRING_PTRA(psz, -1), "SHStrDupA: Caller passed invalid psz"); if (psz) { cch = MultiByteToWideChar(CP_ACP, 0, psz, -1, NULL, 0); pwsz = (WCHAR *)CoTaskMemAlloc((cch + 1) * SIZEOF(WCHAR)); } else pwsz = NULL; *((PVOID UNALIGNED64 *) ppwsz) = pwsz; if (pwsz) { MultiByteToWideChar(CP_ACP, 0, psz, -1, *ppwsz, cch); return S_OK; } return E_OUTOFMEMORY; } // dupe a string using the task allocator for returing from a COM interface // Sometimes, due to structure packing, the pointer we get is not properly // aligned for Win64, so we have to do UNALIGNED64. // STDAPI SHStrDupW(LPCWSTR psz, WCHAR **ppwsz) { WCHAR *pwsz; int cb; RIPMSG(psz && IS_VALID_STRING_PTRW(psz, -1), "SHStrDupW: Caller passed invalid psz"); if (psz) { cb = (lstrlenW(psz) + 1) * SIZEOF(WCHAR); pwsz = (WCHAR *)CoTaskMemAlloc(cb); } else pwsz = NULL; *((PVOID UNALIGNED64 *) ppwsz) = pwsz; if (pwsz) { CopyMemory(pwsz, psz, cb); return S_OK; } return E_OUTOFMEMORY; } STDAPI_(int) StrCmpLogicalW(PCWSTR psz1, PCWSTR psz2) { int iRet = 0; int iCmpNum = 0; while (iRet == 0 && (*psz1 || *psz2)) { int cch1 = 0; int cch2 = 0; BOOL fIsDigit1 = IS_DIGITW(*psz1); BOOL fIsDigit2 = IS_DIGITW(*psz2); ASSERT(fIsDigit1 == TRUE || fIsDigit1 == FALSE); ASSERT(fIsDigit2 == TRUE || fIsDigit2 == FALSE); // using bit wise XOR as logical XOR // if the numbers are mismatched then n if (fIsDigit1 ^ fIsDigit2) { iRet = _StrCmpLocaleW(NORM_IGNORECASE, psz1, -1, psz2, -1); } else if (fIsDigit1 && fIsDigit2) { int cchZero1 = 0; int cchZero2 = 0; // eat leading zeros while (*psz1 == TEXT('0')) { psz1++; cchZero1++; } while (*psz2 == TEXT('0')) { psz2++; cchZero2++; } while (IS_DIGITW(psz1[cch1])) cch1++; while (IS_DIGITW(psz2[cch2])) cch2++; if (cch1 != cch2) { iRet = cch1 > cch2 ? 1 : -1; } else { // remember the first numerical difference iRet = _StrCmpLocaleW(NORM_IGNORECASE, psz1, cch1, psz2, cch2); if (iRet == 0 && iCmpNum == 0 && cchZero1 != cchZero2) { iCmpNum = cchZero2 > cchZero1 ? 1 : -1; } } } else { while (psz1[cch1] && !IS_DIGITW(psz1[cch1])) cch1++; while (psz2[cch2] && !IS_DIGITW(psz2[cch2])) cch2++; iRet = _StrCmpLocaleW(NORM_IGNORECASE, psz1, cch1, psz2, cch2); } // at this point they should be numbers or terminators or different psz1 = &psz1[cch1]; psz2 = &psz2[cch2]; } if (iRet == 0 && iCmpNum) iRet = iCmpNum; return iRet; } STDAPI_(DWORD) StrCatChainW(LPWSTR pszDst, DWORD cchDst, DWORD ichAt, LPCWSTR pszSrc) { RIPMSG(pszDst && IS_VALID_STRING_PTRW(pszDst, -1) && (DWORD)lstrlenW(pszDst) 0) { #ifdef DEBUG if (ichAt < cchDst) DEBUGWhackPathBufferW(pszDst+ichAt, cchDst-ichAt); #endif while (ichAt < cchDst) { if (!(pszDst[ichAt] = *pszSrc++)) break; ichAt++; } // check to make sure we copied a NULL if (ichAt == cchDst) pszDst[ichAt-1] = 0; } return ichAt; }