symbiosis-ssl 2.99 KB
Newer Older
1
2
3
4
5
6
#!/usr/bin/ruby
# 
# NAME
#   symbiosis-ssl - Manage and generate SSL certificates 
#
# SYNOPSIS
7
#   symbiosis-ssl [ --threshold N ] [ --no-generate ] [ --no-rollover ] [ --verbose ] [ --manual ]
8
9
10
#     [ --help ] [ DOMAIN DOMAIN ...]
#
# OPTIONS
11
12
#  --threshold N    Number days before expiry that certificates should be renewed.  Defaults to 10.
#
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
#  --no-generate    Do not try and generate keys or certificates.
#
#  --no-rollover    Do not try and generate keys or certificates.
#
#  --prefix PREFIX  Set the directory prefix for Symbiosis.  Defaults to /srv.
#
#   --help        Show the help information for this script.
#
#   --manual      Show the manual for this script
#
#   --verbose     Show debugging information.
#
# AUTHOR
#   Patrick J. Cherry <patrick@bytemark.co.uk>
#

#
#  Modules we require
#

require 'getoptlong'

opts = GetoptLong.new(
    [ '--help', '-h', GetoptLong::NO_ARGUMENT ],
    [ '--manual', '-m', GetoptLong::NO_ARGUMENT ],
    [ '--verbose', '-v', GetoptLong::NO_ARGUMENT ],
39
    [ '--threshold', '-t', GetoptLong::REQUIRED_ARGUMENT ],
40
41
42
43
44
45
46
    [ '--no-generate', '-G', GetoptLong::NO_ARGUMENT ],
    [ '--no-rollover', '-R', GetoptLong::NO_ARGUMENT ],
    [ '--prefix', '-p', GetoptLong::REQUIRED_ARGUMENT ]
)

manual = help = false
$VERBOSE = false
47
prefix = "/srv"
48
49
do_generate = do_rollover = true
threshold = 10
50
51
52

opts.each do |opt,arg|
  case opt
53
54
55
56
57
58
59
60
61
62
    when '--no-generate'
      do_generate = false
    when '--no-rollover'
      do_rollover = false
    when '--threshold'
      begin
        threshold = Integer(arg)
      rescue ArgumentError
        warn "** Could not parse #{arg.inspect} as an integer for --threshold"
      end
63
64
65
66
    when '--help'
      help = true
    when '--manual'
      manual = true
67
68
    when '--prefix'
      prefix = arg
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
    when '--verbose'
      $VERBOSE = true
  end
end

#
# Output help as required.
#
if help or manual
  require 'symbiosis/utils'
  Symbiosis::Utils.show_help(__FILE__) if help
  Symbiosis::Utils.show_manual(__FILE__) if manual
  exit 0
end

84
85
86
87
88
89
90
91
#
# The required spawn a massive stack of warnings in verbose mode.  So let's
# hide them.
#
v = $VERBOSE
$VERBOSE = false

require 'symbiosis/domains'
92
93
require 'symbiosis/domain/ssl'
require 'symbiosis/ssl'
94
95
96
97
98
99
100
101
require 'symbiosis/ssl/letsencrypt'
require 'symbiosis/ssl/selfsigned'

#
# And unhide.  Ugh.
#
$VERBOSE = v

102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119

domains = []

ARGV.each do |arg|
  domain = Symbiosis::Domains.find(arg.to_s, prefix)

  if domain.nil?
    warn "** Unable to find/parse domain #{arg.inspect}" 
    next
  end

  domains << domain
end

if ARGV.empty?
  domains = Symbiosis::Domains.all(prefix)
end

120
exit_code = 0
121

122
%w(INT TERM).each do |sig|
123
  trap(sig) do
124

125
126
127
    if 0 == Process.uid
      Process.euid = 0
      Process.egid = 0
128
129
    end

130
    exit 1
131
  end
132
end
133

134
now = Time.now
135

136
137
138
139
140
141
142
domains.each do |domain|
  begin
    domain.ssl_magic(threshold, do_generate, do_rollover, now) 
  rescue StandardError => err
    puts "\t!! Failed: #{err.to_s.gsub($/,'')}" if $VERBOSE
    puts err.backtrace.join("\n") if $DEBUG
    exit_code = 1
143
144
145
  end
end

146
147
exit exit_code