From: Jonas Fonseca <fonseca@diku.dk>
Date: Mon, 11 Sep 2006 01:12:15 +0000 (+0200)
Subject: Improve string buffer copy safety
X-Git-Tag: tig-0.5~37
X-Git-Url: https://git.distorted.org.uk/~mdw/tig/commitdiff_plain/9a48919bfae4a744d0f852d834aa93e71293e8f0

Improve string buffer copy safety
---

diff --git a/tig.c b/tig.c
index d3702c4..44fcb1b 100644
--- a/tig.c
+++ b/tig.c
@@ -145,16 +145,22 @@ set_from_int_map(struct int_map *map, size_t map_size,
  */
 
 static inline void
-string_ncopy(char *dst, const char *src, size_t dstlen)
+string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
 {
-	strncpy(dst, src, dstlen - 1);
-	dst[dstlen - 1] = 0;
+	if (srclen > dstlen - 1)
+		srclen = dstlen - 1;
 
+	strncpy(dst, src, srclen);
+	dst[srclen] = 0;
 }
 
-/* Shorthand for safely copying into a fixed buffer. */
+/* Shorthands for safely copying into a fixed buffer. */
+
 #define string_copy(dst, src) \
-	string_ncopy(dst, src, sizeof(dst))
+	string_ncopy_do(dst, sizeof(dst), src, sizeof(dst))
+
+#define string_ncopy(dst, src, srclen) \
+	string_ncopy_do(dst, sizeof(dst), src, srclen)
 
 static char *
 chomp_string(char *name)
@@ -2204,8 +2210,7 @@ pager_draw(struct view *view, struct line *line, unsigned int lineno)
 			string_copy(ref_commit, view->ref);
 
 		} else if (type == LINE_TREE_DIR || type == LINE_TREE_FILE) {
-			strncpy(view->ref, text + STRING_SIZE("100644 blob "), 41);
-			view->ref[40] = 0;
+			string_ncopy(view->ref, text + STRING_SIZE("100644 blob "), 40);
 			string_copy(ref_blob, view->ref);
 		}