forked from OpenFeign/feign
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathUriUtils.java
More file actions
220 lines (188 loc) · 6.62 KB
/
UriUtils.java
File metadata and controls
220 lines (188 loc) · 6.62 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
/*
* Copyright 2012-2023 The Feign Authors
*
* Licensed 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 feign.template;
import feign.Util;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.nio.charset.Charset;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
public class UriUtils {
private static final Pattern PCT_ENCODED_PATTERN = Pattern.compile("%[0-9A-Fa-f][0-9A-Fa-f]");
/**
* Determines if the value is already pct-encoded.
*
* @param value to check.
* @return {@literal true} if the value is already pct-encoded
*/
public static boolean isEncoded(String value, Charset charset) {
for (byte b : value.getBytes(charset)) {
if (!isUnreserved((char) b) && b != '%') {
/* break if there are any unreserved character */
return false;
}
}
return PCT_ENCODED_PATTERN.matcher(value).find();
}
/**
* Uri Encode the value, using the default Charset. Already encoded values are skipped.
*
* @param value to encode.
* @return the encoded value.
*/
public static String encode(String value) {
return encodeChunk(value, Util.UTF_8, false);
}
/**
* Uri Encode the value. Already encoded values are skipped.
*
* @param value to encode.
* @param charset to use.
* @return the encoded value.
*/
public static String encode(String value, Charset charset) {
return encodeChunk(value, charset, false);
}
public static String encode(String value, boolean allowReservedCharacters) {
return encodeInternal(value, Util.UTF_8, allowReservedCharacters);
}
public static String encode(String value, Charset charset, boolean allowReservedCharacters) {
return encodeInternal(value, charset, allowReservedCharacters);
}
/**
* Uri Decode the value.
*
* @param value to decode
* @param charset to use.
* @return the decoded value.
*/
public static String decode(String value, Charset charset) {
try {
/* there is nothing special between uri and url decoding */
return URLDecoder.decode(value, charset.name());
} catch (UnsupportedEncodingException uee) {
/* since the encoding is not supported, return the original value */
return value;
}
}
/**
* Determines if the provided uri is an absolute uri.
*
* @param uri to evaluate.
* @return true if the uri is absolute.
*/
public static boolean isAbsolute(String uri) {
return uri != null && !uri.isEmpty() && uri.startsWith("http");
}
/**
* Encodes the value, preserving all reserved characters.. Values that are already pct-encoded are
* ignored.
*
* @param value inspect.
* @param charset to use.
* @return a new String with the reserved characters preserved.
*/
public static String encodeInternal(String value,
Charset charset,
boolean allowReservedCharacters) {
/* value is encoded, we need to split it up and skip the parts that are already encoded */
Matcher matcher = PCT_ENCODED_PATTERN.matcher(value);
if (!matcher.find()) {
return encodeChunk(value, charset, true);
}
int length = value.length();
StringBuilder encoded = new StringBuilder(length + 8);
int index = 0;
do {
/* split out the value before the encoded value */
String before = value.substring(index, matcher.start());
/* encode it */
encoded.append(encodeChunk(before, charset, allowReservedCharacters));
/* append the encoded value */
encoded.append(matcher.group());
/* update the string search index */
index = matcher.end();
} while (matcher.find());
/* append the rest of the string */
String tail = value.substring(index, length);
encoded.append(encodeChunk(tail, charset, allowReservedCharacters));
return encoded.toString();
}
/**
* Encode a Uri Chunk, ensuring that all reserved characters are also encoded.
*
* @param value to encode.
* @param charset to use.
* @return an encoded uri chunk.
*/
private static String encodeChunk(String value, Charset charset, boolean allowReserved) {
if (isEncoded(value, charset)) {
return value;
}
byte[] data = value.getBytes(charset);
try (ByteArrayOutputStream bos = new ByteArrayOutputStream()) {
for (byte b : data) {
if (isUnreserved((char) b)) {
bos.write(b);
} else if (isReserved((char) b) && allowReserved) {
bos.write(b);
} else {
pctEncode(b, bos);
}
}
return new String(bos.toByteArray(), charset);
} catch (IOException ioe) {
throw new IllegalStateException("Error occurred during encoding of the uri: "
+ ioe.getMessage(), ioe);
}
}
/**
* Percent Encode the provided byte.
*
* @param data to encode
* @param bos with the output stream to use.
*/
private static void pctEncode(byte data, ByteArrayOutputStream bos) {
bos.write('%');
char hex1 = Character.toUpperCase(Character.forDigit((data >> 4) & 0xF, 16));
char hex2 = Character.toUpperCase(Character.forDigit(data & 0xF, 16));
bos.write(hex1);
bos.write(hex2);
}
private static boolean isAlpha(int c) {
return (c >= 'a' && c <= 'z' || c >= 'A' && c <= 'Z');
}
private static boolean isDigit(int c) {
return (c >= '0' && c <= '9');
}
private static boolean isGenericDelimiter(int c) {
return (c == ':') || (c == '/') || (c == '?') || (c == '#') || (c == '[') || (c == ']')
|| (c == '@');
}
private static boolean isSubDelimiter(int c) {
return (c == '!') || (c == '$') || (c == '&') || (c == '\'') || (c == '(') || (c == ')')
|| (c == '*') || (c == '+') || (c == ',') || (c == ';') || (c == '=');
}
private static boolean isUnreserved(int c) {
return isAlpha(c) || isDigit(c) || c == '-' || c == '.' || c == '_' || c == '~';
}
private static boolean isReserved(int c) {
return isGenericDelimiter(c) || isSubDelimiter(c);
}
private boolean isPchar(int c) {
return isUnreserved(c) || isSubDelimiter(c) || c == ':' || c == '@';
}
}