From 7634900b4e12f75bf34ef2c1dea878812df7734a Mon Sep 17 00:00:00 2001
From: suke <suke@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Sat, 10 Feb 2007 11:29:49 +0000
Subject: [PATCH] * ext/win32ole/win32ole.c: add WIN32OLE#ole_query_interface. 
  thanks to Mikael Pahmp.

* test/win32ole/test_win32ole.rb: ditto.


git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@11690 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 ChangeLog                      |  7 +++++
 ext/win32ole/win32ole.c        | 48 +++++++++++++++++++++++++++++++++-
 test/win32ole/test_win32ole.rb |  9 +++++++
 3 files changed, 63 insertions(+), 1 deletion(-)

diff --git a/ChangeLog b/ChangeLog
index 749244cc92..581f32fc3a 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -1,3 +1,10 @@
+Sat Feb 10 20:21:29 2007  Masaki Suketa  <masaki.suketa@nifty.ne.jp>
+
+	* ext/win32ole/win32ole.c: add WIN32OLE#ole_query_interface.
+	  thanks to Mikael Pahmp.
+
+	* test/win32ole/test_win32ole.rb: ditto.
+
 Sat Feb 10 17:46:52 2007  Nobuyoshi Nakada  <nobu@ruby-lang.org>
 
 	* common.mk: targets which depend on yarvcore.h now depend on
diff --git a/ext/win32ole/win32ole.c b/ext/win32ole/win32ole.c
index 7d725e4737..c613a85dae 100644
--- a/ext/win32ole/win32ole.c
+++ b/ext/win32ole/win32ole.c
@@ -80,7 +80,7 @@
 
 #define WC2VSTR(x) ole_wc2vstr((x), TRUE)
 
-#define WIN32OLE_VERSION "0.9.2"
+#define WIN32OLE_VERSION "0.9.3"
 
 typedef HRESULT (STDAPICALLTYPE FNCOCREATEINSTANCEEX)
     (REFCLSID, IUnknown*, DWORD, COSERVERINFO*, DWORD, MULTI_QI*);
@@ -320,6 +320,7 @@ static VALUE make_oletypelib_obj(VALUE guid, VALUE major_version, VALUE minor_ve
 static VALUE ole_typelib_from_itypelib(ITypeLib *pTypeLib);
 static VALUE ole_typelib_from_itypeinfo(ITypeInfo *pTypeInfo);
 static VALUE fole_typelib(VALUE self);
+static VALUE fole_query_interface(VALUE self, VALUE str_iid);
 static HRESULT ole_docinfo_from_type(ITypeInfo *pTypeInfo, BSTR *name, BSTR *helpstr, DWORD *helpcontext, BSTR *helpfile);
 static VALUE ole_usertype2val(ITypeInfo *pTypeInfo, TYPEDESC *pTypeDesc, VALUE typedetails);
 static VALUE ole_typedesc2val();
@@ -3823,6 +3824,50 @@ fole_typelib(VALUE self)
     return vtlib;
 }
 
+/*
+ *  call-seq:
+ *     WIN32OLE#ole_query_interface(iid) -> WIN32OLE object
+ * 
+ *  Returns WIN32OLE object for a specific dispatch or dual
+ *  interface specified by iid.
+ *
+ *      ie = WIN32OLE.new('InternetExplorer.Application')
+ *      ie_web_app = ie.ole_query_interface('{0002DF05-0000-0000-C000-000000000046}') # => WIN32OLE object for dispinterface IWebBrowserApp 
+ */
+static VALUE
+fole_query_interface(VALUE self, VALUE str_iid)
+{
+    HRESULT hr;
+    OLECHAR *pBuf;
+    IID iid;
+    struct oledata *pole;
+    IDispatch *pDispatch;
+         
+    pBuf  = ole_mb2wc(StringValuePtr(str_iid), -1);
+    hr = CLSIDFromString(pBuf, &iid);
+    SysFreeString(pBuf);
+    if(FAILED(hr)) {
+        ole_raise(hr, eWIN32OLERuntimeError, 
+                  "illegal iid: `%s'",
+                  StringValuePtr(str_iid));
+    }
+
+    OLEData_Get_Struct(self, pole);
+    if(!pole->pDispatch) {
+        rb_raise(rb_eRuntimeError, "failed to get dispatch interface");
+    }
+
+    hr = pole->pDispatch->lpVtbl->QueryInterface(pole->pDispatch, &iid,
+                                                 (void **)&pDispatch);
+    if(FAILED(hr)) {
+        ole_raise(hr, eWIN32OLERuntimeError, 
+                  "failed to get interface `%s'", 
+                  StringValuePtr(str_iid));
+    }
+
+    return create_win32ole_object(cWIN32OLE, pDispatch, 0, 0);
+}
+
 static HRESULT
 ole_docinfo_from_type(ITypeInfo *pTypeInfo, BSTR *name, BSTR *helpstr, DWORD *helpcontext, BSTR *helpfile)
 {
@@ -7397,6 +7442,7 @@ Init_win32ole()
     rb_define_method(cWIN32OLE, "ole_type", fole_type, 0);
     rb_define_alias(cWIN32OLE, "ole_obj_help", "ole_type");
     rb_define_method(cWIN32OLE, "ole_typelib", fole_typelib, 0);
+    rb_define_method(cWIN32OLE, "ole_query_interface", fole_query_interface, 1);
 
     rb_define_const(cWIN32OLE, "VERSION", rb_str_new2(WIN32OLE_VERSION));
     rb_define_const(cWIN32OLE, "ARGV", rb_ary_new());
diff --git a/test/win32ole/test_win32ole.rb b/test/win32ole/test_win32ole.rb
index d7a9b368a1..cf39c7681d 100644
--- a/test/win32ole/test_win32ole.rb
+++ b/test/win32ole/test_win32ole.rb
@@ -237,6 +237,15 @@ if defined?(WIN32OLE)
       assert_equal(shell.NameSpace(0).title, shell._invoke(0x60020002, [0], [WIN32OLE::VARIANT::VT_VARIANT]).title)
     end
 
+    def test_ole_query_interface
+      shell=WIN32OLE.new('Shell.Application')
+      assert_raise(ArgumentError) {
+        shell2 = shell.ole_query_interface
+      }
+      shell2 = shell.ole_query_interface('{A4C6892C-3BA9-11D2-9DEA-00C04FB16162}')
+      assert_instance_of(WIN32OLE, shell2)
+    end
+
     def test_s_const_load
       assert(!defined?(CONST1::SsfWINDOWS))
       shell=WIN32OLE.new('Shell.Application')