147 lines
3.6 KiB
Java
147 lines
3.6 KiB
Java
/*
|
|
* Copyright 2012 The Netty Project
|
|
*
|
|
* The Netty Project 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 io.netty.handler.codec.http.multipart;
|
|
|
|
import io.netty.buffer.ByteBuf;
|
|
import io.netty.channel.ChannelException;
|
|
import io.netty.handler.codec.http.HttpConstants;
|
|
import io.netty.util.AbstractReferenceCounted;
|
|
|
|
import java.io.IOException;
|
|
import java.nio.charset.Charset;
|
|
import java.util.regex.Pattern;
|
|
|
|
/**
|
|
* Abstract HttpData implementation
|
|
*/
|
|
public abstract class AbstractHttpData extends AbstractReferenceCounted implements HttpData {
|
|
|
|
private static final Pattern STRIP_PATTERN = Pattern.compile("(?:^\\s+|\\s+$|\\n)");
|
|
private static final Pattern REPLACE_PATTERN = Pattern.compile("[\\r\\t]");
|
|
|
|
private final String name;
|
|
protected long definedSize;
|
|
protected long size;
|
|
private Charset charset = HttpConstants.DEFAULT_CHARSET;
|
|
private boolean completed;
|
|
private long maxSize = DefaultHttpDataFactory.MAXSIZE;
|
|
|
|
protected AbstractHttpData(String name, Charset charset, long size) {
|
|
if (name == null) {
|
|
throw new NullPointerException("name");
|
|
}
|
|
|
|
name = REPLACE_PATTERN.matcher(name).replaceAll(" ");
|
|
name = STRIP_PATTERN.matcher(name).replaceAll("");
|
|
|
|
if (name.isEmpty()) {
|
|
throw new IllegalArgumentException("empty name");
|
|
}
|
|
|
|
this.name = name;
|
|
if (charset != null) {
|
|
setCharset(charset);
|
|
}
|
|
definedSize = size;
|
|
}
|
|
|
|
@Override
|
|
public long getMaxSize() {
|
|
return maxSize;
|
|
}
|
|
|
|
@Override
|
|
public void setMaxSize(long maxSize) {
|
|
this.maxSize = maxSize;
|
|
}
|
|
|
|
@Override
|
|
public void checkSize(long newSize) throws IOException {
|
|
if (maxSize >= 0 && newSize > maxSize) {
|
|
throw new IOException("Size exceed allowed maximum capacity");
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public String getName() {
|
|
return name;
|
|
}
|
|
|
|
@Override
|
|
public boolean isCompleted() {
|
|
return completed;
|
|
}
|
|
|
|
protected void setCompleted() {
|
|
completed = true;
|
|
}
|
|
|
|
@Override
|
|
public Charset getCharset() {
|
|
return charset;
|
|
}
|
|
|
|
@Override
|
|
public void setCharset(Charset charset) {
|
|
if (charset == null) {
|
|
throw new NullPointerException("charset");
|
|
}
|
|
this.charset = charset;
|
|
}
|
|
|
|
@Override
|
|
public long length() {
|
|
return size;
|
|
}
|
|
|
|
@Override
|
|
public long definedLength() {
|
|
return definedSize;
|
|
}
|
|
|
|
@Override
|
|
public ByteBuf content() {
|
|
try {
|
|
return getByteBuf();
|
|
} catch (IOException e) {
|
|
throw new ChannelException(e);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
protected void deallocate() {
|
|
delete();
|
|
}
|
|
|
|
@Override
|
|
public HttpData retain() {
|
|
super.retain();
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public HttpData retain(int increment) {
|
|
super.retain(increment);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public abstract HttpData touch();
|
|
|
|
@Override
|
|
public abstract HttpData touch(Object hint);
|
|
}
|