95 lines
3.4 KiB
Java
95 lines
3.4 KiB
Java
/* ====================================================================
|
|
Licensed to the Apache Software Foundation (ASF) under one or more
|
|
contributor license agreements. See the NOTICE file distributed with
|
|
this work for additional information regarding copyright ownership.
|
|
The ASF licenses this file to You under the Apache License, Version 2.0
|
|
(the "License"); you may not use this file except in compliance with
|
|
the License. You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
==================================================================== */
|
|
|
|
package org.apache.poi.xssf.util;
|
|
|
|
import static org.junit.Assert.assertArrayEquals;
|
|
|
|
import java.io.ByteArrayInputStream;
|
|
import java.io.ByteArrayOutputStream;
|
|
import java.io.IOException;
|
|
import java.nio.charset.Charset;
|
|
|
|
import org.apache.poi.util.IOUtils;
|
|
import org.apache.poi.util.ReplacingInputStream;
|
|
import org.junit.Test;
|
|
|
|
public final class TestEvilUnclosedBRFixingInputStream {
|
|
|
|
static class EvilUnclosedBRFixingInputStream extends ReplacingInputStream {
|
|
public EvilUnclosedBRFixingInputStream(byte[] source) {
|
|
super(new ByteArrayInputStream(source), "<br>", "<br/>");
|
|
}
|
|
}
|
|
|
|
@Test
|
|
public void testOK() throws IOException {
|
|
byte[] ok = getBytes("<p><div>Hello There!</div> <div>Tags!</div></p>");
|
|
|
|
EvilUnclosedBRFixingInputStream inp = new EvilUnclosedBRFixingInputStream(ok);
|
|
|
|
assertArrayEquals(ok, IOUtils.toByteArray(inp));
|
|
inp.close();
|
|
}
|
|
|
|
@Test
|
|
public void testProblem() throws IOException {
|
|
byte[] orig = getBytes("<p><div>Hello<br>There!</div> <div>Tags!</div></p>");
|
|
byte[] fixed = getBytes("<p><div>Hello<br/>There!</div> <div>Tags!</div></p>");
|
|
|
|
EvilUnclosedBRFixingInputStream inp = new EvilUnclosedBRFixingInputStream(orig);
|
|
|
|
assertArrayEquals(fixed, IOUtils.toByteArray(inp));
|
|
inp.close();
|
|
}
|
|
|
|
/**
|
|
* Checks that we can copy with br tags around the buffer boundaries
|
|
*/
|
|
@Test
|
|
public void testBufferSize() throws IOException {
|
|
byte[] orig = getBytes("<p><div>Hello<br> <br>There!</div> <div>Tags!<br><br></div></p>");
|
|
byte[] fixed = getBytes("<p><div>Hello<br/> <br/>There!</div> <div>Tags!<br/><br/></div></p>");
|
|
|
|
// Vary the buffer size, so that we can end up with the br in the
|
|
// overflow or only part in the buffer
|
|
for(int i=5; i<orig.length; i++) {
|
|
EvilUnclosedBRFixingInputStream inp = new EvilUnclosedBRFixingInputStream(orig);
|
|
|
|
ByteArrayOutputStream bout = new ByteArrayOutputStream();
|
|
boolean going = true;
|
|
while(going) {
|
|
byte[] b = new byte[i];
|
|
int r = inp.read(b);
|
|
if(r > 0) {
|
|
bout.write(b, 0, r);
|
|
} else {
|
|
going = false;
|
|
}
|
|
}
|
|
|
|
byte[] result = bout.toByteArray();
|
|
assertArrayEquals(fixed, result);
|
|
inp.close();
|
|
}
|
|
}
|
|
|
|
private static byte[] getBytes(String str) {
|
|
return str.getBytes(Charset.forName("UTF-8"));
|
|
}
|
|
}
|