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 }