/server/src/main/java/com/orientechnologies/orient/server/network/protocol/http/multipart/OHttpMultipartRequestCommand.java

https://github.com/cloudsmith/orientdb · Java · 257 lines · 217 code · 20 blank · 20 comment · 54 complexity · b5c6d4620b18bc8d0b404a6214cb2ffc MD5 · raw file 

    

  1. /*

    2. 

  2.  *

    3. 

  3.  * Copyright 2011 Luca Molino (luca.molino--AT--assetdata.it)

    4. 

  4.  *

    5. 

  5.  * Licensed under the Apache License, Version 2.0 (the "License");

    6. 

  6.  * you may not use this file except in compliance with the License.

    7. 

  7.  * You may obtain a copy of the License at

    8. 

  8.  *

    9. 

  9.  *     http://www.apache.org/licenses/LICENSE-2.0

    10. 

  10.  *

    11. 

  11.  * Unless required by applicable law or agreed to in writing, software

    12. 

  12.  * distributed under the License is distributed on an "AS IS" BASIS,

    13. 

  13.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

    14. 

  14.  * See the License for the specific language governing permissions and

    15. 

  15.  * limitations under the License.

    16. 

  16.  */

    17. 

  17. package com.orientechnologies.orient.server.network.protocol.http.multipart;

    18. 

  18. 

    19. 

  19. import java.io.IOException;

    20. 

  20. import java.util.HashMap;

    21. 

  21. import java.util.LinkedHashMap;

    22. 

  22. 

    23. 

  23. import com.orientechnologies.orient.core.db.record.ODatabaseRecord;

    24. 

  24. import com.orientechnologies.orient.server.network.protocol.http.OHttpRequest;

    25. 

  25. import com.orientechnologies.orient.server.network.protocol.http.OHttpUtils;

    26. 

  26. import com.orientechnologies.orient.server.network.protocol.http.command.OServerCommandAuthenticatedDbAbstract;

    27. 

  27. 

    28. 

  28. /**

    29. 

  29.  * @author luca.molino

    30. 

  30.  * 

    31. 

  31.  */

    32. 

  32. public abstract class OHttpMultipartRequestCommand<B, F> extends OServerCommandAuthenticatedDbAbstract {

    33. 

  33. 

    34. 

  34. 	private STATUS	parseStatus	= STATUS.STATUS_EXPECTED_BOUNDARY;

    35. 

  35. 

    36. 

  36. 	protected static enum STATUS {

    37. 

  37. 		STATUS_EXPECTED_BOUNDARY, STATUS_EXPECTED_BOUNDARY_CRLF, STATUS_EXPECTED_PART_HEADERS, STATUS_EXPECTED_PART_CONTENT, STATUS_EXPECTED_END_REQUEST

    38. 

  38. 	}

    39. 

  39. 

    40. 

  40. 	public void parse(OHttpRequest iRequest, OHttpMultipartContentParser<B> standardContentParser,

    41. 

  41. 			OHttpMultipartContentParser<F> fileContentParser, ODatabaseRecord database) throws IOException {

    42. 

  42. 		int in;

    43. 

  43. 		char currChar;

    44. 

  44. 		boolean endRequest = false;

    45. 

  45. 		OHttpMultipartContentInputStream contentIn = new OHttpMultipartContentInputStream(iRequest.multipartStream, iRequest.boundary);

    46. 

  46. 		HashMap<String, String> headers = new LinkedHashMap<String, String>();

    47. 

  47. 		try {

    48. 

  48. 			while (iRequest.multipartStream.available() > 0 && !endRequest) {

    49. 

  49. 				in = iRequest.multipartStream.read();

    50. 

  50. 				currChar = (char) in;

    51. 

  51. 				switch (parseStatus) {

    52. 

  52. 				case STATUS_EXPECTED_BOUNDARY: {

    53. 

  53. 					readBoundary(iRequest, currChar);

    54. 

  54. 					parseStatus = STATUS.STATUS_EXPECTED_BOUNDARY_CRLF;

    55. 

  55. 					break;

    56. 

  56. 				}

    57. 

  57. 

    58. 

  58. 				case STATUS_EXPECTED_BOUNDARY_CRLF: {

    59. 

  59. 					endRequest = readBoundaryCrLf(iRequest, currChar, endRequest);

    60. 

  60. 					parseStatus = STATUS.STATUS_EXPECTED_PART_HEADERS;

    61. 

  61. 					break;

    62. 

  62. 				}

    63. 

  63. 

    64. 

  64. 				case STATUS_EXPECTED_PART_HEADERS: {

    65. 

  65. 					parsePartHeaders(iRequest, currChar, endRequest, headers);

    66. 

  66. 					parseStatus = STATUS.STATUS_EXPECTED_PART_CONTENT;

    67. 

  67. 					break;

    68. 

  68. 				}

    69. 

  69. 

    70. 

  70. 				case STATUS_EXPECTED_PART_CONTENT: {

    71. 

  71. 					iRequest.multipartStream.setSkipInput(in);

    72. 

  72. 					contentIn.reset();

    73. 

  73. 					if (headers.get(OHttpUtils.MULTIPART_CONTENT_FILENAME) != null) {

    74. 

  74. 						parseFileContent(iRequest, fileContentParser, headers, contentIn, database);

    75. 

  75. 					} else {

    76. 

  76. 						parseBaseContent(iRequest, standardContentParser, headers, contentIn, database);

    77. 

  77. 					}

    78. 

  78. 					break;

    79. 

  79. 				}

    80. 

  80. 

    81. 

  81. 				case STATUS_EXPECTED_END_REQUEST: {

    82. 

  82. 					iRequest.multipartStream.setSkipInput(in);

    83. 

  83. 					endRequest = OHttpMultipartHelper.isEndRequest(iRequest);

    84. 

  84. 					if (!endRequest) {

    85. 

  85. 						parseStatus = STATUS.STATUS_EXPECTED_BOUNDARY_CRLF;

    86. 

  86. 					} else {

    87. 

  87. 						parseStatus = STATUS.STATUS_EXPECTED_BOUNDARY;

    88. 

  88. 					}

    89. 

  89. 					break;

    90. 

  90. 				}

    91. 

  91. 				}

    92. 

  92. 			}

    93. 

  93. 			parseStatus = STATUS.STATUS_EXPECTED_BOUNDARY;

    94. 

  94. 		} catch (Exception e) {

    95. 

  95. 			sendTextContent(iRequest, OHttpUtils.STATUS_INTERNALERROR, e.getMessage(), null, OHttpUtils.CONTENT_TEXT_PLAIN,

    96. 

  96. 					e.getMessage());

    97. 

  97. 		}

    98. 

  98. 	}

    99. 

  99. 

    100. 

  100. 	protected boolean readBoundaryCrLf(final OHttpRequest iRequest, char currChar, boolean endRequest) throws IOException {

    101. 

  101. 		int in;

    102. 

  102. 		if (currChar == '\r') {

    103. 

  103. 			in = iRequest.multipartStream.read();

    104. 

  104. 			currChar = (char) in;

    105. 

  105. 			if (currChar == '\n') {

    106. 

  106. 				return false;

    107. 

  107. 			}

    108. 

  108. 		} else if (currChar == '-') {

    109. 

  109. 			in = iRequest.multipartStream.read();

    110. 

  110. 			currChar = (char) in;

    111. 

  111. 			if (currChar == '-') {

    112. 

  112. 				endRequest = true;

    113. 

  113. 			} else {

    114. 

  114. 				sendTextContent(iRequest, OHttpUtils.STATUS_INVALIDMETHOD_CODE, "Wrong request: Expected -", null,

    115. 

  115. 						OHttpUtils.CONTENT_TEXT_PLAIN, "Wrong request: Expected -");

    116. 

  116. 				endRequest = true;

    117. 

  117. 			}

    118. 

  118. 		} else {

    119. 

  119. 			sendTextContent(iRequest, OHttpUtils.STATUS_INVALIDMETHOD_CODE, "Wrong request: Expected CR/LF", null,

    120. 

  120. 					OHttpUtils.CONTENT_TEXT_PLAIN, "Wrong request: Expected CR/LF");

    121. 

  121. 			endRequest = true;

    122. 

  122. 		}

    123. 

  123. 		return endRequest;

    124. 

  124. 	}

    125. 

  125. 

    126. 

  126. 	protected void readBoundary(final OHttpRequest iRequest, char currChar) throws IOException {

    127. 

  127. 		int in;

    128. 

  128. 		int boundaryCursor = 0;

    129. 

  129. 		for (int i = 0; i < 2; i++) {

    130. 

  130. 			if (currChar != '-') {

    131. 

  131. 				sendTextContent(iRequest, OHttpUtils.STATUS_INVALIDMETHOD_CODE, "Wrong request: Expected boundary", null,

    132. 

  132. 						OHttpUtils.CONTENT_TEXT_PLAIN, "Wrong request: Expected boundary");

    133. 

  133. 				return;

    134. 

  134. 			}

    135. 

  135. 			in = iRequest.multipartStream.read();

    136. 

  136. 			currChar = (char) in;

    137. 

  137. 		}

    138. 

  138. 		while (boundaryCursor < iRequest.boundary.length()) {

    139. 

  139. 			if (currChar != iRequest.boundary.charAt(boundaryCursor)) {

    140. 

  140. 				sendTextContent(iRequest, OHttpUtils.STATUS_INVALIDMETHOD_CODE, "Wrong request: Expected boundary", null,

    141. 

  141. 						OHttpUtils.CONTENT_TEXT_PLAIN, "Wrong request: Expected boundary");

    142. 

  142. 			}

    143. 

  143. 			boundaryCursor++;

    144. 

  144. 			if (boundaryCursor < iRequest.boundary.length()) {

    145. 

  145. 				in = iRequest.multipartStream.read();

    146. 

  146. 				currChar = (char) in;

    147. 

  147. 			}

    148. 

  148. 		}

    149. 

  149. 	}

    150. 

  150. 

    151. 

  151. 	protected void parsePartHeaders(final OHttpRequest iRequest, char currChar, boolean endRequest,

    152. 

  152. 			final HashMap<String, String> headers) throws IOException {

    153. 

  153. 		int in;

    154. 

  154. 		StringBuilder headerName = new StringBuilder();

    155. 

  155. 		boolean endOfHeaders = false;

    156. 

  156. 		while (!endOfHeaders) {

    157. 

  157. 			headerName.append(currChar);

    158. 

  158. 			if (OHttpMultipartHelper.isMultipartPartHeader(headerName)) {

    159. 

  159. 				currChar = parseHeader(iRequest, headers, headerName.toString());

    160. 

  160. 				headerName.setLength(0);

    161. 

  161. 			}

    162. 

  162. 			if (currChar == '\r') {

    163. 

  163. 				in = iRequest.multipartStream.read();

    164. 

  164. 				currChar = (char) in;

    165. 

  165. 				if (currChar == '\n') {

    166. 

  166. 					in = iRequest.multipartStream.read();

    167. 

  167. 					currChar = (char) in;

    168. 

  168. 					if (currChar == '\r') {

    169. 

  169. 						in = iRequest.multipartStream.read();

    170. 

  170. 						currChar = (char) in;

    171. 

  171. 						if (currChar == '\n') {

    172. 

  172. 							endOfHeaders = true;

    173. 

  173. 						}

    174. 

  174. 					}

    175. 

  175. 				}

    176. 

  176. 			} else {

    177. 

  177. 				in = iRequest.multipartStream.read();

    178. 

  178. 				currChar = (char) in;

    179. 

  179. 			}

    180. 

  180. 		}

    181. 

  181. 	}

    182. 

  182. 

    183. 

  183. 	protected char parseHeader(final OHttpRequest iRequest, HashMap<String, String> headers, final String headerName)

    184. 

  184. 			throws IOException {

    185. 

  185. 		final StringBuilder header = new StringBuilder();

    186. 

  186. 		boolean endOfHeader = false;

    187. 

  187. 		int in;

    188. 

  188. 		char currChar;

    189. 

  189. 		in = iRequest.multipartStream.read();

    190. 

  190. 		currChar = (char) in;

    191. 

  191. 		if (currChar == ':') {

    192. 

  192. 			in = iRequest.multipartStream.read();

    193. 

  193. 			currChar = (char) in;

    194. 

  194. 			if (currChar != ' ') {

    195. 

  195. 				sendTextContent(iRequest, OHttpUtils.STATUS_INVALIDMETHOD_CODE, "Wrong request part header: Expected ' ' (header: "

    196. 

  196. 						+ headerName + ")", null, OHttpUtils.CONTENT_TEXT_PLAIN, "Wrong request part header: Expected ' ' (header: "

    197. 

  197. 						+ headerName + ")");

    198. 

  198. 			}

    199. 

  199. 		} else if (currChar != '=') {

    200. 

  200. 			sendTextContent(iRequest, OHttpUtils.STATUS_INVALIDMETHOD_CODE, "Wrong request part header: Expected ':' (header: "

    201. 

  201. 					+ headerName + ")", null, OHttpUtils.CONTENT_TEXT_PLAIN, "Wrong request part header: Expected ':' (header: " + headerName

    202. 

  202. 					+ ")");

    203. 

  203. 		}

    204. 

  204. 		while (!endOfHeader) {

    205. 

  205. 			in = iRequest.multipartStream.read();

    206. 

  206. 			currChar = (char) in;

    207. 

  207. 			if (currChar == ';') {

    208. 

  208. 				if (header.charAt(0) == '"') {

    209. 

  209. 					header.deleteCharAt(0);

    210. 

  210. 				}

    211. 

  211. 				if (header.charAt(header.length() - 1) == '"') {

    212. 

  212. 					header.deleteCharAt(header.length() - 1);

    213. 

  213. 				}

    214. 

  214. 				headers.put(headerName, header.toString());

    215. 

  215. 				in = iRequest.multipartStream.read();

    216. 

  216. 				return (char) in;

    217. 

  217. 			} else if (currChar == '\r') {

    218. 

  218. 				if (header.charAt(0) == '"') {

    219. 

  219. 					header.deleteCharAt(0);

    220. 

  220. 				}

    221. 

  221. 				if (header.charAt(header.length() - 1) == '"') {

    222. 

  222. 					header.deleteCharAt(header.length() - 1);

    223. 

  223. 				}

    224. 

  224. 				headers.put(headerName, header.toString());

    225. 

  225. 				return currChar;

    226. 

  226. 			}

    227. 

  227. 			header.append(currChar);

    228. 

  228. 		}

    229. 

  229. 		return currChar;

    230. 

  230. 	}

    231. 

  231. 

    232. 

  232. 	protected void parseBaseContent(final OHttpRequest iRequest, final OHttpMultipartContentParser<B> contentParser,

    233. 

  233. 			final HashMap<String, String> headers, final OHttpMultipartContentInputStream in, ODatabaseRecord database) throws Exception {

    234. 

  234. 		B result = contentParser.parse(iRequest, headers, in, database);

    235. 

  235. 		parseStatus = STATUS.STATUS_EXPECTED_END_REQUEST;

    236. 

  236. 		processBaseContent(iRequest, result, headers);

    237. 

  237. 		headers.clear();

    238. 

  238. 	}

    239. 

  239. 

    240. 

  240. 	protected void parseFileContent(final OHttpRequest iRequest, final OHttpMultipartContentParser<F> contentParser,

    241. 

  241. 			final HashMap<String, String> headers, final OHttpMultipartContentInputStream in, ODatabaseRecord database) throws Exception {

    242. 

  242. 		F result = contentParser.parse(iRequest, headers, in, database);

    243. 

  243. 		parseStatus = STATUS.STATUS_EXPECTED_END_REQUEST;

    244. 

  244. 		processFileContent(iRequest, result, headers);

    245. 

  245. 		headers.clear();

    246. 

  246. 	}

    247. 

  247. 

    248. 

  248. 	protected abstract void processBaseContent(final OHttpRequest iRequest, B iContentResult, HashMap<String, String> headers)

    249. 

  249. 			throws Exception;

    250. 

  250. 

    251. 

  251. 	protected abstract void processFileContent(final OHttpRequest iRequest, F iContentResult, HashMap<String, String> headers)

    252. 

  252. 			throws Exception;

    253. 

  253. 

    254. 

  254. 	protected abstract String getFileParamenterName();

    255. 

  255. 

    256. 

  256. 	protected abstract String getDocumentParamenterName();

    257. 

  257. }
    258.