From: Arun Prakash Jana <engineerarun@gmail.com>
Date: Thu, 15 Jul 2021 18:52:27 +0000 (+0530)
Subject: Allocate PATH_MAX aligned memory size
X-Git-Tag: v4.2~21
X-Git-Url: http://www.git.stargrave.org/?a=commitdiff_plain;h=8af773eb0d9242272ae0db854305d71dafa09781;p=nnn.git

Allocate PATH_MAX aligned memory size
---

diff --git a/src/nnn.c b/src/nnn.c
index b0ab729d..499073e5 100644
--- a/src/nnn.c
+++ b/src/nnn.c
@@ -163,6 +163,7 @@
 #define ISBLANK(x)      ((x) == ' ' || (x) == '\t')
 #define TOUPPER(ch)     (((ch) >= 'a' && (ch) <= 'z') ? ((ch) - 'a' + 'A') : (ch))
 #define CMD_LEN_MAX     (PATH_MAX + ((NAME_MAX + 1) << 1))
+#define ALIGN_UP(x, A)  ((((x) + (A) - 1) / (A)) * (A))
 #define READLINE_MAX    256
 #define FILTER          '/'
 #define RFILTER         '\\'
@@ -1663,8 +1664,9 @@ static void invertselbuf(const int pathlen)
 	/* Buffer size adjustment */
 	selbufpos -= shrinklen;
 
-	if (alloclen > shrinklen) {
-		pselbuf = xrealloc(pselbuf, selbuflen + (alloclen - shrinklen));
+	if (alloclen > shrinklen && ((selbufpos + (alloclen - shrinklen)) > selbuflen)) {
+		selbuflen = ALIGN_UP(selbufpos + (alloclen - shrinklen), PATH_MAX);
+		pselbuf = xrealloc(pselbuf, selbuflen);
 		if (!pselbuf)
 			errexit();
 	}
@@ -1715,9 +1717,12 @@ static void addtoselbuf(const int pathlen, int startid, int endid)
 			alloclen += pathlen + dentp->nlen;
 	}
 
-	pselbuf = xrealloc(pselbuf, selbuflen + alloclen);
-	if (!pselbuf)
-		errexit();
+	if ((selbufpos + alloclen) > selbuflen) {
+		selbuflen = ALIGN_UP(selbufpos + alloclen, PATH_MAX);
+		pselbuf = xrealloc(pselbuf, selbuflen);
+		if (!pselbuf)
+			errexit();
+	}
 
 	for (int i = startid; i <= endid; ++i) {
 		if (!(pdents[i].flags & FILE_SELECTED)) {