andreacavalli/netty5
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[#4265] Not allow to add/set DefaultHttpHeaders to itself.

Browse Source 
Motivation:

We should prevent to add/set DefaultHttpHeaders to itself to prevent unexpected side-effects.

Modifications:

Throw IllegalArgumentException if user tries to pass the same instance to set/add.

Result:

No surprising side-effects.
This commit is contained in:
Norman Maurer 2015-09-24 08:33:51 +02:00
parent f96777312d
commit d4079d1403
2 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
codec-http/src/main/java/io/netty/handler/codec/http/DefaultHttpHeaders.java
View File

@ -59,6 +59,9 @@ public class DefaultHttpHeaders extends HttpHeaders {
@Override
public HttpHeaders add(HttpHeaders headers) {
if (headers instanceof DefaultHttpHeaders) {
if (headers == this) {
throw new IllegalArgumentException("can't add to itself.");
}
DefaultHttpHeaders defaultHttpHeaders = (DefaultHttpHeaders) headers;
HeaderEntry e = defaultHttpHeaders.head.after;
while (e != defaultHttpHeaders.head) {
@ -74,6 +77,9 @@ public class DefaultHttpHeaders extends HttpHeaders {
@Override
public HttpHeaders set(HttpHeaders headers) {
if (headers instanceof DefaultHttpHeaders) {
if (headers == this) {
throw new IllegalArgumentException("can't add to itself.");
}
clear();
DefaultHttpHeaders defaultHttpHeaders = (DefaultHttpHeaders) headers;
HeaderEntry e = defaultHttpHeaders.head.after;

12
codec-http/src/test/java/io/netty/handler/codec/http/HttpHeadersTest.java
View File

@ -67,4 +67,16 @@ public class HttpHeadersTest {
HttpHeaders headers = new DefaultHttpHeaders(false);
headers.set("test", (CharSequence) null);
}
@Test(expected = IllegalArgumentException.class)
public void testAddSelf() {
HttpHeaders headers = new DefaultHttpHeaders(false);
headers.add(headers);
}
@Test(expected = IllegalArgumentException.class)
public void testSetSelf() {
HttpHeaders headers = new DefaultHttpHeaders(false);
headers.set(headers);
}
}