From e241b45400af88c671b9527937cc9ea603cc5cf1 Mon Sep 17 00:00:00 2001 From: nobu Date: Mon, 21 Sep 2015 15:29:20 +0000 Subject: win32ole.c: inadvertent symbol creation * ext/win32ole/win32ole.c (GetIDsOfNames, fole_missing): avoid inadvertent symbol creation. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@51913 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- ext/win32ole/win32ole.c | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) (limited to 'ext/win32ole') diff --git a/ext/win32ole/win32ole.c b/ext/win32ole/win32ole.c index 4dfcd75edb..1964314484 100644 --- a/ext/win32ole/win32ole.c +++ b/ext/win32ole/win32ole.c @@ -349,7 +349,7 @@ static HRESULT ( STDMETHODCALLTYPE GetIDsOfNames )( Win32OLEIDispatch* p = (Win32OLEIDispatch*)This; */ char* psz = ole_wc2mb(*rgszNames); // support only one method - ID nameid = rb_intern(psz); + ID nameid = rb_check_id_cstr(psz, (long)strlen(psz), cWIN32OLE_enc); free(psz); if ((ID)(DISPID)nameid != nameid) return E_NOINTERFACE; *rgDispId = (DISPID)nameid; @@ -3277,16 +3277,18 @@ fole_each(VALUE self) static VALUE fole_missing(int argc, VALUE *argv, VALUE self) { - ID id; + VALUE mid, sym; const char* mname; - size_t n; + long n; rb_check_arity(argc, 1, UNLIMITED_ARGUMENTS); - id = rb_to_id(argv[0]); - mname = rb_id2name(id); + mid = argv[0]; + sym = rb_check_symbol(&mid); + if (sym) mid = rb_sym2str(sym); + mname = StringValueCStr(mid); if(!mname) { rb_raise(rb_eRuntimeError, "fail: unknown method or property"); } - n = strlen(mname); + n = RSTRING_LEN(mid); #if SIZEOF_SIZE_T > SIZEOF_LONG if (n >= LONG_MAX) { rb_raise(rb_eRuntimeError, "too long method or property name"); @@ -3294,12 +3296,12 @@ fole_missing(int argc, VALUE *argv, VALUE self) #endif if(mname[n-1] == '=') { rb_check_arity(argc, 2, 2); - argv[0] = rb_enc_str_new(mname, (long)(n-1), cWIN32OLE_enc); + argv[0] = rb_enc_str_new(mname, (n-1), cWIN32OLE_enc); return ole_propertyput(self, argv[0], argv[1]); } else { - argv[0] = rb_enc_str_new(mname, (long)n, cWIN32OLE_enc); + argv[0] = rb_enc_str_new(mname, n, cWIN32OLE_enc); return ole_invoke(argc, argv, self, DISPATCH_METHOD|DISPATCH_PROPERTYGET, FALSE); } } -- cgit v1.2.3