* ext/win32ole/win32ole.c (get_ptr_of_variant, ole_set_safe_array,

ole_val2ptr_variant, ole_val2olevariantdata, ole_variant2val,
  ): fix some bugs of WIN32OLE_VARIANT.new when variant type is 
  VT_ARRAY|VT_BSTR or VT_BYREF.

* ext/win32ole/win32ole.c (folevariant_s_array, folevariant_initialize):
  WIN32OLE_VARIANT#[], WIN32OLE_VARIANT#[]=, WIN32OLE_VARIANT#value=
  is defined as instance method of WIN32OLE_VARIANT.

* test/win32ole/test_win32ole_variant.rb: add some test for 
  VT_ARRAY, VT_BYREF variant type.



git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@11959 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

ChangeLog

@@ -1,3 +1,17 @@
+Fri Mar  2 21:17:14 2007  Masaki Suketa  <masaki.suketa@nifty.ne.jp>
+
+	* ext/win32ole/win32ole.c (get_ptr_of_variant, ole_set_safe_array,
+	  ole_val2ptr_variant, ole_val2olevariantdata, ole_variant2val,
+	  ): fix some bugs of WIN32OLE_VARIANT.new when variant type is 
+	  VT_ARRAY|VT_BSTR or VT_BYREF.
+
+	* ext/win32ole/win32ole.c (folevariant_s_array, folevariant_initialize):
+	  WIN32OLE_VARIANT#[], WIN32OLE_VARIANT#[]=, WIN32OLE_VARIANT#value=
+	  is defined as instance method of WIN32OLE_VARIANT.
+
+	* test/win32ole/test_win32ole_variant.rb: add some test for 
+	  VT_ARRAY, VT_BYREF variant type.
+
 Fri Mar  2 07:58:24 2007  Yukihiro Matsumoto  <matz@ruby-lang.org>
 
 	* object.c (rb_obj_ivar_set): RDoc updated according to a

ext/win32ole/win32ole.c

@@ -55,6 +55,30 @@
 #define V_BOOL(X) V_UNION(X,boolVal)
 #endif
 
+#ifndef V_I1REF
+#define V_I1REF(X) V_UNION(X, pcVal)
+#endif
+
+#ifndef U_UI2REF
+#define V_UI2REF(X) V_UNION(X, puiVal)
+#endif
+
+#ifndef V_INT
+#define V_INT(X) V_UNION(X, intVal)
+#endif
+
+#ifndef V_INTREF
+#define V_INTREF(X) V_UNION(X, pintVal)
+#endif
+
+#ifndef V_UINT
+#define V_UINT(X) V_UNION(X, uintVal)
+#endif
+
+#ifndef V_UINTREF
+#define V_UINTREF(X) V_UNION(X, puintVal)
+#endif
+
 #define OLE_RELEASE(X) (X) ? ((X)->lpVtbl->Release(X)) : 0
 
 #define OLE_ADDREF(X) (X) ? ((X)->lpVtbl->AddRef(X)) : 0
@@ -78,9 +102,21 @@
     }\
 }
 
+#ifdef HAVE_LONG_LONG
+#define I8_2_NUM LL2NUM
+#define UI8_2_NUM ULL2NUM
+#define NUM2I8  NUM2LL
+#define NUM2UI8 NUM2ULL
+#else
+#define I8_2_NUM INT2NUM
+#define UI8_2_NUM UINT2NUM
+#define NUM2I8  NUM2INT
+#define NUM2UI8 NUM2UINT
+#endif
+
 #define WC2VSTR(x) ole_wc2vstr((x), TRUE)
 
-#define WIN32OLE_VERSION "0.9.9"
+#define WIN32OLE_VERSION "1.0.0"
 
 typedef HRESULT (STDAPICALLTYPE FNCOCREATEINSTANCEEX)
     (REFCLSID, IUnknown*, DWORD, COSERVERINFO*, DWORD, MULTI_QI*);
@@ -245,6 +281,7 @@ static long dimension(VALUE val);
 static long ary_len_of_dim(VALUE ary, long dim);
 static HRESULT ole_val_ary2variant_ary(VALUE val, VARIANT *var, VARTYPE vt);
 static void ole_val2variant(VALUE val, VARIANT *var);
+static void ole_val2variant_ex(VALUE val, VARIANT *var, VARTYPE vt);
 static void ole_val2ptr_variant(VALUE val, VARIANT *var);
 static void ole_set_byref(VARIANT *realvar, VARIANT *var,  VARTYPE vt);
 static void ole_val2olevariantdata(VALUE val, VARTYPE vt, struct olevariantdata *pvar);
@@ -951,6 +988,9 @@ static void * get_ptr_of_variant(VARIANT *pvar)
     case VT_I2:
         return &V_I2(pvar);
         break;
+    case VT_UI2:
+        return &V_UI2(pvar);
+        break;
     case VT_I4:
         return &V_I4(pvar);
         break;
@@ -963,6 +1003,20 @@ static void * get_ptr_of_variant(VARIANT *pvar)
     case VT_R8:
         return &V_R8(pvar);
         break;
+#if (_MSC_VER >= 1300) || defined(__CYGWIN__) || defined(__MINGW32__)
+    case VT_I8:
+        return &V_I8(pvar);
+        break;
+    case VT_UI8:
+        return &V_UI8(pvar);
+        break;
+#endif
+    case VT_INT:
+        return &V_INT(pvar);
+        break;
+    case VT_UINT:
+        return &V_UINT(pvar);
+        break;
     case VT_CY:
         return &V_CY(pvar);
         break;
@@ -970,10 +1024,10 @@ static void * get_ptr_of_variant(VARIANT *pvar)
         return &V_DATE(pvar);
         break;
     case VT_BSTR:
-        return &V_BSTR(pvar);
+        return V_BSTR(pvar);
         break;
     case VT_DISPATCH:
-        return &V_DISPATCH(pvar);
+        return V_DISPATCH(pvar);
         break;
     case VT_ERROR:
         return &V_ERROR(pvar);
@@ -982,7 +1036,7 @@ static void * get_ptr_of_variant(VARIANT *pvar)
         return &V_BOOL(pvar);
         break;
     case VT_UNKNOWN:
-        return &V_UNKNOWN(pvar);
+        return V_UNKNOWN(pvar);
         break;
     case VT_ARRAY:
         return &V_ARRAY(pvar);
@@ -1018,6 +1072,10 @@ ole_set_safe_array(long n, SAFEARRAY *psa, long *pid, long *pub, VALUE val, long
         val1 = ole_ary_m_entry(val, pid);
         p = val2variant_ptr(val1, &var, vt);
         if (is_all_index_under(pid, pub, dim) == Qtrue) {
+            if (V_VT(&var) == VT_DISPATCH && V_DISPATCH(&var) == NULL ||
+                V_VT(&var) == VT_UNKNOWN && V_UNKNOWN(&var) == NULL) {
+                rb_raise(eWIN32OLERuntimeError, "argument does not have IDispatch or IUnknown Interface");
+            }
             hr = SafeArrayPutElement(psa, pid, p);
         }
         if (FAILED(hr)) {
@@ -1153,7 +1211,6 @@ ole_val2variant(VALUE val, VARIANT *var)
         V_DISPATCH(var) = pole->pDispatch;
         return;
     }
-
     if (rb_obj_is_kind_of(val, cWIN32OLE_VARIANT)) {
         Data_Get_Struct(val, struct olevariantdata, pvar);
         VariantCopy(var, &(pvar->var));
@@ -1208,6 +1265,41 @@ ole_val2variant(VALUE val, VARIANT *var)
     }
 }
 
+static void
+ole_val2variant_ex(VALUE val, VARIANT *var, VARTYPE vt)
+{
+    if (val == Qnil) {
+        if (vt == VT_VARIANT) {
+            ole_val2variant2(val, var);
+        } else {
+            V_VT(var) = (vt & ~VT_BYREF);
+            if (V_VT(var) == VT_DISPATCH) {
+                V_DISPATCH(var) = NULL;
+            } else if (V_VT(var) == VT_UNKNOWN) {
+                V_UNKNOWN(var) = NULL;
+            }
+        }
+        return;
+    }
+#if (_MSC_VER >= 1300) || defined(__CYGWIN__) || defined(__MINGW32__)
+    switch(vt & ~VT_BYREF) {
+    case VT_I8:
+        V_VT(var) = VT_I8;
+        V_I8(var) = NUM2I8 (val);
+        break;
+    case VT_UI8:
+        V_VT(var) = VT_UI8;
+        V_UI8(var) = NUM2UI8(val);
+        break;
+    default:
+        ole_val2variant2(val, var);
+        break;
+    }
+#else  /* (_MSC_VER >= 1300) || defined(__CYGWIN__) || defined(__MINGW32__) */
+    ole_val2variant2(val, var);
+#endif
+}
+
 static void
 ole_val2ptr_variant(VALUE val, VARIANT *var)
 {
@@ -1287,12 +1379,18 @@ ole_set_byref(VARIANT *realvar, VARIANT *var,  VARTYPE vt)
             rb_raise(eWIN32OLERuntimeError, "variant type mismatch");
         }
         switch(vt & ~VT_BYREF) {
+        case VT_I1:
+            V_I1REF(var) = &V_I1(realvar);
+            break;
         case VT_UI1:
             V_UI1REF(var) = &V_UI1(realvar);
             break;
         case VT_I2:
             V_I2REF(var) = &V_I2(realvar);
             break;
+        case VT_UI2:
+            V_UI2REF(var) = &V_UI2(realvar);
+            break;
         case VT_I4:
             V_I4REF(var) = &V_I4(realvar);
             break;
@@ -1305,6 +1403,23 @@ ole_set_byref(VARIANT *realvar, VARIANT *var,  VARTYPE vt)
         case VT_R8:
             V_R8REF(var) = &V_R8(realvar);
             break;
+
+#if (_MSC_VER >= 1300)
+        case VT_I8:
+            V_I8REF(var) = &V_I8(realvar);
+            break;
+        case VT_UI8:
+            V_UI8REF(var) = &V_UI8(realvar);
+            break;
+#endif
+        case VT_INT:
+            V_INTREF(var) = &V_INT(realvar);
+            break;
+
+        case VT_UINT:
+            V_UINTREF(var) = &V_UINT(realvar);
+            break;
+
         case VT_CY:
             V_CYREF(var) = &V_CY(realvar);
             break;
@@ -1330,7 +1445,7 @@ ole_set_byref(VARIANT *realvar, VARIANT *var,  VARTYPE vt)
             V_ARRAYREF(var) = &V_ARRAY(realvar);
             break;
         default:
-            rb_raise(eWIN32OLERuntimeError, "unknown type specified:%d", vt);
+            rb_raise(eWIN32OLERuntimeError, "unknown type specified(setting BYREF):%d", vt);
             break;
         }
     }
@@ -1353,13 +1468,18 @@ ole_val2olevariantdata(VALUE val, VARTYPE vt, struct olevariantdata *pvar)
         if (SUCCEEDED(hr)) {
             memcpy(pdest, RSTRING_PTR(val), len);
             SafeArrayUnaccessData(psa);
-            V_VT(&(pvar->realvar)) = vt;
+            V_VT(&(pvar->realvar)) = (vt & ~VT_BYREF);
             p = V_ARRAY(&(pvar->realvar));
             if (p != NULL) {
                 SafeArrayDestroy(p);
             }
             V_ARRAY(&(pvar->realvar)) = psa;
+            if (vt & VT_BYREF) {
+                V_VT(&(pvar->var)) = vt;
+                V_ARRAYREF(&(pvar->var)) = &(V_ARRAY(&(pvar->realvar)));
+            } else {
                 hr = VariantCopy(&(pvar->var), &(pvar->realvar));
+            }
         } else {
             if (psa) 
                 SafeArrayDestroy(psa);
@@ -1374,13 +1494,22 @@ ole_val2olevariantdata(VALUE val, VARTYPE vt, struct olevariantdata *pvar)
             hr = ole_val_ary2variant_ary(val, &(pvar->realvar), (vt & ~VT_BYREF));
             if (SUCCEEDED(hr)) {
                 if (vt & VT_BYREF) {
-                    V_VT(&(pvar->var)) = V_VT(&(pvar->realvar)) | VT_BYREF;
+                    V_VT(&(pvar->var)) = vt;
                     V_ARRAYREF(&(pvar->var)) = &(V_ARRAY(&(pvar->realvar)));
                 } else {
                     hr = VariantCopy(&(pvar->var), &(pvar->realvar));
                 }
             }
         }
+#if (_MSC_VER >= 1300) || defined(__CYGWIN__) || defined(__MINGW32__)
+    } else if ( (vt & ~VT_BYREF) == VT_I8 || (vt & ~VT_BYREF) == VT_UI8) {
+        ole_val2variant_ex(val, &(pvar->realvar), (vt & ~VT_BYREF));
+        ole_val2variant_ex(val, &(pvar->var), (vt & ~VT_BYREF));
+        V_VT(&(pvar->var)) = vt;
+        if (vt & VT_BYREF) {
+            ole_set_byref(&(pvar->realvar), &(pvar->var), vt);
+        }
+#endif
     } else {
         if (val == Qnil) {
             V_VT(&(pvar->var)) = vt;
@@ -1393,7 +1522,7 @@ ole_val2olevariantdata(VALUE val, VARTYPE vt, struct olevariantdata *pvar)
                 }
             }
         } else {
-            ole_val2variant(val, &(pvar->realvar));
+            ole_val2variant_ex(val, &(pvar->realvar), (vt & ~VT_BYREF));
             if (vt == (VT_BYREF | VT_VARIANT)) {
                 ole_set_byref(&(pvar->realvar), &(pvar->var), vt);
             } else if (vt & VT_BYREF) {
@@ -1579,6 +1708,13 @@ ole_variant2val(VARIANT *pvar)
         break;
     case VT_NULL:
         break;
+    case VT_I1:
+        if(V_ISBYREF(pvar)) 
+            obj = INT2NUM((long)*V_I1REF(pvar));
+        else 
+            obj = INT2NUM((long)V_I1(pvar));
+        break;
+
     case VT_UI1:
         if(V_ISBYREF(pvar)) 
             obj = INT2NUM((long)*V_UI1REF(pvar));
@@ -1593,28 +1729,64 @@ ole_variant2val(VARIANT *pvar)
             obj = INT2NUM((long)V_I2(pvar));
         break;
 
+    case VT_UI2:
+        if(V_ISBYREF(pvar))
+            obj = INT2NUM((long)*V_UI2REF(pvar));
+        else 
+            obj = INT2NUM((long)V_UI2(pvar));
+        break;
+
     case VT_I4:
         if(V_ISBYREF(pvar))
             obj = INT2NUM((long)*V_I4REF(pvar));
         else 
             obj = INT2NUM((long)V_I4(pvar));
         break;
+
+    case VT_UI4:
+        if(V_ISBYREF(pvar))
+            obj = INT2NUM((long)*V_UI4REF(pvar));
+        else 
+            obj = INT2NUM((long)V_UI4(pvar));
+        break;
+
+    case VT_INT:
+        if(V_ISBYREF(pvar))
+            obj = INT2NUM((long)*V_INTREF(pvar));
+        else 
+            obj = INT2NUM((long)V_INT(pvar));
+        break;
+
+    case VT_UINT:
+        if(V_ISBYREF(pvar))
+            obj = INT2NUM((long)*V_UINTREF(pvar));
+        else 
+            obj = INT2NUM((long)V_UINT(pvar));
+        break;
+
 #if (_MSC_VER >= 1300) || defined(__CYGWIN__) || defined(__MINGW32__)
     case VT_I8:
-#ifdef HAVE_LONG_LONG
+        if(V_ISBYREF(pvar))
-        obj = LL2NUM(V_I8(pvar));
+#if (_MSC_VER >= 1300)
+            obj = I8_2_NUM(*V_I8REF(pvar));
 #else 
-        obj = INT2NUM(V_I8(pvar));
+            obj = Qnil;
 #endif
+        else
+            obj = I8_2_NUM(V_I8(pvar));
         break;
     case VT_UI8:
-#ifdef HAVE_LONG_LONG
+        if(V_ISBYREF(pvar))
-        obj = ULL2NUM(V_UI8(pvar));
+#if (_MSC_VER >= 1300)
+            obj = UI8_2_NUM(*V_UI8REF(pvar));
 #else
-        obj = UINT2NUM(V_UI8(pvar));
+            obj = Qnil;
 #endif
+        else
+            obj = UI8_2_NUM(V_UI8(pvar));
         break;
 #endif  /* (_MSC_VER >= 1300) || defined(__CYGWIN__) || defined(__MINGW32__) */
+
     case VT_R4:
         if(V_ISBYREF(pvar))
             obj = rb_float_new(*V_R4REF(pvar));
@@ -3985,12 +4157,14 @@ ole_typedesc2val(ITypeInfo *pTypeInfo, TYPEDESC *pTypeDesc, VALUE typedetails)
     case VT_UI4:
         typestr = rb_str_new2("UI4");
         break;
+#if (_MSC_VER >= 1300) || defined(__CYGWIN__) || defined(__MINGW32__)
     case VT_I8:
         typestr = rb_str_new2("I8");
         break;
     case VT_UI8:
         typestr = rb_str_new2("UI8");
         break;
+#endif
     case VT_INT:
         typestr = rb_str_new2("INT");
         break;
@@ -7324,8 +7498,6 @@ folevariant_s_array(VALUE klass, VALUE elems, VALUE vvt)
         V_ARRAY(&(pvar->var)) = psa;
     }
     if (psab) free(psab);
-    rb_define_singleton_method(obj, "[]", folevariant_ary_aref, -1);
-    rb_define_singleton_method(obj, "[]=", folevariant_ary_aset, -1);
     return obj;
 }
 
@@ -7372,15 +7544,6 @@ folevariant_initialize(VALUE self, VALUE args)
         ole_val2olevariantdata(val, vt, pvar);
     }
     vt = V_VT(&pvar->var);
-    if (vt & VT_ARRAY) {
-        rb_define_singleton_method(self, "[]", folevariant_ary_aref, -1);
-        rb_define_singleton_method(self, "[]=", folevariant_ary_aset, -1);
-        if (((vt & ~VT_BYREF) ==  (VT_ARRAY | VT_UI1)) && TYPE(val) == T_STRING) {
-            rb_define_singleton_method(self, "value=", folevariant_set_value, 1);
-        }
-    } else {
-        rb_define_singleton_method(self, "value=", folevariant_set_value, 1);
-    }
     return self;
 }
 
@@ -7444,11 +7607,14 @@ unlock_safe_array(SAFEARRAY *psa)
  *  WIN32OLE_VARIANT object is VT_ARRAY.
  *
  *  REMARK: 
- *     The all indicies should be 0 or natural number.
+ *     The all indicies should be 0 or natural number and 
+ *     lower than or equal to max indicies.
+ *     (This point is different with Ruby Array indicies.)
  *
  *     obj = WIN32OLE_VARIANT.new([[1,2,3],[4,5,6]])
  *     p obj[0,0] # => 1
  *     p obj[1,0] # => 4
+ *     p obj[2,0] # => WIN32OLERuntimeError
  *     p obj[0, -1] # => WIN32OLERuntimeError
  *
  */     
@@ -7463,6 +7629,10 @@ folevariant_ary_aref(int argc, VALUE *argv, VALUE self)
     HRESULT hr;
 
     Data_Get_Struct(self, struct olevariantdata, pvar);
+    if (!V_ISARRAY(&(pvar->var))) {
+        rb_raise(eWIN32OLERuntimeError, 
+                 "`[]' is not available for this variant type object");
+    }
     psa = get_locked_safe_array(self);
     if (psa == NULL) {
         return val;
@@ -7488,7 +7658,7 @@ val2variant_ptr(VALUE val, VARIANT *var, VARTYPE vt)
 {
     VOID *p = NULL;
     HRESULT hr = S_OK;
-    ole_val2variant2(val, var);
+    ole_val2variant_ex(val, var, vt);
     if ((vt & ~VT_BYREF) == VT_VARIANT) {
         p = var;
     } else {
@@ -7502,7 +7672,7 @@ val2variant_ptr(VALUE val, VARIANT *var, VARTYPE vt)
         p = get_ptr_of_variant(var);
     }
     if (p == NULL) {
-        rb_raise(rb_eRuntimeError, "failed to get ponter of variant");
+        rb_raise(rb_eRuntimeError, "failed to get pointer of variant");
     }
     return p;
 }
@@ -7516,12 +7686,15 @@ val2variant_ptr(VALUE val, VARIANT *var, VARTYPE vt)
  *  WIN32OLE_VARIANT object is VT_ARRAY.
  *
  *  REMARK: 
- *     The all indicies should be 0 or natural number.
+ *     The all indicies should be 0 or natural number and
+ *     lower than or equal to max indicies.
+ *     (This point is different with Ruby Array indicies.)
  *
  *     obj = WIN32OLE_VARIANT.new([[1,2,3],[4,5,6]])
  *     obj[0,0] = 7
  *     obj[1,0] = 8
  *     p obj.value # => [[7,2,3], [8,5,6]]
+ *     obj[2,0] = 9 # => WIN32OLERuntimeError
  *     obj[0, -1] = 9 # => WIN32OLERuntimeError
  *
  */     
@@ -7538,6 +7711,10 @@ folevariant_ary_aset(int argc, VALUE *argv, VALUE self)
     VOID *p = NULL;
 
     Data_Get_Struct(self, struct olevariantdata, pvar);
+    if (!V_ISARRAY(&(pvar->var))) {
+        rb_raise(eWIN32OLERuntimeError, 
+                 "`[]' is not available for this variant type object");
+    }
     psa = get_locked_safe_array(self);
     if (psa == NULL) {
         rb_raise(rb_eRuntimeError, "failed to get SafeArray pointer");
@@ -7545,8 +7722,13 @@ folevariant_ary_aset(int argc, VALUE *argv, VALUE self)
 
     pid = ary2safe_array_index(argc-1, argv, psa);
 
+    VariantInit(&var);
     vt = (V_VT(&(pvar->var)) & ~VT_ARRAY);
     p = val2variant_ptr(argv[argc-1], &var, vt);
+    if (V_VT(&var) == VT_DISPATCH && V_DISPATCH(&var) == NULL ||
+        V_VT(&var) == VT_UNKNOWN && V_UNKNOWN(&var) == NULL) {
+        rb_raise(eWIN32OLERuntimeError, "argument does not have IDispatch or IUnknown Interface");
+    }
     hr = SafeArrayPutElement(psa, pid, p);
     if (FAILED(hr)) {
         ole_raise(hr, eWIN32OLERuntimeError, "failed to SafeArrayPutElement");
@@ -7617,13 +7799,15 @@ folevariant_vartype(VALUE self)
  *  call-seq:
  *     WIN32OLE_VARIANT.value = val #=> set WIN32OLE_VARIANT value to val.
  *
- *  Sets variant value to val.
+ *  Sets variant value to val. If the val type does not match variant value 
+ *  type(vartype), then val is changed to match variant value type(vartype) 
+ *  before setting val.
+ *  Thie method is not available when vartype is VT_ARRAY(except VT_UI1|VT_ARRAY).
+ *  If the vartype is VT_UI1|VT_ARRAY, the val should be String object.
  *
- *     obj = WIN32OLE_VARIANT.new(1)
+ *     obj = WIN32OLE_VARIANT.new(1) # obj.vartype is WIN32OLE::VARIANT::VT_I4
- *     obj.value = 3.2
+ *     obj.value = 3.2 # 3.2 is changed to 3 when setting value.
  *     p obj.value # => 3 
- *
- *
  */     
 static VALUE
 folevariant_set_value(VALUE self, VALUE val)
@@ -7632,6 +7816,10 @@ folevariant_set_value(VALUE self, VALUE val)
     VARTYPE vt;
     Data_Get_Struct(self, struct olevariantdata, pvar);
     vt = V_VT(&(pvar->var));
+    if (V_ISARRAY(&(pvar->var)) && ((vt & ~VT_BYREF) != (VT_UI1|VT_ARRAY) || TYPE(val) != T_STRING)) {
+        rb_raise(eWIN32OLERuntimeError, 
+                 "`value=' is not available for this variant type object");
+    }
     ole_val2olevariantdata(val, vt, pvar);
     return Qnil;
 }
@@ -7735,8 +7923,10 @@ Init_win32ole()
     rb_define_const(mWIN32OLE_VARIANT, "VT_UI1", INT2FIX(VT_UI1));
     rb_define_const(mWIN32OLE_VARIANT, "VT_UI2", INT2FIX(VT_UI2));
     rb_define_const(mWIN32OLE_VARIANT, "VT_UI4", INT2FIX(VT_UI4));
+#if (_MSC_VER >= 1300) || defined(__CYGWIN__) || defined(__MINGW32__)
     rb_define_const(mWIN32OLE_VARIANT, "VT_I8", INT2FIX(VT_I8));
     rb_define_const(mWIN32OLE_VARIANT, "VT_UI8", INT2FIX(VT_UI8));
+#endif
     rb_define_const(mWIN32OLE_VARIANT, "VT_INT", INT2FIX(VT_INT));
     rb_define_const(mWIN32OLE_VARIANT, "VT_UINT", INT2FIX(VT_UINT));
     rb_define_const(mWIN32OLE_VARIANT, "VT_ARRAY", INT2FIX(VT_ARRAY));
@@ -7838,7 +8028,10 @@ Init_win32ole()
     rb_define_singleton_method(cWIN32OLE_VARIANT, "array", folevariant_s_array, 2);
     rb_define_method(cWIN32OLE_VARIANT, "initialize", folevariant_initialize, -2);
     rb_define_method(cWIN32OLE_VARIANT, "value", folevariant_value, 0);
+    rb_define_method(cWIN32OLE_VARIANT, "value=", folevariant_set_value, 1);
     rb_define_method(cWIN32OLE_VARIANT, "vartype", folevariant_vartype, 0);
+    rb_define_method(cWIN32OLE_VARIANT, "[]", folevariant_ary_aref, -1);
+    rb_define_method(cWIN32OLE_VARIANT, "[]=", folevariant_ary_aset, -1);
     rb_define_const(cWIN32OLE_VARIANT, "Empty", rb_funcall(cWIN32OLE_VARIANT, rb_intern("new"), 2, Qnil, INT2FIX(VT_EMPTY)));
     rb_define_const(cWIN32OLE_VARIANT, "Null", rb_funcall(cWIN32OLE_VARIANT, rb_intern("new"), 2, Qnil, INT2FIX(VT_NULL)));
     rb_define_const(cWIN32OLE_VARIANT, "Nothing", rb_funcall(cWIN32OLE_VARIANT, rb_intern("new"), 2, Qnil, INT2FIX(VT_DISPATCH)));

test/win32ole/test_win32ole_variant.rb

@@ -4,16 +4,21 @@ rescue LoadError
 end
 require "test/unit"
 
+$MSGS = []
+def add_skip_message(msg)
+  $MSGS.push msg
+end
+
 if defined?(WIN32OLE_VARIANT)
 
   class TestWIN32OLE_VARIANT < Test::Unit::TestCase
 
-    def test_new
+    def test_s_new
       obj = WIN32OLE_VARIANT.new('foo')
       assert_instance_of(WIN32OLE_VARIANT, obj)
     end
 
-    def test_new_no_argument
+    def test_s_new_no_argument
       ex = nil
       begin
         obj = WIN32OLE_VARIANT.new
@@ -24,7 +29,7 @@ if defined?(WIN32OLE_VARIANT)
       assert_equal("wrong number of arguments (0 for 1..3)", ex.message);
     end
 
-    def test_new_one_argument
+    def test_s_new_one_argument
       ex = nil
       begin
         obj = WIN32OLE_VARIANT.new('foo')
@@ -34,21 +39,264 @@ if defined?(WIN32OLE_VARIANT)
       assert_equal(nil, ex);
     end
 
+    def test_s_new_with_nil
+      obj = WIN32OLE_VARIANT.new(nil, WIN32OLE::VARIANT::VT_I2)
+      assert_equal(0, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_I2, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(nil, WIN32OLE::VARIANT::VT_I4)
+      assert_equal(0, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_I4, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(nil, WIN32OLE::VARIANT::VT_R4)
+      assert_equal(0, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_R4, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(nil, WIN32OLE::VARIANT::VT_R8)
+      assert_equal(0, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_R8, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(nil, WIN32OLE::VARIANT::VT_CY)
+      assert_equal("0", obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_CY, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(nil, WIN32OLE::VARIANT::VT_DATE)
+      assert_equal("1899/12/30 00:00:00", obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_DATE, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(nil, WIN32OLE::VARIANT::VT_BSTR)
+      assert_equal("", obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_BSTR, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(nil, WIN32OLE::VARIANT::VT_DISPATCH)
+      assert_nil(obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_DISPATCH, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(nil, WIN32OLE::VARIANT::VT_BOOL)
+      assert_equal(false, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_BOOL, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(nil, WIN32OLE::VARIANT::VT_VARIANT)
+      assert_nil(obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_VARIANT, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(nil, WIN32OLE::VARIANT::VT_I1)
+      assert_equal(0, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_I1, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(nil, WIN32OLE::VARIANT::VT_UI1)
+      assert_equal(0, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_UI1, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(nil, WIN32OLE::VARIANT::VT_UI2)
+      assert_equal(0, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_UI2, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(nil, WIN32OLE::VARIANT::VT_UI4)
+      assert_equal(0, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_UI4, obj.vartype)
+
+      
+      if defined?(WIN32OLE::VARIANT::VT_I8)
+        obj = WIN32OLE_VARIANT.new(nil, WIN32OLE::VARIANT::VT_I8)
+        assert_equal(0, obj.value)
+        assert_equal(WIN32OLE::VARIANT::VT_I8, obj.vartype)
+      end
+
+      if defined?(WIN32OLE::VARIANT::VT_UI8)
+        obj = WIN32OLE_VARIANT.new(nil, WIN32OLE::VARIANT::VT_UI8)
+        assert_equal(0, obj.value)
+        assert_equal(WIN32OLE::VARIANT::VT_UI8, obj.vartype)
+      end
+      
+      obj = WIN32OLE_VARIANT.new(nil, WIN32OLE::VARIANT::VT_INT)
+      assert_equal(0, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_INT, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(nil, WIN32OLE::VARIANT::VT_UINT)
+      assert_equal(0, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_UINT, obj.vartype)
+    end
+
+    def test_s_new_with_non_nil
+      obj = WIN32OLE_VARIANT.new(2, WIN32OLE::VARIANT::VT_I2)
+      assert_equal(2, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_I2, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(3, WIN32OLE::VARIANT::VT_I4)
+      assert_equal(3, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_I4, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(4.5, WIN32OLE::VARIANT::VT_R4)
+      assert_equal(4.5, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_R4, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(5.5, WIN32OLE::VARIANT::VT_R8)
+      assert_equal(5.5, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_R8, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(600, WIN32OLE::VARIANT::VT_CY)
+      assert_equal("600", obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_CY, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new("2001-06-15 12:17:34", WIN32OLE::VARIANT::VT_DATE)
+      assert_equal("2001/06/15 12:17:34", obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_DATE, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new("foo", WIN32OLE::VARIANT::VT_BSTR)
+      assert_equal("foo", obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_BSTR, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(true, WIN32OLE::VARIANT::VT_BOOL)
+      assert_equal(true, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_BOOL, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(2, WIN32OLE::VARIANT::VT_I1)
+      assert_equal(2, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_I1, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(3, WIN32OLE::VARIANT::VT_UI1)
+      assert_equal(3, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_UI1, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(4, WIN32OLE::VARIANT::VT_UI2)
+      assert_equal(4, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_UI2, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(5, WIN32OLE::VARIANT::VT_UI4)
+      assert_equal(5, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_UI4, obj.vartype)
+
+      if defined?(WIN32OLE::VARIANT::VT_I8)
+        obj = WIN32OLE_VARIANT.new(-123456789012345, WIN32OLE::VARIANT::VT_I8)
+        assert_equal(-123456789012345, obj.value)
+        assert_equal(WIN32OLE::VARIANT::VT_I8, obj.vartype)
+      end
+
+      if defined?(WIN32OLE::VARIANT::VT_UI8)
+        obj = WIN32OLE_VARIANT.new(123456789012345, WIN32OLE::VARIANT::VT_UI8)
+        assert_equal(123456789012345, obj.value)
+        assert_equal(WIN32OLE::VARIANT::VT_UI8, obj.vartype)
+      end
+
+      obj = WIN32OLE_VARIANT.new(4, WIN32OLE::VARIANT::VT_INT)
+      assert_equal(4, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_INT, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(5, WIN32OLE::VARIANT::VT_UINT)
+      assert_equal(5, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_UINT, obj.vartype)
+    end
+
+    def test_s_new_with_non_nil_byref
+      obj = WIN32OLE_VARIANT.new(2, WIN32OLE::VARIANT::VT_I2|WIN32OLE::VARIANT::VT_BYREF)
+      assert_equal(2, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_I2|WIN32OLE::VARIANT::VT_BYREF, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(3, WIN32OLE::VARIANT::VT_I4|WIN32OLE::VARIANT::VT_BYREF)
+      assert_equal(3, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_I4|WIN32OLE::VARIANT::VT_BYREF, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(4.5, WIN32OLE::VARIANT::VT_R4|WIN32OLE::VARIANT::VT_BYREF)
+      assert_equal(4.5, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_R4|WIN32OLE::VARIANT::VT_BYREF, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(5.5, WIN32OLE::VARIANT::VT_R8|WIN32OLE::VARIANT::VT_BYREF)
+      assert_equal(5.5, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_R8|WIN32OLE::VARIANT::VT_BYREF, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(600, WIN32OLE::VARIANT::VT_CY|WIN32OLE::VARIANT::VT_BYREF)
+      assert_equal("600", obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_CY|WIN32OLE::VARIANT::VT_BYREF, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new("2001-06-15 12:17:34", WIN32OLE::VARIANT::VT_DATE|WIN32OLE::VARIANT::VT_BYREF)
+      assert_equal("2001/06/15 12:17:34", obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_DATE|WIN32OLE::VARIANT::VT_BYREF, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new("foo", WIN32OLE::VARIANT::VT_BSTR|WIN32OLE::VARIANT::VT_BYREF)
+      assert_equal("foo", obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_BSTR|WIN32OLE::VARIANT::VT_BYREF, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(true, WIN32OLE::VARIANT::VT_BOOL|WIN32OLE::VARIANT::VT_BYREF)
+      assert_equal(true, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_BOOL|WIN32OLE::VARIANT::VT_BYREF, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(2, WIN32OLE::VARIANT::VT_I1|WIN32OLE::VARIANT::VT_BYREF)
+      assert_equal(2, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_I1|WIN32OLE::VARIANT::VT_BYREF, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(3, WIN32OLE::VARIANT::VT_UI1|WIN32OLE::VARIANT::VT_BYREF)
+      assert_equal(3, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_UI1|WIN32OLE::VARIANT::VT_BYREF, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(4, WIN32OLE::VARIANT::VT_UI2|WIN32OLE::VARIANT::VT_BYREF)
+      assert_equal(4, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_UI2|WIN32OLE::VARIANT::VT_BYREF, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(5, WIN32OLE::VARIANT::VT_UI4|WIN32OLE::VARIANT::VT_BYREF)
+      assert_equal(5, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_UI4|WIN32OLE::VARIANT::VT_BYREF, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(4, WIN32OLE::VARIANT::VT_INT|WIN32OLE::VARIANT::VT_BYREF)
+      assert_equal(4, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_INT|WIN32OLE::VARIANT::VT_BYREF, obj.vartype)
+
+      obj = WIN32OLE_VARIANT.new(5, WIN32OLE::VARIANT::VT_UINT|WIN32OLE::VARIANT::VT_BYREF)
+      assert_equal(5, obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_UINT|WIN32OLE::VARIANT::VT_BYREF, obj.vartype)
+    end
+    
+    # This test is failed in cygwin.
+    # The tagVARIANT definition has no union member pllVal in cygwin.
+    def test_s_new_with_i8_byref
+      if defined?(WIN32OLE::VARIANT::VT_I8) && /mswin/ =~ RUBY_PLATFORM
+        obj = WIN32OLE_VARIANT.new(-123456789012345, WIN32OLE::VARIANT::VT_I8|WIN32OLE::VARIANT::VT_BYREF)
+        assert_equal(-123456789012345, obj.value)
+        assert_equal(WIN32OLE::VARIANT::VT_I8|WIN32OLE::VARIANT::VT_BYREF, obj.vartype)
+      else
+        STDERR.puts("\n#{__FILE__}:#{__LINE__}:#{self.class.name}.test_s_new_with_i8_byref is skipped")
+      end
+    end
+    
+    # This test is failed in cygwin.
+    # The tagVARIANT definition has no union member pullVal in cygwin.
+    def test_s_new_with_ui8_byref
+      if defined?(WIN32OLE::VARIANT::VT_UI8) && /mswin/ =~ RUBY_PLATFORM
+        obj = WIN32OLE_VARIANT.new(123456789012345, WIN32OLE::VARIANT::VT_UI8|WIN32OLE::VARIANT::VT_BYREF)
+        assert_equal(123456789012345, obj.value)
+        assert_equal(WIN32OLE::VARIANT::VT_UI8|WIN32OLE::VARIANT::VT_BYREF, obj.vartype)
+      else
+        STDERR.puts("\n#{__FILE__}:#{__LINE__}:#{self.class.name}.test_s_new_with_ui8_byref is skipped.")
+      end
+    end
+
     def test_value
       obj = WIN32OLE_VARIANT.new('foo')
       assert_equal('foo', obj.value)
     end
 
-    def test_new_2_argument
+    def test_s_new_2_argument
       obj = WIN32OLE_VARIANT.new('foo', WIN32OLE::VARIANT::VT_BSTR|WIN32OLE::VARIANT::VT_BYREF)
       assert_equal('foo', obj.value);
     end
 
-    def test_new_2_argument2
+    def test_s_new_2_argument2
       obj = WIN32OLE_VARIANT.new('foo', WIN32OLE::VARIANT::VT_BSTR)
       assert_equal('foo', obj.value);
     end
 
+    def test_s_new_dispatch_array
+      vt = WIN32OLE::VARIANT::VT_ARRAY|WIN32OLE::VARIANT::VT_DISPATCH
+      obj = WIN32OLE_VARIANT.new(nil, vt)
+      assert_equal(vt, obj.vartype)
+      assert_nil(obj.value)
+
+      vt = WIN32OLE::VARIANT::VT_ARRAY|WIN32OLE::VARIANT::VT_DISPATCH|WIN32OLE::VARIANT::VT_BYREF
+      obj = WIN32OLE_VARIANT.new(nil, vt)
+      assert_equal(vt, obj.vartype)
+      assert_nil(obj.value)
+    end
+
     def test_s_array
       obj = WIN32OLE_VARIANT.array([2,3], WIN32OLE::VARIANT::VT_I4)
       assert_instance_of(WIN32OLE_VARIANT, obj)
@@ -75,6 +323,9 @@ if defined?(WIN32OLE_VARIANT)
       obj[0,1] = "string"
       obj[1,0] = 12.735
       assert_equal([[10, "string"],[12.735, nil],[nil,nil]], obj.value)
+
+      obj = WIN32OLE_VARIANT.array([2,3], WIN32OLE::VARIANT::VT_DISPATCH)
+      assert_equal([[nil, nil, nil],[nil,nil,nil]], obj.value)
     end
 
     def test_s_array_exc
@@ -194,6 +445,18 @@ if defined?(WIN32OLE_VARIANT)
       assert_equal(vartype, obj.vartype)
     end
 
+    def test_create_vt_array_str
+      vartype = WIN32OLE::VARIANT::VT_ARRAY|WIN32OLE::VARIANT::VT_BSTR
+      obj = WIN32OLE_VARIANT.new(["abc", "123"], vartype)
+      assert_equal(vartype, obj.vartype)
+      assert_equal(["abc", "123"], obj.value)
+      
+      vartype = WIN32OLE::VARIANT::VT_ARRAY|WIN32OLE::VARIANT::VT_BYREF|WIN32OLE::VARIANT::VT_BSTR
+      obj = WIN32OLE_VARIANT.new(["abc", "123"], vartype)
+      assert_equal(vartype, obj.vartype)
+      assert_equal(["abc", "123"], obj.value)
+    end
+
     def test_create_vt_array_exc
       exc = assert_raise(TypeError) {
         obj = WIN32OLE_VARIANT.new(1, WIN32OLE::VARIANT::VT_ARRAY);
@@ -213,6 +476,14 @@ if defined?(WIN32OLE_VARIANT)
       obj = WIN32OLE_VARIANT.new([65, 0].pack("C*"), WIN32OLE::VARIANT::VT_ARRAY|WIN32OLE::VARIANT::VT_UI1)
       assert_equal([65, 0].pack("C*"), obj.value)
 
+      obj = WIN32OLE_VARIANT.new("abc", WIN32OLE::VARIANT::VT_ARRAY|WIN32OLE::VARIANT::VT_UI1|WIN32OLE::VARIANT::VT_BYREF)
+      assert_equal("abc", obj.value)
+      obj.value = "DEF"
+      assert_equal("DEF", obj.value)
+      assert_equal(WIN32OLE::VARIANT::VT_ARRAY|WIN32OLE::VARIANT::VT_UI1|WIN32OLE::VARIANT::VT_BYREF, obj.vartype)
+      obj[1] = 71
+      assert_equal("DGF", obj.value)
+
     end
 
     def test_create_vt_array_int
@@ -289,7 +560,7 @@ if defined?(WIN32OLE_VARIANT)
       assert_equal(WIN32OLE::VARIANT::VT_I4, obj.vartype)
 
       obj = WIN32OLE_VARIANT.new([1,2])
-      assert_raise(NoMethodError) {
+      assert_raise(WIN32OLERuntimeError) {
         obj.value = [3,4]
       }
 
@@ -318,3 +589,4 @@ if defined?(WIN32OLE_VARIANT)
 
   end
 end
+