    001/*
    002 * Licensed to the Apache Software Foundation (ASF) under one or more
    003 * contributor license agreements.  See the NOTICE file distributed with
    004 * this work for additional information regarding copyright ownership.
    005 * The ASF licenses this file to You under the Apache License, Version 2.0
    006 * (the "License"); you may not use this file except in compliance with
    007 * the License.  You may obtain a copy of the License at
    008 *
    009 *      http://www.apache.org/licenses/LICENSE-2.0
    010 *
    011 * Unless required by applicable law or agreed to in writing, software
    012 * distributed under the License is distributed on an "AS IS" BASIS,
    013 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    014 * See the License for the specific language governing permissions and
    015 * limitations under the License.
    016 */
    017
    018package org.apache.commons.fileupload2.javax;
    019
    020import static org.junit.jupiter.api.Assertions.assertEquals;
    021import static org.junit.jupiter.api.Assertions.assertTrue;
    022
    023import java.io.ByteArrayOutputStream;
    024import java.io.IOException;
    025import java.nio.charset.StandardCharsets;
    026import java.util.List;
    027
    028import javax.servlet.http.HttpServletRequest;
    029
    030import org.apache.commons.fileupload2.core.AbstractFileUploadTest;
    031import org.apache.commons.fileupload2.core.Constants;
    032import org.apache.commons.fileupload2.core.DiskFileItem;
    033import org.apache.commons.fileupload2.core.DiskFileItemFactory;
    034import org.apache.commons.fileupload2.core.FileUploadException;
    035import org.junit.jupiter.api.Test;
    036
    037/**
    038 * Tests {@link JavaxServletFileUpload}.
    039 *
    040 * @see AbstractFileUploadTest
    041 */
    042public class JavaxServletFileUploadDiskTest extends AbstractFileUploadTest<JavaxServletDiskFileUpload, HttpServletRequest, DiskFileItem, DiskFileItemFactory> {
    043
    044    public JavaxServletFileUploadDiskTest() {
    045        super(new JavaxServletDiskFileUpload());
    046    }
    047
    048    @Override
    049    public List<DiskFileItem> parseUpload(final JavaxServletDiskFileUpload upload, final byte[] bytes, final String contentType) throws FileUploadException {
    050        final HttpServletRequest request = new JavaxMockHttpServletRequest(bytes, contentType);
    051        return upload.parseRequest(new JavaxServletRequestContext(request));
    052    }
    053
    054    /**
    055     * Runs a test with varying file sizes.
    056     */
    057    @Override
    058    @Test
    059    public void testFileUpload() throws IOException, FileUploadException {
    060        final var baos = new ByteArrayOutputStream();
    061        var add = 16;
    062        var num = 0;
    063        for (var i = 0; i < 16384; i += add) {
    064            if (++add == 32) {
    065                add = 16;
    066            }
    067            final var header = "-----1234\r\n" + "Content-Disposition: form-data; name=\"field" + num++ + "\"\r\n" + "\r\n";
    068            baos.write(header.getBytes(StandardCharsets.US_ASCII));
    069            for (var j = 0; j < i; j++) {
    070                baos.write((byte) j);
    071            }
    072            baos.write("\r\n".getBytes(StandardCharsets.US_ASCII));
    073        }
    074        baos.write("-----1234--\r\n".getBytes(StandardCharsets.US_ASCII));
    075
    076        final var fileItems = parseUpload(new JavaxServletDiskFileUpload(), baos.toByteArray());
    077        final var fileIter = fileItems.iterator();
    078        add = 16;
    079        num = 0;
    080        for (var i = 0; i < 16384; i += add) {
    081            if (++add == 32) {
    082                add = 16;
    083            }
    084            final var item = fileIter.next();
    085            assertEquals("field" + num++, item.getFieldName());
    086            final var bytes = item.get();
    087            assertEquals(i, bytes.length);
    088            for (var j = 0; j < i; j++) {
    089                assertEquals((byte) j, bytes[j]);
    090            }
    091        }
    092        assertTrue(!fileIter.hasNext());
    093    }
    094
    095    @Test
    096    public void testParseImpliedUtf8() throws Exception {
    097        // utf8 encoded form-data without explicit content-type encoding
    098        // @formatter:off
    099        final var text = "-----1234\r\n" +
    100                "Content-Disposition: form-data; name=\"utf8Html\"\r\n" +
    101                "\r\n" +
    102                "Thís ís the coñteñt of the fíle\n" +
    103                "\r\n" +
    104                "-----1234--\r\n";
    105        // @formatter:on
    106
    107        final var bytes = text.getBytes(StandardCharsets.UTF_8);
    108        final HttpServletRequest request = new JavaxMockHttpServletRequest(bytes, Constants.CONTENT_TYPE);
    109        // @formatter:off
    110        final var fileItemFactory = DiskFileItemFactory.builder()
    111                .setCharset(StandardCharsets.UTF_8)
    112                .get();
    113        // @formatter:on
    114        final var upload = new JavaxServletFileUpload<>(fileItemFactory);
    115        final var fileItems = upload.parseRequest(request);
    116        final var fileItem = fileItems.get(0);
    117        assertTrue(fileItem.getString().contains("coñteñt"), fileItem.getString());
    118    }
    119
    120    /*
    121     * Test case for <a href="https://issues.apache.org/jira/browse/FILEUPLOAD-210">
    122     */
    123    @Test
    124    public void testParseParameterMap() throws Exception {
    125        // @formatter:off
    126        final var text = "-----1234\r\n" +
    127                      "Content-Disposition: form-data; name=\"file\"; filename=\"foo.tab\"\r\n" +
    128                      "Content-Type: text/whatever\r\n" +
    129                      "\r\n" +
    130                      "This is the content of the file\n" +
    131                      "\r\n" +
    132                      "-----1234\r\n" +
    133                      "Content-Disposition: form-data; name=\"field\"\r\n" +
    134                      "\r\n" +
    135                      "fieldValue\r\n" +
    136                      "-----1234\r\n" +
    137                      "Content-Disposition: form-data; name=\"multi\"\r\n" +
    138                      "\r\n" +
    139                      "value1\r\n" +
    140                      "-----1234\r\n" +
    141                      "Content-Disposition: form-data; name=\"multi\"\r\n" +
    142                      "\r\n" +
    143                      "value2\r\n" +
    144                      "-----1234--\r\n";
    145        // @formatter:on
    146        final var bytes = text.getBytes(StandardCharsets.US_ASCII);
    147        final HttpServletRequest request = new JavaxMockHttpServletRequest(bytes, Constants.CONTENT_TYPE);
    148
    149        final var upload = new JavaxServletFileUpload<>(DiskFileItemFactory.builder().get());
    150        final var mappedParameters = upload.parseParameterMap(request);
    151        assertTrue(mappedParameters.containsKey("file"));
    152        assertEquals(1, mappedParameters.get("file").size());
    153
    154        assertTrue(mappedParameters.containsKey("field"));
    155        assertEquals(1, mappedParameters.get("field").size());
    156
    157        assertTrue(mappedParameters.containsKey("multi"));
    158        assertEquals(2, mappedParameters.get("multi").size());
    159    }
    160
    161}