1 /*
2 * Copyright (C) 2008-2009, Google Inc.
3 * Copyright (C) 2009, Robin Rosenberg <robin.rosenberg@dewire.com>
4 * and other copyright owners as documented in the project's IP log.
5 *
6 * This program and the accompanying materials are made available
7 * under the terms of the Eclipse Distribution License v1.0 which
8 * accompanies this distribution, is reproduced below, and is
9 * available at http://www.eclipse.org/org/documents/edl-v10.php
10 *
11 * All rights reserved.
12 *
13 * Redistribution and use in source and binary forms, with or
14 * without modification, are permitted provided that the following
15 * conditions are met:
16 *
17 * - Redistributions of source code must retain the above copyright
18 * notice, this list of conditions and the following disclaimer.
19 *
20 * - Redistributions in binary form must reproduce the above
21 * copyright notice, this list of conditions and the following
22 * disclaimer in the documentation and/or other materials provided
23 * with the distribution.
24 *
25 * - Neither the name of the Eclipse Foundation, Inc. nor the
26 * names of its contributors may be used to endorse or promote
27 * products derived from this software without specific prior
28 * written permission.
29 *
30 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
31 * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
32 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
33 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
34 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
35 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
36 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
37 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
38 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
39 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
40 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
41 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
42 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
43 */
44
45 package org.eclipse.jgit.transport;
46
47 import java.io.IOException;
48
49 import org.eclipse.jgit.lib.Config;
50 import org.eclipse.jgit.lib.Repository;
51 import org.eclipse.jgit.lib.Config.SectionParser;
52 import org.eclipse.jgit.transport.resolver.ServiceNotAuthorizedException;
53 import org.eclipse.jgit.transport.resolver.ServiceNotEnabledException;
54
55 /** A service exposed by {@link Daemon} over anonymous <code>git://</code>. */
56 public abstract class DaemonService {
57 private final String command;
58
59 private final SectionParser<ServiceConfig> configKey;
60
61 private boolean enabled;
62
63 private boolean overridable;
64
65 DaemonService(final String cmdName, final String cfgName) {
66 command = cmdName.startsWith("git-") ? cmdName : "git-" + cmdName; //$NON-NLS-1$ //$NON-NLS-2$
67 configKey = new SectionParser<ServiceConfig>() {
68 public ServiceConfig parse(final Config cfg) {
69 return new ServiceConfig(DaemonService.this, cfg, cfgName);
70 }
71 };
72 overridable = true;
73 }
74
75 private static class ServiceConfig {
76 final boolean enabled;
77
78 ServiceConfig(final DaemonService service, final Config cfg,
79 final String name) {
80 enabled = cfg.getBoolean("daemon", name, service.isEnabled()); //$NON-NLS-1$
81 }
82 }
83
84 /** @return is this service enabled for invocation? */
85 public boolean isEnabled() {
86 return enabled;
87 }
88
89 /**
90 * @param on
91 * true to allow this service to be used; false to deny it.
92 */
93 public void setEnabled(final boolean on) {
94 enabled = on;
95 }
96
97 /** @return can this service be configured in the repository config file? */
98 public boolean isOverridable() {
99 return overridable;
100 }
101
102 /**
103 * @param on
104 * true to permit repositories to override this service's enabled
105 * state with the <code>daemon.servicename</code> config setting.
106 */
107 public void setOverridable(final boolean on) {
108 overridable = on;
109 }
110
111 /** @return name of the command requested by clients. */
112 public String getCommandName() {
113 return command;
114 }
115
116 /**
117 * Determine if this service can handle the requested command.
118 *
119 * @param commandLine
120 * input line from the client.
121 * @return true if this command can accept the given command line.
122 */
123 public boolean handles(final String commandLine) {
124 return command.length() + 1 < commandLine.length()
125 && commandLine.charAt(command.length()) == ' '
126 && commandLine.startsWith(command);
127 }
128
129 void execute(final DaemonClient client, final String commandLine)
130 throws IOException, ServiceNotEnabledException,
131 ServiceNotAuthorizedException {
132 final String name = commandLine.substring(command.length() + 1);
133 Repository db;
134 try {
135 db = client.getDaemon().openRepository(client, name);
136 } catch (ServiceMayNotContinueException e) {
137 // An error when opening the repo means the client is expecting a ref
138 // advertisement, so use that style of error.
139 PacketLineOut pktOut = new PacketLineOut(client.getOutputStream());
140 pktOut.writeString("ERR " + e.getMessage() + "\n"); //$NON-NLS-1$ //$NON-NLS-2$
141 db = null;
142 }
143 if (db == null)
144 return;
145 try {
146 if (isEnabledFor(db))
147 execute(client, db);
148 } finally {
149 db.close();
150 }
151 }
152
153 private boolean isEnabledFor(final Repository db) {
154 if (isOverridable())
155 return db.getConfig().get(configKey).enabled;
156 return isEnabled();
157 }
158
159 abstract void execute(DaemonClient client, Repository db)
160 throws IOException, ServiceNotEnabledException,
161 ServiceNotAuthorizedException;
162 }