Copy dtmhost string on call to TuneToDtm.

kvap · kvap · commit 57f620d4253c · 2015-10-07T14:32:45.000+03:00
diff --git a/contrib/pg_dtm/libdtm.c b/contrib/pg_dtm/libdtm.c
@@ -231,11 +231,15 @@ static bool dtm_query(DTMConn dtm, char cmd, int argc, ...)
 	return true;
 }
 
-static char *dtmhost = "127.0.0.1";
-static int dtmport = 5431;
+static char *dtmhost = NULL;
+static int dtmport = 0;
 
 void TuneToDtm(char *host, int port) {
-	dtmhost = host;
+	if (dtmhost) {
+		free(dtmhost);
+		dtmhost = NULL;
+	}
+	dtmhost = strdup(host);
 	dtmport = port;
 }
 
@@ -244,10 +248,15 @@ static DTMConn GetConnection()
 	static DTMConn dtm = NULL;
 	if (dtm == NULL)
 	{
-		dtm = DtmConnect(dtmhost, dtmport);
-        if (dtm == NULL) { 
-            elog(ERROR, "Failed to connect to DTMD");
-        }
+		if (dtmhost) {
+			dtm = DtmConnect(dtmhost, dtmport);
+			if (dtm == NULL)
+			{
+				elog(ERROR, "Failed to connect to DTMD %s:%d", dtmhost, dtmport);
+			}
+		} else {
+			elog(ERROR, "DTMD address not specified");
+		}
 	}
 	return dtm;
 }
