// -------------------------------------------------------------------------- // Module Name: UserPict.cpp // // Copyright (c) 2000, Microsoft Corporation // // Functions that implement user picture manipulation. // // History: 2000-03-24 vtan created // 2000-05-03 jeffreys reworked using DIB sections // 2000-10-26 jeffreys switched from %ALLUSERSPROFILE%\Pictures // to CSIDL_COMMON_APPDATA\User Account Pictures // -------------------------------------------------------------------------- #include "shellprv.h" #include #include #include // for SetNamedSecurityInfo #include // for ILogonUser #pragma warning(push,4) // -------------------------------------------------------------------------- // SaveDIBSectionToFile // // Arguments: hbm = Source image (DIB section) to save // hdc = Device Context containing hbm. May be NULL // if hbm is not selected in any DC or hbm // is known to have no color table. // pszFile = Target image file. // // Returns: BOOL // // Purpose: Write a DIB to disk in the proper format // // History: 2000-05-03 jeffreys created // -------------------------------------------------------------------------- #define DIB_HEADER_MARKER ((WORD) ('M' << 8) | 'B') BOOL SaveDIBSectionToFile(HBITMAP hbm, HDC hdc, LPCTSTR pszFile) { BOOL bResult; DIBSECTION ds; HANDLE hFile; BITMAPFILEHEADER bf; DWORD cbWritten; bResult = FALSE; // Get the details about the bitmap. This also validates hbm. if (GetObject(hbm, sizeof(ds), &ds) == 0) return FALSE; // Fill in a couple of optional fields if necessary if (ds.dsBmih.biSizeImage == 0) ds.dsBmih.biSizeImage = ds.dsBmih.biHeight * ds.dsBm.bmWidthBytes; if (ds.dsBmih.biBitCount <= 8 && ds.dsBmih.biClrUsed == 0) ds.dsBmih.biClrUsed = 1 << ds.dsBmih.biBitCount; // Open the target file. This also validates pszFile. hFile = CreateFile(pszFile, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); if (INVALID_HANDLE_VALUE == hFile) return FALSE; // Prepare the BITMAPFILEHEADER for writing bf.bfType = DIB_HEADER_MARKER; bf.bfReserved1 = 0; bf.bfReserved2 = 0; // The bit offset is the cumulative size of all of the header stuff bf.bfOffBits = sizeof(bf) + sizeof(ds.dsBmih) + (ds.dsBmih.biClrUsed*sizeof(RGBQUAD)); if (ds.dsBmih.biCompression == BI_BITFIELDS) bf.bfOffBits += sizeof(ds.dsBitfields); // Round up to the next 16-byte boundary. This isn't strictly necessary, // but it makes the file layout cleaner. (You can create a file mapping // and pass it to CreateDIBSection this way.) bf.bfOffBits = ((bf.bfOffBits + 15) & ~15); // The file size is the bit offset + the size of the bits bf.bfSize = bf.bfOffBits + ds.dsBmih.biSizeImage; // Write the BITMAPFILEHEADER first bResult = WriteFile(hFile, &bf, sizeof(bf), &cbWritten, NULL); if (bResult) { // Next is the BITMAPINFOHEADER bResult = WriteFile(hFile, &ds.dsBmih, sizeof(ds.dsBmih), &cbWritten, NULL); if (bResult) { // Then the 3 bitfields, if necessary if (ds.dsBmih.biCompression == BI_BITFIELDS) { bResult = WriteFile(hFile, &ds.dsBitfields, sizeof(ds.dsBitfields), &cbWritten, NULL); } if (bResult) { // Now the color table, if any if (ds.dsBmih.biClrUsed != 0) { RGBQUAD argb[256]; HDC hdcDelete; HBITMAP hbmOld; // Assume failure here bResult = FALSE; hdcDelete = NULL; if (!hdc) { hdcDelete = CreateCompatibleDC(NULL); if (hdcDelete) { hbmOld = (HBITMAP)SelectObject(hdcDelete, hbm); hdc = hdcDelete; } } if (hdc && GetDIBColorTable(hdc, 0, ARRAYSIZE(argb), argb) == ds.dsBmih.biClrUsed) { bResult = WriteFile(hFile, argb, ds.dsBmih.biClrUsed*sizeof(RGBQUAD), &cbWritten, NULL); } if (hdcDelete) { SelectObject(hdcDelete, hbmOld); DeleteDC(hdcDelete); } } // Finally, write the bits if (bResult) { SetFilePointer(hFile, bf.bfOffBits, NULL, FILE_BEGIN); bResult = WriteFile(hFile, ds.dsBm.bmBits, ds.dsBmih.biSizeImage, &cbWritten, NULL); SetEndOfFile(hFile); } } } } CloseHandle(hFile); if (!bResult) { // Something failed, clean up DeleteFile(pszFile); } return bResult; } // -------------------------------------------------------------------------- // MakeDIBSection // // Arguments: pImage = Source image // // Returns: HBITMAP // // Purpose: Create a DIB section containing the given image // on a white background // // History: 2000-05-03 jeffreys created // -------------------------------------------------------------------------- HBITMAP MakeDIBSection(IShellImageData *pImage, ULONG cx, ULONG cy) { HBITMAP hbm; HDC hdc; BITMAPINFO dib; hdc = CreateCompatibleDC(NULL); if (hdc == NULL) return NULL; dib.bmiHeader.biSize = sizeof(BITMAPINFOHEADER); dib.bmiHeader.biWidth = cx; dib.bmiHeader.biHeight = cy; dib.bmiHeader.biPlanes = 1; dib.bmiHeader.biBitCount = 24; dib.bmiHeader.biCompression = BI_RGB; dib.bmiHeader.biSizeImage = 0; dib.bmiHeader.biXPelsPerMeter = 0; dib.bmiHeader.biYPelsPerMeter = 0; dib.bmiHeader.biClrUsed = 0; dib.bmiHeader.biClrImportant = 0; hbm = CreateDIBSection(hdc, &dib, DIB_RGB_COLORS, NULL, NULL, 0); if (hbm) { HBITMAP hbmOld; RECT rc; hbmOld = (HBITMAP)SelectObject(hdc, hbm); // Initialize the entire image with white PatBlt(hdc, 0, 0, cx, cy, WHITENESS); rc.left = 0; rc.top = 0; rc.right = cx; rc.bottom = cy; // Draw the source image into the DIB section HRESULT hr = pImage->Draw(hdc, &rc, NULL); SelectObject(hdc, hbmOld); if (FAILED(hr)) { DeleteObject(hbm); hbm = NULL; SetLastError(hr); } } DeleteDC(hdc); return hbm; } // -------------------------------------------------------------------------- // ConvertAndResizeImage // // Arguments: pszFileSource = Source image file. // pszFileTarget = Target image file (resized). // // Returns: HRESULT // // Purpose: Uses GDI+ via COM interfaces to convert the given image file // to a bmp sized at 96x96. // // History: 2000-03-24 vtan created // 2000-05-03 jeffreys reworked using DIB sections // -------------------------------------------------------------------------- HRESULT ConvertAndResizeImage (LPCTSTR pszFileSource, LPCTSTR pszFileTarget) { HRESULT hr; IShellImageDataFactory *pImagingFactory; hr = CoCreateInstance(CLSID_ShellImageDataFactory, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARG(IShellImageDataFactory, &pImagingFactory)); if (SUCCEEDED(hr)) { IShellImageData *pImage; hr = pImagingFactory->CreateImageFromFile(pszFileSource, &pImage); if (SUCCEEDED(hr)) { hr = pImage->Decode(SHIMGDEC_DEFAULT, 0, 0); if (SUCCEEDED(hr)) { SIZE sizeImg; ULONG cxDest, cyDest; HDC hdc; HBITMAP hbm; DWORD dwErr; // The default dimensions are based on the screen resolution hdc = GetDC(NULL); if (hdc != NULL) { // Make it 1/2 inch square by default cxDest = GetDeviceCaps(hdc, LOGPIXELSX) / 2; cyDest = GetDeviceCaps(hdc, LOGPIXELSY) / 2; ReleaseDC(NULL, hdc); } else { // Most common display modes run at 96dpi ("small fonts") cxDest = cyDest = 48; } // Get the current image dimensions so we can maintain aspect ratio if ( SUCCEEDED(pImage->GetSize(&sizeImg)) ) { // Don't want to make small images bigger cxDest = min(cxDest, (ULONG)sizeImg.cx); cyDest = min(cyDest, (ULONG)sizeImg.cy); // If it's not square, scale the smaller dimension // to maintain the aspect ratio. if (sizeImg.cx > sizeImg.cy) { cyDest = MulDiv(cxDest, sizeImg.cy, sizeImg.cx); } else if (sizeImg.cx < sizeImg.cy) { cxDest = MulDiv(cyDest, sizeImg.cx, sizeImg.cy); } } // Resize the image // Note that this gives better results than scaling while drawing // into the DIB section (see MakeDIBSection). // // However, it doesn't always work. For example, animated images // result in E_NOTVALIDFORANIMATEDIMAGE. So ignore the return // value and the scaling will be done in MakeDIBSection if necessary. pImage->Scale(cxDest, cyDest, 0); hbm = MakeDIBSection(pImage, cxDest, cyDest); if (hbm) { // Save the DIB section to disk if (!SaveDIBSectionToFile(hbm, NULL, pszFileTarget)) { dwErr = GetLastError(); hr = HRESULT_FROM_WIN32(dwErr); } DeleteObject(hbm); } else { dwErr = GetLastError(); hr = HRESULT_FROM_WIN32(dwErr); } } pImage->Release(); } pImagingFactory->Release(); } return(hr); } // -------------------------------------------------------------------------- // SetExplicitAccessToObject // // Arguments: pszTarget = Target object // seType = Type of object // pszUser = User to grant access to // dwMask = Permissions granted // dwFlags = Inheritance flags // // Returns: BOOL // // Purpose: Grants Read/Write/Execute/Delete access to the // specified user on the specified file. // // Note that this stomps existing explicit entries in the DACL. // Multiple calls are not cumulative. // // History: 2000-05-19 jeffreys created // -------------------------------------------------------------------------- DWORD SetExplicitAccessToObject(LPTSTR pszTarget, SE_OBJECT_TYPE seType, LPCTSTR pszUser, DWORD dwMask, DWORD dwFlags) { BOOL bResult; // 84 bytes BYTE rgAclBuffer[sizeof(ACL) + (sizeof(ACCESS_ALLOWED_ACE) - sizeof(ULONG)) + (sizeof(SID) + (SID_MAX_SUB_AUTHORITIES-1)*sizeof(ULONG))]; PACL pDacl = (PACL)rgAclBuffer; if (!InitializeAcl(pDacl, sizeof(rgAclBuffer), ACL_REVISION)) return FALSE; pDacl->AceCount = 1; PACCESS_ALLOWED_ACE pAce = (PACCESS_ALLOWED_ACE)(pDacl+1); pAce->Header.AceType = ACCESS_ALLOWED_ACE_TYPE; pAce->Header.AceFlags = (UCHAR)(dwFlags & 0xFF); pAce->Mask = dwMask; SID_NAME_USE snu; TCHAR szDomainName[MAX_PATH]; DWORD cbDomainName = ARRAYSIZE(szDomainName); DWORD cbSid = sizeof(SID) + (SID_MAX_SUB_AUTHORITIES-1)*sizeof(ULONG); bResult = LookupAccountName( NULL, pszUser, (PSID)&(pAce->SidStart), &cbSid, szDomainName, &cbDomainName, &snu); if (bResult) { DWORD dwErr; // LookupAccountName doesn't return the SID length on success cbSid = GetLengthSid((PSID)&(pAce->SidStart)); // Update the ACE size pAce->Header.AceSize = (USHORT)(sizeof(ACCESS_ALLOWED_ACE) - sizeof(ULONG) + cbSid); dwErr = SetNamedSecurityInfo( pszTarget, seType, DACL_SECURITY_INFORMATION | UNPROTECTED_DACL_SECURITY_INFORMATION, NULL, NULL, pDacl, NULL); if (ERROR_SUCCESS != dwErr) { SetLastError(dwErr); bResult = FALSE; } } return bResult; } // -------------------------------------------------------------------------- // SetDefaultUserPicture // // Arguments: pszUsername = Desired user (NULL for current user). // // Returns: HRESULT // // Purpose: Picks one of the default user pictures at random and // assigns it to the specified user. // // History: 2001-03-27 reinerf created // -------------------------------------------------------------------------- HRESULT SetDefaultUserPicture(LPCTSTR pszUsername) { HRESULT hr; TCHAR szPath[MAX_PATH]; hr = SHGetUserPicturePath(NULL, SHGUPP_FLAG_DEFAULTPICSPATH, szPath); if (SUCCEEDED(hr)) { BOOL bFound = FALSE; // assume we won't find a picture // Assume everything in the dir is a vaild image file if (PathAppend(szPath, TEXT("*.*"))) { static DWORD dwSeed = 0; WIN32_FIND_DATA fd; HANDLE hFind = FindFirstFile(szPath, &fd); if (dwSeed == 0) { dwSeed = GetTickCount(); } if (hFind != INVALID_HANDLE_VALUE) { DWORD dwCount = 0; // use a probability collector algorithim (with a limit of 100 files) do { if (!PathIsDotOrDotDot(fd.cFileName)) { dwCount++; // although RtlRandom returns a ULONG it is distributed from 0...MAXLONG if (RtlRandomEx(&dwSeed) <= (MAXLONG / dwCount)) { bFound = TRUE; PathRemoveFileSpec(szPath); PathAppend(szPath, fd.cFileName); } } } while (FindNextFile(hFind, &fd) && (dwCount < 100)); FindClose(hFind); } } if (bFound) { hr = SHSetUserPicturePath(pszUsername, 0, szPath); } else { hr = E_FAIL; } } return hr; } // -------------------------------------------------------------------------- // ::SHGetUserPicturePath // // Arguments: pszUsername = Desired user (NULL for current user). // dwFlags = Flags. // pszPath = Path to user picture. // // Returns: HRESULT // // Purpose: Returns the user's picture path (absolute). Does parameter // validation as well. This function only supports .bmp files. // // Use SHGUPP_FLAG_BASEPATH to return the base to the pictures // directory. // // Use SHGUPP_FLAG_DEFAULTPICSPATH to return the path to the // default pictures directory. // // Use SHGUPP_FLAG_CREATE to create the user picture directory. // // If neither SHGUPP_FLAG_BASEPATH or SHGUPP_FLAG_DEFAULTPICSPATH // is specified, and the user has no picture, SHGUPP_FLAG_CREATE // will select one of the default pictures at random. // // History: 2000-02-22 vtan created // 2000-03-24 vtan moved from folder.cpp // -------------------------------------------------------------------------- #define UASTR_PATH_PICTURES TEXT("Microsoft\\User Account Pictures") #define UASTR_PATH_DEFPICS UASTR_PATH_PICTURES TEXT("\\Default Pictures") STDAPI SHGetUserPicturePath (LPCTSTR pszUsername, DWORD dwFlags, LPTSTR pszPath) { HRESULT hr; TCHAR szPath[MAX_PATH]; // Validate dwFlags. if ((dwFlags & SHGUPP_FLAG_INVALID_MASK) != 0) { return(E_INVALIDARG); } // Validate pszPath. This must not be NULL. if ((pszPath == NULL) || IsBadWritePtr(pszPath, MAX_PATH * sizeof(TCHAR))) { return(E_INVALIDARG); } // Start by getting the base picture path hr = SHGetFolderPathAndSubDir(NULL, (dwFlags & SHGUPP_FLAG_CREATE) ? (CSIDL_COMMON_APPDATA | CSIDL_FLAG_CREATE) : CSIDL_COMMON_APPDATA, NULL, SHGFP_TYPE_CURRENT, (dwFlags & SHGUPP_FLAG_DEFAULTPICSPATH) ? UASTR_PATH_DEFPICS : UASTR_PATH_PICTURES, szPath); // If the base path is requested this function is done. if (S_OK == hr && 0 == (dwFlags & (SHGUPP_FLAG_BASEPATH | SHGUPP_FLAG_DEFAULTPICSPATH))) { TCHAR szUsername[UNLEN + sizeof('\0')]; if (pszUsername == NULL) { DWORD dwUsernameSize; dwUsernameSize = ARRAYSIZE(szUsername); if (GetUserName(szUsername, &dwUsernameSize) != FALSE) { pszUsername = szUsername; } else { hr = HRESULT_FROM_WIN32(GetLastError()); } } if (pszUsername != NULL) { // Append the user name to the picture path. Then look for // .bmp. This function only supports bmp. PathAppend(szPath, pszUsername); lstrcatn(szPath, TEXT(".bmp"), ARRAYSIZE(szPath)); if (PathFileExistsAndAttributes(szPath, NULL) != FALSE) { hr = S_OK; } else if (dwFlags & SHGUPP_FLAG_CREATE) { // No picture has been set for this user. Select one // of the default pictures at random. hr = SetDefaultUserPicture(pszUsername); ASSERT(FAILED(hr) || PathFileExistsAndAttributes(szPath, NULL)); } else { hr = HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND); } } } if (S_OK == hr) { lstrcpyn(pszPath, szPath, MAX_PATH); } return(hr); } // -------------------------------------------------------------------------- // ::SHSetUserPicturePath // // Arguments: pszUsername = Desired user (NULL for current user). // dwFlags = Flags. // pszPath = Path to NEW user picture. // // Returns: HRESULT // // Purpose: Sets the specified user's picture as a copy of the given // image file. The image file may be any supported standard image // file (.gif / .jpg / .bmp). The file is converted to a 96x96 // .bmp file in the user picture directory. // // History: 2000-02-22 vtan created // 2000-03-24 vtan moved from folder.cpp // 2000-04-27 jeffreys restore old image on conversion failure // -------------------------------------------------------------------------- STDAPI SHSetUserPicturePath (LPCTSTR pszUsername, DWORD dwFlags, LPCTSTR pszPath) { HRESULT hr; TCHAR szPath[MAX_PATH]; TCHAR szUsername[UNLEN + sizeof('\0')]; DWORD dwUsernameSize; hr = E_FAIL; // Validate dwFlags. Currently no valid flags so this must be 0x00000000. if ((dwFlags & SHSUPP_FLAG_INVALID_MASK) != 0) { return(E_INVALIDARG); } dwUsernameSize = ARRAYSIZE(szUsername); if (GetUserName(szUsername, &dwUsernameSize) == FALSE) { return(HRESULT_FROM_WIN32(GetLastError())); } if (pszUsername != NULL) { // Privilege check. Must be an administrator to use this function when // pszUsername is not NULL (i.e. for somebody else). if ((lstrcmpi(pszUsername, szUsername) != 0) && (SHTestTokenMembership(NULL, DOMAIN_ALIAS_RID_ADMINS) == FALSE)) { static const SID c_SystemSid = {SID_REVISION,1,SECURITY_NT_AUTHORITY,{SECURITY_LOCAL_SYSTEM_RID}}; BOOL bSystem = FALSE; // One more check. Allow local system through since we may // get called from the logon screen. if (!CheckTokenMembership(NULL, (PSID)&c_SystemSid, &bSystem) || !bSystem) { return(E_ACCESSDENIED); } } } else { pszUsername = szUsername; } // Start by getting the base picture path hr = SHGetFolderPathAndSubDir(NULL, CSIDL_COMMON_APPDATA | CSIDL_FLAG_CREATE, NULL, SHGFP_TYPE_CURRENT, UASTR_PATH_PICTURES, szPath); if (S_OK == hr) { // Before attempt to delete what's there try to access the // new file. If this fails deleting what's currently installed // could leave the user without a picture. Fail the API before // anything is lost. if ((pszPath == NULL) || (PathFileExistsAndAttributes(pszPath, NULL) != FALSE)) { TCHAR szTemp[MAX_PATH]; PathAppend(szPath, pszUsername); lstrcpyn(szTemp, szPath, ARRAYSIZE(szTemp)); lstrcatn(szPath, TEXT(".bmp"), ARRAYSIZE(szPath)); lstrcatn(szTemp, TEXT(".tmp"), ARRAYSIZE(szTemp)); if ((pszPath == NULL) || lstrcmpi(pszPath, szPath) != 0) { // If present, rename .Bmp to .Tmp. // First reset the attributes so file ops work. Don't use // trace macros because failure is expected. (BOOL)SetFileAttributes(szPath, 0); (BOOL)SetFileAttributes(szTemp, 0); (BOOL)MoveFileEx(szPath, szTemp, MOVEFILE_REPLACE_EXISTING); // Convert the given image to a bmp and resize it // using the helper function which does all the goo. if (pszPath != NULL) { hr = ConvertAndResizeImage(pszPath, szPath); if (SUCCEEDED(hr)) { // Since this may be an admin setting someone else's // picture, we need to grant that person access to // modify/delete the file so they can change it // themselves later. (BOOL)SetExplicitAccessToObject(szPath, SE_FILE_OBJECT, pszUsername, GENERIC_READ | GENERIC_EXECUTE | GENERIC_WRITE | DELETE, 0); } } else { hr = S_OK; } if (SUCCEEDED(hr)) { // Delete the old picture (BOOL)DeleteFile(szTemp); } else { // Restore the old picture (BOOL)MoveFileEx(szTemp, szPath, MOVEFILE_REPLACE_EXISTING); } // Notify everyone that a user picture has changed SHChangeDWORDAsIDList dwidl; dwidl.cb = SIZEOF(dwidl) - SIZEOF(dwidl.cbZero); dwidl.dwItem1 = SHCNEE_USERINFOCHANGED; dwidl.dwItem2 = 0; dwidl.cbZero = 0; SHChangeNotify(SHCNE_EXTENDED_EVENT, SHCNF_FLUSH, (LPCITEMIDLIST)&dwidl, NULL); } else { // Source and destination are the same, nothing to do. hr = S_FALSE; } } else { hr = HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND); } } return(hr); } #pragma warning(pop)