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.BufferedOutputStream;
48 import java.io.IOException;
49 import java.io.InputStream;
50 import java.io.OutputStream;
51 import java.net.InetAddress;
52 import java.net.Socket;
53 import java.util.Arrays;
54 import java.util.Collection;
55
56 import org.eclipse.jgit.transport.resolver.ServiceNotAuthorizedException;
57 import org.eclipse.jgit.transport.resolver.ServiceNotEnabledException;
58
59 /**
60 * Active network client of {@link org.eclipse.jgit.transport.Daemon}.
61 */
62 public class DaemonClient {
63 private final Daemon daemon;
64
65 private InetAddress peer;
66
67 private InputStream rawIn;
68
69 private OutputStream rawOut;
70
71 DaemonClient(Daemon d) {
72 daemon = d;
73 }
74
75 void setRemoteAddress(InetAddress ia) {
76 peer = ia;
77 }
78
79 /**
80 * Get the daemon which spawned this client.
81 *
82 * @return the daemon which spawned this client.
83 */
84 public Daemon getDaemon() {
85 return daemon;
86 }
87
88 /**
89 * Get Internet address of the remote client.
90 *
91 * @return Internet address of the remote client.
92 */
93 public InetAddress getRemoteAddress() {
94 return peer;
95 }
96
97 /**
98 * Get input stream to read from the connected client.
99 *
100 * @return input stream to read from the connected client.
101 */
102 public InputStream getInputStream() {
103 return rawIn;
104 }
105
106 /**
107 * Get output stream to send data to the connected client.
108 *
109 * @return output stream to send data to the connected client.
110 */
111 public OutputStream getOutputStream() {
112 return rawOut;
113 }
114
115 void execute(Socket sock) throws IOException,
116 ServiceNotEnabledException, ServiceNotAuthorizedException {
117 rawIn = new BufferedInputStream(sock.getInputStream());
118 rawOut = new BufferedOutputStream(sock.getOutputStream());
119
120 if (0 < daemon.getTimeout())
121 sock.setSoTimeout(daemon.getTimeout() * 1000);
122 String cmd = new PacketLineIn(rawIn).readStringRaw();
123
124 Collection<String> extraParameters = null;
125
126 int nulnul = cmd.indexOf("\0\0"); //$NON-NLS-1$
127 if (nulnul != -1) {
128 extraParameters = Arrays.asList(cmd.substring(nulnul + 2).split("\0")); //$NON-NLS-1$
129 }
130
131 final int nul = cmd.indexOf('\0');
132 if (nul >= 0) {
133 // Newer clients hide a "host" header behind this byte.
134 // Currently we don't use it for anything, so we ignore
135 // this portion of the command.
136 //
137 cmd = cmd.substring(0, nul);
138 }
139
140 final DaemonService srv = getDaemon().matchService(cmd);
141 if (srv == null)
142 return;
143 sock.setSoTimeout(0);
144 srv.execute(this, cmd, extraParameters);
145 }
146 }