python/mysqlclient
1
0
Fork 0
mirror of https://github.com/PyMySQL/mysqlclient.git synced 2025-08-15 02:54:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use mysql_real_escape_string_quote() if exists.

Browse Source
This commit is contained in:
INADA Naoki
2016-09-05 01:16:12 +09:00
parent 05fbc500d1
commit a83baad1d0
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
_mysql.c
View File

@ -1063,10 +1063,15 @@ _mysql_escape_string(
if (self && PyModule_Check((PyObject*)self)) if (self && PyModule_Check((PyObject*)self))
self = NULL; self = NULL;
if (self && self->open) if (self && self->open) {
#if MYSQL_VERSION_ID >= 50706
len = mysql_real_escape_string_quote(&(self->connection), out, in, size, '\'');
#else
len = mysql_real_escape_string(&(self->connection), out, in, size); len = mysql_real_escape_string(&(self->connection), out, in, size);
else #endif
} else {
len = mysql_escape_string(out, in, size); len = mysql_escape_string(out, in, size);
}
if (_PyBytes_Resize(&str, len) < 0) return NULL; if (_PyBytes_Resize(&str, len) < 0) return NULL;
return (str); return (str);
} }
@ -1117,7 +1122,11 @@ _mysql_string_literal(
out = PyBytes_AS_STRING(str); out = PyBytes_AS_STRING(str);
check_server_init(NULL); check_server_init(NULL);
if (self && self->open) { if (self && self->open) {
#if MYSQL_VERSION_ID >= 50706
len = mysql_real_escape_string_quote(&(self->connection), out+1, in, size, '\'');
#else
len = mysql_real_escape_string(&(self->connection), out+1, in, size); len = mysql_real_escape_string(&(self->connection), out+1, in, size);
#endif
} else { } else {
len = mysql_escape_string(out+1, in, size); len = mysql_escape_string(out+1, in, size);
} }