001    /*
002     * Copyright (C) 2003-2009 eXo Platform SAS.
003     *
004     * This is free software; you can redistribute it and/or modify it
005     * under the terms of the GNU Lesser General Public License as
006     * published by the Free Software Foundation; either version 2.1 of
007     * the License, or (at your option) any later version.
008     *
009     * This software is distributed in the hope that it will be useful,
010     * but WITHOUT ANY WARRANTY; without even the implied warranty of
011     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
012     * Lesser General Public License for more details.
013     *
014     * You should have received a copy of the GNU Lesser General Public
015     * License along with this software; if not, write to the Free
016     * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
017     * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
018     */
019    package org.crsh.ssh.term;
020    
021    import org.apache.sshd.common.keyprovider.AbstractKeyPairProvider;
022    import org.apache.sshd.common.util.SecurityUtils;
023    import org.bouncycastle.openssl.PEMReader;
024    import org.slf4j.Logger;
025    import org.slf4j.LoggerFactory;
026    
027    import java.io.InputStream;
028    import java.io.InputStreamReader;
029    import java.net.URL;
030    import java.security.KeyPair;
031    import java.util.ArrayList;
032    import java.util.List;
033    
034    /**
035     * @author <a href="mailto:julien.viet@exoplatform.com">Julien Viet</a>
036     */
037    public class URLKeyPairProvider extends AbstractKeyPairProvider {
038    
039      /** . */
040      private static final Logger LOG = LoggerFactory.getLogger(URLKeyPairProvider.class);
041    
042      /** . */
043      private final URL url;
044    
045      public URLKeyPairProvider(URL url) {
046        this.url = url;
047      }
048    
049      @Override
050      protected KeyPair[] loadKeys() {
051        if (!SecurityUtils.isBouncyCastleRegistered()) {
052          throw new IllegalStateException("BouncyCastle must be registered as a JCE provider");
053        }
054        List<KeyPair> keys = new ArrayList<KeyPair>();
055        if (url != null) {
056          try {
057            InputStream in = url.openStream();
058            PEMReader r = new PEMReader(new InputStreamReader(in));
059            try {
060              Object o = r.readObject();
061              if (o instanceof KeyPair) {
062                keys.add((KeyPair) o);
063              }
064            } finally {
065              r.close();
066            }
067          } catch (Exception e) {
068            LOG.info("Unable to read key {}: {}", url, e);
069          }
070        }
071        return keys.toArray(new KeyPair[keys.size()]);
072      }
073    }