github-migration/mysql2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

compare length of original and escaped string, if they match return the original

Browse Source
This commit is contained in:
Brian Lopez 2010-04-03 01:42:15 -07:00
parent 1b9e35712f
commit 677c309963
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
ext/mysql2_ext.c
View File

@ -125,17 +125,25 @@ static VALUE rb_mysql_client_query(VALUE self, VALUE sql) {
static VALUE rb_mysql_client_escape(VALUE self, VALUE str) { static VALUE rb_mysql_client_escape(VALUE self, VALUE str) {
MYSQL * client; MYSQL * client;
VALUE newStr; VALUE newStr;
unsigned long newLen, oldLen;
Check_Type(str, T_STRING); Check_Type(str, T_STRING);
char escaped[(RSTRING_LEN(str)*2)+1]; oldLen = RSTRING_LEN(str);
char escaped[(oldLen*2)+1];
GetMysql2Client(self, client); GetMysql2Client(self, client);
newStr = rb_str_new(escaped, mysql_real_escape_string(client, escaped, RSTRING_PTR(str), RSTRING_LEN(str))); newLen = mysql_real_escape_string(client, escaped, RSTRING_PTR(str), RSTRING_LEN(str));
if (newLen == oldLen) {
// no need to return a new ruby string if nothing changed
return str;
} else {
newStr = rb_str_new(escaped, newLen);
#ifdef HAVE_RUBY_ENCODING_H #ifdef HAVE_RUBY_ENCODING_H
rb_enc_associate_index(newStr, utf8Encoding); rb_enc_associate_index(newStr, utf8Encoding);
#endif #endif
return newStr; return newStr;
}
} }
/* Mysql2::Result */ /* Mysql2::Result */