about summary refs log tree commit diff
diff options
context:
space:
mode:
-rw-r--r--stdio/linewrap.c176
1 files changed, 122 insertions, 54 deletions
diff --git a/stdio/linewrap.c b/stdio/linewrap.c
index 7f6576f29b..e22640da0f 100644
--- a/stdio/linewrap.c
+++ b/stdio/linewrap.c
@@ -50,6 +50,32 @@ unwrap_stream (FILE *stream, struct line_wrap_data *d)
   stream->__io_funcs.__fileno = d->fileno;
   stream->__io_funcs.__seek = d->seek;
 }
+
+/* If WRAPPER_COOKIE points to a 0 pointer, then STREAM is assumed to be
+   wrapped, and will be unwrapped, storing the wrapper cookie into
+   WRAPPER_COOKIE.  Otherwise, nothing is done.  */
+static inline void
+ensure_unwrapped (FILE *stream, struct line_wrap_data **wrapper_cookie)
+{
+  if (*wrapper_cookie == 0)
+    {
+      *wrapper_cookie = stream->__cookie;
+      unwrap_stream (stream, *wrapper_cookie);
+    }
+}
+
+/* If WRAPPER_COOKIE points to a non-0 pointer, then STREAM is assumed to
+   *have been unwrapped with ensure_unwrapped, will be wrapped with
+   *WRAPPER_COOKIE, and *WRAPPER_COOKIE zeroed.  Otherwise, nothing is done. */
+static inline void
+ensure_wrapped (FILE *stream, struct line_wrap_data **wrapper_cookie)
+{
+  if (*wrapper_cookie)
+    {
+      wrap_stream (stream, *wrapper_cookie);
+      *wrapper_cookie = 0;
+    }
+}
 
 /* Cookie io functions that might get called on a wrapped stream.
    Must pass the original cookie to the original functions.  */
@@ -68,26 +94,26 @@ lwfileno (void *cookie)
   return (*d->fileno) (d->cookie);
 }
 
-/* This function is called when STREAM must be flushed.
-   C is EOF or a character to be appended to the buffer contents.  */
-void
-__line_wrap_output (FILE *stream, int c)
+/* Process STREAM's buffer so that line wrapping is done from POINT_OFFS to
+   the end of its buffer.  If WRAPPER_COOKIE is 0, and it's necessary to
+   flush some data, STREAM is unwrapped, and the line wrap stdio cookie
+   stored in WRAPPER_COOKIE; otherwise, stream is assumed to already be
+   unwrapped, and WRAPPER_COOKIE to point to the line wrap data.  Returns C
+   or EOF if C was output.  */
+static inline int
+lwupdate (FILE *stream, int c, struct line_wrap_data **wrapper_cookie)
 {
   char *buf, *nl;
   size_t len;
-
-  /* Extract our data and restore the stream's original cookie
-     and output function so writes we do really go out.  */
-  struct line_wrap_data *d = stream->__cookie;
-  unwrap_stream (stream, d);
+  struct line_wrap_data *d = *wrapper_cookie ?: stream->__cookie;
 
   /* Scan the buffer for newlines.  */
-  buf = stream->__buffer;
+  buf = stream->__buffer + d->point_offs;
   while ((buf < stream->__bufp || (c != EOF && c != '\n')) && !stream->__error)
     {
       size_t r;
 
-      if (d->point == 0 && d->lmargin != 0)
+      if (d->point_col == 0 && d->lmargin != 0)
 	{
 	  /* We are starting a new line.  Print spaces to the left margin.  */
 	  const size_t pad = d->lmargin;
@@ -106,16 +132,18 @@ __line_wrap_output (FILE *stream, int c)
 	      size_t i;
 	      char *olimit;
 
+	      ensure_unwrapped (stream, wrapper_cookie);
+
 	      len = stream->__bufp - buf;
 	      olimit = stream->__put_limit;
 	      stream->__bufp = stream->__put_limit = buf;
 	      for (i = 0; i < pad; ++i)
 		(*d->output) (stream, ' ');
 	      stream->__put_limit = olimit;
-	      memcpy (stream->__bufp, buf, len);
+	      memmove (stream->__bufp, buf, len);
 	      stream->__bufp += len;
 	    }
-	  d->point = pad;
+	  d->point_col = pad;
 	}
 
       len = stream->__bufp - buf;
@@ -125,12 +153,12 @@ __line_wrap_output (FILE *stream, int c)
 	{
 	  /* The buffer ends in a partial line.  */
 
-	  if (d->point + len + (c != EOF && c != '\n') <= d->rmargin)
+	  if (d->point_col + len + (c != EOF && c != '\n') <= d->rmargin)
 	    {
 	      /* The remaining buffer text is a partial line and fits
 		 within the maximum line width.  Advance point for the
 		 characters to be written and stop scanning.  */
-	      d->point += len;
+	      d->point_col += len;
 	      break;
 	    }
 	  else
@@ -138,11 +166,11 @@ __line_wrap_output (FILE *stream, int c)
 	       the end of the buffer.  */
 	    nl = stream->__bufp;
 	}
-      else if (d->point + (nl - buf) <= d->rmargin)
+      else if (d->point_col + (nl - buf) <= d->rmargin)
 	{
 	  /* The buffer contains a full line that fits within the maximum
 	     line width.  Reset point and scan the next line.  */
-	  d->point = 0;
+	  d->point_col = 0;
 	  buf = nl + 1;
 	  continue;
 	}
@@ -156,10 +184,10 @@ __line_wrap_output (FILE *stream, int c)
 	     newline and anything after it in the buffer.  */
 	  if (nl < stream->__bufp)
 	    {
-	      memcpy (buf + (r - d->point), nl, stream->__bufp - nl);
-	      stream->__bufp -= buf + (r - d->point) - nl;
+	      memmove (buf + (r - d->point_col), nl, stream->__bufp - nl);
+	      stream->__bufp -= buf + (r - d->point_col) - nl;
 	      /* Reset point for the next line and start scanning it.  */
-	      d->point = 0;
+	      d->point_col = 0;
 	      buf += r + 1; /* Skip full line plus \n. */
 	    }
 	  else
@@ -167,8 +195,8 @@ __line_wrap_output (FILE *stream, int c)
 	      /* The buffer ends with a partial line that is beyond the
 		 maximum line width.  Advance point for the characters
 		 written, and discard those past the max from the buffer.  */
-	      d->point += len;
-	      stream->__bufp -= d->point - r;
+	      d->point_col += len;
+	      stream->__bufp -= d->point_col - r;
 	      if (c != '\n')
 		/* Swallow the extra character too.  */
 		c = EOF;
@@ -184,7 +212,7 @@ __line_wrap_output (FILE *stream, int c)
 	  char *p, *nextline;
 	  int i;
 
-	  p = buf + (r + 1 - d->point);
+	  p = buf + (r + 1 - d->point_col);
 	  while (p >= buf && !isblank (*p))
 	    --p;
 	  nextline = p + 1;	/* This will begin the next line.  */
@@ -201,7 +229,7 @@ __line_wrap_output (FILE *stream, int c)
 	    {
 	      /* A single word that is greater than the maximum line width.
 		 Oh well.  Put it on an overlong line by itself.  */
-	      p = buf + (r + 1 - d->point);
+	      p = buf + (r + 1 - d->point_col);
 	      /* Find the end of the long word.  */
 	      do
 		++p;
@@ -209,7 +237,7 @@ __line_wrap_output (FILE *stream, int c)
 	      if (p == nl)
 		{
 		  /* It already ends a line.  No fussing required.  */
-		  d->point = 0;
+		  d->point_col = 0;
 		  buf = nl + 1;
 		  continue;
 		}
@@ -228,14 +256,17 @@ __line_wrap_output (FILE *stream, int c)
 	  if (nextline - (nl + 1) < d->wmargin)
 	    /* The margin needs more blanks than we removed.
 	       Output the first line so we can use the space.  */
-	    (*d->output) (stream, '\n');
+	    {
+	      ensure_unwrapped (stream, wrapper_cookie);
+	      (*d->output) (stream, '\n');
+	    }
 	  else
 	    /* We can fit the newline and blanks in before
 	       the next word.  */
 	    *stream->__bufp++ = '\n';
 
 	  /* Reset the counter of what has been output this line.  */
-	  d->point = 0;
+	  d->point_col = 0;
 
 	  /* Add blanks up to the wrap margin column.  */
 	  for (i = 0; i < d->wmargin; ++i)
@@ -244,7 +275,7 @@ __line_wrap_output (FILE *stream, int c)
 	  /* Copy the tail of the original buffer into the current buffer
 	     position.  */
 	  if (stream->__bufp != nextline)
-	    memcpy (stream->__bufp, nextline, buf + len - nextline);
+	    memmove (stream->__bufp, nextline, buf + len - nextline);
 	  len -= nextline - buf;
 
 	  /* Continue the scan on the remaining lines in the buffer.  */
@@ -255,16 +286,33 @@ __line_wrap_output (FILE *stream, int c)
 	}
     }
 
+  /* Remember that we've scanned as far as the end of the buffer.  */
+  d->point_offs = stream->__bufp - stream->__buffer;
+
+  return c;
+}
+
+/* This function is called when STREAM must be flushed.
+   C is EOF or a character to be appended to the buffer contents.  */
+void
+__line_wrap_output (FILE *stream, int c)
+{
+  struct line_wrap_data *d = 0;
+
+  c = lwupdate (stream, c, &d);
+
   if (!stream->__error)
     {
+      ensure_unwrapped (stream, &d);
       (*d->output) (stream, c);
+      d->point_offs = 0;	/* The buffer now holds nothing.  */
       if (c == '\n')
-	d->point = 0;
+	d->point_col = 0;
       else if (c != EOF)
-	++d->point;
+	++d->point_col;
     }
 
-  wrap_stream (stream, d);
+  ensure_wrapped (stream, &d);
 }
 
 /* Modify STREAM so that it prefixes lines written on it with LMARGIN spaces
@@ -285,7 +333,8 @@ line_wrap_stream (FILE *stream, size_t lmargin, size_t rmargin, size_t wmargin)
   fflush (stream);
 
   /* Initialize our wrapping state.  */
-  d->point = 0;
+  d->point_col = 0;
+  d->point_offs = 0;
 
   /* Save the original cookie and output and close hooks.  */
   d->cookie = stream->__cookie;
@@ -326,8 +375,28 @@ line_wrapped (FILE *stream)
   return (stream->__room_funcs.__output == &__line_wrap_output);
 }
 
+/* If STREAM is not line-wrapped, return 0.  Otherwise all pending text
+   buffered text in STREAM so that the POINT_OFFS field refers to the last
+   position in the stdio buffer, and return the line wrap state object for
+   STREAM.  Since all text has been processed, this means that (1) the
+   POINT_COL field refers to the column at which any new text would be added,
+   and (2) any changes to the margin parameters will only affect new text.  */
+struct line_wrap_data *
+__line_wrap_update (FILE *stream)
+{
+  if (line_wrapped (stream))
+    {
+      struct line_wrap_data *d = stream->__cookie, *wc = 0;
+      lwupdate (stream, EOF, &wc);
+      ensure_wrapped (stream, &wc);
+      return d;
+    }
+  else
+    return 0;
+}
+
 /* If STREAM is not line-wrapped return -1, else return its left margin.  */
-size_t
+inline size_t
 line_wrap_lmargin (FILE *stream)
 {
   if (! line_wrapped (stream))
@@ -337,22 +406,22 @@ line_wrap_lmargin (FILE *stream)
 
 /* If STREAM is not line-wrapped return -1, else set its left margin to
    LMARGIN and return the old value.  */
-size_t
+inline size_t
 line_wrap_set_lmargin (FILE *stream, size_t lmargin)
 {
-  if (! line_wrapped (stream))
-    return -1;
-  else
+  struct line_wrap_data *d = __line_wrap_update (stream);
+  if (d)
     {
-      struct line_wrap_data *d = stream->__cookie;
       size_t old = d->lmargin;
       d->lmargin = lmargin;
       return old;
     }
+  else
+    return -1;
 }
 
 /* If STREAM is not line-wrapped return -1, else return its left margin.  */
-size_t
+inline size_t
 line_wrap_rmargin (FILE *stream)
 {
   if (! line_wrapped (stream))
@@ -362,22 +431,22 @@ line_wrap_rmargin (FILE *stream)
 
 /* If STREAM is not line-wrapped return -1, else set its right margin to
    RMARGIN and return the old value.  */
-size_t
+inline size_t
 line_wrap_set_rmargin (FILE *stream, size_t rmargin)
 {
-  if (! line_wrapped (stream))
-    return -1;
-  else
+  struct line_wrap_data *d = __line_wrap_update (stream);
+  if (d)
     {
-      struct line_wrap_data *d = stream->__cookie;
       size_t old = d->rmargin;
       d->rmargin = rmargin;
       return old;
     }
+  else
+    return -1;
 }
 
 /* If STREAM is not line-wrapped return -1, else return its wrap margin.  */
-size_t
+inline size_t
 line_wrap_wmargin (FILE *stream)
 {
   if (! line_wrapped (stream))
@@ -387,28 +456,27 @@ line_wrap_wmargin (FILE *stream)
 
 /* If STREAM is not line-wrapped return -1, else set its left margin to
    WMARGIN and return the old value.  */
-size_t
+inline size_t
 line_wrap_set_wmargin (FILE *stream, size_t wmargin)
 {
-  if (! line_wrapped (stream))
-    return -1;
-  else
+  struct line_wrap_data *d = __line_wrap_update (stream);
+  if (d)
     {
-      struct line_wrap_data *d = stream->__cookie;
       size_t old = d->wmargin;
       d->wmargin = wmargin;
       return old;
     }
+  else
+    return -1;
 }
 
 /* If STREAM is not line-wrapped return -1, else return the column number of
    the current output point.  */
-size_t
+inline size_t
 line_wrap_point (FILE *stream)
 {
-  if (! line_wrapped (stream))
-    return -1;
-  return ((struct line_wrap_data *)stream->__cookie)->point;
+  struct line_wrap_data *d = __line_wrap_update (stream);
+  return d ? d->point_col : -1;
 }
 
 #ifdef TEST