Repair badly broken estimation of output buffer size in lquery_out().

tglsfdc · tglsfdc · commit a7e383d24158 · 2003-12-17T20:15:41.000Z
diff --git a/contrib/ltree/ltree_io.c b/contrib/ltree/ltree_io.c
@@ -498,22 +498,21 @@ lquery_out(PG_FUNCTION_ARGS)
 			   *ptr;
 	int			i,
 				j,
-				totallen = 0;
+				totallen = 1;
 	lquery_level *curqlevel;
 	lquery_variant *curtlevel;
 
 	curqlevel = LQUERY_FIRST(in);
 	for (i = 0; i < in->numlevel; i++)
 	{
+		totallen++;
 		if (curqlevel->numvar)
-			totallen = (curqlevel->numvar * 4) + 1 + curqlevel->totallen;
+			totallen += 1 + (curqlevel->numvar * 4) + curqlevel->totallen;
 		else
-			totallen = 2 * 11 + 4;
-		totallen++;
+			totallen += 2 * 11 + 4;
 		curqlevel = LQL_NEXT(curqlevel);
 	}
 
-
 	ptr = buf = (char *) palloc(totallen);
 	curqlevel = LQUERY_FIRST(in);
 	for (i = 0; i < in->numlevel; i++)
