Merge pull request #1073 from gitblit/1062-DocEditorUpdates

Paul Martin

2016-04-30 a502d96a860456ec5e8c96761db70f7cabb74751

 src/main/java/com/gitblit/transport/ssh/SshKey.java

@@ -1,14 +1,34 @@
/*
 * 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.apache.sshd.common.util.buffer.Buffer;
import org.apache.sshd.common.util.buffer.ByteArrayBuffer;
import org.eclipse.jgit.lib.Constants;

import com.gitblit.Constants.AccessPermission;
import com.gitblit.utils.StringUtils;
import com.google.common.base.Joiner;

/**
 * Class that encapsulates a public SSH key and it's metadata.
@@ -30,13 +50,18 @@

   private String toString;

   private AccessPermission permission;

   public SshKey(String data) {
      this.rawData = data;
      // strip out line breaks (issue-571)
      this.rawData = Joiner.on("").join(data.replace("\r\n", "\n").split("\n"));
      this.permission = AccessPermission.PUSH;
   }

   public SshKey(PublicKey key) {
      this.publicKey = key;
      this.comment = "";
      this.permission = AccessPermission.PUSH;
   }

   public PublicKey getPublicKey() {
@@ -48,9 +73,9 @@
         }
         final byte[] bin = Base64.decodeBase64(Constants.encodeASCII(parts[1]));
         try {
            publicKey = new Buffer(bin).getRawPublicKey();
            publicKey = new ByteArrayBuffer(bin).getRawPublicKey();
         } catch (SshException e) {
            e.printStackTrace();
            throw new RuntimeException(e);
         }
      }
      return publicKey;
@@ -78,10 +103,50 @@
      }
   }

   /**
    * 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() {
      if (rawData == null && publicKey != null) {
         // build the raw data manually from the public key
         Buffer buf = new Buffer();
         Buffer buf = new ByteArrayBuffer();

         // 1: identify the algorithm
         buf.putRawPublicKey(publicKey);
@@ -108,8 +173,8 @@
            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(':');