* ext/win32ole/win32ole.c (fole_initialize): support licensed COM server. Thanks to Gray Wolf. [Feature :#13828]

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@59665 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
suke 2017-08-26 23:39:43 +00:00
Родитель 6582a2f6f1
Коммит 917beef327
1 изменённых файлов: 41 добавлений и 5 удалений

Просмотреть файл

@ -26,7 +26,7 @@
const IID IID_IMultiLanguage2 = {0xDCCFC164, 0x2B38, 0x11d2, {0xB7, 0xEC, 0x00, 0xC0, 0x4F, 0x8F, 0x5D, 0x9A}}; const IID IID_IMultiLanguage2 = {0xDCCFC164, 0x2B38, 0x11d2, {0xB7, 0xEC, 0x00, 0xC0, 0x4F, 0x8F, 0x5D, 0x9A}};
#endif #endif
#define WIN32OLE_VERSION "1.8.5" #define WIN32OLE_VERSION "1.8.6"
typedef HRESULT (STDAPICALLTYPE FNCOCREATEINSTANCEEX) typedef HRESULT (STDAPICALLTYPE FNCOCREATEINSTANCEEX)
(REFCLSID, IUnknown*, DWORD, COSERVERINFO*, DWORD, MULTI_QI*); (REFCLSID, IUnknown*, DWORD, COSERVERINFO*, DWORD, MULTI_QI*);
@ -2440,12 +2440,16 @@ fole_s_ole_uninitialize(VALUE self)
/* /*
* call-seq: * call-seq:
* WIN32OLE.new(server, [host]) -> WIN32OLE object * WIN32OLE.new(server, [host]) -> WIN32OLE object
* WIN32OLE.new(server, license: 'key') -> WIN32OLE object
* *
* Returns a new WIN32OLE object(OLE Automation object). * Returns a new WIN32OLE object(OLE Automation object).
* The first argument server specifies OLE Automation server. * The first argument server specifies OLE Automation server.
* The first argument should be CLSID or PROGID. * The first argument should be CLSID or PROGID.
* If second argument host specified, then returns OLE Automation * If second argument host specified, then returns OLE Automation
* object on host. * object on host.
* If :license keyword argument is provided,
* IClassFactory2::CreateInstanceLic is used to create instance of
* licensed server.
* *
* WIN32OLE.new('Excel.Application') # => Excel OLE Automation WIN32OLE object. * WIN32OLE.new('Excel.Application') # => Excel OLE Automation WIN32OLE object.
* WIN32OLE.new('{00024500-0000-0000-C000-000000000046}') # => Excel OLE Automation WIN32OLE object. * WIN32OLE.new('{00024500-0000-0000-C000-000000000046}') # => Excel OLE Automation WIN32OLE object.
@ -2456,13 +2460,19 @@ fole_initialize(int argc, VALUE *argv, VALUE self)
VALUE svr_name; VALUE svr_name;
VALUE host; VALUE host;
VALUE others; VALUE others;
VALUE opts;
HRESULT hr; HRESULT hr;
CLSID clsid; CLSID clsid;
OLECHAR *pBuf; OLECHAR *pBuf;
OLECHAR *key_buf;
IDispatch *pDispatch; IDispatch *pDispatch;
IClassFactory2 * pIClassFactory2;
void *p; void *p;
static ID keyward_ids[1];
VALUE kwargs[1];
rb_call_super(0, 0); rb_call_super(0, 0);
rb_scan_args(argc, argv, "11*", &svr_name, &host, &others); rb_scan_args(argc, argv, "11*:", &svr_name, &host, &others, &opts);
StringValue(svr_name); StringValue(svr_name);
if (rb_safe_level() > 0 && OBJ_TAINTED(svr_name)) { if (rb_safe_level() > 0 && OBJ_TAINTED(svr_name)) {
@ -2491,9 +2501,35 @@ fole_initialize(int argc, VALUE *argv, VALUE self)
StringValuePtr(svr_name)); StringValuePtr(svr_name));
} }
/* get IDispatch interface */ if (!keyward_ids[0]) {
hr = CoCreateInstance(&clsid, NULL, CLSCTX_INPROC_SERVER | CLSCTX_LOCAL_SERVER, keyward_ids[0] = rb_intern_const("license");
&IID_IDispatch, &p); }
rb_get_kwargs(opts, keyward_ids, 0, 1, kwargs);
if (kwargs[0] == Qundef) {
/* get IDispatch interface */
hr = CoCreateInstance(
&clsid,
NULL,
CLSCTX_INPROC_SERVER | CLSCTX_LOCAL_SERVER,
&IID_IDispatch,
&p
);
} else {
hr = CoGetClassObject(
&clsid,
CLSCTX_INPROC_SERVER | CLSCTX_LOCAL_SERVER,
NULL,
&IID_IClassFactory2,
(LPVOID)&pIClassFactory2
);
if (hr == S_OK) {
key_buf = ole_vstr2wc(kwargs[0]);
hr = pIClassFactory2->lpVtbl->CreateInstanceLic(pIClassFactory2, NULL, NULL, &IID_IDispatch, key_buf, &p);
SysFreeString(key_buf);
OLE_RELEASE(pIClassFactory2);
}
}
pDispatch = p; pDispatch = p;
if(FAILED(hr)) { if(FAILED(hr)) {
ole_raise(hr, eWIN32OLERuntimeError, ole_raise(hr, eWIN32OLERuntimeError,