trilogy 2.13.1 → 2.13.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 50f7b9c6e6db317bed3d230ffc0714c109ac78cce85a391a3530cee82c8676e2
-  data.tar.gz: 7c95de9ace5aa5669ce8fac0bd95513be353cce52bed60a6441a9558dd18abe5
+  metadata.gz: 4ea4b038d25e87b1d26c6dc0af6aa37b3cfa2b7244ea4788155bb392f7a5e93e
+  data.tar.gz: c4201dd5f2c3974244cea9ae97af6052d3d1991509e64c2dde8319882b00a53e
 SHA512:
-  metadata.gz: 88a4f0114420039839374a8526dea80982862772d7b89e940dd509109b61500191723912372d3798415c02b8af9168b8f7d5dceaec79518577096a0314c1d0f5
-  data.tar.gz: 16d595b50c2804a377cdb8899add3ac9158b2991297fe6163c83d7699b216222fc3991c84c5dc67a8331bbe1480cd7c3afc24938297a46a34630d4bb47452602
+  metadata.gz: 2e58cdf8bf20b4d31e43b811ceb7972a772c2ccb44133f715a4d9c0723c1c1ed4b3139c5e7be68575b119932348f5d8d393d088b88fd73ba058eeef9d8203796
+  data.tar.gz: ef9527a11371f0b6dbab901eb6c682e5a6e7cbf48d81e483b647682355b825a08ca342491a523859f543bfcaaf598588a713c5859baf8a883fe5adcda14f59ea

data/ext/trilogy-ruby/cast.c

@@ -56,8 +56,11 @@ static const char *ruby_encoding_name_map[] = {
 
 static rb_encoding * encoding_for_charset(TRILOGY_CHARSET_t charset)
 {
-    static rb_encoding * map[TRILOGY_CHARSET_MAX];
+    if (RB_UNLIKELY(charset >= TRILOGY_CHARSET_MAX)) {
+        return rb_enc_find(ruby_encoding_name_map[trilogy_encoding_from_charset(charset)]);
+    }
 
+    static rb_encoding * map[TRILOGY_CHARSET_MAX];
     if (RB_LIKELY(map[charset])) {
         return map[charset];
     }

data/ext/trilogy-ruby/cext.c

@@ -1128,7 +1128,6 @@ static VALUE rb_trilogy_query(VALUE self, VALUE query)
     struct trilogy_ctx *ctx = get_open_ctx(self);
 
     StringValue(query);
-    query = rb_str_export_to_enc(query, ctx->encoding);
     query = rb_str_new_frozen(query);
 
     rb_trilogy_acquire_buffer(ctx);
@@ -1189,13 +1188,6 @@ static VALUE rb_trilogy_escape(VALUE self, VALUE str)
     struct trilogy_ctx *ctx = get_open_ctx(self);
 
     StringValue(str);
-    str = rb_str_new_frozen(str);
-
-    rb_encoding *str_enc = rb_enc_get(str);
-
-    if (!rb_enc_asciicompat(str_enc)) {
-        rb_raise(rb_eEncCompatError, "input string must be ASCII-compatible");
-    }
 
     const char *escaped_str;
     size_t escaped_len;
@@ -1208,7 +1200,7 @@ static VALUE rb_trilogy_escape(VALUE self, VALUE str)
         handle_trilogy_error(ctx, rc, "trilogy_escape");
     }
 
-    VALUE escaped_string = rb_enc_str_new(escaped_str, escaped_len, str_enc);
+    VALUE escaped_string = rb_enc_str_new(escaped_str, escaped_len, ctx->encoding);
 
     rb_trilogy_release_buffer(ctx);
 
@@ -1371,9 +1363,9 @@ RUBY_FUNC_EXPORTED void Init_cext(void)
     rb_define_private_method(Trilogy, "_connect", rb_trilogy_connect, 4);
     rb_define_method(Trilogy, "change_db", rb_trilogy_change_db, 1);
     rb_define_alias(Trilogy, "select_db", "change_db");
-    rb_define_method(Trilogy, "query", rb_trilogy_query, 1);
+    rb_define_private_method(Trilogy, "_query", rb_trilogy_query, 1);
     rb_define_method(Trilogy, "ping", rb_trilogy_ping, 0);
-    rb_define_method(Trilogy, "escape", rb_trilogy_escape, 1);
+    rb_define_private_method(Trilogy, "_escape", rb_trilogy_escape, 1);
     rb_define_method(Trilogy, "close", rb_trilogy_close, 0);
     rb_define_method(Trilogy, "closed?", rb_trilogy_closed, 0);
     rb_define_method(Trilogy, "check", rb_trilogy_check, 0);

data/lib/trilogy/encoding.rb

@@ -84,7 +84,13 @@ class Trilogy
       "binary"   => CHARSET_BINARY,
     }.freeze
 
+    UNSUPPORTED_ENCODINGS = %w[big5 sjis gbk cp932 gb18030 ucs2 utf16 utf16le utf32].freeze
+
     def self.find(mysql_encoding)
+      if UNSUPPORTED_ENCODINGS.include?(mysql_encoding)
+        raise ArgumentError, "Unsupported encoding: #{mysql_encoding}"
+      end
+
       unless rb_encoding = RUBY_ENCODINGS[mysql_encoding]
         raise ArgumentError, "Unknown or unsupported encoding: #{mysql_encoding}"
       end

data/lib/trilogy/version.rb

@@ -1,3 +1,3 @@
 class Trilogy
-  VERSION = "2.13.1"
+  VERSION = "2.13.2"
 end

data/lib/trilogy.rb

@@ -16,6 +16,27 @@ class Trilogy
     end
   private_constant :IO_TIMEOUT_ERROR
 
+  def query(arg)
+    str = String.try_convert(arg) or raise TypeError, "no implicit conversion of #{arg.class} into String"
+    str = str.dup if str.equal?(arg)
+    str.freeze
+
+    str_encoding = str.encoding
+    if @encoding != str_encoding && ::Encoding::BINARY != str_encoding && !(str.ascii_only? && @encoding.ascii_compatible?)
+      raise ::Encoding::CompatibilityError, "Trilogy connection uses #{@encoding} but the query was provided in #{str_encoding}"
+    end
+
+    _query(str)
+  end
+
+  def escape(arg)
+    str = String.try_convert(arg) or raise TypeError, "no implicit conversion of #{arg.class} into String"
+    if @encoding != str.encoding && !(str.ascii_only? && @encoding.ascii_compatible?) && str.encoding != ::Encoding::ASCII_8BIT
+      str = str.encode(@encoding)
+    end
+    _escape(str)
+  end
+
   module Synchronization
     def initialize(...)
       @mutex = Mutex.new
@@ -47,10 +68,12 @@ class Trilogy
 
   prepend(Synchronization)
 
+  attr_reader :encoding
+
   def initialize(options = {})
     options[:port] = options[:port].to_i if options[:port]
     mysql_encoding = options[:encoding] || "utf8mb4"
-    encoding = Trilogy::Encoding.find(mysql_encoding)
+    @encoding = Trilogy::Encoding.find(mysql_encoding)
     charset = Trilogy::Encoding.charset(mysql_encoding)
     @connection_options = options
     @connected_host = nil
@@ -99,7 +122,7 @@ class Trilogy
       end
     end
 
-    _connect(socket, encoding, charset, options)
+    _connect(socket, @encoding, charset, options)
   ensure
     # Socket's fd will be dup'd in C
     socket&.close

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: trilogy
 version: !ruby/object:Gem::Version
-  version: 2.13.1
+  version: 2.13.2
 platform: ruby
 authors:
 - GitHub Engineering
@@ -87,7 +87,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 4.0.6
+rubygems_version: 4.0.20
 specification_version: 4
 summary: A friendly MySQL-compatible library for Ruby, binding to libtrilogy
 test_files: []