From 11eb4b51accebf8a163e8f46f99b067eda80f7b9 Mon Sep 17 00:00:00 2001 From: James Moger <james.moger@gitblit.com> Date: Fri, 23 May 2014 08:29:14 -0400 Subject: [PATCH] Remove GitBlit static singleton reference from localclone.groovy --- src/main/java/com/gitblit/transport/ssh/SshKey.java | 117 +++++++++++++++++++++++++++++++++++++++++++++++----------- 1 files changed, 94 insertions(+), 23 deletions(-) diff --git a/src/main/java/com/gitblit/transport/ssh/SshKey.java b/src/main/java/com/gitblit/transport/ssh/SshKey.java index cac6c41..ab44854 100644 --- a/src/main/java/com/gitblit/transport/ssh/SshKey.java +++ b/src/main/java/com/gitblit/transport/ssh/SshKey.java @@ -1,13 +1,31 @@ +/* + * Copyright 2014 gitblit.com. + * + * Licensed 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 com.gitblit.transport.ssh; import java.io.Serializable; import java.security.PublicKey; +import java.util.Arrays; +import java.util.List; import org.apache.commons.codec.binary.Base64; import org.apache.sshd.common.SshException; import org.apache.sshd.common.util.Buffer; import org.eclipse.jgit.lib.Constants; +import com.gitblit.Constants.AccessPermission; import com.gitblit.utils.StringUtils; /** @@ -28,13 +46,19 @@ private String fingerprint; + private String toString; + + private AccessPermission permission; + public SshKey(String data) { this.rawData = data; + this.permission = AccessPermission.PUSH; } public SshKey(PublicKey key) { this.publicKey = key; this.comment = ""; + this.permission = AccessPermission.PUSH; } public PublicKey getPublicKey() { @@ -48,7 +72,7 @@ try { publicKey = new Buffer(bin).getRawPublicKey(); } catch (SshException e) { - e.printStackTrace(); + throw new RuntimeException(e); } } return publicKey; @@ -71,6 +95,49 @@ public void setComment(String comment) { this.comment = comment; + if (rawData != null) { + rawData = null; + } + } + + /** + * Returns true if this key may be used to clone or fetch. + * + * @return true if this key can be used to clone or fetch + */ + public boolean canClone() { + return permission.atLeast(AccessPermission.CLONE); + } + + /** + * Returns true if this key may be used to push changes. + * + * @return true if this key can be used to push changes + */ + public boolean canPush() { + return permission.atLeast(AccessPermission.PUSH); + } + + /** + * Returns the access permission for the key. + * + * @return the access permission for the key + */ + public AccessPermission getPermission() { + return permission; + } + + /** + * Control the access permission assigned to this key. + * + * @param value + */ + public void setPermission(AccessPermission value) throws IllegalArgumentException { + List<AccessPermission> permitted = Arrays.asList(AccessPermission.SSHPERMISSIONS); + if (!permitted.contains(value)) { + throw new IllegalArgumentException("Illegal SSH public key permission specified: " + value); + } + this.permission = value; } public String getRawData() { @@ -96,12 +163,6 @@ public String getFingerprint() { if (fingerprint == null) { StringBuilder sb = new StringBuilder(); - // TODO append the keysize - int keySize = 0; - if (keySize > 0) { - sb.append(keySize).append(' '); - } - // append the key hash as colon-separated pairs String hash; if (rawData != null) { @@ -109,26 +170,13 @@ final byte [] bin = Base64.decodeBase64(Constants.encodeASCII(parts[1])); hash = StringUtils.getMD5(bin); } else { - // TODO get hash from publickey - hash = "todo"; + // TODO calculate the correct hash from a PublicKey instance + hash = StringUtils.getMD5(getPublicKey().getEncoded()); } for (int i = 0; i < hash.length(); i += 2) { sb.append(hash.charAt(i)).append(hash.charAt(i + 1)).append(':'); } sb.setLength(sb.length() - 1); - - // append the comment - String c = getComment(); - if (!StringUtils.isEmpty(c)) { - sb.append(' '); - sb.append(c); - } - - // append the algorithm - String alg = getAlgorithm(); - if (!StringUtils.isEmpty(alg)) { - sb.append(" (").append(alg).append(")"); - } fingerprint = sb.toString(); } return fingerprint; @@ -151,6 +199,29 @@ @Override public String toString() { - return getFingerprint(); + if (toString == null) { + StringBuilder sb = new StringBuilder(); + // TODO append the keysize + int keySize = 0; + if (keySize > 0) { + sb.append(keySize).append(' '); + } + // append fingerprint + sb.append(' '); + sb.append(getFingerprint()); + // append the comment + String c = getComment(); + if (!StringUtils.isEmpty(c)) { + sb.append(' '); + sb.append(c); + } + // append algorithm + String alg = getAlgorithm(); + if (!StringUtils.isEmpty(alg)) { + sb.append(" (").append(alg).append(")"); + } + toString = sb.toString(); + } + return toString; } } -- Gitblit v1.9.1