View Javadoc
1   package org.apache.maven.wagon.providers.ssh.knownhost;
2   
3   /*
4    * Licensed to the Apache Software Foundation (ASF) under one
5    * or more contributor license agreements.  See the NOTICE file
6    * distributed with this work for additional information
7    * regarding copyright ownership.  The ASF licenses this file
8    * to you under the Apache License, Version 2.0 (the
9    * "License"); you may not use this file except in compliance
10   * with the License.  You may obtain a copy of the License at
11   *
12   *   http://www.apache.org/licenses/LICENSE-2.0
13   *
14   * Unless required by applicable law or agreed to in writing,
15   * software distributed under the License is distributed on an
16   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
17   * KIND, either express or implied.  See the License for the
18   * specific language governing permissions and limitations
19   * under the License.
20   */
21  
22  import java.io.ByteArrayInputStream;
23  import java.io.File;
24  import java.io.FileInputStream;
25  import java.io.IOException;
26  import java.io.InputStream;
27  import java.util.Set;
28  
29  import org.codehaus.plexus.util.FileUtils;
30  
31  /**
32   * Provides known hosts from a file
33   *
34   * @author Juan F. Codagnone
35   * @since Sep 12, 2005
36   *
37   * @plexus.component role="org.apache.maven.wagon.providers.ssh.knownhost.KnownHostsProvider"
38   *    role-hint="file"
39   *    instantiation-strategy="per-lookup"
40   */
41  public class FileKnownHostsProvider
42      extends StreamKnownHostsProvider
43  {
44      private final File file;
45  
46      /**
47       * Creates the FileKnownHostsProvider.
48       *
49       * @param file the file that holds the known hosts, in the openssh format
50       * @throws IOException
51       */
52      public FileKnownHostsProvider( File file )
53          throws IOException
54      {
55          super( file.exists() ? (InputStream) new FileInputStream( file ) : new ByteArrayInputStream( "".getBytes() ) );
56          this.file = file;
57      }
58  
59      /**
60       * Creates a FileKnownHostsProvider using as file openssh knwon_host
61       *
62       * @throws IOException
63       * @see #FileKnownHostsProvider(File)
64       */
65      public FileKnownHostsProvider()
66          throws IOException
67      {
68          this( new File( System.getProperty( "user.home" ), ".ssh/known_hosts" ) );
69      }
70  
71      public void storeKnownHosts( String contents )
72          throws IOException
73      {
74          Set<KnownHostEntry> hosts = this.loadKnownHosts( contents );
75  
76          if ( ! this.knownHosts.equals( hosts ) )
77          {
78              file.getParentFile().mkdirs();
79              FileUtils.fileWrite( file.getAbsolutePath(), contents );
80              this.knownHosts = hosts;
81          }
82      }
83  
84      public void addKnownHost( KnownHostEntry knownHostEntry )
85          throws IOException
86      {
87          if ( !this.knownHosts.contains( knownHostEntry ) )
88          {
89              String knownHost = knownHostEntry.getHostName() + " " + knownHostEntry.getKeyType() + " "
90                  + knownHostEntry.getKeyValue() + "\n";
91              FileUtils.fileAppend( file.getAbsolutePath(), knownHost );
92          }
93      }
94  
95      public File getFile()
96      {
97          return file;
98      }
99  }