X-Git-Url: https://diplodocus.org/git/nmh/blobdiff_plain/a59946091a8ee121ad89708efb0b719e0d1e179a..7ebe4fe7bb8bf17b4e1058f687efbe9535d11df7:/sbr/fmt_scan.c

diff --git a/sbr/fmt_scan.c b/sbr/fmt_scan.c
index ac4d23af..0d9e1e90 100644
--- a/sbr/fmt_scan.c
+++ b/sbr/fmt_scan.c
@@ -118,7 +118,8 @@ cpnumber(char **dest, int num, unsigned int wid, char fill, size_t n) {
  * no more than n bytes are copied
  */
 static void
-cptrimmed(char **dest, char *str, unsigned int wid, char fill, size_t n) {
+cptrimmed(char **dest, char *str, unsigned int wid, char fill, size_t n,
+	  size_t max) {
     int remaining;     /* remaining output width available */
     int c, ljust;
     int end;           /* number of input bytes remaining in str */
@@ -129,7 +130,8 @@ cptrimmed(char **dest, char *str, unsigned int wid, char fill, size_t n) {
 #endif
     char *sp;          /* current position in source string */
     char *cp = *dest;  /* current position in destination string */
-    char *ep = cp + n; /* end of destination buffer */
+    char *ep = cp + n; /* end of destination buffer based on desired width */
+    char *epmax = cp + max; /* true end of destination buffer */
     int prevCtrl = 1;
 
     /* get alignment */
@@ -144,6 +146,13 @@ cptrimmed(char **dest, char *str, unsigned int wid, char fill, size_t n) {
 	while (*sp && remaining > 0 && end > 0) {
 #ifdef MULTIBYTE_SUPPORT
 	    char_len = mbtowc(&wide_char, sp, end);
+
+	    /* Account for multibyte characters taking only one character's
+	       width of output. */
+	    if (char_len > 1  &&  epmax - ep >= char_len - 1) {
+		ep += char_len - 1;
+	    }
+
 	    if (char_len <= 0 || (cp + char_len > ep))
 		break;
 
@@ -208,7 +217,7 @@ cptrimmed(char **dest, char *str, unsigned int wid, char fill, size_t n) {
 }
 
 static void
-cpstripped (char **dest, char *end, char *str)
+cpstripped (char **dest, char *end, char *max, char *str)
 {
     int prevCtrl = 1;	/* This is 1 so we strip out leading spaces */
     int len;
@@ -235,6 +244,12 @@ cpstripped (char **dest, char *end, char *str)
 #ifdef MULTIBYTE_SUPPORT
     	char_len = mbtowc(&wide_char, str, len);
 
+	/* Account for multibyte characters taking only one character's
+	   width of output. */
+	if (char_len > 1  &&  max - end >= char_len - 1) {
+	    end += char_len - 1;
+	}
+
 	if (char_len <= 0 || *dest + char_len > end)
 	    break;
 
@@ -373,10 +388,11 @@ fmt_scan (struct format *format, char *scanl, size_t max, int width, int *dat)
 	switch (fmt->f_type) {
 
 	case FT_COMP:
-	    cpstripped (&cp, ep, fmt->f_comp->c_text);
+	    cpstripped (&cp, ep, scanl + max - 1, fmt->f_comp->c_text);
 	    break;
 	case FT_COMPF:
-	    cptrimmed (&cp, fmt->f_comp->c_text, fmt->f_width, fmt->f_fill, ep - cp);
+	    cptrimmed (&cp, fmt->f_comp->c_text, fmt->f_width, fmt->f_fill,
+		       ep - cp, scanl - cp + max - 1);
 	    break;
 
 	case FT_LIT:
@@ -399,10 +415,11 @@ fmt_scan (struct format *format, char *scanl, size_t max, int width, int *dat)
 	    break;
 
 	case FT_STR:
-	    cpstripped (&cp, ep, str);
+	    cpstripped (&cp, ep, scanl + max - 1, str);
 	    break;
 	case FT_STRF:
-	    cptrimmed (&cp, str, fmt->f_width, fmt->f_fill, ep - cp);
+	    cptrimmed (&cp, str, fmt->f_width, fmt->f_fill, ep - cp,
+		       scanl - cp + max - 1);
 	    break;
 	case FT_STRLIT:
 	    sp = str;
@@ -926,7 +943,7 @@ fmt_scan (struct format *format, char *scanl, size_t max, int width, int *dat)
 			*cp++ = ' ';
 		}
 	    }
-	    cpstripped (&cp, ep, lp);
+	    cpstripped (&cp, ep, scanl + max - 1, lp);
 	    }
 	    break;