123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229 |
- /*
- * Copyright (C) 2009-2010, Google Inc.
- * and other copyright owners as documented in the project's IP log.
- *
- * This program and the accompanying materials are made available
- * under the terms of the Eclipse Distribution License v1.0 which
- * accompanies this distribution, is reproduced below, and is
- * available at http://www.eclipse.org/org/documents/edl-v10.php
- *
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or
- * without modification, are permitted provided that the following
- * conditions are met:
- *
- * - Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- *
- * - Redistributions in binary form must reproduce the above
- * copyright notice, this list of conditions and the following
- * disclaimer in the documentation and/or other materials provided
- * with the distribution.
- *
- * - Neither the name of the Eclipse Foundation, Inc. nor the
- * names of its contributors may be used to endorse or promote
- * products derived from this software without specific prior
- * written permission.
- *
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
- * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
- * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
- * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
- * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
- * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
- * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
- * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
- * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
- * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
- * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- */
-
- package org.eclipse.jgit.http.server;
-
- import static javax.servlet.http.HttpServletResponse.SC_BAD_REQUEST;
- import static javax.servlet.http.HttpServletResponse.SC_FORBIDDEN;
- import static javax.servlet.http.HttpServletResponse.SC_INTERNAL_SERVER_ERROR;
- import static javax.servlet.http.HttpServletResponse.SC_UNAUTHORIZED;
- import static javax.servlet.http.HttpServletResponse.SC_UNSUPPORTED_MEDIA_TYPE;
- import static org.eclipse.jgit.http.server.ClientVersionUtil.hasChunkedEncodingRequestBug;
- import static org.eclipse.jgit.http.server.ClientVersionUtil.parseVersion;
- import static org.eclipse.jgit.http.server.GitSmartHttpTools.UPLOAD_PACK;
- import static org.eclipse.jgit.http.server.GitSmartHttpTools.UPLOAD_PACK_REQUEST_TYPE;
- import static org.eclipse.jgit.http.server.GitSmartHttpTools.UPLOAD_PACK_RESULT_TYPE;
- import static org.eclipse.jgit.http.server.GitSmartHttpTools.sendError;
- import static org.eclipse.jgit.http.server.ServletUtils.ATTRIBUTE_HANDLER;
- import static org.eclipse.jgit.http.server.ServletUtils.consumeRequestBody;
- import static org.eclipse.jgit.http.server.ServletUtils.getInputStream;
- import static org.eclipse.jgit.http.server.ServletUtils.getRepository;
- import static org.eclipse.jgit.util.HttpSupport.HDR_USER_AGENT;
-
- import java.io.IOException;
- import java.text.MessageFormat;
- import java.util.List;
-
- import javax.servlet.Filter;
- import javax.servlet.FilterChain;
- import javax.servlet.FilterConfig;
- import javax.servlet.ServletException;
- import javax.servlet.ServletRequest;
- import javax.servlet.ServletResponse;
- import javax.servlet.http.HttpServlet;
- import javax.servlet.http.HttpServletRequest;
- import javax.servlet.http.HttpServletResponse;
-
- import org.eclipse.jgit.lib.Repository;
- import org.eclipse.jgit.transport.InternalHttpServerGlue;
- import org.eclipse.jgit.transport.RefAdvertiser.PacketLineOutRefAdvertiser;
- import org.eclipse.jgit.transport.ServiceMayNotContinueException;
- import org.eclipse.jgit.transport.UploadPack;
- import org.eclipse.jgit.transport.UploadPackInternalServerErrorException;
- import org.eclipse.jgit.transport.resolver.ServiceNotAuthorizedException;
- import org.eclipse.jgit.transport.resolver.ServiceNotEnabledException;
- import org.eclipse.jgit.transport.resolver.UploadPackFactory;
-
- /** Server side implementation of smart fetch over HTTP. */
- class UploadPackServlet extends HttpServlet {
- private static final long serialVersionUID = 1L;
-
- static class InfoRefs extends SmartServiceInfoRefs {
- private final UploadPackFactory<HttpServletRequest> uploadPackFactory;
-
- InfoRefs(UploadPackFactory<HttpServletRequest> uploadPackFactory,
- List<Filter> filters) {
- super(UPLOAD_PACK, filters);
- this.uploadPackFactory = uploadPackFactory;
- }
-
- @Override
- protected void begin(HttpServletRequest req, Repository db)
- throws IOException, ServiceNotEnabledException,
- ServiceNotAuthorizedException {
- UploadPack up = uploadPackFactory.create(req, db);
- InternalHttpServerGlue.setPeerUserAgent(
- up,
- req.getHeader(HDR_USER_AGENT));
- req.setAttribute(ATTRIBUTE_HANDLER, up);
- }
-
- @Override
- protected void advertise(HttpServletRequest req,
- PacketLineOutRefAdvertiser pck) throws IOException,
- ServiceNotEnabledException, ServiceNotAuthorizedException {
- UploadPack up = (UploadPack) req.getAttribute(ATTRIBUTE_HANDLER);
- try {
- up.setBiDirectionalPipe(false);
- up.sendAdvertisedRefs(pck);
- } finally {
- up.getRevWalk().close();
- }
- }
- }
-
- static class Factory implements Filter {
- private final UploadPackFactory<HttpServletRequest> uploadPackFactory;
-
- Factory(UploadPackFactory<HttpServletRequest> uploadPackFactory) {
- this.uploadPackFactory = uploadPackFactory;
- }
-
- @Override
- public void doFilter(ServletRequest request, ServletResponse response,
- FilterChain chain) throws IOException, ServletException {
- HttpServletRequest req = (HttpServletRequest) request;
- HttpServletResponse rsp = (HttpServletResponse) response;
- UploadPack rp;
- try {
- rp = uploadPackFactory.create(req, getRepository(req));
- } catch (ServiceNotAuthorizedException e) {
- rsp.sendError(SC_UNAUTHORIZED, e.getMessage());
- return;
- } catch (ServiceNotEnabledException e) {
- sendError(req, rsp, SC_FORBIDDEN, e.getMessage());
- return;
- }
-
- try {
- req.setAttribute(ATTRIBUTE_HANDLER, rp);
- chain.doFilter(req, rsp);
- } finally {
- req.removeAttribute(ATTRIBUTE_HANDLER);
- }
- }
-
- @Override
- public void init(FilterConfig filterConfig) throws ServletException {
- // Nothing.
- }
-
- @Override
- public void destroy() {
- // Nothing.
- }
- }
-
- /** {@inheritDoc} */
- @Override
- public void doPost(final HttpServletRequest req,
- final HttpServletResponse rsp) throws IOException {
- if (!UPLOAD_PACK_REQUEST_TYPE.equals(req.getContentType())) {
- rsp.sendError(SC_UNSUPPORTED_MEDIA_TYPE);
- return;
- }
-
- int[] version = parseVersion(req.getHeader(HDR_USER_AGENT));
- if (hasChunkedEncodingRequestBug(version, req)) {
- GitSmartHttpTools.sendError(req, rsp, SC_BAD_REQUEST, "\n\n"
- + HttpServerText.get().clientHas175ChunkedEncodingBug);
- return;
- }
-
- SmartOutputStream out = new SmartOutputStream(req, rsp, false) {
- @Override
- public void flush() throws IOException {
- doFlush();
- }
- };
-
- UploadPack up = (UploadPack) req.getAttribute(ATTRIBUTE_HANDLER);
- try {
- up.setBiDirectionalPipe(false);
- rsp.setContentType(UPLOAD_PACK_RESULT_TYPE);
-
- up.upload(getInputStream(req), out, null);
- out.close();
-
- } catch (ServiceMayNotContinueException e) {
- if (e.isOutput()) {
- consumeRequestBody(req);
- out.close();
- } else if (!rsp.isCommitted()) {
- rsp.reset();
- sendError(req, rsp, e.getStatusCode(), e.getMessage());
- }
- return;
-
- } catch (UploadPackInternalServerErrorException e) {
- // Special case exception, error message was sent to client.
- log(up.getRepository(), e.getCause());
- consumeRequestBody(req);
- out.close();
-
- } catch (Throwable e) {
- log(up.getRepository(), e);
- if (!rsp.isCommitted()) {
- rsp.reset();
- sendError(req, rsp, SC_INTERNAL_SERVER_ERROR);
- }
- return;
- }
- }
-
- private void log(Repository git, Throwable e) {
- getServletContext().log(MessageFormat.format(
- HttpServerText.get().internalErrorDuringUploadPack,
- ServletUtils.identify(git)), e);
- }
- }
|