From d00f6f09a522ba6d13b2a341e317249bbff117ec Mon Sep 17 00:00:00 2001
From: Brian Clozel <bclozel@pivotal.io>
Date: Mon, 9 Jul 2018 19:15:29 +0200
Subject: [PATCH] Polish ContentCachingRequestWrapper

Issue: SPR-15762
---
 .../util/ContentCachingRequestWrapper.java    | 71 +++++++------------
 1 file changed, 27 insertions(+), 44 deletions(-)

diff --git a/spring-web/src/main/java/org/springframework/web/util/ContentCachingRequestWrapper.java b/spring-web/src/main/java/org/springframework/web/util/ContentCachingRequestWrapper.java
index fc8ec32f32..441155c9b1 100644
--- a/spring-web/src/main/java/org/springframework/web/util/ContentCachingRequestWrapper.java
+++ b/spring-web/src/main/java/org/springframework/web/util/ContentCachingRequestWrapper.java
@@ -1,5 +1,5 @@
 /*
- * Copyright 2002-2017 the original author or authors.
+ * Copyright 2002-2018 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -214,69 +214,52 @@ public class ContentCachingRequestWrapper extends HttpServletRequestWrapper {
 		}
 
 		@Override
-		public int readLine(final byte[] b, final int off, final int len) throws IOException {
-			int count = is.readLine(b, off, len);
-			cache(b, off, count);
-			return count;
-		}
-
-		private void cache(final byte[] b, final int off, final int count) {
-			if (contentCacheLimit != null && cachedContent.size() == contentCacheLimit) {
-				this.overflow = true;
-				handleContentOverflow(contentCacheLimit);
-			} else {
-				int sizeToCache = contentCacheLimit == null || count + cachedContent.size() < contentCacheLimit
-						? count
-						: contentCacheLimit - cachedContent.size();
-				cachedContent.write(b, off, sizeToCache);
-				if (sizeToCache < count) {
+		public int read() throws IOException {
+			int ch = this.is.read();
+			if (ch != -1 && !this.overflow) {
+				if (contentCacheLimit != null && cachedContent.size() == contentCacheLimit) {
 					this.overflow = true;
 					handleContentOverflow(contentCacheLimit);
 				}
+				else {
+					cachedContent.write(ch);
+				}
 			}
+			return ch;
 		}
 
 		@Override
-		public int read(final byte[] b) throws IOException {
-			int count = super.read(b);
+		public int read(byte[] b) throws IOException {
+			int count = this.is.read(b);
+			writeToCache(b, 0, count);
+			return count;
+		}
+
+		private void writeToCache(final byte[] b, final int off, int count) {
 			if (!this.overflow && count > 0) {
-				if (contentCacheLimit != null && cachedContent.size() == contentCacheLimit) {
+				if (contentCacheLimit != null &&
+						count + cachedContent.size() > contentCacheLimit) {
 					this.overflow = true;
+					cachedContent.write(b, off, contentCacheLimit - cachedContent.size());
 					handleContentOverflow(contentCacheLimit);
-				} else {
-                    int sizeToCache = contentCacheLimit == null || count + cachedContent.size() < contentCacheLimit
-                            ? count
-                            : contentCacheLimit - cachedContent.size();
-                    cachedContent.write(b, 0, sizeToCache);
-                    if (sizeToCache < count) {
-                    	this.overflow = true;
-                    	handleContentOverflow(contentCacheLimit);
-					}
+					return;
 				}
+				cachedContent.write(b, off, count);
 			}
-			return count;
 		}
 
 		@Override
 		public int read(final byte[] b, final int off, final int len) throws IOException {
-			int count = is.read(b, off, len);
-			cache(b, off, count);
+			int count = this.is.read(b, off, len);
+			writeToCache(b, off, count);
 			return count;
 		}
 
 		@Override
-		public int read() throws IOException {
-			int ch = this.is.read();
-			if (ch != -1 && !this.overflow) {
-				if (contentCacheLimit != null && cachedContent.size() == contentCacheLimit) {
-					this.overflow = true;
-					handleContentOverflow(contentCacheLimit);
-				}
-				else {
-					cachedContent.write(ch);
-				}
-			}
-			return ch;
+		public int readLine(final byte[] b, final int off, final int len) throws IOException {
+			int count = this.is.readLine(b, off, len);
+			writeToCache(b, off, count);
+			return count;
 		}
 
 		@Override