Search in sources :

Example 1 with FileUploadIOException

use of com.helger.web.fileupload.exception.FileUploadIOException in project ph-web by phax.

the class AbstractFileUploadBase method parseRequest.

/**
 * Processes an <a href="http://www.ietf.org/rfc/rfc1867.txt">RFC 1867</a>
 * compliant <code>multipart/form-data</code> stream.
 *
 * @param aCtx
 *        The context for the request to be parsed.
 * @return A list of <code>FileItem</code> instances parsed from the request,
 *         in the order that they were transmitted.
 * @throws FileUploadException
 *         if there are problems reading/parsing the request or storing files.
 */
@Nonnull
@ReturnsMutableCopy
public ICommonsList<IFileItem> parseRequest(@Nonnull final IRequestContext aCtx) throws FileUploadException {
    final ICommonsList<IFileItem> aItems = new CommonsArrayList<>();
    boolean bSuccessful = false;
    try {
        final IFileItemIterator aItemIter = getItemIterator(aCtx);
        final IFileItemFactory aFileItemFactory = getFileItemFactory();
        if (aFileItemFactory == null)
            throw new IllegalStateException("No FileItemFactory has been set.");
        while (aItemIter.hasNext()) {
            final IFileItemStream aFileItemStream = aItemIter.next();
            // Don't use getName() here to prevent an InvalidFileNameException.
            final IFileItem aFileItem = aFileItemFactory.createItem(aFileItemStream.getFieldName(), aFileItemStream.getContentType(), aFileItemStream.isFormField(), aFileItemStream.getNameUnchecked());
            aItems.add(aFileItem);
            try (final InputStream aIS = aFileItemStream.openStream();
                final OutputStream aOS = aFileItem.getOutputStream()) {
                final byte[] aBuffer = new byte[8192];
                int nBytesRead;
                // potentially blocking read
                while ((nBytesRead = aIS.read(aBuffer, 0, aBuffer.length)) > -1) {
                    aOS.write(aBuffer, 0, nBytesRead);
                }
            } catch (final FileUploadIOException ex) {
                throw (FileUploadException) ex.getCause();
            } catch (final IOException ex) {
                throw new IOFileUploadException("Processing of " + RequestHelper.MULTIPART_FORM_DATA + " request failed. " + ex.getMessage(), ex);
            }
            if (aFileItem instanceof IFileItemHeadersSupport) {
                final IFileItemHeaders aFileItemHeaders = aFileItemStream.getHeaders();
                ((IFileItemHeadersSupport) aFileItem).setHeaders(aFileItemHeaders);
            }
        }
        bSuccessful = true;
        return aItems;
    } catch (final FileUploadIOException ex) {
        throw (FileUploadException) ex.getCause();
    } catch (final IOException ex) {
        throw new FileUploadException(ex.getMessage(), ex);
    } finally {
        if (!bSuccessful) {
            // Delete all file items
            for (final IFileItem aFileItem : aItems) {
                try {
                    aFileItem.delete();
                } catch (final Exception ex) {
                    // ignore it
                    if (LOGGER.isErrorEnabled())
                        LOGGER.error("Failed to delete fileItem " + aFileItem, ex);
                }
            }
        }
    }
}
Also used : IFileItemHeaders(com.helger.web.fileupload.IFileItemHeaders) AbstractLimitedInputStream(com.helger.web.fileupload.io.AbstractLimitedInputStream) InputStream(java.io.InputStream) OutputStream(java.io.OutputStream) FileUploadIOException(com.helger.web.fileupload.exception.FileUploadIOException) IOException(java.io.IOException) SizeLimitExceededException(com.helger.web.fileupload.exception.SizeLimitExceededException) InvalidContentTypeException(com.helger.web.fileupload.exception.InvalidContentTypeException) FileUploadException(com.helger.web.fileupload.exception.FileUploadException) FileUploadIOException(com.helger.web.fileupload.exception.FileUploadIOException) NoSuchElementException(java.util.NoSuchElementException) IOException(java.io.IOException) IOFileUploadException(com.helger.web.fileupload.exception.IOFileUploadException) FileUploadIOException(com.helger.web.fileupload.exception.FileUploadIOException) IFileItemStream(com.helger.web.fileupload.IFileItemStream) IFileItemIterator(com.helger.web.fileupload.IFileItemIterator) IOFileUploadException(com.helger.web.fileupload.exception.IOFileUploadException) IFileItemHeadersSupport(com.helger.web.fileupload.IFileItemHeadersSupport) IFileItem(com.helger.web.fileupload.IFileItem) CommonsArrayList(com.helger.commons.collection.impl.CommonsArrayList) FileUploadException(com.helger.web.fileupload.exception.FileUploadException) IOFileUploadException(com.helger.web.fileupload.exception.IOFileUploadException) IFileItemFactory(com.helger.web.fileupload.IFileItemFactory) ReturnsMutableCopy(com.helger.commons.annotation.ReturnsMutableCopy) Nonnull(javax.annotation.Nonnull)

Aggregations

ReturnsMutableCopy (com.helger.commons.annotation.ReturnsMutableCopy)1 CommonsArrayList (com.helger.commons.collection.impl.CommonsArrayList)1 IFileItem (com.helger.web.fileupload.IFileItem)1 IFileItemFactory (com.helger.web.fileupload.IFileItemFactory)1 IFileItemHeaders (com.helger.web.fileupload.IFileItemHeaders)1 IFileItemHeadersSupport (com.helger.web.fileupload.IFileItemHeadersSupport)1 IFileItemIterator (com.helger.web.fileupload.IFileItemIterator)1 IFileItemStream (com.helger.web.fileupload.IFileItemStream)1 FileUploadException (com.helger.web.fileupload.exception.FileUploadException)1 FileUploadIOException (com.helger.web.fileupload.exception.FileUploadIOException)1 IOFileUploadException (com.helger.web.fileupload.exception.IOFileUploadException)1 InvalidContentTypeException (com.helger.web.fileupload.exception.InvalidContentTypeException)1 SizeLimitExceededException (com.helger.web.fileupload.exception.SizeLimitExceededException)1 AbstractLimitedInputStream (com.helger.web.fileupload.io.AbstractLimitedInputStream)1 IOException (java.io.IOException)1 InputStream (java.io.InputStream)1 OutputStream (java.io.OutputStream)1 NoSuchElementException (java.util.NoSuchElementException)1 Nonnull (javax.annotation.Nonnull)1