| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| package org.apache.sshd.agent.local; |
| |
| import java.io.IOException; |
| import java.util.Objects; |
| import java.util.UUID; |
| import java.util.concurrent.atomic.AtomicBoolean; |
| |
| import org.apache.sshd.agent.SshAgent; |
| import org.apache.sshd.agent.SshAgentServer; |
| import org.apache.sshd.common.session.ConnectionService; |
| import org.apache.sshd.common.session.Session; |
| import org.apache.sshd.common.util.logging.AbstractLoggingBean; |
| import org.apache.sshd.core.CoreModuleProperties; |
| |
| /** |
| * The server side fake agent, acting as an agent, but actually forwarding the requests to the auth channel on the |
| * client side. |
| */ |
| public class AgentServerProxy extends AbstractLoggingBean implements SshAgentServer { |
| private final ConnectionService service; |
| private final String id; |
| private final AtomicBoolean open = new AtomicBoolean(true); |
| |
| public AgentServerProxy(ConnectionService service) throws IOException { |
| this.service = Objects.requireNonNull(service, "No connection service provided"); |
| this.id = UUID.randomUUID().toString(); |
| } |
| |
| public SshAgent createClient() throws IOException { |
| try { |
| Session session = this.service.getSession(); |
| String channelType = CoreModuleProperties.PROXY_CHANNEL_TYPE.getRequired(session); |
| AgentForwardedChannel channel = new AgentForwardedChannel(channelType); |
| this.service.registerChannel(channel); |
| channel.open().verify(CoreModuleProperties.CHANNEL_OPEN_TIMEOUT.getRequired(channel)); |
| return channel.getAgent(); |
| } catch (Throwable t) { |
| if (log.isDebugEnabled()) { |
| log.warn("createClient({})[{}] failed to create client: {}", service.getSession(), getId(), t.toString()); |
| } |
| |
| if (t instanceof IOException) { |
| throw (IOException) t; |
| } |
| |
| throw new IOException("Failed (" + t.getClass().getSimpleName() + ") to create client: " + t.getMessage(), t); |
| } |
| } |
| |
| @Override |
| public String getId() { |
| return id; |
| } |
| |
| @Override |
| public boolean isOpen() { |
| return open.get(); |
| } |
| |
| @Override |
| public void close() throws IOException { |
| if (open.getAndSet(false)) { |
| if (log.isDebugEnabled()) { |
| log.debug("closed({})[{}]", service.getSession(), getId()); |
| } |
| } |
| } |
| } |