mirror of
https://gitee.com/chinabugotech/hutool.git
synced 2025-05-09 23:51:34 +08:00
修复getFileNameFromDisposition不符合规范问题
This commit is contained in:
parent
f806e3b855
commit
dca342c28f
@ -20,13 +20,15 @@ import org.dromara.hutool.core.collection.CollUtil;
|
|||||||
import org.dromara.hutool.core.map.CaseInsensitiveMap;
|
import org.dromara.hutool.core.map.CaseInsensitiveMap;
|
||||||
import org.dromara.hutool.core.regex.ReUtil;
|
import org.dromara.hutool.core.regex.ReUtil;
|
||||||
import org.dromara.hutool.core.text.StrUtil;
|
import org.dromara.hutool.core.text.StrUtil;
|
||||||
|
import org.dromara.hutool.core.text.split.SplitUtil;
|
||||||
import org.dromara.hutool.core.util.ObjUtil;
|
import org.dromara.hutool.core.util.ObjUtil;
|
||||||
|
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* HTTP头相关方法
|
* HTTP头相关方法<br>
|
||||||
|
* 相关规范见:https://www.rfc-editor.org/rfc/rfc5987
|
||||||
*
|
*
|
||||||
* @author Looly
|
* @author Looly
|
||||||
* @since 6.0.0
|
* @since 6.0.0
|
||||||
@ -67,7 +69,6 @@ public class HttpHeaderUtil {
|
|||||||
final List<String> dispositions = headerList(headers, HeaderName.CONTENT_DISPOSITION.getValue());
|
final List<String> dispositions = headerList(headers, HeaderName.CONTENT_DISPOSITION.getValue());
|
||||||
String fileName = null;
|
String fileName = null;
|
||||||
if (CollUtil.isNotEmpty(dispositions)) {
|
if (CollUtil.isNotEmpty(dispositions)) {
|
||||||
|
|
||||||
// filename* 采用了 RFC 5987 中规定的编码方式,优先读取
|
// filename* 采用了 RFC 5987 中规定的编码方式,优先读取
|
||||||
fileName = getFileNameFromDispositions(dispositions, StrUtil.addSuffixIfNot(paramName, "*"));
|
fileName = getFileNameFromDispositions(dispositions, StrUtil.addSuffixIfNot(paramName, "*"));
|
||||||
if ((!StrUtil.endWith(fileName, "*")) && StrUtil.isBlank(fileName)) {
|
if ((!StrUtil.endWith(fileName, "*")) && StrUtil.isBlank(fileName)) {
|
||||||
@ -86,13 +87,38 @@ public class HttpHeaderUtil {
|
|||||||
* @return 文件名,empty表示无
|
* @return 文件名,empty表示无
|
||||||
*/
|
*/
|
||||||
private static String getFileNameFromDispositions(final List<String> dispositions, String paramName) {
|
private static String getFileNameFromDispositions(final List<String> dispositions, String paramName) {
|
||||||
|
// 正则转义
|
||||||
|
paramName = StrUtil.replace(paramName, "*", "\\*");
|
||||||
String fileName = null;
|
String fileName = null;
|
||||||
for (final String disposition : dispositions) {
|
for (final String disposition : dispositions) {
|
||||||
fileName = ReUtil.getGroup1(paramName + "=\"(.*?)\"", disposition);
|
fileName = ReUtil.getGroup1(paramName + "=([^;]+)", disposition);
|
||||||
if (StrUtil.isNotBlank(fileName)) {
|
if (StrUtil.isNotBlank(fileName)) {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return fileName;
|
return getRfc5987Value(fileName);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取rfc5987标准的值,标准见:https://www.rfc-editor.org/rfc/rfc5987#section-3.2.1<br>
|
||||||
|
* 包括:
|
||||||
|
*
|
||||||
|
*<ul>
|
||||||
|
* <li>Non-extended:无双引号包裹的值</li>
|
||||||
|
* <li>Non-extended:双引号包裹的值</li>
|
||||||
|
* <li>Extended notation:编码'语言'值</li>
|
||||||
|
*</ul>
|
||||||
|
*
|
||||||
|
* @param value 值
|
||||||
|
* @return 结果值
|
||||||
|
*/
|
||||||
|
private static String getRfc5987Value(final String value){
|
||||||
|
final List<String> split = SplitUtil.split(value, "'");
|
||||||
|
if(3 == split.size()){
|
||||||
|
return split.get(2);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 普通值
|
||||||
|
return StrUtil.unWrap(value, '"');
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -0,0 +1,56 @@
|
|||||||
|
package org.dromara.hutool.http.meta;
|
||||||
|
|
||||||
|
import org.dromara.hutool.core.collection.ListUtil;
|
||||||
|
import org.junit.jupiter.api.Assertions;
|
||||||
|
import org.junit.jupiter.api.Test;
|
||||||
|
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
public class HttpHeaderUtilTest {
|
||||||
|
@Test
|
||||||
|
void getFileNameFromDispositionTest() {
|
||||||
|
final Map<String, List<String>> headers = new HashMap<>();
|
||||||
|
headers.put(HeaderName.CONTENT_DISPOSITION.getValue(),
|
||||||
|
ListUtil.of("attachment; filename*=utf-8''%E6%B5%8B%E8%AF%95.xlsx; filename=\"æµ\u008Bè¯\u0095.xlsx\""));
|
||||||
|
final String fileNameFromDisposition = HttpHeaderUtil.getFileNameFromDisposition(headers, null);
|
||||||
|
Assertions.assertEquals("%E6%B5%8B%E8%AF%95.xlsx", fileNameFromDisposition);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void getFileNameFromDispositionTest2() {
|
||||||
|
final Map<String, List<String>> headers = new HashMap<>();
|
||||||
|
headers.put(HeaderName.CONTENT_DISPOSITION.getValue(),
|
||||||
|
ListUtil.of("attachment; filename*=utf-8''%E6%B5%8B%E8%AF%95.xlsx"));
|
||||||
|
final String fileNameFromDisposition = HttpHeaderUtil.getFileNameFromDisposition(headers, null);
|
||||||
|
Assertions.assertEquals("%E6%B5%8B%E8%AF%95.xlsx", fileNameFromDisposition);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void getFileNameFromDispositionTest3() {
|
||||||
|
final Map<String, List<String>> headers = new HashMap<>();
|
||||||
|
headers.put(HeaderName.CONTENT_DISPOSITION.getValue(),
|
||||||
|
ListUtil.of("attachment; filename*=\"%E6%B5%8B%E8%AF%95.xlsx\""));
|
||||||
|
final String fileNameFromDisposition = HttpHeaderUtil.getFileNameFromDisposition(headers, null);
|
||||||
|
Assertions.assertEquals("%E6%B5%8B%E8%AF%95.xlsx", fileNameFromDisposition);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void getFileNameFromDispositionTest4() {
|
||||||
|
final Map<String, List<String>> headers = new HashMap<>();
|
||||||
|
headers.put(HeaderName.CONTENT_DISPOSITION.getValue(),
|
||||||
|
ListUtil.of("attachment; filename=\"%E6%B5%8B%E8%AF%95.xlsx\""));
|
||||||
|
final String fileNameFromDisposition = HttpHeaderUtil.getFileNameFromDisposition(headers, null);
|
||||||
|
Assertions.assertEquals("%E6%B5%8B%E8%AF%95.xlsx", fileNameFromDisposition);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void getFileNameFromDispositionTest5() {
|
||||||
|
final Map<String, List<String>> headers = new HashMap<>();
|
||||||
|
headers.put(HeaderName.CONTENT_DISPOSITION.getValue(),
|
||||||
|
ListUtil.of("attachment; filename=%E6%B5%8B%E8%AF%95.xlsx"));
|
||||||
|
final String fileNameFromDisposition = HttpHeaderUtil.getFileNameFromDisposition(headers, null);
|
||||||
|
Assertions.assertEquals("%E6%B5%8B%E8%AF%95.xlsx", fileNameFromDisposition);
|
||||||
|
}
|
||||||
|
}
|
Loading…
x
Reference in New Issue
Block a user