sshkey 1.3.3 → 1.4.0

data/lib/sshkey.rb

@@ -193,6 +193,63 @@ class SSHKey
     key_object.to_text.match(/Private-Key:\s\((\d*)\sbit\)/)[1].to_i
   end
 
+  # Randomart
+  #
+  # Generate OpenSSH compatible ASCII art fingerprints
+  # See http://www.opensource.apple.com/source/OpenSSH/OpenSSH-175/openssh/key.c (key_fingerprint_randomart function)
+  #
+  # Example:
+  # +--[ RSA 2048]----+
+  # |o+ o..           |
+  # |..+.o            |
+  # | ooo             |
+  # |.++. o           |
+  # |+o+ +   S        |
+  # |.. + o .         |
+  # |  . + .          |
+  # |   . .           |
+  # |    Eo.          |
+  # +-----------------+
+  def randomart
+    fieldsize_x = 17
+    fieldsize_y = 9
+    x = fieldsize_x / 2
+    y = fieldsize_y / 2
+    raw_digest = Digest::MD5.digest(ssh_public_key_conversion)
+    num_bytes = raw_digest.bytesize
+
+    field = Array.new(fieldsize_x) { Array.new(fieldsize_y) {0} }
+
+    raw_digest.bytes.each do |byte|
+      4.times do
+        x += (byte & 0x1 != 0) ? 1 : -1
+        y += (byte & 0x2 != 0) ? 1 : -1
+
+        x = [[x, 0].max, fieldsize_x - 1].min
+        y = [[y, 0].max, fieldsize_y - 1].min
+
+        field[x][y] += 1 if (field[x][y] < num_bytes - 2)
+
+        byte >>= 2
+      end
+    end
+
+    field[fieldsize_x / 2][fieldsize_y / 2] = num_bytes - 1
+    field[x][y] = num_bytes
+    augmentation_string = " .o+=*BOX@%&#/^SE"
+    output = "+--#{sprintf("[%4s %4u]", type.upcase, bits)}----+\n"
+    fieldsize_y.times do |y|
+      output << "|"
+      fieldsize_x.times do |x|
+        output << augmentation_string[[field[x][y], num_bytes].min]
+      end
+      output << "|"
+      output << "\n"
+    end
+    output << "+#{"-" * fieldsize_x}+"
+    output
+  end
+
   private
 
   # SSH Public Key Conversion

data/lib/sshkey/version.rb

@@ -1,3 +1,3 @@
 class SSHKey
-  VERSION = "1.3.3"
+  VERSION = "1.4.0"
 end

data/test/sshkey_test.rb

@@ -87,6 +87,48 @@ EOF
   KEY2_SHA1_FINGERPRINT = "9a:52:78:2b:6b:cb:39:b7:85:ed:90:8a:28:62:aa:b3:98:88:e6:07"
   KEY3_SHA1_FINGERPRINT = "15:68:c6:72:ac:18:d1:fc:ab:a2:b7:b5:8c:d1:fe:8f:b9:ae:a9:47"
 
+  KEY1_RANDOMART = <<-EOF.rstrip
++--[ RSA 2048]----+
+|o+ o..           |
+|..+.o            |
+| ooo             |
+|.++. o           |
+|+o+ +   S        |
+|.. + o .         |
+|  . + .          |
+|   . .           |
+|    Eo.          |
++-----------------+
+EOF
+
+  KEY2_RANDOMART = <<-EOF.rstrip
++--[ RSA 2048]----+
+|  ..o..          |
+|   ..+ .         |
+|    o   .        |
+|     . o         |
+|    . o S .      |
+|     + o O o     |
+|      + + O .    |
+|       = o .     |
+|       .E        |
++-----------------+
+EOF
+
+  KEY3_RANDOMART = <<-EOF.rstrip
++--[ DSA 1024]----+
+|       .=o.      |
+|      .+.o .     |
+|    + =.o . .    |
+|   + * + . .     |
+|    + = S . E    |
+|     + = . .     |
+|      .          |
+|                 |
+|                 |
++-----------------+
+EOF
+
   def setup
     @key1 = SSHKey.new(SSH_PRIVATE_KEY1, :comment => "me@example.com")
     @key2 = SSHKey.new(SSH_PRIVATE_KEY2, :comment => "me@example.com")
@@ -224,7 +266,13 @@ EOF
     assert_equal 2048, @key1.bits
     assert_equal 2048, @key2.bits
     assert_equal 1024, @key3.bits
-    assert_equal 512, SSHKey.generate(bits: 512).bits
+    assert_equal 512, SSHKey.generate(:bits => 512).bits
+  end
+
+  def test_randomart
+    assert_equal KEY1_RANDOMART, @key1.randomart
+    assert_equal KEY2_RANDOMART, @key2.randomart
+    assert_equal KEY3_RANDOMART, @key3.randomart
   end
 
   def test_to_byte_array

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sshkey
 version: !ruby/object:Gem::Version
-  version: 1.3.3
+  version: 1.4.0
   prerelease: 
 platform: ruby
 authors: