123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493 |
- /*
- * Copyright (C) 2008-2009, 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.transport;
-
- import java.io.IOException;
- import java.io.InputStream;
- import java.io.OutputStream;
- import java.net.InetAddress;
- import java.net.InetSocketAddress;
- import java.net.ServerSocket;
- import java.net.Socket;
- import java.net.SocketAddress;
- import java.net.SocketException;
- import java.util.concurrent.atomic.AtomicBoolean;
- import java.util.Collection;
-
- import org.eclipse.jgit.annotations.Nullable;
- import org.eclipse.jgit.errors.RepositoryNotFoundException;
- import org.eclipse.jgit.internal.JGitText;
- import org.eclipse.jgit.lib.PersonIdent;
- import org.eclipse.jgit.lib.Repository;
- import org.eclipse.jgit.storage.pack.PackConfig;
- import org.eclipse.jgit.transport.resolver.ReceivePackFactory;
- import org.eclipse.jgit.transport.resolver.RepositoryResolver;
- import org.eclipse.jgit.transport.resolver.ServiceNotAuthorizedException;
- import org.eclipse.jgit.transport.resolver.ServiceNotEnabledException;
- import org.eclipse.jgit.transport.resolver.UploadPackFactory;
-
- /**
- * Basic daemon for the anonymous <code>git://</code> transport protocol.
- */
- public class Daemon {
- /** 9418: IANA assigned port number for Git. */
- public static final int DEFAULT_PORT = 9418;
-
- private static final int BACKLOG = 5;
-
- private InetSocketAddress myAddress;
-
- private final DaemonService[] services;
-
- private final ThreadGroup processors;
-
- private Acceptor acceptThread;
-
- private int timeout;
-
- private PackConfig packConfig;
-
- private volatile RepositoryResolver<DaemonClient> repositoryResolver;
-
- volatile UploadPackFactory<DaemonClient> uploadPackFactory;
-
- volatile ReceivePackFactory<DaemonClient> receivePackFactory;
-
- /**
- * Configure a daemon to listen on any available network port.
- */
- public Daemon() {
- this(null);
- }
-
- /**
- * Configure a new daemon for the specified network address.
- *
- * @param addr
- * address to listen for connections on. If null, any available
- * port will be chosen on all network interfaces.
- */
- @SuppressWarnings("unchecked")
- public Daemon(InetSocketAddress addr) {
- myAddress = addr;
- processors = new ThreadGroup("Git-Daemon"); //$NON-NLS-1$
-
- repositoryResolver = (RepositoryResolver<DaemonClient>) RepositoryResolver.NONE;
-
- uploadPackFactory = new UploadPackFactory<DaemonClient>() {
- @Override
- public UploadPack create(DaemonClient req, Repository db)
- throws ServiceNotEnabledException,
- ServiceNotAuthorizedException {
- UploadPack up = new UploadPack(db);
- up.setTimeout(getTimeout());
- up.setPackConfig(getPackConfig());
- return up;
- }
- };
-
- receivePackFactory = new ReceivePackFactory<DaemonClient>() {
- @Override
- public ReceivePack create(DaemonClient req, Repository db)
- throws ServiceNotEnabledException,
- ServiceNotAuthorizedException {
- ReceivePack rp = new ReceivePack(db);
-
- InetAddress peer = req.getRemoteAddress();
- String host = peer.getCanonicalHostName();
- if (host == null)
- host = peer.getHostAddress();
- String name = "anonymous"; //$NON-NLS-1$
- String email = name + "@" + host; //$NON-NLS-1$
- rp.setRefLogIdent(new PersonIdent(name, email));
- rp.setTimeout(getTimeout());
-
- return rp;
- }
- };
-
- services = new DaemonService[] {
- new DaemonService("upload-pack", "uploadpack") { //$NON-NLS-1$ //$NON-NLS-2$
- {
- setEnabled(true);
- }
-
- @Override
- protected void execute(final DaemonClient dc,
- final Repository db,
- @Nullable Collection<String> extraParameters)
- throws IOException,
- ServiceNotEnabledException,
- ServiceNotAuthorizedException {
- UploadPack up = uploadPackFactory.create(dc, db);
- InputStream in = dc.getInputStream();
- OutputStream out = dc.getOutputStream();
- if (extraParameters != null) {
- up.setExtraParameters(extraParameters);
- }
- up.upload(in, out, null);
- }
- }, new DaemonService("receive-pack", "receivepack") { //$NON-NLS-1$ //$NON-NLS-2$
- {
- setEnabled(false);
- }
-
- @Override
- protected void execute(final DaemonClient dc,
- final Repository db,
- @Nullable Collection<String> extraParameters)
- throws IOException,
- ServiceNotEnabledException,
- ServiceNotAuthorizedException {
- ReceivePack rp = receivePackFactory.create(dc, db);
- InputStream in = dc.getInputStream();
- OutputStream out = dc.getOutputStream();
- rp.receive(in, out, null);
- }
- } };
- }
-
- /**
- * Get the address connections are received on.
- *
- * @return the address connections are received on.
- */
- public synchronized InetSocketAddress getAddress() {
- return myAddress;
- }
-
- /**
- * Lookup a supported service so it can be reconfigured.
- *
- * @param name
- * name of the service; e.g. "receive-pack"/"git-receive-pack" or
- * "upload-pack"/"git-upload-pack".
- * @return the service; null if this daemon implementation doesn't support
- * the requested service type.
- */
- public synchronized DaemonService getService(String name) {
- if (!name.startsWith("git-")) //$NON-NLS-1$
- name = "git-" + name; //$NON-NLS-1$
- for (DaemonService s : services) {
- if (s.getCommandName().equals(name))
- return s;
- }
- return null;
- }
-
- /**
- * Get timeout (in seconds) before aborting an IO operation.
- *
- * @return timeout (in seconds) before aborting an IO operation.
- */
- public int getTimeout() {
- return timeout;
- }
-
- /**
- * Set the timeout before willing to abort an IO call.
- *
- * @param seconds
- * number of seconds to wait (with no data transfer occurring)
- * before aborting an IO read or write operation with the
- * connected client.
- */
- public void setTimeout(int seconds) {
- timeout = seconds;
- }
-
- /**
- * Get configuration controlling packing, may be null.
- *
- * @return configuration controlling packing, may be null.
- */
- public PackConfig getPackConfig() {
- return packConfig;
- }
-
- /**
- * Set the configuration used by the pack generator.
- *
- * @param pc
- * configuration controlling packing parameters. If null the
- * source repository's settings will be used.
- */
- public void setPackConfig(PackConfig pc) {
- this.packConfig = pc;
- }
-
- /**
- * Set the resolver used to locate a repository by name.
- *
- * @param resolver
- * the resolver instance.
- */
- public void setRepositoryResolver(RepositoryResolver<DaemonClient> resolver) {
- repositoryResolver = resolver;
- }
-
- /**
- * Set the factory to construct and configure per-request UploadPack.
- *
- * @param factory
- * the factory. If null upload-pack is disabled.
- */
- @SuppressWarnings("unchecked")
- public void setUploadPackFactory(UploadPackFactory<DaemonClient> factory) {
- if (factory != null)
- uploadPackFactory = factory;
- else
- uploadPackFactory = (UploadPackFactory<DaemonClient>) UploadPackFactory.DISABLED;
- }
-
- /**
- * Get the factory used to construct per-request ReceivePack.
- *
- * @return the factory.
- * @since 4.3
- */
- public ReceivePackFactory<DaemonClient> getReceivePackFactory() {
- return receivePackFactory;
- }
-
- /**
- * Set the factory to construct and configure per-request ReceivePack.
- *
- * @param factory
- * the factory. If null receive-pack is disabled.
- */
- @SuppressWarnings("unchecked")
- public void setReceivePackFactory(ReceivePackFactory<DaemonClient> factory) {
- if (factory != null)
- receivePackFactory = factory;
- else
- receivePackFactory = (ReceivePackFactory<DaemonClient>) ReceivePackFactory.DISABLED;
- }
-
- private class Acceptor extends Thread {
-
- private final ServerSocket listenSocket;
-
- private final AtomicBoolean running = new AtomicBoolean(true);
-
- public Acceptor(ThreadGroup group, String name, ServerSocket socket) {
- super(group, name);
- this.listenSocket = socket;
- }
-
- @Override
- public void run() {
- setUncaughtExceptionHandler((thread, throwable) -> terminate());
- while (isRunning()) {
- try {
- startClient(listenSocket.accept());
- } catch (SocketException e) {
- // Test again to see if we should keep accepting.
- } catch (IOException e) {
- break;
- }
- }
-
- terminate();
- }
-
- private void terminate() {
- try {
- shutDown();
- } finally {
- clearThread();
- }
- }
-
- public boolean isRunning() {
- return running.get();
- }
-
- public void shutDown() {
- running.set(false);
- try {
- listenSocket.close();
- } catch (IOException err) {
- //
- }
- }
-
- }
-
- /**
- * Start this daemon on a background thread.
- *
- * @throws java.io.IOException
- * the server socket could not be opened.
- * @throws java.lang.IllegalStateException
- * the daemon is already running.
- */
- public synchronized void start() throws IOException {
- if (acceptThread != null) {
- throw new IllegalStateException(JGitText.get().daemonAlreadyRunning);
- }
- ServerSocket socket = new ServerSocket();
- socket.setReuseAddress(true);
- if (myAddress != null) {
- socket.bind(myAddress, BACKLOG);
- } else {
- socket.bind(new InetSocketAddress((InetAddress) null, 0), BACKLOG);
- }
- myAddress = (InetSocketAddress) socket.getLocalSocketAddress();
-
- acceptThread = new Acceptor(processors, "Git-Daemon-Accept", socket); //$NON-NLS-1$
- acceptThread.start();
- }
-
- private synchronized void clearThread() {
- acceptThread = null;
- }
-
- /**
- * Whether this daemon is receiving connections.
- *
- * @return {@code true} if this daemon is receiving connections.
- */
- public synchronized boolean isRunning() {
- return acceptThread != null && acceptThread.isRunning();
- }
-
- /**
- * Stop this daemon.
- */
- public synchronized void stop() {
- if (acceptThread != null) {
- acceptThread.shutDown();
- }
- }
-
- /**
- * Stops this daemon and waits until it's acceptor thread has finished.
- *
- * @throws java.lang.InterruptedException
- * if waiting for the acceptor thread is interrupted
- * @since 4.9
- */
- public void stopAndWait() throws InterruptedException {
- Thread acceptor = null;
- synchronized (this) {
- acceptor = acceptThread;
- stop();
- }
- if (acceptor != null) {
- acceptor.join();
- }
- }
-
- void startClient(Socket s) {
- final DaemonClient dc = new DaemonClient(this);
-
- final SocketAddress peer = s.getRemoteSocketAddress();
- if (peer instanceof InetSocketAddress)
- dc.setRemoteAddress(((InetSocketAddress) peer).getAddress());
-
- new Thread(processors, "Git-Daemon-Client " + peer.toString()) { //$NON-NLS-1$
- @Override
- public void run() {
- try {
- dc.execute(s);
- } catch (ServiceNotEnabledException e) {
- // Ignored. Client cannot use this repository.
- } catch (ServiceNotAuthorizedException e) {
- // Ignored. Client cannot use this repository.
- } catch (IOException e) {
- // Ignore unexpected IO exceptions from clients
- } finally {
- try {
- s.getInputStream().close();
- } catch (IOException e) {
- // Ignore close exceptions
- }
- try {
- s.getOutputStream().close();
- } catch (IOException e) {
- // Ignore close exceptions
- }
- }
- }
- }.start();
- }
-
- synchronized DaemonService matchService(String cmd) {
- for (DaemonService d : services) {
- if (d.handles(cmd))
- return d;
- }
- return null;
- }
-
- Repository openRepository(DaemonClient client, String name)
- throws ServiceMayNotContinueException {
- // Assume any attempt to use \ was by a Windows client
- // and correct to the more typical / used in Git URIs.
- //
- name = name.replace('\\', '/');
-
- // git://thishost/path should always be name="/path" here
- //
- if (!name.startsWith("/")) //$NON-NLS-1$
- return null;
-
- try {
- return repositoryResolver.open(client, name.substring(1));
- } catch (RepositoryNotFoundException e) {
- // null signals it "wasn't found", which is all that is suitable
- // for the remote client to know.
- return null;
- } catch (ServiceNotAuthorizedException e) {
- // null signals it "wasn't found", which is all that is suitable
- // for the remote client to know.
- return null;
- } catch (ServiceNotEnabledException e) {
- // null signals it "wasn't found", which is all that is suitable
- // for the remote client to know.
- return null;
- }
- }
- }
|