View Javadoc
1   /*
2    * Copyright (C) 2008-2009, 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  
44  package org.eclipse.jgit.transport;
45  
46  import java.io.BufferedInputStream;
47  import java.io.IOException;
48  import java.io.InputStream;
49  import java.io.OutputStream;
50  import java.net.InetAddress;
51  import java.net.Socket;
52  
53  import org.eclipse.jgit.transport.resolver.ServiceNotAuthorizedException;
54  import org.eclipse.jgit.transport.resolver.ServiceNotEnabledException;
55  import org.eclipse.jgit.util.io.SafeBufferedOutputStream;
56  
57  /** Active network client of {@link Daemon}. */
58  public class DaemonClient {
59  	private final Daemon daemon;
60  
61  	private InetAddress peer;
62  
63  	private InputStream rawIn;
64  
65  	private OutputStream rawOut;
66  
67  	DaemonClient(final Daemon d) {
68  		daemon = d;
69  	}
70  
71  	void setRemoteAddress(final InetAddress ia) {
72  		peer = ia;
73  	}
74  
75  	/** @return the daemon which spawned this client. */
76  	public Daemon getDaemon() {
77  		return daemon;
78  	}
79  
80  	/** @return Internet address of the remote client. */
81  	public InetAddress getRemoteAddress() {
82  		return peer;
83  	}
84  
85  	/** @return input stream to read from the connected client. */
86  	public InputStream getInputStream() {
87  		return rawIn;
88  	}
89  
90  	/** @return output stream to send data to the connected client. */
91  	public OutputStream getOutputStream() {
92  		return rawOut;
93  	}
94  
95  	void execute(final Socket sock) throws IOException,
96  			ServiceNotEnabledException, ServiceNotAuthorizedException {
97  		rawIn = new BufferedInputStream(sock.getInputStream());
98  		rawOut = new SafeBufferedOutputStream(sock.getOutputStream());
99  
100 		if (0 < daemon.getTimeout())
101 			sock.setSoTimeout(daemon.getTimeout() * 1000);
102 		String cmd = new PacketLineIn(rawIn).readStringRaw();
103 		final int nul = cmd.indexOf('\0');
104 		if (nul >= 0) {
105 			// Newer clients hide a "host" header behind this byte.
106 			// Currently we don't use it for anything, so we ignore
107 			// this portion of the command.
108 			//
109 			cmd = cmd.substring(0, nul);
110 		}
111 
112 		final DaemonService srv = getDaemon().matchService(cmd);
113 		if (srv == null)
114 			return;
115 		sock.setSoTimeout(0);
116 		srv.execute(this, cmd);
117 	}
118 }