1 files changed, 11 insertions, 5 deletions

diff --git a/usr.bin/top/display.c b/usr.bin/top/display.c
index 708855fb095f..539ddb4d536e 100644
--- a/usr.bin/top/display.c
+++ b/usr.bin/top/display.c
@@ -1347,7 +1347,8 @@ i_uptime(struct timeval *bt, time_t *tod)
 static char *
 setup_buffer(char *buffer, int addlen)
 {
-    size_t len;
+    size_t len, old_len;
+    char *new_buffer;
 
     setup_buffer_bufsiz = screen_width;
     if (setup_buffer_bufsiz < SETUPBUFFER_MIN_SCREENWIDTH)
@@ -1355,13 +1356,18 @@ setup_buffer(char *buffer, int addlen)
 	setup_buffer_bufsiz = SETUPBUFFER_MIN_SCREENWIDTH;
     }
 
-    free(buffer);
     len = setup_buffer_bufsiz + addlen + SETUPBUFFER_REQUIRED_ADDBUFSIZ;
-    buffer = calloc(len, sizeof(char));
-    if (buffer == NULL)
+    new_buffer = calloc(len, sizeof(char));
+    if (new_buffer == NULL)
     {
 	errx(4, "can't allocate sufficient memory");
     }
+    if (buffer != NULL)
+    {
+	old_len = strlen(buffer);
+	memcpy(new_buffer, buffer, old_len < len - 1 ? old_len : len - 1);
+	free(buffer);
+    }
 
-    return buffer;
+    return new_buffer;
 }