ruby-oci8 2.0.6 → 2.1.0

data/ext/oci8/bind.c

@@ -2,19 +2,31 @@
 /*
  * bind.c
  *
- * $Author: kubo $
- * $Date: 2010-12-14 20:30:44 +0900 (Tue, 14 Dec 2010) $
- *
- * Copyright (C) 2002-2008 KUBO Takehiro <kubo@jiubao.org>
+ * Copyright (C) 2002-2011 KUBO Takehiro <kubo@jiubao.org>
  */
 #include "oci8.h"
 
+#ifndef OCI_ATTR_MAXCHAR_SIZE
+#define OCI_ATTR_MAXCHAR_SIZE 164
+#endif
+
 static ID id_bind_type;
+static VALUE sym_length;
+static VALUE sym_length_semantics;
+static VALUE sym_char;
+static VALUE sym_nchar;
 
 static VALUE cOCI8BindTypeBase;
 
+typedef struct {
+    oci8_bind_t obind;
+    sb4 bytelen;
+    sb4 charlen;
+    ub1 csfrm;
+} oci8_bind_string_t;
+
 /*
- * bind_char and bind_nchar
+ * bind_string
  */
 static VALUE bind_string_get(oci8_bind_t *obind, void *data, void *null_struct)
 {
@@ -24,67 +36,87 @@ static VALUE bind_string_get(oci8_bind_t *obind, void *data, void *null_struct)
 
 static void bind_string_set(oci8_bind_t *obind, void *data, void **null_structp, VALUE val)
 {
+    oci8_bind_string_t *obs = (oci8_bind_string_t *)obind;
     oci8_vstr_t *vstr = (oci8_vstr_t *)data;
 
     OCI8StringValue(val);
-    if (RSTRING_LEN(val) > obind->value_sz - sizeof(vstr->size)) {
-        rb_raise(rb_eArgError, "too long String to set. (%ld for %d)", RSTRING_LEN(val), obind->value_sz - (sb4)sizeof(vstr->size));
+    if (RSTRING_LEN(val) > obs->bytelen) {
+        rb_raise(rb_eArgError, "too long String to set. (%ld for %d)", RSTRING_LEN(val), obs->bytelen);
     }
     memcpy(vstr->buf, RSTRING_PTR(val), RSTRING_LEN(val));
     vstr->size = RSTRING_LEN(val);
 }
 
-static void bind_string_init(oci8_bind_t *obind, VALUE svc, VALUE val, VALUE length)
+static void bind_string_init(oci8_bind_t *obind, VALUE svc, VALUE val, VALUE param)
 {
+    oci8_bind_string_t *obs = (oci8_bind_string_t *)obind;
+    VALUE length;
+    VALUE length_semantics;
+    VALUE nchar;
     sb4 sz;
-    if (NIL_P(length)) {
-        if (NIL_P(val)) {
-            rb_raise(rb_eArgError, "value and length are both null.");
+
+    Check_Type(param, T_HASH);
+    length = rb_hash_aref(param, sym_length);
+    length_semantics = rb_hash_aref(param, sym_length_semantics);
+    nchar = rb_hash_aref(param, sym_nchar);
+
+    sz = NUM2INT(length);
+    if (sz < 0) {
+        rb_raise(rb_eArgError, "invalid bind length %d", sz);
+    }
+    if (length_semantics == sym_char) {
+        /* character semantics */
+        obs->charlen = sz;
+        obs->bytelen = sz = sz * oci8_nls_ratio;
+        if (oci8_nls_ratio == 1) {
+            /* sz must be bigger than charlen to suppress ORA-06502.
+             * I don't know the reason...
+             */
+            sz *= 2;
         }
-        StringValue(val);
-        sz = RSTRING_LEN(val);
     } else {
-        sz = NUM2INT(length);
+        /* byte semantics */
+        obs->bytelen = sz;
+        obs->charlen = 0;
     }
-    if (sz < 0) {
-        rb_raise(rb_eArgError, "invalid bind length %d", sz);
+    if (RTEST(nchar)) {
+        obs->csfrm = SQLCS_NCHAR; /* bind as NCHAR/NVARCHAR2 */
+    } else {
+        obs->csfrm = SQLCS_IMPLICIT; /* bind as CHAR/VARCHAR2 */
+    }
+    if (sz == 0) {
+        sz = 1; /* to avoid ORA-01459. */
     }
     sz += sizeof(sb4);
     obind->value_sz = sz;
     obind->alloc_sz = (sz + (sizeof(sb4) - 1)) & ~(sizeof(sb4) - 1);
 }
 
-static const oci8_bind_class_t bind_char_class = {
-    {
-        NULL,
-        oci8_bind_free,
-        sizeof(oci8_bind_t)
-    },
-    bind_string_get,
-    bind_string_set,
-    bind_string_init,
-    NULL,
-    NULL,
-    NULL,
-    NULL,
-    SQLT_LVC
-};
+static void bind_string_post_bind_hook(oci8_bind_t *obind)
+{
+    oci8_bind_string_t *obs = (oci8_bind_string_t *)obind;
 
-static const oci8_bind_class_t bind_nchar_class = {
+    if (obs->charlen != 0) {
+        chker2(OCIAttrSet(obind->base.hp.ptr, obind->base.type, (void*)&obs->charlen, 0, OCI_ATTR_MAXCHAR_SIZE, oci8_errhp),
+               &obind->base);
+    }
+    chker2(OCIAttrSet(obind->base.hp.ptr, obind->base.type, (void*)&obs->csfrm, 0, OCI_ATTR_CHARSET_FORM, oci8_errhp),
+           &obind->base);
+}
+
+static const oci8_bind_vtable_t bind_string_vtable = {
     {
         NULL,
         oci8_bind_free,
-        sizeof(oci8_bind_t)
+        sizeof(oci8_bind_string_t)
     },
     bind_string_get,
     bind_string_set,
     bind_string_init,
     NULL,
     NULL,
-    NULL,
-    NULL,
     SQLT_LVC,
-    SQLCS_NCHAR,
+    bind_string_post_bind_hook,
 };
 
 /*
@@ -98,225 +130,63 @@ static VALUE bind_raw_get(oci8_bind_t *obind, void *data, void *null_struct)
 
 static void bind_raw_set(oci8_bind_t *obind, void *data, void **null_structp, VALUE val)
 {
+    oci8_bind_string_t *obs = (oci8_bind_string_t *)obind;
     oci8_vstr_t *vstr = (oci8_vstr_t *)data;
 
     StringValue(val);
-    if (RSTRING_LEN(val) > obind->value_sz - sizeof(vstr->size)) {
-        rb_raise(rb_eArgError, "too long String to set. (%ld for %d)", RSTRING_LEN(val), obind->value_sz - (sb4)sizeof(vstr->size));
+    if (RSTRING_LEN(val) > obs->bytelen) {
+        rb_raise(rb_eArgError, "too long String to set. (%ld for %d)", RSTRING_LEN(val), obs->bytelen);
     }
     memcpy(vstr->buf, RSTRING_PTR(val), RSTRING_LEN(val));
     vstr->size = RSTRING_LEN(val);
 }
 
-static const oci8_bind_class_t bind_raw_class = {
+static const oci8_bind_vtable_t bind_raw_vtable = {
     {
         NULL,
         oci8_bind_free,
-        sizeof(oci8_bind_t)
+        sizeof(oci8_bind_string_t)
     },
     bind_raw_get,
     bind_raw_set,
     bind_string_init,
     NULL,
     NULL,
-    NULL,
-    NULL,
     SQLT_LVB
 };
 
-#ifdef USE_DYNAMIC_FETCH /* don't use DYNAMIC_FETCH. It doesn't work well... */
 /*
- * bind_long
+ * bind_binary_double
  */
-typedef struct {
-    VALUE obj;
-    ub4 alen;
-    char buf[1];
-} bind_long_t;
-#define bind_long_offset ((size_t)((bind_long_t*)0)->buf)
-
-static void bind_long_mark(oci8_base_t *base)
-{
-    oci8_bind_t *obind = (oci8_bind_t*)base;
-    ub4 idx = 0;
-
-    if (obind->valuep == NULL)
-        return;
-    do {
-        bind_long_t *bl = (bind_long_t *)((char*)obind->valuep + obind->alloc_sz * idx);
-        rb_gc_mark(bl->obj);
-    } while (++idx < obind->maxar_sz);
-}
-
-static VALUE bind_long_get(oci8_bind_t *obind, void *data, void *null_struct)
-{
-    bind_long_t *bl = (bind_long_t *)data;
-    return RTEST(bl->obj) ? rb_str_dup(bl->obj) : Qnil;
-}
-
-static void bind_long_set(oci8_bind_t *obind, void *data, void **null_structp, VALUE val)
-{
-    bind_long_t *bl = (bind_long_t *)data;
-    bl->obj = rb_str_dup(val);
-}
-
-static void bind_long_init(oci8_bind_t *obind, VALUE svc, VALUE val, VALUE length)
-{
-    sb4 sz = 0;
-
-    if (!NIL_P(length)) {
-        sz = NUM2INT(length);
-    }
-    if (sz < 4000) {
-        sz = 4000;
-    }
-    sz += bind_long_offset;
-    obind->value_sz = INT_MAX;
-    obind->alloc_sz = (sz + (sizeof(VALUE) - 1)) & ~(sizeof(VALUE) - 1);
-}
-
-static void bind_long_init_elem(oci8_bind_t *obind, VALUE svc)
-{
-    ub4 idx = 0;
-
-    do {
-        bind_long_t *bl = (bind_long_t *)((char*)obind->valuep + obind->alloc_sz * idx);
-        bl->obj = Qnil;
-    } while (++idx < obind->maxar_sz);
-}
-
-static ub1 bind_long_in(oci8_bind_t *obind, ub4 idx, ub1 piece, void **valuepp, ub4 **alenpp, void **indpp)
-{
-    bind_long_t *bl = (bind_long_t *)((char*)obind->valuep + obind->alloc_sz * idx);
-
-    *alenpp = &bl->alen;
-    *indpp = &obind->u.inds[idx];
-    if (NIL_P(bl->obj)) {
-        *valuepp = NULL;
-        bl->alen = 0;
-        obind->u.inds[idx] = -1;
-    } else {
-        StringValue(bl->obj);
-        *valuepp = RSTRING_PTR(bl->obj);
-        bl->alen = RSTRING_LEN(bl->obj);
-        obind->u.inds[idx] = 0;
-    }
-    return OCI_ONE_PIECE;
-}
-
-static void bind_long_out(oci8_bind_t *obind, ub4 idx, ub1 piece, void **valuepp, ub4 **alenpp, void **indpp)
-{
-    bind_long_t *bl = (bind_long_t *)((char*)obind->valuep + obind->alloc_sz * idx);
-
-    switch (piece) {
-    case OCI_NEXT_PIECE:
-    case OCI_LAST_PIECE:
-        if (bl->alen > 0) {
-            if (!RTEST(bl->obj)) {
-                bl->obj = rb_str_buf_new(bl->alen);
-            }
-            rb_str_buf_cat(bl->obj, bl->buf, bl->alen);
-        }
-        break;
-    default:
-        /* OCI_FIRST_PIECE is passed at the first call according to manuals.
-         * But OCI_ONE_PIECE is passed on Oracle 8 and 8i on Windows...
-         */
-        bl->obj = Qnil;
-    }
-    *valuepp = bl->buf;
-    *alenpp = &bl->alen;
-    *indpp = &obind->u.inds[idx];
-    bl->alen = obind->alloc_sz - bind_long_offset;
-    obind->u.inds[idx] = 0;
-}
-
-static const oci8_bind_class_t bind_long_class = {
-    {
-        bind_long_mark,
-        oci8_bind_free,
-        sizeof(oci8_bind_t)
-    },
-    bind_long_get,
-    bind_long_set,
-    bind_long_init,
-    bind_long_init_elem,
-    bind_long_in,
-    bind_long_out,
-    NULL,
-    SQLT_CHR
-};
-
-/*
- * bind_long_raw
- */
-static const oci8_bind_class_t bind_long_raw_class = {
-    {
-        bind_long_mark,
-        oci8_bind_free,
-        sizeof(oci8_bind_t)
-    },
-    bind_long_get,
-    bind_long_set,
-    bind_long_init,
-    bind_long_init_elem,
-    bind_long_in,
-    bind_long_out,
-    NULL,
-    SQLT_BIN
-};
-#endif /* USE_DYNAMIC_FETCH */
-
-/*
- * bind_float
- */
-static VALUE bind_float_get(oci8_bind_t *obind, void *data, void *null_struct)
+static VALUE bind_binary_double_get(oci8_bind_t *obind, void *data, void *null_struct)
 {
     return rb_float_new(*(double*)data);
 }
 
-static void bind_float_set(oci8_bind_t *obind, void *data, void **null_structp, VALUE val)
+static void bind_binary_double_set(oci8_bind_t *obind, void *data, void **null_structp, VALUE val)
 {
     /* val is converted to Float if it isn't Float. */
     *(double*)data = RFLOAT_VALUE(rb_Float(val));
 }
 
-static void bind_float_init(oci8_bind_t *obind, VALUE svc, VALUE val, VALUE length)
+static void bind_binary_double_init(oci8_bind_t *obind, VALUE svc, VALUE val, VALUE length)
 {
     obind->value_sz = sizeof(double);
     obind->alloc_sz = sizeof(double);
 }
 
-static const oci8_bind_class_t bind_float_class = {
-    {
-        NULL,
-        oci8_bind_free,
-        sizeof(oci8_bind_t)
-    },
-    bind_float_get,
-    bind_float_set,
-    bind_float_init,
-    NULL,
-    NULL,
-    NULL,
-    NULL,
-    SQLT_FLT
-};
-
 #ifndef SQLT_BDOUBLE
 #define SQLT_BDOUBLE 22
 #endif
-static const oci8_bind_class_t bind_binary_double_class = {
+static const oci8_bind_vtable_t bind_binary_double_vtable = {
     {
         NULL,
         oci8_bind_free,
         sizeof(oci8_bind_t)
     },
-    bind_float_get,
-    bind_float_set,
-    bind_float_init,
-    NULL,
-    NULL,
+    bind_binary_double_get,
+    bind_binary_double_set,
+    bind_binary_double_init,
     NULL,
     NULL,
     SQLT_BDOUBLE
@@ -325,7 +195,7 @@ static const oci8_bind_class_t bind_binary_double_class = {
 static VALUE oci8_bind_get(VALUE self)
 {
     oci8_bind_t *obind = DATA_PTR(self);
-    const oci8_bind_class_t *obc = (const oci8_bind_class_t *)obind->base.klass;
+    const oci8_bind_vtable_t *vptr = (const oci8_bind_vtable_t *)obind->base.vptr;
     ub4 idx = obind->curar_idx;
     void **null_structp = NULL;
 
@@ -333,7 +203,7 @@ static VALUE oci8_bind_get(VALUE self)
         if (obind->u.inds[idx] != 0)
             return Qnil;
     }
-    return obc->get(obind, (void*)((size_t)obind->valuep + obind->alloc_sz * idx), null_structp);
+    return vptr->get(obind, (void*)((size_t)obind->valuep + obind->alloc_sz * idx), null_structp);
 }
 
 VALUE oci8_bind_get_data(VALUE self)
@@ -358,7 +228,7 @@ VALUE oci8_bind_get_data(VALUE self)
 static VALUE oci8_bind_set(VALUE self, VALUE val)
 {
     oci8_bind_t *obind = DATA_PTR(self);
-    const oci8_bind_class_t *obc = (const oci8_bind_class_t *)obind->base.klass;
+    const oci8_bind_vtable_t *vptr = (const oci8_bind_vtable_t *)obind->base.vptr;
     ub4 idx = obind->curar_idx;
 
     if (NIL_P(val)) {
@@ -377,7 +247,7 @@ static VALUE oci8_bind_set(VALUE self, VALUE val)
             null_structp = &obind->u.null_structs[idx];
             *(OCIInd*)obind->u.null_structs[idx] = 0;
         }
-        obc->set(obind, (void*)((size_t)obind->valuep + obind->alloc_sz * idx), null_structp, val);
+        vptr->set(obind, (void*)((size_t)obind->valuep + obind->alloc_sz * idx), null_structp, val);
     }
     return self;
 }
@@ -409,7 +279,7 @@ void oci8_bind_set_data(VALUE self, VALUE val)
 static VALUE oci8_bind_initialize(VALUE self, VALUE svc, VALUE val, VALUE length, VALUE max_array_size)
 {
     oci8_bind_t *obind = DATA_PTR(self);
-    const oci8_bind_class_t *bind_class = (const oci8_bind_class_t *)obind->base.klass;
+    const oci8_bind_vtable_t *bind_class = (const oci8_bind_vtable_t *)obind->base.vptr;
     ub4 cnt = 1;
 
     obind->tdo = Qnil;
@@ -471,22 +341,20 @@ void Init_oci8_bind(VALUE klass)
 {
     cOCI8BindTypeBase = klass;
     id_bind_type = rb_intern("bind_type");
+    sym_length = ID2SYM(rb_intern("length"));
+    sym_length_semantics = ID2SYM(rb_intern("length_semantics"));
+    sym_char = ID2SYM(rb_intern("char"));
+    sym_nchar = ID2SYM(rb_intern("nchar"));
 
     rb_define_method(cOCI8BindTypeBase, "initialize", oci8_bind_initialize, 4);
     rb_define_method(cOCI8BindTypeBase, "get", oci8_bind_get, 0);
     rb_define_method(cOCI8BindTypeBase, "set", oci8_bind_set, 1);
 
     /* register primitive data types. */
-    oci8_define_bind_class("CHAR", &bind_char_class);
-    oci8_define_bind_class("NCHAR", &bind_nchar_class);
-    oci8_define_bind_class("RAW", &bind_raw_class);
-#ifdef USE_DYNAMIC_FETCH
-    oci8_define_bind_class("Long", &bind_long_class);
-    oci8_define_bind_class("LongRaw", &bind_long_raw_class);
-#endif /* USE_DYNAMIC_FETCH */
-    oci8_define_bind_class("Float", &bind_float_class);
+    oci8_define_bind_class("String", &bind_string_vtable);
+    oci8_define_bind_class("RAW", &bind_raw_vtable);
     if (oracle_client_version >= ORAVER_10_1) {
-        oci8_define_bind_class("BinaryDouble", &bind_binary_double_class);
+        oci8_define_bind_class("BinaryDouble", &bind_binary_double_vtable);
     }
 }