atl: Added support for category maps in AtlModuleRegisterServer.

732376d5 · Jacek Caban · Alexandre Julliard · cc18fc0f · 732376d5
Commit 732376d5 authored Jan 04, 2013 by Jacek Caban Committed by Alexandre Julliard Jan 04, 2013
Hide whitespace changes
Inline Side-by-side

Showing with 11 additions and 0 deletions

atl_main.c dlls/atl/atl_main.c +11 -0

No files found.
--- a/dlls/atl/atl_main.c
+++ b/dlls/atl/atl_main.c
@@ -220,6 +220,17 @@ HRESULT WINAPI AtlModuleRegisterServer(_ATL_MODULEW* pM, BOOL bRegTypeLib, const
            hRes = obj->pfnUpdateRegistry(TRUE); /* register */
            if (FAILED(hRes))
                return hRes;
+
+            if(pM->cbSize > ATLVer1Size) {
+                const struct _ATL_CATMAP_ENTRY *catmap;
+
+                catmap = ((const _ATL_OBJMAP_ENTRYW*)obj)->pfnGetCategoryMap();
+                if(catmap) {
+                    hRes = AtlRegisterClassCategoriesHelper(obj->pclsid, catmap, TRUE);
+                    if(FAILED(hRes))
+                        return hRes;
+                }
+            }
        }
    }