You've already forked mariadb-connector-c
mirror of
https://github.com/mariadb-corporation/mariadb-connector-c.git
synced 2025-08-07 02:42:49 +03:00
Fix for CONC155: return trailing zero when fetching from binary columns
into string
This commit is contained in:
@@ -813,9 +813,19 @@ void ps_fetch_bin(MYSQL_BIND *r_param, const MYSQL_FIELD *field,
|
||||
ulong field_length= net_field_length(row);
|
||||
size_t copylen;
|
||||
|
||||
/* Bug conc-155: For text columns we need to store terminating zero character */
|
||||
if (!(field->flags & BINARY_FLAG) && r_param->buffer_type == MYSQL_TYPE_STRING)
|
||||
field_length++;
|
||||
|
||||
copylen= MIN(field_length, r_param->buffer_length);
|
||||
memcpy(r_param->buffer, *row, copylen);
|
||||
*r_param->error= copylen < field_length;
|
||||
|
||||
/* don't count trailing zero if we fetch into string */
|
||||
if (r_param->buffer_type == MYSQL_TYPE_STRING &&
|
||||
!*r_param->error)
|
||||
field_length--;
|
||||
|
||||
*r_param->length= field_length;
|
||||
|
||||
(*row) += field_length;
|
||||
|
@@ -4012,7 +4012,52 @@ static int test_conc154(MYSQL *mysql)
|
||||
return OK;
|
||||
}
|
||||
|
||||
static int test_conc155(MYSQL *mysql)
|
||||
{
|
||||
MYSQL_STMT *stmt;
|
||||
MYSQL_BIND bind;
|
||||
char buffer[50];
|
||||
int rc;
|
||||
|
||||
rc= mysql_query(mysql, "DROP TABLE IF EXISTS t1");
|
||||
check_mysql_rc(rc, mysql);
|
||||
rc= mysql_query(mysql, "CREATE TABLE t1 (a TEXT)");
|
||||
check_mysql_rc(rc, mysql);
|
||||
rc= mysql_query(mysql, "INSERT INTO t1 VALUES ('zero terminated string')");
|
||||
check_mysql_rc(rc, mysql);
|
||||
|
||||
stmt= mysql_stmt_init(mysql);
|
||||
rc= mysql_stmt_prepare(stmt, "SELECT a FROM t1", strlen("SELECT a FROM t1"));
|
||||
check_stmt_rc(rc, stmt);
|
||||
|
||||
rc= mysql_stmt_execute(stmt);
|
||||
check_stmt_rc(rc, stmt);
|
||||
|
||||
memset(buffer, 'X', 50);
|
||||
memset(&bind, 0, sizeof(MYSQL_BIND));
|
||||
|
||||
bind.buffer= buffer;
|
||||
bind.buffer_length= 50;
|
||||
bind.buffer_type= MYSQL_TYPE_STRING;
|
||||
|
||||
rc= mysql_stmt_bind_result(stmt, &bind);
|
||||
check_stmt_rc(rc, stmt);
|
||||
|
||||
rc= mysql_stmt_fetch(stmt);
|
||||
check_stmt_rc(rc, stmt);
|
||||
|
||||
if (strlen(buffer) != strlen("zero terminated string"))
|
||||
{
|
||||
diag("Wrong buffer length");
|
||||
return FAIL;
|
||||
}
|
||||
|
||||
mysql_stmt_close(stmt);
|
||||
return OK;
|
||||
}
|
||||
|
||||
struct my_tests_st my_tests[] = {
|
||||
{"test_conc155", test_conc155, TEST_CONNECTION_DEFAULT, 0, NULL, NULL},
|
||||
{"test_conc154", test_conc154, TEST_CONNECTION_DEFAULT, 0, NULL , NULL},
|
||||
{"test_conc141", test_conc141, TEST_CONNECTION_NEW, 0, NULL , NULL},
|
||||
{"test_conc67", test_conc67, TEST_CONNECTION_DEFAULT, 0, NULL , NULL},
|
||||
@@ -4083,3 +4128,4 @@ int main(int argc, char **argv)
|
||||
|
||||
return(exit_status());
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user