12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394 |
- /* ====================================================================
- 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"));
- }
- }
|