Ros-diffs July 2023

ros-diffs@reactos.org

26 participants
189 discussions

[reactos] 01/01: [ATL] CImage: Respect CODEC info (#5408)

by Katayama Hirofumi MZ

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=50774f865c4f9d9b926f3… commit 50774f865c4f9d9b926f32a8bcfc20bbeced8fb5 Author: Katayama Hirofumi MZ <katayama.hirofumi.mz(a)gmail.com> AuthorDate: Mon Jul 31 07:22:38 2023 +0900 Commit: GitHub <noreply(a)github.com> CommitDate: Mon Jul 31 07:22:38 2023 +0900 [ATL] CImage: Respect CODEC info (#5408) Add _getAllEncoders and _getAllDecoders helper functions and use them in GetImporterFilterString, GetExporterFilterString, FileTypeFromExtension and GetClsidFromFileType functions. CORE-18867, CORE-19008 --- sdk/lib/atl/atlimage.h | 490 +++++++++++++++++++++++++++---------------------- 1 file changed, 266 insertions(+), 224 deletions(-) diff --git a/sdk/lib/atl/atlimage.h b/sdk/lib/atl/atlimage.h index 9765b0e70d5..236defea834 100644 --- a/sdk/lib/atl/atlimage.h +++ b/sdk/lib/atl/atlimage.h @@ -387,14 +387,10 @@ public: return E_FAIL; } - // TODO & FIXME: get parameters (m_rgbTransColor etc.) - // get bitmap handle HBITMAP hbm = NULL; Color color(0xFF, 0xFF, 0xFF); - Gdiplus::Status status; - status = GetCommon().CreateHBITMAPFromBitmap( - pBitmap, &hbm, color.GetValue()); + Status status = GetCommon().CreateHBITMAPFromBitmap(pBitmap, &hbm, color.GetValue()); // delete GpBitmap GetCommon().DisposeImage(pBitmap); @@ -414,14 +410,10 @@ public: return E_FAIL; } - // TODO & FIXME: get parameters (m_rgbTransColor etc.) - // get bitmap handle HBITMAP hbm = NULL; Color color(0xFF, 0xFF, 0xFF); - Gdiplus::Status status; - status = GetCommon().CreateHBITMAPFromBitmap( - pBitmap, &hbm, color.GetValue()); + Status status = GetCommon().CreateHBITMAPFromBitmap(pBitmap, &hbm, color.GetValue()); // delete Bitmap GetCommon().DisposeImage(pBitmap); @@ -525,10 +517,13 @@ public: using namespace Gdiplus; ATLASSERT(m_hbm); - // TODO & FIXME: set parameters (m_rgbTransColor etc.) - CLSID clsid; - if (!GetClsidFromFileType(&clsid, guidFileType)) - return E_FAIL; + // Get encoders + UINT cEncoders = 0; + ImageCodecInfo* pEncoders = _getAllEncoders(cEncoders); + + // Get Codec + CLSID clsid = FindCodecForFileType(*guidFileType, pEncoders, cEncoders); + delete[] pEncoders; // create a GpBitmap from HBITMAP GpBitmap *pBitmap = NULL; @@ -550,31 +545,32 @@ public: using namespace Gdiplus; ATLASSERT(m_hbm); - // TODO & FIXME: set parameters (m_rgbTransColor etc.) - // convert the file name string into Unicode CStringW pszNameW(pszFileName); + // Get encoders + UINT cEncoders = 0; + ImageCodecInfo* pEncoders = _getAllEncoders(cEncoders); + // if the file type is null, get the file type from extension - const GUID *FileType = &guidFileType; + CLSID clsid; if (::IsEqualGUID(guidFileType, GUID_NULL)) { - LPCWSTR pszExt = GetFileExtension(pszNameW); - FileType = FileTypeFromExtension(pszExt); + CString strExt(GetFileExtension(pszNameW)); + clsid = FindCodecForExtension(strExt, pEncoders, cEncoders); } - - // get CLSID from file type - CLSID clsid; - if (!GetClsidFromFileType(&clsid, FileType)) - return E_FAIL; + else + { + clsid = FindCodecForFileType(guidFileType, pEncoders, cEncoders); + } + delete[] pEncoders; // create a GpBitmap from HBITMAP GpBitmap *pBitmap = NULL; GetCommon().CreateBitmapFromHBITMAP(m_hbm, NULL, &pBitmap); // save to file - Status status; - status = GetCommon().SaveImageToFile(pBitmap, pszNameW, &clsid, NULL); + Status status = GetCommon().SaveImageToFile(pBitmap, pszNameW, &clsid, NULL); // destroy GpBitmap GetCommon().DisposeImage(pBitmap); @@ -714,73 +710,96 @@ public: excludeDefaultSave = excludeIcon | excludeEMF | excludeWMF }; - struct FILTER_DATA { - DWORD dwExclude; - const TCHAR *title; - const TCHAR *extensions; - const GUID *guid; - }; - protected: - static HRESULT GetCommonFilterString( + static bool ShouldExcludeFormat(REFGUID guidFileType, DWORD dwExclude) + { + if (::IsEqualGUID(guidFileType, Gdiplus::ImageFormatGIF)) + return !!(dwExclude & excludeGIF); + if (::IsEqualGUID(guidFileType, Gdiplus::ImageFormatBMP)) + return !!(dwExclude & excludeBMP); + if (::IsEqualGUID(guidFileType, Gdiplus::ImageFormatEMF)) + return !!(dwExclude & excludeEMF); + if (::IsEqualGUID(guidFileType, Gdiplus::ImageFormatWMF)) + return !!(dwExclude & excludeWMF); + if (::IsEqualGUID(guidFileType, Gdiplus::ImageFormatJPEG)) + return !!(dwExclude & excludeJPEG); + if (::IsEqualGUID(guidFileType, Gdiplus::ImageFormatPNG)) + return !!(dwExclude & excludePNG); + if (::IsEqualGUID(guidFileType, Gdiplus::ImageFormatTIFF)) + return !!(dwExclude & excludeTIFF); + if (::IsEqualGUID(guidFileType, Gdiplus::ImageFormatIcon)) + return !!(dwExclude & excludeIcon); + return ((dwExclude & excludeOther) == excludeOther); + } + + static HRESULT BuildCodecFilterString( + const Gdiplus::ImageCodecInfo* pCodecs, + UINT cCodecs, CSimpleString& strFilter, CSimpleArray<GUID>& aguidFileTypes, LPCTSTR pszAllFilesDescription, DWORD dwExclude, TCHAR chSeparator) { - static const FILTER_DATA table[] = + if (!pCodecs || !cCodecs) { - {excludeBMP, TEXT("BMP"), TEXT("*.BMP;*.DIB;*.RLE"), &Gdiplus::ImageFormatBMP}, - {excludeJPEG, TEXT("JPEG"), TEXT("*.JPG;*.JPEG;*.JPE;*.JFIF"), &Gdiplus::ImageFormatJPEG}, - {excludeGIF, TEXT("GIF"), TEXT("*.GIF"), &Gdiplus::ImageFormatGIF}, - {excludeEMF, TEXT("EMF"), TEXT("*.EMF"), &Gdiplus::ImageFormatEMF}, - {excludeWMF, TEXT("WMF"), TEXT("*.WMF"), &Gdiplus::ImageFormatWMF}, - {excludeTIFF, TEXT("TIFF"), TEXT("*.TIF;*.TIFF"), &Gdiplus::ImageFormatTIFF}, - {excludePNG, TEXT("PNG"), TEXT("*.PNG"), &Gdiplus::ImageFormatPNG}, - {excludeIcon, TEXT("ICO"), TEXT("*.ICO"), &Gdiplus::ImageFormatIcon} - }; + strFilter += chSeparator; + return E_FAIL; + } if (pszAllFilesDescription) { strFilter += pszAllFilesDescription; - strFilter += chSeparator; BOOL bFirst = TRUE; - for (size_t i = 0; i < _countof(table); ++i) + CString extensions; + for (UINT i = 0; i < cCodecs; ++i) { - if ((dwExclude & table[i].dwExclude) != 0) + if (ShouldExcludeFormat(pCodecs[i].FormatID, dwExclude)) continue; if (bFirst) bFirst = FALSE; else - strFilter += TEXT(';'); + extensions += TEXT(';'); - strFilter += table[i].extensions; + CString ext(pCodecs[i].FilenameExtension); + extensions += ext; } + extensions.MakeLower(); + + strFilter += TEXT(" ("); + strFilter += extensions; + strFilter += TEXT(")"); + strFilter += chSeparator; + + strFilter += extensions; strFilter += chSeparator; aguidFileTypes.Add(GUID_NULL); } - for (size_t i = 0; i < _countof(table); ++i) + for (UINT i = 0; i < cCodecs; ++i) { - if ((dwExclude & table[i].dwExclude) != 0) + if (ShouldExcludeFormat(pCodecs[i].FormatID, dwExclude)) continue; - strFilter += table[i].title; + + CString extensions(pCodecs[i].FilenameExtension); + extensions.MakeLower(); + + CString desc(pCodecs[i].FormatDescription); + strFilter += desc; strFilter += TEXT(" ("); - strFilter += table[i].extensions; + strFilter += extensions; strFilter += TEXT(")"); strFilter += chSeparator; - strFilter += table[i].extensions; + strFilter += extensions; strFilter += chSeparator; - aguidFileTypes.Add(*table[i].guid); + aguidFileTypes.Add(pCodecs[i].FormatID); } strFilter += chSeparator; - return S_OK; } @@ -792,11 +811,17 @@ public: DWORD dwExclude = excludeDefaultLoad, TCHAR chSeparator = TEXT('|')) { - return GetCommonFilterString(strImporters, - aguidFileTypes, - pszAllFilesDescription, - dwExclude, - chSeparator); + UINT cDecoders = 0; + Gdiplus::ImageCodecInfo* pDecoders = _getAllDecoders(cDecoders); + HRESULT hr = BuildCodecFilterString(pDecoders, + cDecoders, + strImporters, + aguidFileTypes, + pszAllFilesDescription, + dwExclude, + chSeparator); + delete[] pDecoders; + return hr; } static HRESULT GetExporterFilterString( @@ -806,30 +831,27 @@ public: DWORD dwExclude = excludeDefaultSave, TCHAR chSeparator = TEXT('|')) { - return GetCommonFilterString(strExporters, - aguidFileTypes, - pszAllFilesDescription, - dwExclude, - chSeparator); + UINT cEncoders = 0; + Gdiplus::ImageCodecInfo* pEncoders = _getAllEncoders(cEncoders); + HRESULT hr = BuildCodecFilterString(pEncoders, + cEncoders, + strExporters, + aguidFileTypes, + pszAllFilesDescription, + dwExclude, + chSeparator); + delete[] pEncoders; + return hr; } protected: // an extension of BITMAPINFO - struct MYBITMAPINFOEX + struct MYBITMAPINFOEX : BITMAPINFO { - BITMAPINFOHEADER bmiHeader; - RGBQUAD bmiColors[256]; - BITMAPINFO *get() - { - return reinterpret_cast<BITMAPINFO *>(this); - } - const BITMAPINFO *get() const - { - return reinterpret_cast<const BITMAPINFO *>(this); - } + RGBQUAD bmiColorsExtra[256 - 1]; }; - // abbreviations of GDI+ basic types + // abbreviations of GDI+ basic types. FIXME: Delete us typedef Gdiplus::GpStatus St; typedef Gdiplus::GpBitmap Bm; typedef Gdiplus::GpImage Im; @@ -837,69 +859,45 @@ protected: // The common data of atlimage struct COMMON { - // abbreviations of GDI+ basic types - typedef Gdiplus::ImageCodecInfo ICI; - typedef Gdiplus::EncoderParameters EncParams; - typedef Gdiplus::ARGB ARGB; - typedef HBITMAP HBM; - typedef Gdiplus::GdiplusStartupInput GSI; - typedef Gdiplus::GdiplusStartupOutput GSO; - // GDI+ function types -#undef API -#undef CST -#define API WINGDIPAPI -#define CST GDIPCONST - typedef St (WINAPI *STARTUP)(ULONG_PTR *, const GSI *, GSO *); - typedef void (WINAPI *SHUTDOWN)(ULONG_PTR); - typedef St (API *GETIMAGEENCODERSSIZE)(UINT *, UINT *); - typedef St (API *GETIMAGEENCODERS)(UINT, UINT, ICI *); - typedef St (API *CREATEBITMAPFROMFILE)(CST WCHAR*, Bm **); - typedef St (API *CREATEHBITMAPFROMBITMAP)(Bm *, HBM *, ARGB); - typedef St (API *CREATEBITMAPFROMSTREAM)(IStream *, Bm **); - typedef St (API *CREATEBITMAPFROMHBITMAP)(HBM, HPALETTE, Bm **); - typedef St (API *SAVEIMAGETOSTREAM)(Im *, IStream *, CST CLSID *, - CST EncParams *); - typedef St (API *SAVEIMAGETOFILE)(Im *, CST WCHAR *, CST CLSID *, - CST EncParams *); - typedef St (API *DISPOSEIMAGE)(Im*); -#undef API -#undef CST + using FUN_Startup = decltype(&Gdiplus::GdiplusStartup); + using FUN_Shutdown = decltype(&Gdiplus::GdiplusShutdown); + using FUN_GetImageEncoderSize = decltype(&Gdiplus::DllExports::GdipGetImageEncodersSize); + using FUN_GetImageEncoder = decltype(&Gdiplus::DllExports::GdipGetImageEncoders); + using FUN_GetImageDecoderSize = decltype(&Gdiplus::DllExports::GdipGetImageDecodersSize); + using FUN_GetImageDecoder = decltype(&Gdiplus::DllExports::GdipGetImageDecoders); + using FUN_CreateBitmapFromFile = decltype(&Gdiplus::DllExports::GdipCreateBitmapFromFile); + using FUN_CreateHBITMAPFromBitmap = decltype(&Gdiplus::DllExports::GdipCreateHBITMAPFromBitmap); + using FUN_CreateBitmapFromStream = decltype(&Gdiplus::DllExports::GdipCreateBitmapFromStream); + using FUN_CreateBitmapFromHBITMAP = decltype(&Gdiplus::DllExports::GdipCreateBitmapFromHBITMAP); + using FUN_SaveImageToStream = decltype(&Gdiplus::DllExports::GdipSaveImageToStream); + using FUN_SaveImageToFile = decltype(&Gdiplus::DllExports::GdipSaveImageToFile); + using FUN_DisposeImage = decltype(&Gdiplus::DllExports::GdipDisposeImage); // members - int count; - HINSTANCE hinstGdiPlus; - ULONG_PTR gdiplusToken; + int count = 0; + HINSTANCE hinstGdiPlus = NULL; + ULONG_PTR gdiplusToken = 0; // GDI+ functions - STARTUP Startup; - SHUTDOWN Shutdown; - GETIMAGEENCODERSSIZE GetImageEncodersSize; - GETIMAGEENCODERS GetImageEncoders; - CREATEBITMAPFROMFILE CreateBitmapFromFile; - CREATEHBITMAPFROMBITMAP CreateHBITMAPFromBitmap; - CREATEBITMAPFROMSTREAM CreateBitmapFromStream; - CREATEBITMAPFROMHBITMAP CreateBitmapFromHBITMAP; - SAVEIMAGETOSTREAM SaveImageToStream; - SAVEIMAGETOFILE SaveImageToFile; - DISPOSEIMAGE DisposeImage; + FUN_Startup Startup = NULL; + FUN_Shutdown Shutdown = NULL; + FUN_GetImageEncoderSize GetImageEncodersSize = NULL; + FUN_GetImageEncoder GetImageEncoders = NULL; + FUN_GetImageDecoderSize GetImageDecodersSize = NULL; + FUN_GetImageDecoder GetImageDecoders = NULL; + FUN_CreateBitmapFromFile CreateBitmapFromFile = NULL; + FUN_CreateHBITMAPFromBitmap CreateHBITMAPFromBitmap = NULL; + FUN_CreateBitmapFromStream CreateBitmapFromStream = NULL; + FUN_CreateBitmapFromHBITMAP CreateBitmapFromHBITMAP = NULL; + FUN_SaveImageToStream SaveImageToStream = NULL; + FUN_SaveImageToFile SaveImageToFile = NULL; + FUN_DisposeImage DisposeImage = NULL; COMMON() { - count = 0; - hinstGdiPlus = NULL; - Startup = NULL; - Shutdown = NULL; - GetImageEncodersSize = NULL; - GetImageEncoders = NULL; - CreateBitmapFromFile = NULL; - CreateHBITMAPFromBitmap = NULL; - CreateBitmapFromStream = NULL; - CreateBitmapFromHBITMAP = NULL; - SaveImageToStream = NULL; - SaveImageToFile = NULL; - DisposeImage = NULL; } + ~COMMON() { FreeLib(); @@ -915,11 +913,13 @@ protected: } // get procedure address of the DLL - template <typename TYPE> - TYPE AddrOf(const char *name) + template <typename FUN_T> + FUN_T _getFUN(FUN_T& fun, const char *name) { - FARPROC proc = ::GetProcAddress(hinstGdiPlus, name); - return reinterpret_cast<TYPE>(proc); + if (fun) + return fun; + fun = reinterpret_cast<FUN_T>(::GetProcAddress(hinstGdiPlus, name)); + return fun; } HINSTANCE LoadLib() @@ -929,24 +929,19 @@ protected: hinstGdiPlus = ::LoadLibraryA("gdiplus.dll"); - // get procedure addresses from the DLL - Startup = AddrOf<STARTUP>("GdiplusStartup"); - Shutdown = AddrOf<SHUTDOWN>("GdiplusShutdown"); - GetImageEncodersSize = - AddrOf<GETIMAGEENCODERSSIZE>("GdipGetImageEncodersSize"); - GetImageEncoders = AddrOf<GETIMAGEENCODERS>("GdipGetImageEncoders"); - CreateBitmapFromFile = - AddrOf<CREATEBITMAPFROMFILE>("GdipCreateBitmapFromFile"); - CreateHBITMAPFromBitmap = - AddrOf<CREATEHBITMAPFROMBITMAP>("GdipCreateHBITMAPFromBitmap"); - CreateBitmapFromStream = - AddrOf<CREATEBITMAPFROMSTREAM>("GdipCreateBitmapFromStream"); - CreateBitmapFromHBITMAP = - AddrOf<CREATEBITMAPFROMHBITMAP>("GdipCreateBitmapFromHBITMAP"); - SaveImageToStream = - AddrOf<SAVEIMAGETOSTREAM>("GdipSaveImageToStream"); - SaveImageToFile = AddrOf<SAVEIMAGETOFILE>("GdipSaveImageToFile"); - DisposeImage = AddrOf<DISPOSEIMAGE>("GdipDisposeImage"); + _getFUN(Startup, "GdiplusStartup"); + _getFUN(Shutdown, "GdiplusShutdown"); + _getFUN(GetImageEncodersSize, "GdipGetImageEncodersSize"); + _getFUN(GetImageEncoders, "GdipGetImageEncoders"); + _getFUN(GetImageDecodersSize, "GdipGetImageDecodersSize"); + _getFUN(GetImageDecoders, "GdipGetImageDecoders"); + _getFUN(CreateBitmapFromFile, "GdipCreateBitmapFromFile"); + _getFUN(CreateHBITMAPFromBitmap, "GdipCreateHBITMAPFromBitmap"); + _getFUN(CreateBitmapFromStream, "GdipCreateBitmapFromStream"); + _getFUN(CreateBitmapFromHBITMAP, "GdipCreateBitmapFromHBITMAP"); + _getFUN(SaveImageToStream, "GdipSaveImageToStream"); + _getFUN(SaveImageToFile, "GdipSaveImageToFile"); + _getFUN(DisposeImage, "GdipDisposeImage"); if (hinstGdiPlus && Startup) { @@ -966,6 +961,8 @@ protected: Shutdown = NULL; GetImageEncodersSize = NULL; GetImageEncoders = NULL; + GetImageDecodersSize = NULL; + GetImageDecoders = NULL; CreateBitmapFromFile = NULL; CreateHBITMAPFromBitmap = NULL; CreateBitmapFromStream = NULL; @@ -1024,96 +1021,143 @@ protected: return RGB(quad.rgbRed, quad.rgbGreen, quad.rgbBlue); } - struct EXTENSION_ENTRY + static CLSID + FindCodecForExtension(LPCTSTR dotext, const Gdiplus::ImageCodecInfo *pCodecs, UINT nCodecs) { - LPCWSTR pszExt; - GUID guid; - }; - - const GUID *FileTypeFromExtension(LPCWSTR pszExt) const - { - static const EXTENSION_ENTRY table[] = - { - {L".jpg", Gdiplus::ImageFormatJPEG}, - {L".png", Gdiplus::ImageFormatPNG}, - {L".bmp", Gdiplus::ImageFormatBMP}, - {L".gif", Gdiplus::ImageFormatGIF}, - {L".tif", Gdiplus::ImageFormatTIFF}, - {L".jpeg", Gdiplus::ImageFormatJPEG}, - {L".jpe", Gdiplus::ImageFormatJPEG}, - {L".jfif", Gdiplus::ImageFormatJPEG}, - {L".dib", Gdiplus::ImageFormatBMP}, - {L".rle", Gdiplus::ImageFormatBMP}, - {L".tiff", Gdiplus::ImageFormatTIFF} - }; - const size_t count = _countof(table); - for (size_t i = 0; i < count; ++i) + for (UINT i = 0; i < nCodecs; ++i) { - if (::lstrcmpiW(table[i].pszExt, pszExt) == 0) - return &table[i].guid; + CString strSpecs(pCodecs[i].FilenameExtension); + int ichOld = 0, ichSep; + for (;;) + { + ichSep = strSpecs.Find(TEXT(';'), ichOld); + + CString strSpec; + if (ichSep < 0) + strSpec = strSpecs.Mid(ichOld); + else + strSpec = strSpecs.Mid(ichOld, ichSep - ichOld); + + int ichDot = strSpec.ReverseFind(TEXT('.')); + if (ichDot >= 0) + strSpec = strSpec.Mid(ichDot); + + if (!dotext || strSpec.CompareNoCase(dotext) == 0) + return pCodecs[i].Clsid; + + if (ichSep < 0) + break; + + ichOld = ichSep + 1; + } } - return NULL; + return CLSID_NULL; } - struct FORMAT_ENTRY + // Deprecated. Don't use this + static const GUID *FileTypeFromExtension(LPCTSTR dotext) { - GUID guid; - LPCWSTR mime; - }; + UINT cEncoders = 0; + Gdiplus::ImageCodecInfo* pEncoders = _getAllEncoders(cEncoders); - bool GetClsidFromFileType(CLSID *clsid, const GUID *guid) const - { - static const FORMAT_ENTRY table[] = - { - {Gdiplus::ImageFormatJPEG, L"image/jpeg"}, - {Gdiplus::ImageFormatPNG, L"image/png"}, - {Gdiplus::ImageFormatBMP, L"image/bmp"}, - {Gdiplus::ImageFormatGIF, L"image/gif"}, - {Gdiplus::ImageFormatTIFF, L"image/tiff"} - }; - const size_t count = _countof(table); - for (size_t i = 0; i < count; ++i) + for (UINT i = 0; i < cEncoders; ++i) { - if (::IsEqualGUID(table[i].guid, *guid)) + CString strSpecs(pEncoders[i].FilenameExtension); + int ichOld = 0, ichSep; + for (;;) { - int num = GetEncoderClsid(table[i].mime, clsid); - if (num >= 0) + ichSep = strSpecs.Find(TEXT(';'), ichOld); + + CString strSpec; + if (ichSep < 0) + strSpec = strSpecs.Mid(ichOld); + else + strSpec = strSpecs.Mid(ichOld, ichSep - ichOld); + + int ichDot = strSpec.ReverseFind(TEXT('.')); + if (ichDot >= 0) + strSpec = strSpec.Mid(ichDot); + + if (!dotext || strSpec.CompareNoCase(dotext) == 0) { - return true; + static GUID s_guid; + s_guid = pEncoders[i].FormatID; + delete[] pEncoders; + return &s_guid; } + + if (ichSep < 0) + break; + + ichOld = ichSep + 1; } } - return false; + + delete[] pEncoders; + return NULL; + } + + static CLSID + FindCodecForFileType(REFGUID guidFileType, const Gdiplus::ImageCodecInfo *pCodecs, UINT nCodecs) + { + for (UINT iInfo = 0; iInfo < nCodecs; ++iInfo) + { + if (::IsEqualGUID(pCodecs[iInfo].FormatID, guidFileType)) + return pCodecs[iInfo].Clsid; + } + return CLSID_NULL; } - int GetEncoderClsid(LPCWSTR mime, CLSID *clsid) const + // Deprecated. Don't use this + static bool GetClsidFromFileType(CLSID *clsid, const GUID *guid) { - UINT count = 0, total_size = 0; - GetCommon().GetImageEncodersSize(&count, &total_size); + UINT cEncoders = 0; + Gdiplus::ImageCodecInfo* pEncoders = _getAllEncoders(cEncoders); + *clsid = FindCodecForFileType(*guid, pEncoders, cEncoders); + delete[] pEncoders; + return true; + } + + static Gdiplus::ImageCodecInfo* _getAllEncoders(UINT& cEncoders) + { + CImage image; // Initialize common + + UINT total_size = 0; + GetCommon().GetImageEncodersSize(&cEncoders, &total_size); if (total_size == 0) - return -1; // failure + return NULL; // failure - Gdiplus::ImageCodecInfo *pInfo; - BYTE *pb = new BYTE[total_size]; - ATLASSERT(pb); - pInfo = reinterpret_cast<Gdiplus::ImageCodecInfo *>(pb); - if (pInfo == NULL) - return -1; // failure + Gdiplus::ImageCodecInfo *ret; + ret = new Gdiplus::ImageCodecInfo[total_size / sizeof(ret[0])]; + if (ret == NULL) + { + cEncoders = 0; + return NULL; // failure + } - GetCommon().GetImageEncoders(count, total_size, pInfo); + GetCommon().GetImageEncoders(cEncoders, total_size, ret); + return ret; // needs delete[] + } + + static Gdiplus::ImageCodecInfo* _getAllDecoders(UINT& cDecoders) + { + CImage image; // Initialize common - for (UINT iInfo = 0; iInfo < count; ++iInfo) + UINT total_size = 0; + GetCommon().GetImageDecodersSize(&cDecoders, &total_size); + if (total_size == 0) + return NULL; // failure + + Gdiplus::ImageCodecInfo *ret; + ret = new Gdiplus::ImageCodecInfo[total_size / sizeof(ret[0])]; + if (ret == NULL) { - if (::lstrcmpiW(pInfo[iInfo].MimeType, mime) == 0) - { - *clsid = pInfo[iInfo].Clsid; - delete[] pb; - return iInfo; // success - } + cDecoders = 0; + return NULL; // failure } - delete[] pb; - return -1; // failure + GetCommon().GetImageDecoders(cDecoders, total_size, ret); + return ret; // needs delete[] } void AttachInternal(HBITMAP hBitmap, DIBOrientation eOrientation, @@ -1186,9 +1230,7 @@ protected: // create a DIB section HDC hDC = ::CreateCompatibleDC(NULL); ATLASSERT(hDC); - LPVOID pvBits; - HBITMAP hbm = ::CreateDIBSection(hDC, bi.get(), DIB_RGB_COLORS, - &pvBits, NULL, 0); + HBITMAP hbm = ::CreateDIBSection(hDC, &bi, DIB_RGB_COLORS, NULL, NULL, 0); ATLASSERT(hbm); ::DeleteDC(hDC);

1 year, 4 months

[reactos] 01/01: [IP] Don't reference uninitialized PCB to avoid BSOD. CORE-18982

by David L Bean

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=f8a6542b15efa996d6726… commit f8a6542b15efa996d672609e651485f50310a6c0 Author: David L Bean <d_bean(a)hotmail.com> AuthorDate: Mon Jul 24 13:18:16 2023 -0400 Commit: Thomas Faber <thomas.faber(a)reactos.org> CommitDate: Sun Jul 30 16:52:55 2023 -0400 [IP] Don't reference uninitialized PCB to avoid BSOD. CORE-18982 transport calls to LibTCPConnect that suffer certain early failures like parameter errors or early route lookup failures return without initializing the pcb. In order to avoid later BSOD's this change clears the ConnectionRequest bucket in those cases. --- drivers/network/tcpip/ip/transport/tcp/tcp.c | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/drivers/network/tcpip/ip/transport/tcp/tcp.c b/drivers/network/tcpip/ip/transport/tcp/tcp.c index 3d65e9a46b9..89b9184fdca 100644 --- a/drivers/network/tcpip/ip/transport/tcp/tcp.c +++ b/drivers/network/tcpip/ip/transport/tcp/tcp.c @@ -411,7 +411,13 @@ NTSTATUS TCPConnect Status = TCPTranslateError(LibTCPConnect(Connection, &connaddr, RemotePort)); - + if (!NT_SUCCESS(Status)) + { + LockObject(Connection); + RemoveEntryList(&Bucket->Entry); + UnlockObject(Connection); + ExFreeToNPagedLookasideList(&TdiBucketLookasideList, Bucket); + } TI_DbgPrint(DEBUG_TCP,("[IP, TCPConnect] Leaving. Status = 0x%x\n", Status)); return Status;

1 year, 4 months

[reactos] 01/01: [REACTOS] Optimize PNG images (#5492)

by Andrew Dent

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=e5993f13f0223eb1a068f… commit e5993f13f0223eb1a068ff7cf19f6449f6f574d3 Author: Andrew Dent <dent.ace(a)gmail.com> AuthorDate: Sun Jul 30 21:32:34 2023 +0100 Commit: GitHub <noreply(a)github.com> CommitDate: Sun Jul 30 22:32:34 2023 +0200 [REACTOS] Optimize PNG images (#5492) * [PIFMGR] Optimize png images - Lossless optimization with `pngslim`. Saves ~50%. - Restricted to RGBA color mode (‘c6’), for maximum compatibility at the cost of some compression. * [ROSAPPS] Optimize png image - Lossless optimization with `pngslim`. Saves ~10%. * [MSTSC] Optimize png image - Lossless optimization with `pngslim`. Saves ~56%. - Restricted to RGBA color mode (‘c6’), for maximum compatibility at the cost of some compression. * [Themes] Optimize png image - Lossless optimization with `pngslim`. Saves ~20%. - Restricted to RGB color mode (‘c2’), for maximum compatibility. --- base/applications/mstsc/res/video-display.png | Bin 33353 -> 14635 bytes .../Blackshade/blackshade.msstyles/BlackShade.png | Bin 426091 -> 340598 bytes .../net/tsclient/rdesktop/doc/keynums.png | Bin 9173 -> 8153 bytes subsystems/mvdm/pifmgr/res/01.png | Bin 953 -> 467 bytes subsystems/mvdm/pifmgr/res/01vdm.png | Bin 879 -> 449 bytes subsystems/mvdm/pifmgr/res/02.png | Bin 787 -> 365 bytes subsystems/mvdm/pifmgr/res/03.png | Bin 685 -> 315 bytes subsystems/mvdm/pifmgr/res/04.png | Bin 582 -> 272 bytes subsystems/mvdm/pifmgr/res/05.png | Bin 568 -> 290 bytes subsystems/mvdm/pifmgr/res/06.png | Bin 621 -> 310 bytes subsystems/mvdm/pifmgr/res/09.png | Bin 657 -> 288 bytes subsystems/mvdm/pifmgr/res/11.png | Bin 635 -> 292 bytes subsystems/mvdm/pifmgr/res/12.png | Bin 813 -> 383 bytes subsystems/mvdm/pifmgr/res/13.png | Bin 738 -> 374 bytes subsystems/mvdm/pifmgr/res/14.png | Bin 457 -> 246 bytes subsystems/mvdm/pifmgr/res/15.png | Bin 585 -> 293 bytes subsystems/mvdm/pifmgr/res/17.png | Bin 796 -> 392 bytes subsystems/mvdm/pifmgr/res/18.png | Bin 591 -> 308 bytes subsystems/mvdm/pifmgr/res/19.png | Bin 509 -> 228 bytes subsystems/mvdm/pifmgr/res/20.png | Bin 618 -> 339 bytes subsystems/mvdm/pifmgr/res/22.png | Bin 762 -> 367 bytes subsystems/mvdm/pifmgr/res/23.png | Bin 725 -> 337 bytes subsystems/mvdm/pifmgr/res/34.png | Bin 640 -> 302 bytes 23 files changed, 0 insertions(+), 0 deletions(-) diff --git a/base/applications/mstsc/res/video-display.png b/base/applications/mstsc/res/video-display.png old mode 100644 new mode 100755 index 93c3f3bfec1..f5632869a40 Binary files a/base/applications/mstsc/res/video-display.png and b/base/applications/mstsc/res/video-display.png differ diff --git a/media/themes/Blackshade/blackshade.msstyles/BlackShade.png b/media/themes/Blackshade/blackshade.msstyles/BlackShade.png index 64394a6b98c..f45a2b9227e 100644 Binary files a/media/themes/Blackshade/blackshade.msstyles/BlackShade.png and b/media/themes/Blackshade/blackshade.msstyles/BlackShade.png differ diff --git a/modules/rosapps/applications/net/tsclient/rdesktop/doc/keynums.png b/modules/rosapps/applications/net/tsclient/rdesktop/doc/keynums.png old mode 100644 new mode 100755 index 401601b9e06..1f18001f781 Binary files a/modules/rosapps/applications/net/tsclient/rdesktop/doc/keynums.png and b/modules/rosapps/applications/net/tsclient/rdesktop/doc/keynums.png differ diff --git a/subsystems/mvdm/pifmgr/res/01.png b/subsystems/mvdm/pifmgr/res/01.png index dbde11820b1..ab1e8426238 100644 Binary files a/subsystems/mvdm/pifmgr/res/01.png and b/subsystems/mvdm/pifmgr/res/01.png differ diff --git a/subsystems/mvdm/pifmgr/res/01vdm.png b/subsystems/mvdm/pifmgr/res/01vdm.png index 61db4f8a183..133ef2336a8 100644 Binary files a/subsystems/mvdm/pifmgr/res/01vdm.png and b/subsystems/mvdm/pifmgr/res/01vdm.png differ diff --git a/subsystems/mvdm/pifmgr/res/02.png b/subsystems/mvdm/pifmgr/res/02.png index e89a8af3c9e..50cadd0f9ce 100644 Binary files a/subsystems/mvdm/pifmgr/res/02.png and b/subsystems/mvdm/pifmgr/res/02.png differ diff --git a/subsystems/mvdm/pifmgr/res/03.png b/subsystems/mvdm/pifmgr/res/03.png index 35bb17c25f7..1e0598b9103 100644 Binary files a/subsystems/mvdm/pifmgr/res/03.png and b/subsystems/mvdm/pifmgr/res/03.png differ diff --git a/subsystems/mvdm/pifmgr/res/04.png b/subsystems/mvdm/pifmgr/res/04.png index 6392a475382..e86e3cda723 100644 Binary files a/subsystems/mvdm/pifmgr/res/04.png and b/subsystems/mvdm/pifmgr/res/04.png differ diff --git a/subsystems/mvdm/pifmgr/res/05.png b/subsystems/mvdm/pifmgr/res/05.png index fa84c7594a4..ae5a7f9ad0c 100644 Binary files a/subsystems/mvdm/pifmgr/res/05.png and b/subsystems/mvdm/pifmgr/res/05.png differ diff --git a/subsystems/mvdm/pifmgr/res/06.png b/subsystems/mvdm/pifmgr/res/06.png index f30f78c6b38..ac45947dd63 100644 Binary files a/subsystems/mvdm/pifmgr/res/06.png and b/subsystems/mvdm/pifmgr/res/06.png differ diff --git a/subsystems/mvdm/pifmgr/res/09.png b/subsystems/mvdm/pifmgr/res/09.png index 7e956ded5c0..d6a1e775a2d 100644 Binary files a/subsystems/mvdm/pifmgr/res/09.png and b/subsystems/mvdm/pifmgr/res/09.png differ diff --git a/subsystems/mvdm/pifmgr/res/11.png b/subsystems/mvdm/pifmgr/res/11.png index 017f2b4dda5..9580686d7da 100644 Binary files a/subsystems/mvdm/pifmgr/res/11.png and b/subsystems/mvdm/pifmgr/res/11.png differ diff --git a/subsystems/mvdm/pifmgr/res/12.png b/subsystems/mvdm/pifmgr/res/12.png index e393ff8fa8a..4a53da8793d 100644 Binary files a/subsystems/mvdm/pifmgr/res/12.png and b/subsystems/mvdm/pifmgr/res/12.png differ diff --git a/subsystems/mvdm/pifmgr/res/13.png b/subsystems/mvdm/pifmgr/res/13.png index 97cc0d45639..678589a5abf 100644 Binary files a/subsystems/mvdm/pifmgr/res/13.png and b/subsystems/mvdm/pifmgr/res/13.png differ diff --git a/subsystems/mvdm/pifmgr/res/14.png b/subsystems/mvdm/pifmgr/res/14.png index 1a395a1ad8f..b0673e29a38 100644 Binary files a/subsystems/mvdm/pifmgr/res/14.png and b/subsystems/mvdm/pifmgr/res/14.png differ diff --git a/subsystems/mvdm/pifmgr/res/15.png b/subsystems/mvdm/pifmgr/res/15.png index 9de4b5e5c6f..d62c147963e 100644 Binary files a/subsystems/mvdm/pifmgr/res/15.png and b/subsystems/mvdm/pifmgr/res/15.png differ diff --git a/subsystems/mvdm/pifmgr/res/17.png b/subsystems/mvdm/pifmgr/res/17.png index 17fa0a20ebb..f944fb5ef80 100644 Binary files a/subsystems/mvdm/pifmgr/res/17.png and b/subsystems/mvdm/pifmgr/res/17.png differ diff --git a/subsystems/mvdm/pifmgr/res/18.png b/subsystems/mvdm/pifmgr/res/18.png index 1f766dd2dc5..b2468728a19 100644 Binary files a/subsystems/mvdm/pifmgr/res/18.png and b/subsystems/mvdm/pifmgr/res/18.png differ diff --git a/subsystems/mvdm/pifmgr/res/19.png b/subsystems/mvdm/pifmgr/res/19.png index 9b7bdea8d3b..6642158404b 100644 Binary files a/subsystems/mvdm/pifmgr/res/19.png and b/subsystems/mvdm/pifmgr/res/19.png differ diff --git a/subsystems/mvdm/pifmgr/res/20.png b/subsystems/mvdm/pifmgr/res/20.png index 8bbd3a36415..622a09ba937 100644 Binary files a/subsystems/mvdm/pifmgr/res/20.png and b/subsystems/mvdm/pifmgr/res/20.png differ diff --git a/subsystems/mvdm/pifmgr/res/22.png b/subsystems/mvdm/pifmgr/res/22.png index 252de311206..728b896ab83 100644 Binary files a/subsystems/mvdm/pifmgr/res/22.png and b/subsystems/mvdm/pifmgr/res/22.png differ diff --git a/subsystems/mvdm/pifmgr/res/23.png b/subsystems/mvdm/pifmgr/res/23.png index 2a95e45e8b3..f1326f8eaea 100644 Binary files a/subsystems/mvdm/pifmgr/res/23.png and b/subsystems/mvdm/pifmgr/res/23.png differ diff --git a/subsystems/mvdm/pifmgr/res/34.png b/subsystems/mvdm/pifmgr/res/34.png index d11d0f7e1b7..0d7665bcf6d 100644 Binary files a/subsystems/mvdm/pifmgr/res/34.png and b/subsystems/mvdm/pifmgr/res/34.png differ

1 year, 4 months

[reactos] 01/01: [NTUSER] co_IntSendMessageWithCallBack: Formatting only, no functional changes

by Joachim Henze

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=6b7efb331c0cd20c8610c… commit 6b7efb331c0cd20c8610cd24d5e7221cff59969a Author: Joachim Henze <Joachim.Henze(a)reactos.org> AuthorDate: Sun Jul 30 21:39:10 2023 +0200 Commit: Joachim Henze <Joachim.Henze(a)reactos.org> CommitDate: Sun Jul 30 21:39:10 2023 +0200 [NTUSER] co_IntSendMessageWithCallBack: Formatting only, no functional changes e.g. wrong indentation. Want to port something back here, and the slight tweaking allows master to remain in sync with previous releases. --- win32ss/user/ntuser/message.c | 76 ++++++++++++++++++++----------------------- 1 file changed, 36 insertions(+), 40 deletions(-) diff --git a/win32ss/user/ntuser/message.c b/win32ss/user/ntuser/message.c index fbee45fec6b..78a7956d88b 100644 --- a/win32ss/user/ntuser/message.c +++ b/win32ss/user/ntuser/message.c @@ -2,7 +2,6 @@ * COPYRIGHT: See COPYING in the top level directory * PROJECT: ReactOS Win32k subsystem * PURPOSE: Messages - * FILE: win32ss/user/ntuser/message.c * PROGRAMER: Casper S. Hornstrup (chorns(a)users.sourceforge.net) */ @@ -1713,12 +1712,12 @@ co_IntSendMessageNoWait(HWND hWnd, LPARAM lParam) { ULONG_PTR Result = 0; - return co_IntSendMessageWithCallBack( hWnd, - Msg, - wParam, - lParam, - NULL, - 0, + return co_IntSendMessageWithCallBack(hWnd, + Msg, + wParam, + lParam, + NULL, + 0, &Result); } /* MSDN: @@ -1729,7 +1728,7 @@ co_IntSendMessageNoWait(HWND hWnd, process the message and the sender will free the memory before it is used. */ LRESULT FASTCALL -co_IntSendMessageWithCallBack( HWND hWnd, +co_IntSendMessageWithCallBack(HWND hWnd, UINT Msg, WPARAM wParam, LPARAM lParam, @@ -1750,7 +1749,7 @@ co_IntSendMessageWithCallBack( HWND hWnd, if (!(Window = UserGetWindowObject(hWnd))) { - TRACE("SendMessageWithCallBack: Invalid handle 0x%p!\n",hWnd); + TRACE("SendMessageWithCallBack: Invalid handle 0x%p\n",hWnd); RETURN(FALSE); } @@ -1759,29 +1758,26 @@ co_IntSendMessageWithCallBack( HWND hWnd, if (Window->state & WNDS_DESTROYED) { /* FIXME: last error? */ - ERR("Attempted to send message to window %p that is being destroyed!\n", hWnd); + ERR("Attempted to send message to window %p that is being destroyed\n", hWnd); RETURN(FALSE); } Win32Thread = PsGetCurrentThreadWin32Thread(); - if (Win32Thread == NULL || - Win32Thread->TIF_flags & TIF_INCLEANUP) - { + if (Win32Thread == NULL || Win32Thread->TIF_flags & TIF_INCLEANUP) RETURN(FALSE); - } ptiSendTo = IntSendTo(Window, Win32Thread, Msg); if (Msg & 0x80000000 && !ptiSendTo) { - if (Win32Thread->TIF_flags & TIF_INCLEANUP) RETURN( FALSE); + if (Win32Thread->TIF_flags & TIF_INCLEANUP) RETURN(FALSE); - TRACE("SMWCB: Internal Message!\n"); - Result = (ULONG_PTR)handle_internal_message( Window, Msg, wParam, lParam ); - if (uResult) *uResult = Result; - RETURN( TRUE); + TRACE("SMWCB: Internal Message\n"); + Result = (ULONG_PTR)handle_internal_message(Window, Msg, wParam, lParam); + if (uResult) *uResult = Result; + RETURN(TRUE); } /* See if this message type is present in the table */ @@ -1796,14 +1792,14 @@ co_IntSendMessageWithCallBack( HWND hWnd, if (!lParamBufferSize) lParamBufferSize = -1; } - if (! NT_SUCCESS(PackParam(&lParamPacked, Msg, wParam, lParam, !!ptiSendTo))) + if (!NT_SUCCESS(PackParam(&lParamPacked, Msg, wParam, lParam, !!ptiSendTo))) { ERR("Failed to pack message parameters\n"); - RETURN( FALSE); + RETURN(FALSE); } /* If it can be sent now, then send it. */ - if ( !ptiSendTo ) + if (!ptiSendTo) { if (Win32Thread->TIF_flags & TIF_INCLEANUP) { @@ -1812,9 +1808,9 @@ co_IntSendMessageWithCallBack( HWND hWnd, RETURN(FALSE); } - IntCallWndProc( Window, hWnd, Msg, wParam, lParam); + IntCallWndProc(Window, hWnd, Msg, wParam, lParam); - if ( Window->state & WNDS_SERVERSIDEWINDOWPROC ) + if (Window->state & WNDS_SERVERSIDEWINDOWPROC) { TRACE("SMWCB: Server Side Window Procedure\n"); switch(Window->fnid) @@ -1823,28 +1819,28 @@ co_IntSendMessageWithCallBack( HWND hWnd, DoCallBack = !DesktopWindowProc(Window, Msg, wParam, lParamPacked, (LRESULT*)&Result); break; case FNID_MESSAGEWND: - DoCallBack = !UserMessageWindowProc(Window, Msg, wParam, lParam,(LRESULT*)&Result); + DoCallBack = !UserMessageWindowProc(Window, Msg, wParam, lParam, (LRESULT*)&Result); break; case FNID_MENU: - DoCallBack = !PopupMenuWndProc( Window, Msg, wParam, lParam,(LRESULT*)&Result); + DoCallBack = !PopupMenuWndProc(Window, Msg, wParam, lParam, (LRESULT*)&Result); break; } } if (DoCallBack) - Result = (ULONG_PTR)co_IntCallWindowProc( Window->lpfnWndProc, - !Window->Unicode, - hWnd, - Msg, - wParam, - lParamPacked, - lParamBufferSize ); + Result = (ULONG_PTR)co_IntCallWindowProc(Window->lpfnWndProc, + !Window->Unicode, + hWnd, + Msg, + wParam, + lParamPacked, + lParamBufferSize); if(uResult) { *uResult = Result; } - IntCallWndProcRet( Window, hWnd, Msg, wParam, lParam, (LRESULT *)uResult); + IntCallWndProcRet(Window, hWnd, Msg, wParam, lParam, (LRESULT *)uResult); if (CompletionCallback) { @@ -1856,9 +1852,9 @@ co_IntSendMessageWithCallBack( HWND hWnd, } } - if ( !ptiSendTo) + if (!ptiSendTo) { - if (! NT_SUCCESS(UnpackParam(lParamPacked, Msg, wParam, lParam, FALSE))) + if (!NT_SUCCESS(UnpackParam(lParamPacked, Msg, wParam, lParam, FALSE))) { ERR("Failed to unpack message parameters\n"); } @@ -1867,7 +1863,7 @@ co_IntSendMessageWithCallBack( HWND hWnd, if(!(Message = AllocateUserMessage(FALSE))) { - ERR("MsqSendMessage(): Not enough memory to allocate a message\n"); + ERR("Failed to allocate message\n"); RETURN(FALSE); } @@ -1879,7 +1875,7 @@ co_IntSendMessageWithCallBack( HWND hWnd, Message->lResult = 0; Message->QS_Flags = 0; Message->ptiReceiver = ptiSendTo; - Message->ptiSender = NULL; // mjmartin, you are right! This is null. + Message->ptiSender = NULL; Message->ptiCallBackSender = Win32Thread; Message->CompletionCallback = CompletionCallback; Message->CompletionCallbackContext = CompletionCallbackContext; @@ -1889,9 +1885,9 @@ co_IntSendMessageWithCallBack( HWND hWnd, Message->flags = SMF_RECEIVERFREE; if (Msg & 0x80000000) // Higher priority event message! - InsertHeadList(&ptiSendTo->SentMessagesListHead, &Message->ListEntry); + InsertHeadList(&ptiSendTo->SentMessagesListHead, &Message->ListEntry); else - InsertTailList(&ptiSendTo->SentMessagesListHead, &Message->ListEntry); + InsertTailList(&ptiSendTo->SentMessagesListHead, &Message->ListEntry); MsqWakeQueue(ptiSendTo, QS_SENDMESSAGE, TRUE); RETURN(TRUE);

1 year, 4 months

[reactos] 01/01: [ATL] Overload const-return-versions of FindChar/FindString/FindOneOf (#5473)

by Katayama Hirofumi MZ

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=985680068c41fe90d3d19… commit 985680068c41fe90d3d19ee6777f2f982672b931 Author: Katayama Hirofumi MZ <katayama.hirofumi.mz(a)gmail.com> AuthorDate: Sun Jul 30 07:59:01 2023 +0900 Commit: GitHub <noreply(a)github.com> CommitDate: Sun Jul 30 07:59:01 2023 +0900 [ATL] Overload const-return-versions of FindChar/FindString/FindOneOf (#5473) Overload the const-return-versions of FindChar/FindCharReverse/FindString/FindOneOf. Modify CStringT::Replace. --- sdk/lib/atl/atlsimpstr.h | 4 ++++ sdk/lib/atl/cstringt.h | 50 +++++++++++++++++++++++++++++++++++++++++++++++- 2 files changed, 53 insertions(+), 1 deletion(-) diff --git a/sdk/lib/atl/atlsimpstr.h b/sdk/lib/atl/atlsimpstr.h index c06a8e46cc5..7d78f41abc4 100644 --- a/sdk/lib/atl/atlsimpstr.h +++ b/sdk/lib/atl/atlsimpstr.h @@ -358,6 +358,10 @@ public: return GetData()->nDataLength; } + PXSTR GetString() throw() + { + return m_pszData; + } PCXSTR GetString() const throw() { return m_pszData; diff --git a/sdk/lib/atl/cstringt.h b/sdk/lib/atl/cstringt.h index 748cbcba993..2e5a18b1a54 100644 --- a/sdk/lib/atl/cstringt.h +++ b/sdk/lib/atl/cstringt.h @@ -99,6 +99,12 @@ public: } static LPWSTR __cdecl FindString( + _In_z_ LPWSTR pszSource, + _In_z_ LPCWSTR pszSub) + { + return ::wcsstr(pszSource, pszSub); + } + static LPCWSTR __cdecl FindString( _In_z_ LPCWSTR pszSource, _In_z_ LPCWSTR pszSub) { @@ -106,6 +112,12 @@ public: } static LPWSTR __cdecl FindChar( + _In_z_ LPWSTR pszSource, + _In_ WCHAR ch) + { + return ::wcschr(pszSource, ch); + } + static LPCWSTR __cdecl FindChar( _In_z_ LPCWSTR pszSource, _In_ WCHAR ch) { @@ -113,6 +125,12 @@ public: } static LPWSTR __cdecl FindCharReverse( + _In_z_ LPWSTR pszSource, + _In_ WCHAR ch) + { + return ::wcsrchr(pszSource, ch); + } + static LPCWSTR __cdecl FindCharReverse( _In_z_ LPCWSTR pszSource, _In_ WCHAR ch) { @@ -120,6 +138,12 @@ public: } static LPWSTR __cdecl FindOneOf( + _In_z_ LPWSTR pszSource, + _In_z_ LPCWSTR pszCharSet) + { + return ::wcspbrk(pszSource, pszCharSet); + } + static LPCWSTR __cdecl FindOneOf( _In_z_ LPCWSTR pszSource, _In_z_ LPCWSTR pszCharSet) { @@ -262,6 +286,12 @@ public: } static LPSTR __cdecl FindString( + _In_z_ LPSTR pszSource, + _In_z_ LPCSTR pszSub) + { + return ::strstr(pszSource, pszSub); + } + static LPCSTR __cdecl FindString( _In_z_ LPCSTR pszSource, _In_z_ LPCSTR pszSub) { @@ -269,6 +299,12 @@ public: } static LPSTR __cdecl FindChar( + _In_z_ LPSTR pszSource, + _In_ CHAR ch) + { + return ::strchr(pszSource, ch); + } + static LPCSTR __cdecl FindChar( _In_z_ LPCSTR pszSource, _In_ CHAR ch) { @@ -276,6 +312,12 @@ public: } static LPSTR __cdecl FindCharReverse( + _In_z_ LPSTR pszSource, + _In_ CHAR ch) + { + return ::strrchr(pszSource, ch); + } + static LPCSTR __cdecl FindCharReverse( _In_z_ LPCSTR pszSource, _In_ CHAR ch) { @@ -283,6 +325,12 @@ public: } static LPSTR __cdecl FindOneOf( + _In_z_ LPSTR pszSource, + _In_z_ LPCSTR pszCharSet) + { + return ::strpbrk(pszSource, pszCharSet); + } + static LPCSTR __cdecl FindOneOf( _In_z_ LPCSTR pszSource, _In_z_ LPCSTR pszCharSet) { @@ -876,7 +924,7 @@ public: int Replace(XCHAR chOld, XCHAR chNew) { - PCXSTR pszString = CThisSimpleString::GetString(); + PXSTR pszString = CThisSimpleString::GetString(); PXSTR pszFirst = StringTraits::FindChar(pszString, chOld); if (!pszFirst) return 0;

1 year, 4 months

[reactos] 01/01: [IP] Unlock Connection when TCPAllocatePort() fails. CORE-18371

by Serge Gautherie

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=2010a5b8d915beebd64c6… commit 2010a5b8d915beebd64c68cb568b03c29ab64d75 Author: Serge Gautherie <reactos-git_serge_171003(a)gautherie.fr> AuthorDate: Mon Jul 24 22:38:27 2023 +0200 Commit: Thomas Faber <thomas.faber(a)reactos.org> CommitDate: Sat Jul 29 11:23:13 2023 -0400 [IP] Unlock Connection when TCPAllocatePort() fails. CORE-18371 Addendum to 874d317 (0.4.15-dev-5707, #5074). --- drivers/network/tcpip/ip/transport/tcp/accept.c | 1 + drivers/network/tcpip/ip/transport/tcp/tcp.c | 1 + 2 files changed, 2 insertions(+) diff --git a/drivers/network/tcpip/ip/transport/tcp/accept.c b/drivers/network/tcpip/ip/transport/tcp/accept.c index 4d12c527fae..cfab933003d 100644 --- a/drivers/network/tcpip/ip/transport/tcp/accept.c +++ b/drivers/network/tcpip/ip/transport/tcp/accept.c @@ -82,6 +82,7 @@ NTSTATUS TCPListen(PCONNECTION_ENDPOINT Connection, UINT Backlog) if (AllocatedPort == (UINT) -1) { DbgPrint("ERR: No more ports available.\n"); + UnlockObject(Connection); return STATUS_TOO_MANY_ADDRESSES; } Connection->AddressFile->Port = AllocatedPort; diff --git a/drivers/network/tcpip/ip/transport/tcp/tcp.c b/drivers/network/tcpip/ip/transport/tcp/tcp.c index cd28ffaf71a..3d65e9a46b9 100644 --- a/drivers/network/tcpip/ip/transport/tcp/tcp.c +++ b/drivers/network/tcpip/ip/transport/tcp/tcp.c @@ -386,6 +386,7 @@ NTSTATUS TCPConnect if (AllocatedPort == (UINT) -1) { DbgPrint("ERR: No more ports available.\n"); + UnlockObject(Connection); return STATUS_TOO_MANY_ADDRESSES; } Connection->AddressFile->Port = AllocatedPort;

1 year, 4 months

[reactos] 14/14: [NTOS:Mm] Fix some 64 bit issues

by Timo Kreuzer

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=539c31658985005bf22a9… commit 539c31658985005bf22a9fb03b967d7bbae1417d Author: Timo Kreuzer <timo.kreuzer(a)reactos.org> AuthorDate: Fri Apr 14 10:34:21 2023 +0300 Commit: Timo Kreuzer <timo.kreuzer(a)reactos.org> CommitDate: Sat Jul 29 14:00:44 2023 +0300 [NTOS:Mm] Fix some 64 bit issues --- ntoskrnl/mm/i386/page.c | 2 +- ntoskrnl/mm/pagefile.c | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/ntoskrnl/mm/i386/page.c b/ntoskrnl/mm/i386/page.c index 38cd5066256..e0f7a304c8b 100644 --- a/ntoskrnl/mm/i386/page.c +++ b/ntoskrnl/mm/i386/page.c @@ -598,7 +598,7 @@ MmCreatePageFileMapping(PEPROCESS Process, /* And we don't support creating for other process */ ASSERT(Process == PsGetCurrentProcess()); - if (SwapEntry & (1 << 31)) + if (SwapEntry & (1 << (sizeof(SWAPENTRY) - 1))) { KeBugCheck(MEMORY_MANAGEMENT); } diff --git a/ntoskrnl/mm/pagefile.c b/ntoskrnl/mm/pagefile.c index a57c4de0304..ae15771cad2 100644 --- a/ntoskrnl/mm/pagefile.c +++ b/ntoskrnl/mm/pagefile.c @@ -152,7 +152,7 @@ MmWriteToSwapPage(SWAPENTRY SwapEntry, PFN_NUMBER Page) IO_STATUS_BLOCK Iosb; NTSTATUS Status; KEVENT Event; - UCHAR MdlBase[sizeof(MDL) + sizeof(ULONG)]; + UCHAR MdlBase[sizeof(MDL) + sizeof(PFN_NUMBER)]; PMDL Mdl = (PMDL)MdlBase; DPRINT("MmWriteToSwapPage\n"); @@ -217,7 +217,7 @@ MiReadPageFile( IO_STATUS_BLOCK Iosb; NTSTATUS Status; KEVENT Event; - UCHAR MdlBase[sizeof(MDL) + sizeof(ULONG)]; + UCHAR MdlBase[sizeof(MDL) + sizeof(PFN_NUMBER)]; PMDL Mdl = (PMDL)MdlBase; PMMPAGING_FILE PagingFile;

1 year, 4 months

[reactos] 13/14: NTOS:CC Free some VACBs when we're under memory pressure

by Jérôme Gardou

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=69daff72d2607e514a4c8… commit 69daff72d2607e514a4c8590c26c009da7f9d5b1 Author: Jérôme Gardou <jerome.gardou(a)reactos.org> AuthorDate: Sat Oct 1 22:48:46 2022 +0200 Commit: Timo Kreuzer <timo.kreuzer(a)reactos.org> CommitDate: Sat Jul 29 14:00:44 2023 +0300 NTOS:CC Free some VACBs when we're under memory pressure --- ntoskrnl/cc/lazywrite.c | 10 +++++ ntoskrnl/cc/view.c | 88 ++++++++++++++---------------------------- ntoskrnl/include/internal/cc.h | 4 ++ 3 files changed, 44 insertions(+), 58 deletions(-) diff --git a/ntoskrnl/cc/lazywrite.c b/ntoskrnl/cc/lazywrite.c index 607a469d20d..87216bf3bef 100644 --- a/ntoskrnl/cc/lazywrite.c +++ b/ntoskrnl/cc/lazywrite.c @@ -130,6 +130,16 @@ CcWriteBehind(VOID) ++CcLazyWriteIos; DPRINT("Lazy writer done (%d)\n", Count); } + + /* Make sure we're not throttling writes after this */ + while (MmAvailablePages < MmThrottleTop) + { + /* Break if we can't even find one to free */ + if (!CcRosFreeOneUnusedVacb()) + { + break; + } + } } VOID diff --git a/ntoskrnl/cc/view.c b/ntoskrnl/cc/view.c index a40272f8f34..25d066abd03 100644 --- a/ntoskrnl/cc/view.c +++ b/ntoskrnl/cc/view.c @@ -595,34 +595,26 @@ CcRosUnmarkDirtyVacb ( } } -static BOOLEAN -CcRosFreeUnusedVacb ( - PULONG Count) +CcRosFreeOneUnusedVacb( + VOID) { - ULONG cFreed; - BOOLEAN Freed; KIRQL oldIrql; - PROS_VACB current; - LIST_ENTRY FreeList; PLIST_ENTRY current_entry; - - cFreed = 0; - Freed = FALSE; - InitializeListHead(&FreeList); + PROS_VACB to_free = NULL; oldIrql = KeAcquireQueuedSpinLock(LockQueueMasterLock); /* Browse all the available VACB */ current_entry = VacbLruListHead.Flink; - while (current_entry != &VacbLruListHead) + while ((current_entry != &VacbLruListHead) && (to_free == NULL)) { ULONG Refs; + PROS_VACB current; current = CONTAINING_RECORD(current_entry, ROS_VACB, VacbLruListEntry); - current_entry = current_entry->Flink; KeAcquireSpinLockAtDpcLevel(&current->SharedCacheMap->CacheMapLock); @@ -634,47 +626,32 @@ CcRosFreeUnusedVacb ( ASSERT(!current->MappedCount); ASSERT(Refs == 1); - /* Reset and move to free list */ + /* Reset it, this is the one we want to free */ RemoveEntryList(&current->CacheMapVacbListEntry); + InitializeListHead(&current->CacheMapVacbListEntry); RemoveEntryList(&current->VacbLruListEntry); InitializeListHead(&current->VacbLruListEntry); - InsertHeadList(&FreeList, &current->CacheMapVacbListEntry); + + to_free = current; } KeReleaseSpinLockFromDpcLevel(&current->SharedCacheMap->CacheMapLock); + current_entry = current_entry->Flink; } KeReleaseQueuedSpinLock(LockQueueMasterLock, oldIrql); - /* And now, free any of the found VACB, that'll free memory! */ - while (!IsListEmpty(&FreeList)) + /* And now, free the VACB that we found, if any. */ + if (to_free == NULL) { - ULONG Refs; - - current_entry = RemoveHeadList(&FreeList); - current = CONTAINING_RECORD(current_entry, - ROS_VACB, - CacheMapVacbListEntry); - InitializeListHead(&current->CacheMapVacbListEntry); - Refs = CcRosVacbDecRefCount(current); - ASSERT(Refs == 0); - ++cFreed; - } - - /* If we freed at least one VACB, return success */ - if (cFreed != 0) - { - Freed = TRUE; + return FALSE; } - /* If caller asked for free count, return it */ - if (Count != NULL) - { - *Count = cFreed; - } + /* This must be its last ref */ + NT_VERIFY(CcRosVacbDecRefCount(to_free) == 0); - return Freed; + return TRUE; } static @@ -690,7 +667,6 @@ CcRosCreateVacb ( NTSTATUS Status; KIRQL oldIrql; ULONG Refs; - BOOLEAN Retried; SIZE_T ViewSize = VACB_MAPPING_GRANULARITY; ASSERT(SharedCacheMap); @@ -711,28 +687,24 @@ CcRosCreateVacb ( CcRosVacbIncRefCount(current); - Retried = FALSE; -Retry: - /* Map VACB in system space */ - Status = MmMapViewInSystemSpaceEx(SharedCacheMap->Section, &current->BaseAddress, &ViewSize, &current->FileOffset, 0); - - if (!NT_SUCCESS(Status)) + while (TRUE) { - ULONG Freed; - /* If no space left, try to prune unused VACB - * to recover space to map our VACB - * If it succeed, retry to map, otherwise - * just fail. - */ - if (!Retried && CcRosFreeUnusedVacb(&Freed)) + /* Map VACB in system space */ + Status = MmMapViewInSystemSpaceEx(SharedCacheMap->Section, &current->BaseAddress, &ViewSize, &current->FileOffset, 0); + if (NT_SUCCESS(Status)) { - DPRINT("Prunned %d VACB, trying again\n", Freed); - Retried = TRUE; - goto Retry; + break; } - ExFreeToNPagedLookasideList(&VacbLookasideList, current); - return Status; + /* + * If no space left, try to prune one unused VACB to recover space to map our VACB. + * If it succeeds, retry to map, otherwise just fail. + */ + if (!CcRosFreeOneUnusedVacb()) + { + ExFreeToNPagedLookasideList(&VacbLookasideList, current); + return Status; + } } #if DBG diff --git a/ntoskrnl/include/internal/cc.h b/ntoskrnl/include/internal/cc.h index a56118a8558..90c315901d1 100644 --- a/ntoskrnl/include/internal/cc.h +++ b/ntoskrnl/include/internal/cc.h @@ -509,3 +509,7 @@ CcRosVacbDecRefCount( } #define CcRosVacbGetRefCount(vacb) InterlockedCompareExchange((PLONG)&(vacb)->ReferenceCount, 0, 0) #endif + +BOOLEAN +CcRosFreeOneUnusedVacb( + VOID);

1 year, 4 months

[reactos] 12/14: Revert "[NTOS:CC] Wake up lazy scan after inserting elements in the list."

by Jérôme Gardou

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=470ad188258a3561ef595… commit 470ad188258a3561ef595fc281f1643eda24e880 Author: Jérôme Gardou <jerome.gardou(a)reactos.org> AuthorDate: Fri Sep 23 23:00:18 2022 +0200 Commit: Timo Kreuzer <timo.kreuzer(a)reactos.org> CommitDate: Sat Jul 29 14:00:44 2023 +0300 Revert "[NTOS:CC] Wake up lazy scan after inserting elements in the list." This reverts commit 6e4f83da700fbfac5450dc9ef667585ddc870bc9. --- ntoskrnl/cc/copy.c | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/ntoskrnl/cc/copy.c b/ntoskrnl/cc/copy.c index 7b43ab51e60..e01f2b46865 100644 --- a/ntoskrnl/cc/copy.c +++ b/ntoskrnl/cc/copy.c @@ -400,6 +400,16 @@ CcCanIWrite ( return FALSE; } + /* Otherwise, if there are no deferred writes yet, start the lazy writer */ + if (IsListEmpty(&CcDeferredWrites)) + { + KIRQL OldIrql; + + OldIrql = KeAcquireQueuedSpinLock(LockQueueMasterLock); + CcScheduleLazyWriteScan(TRUE); + KeReleaseQueuedSpinLock(LockQueueMasterLock, OldIrql); + } + /* Initialize our wait event */ KeInitializeEvent(&WaitEvent, NotificationEvent, FALSE); @@ -427,12 +437,6 @@ CcCanIWrite ( &CcDeferredWriteSpinLock); } - /* Now make sure that the lazy scan writer will be active */ - OldIrql = KeAcquireQueuedSpinLock(LockQueueMasterLock); - if (!LazyWriter.ScanActive) - CcScheduleLazyWriteScan(TRUE); - KeReleaseQueuedSpinLock(LockQueueMasterLock, OldIrql); - #if DBG DPRINT1("Actively deferring write for: %p\n", FileObject); DPRINT1("Because:\n");

1 year, 4 months

[reactos] 11/14: [NTOS:Mm] Use MmRebalanceMemoryConsumersAndWait in the page fault handler

by Timo Kreuzer

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=876769fdd50c911638bd9… commit 876769fdd50c911638bd91329ceed56ee6eaa1ce Author: Timo Kreuzer <timo.kreuzer(a)reactos.org> AuthorDate: Wed Apr 5 23:02:43 2023 +0300 Commit: Timo Kreuzer <timo.kreuzer(a)reactos.org> CommitDate: Sat Jul 29 14:00:44 2023 +0300 [NTOS:Mm] Use MmRebalanceMemoryConsumersAndWait in the page fault handler --- ntoskrnl/mm/ARM3/pagfault.c | 19 ++++++++++++++++++- ntoskrnl/mm/mmfault.c | 18 ++++++++++++++++-- 2 files changed, 34 insertions(+), 3 deletions(-) diff --git a/ntoskrnl/mm/ARM3/pagfault.c b/ntoskrnl/mm/ARM3/pagfault.c index a312f4a8bdd..b9a855662d9 100644 --- a/ntoskrnl/mm/ARM3/pagfault.c +++ b/ntoskrnl/mm/ARM3/pagfault.c @@ -15,6 +15,10 @@ #define MODULE_INVOLVED_IN_ARM3 #include <mm/ARM3/miarm.h> +VOID +NTAPI +MmRebalanceMemoryConsumersAndWait(VOID); + /* GLOBALS ********************************************************************/ #define HYDRA_PROCESS (PEPROCESS)1 @@ -1940,7 +1944,7 @@ _WARN("Session space stuff is not implemented yet!") return STATUS_IN_PAGE_ERROR | 0x10000000; } } - +RetryKernel: /* Acquire the working set lock */ KeRaiseIrql(APC_LEVEL, &LockIrql); MiLockWorkingSet(CurrentThread, WorkingSet); @@ -2107,6 +2111,12 @@ _WARN("Session space stuff is not implemented yet!") MiUnlockWorkingSet(CurrentThread, WorkingSet); KeLowerIrql(LockIrql); + if (Status == STATUS_NO_MEMORY) + { + MmRebalanceMemoryConsumersAndWait(); + goto RetryKernel; + } + /* We are done! */ DPRINT("Fault resolved with status: %lx\n", Status); return Status; @@ -2616,6 +2626,13 @@ ExitUser: /* Return the status */ ASSERT(KeGetCurrentIrql() <= APC_LEVEL); MiUnlockProcessWorkingSet(CurrentProcess, CurrentThread); + + if (Status == STATUS_NO_MEMORY) + { + MmRebalanceMemoryConsumersAndWait(); + goto UserFault; + } + return Status; } diff --git a/ntoskrnl/mm/mmfault.c b/ntoskrnl/mm/mmfault.c index 82633916a16..7937f7b5e5b 100644 --- a/ntoskrnl/mm/mmfault.c +++ b/ntoskrnl/mm/mmfault.c @@ -200,6 +200,10 @@ MmNotPresentFault(KPROCESSOR_MODE Mode, extern BOOLEAN Mmi386MakeKernelPageTableGlobal(PVOID Address); +VOID +NTAPI +MmRebalanceMemoryConsumersAndWait(VOID); + NTSTATUS NTAPI MmAccessFault(IN ULONG FaultCode, @@ -208,6 +212,7 @@ MmAccessFault(IN ULONG FaultCode, IN PVOID TrapInformation) { PMEMORY_AREA MemoryArea = NULL; + NTSTATUS Status; /* Cute little hack for ROS */ if ((ULONG_PTR)Address >= (ULONG_PTR)MmSystemRangeStart) @@ -252,16 +257,25 @@ MmAccessFault(IN ULONG FaultCode, return MmArmAccessFault(FaultCode, Address, Mode, TrapInformation); } +Retry: /* Keep same old ReactOS Behaviour */ if (!MI_IS_NOT_PRESENT_FAULT(FaultCode)) { /* Call access fault */ - return MmpAccessFault(Mode, (ULONG_PTR)Address, TrapInformation ? FALSE : TRUE); + Status = MmpAccessFault(Mode, (ULONG_PTR)Address, TrapInformation ? FALSE : TRUE); } else { /* Call not present */ - return MmNotPresentFault(Mode, (ULONG_PTR)Address, TrapInformation ? FALSE : TRUE); + Status = MmNotPresentFault(Mode, (ULONG_PTR)Address, TrapInformation ? FALSE : TRUE); } + + if (Status == STATUS_NO_MEMORY) + { + MmRebalanceMemoryConsumersAndWait(); + goto Retry; + } + + return Status; }

1 year, 4 months

Jump to page:

2024

2023

2022

2021

2020

2019

2018

2017

2016

2015

2014

2013

2012

2011

2010

2009

2008

2007

2006

2005

2004

Ros-diffs July 2023