From f75f70ecc3a72e771bee6e1a73944836d0883c53 Mon Sep 17 00:00:00 2001
From: Bibiko <bibiko@eva.mpg.de>
Date: Thu, 11 Jun 2009 18:41:49 +0000
Subject: =?UTF-8?q?=E2=80=A2=20fix=20to=20allow=20insert=20large=20blob=20?=
 =?UTF-8?q?data=20If=20a=20query=20gave=20the=20error=202006=20(server=20h?=
 =?UTF-8?q?as=20gone=20away)=20try=20to=20increase=20max=5Fallowed=5Fpacke?=
 =?UTF-8?q?t=20-=20if=20the=20user=20has=20SUPER=20privileges=20-=20and=20?=
 =?UTF-8?q?redo=20the=20query.=20If=20max=5Fallowed=5Fpacket=20was=20chang?=
 =?UTF-8?q?ed=20reset=20it=20to=20default=20after=20executing.=20In=20addi?=
 =?UTF-8?q?tion=20if=20error=202006=20occurred=20and=20max=5Fallowed=5Fpac?=
 =?UTF-8?q?ket=20can't=20be=20set=20write=20an=20additional=20error=20mess?=
 =?UTF-8?q?age=20to=20the=20log=20"(Please=20check=20if=20query=20size=20<?=
 =?UTF-8?q?=20max=5Fallowed=5Fpacket)".=20-=2010MB=20of=20data=20are=20no?=
 =?UTF-8?q?=20problem=20-=20while=20trying=20to=20insert=2067MB=20MySQL=20?=
 =?UTF-8?q?gave=20the=20error=20:=20...=20X'(null)'=20plus=20malloc:=20mma?=
 =?UTF-8?q?p(size=3D141287424)=20failed=20(error=20code=3D12);=20can't=20a?=
 =?UTF-8?q?llocate=20region?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Source/CMMCPConnection.h |  2 ++
 Source/CMMCPConnection.m | 56 ++++++++++++++++++++++++++++++------------------
 2 files changed, 37 insertions(+), 21 deletions(-)

diff --git a/Source/CMMCPConnection.h b/Source/CMMCPConnection.h
index 4b0330ec..2f2552e4 100644
--- a/Source/CMMCPConnection.h
+++ b/Source/CMMCPConnection.h
@@ -57,6 +57,8 @@
 	BOOL useKeepAlive;
 	float keepAliveInterval;
 	
+	BOOL isMaxAllowedPacketEditable;
+	
 	NSString *serverVersionString;
 	
 	NSStringEncoding workerStringEncoding;
diff --git a/Source/CMMCPConnection.m b/Source/CMMCPConnection.m
index 540cfa8c..c0eadc5e 100644
--- a/Source/CMMCPConnection.m
+++ b/Source/CMMCPConnection.m
@@ -246,6 +246,9 @@ static void forcePingTimeout(int signalNumber);
 	mConnected = YES;
 	mEncoding = [MCPConnection encodingForMySQLEncoding:mysql_character_set_name(mConnection)];
 	[self timeZone]; // Getting the timezone used by the server.
+	
+	isMaxAllowedPacketEditable = [self isMaxAllowedPacketEditable];
+	
 	return mConnected;
 }
 
@@ -638,6 +641,8 @@ static void forcePingTimeout(int signalNumber);
 	NSDate		*queryStartDate;
 	NSThread	*queryThread;
 	BOOL connectionHasTimedOut = NO, connectionChecked = NO;
+	
+	int currentMaxAllowedPacket = -1;
 
 	// If no connection is present, return nil.
 	if (!mConnected) return nil;
@@ -677,6 +682,12 @@ static void forcePingTimeout(int signalNumber);
 	// If there was an error, check whether it was a connection-related error
 	if (connectionHasTimedOut || (0 != workerQueryResultCode && [CMMCPConnection isErrorNumberConnectionError:[self getLastErrorID]])) {
 
+		// Try to increase max_allowed_packet for error 2006
+		if(isMaxAllowedPacketEditable && !connectionHasTimedOut && workerQueryResultCode == 1 && [self getLastErrorID] == 2006) {
+			currentMaxAllowedPacket = [self getMaxAllowedPacket];
+			[self setMaxAllowedPacketTo:strlen([self cStringFromString:query])+1024];
+		}
+
 		// Check the connection and see if it can be restored.  This triggers reconnects as necessary, and
 		// should only return false if a disconnection has been requested - in which case return nil.
 		if (![self checkConnection]) return nil;
@@ -691,6 +702,10 @@ static void forcePingTimeout(int signalNumber);
 		[queryThread release];
 	}
 
+	// If max_allowed_packet was changes reset it to default
+	if(currentMaxAllowedPacket > -1)
+		[self setMaxAllowedPacketTo:currentMaxAllowedPacket];
+
 	// Retrieve the result or error appropriately.
 	if (0 == workerQueryResultCode) {
 		if (mysql_field_count(mConnection) != 0) {
@@ -704,12 +719,16 @@ static void forcePingTimeout(int signalNumber);
 
 		// Inform the delegate about errors
 		if (delegate && [delegate respondsToSelector:@selector(queryGaveError:)]) {
-			[delegate queryGaveError:[self getLastErrorMessage]];
+			if(workerQueryResultCode == 1 && [self getLastErrorID] == 2006)
+				// very likely that query length > max_allowed_packet 
+				[delegate queryGaveError:[NSString stringWithFormat:@"%@ (Please check if query size < max_allowed_packet)", [self getLastErrorMessage]]];
+			else
+				[delegate queryGaveError:[self getLastErrorMessage]];
 		}
 
 		return nil;
 	}
-	
+
 	[self startKeepAliveTimerResettingState:YES];
 
 	return [theResult autorelease];
@@ -1048,17 +1067,18 @@ static void forcePingTimeout(int signalNumber)
  */
 - (int) getMaxAllowedPacket
 {
-	CMMCPResult * r;
-	r = [self queryString:@"SELECT @@global.max_allowed_packet" usingEncoding:mEncoding];
-	if (![[self getLastErrorMessage] isEqualToString:@""]) {
-		if ([self isConnected])
-			NSRunAlertPanel(@"Error", [NSString stringWithFormat:@"An error occured while retrieving max_allowed_packet size:\n\n%@", [self getLastErrorMessage]], @"OK", nil, nil);
-		return -1;
+
+	if (0 == mysql_query(mConnection, "SELECT @@global.max_allowed_packet")) {
+		if (mysql_field_count(mConnection) != 0) {
+			CMMCPResult *r = [[CMMCPResult alloc] initWithMySQLPtr:mConnection encoding:mEncoding timeZone:mTimeZone];
+			NSArray *a = [r fetchRowAsArray];
+			[r autorelease];
+			if([a count]) return [[a objectAtIndex:0] intValue];
+		}
 	}
-	NSArray *a = [r fetchRowAsArray];
-	if([a count])
-		return [[a objectAtIndex:0] intValue];
-	
+
+	NSRunAlertPanel(@"Error", [NSString stringWithFormat:@"An error occured while retrieving max_allowed_packet size:\n\n%@", [self getLastErrorMessage]], @"OK", nil, nil);
+
 	return -1;
 }
 
@@ -1070,14 +1090,9 @@ static void forcePingTimeout(int signalNumber)
  */
 - (int) setMaxAllowedPacketTo:(int)newSize
 {
-	
-	if(![self isMaxAllowedPacketEditable]) return [self getMaxAllowedPacket];
+	if(![self isMaxAllowedPacketEditable] || newSize < 1024) return [self getMaxAllowedPacket];
 
-	[self queryString:[NSString stringWithFormat:@"SET GLOBAL max_allowed_packet = %d", newSize] usingEncoding:mEncoding];
-	if (![[self getLastErrorMessage] isEqualToString:@""]) {
-		if ([self isConnected])
-			NSRunAlertPanel(@"Error", [NSString stringWithFormat:@"An error occured while setting max_allowed_packet size:\n\n%@", [self getLastErrorMessage]], @"OK", nil, nil);
-	}
+	mysql_query(mConnection, [[NSString stringWithFormat:@"SET GLOBAL max_allowed_packet = %d", newSize] UTF8String]);
 
 	return [self getMaxAllowedPacket];
 }
@@ -1088,8 +1103,7 @@ static void forcePingTimeout(int signalNumber)
  */
 - (BOOL) isMaxAllowedPacketEditable
 {
-	[self queryString:[NSString stringWithFormat:@"SET GLOBAL max_allowed_packet = %d", [self getMaxAllowedPacket]] usingEncoding:mEncoding];
-	return ([[self getLastErrorMessage] isEqualToString:@""]);
+	return(!mysql_query(mConnection, "SET GLOBAL max_allowed_packet = @@global.max_allowed_packet"));
 }
 
 
-- 
cgit v1.2.3