rdo-sqlite 0.0.3 → 0.0.4

data/ext/rdo_sqlite/driver.c

@@ -74,10 +74,34 @@ static VALUE rdo_sqlite_driver_prepare(VALUE self, VALUE cmd) {
 /** Quote a string literal for interpolation into a statement */
 static VALUE rdo_sqlite_driver_quote(VALUE self, VALUE str) {
   Check_Type(str, T_STRING);
-  char * quoted = sqlite3_mprintf("%q", RSTRING_PTR(str));
-  VALUE new_str = rb_str_new2(quoted);
-  sqlite3_free(quoted);
-  return new_str;
+
+  char          * raw = RSTRING_PTR(str);
+  unsigned long   len = RSTRING_LEN(str);
+  char          * buf = malloc(sizeof(char) * len * 2);
+  char          * b   = buf;
+  char          * s   = raw;
+
+  // not using sqlite3_mprintf() due to \0 check & performance
+  for (; (unsigned long) (s - raw) < len; ++s, ++b) {
+    switch (*s) {
+      case '\0':
+        free(buf);
+        rb_raise(rb_eArgError,
+            "Cannot #quote binary data. Use #prepare, #execute or a hex X'AABB' literal.");
+        break;
+
+      case '\'':
+        *(b++) = *s;
+
+      default:
+        *b = *s;
+    }
+  }
+
+  VALUE quoted = rb_str_new(buf, b - buf);
+  free(buf);
+
+  return quoted;
 }
 
 /** Initialize driver class */

data/lib/rdo/sqlite/version.rb

@@ -7,6 +7,6 @@
 
 module RDO
   module SQLite
-    VERSION = "0.0.3"
+    VERSION = "0.0.4"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rdo-sqlite
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-10-10 00:00:00.000000000 Z
+date: 2012-10-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rdo