| /* |
| * 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.server.config.keys; |
| |
| import java.nio.file.Files; |
| import java.nio.file.LinkOption; |
| import java.nio.file.Path; |
| import java.security.KeyPair; |
| import java.util.ArrayList; |
| import java.util.Collection; |
| import java.util.EnumSet; |
| import java.util.Map; |
| import java.util.Properties; |
| |
| import org.apache.sshd.common.config.keys.BuiltinIdentities; |
| import org.apache.sshd.common.config.keys.IdentityUtils; |
| import org.apache.sshd.common.keyprovider.KeyPairProvider; |
| import org.apache.sshd.common.util.GenericUtils; |
| import org.apache.sshd.common.util.MapEntryUtils; |
| import org.apache.sshd.common.util.io.IoUtils; |
| import org.apache.sshd.util.test.BaseTestSupport; |
| import org.apache.sshd.util.test.NoIoTestCase; |
| import org.junit.FixMethodOrder; |
| import org.junit.Test; |
| import org.junit.experimental.categories.Category; |
| import org.junit.runners.MethodSorters; |
| |
| /** |
| * @author <a href="mailto:dev@mina.apache.org">Apache MINA SSHD Project</a> |
| */ |
| @FixMethodOrder(MethodSorters.NAME_ASCENDING) |
| @Category({ NoIoTestCase.class }) |
| public class ServerIdentityTest extends BaseTestSupport { |
| public ServerIdentityTest() { |
| super(); |
| } |
| |
| @Test |
| public void testLoadServerIdentities() throws Exception { |
| Path resFolder = getTestResourcesFolder(); |
| Collection<Path> paths = new ArrayList<>(BuiltinIdentities.VALUES.size()); |
| LinkOption[] options = IoUtils.getLinkOptions(true); |
| Collection<BuiltinIdentities> expected = EnumSet.noneOf(BuiltinIdentities.class); |
| for (BuiltinIdentities type : BuiltinIdentities.VALUES) { |
| String fileName = ServerIdentity.getIdentityFileName(type); |
| Path file = resFolder.resolve(fileName); |
| if (!Files.exists(file, options)) { |
| System.out.println("Skip non-existing identity file " + file); |
| continue; |
| } |
| |
| if (!type.isSupported()) { |
| System.out.println("Skip unsupported identity file " + file); |
| continue; |
| } |
| |
| paths.add(file); |
| expected.add(type); |
| } |
| |
| Properties props = new Properties(); |
| props.setProperty(ServerIdentity.HOST_KEY_CONFIG_PROP, GenericUtils.join(paths, ',')); |
| |
| Map<String, KeyPair> ids = ServerIdentity.loadIdentities(props, options); |
| assertEquals("Mismatched loaded ids count", GenericUtils.size(paths), MapEntryUtils.size(ids)); |
| |
| Collection<KeyPair> pairs = new ArrayList<>(ids.size()); |
| for (BuiltinIdentities type : BuiltinIdentities.VALUES) { |
| if (expected.contains(type)) { |
| KeyPair kp = ids.get(type.getName()); |
| assertNotNull("No key pair loaded for " + type, kp); |
| pairs.add(kp); |
| } |
| } |
| |
| KeyPairProvider provider = IdentityUtils.createKeyPairProvider(ids, true /* supported only */); |
| assertNotNull("No provider generated", provider); |
| |
| Iterable<KeyPair> keys = provider.loadKeys(null); |
| for (KeyPair kp : keys) { |
| assertTrue("Unexpected loaded key: " + kp, pairs.remove(kp)); |
| } |
| |
| assertEquals("Not all pairs listed", 0, pairs.size()); |
| } |
| } |