areadlinkat-with-size: Don't return an excessive memory allocation.

* lib/areadlinkat-with-size.c (areadlinkat_with_size): Shrink the buffer
before returning it.

1 files changed, 7 insertions, 0 deletions

diff --git a/lib/areadlinkat-with-size.c b/lib/areadlinkat-with-size.c
index ed00b98e4d..5b2bccc4a5 100644
--- a/lib/areadlinkat-with-size.c
+++ b/lib/areadlinkat-with-size.c
@@ -92,6 +92,13 @@ areadlinkat_with_size (int fd, char const *file, size_t size)
       if (link_length < buf_size)
         {
           buffer[link_length] = 0;
+          /* Shrink BUFFER before returning it.  */
+          if (link_length + 1 < buf_size)
+            {
+              char *shrinked_buffer = realloc (buffer, link_length + 1);
+              if (shrinked_buffer != NULL)
+                buffer = shrinked_buffer;
+            }
           return buffer;
         }