1 file changed, 34 insertions, 18 deletions
diff --git a/source/ui-ssdiff.c b/source/ui-ssdiff.c
index 750dc3d..7cea09e 100644
--- a/source/ui-ssdiff.c
+++ b/source/ui-ssdiff.c
@@ -91,22 +91,21 @@ static char *longest_common_subsequence(char *A, char *B)
 
 static int line_from_hunk(char *line, char type)
 {
-	char *buf1, *buf2;
-	int len, res;
+	char *p;
+	long res;
 
-	buf1 = strchr(line, type);
-	if (buf1 == NULL)
+	p = strchr(line, type);
+	if (p == NULL)
 		return 0;
-	buf1 += 1;
-	buf2 = strchr(buf1, ',');
-	if (buf2 == NULL)
+	p += 1;
+	// git omits the length when a hunk covers a single line, as in
+	// "@@ -1 +1 @@", so the number runs to whatever follows it rather than
+	// to a comma that may belong to the other side of the header or be
+	// missing altogether.
+	res = strtol(p, NULL, 10);
+	if (res < 0 || res > INT_MAX)
 		return 0;
-	len = buf2 - buf1;
-	buf2 = xmalloc(len + 1);
-	strlcpy(buf2, buf1, len + 1);
-	res = atoi(buf2);
-	free(buf2);
-	return res;
+	return (int)res;
 }
 
 static char *replace_tabs(char *line)
@@ -190,7 +189,24 @@ static void deferred_new_add(char *line, int line_no)
 	}
 }
 
-static void print_part_with_lcs(char *class, char *line, char *lcs)
+/* The item owns the copy of the line taken when it was deferred, so both go
+ * together. print_ssdiff_line only reads the line and frees what it derives
+ * from it, and never keeps the pointer it was handed. */
+static void free_deferred(struct deferred_lines *item)
+{
+	free(item->line);
+	free(item);
+}
+
+static void flush_run(struct strbuf *run)
+{
+	if (!run->len)
+		return;
+	html_txt(run->buf);
+	strbuf_reset(run);
+}
+
+static void print_part_with_lcs(const char *class, char *line, char *lcs)
 {
 	int line_len = strlen(line);
 	int i, j;
@@ -309,7 +325,7 @@ static void print_deferred_old_lines(void)
 		print_ssdiff_line("del", iter_old->line_no,
 				  iter_old->line, -1, NULL, 0);
 		tmp = iter_old->next;
-		free(iter_old);
+		free_deferred(iter_old);
 		iter_old = tmp;
 	}
 }
@@ -322,7 +338,7 @@ static void print_deferred_new_lines(void)
 		print_ssdiff_line("add", -1, NULL,
 				  iter_new->line_no, iter_new->line, 0);
 		tmp = iter_new->next;
-		free(iter_new);
+		free_deferred(iter_new);
 		iter_new = tmp;
 	}
 }
@@ -350,13 +366,13 @@ static void print_deferred_changed_lines(void)
 					  iter_new->line_no, iter_new->line, 0);
 		if (iter_old) {
 			tmp = iter_old->next;
-			free(iter_old);
+			free_deferred(iter_old);
 			iter_old = tmp;
 		}
 
 		if (iter_new) {
 			tmp = iter_new->next;
-			free(iter_new);
+			free_deferred(iter_new);
 			iter_new = tmp;
 		}
 	}