proxysql/deps/mariadb-client-library/libmariadb.c.patch

@@ -2039,10 +2039,16 @@
 
   mysql->client_flag= client_flag;
 
+	{
+		volatile my_bool net_blocking = vio_is_blocking(net->vio);
+		if (!net_blocking)
+			vio_blocking(net->vio, TRUE, 0);
   if (run_plugin_auth(mysql, scramble_data, scramble_len,
                              scramble_plugin, db))
     goto error;
-
+		if (!net_blocking)
+			vio_blocking(net->vio, FALSE, 0);
+	}
   if (mysql->client_flag & CLIENT_COMPRESS)
     net->compress= 1;
 
@@ -2210,36 +2216,39 @@
   else
     mysql->charset=default_charset_info;
 
-  mysql->user= (char *)user;
-  mysql->passwd= (char *)passwd;
-  mysql->db= (char *)db;
+	mysql->user= strdup(user ? user : "");
+  mysql->passwd= strdup(passwd ? passwd : "");
+  mysql->db= 0;
+
   rc= run_plugin_auth(mysql, 0, 0, 0, db);
 
   if (rc==0)
   {
-    LIST *li_stmt= mysql->stmts;
+    //LIST *li_stmt= mysql->stmts; // see bug #897
     my_free(s_user);
     my_free(s_passwd);
     my_free(s_db);
 
-    if (!(mysql->user=  my_strdup(user,MYF(MY_WME))) ||
-        !(mysql->passwd=my_strdup(passwd,MYF(MY_WME))) ||
-        !(mysql->db= db ? my_strdup(db,MYF(MY_WME)) : 0))
+		if (db && !(mysql->db= strdup(db)))
     {
       SET_CLIENT_ERROR(mysql, CR_OUT_OF_MEMORY, unknown_sqlstate, 0);
       rc= 1;
     }
 
-    for (;li_stmt;li_stmt= li_stmt->next)
-    {
-      MYSQL_STMT *stmt= (MYSQL_STMT *)li_stmt->data;
-      stmt->mysql= NULL;
-      SET_CLIENT_STMT_ERROR(stmt, CR_SERVER_LOST, SQLSTATE_UNKNOWN, 0);
-    }/* detach stmts */
+    //for (;li_stmt;li_stmt= li_stmt->next) // see bug #897
+    //{
+    //  MYSQL_STMT *stmt= (MYSQL_STMT *)li_stmt->data;
+    //  stmt->mysql= NULL;
+    //  SET_CLIENT_STMT_ERROR(stmt, CR_SERVER_LOST, SQLSTATE_UNKNOWN, 0);
+    //}/* detach stmts */
     mysql->stmts= NULL;
     
   } else
   {
+    free(mysql->user);
+    free(mysql->passwd);
+    free(mysql->db);
+
     mysql->user= s_user;
     mysql->passwd= s_passwd;
     mysql->db= s_db;
@@ -2396,6 +2405,36 @@
     bzero((char*) &mysql->options,sizeof(mysql->options));
     mysql->net.vio= 0;
     if (mysql->free_me)
+      my_free((gptr) mysql);
+  }
+  DBUG_VOID_RETURN;
+}
+
+void STDCALL
+mysql_close_no_command(MYSQL *mysql)
+{
+  //MYSQL_STMT *stmt;
+  DBUG_ENTER("mysql_close");
+  if (mysql)					/* Some simple safety */
+  {
+    //LIST *li_stmt= mysql->stmts;
+
+    if (mysql->methods) {
+			if (mysql->net.vio) {
+				free_old_query(mysql);
+				mysql->status=MYSQL_STATUS_READY; /* Force command */
+    		mysql->reconnect=0;
+				end_server(mysql);
+			}
+		}
+    mysql_close_memory(mysql);
+    mysql_close_options(mysql);
+    mysql->host_info=mysql->user=mysql->passwd=mysql->db=0;
+
+    /* Clear pointers for better safety */
+    bzero((char*) &mysql->options,sizeof(mysql->options));
+    mysql->net.vio= 0;
+    if (mysql->free_me)
       my_free(mysql);
   }
   DBUG_VOID_RETURN;
@@ -3510,19 +3549,27 @@
   DBUG_VOID_RETURN;
 }
 
-int STDCALL mysql_set_character_set(MYSQL *mysql, const char *csname)
+int STDCALL mysql_set_character_set(MYSQL *mysql, const char *csname, uint charsetnr)
 {
   const CHARSET_INFO *cs;
   DBUG_ENTER("mysql_set_character_set");
 
-  if (!csname)
+  if (!csname && !charsetnr)
     goto error;
 
-  if ((cs= mysql_find_charset_name(csname)))
-  {
-    char buff[64];
-
-    my_snprintf(buff, 63, "SET NAMES %s", cs->csname);
+  if (csname) {
+	cs = mysql_find_charset_name(csname);
+  } else {
+    cs = mysql_find_charset_nr(charsetnr); 
+  }
+  if (cs)
+  {
+    char buff[128];
+    if (csname) { // default behavior
+      my_snprintf(buff, 63, "SET NAMES %s", cs->csname);
+    } else {
+      my_snprintf(buff, 63, "SET NAMES %s COLLATE %s", cs->csname, cs->name);
+    }
     if (!mysql_real_query(mysql, buff, (uint)strlen(buff)))
     {
       mysql->charset= cs;