From 04eb92153f04930b348f85120255c5715ec55b6f Mon Sep 17 00:00:00 2001
From: Bruce Hill <bruce@bruce-hill.com>
Date: Thu, 23 Sep 2021 17:31:21 -0700
Subject: WIP

---
 files.c | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'files.c')

diff --git a/files.c b/files.c
index d391eaf..3e90f8c 100644
--- a/files.c
+++ b/files.c
@@ -36,7 +36,7 @@ static void populate_lines(file_t *f)
             f->lines = grow(f->lines, linecap *= 2);
         f->lines[n] = p;
         do {
-            char *nl = strchr(p, '\n');
+            char *nl = memchr(p, '\n', (size_t)(f->end - p));
             if (nl) {
                 p = nl+1;
                 break;
@@ -234,7 +234,7 @@ void fprint_line(FILE *dest, file_t *f, const char *start, const char *end, cons
     va_end(args);
     (void)fputc('\n', dest);
 
-    const char *eol = linenum == f->nlines ? strchr(line, '\0') : strchr(line, '\n');
+    const char *eol = linenum == f->nlines ? f->end : strchr(line, '\n');
     if (end == NULL || end > eol) end = eol;
     fprintf(dest, "\033[2m%5lu\033(0\x78\033(B\033[m%.*s\033[41;30m%.*s\033[m%.*s\n",
             linenum,
-- 
cgit v1.2.3