1 files changed, 78 insertions, 3 deletions

diff --git a/ext/win32ole/win32ole.c b/ext/win32ole/win32ole.c
index dd7d7b13a9..2de548a4cb 100644
--- a/ext/win32ole/win32ole.c
+++ b/ext/win32ole/win32ole.c
@@ -118,7 +118,7 @@
 
 #define WC2VSTR(x) ole_wc2vstr((x), TRUE)
 
-#define WIN32OLE_VERSION "1.2.1"
+#define WIN32OLE_VERSION "1.2.2"
 
 typedef HRESULT (STDAPICALLTYPE FNCOCREATEINSTANCEEX)
     (REFCLSID, IUnknown*, DWORD, COSERVERINFO*, DWORD, MULTI_QI*);
@@ -7606,10 +7606,18 @@ find_default_source(VALUE ole, IID *piid, ITypeInfo **ppTypeInfo)
 
     IDispatch *pDispatch;
     ITypeInfo *pTypeInfo;
+    ITypeInfo *pTypeInfo2;
+    ITypeInfo *pRefTypeInfo;
+    ITypeLib  *pTypeLib;
     TYPEATTR *pTypeAttr;
-    int i;
+    TYPEATTR *pTypeAttr2;
+    TYPEATTR *pRefTypeAttr;
+    int i, j;
     int iFlags;
-    HREFTYPE hRefType;
+    int flags;
+    HREFTYPE hRefType, href;
+    int count;
+    BOOL found = FALSE;
 
     struct oledata *pole;
 
@@ -7649,6 +7657,73 @@ find_default_source(VALUE ole, IID *piid, ITypeInfo **ppTypeInfo)
         OLE_RELEASE(pTypeInfo);
         return hr;
     }
+
+    /*
+     *  if the ole is not COCLASS, then try to find COCLASS which
+     *  has ole as [default] interface.
+     */
+    if (pTypeAttr->typekind != TKIND_COCLASS) {
+        /* 
+         * search coclass
+         */
+        hr = pTypeInfo->lpVtbl->GetContainingTypeLib(pTypeInfo, &pTypeLib, NULL);
+        if (FAILED(hr)) {
+            OLE_RELEASE_TYPEATTR(pTypeInfo, pTypeAttr);
+            OLE_RELEASE(pTypeInfo);
+            return hr;
+        }
+        count = pTypeLib->lpVtbl->GetTypeInfoCount(pTypeLib);
+
+        for (i = 0; i < count && !found; i++) {
+             hr = pTypeLib->lpVtbl->GetTypeInfo(pTypeLib, i, &pTypeInfo2);
+             if (FAILED(hr))
+                 continue;
+             hr = OLE_GET_TYPEATTR(pTypeInfo2, &pTypeAttr2);
+             if (FAILED(hr)) {
+                 OLE_RELEASE(pTypeInfo2);
+                 continue;
+             }
+             if (pTypeAttr2->typekind != TKIND_COCLASS) {
+                 OLE_RELEASE_TYPEATTR(pTypeInfo2, pTypeAttr2);
+                 OLE_RELEASE(pTypeInfo2);
+                 continue;
+             }
+             for (j = 0; j < pTypeAttr2->cImplTypes && !found; j++) {
+                 hr = pTypeInfo2->lpVtbl->GetImplTypeFlags(pTypeInfo2, j, &flags);
+                 if (FAILED(hr))
+                     continue;
+                 if (!(flags & IMPLTYPEFLAG_FDEFAULT)) 
+                     continue;
+                 hr = pTypeInfo2->lpVtbl->GetRefTypeOfImplType(pTypeInfo2, j, &href);
+                 if (FAILED(hr))
+                     continue;
+                 hr = pTypeInfo2->lpVtbl->GetRefTypeInfo(pTypeInfo2, href, &pRefTypeInfo);
+                 if (FAILED(hr)) 
+                     continue;
+                 hr = OLE_GET_TYPEATTR(pRefTypeInfo, &pRefTypeAttr);
+                 if (FAILED(hr))  {
+                     OLE_RELEASE(pRefTypeInfo);
+                     continue;
+                 }
+                 if (IsEqualGUID(&(pTypeAttr->guid), &(pRefTypeAttr->guid))) {
+                     found = TRUE;
+                 }
+                 OLE_RELEASE_TYPEATTR(pRefTypeInfo, pRefTypeAttr);
+                 OLE_RELEASE(pRefTypeInfo);
+             }
+             if (!found) {
+                 OLE_RELEASE_TYPEATTR(pTypeInfo2, pTypeAttr2);
+                 OLE_RELEASE(pTypeInfo2);
+             }
+        }
+        if (found) {
+            OLE_RELEASE_TYPEATTR(pTypeInfo, pTypeAttr);
+            OLE_RELEASE(pTypeInfo);
+            pTypeInfo = pTypeInfo2;
+            pTypeAttr = pTypeAttr2;
+        }
+    }
+
     /* Enumerate all implemented types of the COCLASS */
     for (i = 0; i < pTypeAttr->cImplTypes; i++) {
         hr = pTypeInfo->lpVtbl->GetImplTypeFlags(pTypeInfo, i, &iFlags);