org.apache.commons.fileupload.disk.DiskFileItem Java Examples

public static PopupForm fromRequest(String uuid, HttpServletRequest request) {
    String descriptor = request.getParameter("descriptor");
    boolean isOpenCampaign = "open-campaign".equals(request.getParameter("open-campaign"));

    long startTime = "".equals(request.getParameter("start_time")) ? 0 : parseTime(request.getParameter("start_time_selected_datetime"));
    long endTime = "".equals(request.getParameter("end_time")) ? 0 : parseTime(request.getParameter("end_time_selected_datetime"));

    List<String> assignToEids = new ArrayList<String>();
    if (request.getParameter("distribution") != null) {
        for (String user : request.getParameter("distribution").split("[\r\n]+")) {
            if (!user.isEmpty()) {
                assignToEids.add(user);
            }
        }
    }

    Optional<DiskFileItem> templateItem = Optional.empty();
    DiskFileItem dfi = (DiskFileItem) request.getAttribute("template");
    if (dfi != null && dfi.getSize() > 0) {
        templateItem = Optional.of(dfi);
    }

    return new PopupForm(uuid, descriptor, startTime, endTime, isOpenCampaign, assignToEids, templateItem);
}
 

/**
 * Create an instance of FileItem in a temporary file.
 * 
 * @param fieldName
 * @param fileName
 * @param classPathResource
 * @return
 * @throws IOException
 * @throws URISyntaxException
 */
private FileItem createFileItemFromClassPathResource(String fieldName, String fileName, String classPathResource)
        throws IOException, URISyntaxException {
    InputStream inputStream = getClass().getResourceAsStream(classPathResource);
    Path inputPath = Paths.get(getClass().getResource(classPathResource).toURI());
    int availableBytes = inputStream.available();

    // Write the inputStream to a FileItem
    File outFile = codedefendersHome.createTempFile("ClassUploadManagerTest","");
    outFile.deleteOnExit();
    
    String contentType = null;
    if( classPathResource.endsWith(".zip") ){
        contentType = "application/zip";
    } else {
        contentType = "plain/text";
    }
    FileItem fileItem = new DiskFileItem(fieldName, contentType, false, fileName, availableBytes, outFile);
    Files.copy(inputPath, fileItem.getOutputStream());

    return fileItem;
}
 

public DiskFileItem getObject ( String command ) throws Exception {

        String[] parts = command.split(";");

        if ( parts.length == 3 && "copyAndDelete".equals(parts[ 0 ]) ) {
            return copyAndDelete(parts[ 1 ], parts[ 2 ]);
        }
        else if ( parts.length == 3 && "write".equals(parts[ 0 ]) ) {
            return write(parts[ 1 ], parts[ 2 ].getBytes("US-ASCII"));
        }
        else if ( parts.length == 3 && "writeB64".equals(parts[ 0 ]) ) {
            return write(parts[ 1 ], Base64.decodeBase64(parts[ 2 ]));
        }
        else if ( parts.length == 3 && "writeOld".equals(parts[ 0 ]) ) {
            return writePre131(parts[ 1 ], parts[ 2 ].getBytes("US-ASCII"));
        }
        else if ( parts.length == 3 && "writeOldB64".equals(parts[ 0 ]) ) {
            return writePre131(parts[ 1 ], Base64.decodeBase64(parts[ 2 ]));
        }
        else {
            throw new IllegalArgumentException("Unsupported command " + command + " " + Arrays.toString(parts));
        }
    }
 

public static PopupForm fromRequest(String uuid, HttpServletRequest request) {
    String descriptor = request.getParameter("descriptor");
    boolean isOpenCampaign = "open-campaign".equals(request.getParameter("open-campaign"));

    long startTime = "".equals(request.getParameter("start_time")) ? 0 : parseTime(request.getParameter("start_time_selected_datetime"));
    long endTime = "".equals(request.getParameter("end_time")) ? 0 : parseTime(request.getParameter("end_time_selected_datetime"));

    List<String> assignToEids = new ArrayList<String>();
    if (request.getParameter("distribution") != null) {
        for (String user : request.getParameter("distribution").split("[\r\n]+")) {
            if (!user.isEmpty()) {
                assignToEids.add(user);
            }
        }
    }

    Optional<DiskFileItem> templateItem = Optional.empty();
    DiskFileItem dfi = (DiskFileItem) request.getAttribute("template");
    if (dfi != null && dfi.getSize() > 0) {
        templateItem = Optional.of(dfi);
    }

    return new PopupForm(uuid, descriptor, startTime, endTime, isOpenCampaign, assignToEids, templateItem);
}
 

public DiskFileItem getObject ( CmdExecuteHelper cmdHelper ) throws Exception {

        String[] parts = cmdHelper.getCommand().split(";");

        if ( parts.length == 3 && "copyAndDelete".equals(parts[ 0 ]) ) {
            return copyAndDelete(parts[ 1 ], parts[ 2 ]);
        }
        else if ( parts.length == 3 && "write".equals(parts[ 0 ]) ) {
            return write(parts[ 1 ], parts[ 2 ].getBytes("US-ASCII"));
        }
        else if ( parts.length == 3 && "writeB64".equals(parts[ 0 ]) ) {
            return write(parts[ 1 ], Base64.decodeBase64(parts[ 2 ]));
        }
        else if ( parts.length == 3 && "writeOld".equals(parts[ 0 ]) ) {
            return writePre131(parts[ 1 ], parts[ 2 ].getBytes("US-ASCII"));
        }
        else if ( parts.length == 3 && "writeOldB64".equals(parts[ 0 ]) ) {
            return writePre131(parts[ 1 ], Base64.decodeBase64(parts[ 2 ]));
        }
        else {
            throw new IllegalArgumentException("Unsupported command " + cmdHelper.getCommand() + " " + Arrays.toString(parts));
        }
    }
 

public void testDoImportUsersFromFileNoToken( ) throws AccessDeniedException, UserNotSignedException, IOException
{
    AdminUserJspBean bean = new AdminUserJspBean( );
    MockHttpServletRequest request = new MockHttpServletRequest( );
    AdminUser user = getUserToModify( );
    AdminAuthenticationService.getInstance( ).registerUser( request, user );
    bean.init( request, "CORE_USERS_MANAGEMENT" );
    Map<String, List<FileItem>> multipartFiles = new HashMap<>( );
    List<FileItem> fileItems = new ArrayList<>( );
    FileItem file = new DiskFileItem( "import_file", "application/csv", true, "junit.csv", 1024, new File( System.getProperty( "java.io.tmpdir" ) ) );
    OutputStreamWriter writer = new OutputStreamWriter( file.getOutputStream( ), Charset.forName( "UTF-8" ) );
    writer.write( "test;test;test;[email protected];" + AdminUser.ACTIVE_CODE + ";" + Locale.FRANCE + ";0;false;false;;;" );
    writer.close( );
    fileItems.add( file );
    multipartFiles.put( "import_file", fileItems );
    Map<String, String [ ]> parameters = request.getParameterMap( );
    MultipartHttpServletRequest multipartRequest = new MultipartHttpServletRequest( request, multipartFiles, parameters );
    bean.getImportUsersFromFile( request ); // initialize _importAdminUserService
    AdminUser importedUser = null;
    try
    {
        bean.doImportUsersFromFile( multipartRequest );
        fail( "Should have thrown" );
    }
    catch( AccessDeniedException e )
    {
        importedUser = AdminUserHome.findUserByLogin( "test" );
        assertNull( importedUser );
    }
    finally
    {
        if ( importedUser != null )
        {
            disposeOfUser( importedUser );
        }
        disposeOfUser( user );
    }
}
 

/**
 * Return a description for the storage location of the multipart content.
 * Tries to be as specific as possible: mentions the file location in case
 * of a temporary file.
 */
public String getStorageDescription() {
	if (this.fileItem.isInMemory()) {
		return "in memory";
	}
	else if (this.fileItem instanceof DiskFileItem) {
		return "at [" + ((DiskFileItem) this.fileItem).getStoreLocation().getAbsolutePath() + "]";
	}
	else {
		return "on disk";
	}
}
 

/**
 * Determine whether the multipart content is still available.
 * If a temporary file has been moved, the content is no longer available.
 */
protected boolean isAvailable() {
	// If in memory, it's available.
	if (this.fileItem.isInMemory()) {
		return true;
	}
	// Check actual existence of temporary file.
	if (this.fileItem instanceof DiskFileItem) {
		return ((DiskFileItem) this.fileItem).getStoreLocation().exists();
	}
	// Check whether current file size is different than original one.
	return (this.fileItem.getSize() == this.size);
}
 

private static DiskFileItem makePayload ( int thresh, String repoPath, String filePath, byte[] data ) throws IOException, Exception {
    // if thresh < written length, delete outputFile after copying to repository temp file
    // otherwise write the contents to repository temp file
    File repository = new File(repoPath);
    DiskFileItem diskFileItem = new DiskFileItem("test", "application/octet-stream", false, "test", 100000, repository);
    File outputFile = new File(filePath);
    DeferredFileOutputStream dfos = new DeferredFileOutputStream(thresh, outputFile);
    OutputStream os = (OutputStream) Reflections.getFieldValue(dfos, "memoryOutputStream");
    os.write(data);
    Reflections.getField(ThresholdingOutputStream.class, "written").set(dfos, data.length);
    Reflections.setFieldValue(diskFileItem, "dfos", dfos);
    Reflections.setFieldValue(diskFileItem, "sizeThreshold", 0);
    return diskFileItem;
}
 

public static String getTextParameter(DiskFileItem diskFileItem, String characterEncoding)
        throws Exception {
    String encoding = diskFileItem.getCharSet();
    if (encoding == null) {
        encoding = characterEncoding;
    }
    String textValue;
    if (encoding == null) {
        textValue = new String(diskFileItem.get());
    } else {
        textValue = new String(diskFileItem.get(), encoding);
    }
    return textValue;
}
 

public void testDoImportUsersFromFile( ) throws AccessDeniedException, UserNotSignedException, IOException
{
    AdminUserJspBean bean = new AdminUserJspBean( );
    MockHttpServletRequest request = new MockHttpServletRequest( );
    request.addParameter( SecurityTokenService.PARAMETER_TOKEN, SecurityTokenService.getInstance( ).getToken( request, "jsp/admin/user/ImportUser.jsp" ) );
    AdminUser user = getUserToModify( );
    AdminAuthenticationService.getInstance( ).registerUser( request, user );
    bean.init( request, "CORE_USERS_MANAGEMENT" );
    Map<String, List<FileItem>> multipartFiles = new HashMap<>( );
    List<FileItem> fileItems = new ArrayList<>( );
    FileItem file = new DiskFileItem( "import_file", "application/csv", true, "junit.csv", 1024, new File( System.getProperty( "java.io.tmpdir" ) ) );
    OutputStreamWriter writer = new OutputStreamWriter( file.getOutputStream( ), Charset.forName( "UTF-8" ) );
    writer.write( "test;test;test;[email protected];" + AdminUser.ACTIVE_CODE + ";" + Locale.FRANCE + ";0;false;false;;;" );
    writer.close( );
    fileItems.add( file );
    multipartFiles.put( "import_file", fileItems );
    Map<String, String [ ]> parameters = request.getParameterMap( );
    MultipartHttpServletRequest multipartRequest = new MultipartHttpServletRequest( request, multipartFiles, parameters );
    bean.getImportUsersFromFile( request ); // initialize _importAdminUserService
    AdminUser importedUser = null;
    try
    {
        bean.doImportUsersFromFile( multipartRequest );
        importedUser = AdminUserHome.findUserByLogin( "test" );
        assertNotNull( importedUser );
        assertEquals( "test", importedUser.getAccessCode( ) );
        assertEquals( "test", importedUser.getFirstName( ) );
        assertEquals( "test", importedUser.getLastName( ) );
        assertEquals( "[email protected]", importedUser.getEmail( ) );
        assertEquals( AdminUser.ACTIVE_CODE, importedUser.getStatus( ) );
    }
    finally
    {
        if ( importedUser != null )
        {
            disposeOfUser( importedUser );
        }
        disposeOfUser( user );
    }
}
 

public void testDoImportUsersFromFileInvalidToken( ) throws AccessDeniedException, UserNotSignedException, IOException
{
    AdminUserJspBean bean = new AdminUserJspBean( );
    MockHttpServletRequest request = new MockHttpServletRequest( );
    request.addParameter( SecurityTokenService.PARAMETER_TOKEN, SecurityTokenService.getInstance( ).getToken( request, "jsp/admin/user/ImportUser.jsp" )
            + "b" );
    AdminUser user = getUserToModify( );
    AdminAuthenticationService.getInstance( ).registerUser( request, user );
    bean.init( request, "CORE_USERS_MANAGEMENT" );
    Map<String, List<FileItem>> multipartFiles = new HashMap<>( );
    List<FileItem> fileItems = new ArrayList<>( );
    FileItem file = new DiskFileItem( "import_file", "application/csv", true, "junit.csv", 1024, new File( System.getProperty( "java.io.tmpdir" ) ) );
    OutputStreamWriter writer = new OutputStreamWriter( file.getOutputStream( ), Charset.forName( "UTF-8" ) );
    writer.write( "test;test;test;[email protected];" + AdminUser.ACTIVE_CODE + ";" + Locale.FRANCE + ";0;false;false;;;" );
    writer.close( );
    fileItems.add( file );
    multipartFiles.put( "import_file", fileItems );
    Map<String, String [ ]> parameters = request.getParameterMap( );
    MultipartHttpServletRequest multipartRequest = new MultipartHttpServletRequest( request, multipartFiles, parameters );
    bean.getImportUsersFromFile( request ); // initialize _importAdminUserService
    AdminUser importedUser = null;
    try
    {
        bean.doImportUsersFromFile( multipartRequest );
        fail( "Should have thrown" );
    }
    catch( AccessDeniedException e )
    {
        importedUser = AdminUserHome.findUserByLogin( "test" );
        assertNull( importedUser );
    }
    finally
    {
        if ( importedUser != null )
        {
            disposeOfUser( importedUser );
        }
        disposeOfUser( user );
    }
}
 

private PopupForm(String uuid,
                  String descriptor,
                  long startTime, long endTime,
                  boolean isOpenCampaign,
                  List<String> assignToEids,
                  Optional<DiskFileItem> templateItem) {
    this.uuid = uuid;
    this.descriptor = descriptor;
    this.startTime = startTime;
    this.endTime = endTime;
    this.isOpenCampaign = isOpenCampaign;
    this.assignToEids = assignToEids;
    this.templateItem = templateItem;
}
 

/**
 * Returns an identifier that is unique within the class loader used to
 * load this class, but does not have random-like apearance.
 *
 * @return A String with the non-random looking instance identifier.
 */
private static String getUniqueId() {
    int current;
    synchronized (DiskFileItem.class) {
        current = counter++;
    }
    String id = Integer.toString(current);

    // If you manage to get more than 100 million of ids, you'll
    // start getting ids longer than 8 characters.
    if (current < 100000000) {
        id = ("00000000" + id).substring(id.length());
    }
    return id;
}
 

private PopupForm(String uuid,
                  String descriptor,
                  long startTime, long endTime,
                  boolean isOpenCampaign,
                  List<String> assignToEids,
                  Optional<DiskFileItem> templateItem) {
    this.uuid = uuid;
    this.descriptor = descriptor;
    this.startTime = startTime;
    this.endTime = endTime;
    this.isOpenCampaign = isOpenCampaign;
    this.assignToEids = assignToEids;
    this.templateItem = templateItem;
}
 

@Override
public Collection<String> getHeaderNames() {
    if (fileItem instanceof DiskFileItem) {
        final Set<String> headerNames = new LinkedHashSet<>();
        final Iterator<String> iter = fileItem.getHeaders().getHeaderNames();
        while (iter.hasNext()) {
            headerNames.add(iter.next());
        }
        return headerNames;
    }
    return Collections.emptyList();
}
 

private MultipartFile createMultipartFile(String filename) throws IOException {
  File file = new File("/src/test/resources/" + filename);

  DiskFileItem fileItem =
      new DiskFileItem(
          "file", "text/plain", false, file.getName(), (int) file.length(), file.getParentFile());
  fileItem.getOutputStream();
  return new CommonsMultipartFile(fileItem);
}
 

@Test
public void testFullFormWithImageTooWide() throws Exception {

	// FIXME imageFileItem does not get any data
	File image = new File("testImage_tooWide.jpg");

	if (image.length() > 0) {

		FileItem imageFileItem = new DiskFileItem("image", "image/JPEG",
				true, "file", 50000, null);
		// imageFileItem = image.toFileItem();

		ProcessCreateResponse testResponse = this
				.processTestRequest(
						ProtocolTestConstants.VALID_SUGGESTION_KEY,
						ProtocolTestConstants.VALID_SUGGESTION_STRING,
						ProtocolTestConstants.VALID_SUGGESTION_WEIGHT,
						ProtocolTestConstants.VALID_SUGGESTION_INDEX,
						imageFileItem);

		// TODO: Insert Assertion

		System.out.println(testResponse.getContainer().getComponents()
				.getImageBase64());
	} else {
		fail("no image to test with provided!");
	}
}
 

@Test
public void testFullFormWithImageTooHigh() throws Exception {

	// FIXME imageFileItem does not get any data
	File image = new File("testImage_tooHigh.jpg");

	if (image.length() > 0) {

		FileItem imageFileItem = new DiskFileItem("image", "image/JPEG",
				true, "file", 50000, null);
		// imageFileItem = image.toFileItem();

		ProcessCreateResponse testResponse = this
				.processTestRequest(
						ProtocolTestConstants.VALID_SUGGESTION_KEY,
						ProtocolTestConstants.VALID_SUGGESTION_STRING,
						ProtocolTestConstants.VALID_SUGGESTION_WEIGHT,
						ProtocolTestConstants.VALID_SUGGESTION_INDEX,
						imageFileItem);

		// TODO: Insert Assertion

		System.out.println(testResponse.getContainer().getComponents()
				.getImageBase64());
	} else {
		fail("no image to test with provided!");
	}
}
 

@Test
public void testFullFormWithImageWrongType() throws Exception {

	// FIXME imageFileItem does not get any data
	File image = new File("testImage_wrongType.png");

	if (image.length() > 0) {

		FileItem imageFileItem = new DiskFileItem("image", "image/JPEG",
				true, "file", 50000, null);
		// imageFileItem = image.toFileItem();

		ProcessCreateResponse testResponse = this
				.processTestRequest(
						ProtocolTestConstants.VALID_SUGGESTION_KEY,
						ProtocolTestConstants.VALID_SUGGESTION_STRING,
						ProtocolTestConstants.VALID_SUGGESTION_WEIGHT,
						ProtocolTestConstants.VALID_SUGGESTION_INDEX,
						imageFileItem);

		// TODO: Insert Assertion

		System.out.println(testResponse.getContainer().getComponents()
				.getImageBase64());
	} else {
		fail("no image to test with provided!");
	}
}
 

/**
 * Return a description for the storage location of the multipart content.
 * Tries to be as specific as possible: mentions the file location in case
 * of a temporary file.
 */
public String getStorageDescription() {
	if (this.fileItem.isInMemory()) {
		return "in memory";
	}
	else if (this.fileItem instanceof DiskFileItem) {
		return "at [" + ((DiskFileItem) this.fileItem).getStoreLocation().getAbsolutePath() + "]";
	}
	else {
		return "on disk";
	}
}
 

/**
 * Determine whether the multipart content is still available.
 * If a temporary file has been moved, the content is no longer available.
 */
protected boolean isAvailable() {
	// If in memory, it's available.
	if (this.fileItem.isInMemory()) {
		return true;
	}
	// Check actual existence of temporary file.
	if (this.fileItem instanceof DiskFileItem) {
		return ((DiskFileItem) this.fileItem).getStoreLocation().exists();
	}
	// Check whether current file size is different than original one.
	return (this.fileItem.getSize() == this.size);
}
 

public static String getTextParameter(DiskFileItem diskFileItem, String characterEncoding)
        throws Exception {

    String encoding = diskFileItem.getCharSet();
    if (encoding == null) {
        encoding = characterEncoding;
    }
    String textValue;
    if (encoding == null) {
        textValue = new String(diskFileItem.get(), StandardCharsets.UTF_8);
    } else {
        textValue = new String(diskFileItem.get(), encoding);
    }
    return textValue;
}
 

/**
 * Determine whether the multipart content is still available. If a temporary file has been moved, the content is no
 * longer available.
 */
protected boolean isAvailable() {
    // If in memory, it's available.
    if (fileItem.isInMemory()) {
        return true;
    }
    // Check actual existence of temporary file.
    if (fileItem instanceof DiskFileItem) {
        return ((DiskFileItem) fileItem).getStoreLocation().exists();
    }
    // Check whether current file size is different than original one.
    return (fileItem.getSize() == size);
}
 

/**
 * Return a description for the storage location of the multipart content.
 * Tries to be as specific as possible: mentions the file location in case
 * of a temporary file.
 */
public String getStorageDescription() {
    if (this.fileItem.isInMemory()) {
        return "in memory";
    }
    else if (this.fileItem instanceof DiskFileItem) {
        return "at [" + ((DiskFileItem) this.fileItem).getStoreLocation().getAbsolutePath() + "]";
    }
    else {
        return "on disk";
    }
}
 

/**
 * Return a description for the storage location of the multipart content.
 * Tries to be as specific as possible: mentions the file location in case
 * of a temporary file.
 */
public String getStorageDescription() {
	if (this.fileItem.isInMemory()) {
		return "in memory";
	}
	else if (this.fileItem instanceof DiskFileItem) {
		return "at [" + ((DiskFileItem) this.fileItem).getStoreLocation().getAbsolutePath() + "]";
	}
	else {
		return "on disk";
	}
}
 

/**
 * Test delete method.
 *
 * @throws Exception when a serious error occurs.
 */
@Test
public void testDelete() throws Exception {
    File file = new File("delete.me");
    file.createNewFile();
    assertTrue(file.exists());
    FileItem item = new DiskFileItem("fieldName", "text/html", false,
            "delete.me", 0, file);
    ApacheMultiPart part = new ApacheMultiPart(item);
    part.delete();
    file.delete();
}
 

/**
 * Test getContentType method.
 * 
 * @throws Exception when a serious error occurs.
 */
@Test
public void testGetContentType() throws Exception {
    File file = new File("delete.me");
    file.createNewFile();
    assertTrue(file.exists());
    FileItem item = new DiskFileItem("fieldName", "text/html", false,
            "delete.me", 0, file);
    ApacheMultiPart part = new ApacheMultiPart(item);
    assertEquals(part.getContentType(), "text/html");
    file.delete();
}
 

/**
 * Determine whether the multipart content is still available.
 * If a temporary file has been moved, the content is no longer available.
 */
protected boolean isAvailable() {
    // If in memory, it's available.
    if (this.fileItem.isInMemory()) {
        return true;
    }
    // Check actual existence of temporary file.
    if (this.fileItem instanceof DiskFileItem) {
        return ((DiskFileItem) this.fileItem).getStoreLocation().exists();
    }
    // Check whether current file size is different than original one.
    return (this.fileItem.getSize() == this.size);
}
 

private static DiskFileItem makePayload ( int thresh, String repoPath, String filePath, byte[] data ) throws IOException, Exception {
    // if thresh < written length, delete outputFile after copying to repository temp file
    // otherwise write the contents to repository temp file
    File repository = new File(repoPath);
    DiskFileItem diskFileItem = new DiskFileItem("test", "application/octet-stream", false, "test", 100000, repository);
    File outputFile = new File(filePath);
    DeferredFileOutputStream dfos = new DeferredFileOutputStream(thresh, outputFile);
    OutputStream os = (OutputStream) Reflections.getFieldValue(dfos, "memoryOutputStream");
    os.write(data);
    Reflections.getField(ThresholdingOutputStream.class, "written").set(dfos, data.length);
    Reflections.setFieldValue(diskFileItem, "dfos", dfos);
    Reflections.setFieldValue(diskFileItem, "sizeThreshold", 0);
    return diskFileItem;
}