summaryrefslogtreecommitdiff
path: root/lib/rubygems/commands/cert_command.rb
blob: f5b698855bb4f2cd054182b2aa55ab0fa6d34b3f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
require 'rubygems/command'
require 'rubygems/security'

class Gem::Commands::CertCommand < Gem::Command

  def initialize
    super 'cert', 'Manage RubyGems certificates and signing settings'

    add_option('-a', '--add CERT',
               'Add a trusted certificate.') do |value, options|
      cert = OpenSSL::X509::Certificate.new(File.read(value))
      Gem::Security.add_trusted_cert(cert)
      say "Added '#{cert.subject.to_s}'"
    end

    add_option('-l', '--list',
               'List trusted certificates.') do |value, options|
      glob_str = File::join(Gem::Security::OPT[:trust_dir], '*.pem')
      Dir::glob(glob_str) do |path|
        begin
          cert = OpenSSL::X509::Certificate.new(File.read(path))
          # this could probably be formatted more gracefully
          say cert.subject.to_s
        rescue OpenSSL::X509::CertificateError
          next
        end
      end
    end

    add_option('-r', '--remove STRING',
               'Remove trusted certificates containing',
               'STRING.') do |value, options|
      trust_dir = Gem::Security::OPT[:trust_dir]
      glob_str = File::join(trust_dir, '*.pem')

      Dir::glob(glob_str) do |path|
        begin
          cert = OpenSSL::X509::Certificate.new(File.read(path))
          if cert.subject.to_s.downcase.index(value)
            say "Removed '#{cert.subject.to_s}'"
            File.unlink(path)
          end
        rescue OpenSSL::X509::CertificateError
          next
        end
      end
    end

    add_option('-b', '--build EMAIL_ADDR',
               'Build private key and self-signed',
               'certificate for EMAIL_ADDR.') do |value, options|
      vals = Gem::Security.build_self_signed_cert(value)
      File.chmod 0600, vals[:key_path]
      say "Public Cert: #{vals[:cert_path]}"
      say "Private Key: #{vals[:key_path]}"
      say "Don't forget to move the key file to somewhere private..."
    end

    add_option('-C', '--certificate CERT',
               'Certificate for --sign command.') do |value, options|
      cert = OpenSSL::X509::Certificate.new(File.read(value))
      Gem::Security::OPT[:issuer_cert] = cert
    end

    add_option('-K', '--private-key KEY',
               'Private key for --sign command.') do |value, options|
      key = OpenSSL::PKey::RSA.new(File.read(value))
      Gem::Security::OPT[:issuer_key] = key
    end

    add_option('-s', '--sign NEWCERT',
               'Sign a certificate with my key and',
               'certificate.') do |value, options|
      cert = OpenSSL::X509::Certificate.new(File.read(value))
      my_cert = Gem::Security::OPT[:issuer_cert]
      my_key = Gem::Security::OPT[:issuer_key]
      cert = Gem::Security.sign_cert(cert, my_key, my_cert)
      File.open(value, 'wb') { |file| file.write(cert.to_pem) }
    end
  end

  def execute
  end

end