Jacek Caban : atl: Added support for category maps in AtlModuleRegisterServer.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 732376d5b9c06987dca08c0894806611395fe802
URL:    http://source.winehq.org/git/wine.git/?a=commit;h=732376d5b9c06987dca08c0894...

Author: Jacek Caban <jacek(a)codeweavers.com>
Date:   Fri Jan  4 11:34:15 2013 +0100

atl: Added support for category maps in AtlModuleRegisterServer.

---

 dlls/atl/atl_main.c |   11 +++++++++++
 1 files changed, 11 insertions(+), 0 deletions(-)

diff --git a/dlls/atl/atl_main.c b/dlls/atl/atl_main.c
index f75ecee..2a96369 100644
--- 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;
+                }
+            }
         }
     }