You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

UploadPackServlet.java 8.4KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239
  1. /*
  2. * Copyright (C) 2009-2010, Google Inc.
  3. * and other copyright owners as documented in the project's IP log.
  4. *
  5. * This program and the accompanying materials are made available
  6. * under the terms of the Eclipse Distribution License v1.0 which
  7. * accompanies this distribution, is reproduced below, and is
  8. * available at http://www.eclipse.org/org/documents/edl-v10.php
  9. *
  10. * All rights reserved.
  11. *
  12. * Redistribution and use in source and binary forms, with or
  13. * without modification, are permitted provided that the following
  14. * conditions are met:
  15. *
  16. * - Redistributions of source code must retain the above copyright
  17. * notice, this list of conditions and the following disclaimer.
  18. *
  19. * - Redistributions in binary form must reproduce the above
  20. * copyright notice, this list of conditions and the following
  21. * disclaimer in the documentation and/or other materials provided
  22. * with the distribution.
  23. *
  24. * - Neither the name of the Eclipse Foundation, Inc. nor the
  25. * names of its contributors may be used to endorse or promote
  26. * products derived from this software without specific prior
  27. * written permission.
  28. *
  29. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
  30. * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
  31. * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  32. * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  33. * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
  34. * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  35. * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  36. * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  37. * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
  38. * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  39. * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  40. * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  41. * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  42. */
  43. package org.eclipse.jgit.http.server;
  44. import static javax.servlet.http.HttpServletResponse.SC_FORBIDDEN;
  45. import static javax.servlet.http.HttpServletResponse.SC_INTERNAL_SERVER_ERROR;
  46. import static javax.servlet.http.HttpServletResponse.SC_UNAUTHORIZED;
  47. import static javax.servlet.http.HttpServletResponse.SC_UNSUPPORTED_MEDIA_TYPE;
  48. import static org.eclipse.jgit.http.server.GitSmartHttpTools.UPLOAD_PACK;
  49. import static org.eclipse.jgit.http.server.GitSmartHttpTools.UPLOAD_PACK_REQUEST_TYPE;
  50. import static org.eclipse.jgit.http.server.GitSmartHttpTools.UPLOAD_PACK_RESULT_TYPE;
  51. import static org.eclipse.jgit.http.server.GitSmartHttpTools.sendError;
  52. import static org.eclipse.jgit.http.server.ServletUtils.ATTRIBUTE_HANDLER;
  53. import static org.eclipse.jgit.http.server.ServletUtils.consumeRequestBody;
  54. import static org.eclipse.jgit.http.server.ServletUtils.getInputStream;
  55. import static org.eclipse.jgit.http.server.ServletUtils.getRepository;
  56. import static org.eclipse.jgit.util.HttpSupport.HDR_USER_AGENT;
  57. import java.io.IOException;
  58. import java.text.MessageFormat;
  59. import java.util.List;
  60. import javax.servlet.Filter;
  61. import javax.servlet.FilterChain;
  62. import javax.servlet.FilterConfig;
  63. import javax.servlet.ServletException;
  64. import javax.servlet.ServletRequest;
  65. import javax.servlet.ServletResponse;
  66. import javax.servlet.http.HttpServlet;
  67. import javax.servlet.http.HttpServletRequest;
  68. import javax.servlet.http.HttpServletResponse;
  69. import org.eclipse.jgit.lib.Repository;
  70. import org.eclipse.jgit.transport.InternalHttpServerGlue;
  71. import org.eclipse.jgit.transport.PacketLineOut;
  72. import org.eclipse.jgit.transport.RefAdvertiser.PacketLineOutRefAdvertiser;
  73. import org.eclipse.jgit.transport.ServiceMayNotContinueException;
  74. import org.eclipse.jgit.transport.UploadPack;
  75. import org.eclipse.jgit.transport.UploadPackInternalServerErrorException;
  76. import org.eclipse.jgit.transport.resolver.ServiceNotAuthorizedException;
  77. import org.eclipse.jgit.transport.resolver.ServiceNotEnabledException;
  78. import org.eclipse.jgit.transport.resolver.UploadPackFactory;
  79. /** Server side implementation of smart fetch over HTTP. */
  80. class UploadPackServlet extends HttpServlet {
  81. private static final long serialVersionUID = 1L;
  82. static class InfoRefs extends SmartServiceInfoRefs {
  83. private final UploadPackFactory<HttpServletRequest> uploadPackFactory;
  84. InfoRefs(UploadPackFactory<HttpServletRequest> uploadPackFactory,
  85. List<Filter> filters) {
  86. super(UPLOAD_PACK, filters);
  87. this.uploadPackFactory = uploadPackFactory;
  88. }
  89. @Override
  90. protected void begin(HttpServletRequest req, Repository db)
  91. throws IOException, ServiceNotEnabledException,
  92. ServiceNotAuthorizedException {
  93. UploadPack up = uploadPackFactory.create(req, db);
  94. InternalHttpServerGlue.setPeerUserAgent(
  95. up,
  96. req.getHeader(HDR_USER_AGENT));
  97. req.setAttribute(ATTRIBUTE_HANDLER, up);
  98. }
  99. @Override
  100. protected void advertise(HttpServletRequest req,
  101. PacketLineOutRefAdvertiser pck) throws IOException,
  102. ServiceNotEnabledException, ServiceNotAuthorizedException {
  103. UploadPack up = (UploadPack) req.getAttribute(ATTRIBUTE_HANDLER);
  104. try {
  105. up.setBiDirectionalPipe(false);
  106. up.sendAdvertisedRefs(pck);
  107. } finally {
  108. // TODO(jonathantanmy): Move responsibility for closing the
  109. // RevWalk to UploadPack, either by making it AutoCloseable
  110. // or by making sendAdvertisedRefs clean up after itself.
  111. up.getRevWalk().close();
  112. }
  113. }
  114. @Override
  115. protected void respond(HttpServletRequest req,
  116. PacketLineOut pckOut, String serviceName) throws IOException,
  117. ServiceNotEnabledException, ServiceNotAuthorizedException {
  118. UploadPack up = (UploadPack) req.getAttribute(ATTRIBUTE_HANDLER);
  119. try {
  120. up.setBiDirectionalPipe(false);
  121. up.sendAdvertisedRefs(new PacketLineOutRefAdvertiser(pckOut), serviceName);
  122. } finally {
  123. // TODO(jonathantanmy): Move responsibility for closing the
  124. // RevWalk to UploadPack, either by making it AutoCloseable
  125. // or by making sendAdvertisedRefs clean up after itself.
  126. up.getRevWalk().close();
  127. }
  128. }
  129. }
  130. static class Factory implements Filter {
  131. private final UploadPackFactory<HttpServletRequest> uploadPackFactory;
  132. Factory(UploadPackFactory<HttpServletRequest> uploadPackFactory) {
  133. this.uploadPackFactory = uploadPackFactory;
  134. }
  135. @Override
  136. public void doFilter(ServletRequest request, ServletResponse response,
  137. FilterChain chain) throws IOException, ServletException {
  138. HttpServletRequest req = (HttpServletRequest) request;
  139. HttpServletResponse rsp = (HttpServletResponse) response;
  140. UploadPack rp;
  141. try {
  142. rp = uploadPackFactory.create(req, getRepository(req));
  143. } catch (ServiceNotAuthorizedException e) {
  144. rsp.sendError(SC_UNAUTHORIZED, e.getMessage());
  145. return;
  146. } catch (ServiceNotEnabledException e) {
  147. sendError(req, rsp, SC_FORBIDDEN, e.getMessage());
  148. return;
  149. }
  150. try {
  151. req.setAttribute(ATTRIBUTE_HANDLER, rp);
  152. chain.doFilter(req, rsp);
  153. } finally {
  154. req.removeAttribute(ATTRIBUTE_HANDLER);
  155. }
  156. }
  157. @Override
  158. public void init(FilterConfig filterConfig) throws ServletException {
  159. // Nothing.
  160. }
  161. @Override
  162. public void destroy() {
  163. // Nothing.
  164. }
  165. }
  166. /** {@inheritDoc} */
  167. @Override
  168. public void doPost(final HttpServletRequest req,
  169. final HttpServletResponse rsp) throws IOException {
  170. if (!UPLOAD_PACK_REQUEST_TYPE.equals(req.getContentType())) {
  171. rsp.sendError(SC_UNSUPPORTED_MEDIA_TYPE);
  172. return;
  173. }
  174. SmartOutputStream out = new SmartOutputStream(req, rsp, false) {
  175. @Override
  176. public void flush() throws IOException {
  177. doFlush();
  178. }
  179. };
  180. UploadPack up = (UploadPack) req.getAttribute(ATTRIBUTE_HANDLER);
  181. try {
  182. up.setBiDirectionalPipe(false);
  183. rsp.setContentType(UPLOAD_PACK_RESULT_TYPE);
  184. up.upload(getInputStream(req), out, null);
  185. out.close();
  186. } catch (ServiceMayNotContinueException e) {
  187. if (e.isOutput()) {
  188. consumeRequestBody(req);
  189. out.close();
  190. } else if (!rsp.isCommitted()) {
  191. rsp.reset();
  192. sendError(req, rsp, e.getStatusCode(), e.getMessage());
  193. }
  194. return;
  195. } catch (UploadPackInternalServerErrorException e) {
  196. // Special case exception, error message was sent to client.
  197. log(up.getRepository(), e.getCause());
  198. consumeRequestBody(req);
  199. out.close();
  200. } catch (Throwable e) {
  201. log(up.getRepository(), e);
  202. if (!rsp.isCommitted()) {
  203. rsp.reset();
  204. sendError(req, rsp, SC_INTERNAL_SERVER_ERROR);
  205. }
  206. return;
  207. }
  208. }
  209. private void log(Repository git, Throwable e) {
  210. getServletContext().log(MessageFormat.format(
  211. HttpServerText.get().internalErrorDuringUploadPack,
  212. ServletUtils.identify(git)), e);
  213. }
  214. }