class Net::SMTP

# File net/smtp.rb, line 176
def SMTP.default_port
  25
end

# File net/smtp.rb, line 194
def SMTP.default_ssl_context(verify_peer=true)
  context = OpenSSL::SSL::SSLContext.new
  context.verify_mode = verify_peer ? OpenSSL::SSL::VERIFY_PEER : OpenSSL::SSL::VERIFY_NONE
  store = OpenSSL::X509::Store.new
  store.set_default_paths
  context.cert_store = store
  context
end

# File net/smtp.rb, line 181
def SMTP.default_submission_port
  587
end

# File net/smtp.rb, line 186
def SMTP.default_tls_port
  465
end

# File net/smtp.rb, line 214
def initialize(address, port = nil)
  @address = address
  @port = (port || SMTP.default_port)
  @esmtp = true
  @capabilities = nil
  @socket = nil
  @started = false
  @open_timeout = 30
  @read_timeout = 60
  @error_occurred = false
  @debug_output = nil
  @tls = false
  @starttls = :auto
  @ssl_context_tls = nil
  @ssl_context_starttls = nil
end

# File net/smtp.rb, line 466
def SMTP.start(address, port = nil, *args, helo: nil,
               user: nil, secret: nil, password: nil, authtype: nil,
               tls_verify: true, tls_hostname: nil,
               &block)
  raise ArgumentError, "wrong number of arguments (given #{args.size + 2}, expected 1..6)" if args.size > 4
  helo ||= args[0] || 'localhost'
  user ||= args[1]
  secret ||= password || args[2]
  authtype ||= args[3]
  new(address, port).start(helo: helo, user: user, secret: secret, authtype: authtype, tls_verify: tls_verify, tls_hostname: tls_hostname, &block)
end

# File net/smtp.rb, line 791
def auth_cram_md5(user, secret)
  check_auth_args user, secret
  res = critical {
    res0 = get_response('AUTH CRAM-MD5')
    check_auth_continue res0
    crammed = cram_md5_response(secret, res0.cram_md5_challenge)
    get_response(base64_encode("#{user} #{crammed}"))
  }
  check_auth_response res
  res
end

# File net/smtp.rb, line 780
def auth_login(user, secret)
  check_auth_args user, secret
  res = critical {
    check_auth_continue get_response('AUTH LOGIN')
    check_auth_continue get_response(base64_encode(user))
    get_response(base64_encode(secret))
  }
  check_auth_response res
  res
end

# File net/smtp.rb, line 771
def auth_plain(user, secret)
  check_auth_args user, secret
  res = critical {
    get_response('AUTH PLAIN ' + base64_encode("\0#{user}\0#{secret}"))
  }
  check_auth_response res
  res
end

# File net/smtp.rb, line 765
def authenticate(user, secret, authtype = DEFAULT_AUTH_TYPE)
  check_auth_method authtype
  check_auth_args user, secret
  public_send auth_method(authtype), user, secret
end

# File net/smtp.rb, line 287
def capable_auth_types
  return [] unless @capabilities
  return [] unless @capabilities['AUTH']
  @capabilities['AUTH']
end

# File net/smtp.rb, line 274
def capable_cram_md5_auth?
  auth_capable?('CRAM-MD5')
end

# File net/smtp.rb, line 268
def capable_login_auth?
  auth_capable?('LOGIN')
end

# File net/smtp.rb, line 262
def capable_plain_auth?
  auth_capable?('PLAIN')
end

# File net/smtp.rb, line 250
def capable_starttls?
  capable?('STARTTLS')
end

# File net/smtp.rb, line 925
def data(msgstr = nil, &block)   #:yield: stream
  if msgstr and block
    raise ArgumentError, "message and block are exclusive"
  end
  unless msgstr or block
    raise ArgumentError, "message or block is required"
  end
  res = critical {
    check_continue get_response('DATA')
    socket_sync_bak = @socket.io.sync
    begin
      @socket.io.sync = false
      if msgstr
        @socket.write_message msgstr
      else
        @socket.write_message_by_block(&block)
      end
    ensure
      @socket.io.flush
      @socket.io.sync = socket_sync_bak
    end
    recv_response()
  }
  check_response res
  res
end

# File net/smtp.rb, line 400
def debug_output=(arg)
  @debug_output = arg
end

# File net/smtp.rb, line 358
def disable_starttls
  @starttls = false
  @ssl_context_starttls = nil
end

# File net/smtp.rb, line 314
def disable_tls
  @tls = false
  @ssl_context_tls = nil
end

# File net/smtp.rb, line 869
def ehlo(domain)
  getok("EHLO #{domain}")
end

# File net/smtp.rb, line 340
def enable_starttls(context = nil)
  raise 'openssl library not installed' unless defined?(OpenSSL)
  raise ArgumentError, "SMTPS and STARTTLS is exclusive" if @tls
  @starttls = :always
  @ssl_context_starttls = context
end

# File net/smtp.rb, line 349
def enable_starttls_auto(context = nil)
  raise 'openssl library not installed' unless defined?(OpenSSL)
  raise ArgumentError, "SMTPS and STARTTLS is exclusive" if @tls
  @starttls = :auto
  @ssl_context_starttls = context
end

# File net/smtp.rb, line 303
def enable_tls(context = nil)
  raise 'openssl library not installed' unless defined?(OpenSSL)
  raise ArgumentError, "SMTPS and STARTTLS is exclusive" if @starttls == :always
  @tls = true
  @ssl_context_tls = context
end

# File net/smtp.rb, line 570
def finish
  raise IOError, 'not yet started' unless started?
  do_finish
end

# File net/smtp.rb, line 865
def helo(domain)
  getok("HELO #{domain}")
end

# File net/smtp.rb, line 232
def inspect
  "#<#{self.class} #{@address}:#{@port} started=#{@started}>"
end

# File net/smtp.rb, line 873
def mailfrom(from_addr)
  getok("MAIL FROM:<#{from_addr}>")
end

# File net/smtp.rb, line 749
def open_message_stream(from_addr, *to_addrs, &block)   # :yield: stream
  raise IOError, 'closed session' unless @socket
  mailfrom from_addr
  rcptto_list(to_addrs) {data(&block)}
end

# File net/smtp.rb, line 952
def quit
  getok('QUIT')
end

# File net/smtp.rb, line 898
def rcptto(to_addr)
  getok("RCPT TO:<#{to_addr}>")
end

# File net/smtp.rb, line 877
def rcptto_list(to_addrs)
  raise ArgumentError, 'mail destination not given' if to_addrs.empty?
  ok_users = []
  unknown_users = []
  to_addrs.flatten.each do |addr|
    begin
      rcptto addr
    rescue SMTPAuthenticationError
      unknown_users << addr.dump
    else
      ok_users << addr
    end
  end
  raise ArgumentError, 'mail destination not given' if ok_users.empty?
  ret = yield
  unless unknown_users.empty?
    raise SMTPAuthenticationError, "failed to deliver for #{unknown_users.join(', ')}"
  end
  ret
end

# File net/smtp.rb, line 381
def read_timeout=(sec)
  @socket.read_timeout = sec if @socket
  @read_timeout = sec
end

# File net/smtp.rb, line 857
def rset
  getok('RSET')
end

# File net/smtp.rb, line 696
def send_message(msgstr, from_addr, *to_addrs)
  raise IOError, 'closed session' unless @socket
  mailfrom from_addr
  rcptto_list(to_addrs) {data msgstr}
end

# File net/smtp.rb, line 541
def start(*args, helo: nil,
          user: nil, secret: nil, password: nil, authtype: nil, tls_verify: true, tls_hostname: nil)
  raise ArgumentError, "wrong number of arguments (given #{args.size}, expected 0..4)" if args.size > 4
  helo ||= args[0] || 'localhost'
  user ||= args[1]
  secret ||= password || args[2]
  authtype ||= args[3]
  if @tls && @ssl_context_tls.nil?
    @ssl_context_tls = SMTP.default_ssl_context(tls_verify)
  end
  if @starttls && @ssl_context_starttls.nil?
    @ssl_context_starttls = SMTP.default_ssl_context(tls_verify)
  end
  @tls_hostname = tls_hostname
  if block_given?
    begin
      do_start helo, user, secret, authtype
      return yield(self)
    ensure
      do_finish
    end
  else
    do_start helo, user, secret, authtype
    return self
  end
end

# File net/smtp.rb, line 479
def started?
  @started
end

# File net/smtp.rb, line 861
def starttls
  getok('STARTTLS')
end

# File net/smtp.rb, line 324
def starttls?
  @starttls
end

# File net/smtp.rb, line 329
def starttls_always?
  @starttls == :always
end

# File net/smtp.rb, line 334
def starttls_auto?
  @starttls == :auto
end

# File net/smtp.rb, line 294
def tls?
  @tls
end

# File net/smtp.rb, line 278
def auth_capable?(type)
  return nil unless @capabilities
  return false unless @capabilities['AUTH']
  @capabilities['AUTH'].include?(type)
end

# File net/smtp.rb, line 811
def auth_method(type)
  "auth_#{type.to_s.downcase}".intern
end

# File net/smtp.rb, line 824
def base64_encode(str)
  # expects "str" may not become too long
  [str].pack('m0')
end

# File net/smtp.rb, line 254
def capable?(key)
  return nil unless @capabilities
  @capabilities[key] ? true : false
end

# File net/smtp.rb, line 815
def check_auth_args(user, secret, authtype = DEFAULT_AUTH_TYPE)
  unless user
    raise ArgumentError, 'SMTP-AUTH requested but missing user name'
  end
  unless secret
    raise ArgumentError, 'SMTP-AUTH requested but missing secret phrase'
  end
end

# File net/smtp.rb, line 1019
def check_auth_continue(res)
  unless res.continue?
    raise res.exception_class, res.message
  end
end

# File net/smtp.rb, line 805
def check_auth_method(type)
  unless respond_to?(auth_method(type), true)
    raise ArgumentError, "wrong authentication type #{type}"
  end
end

# File net/smtp.rb, line 1013
def check_auth_response(res)
  unless res.success?
    raise SMTPAuthenticationError, res.message
  end
end

# File net/smtp.rb, line 1007
def check_continue(res)
  unless res.continue?
    raise SMTPUnknownError, "could not get 3xx (#{res.status}: #{res.string})"
  end
end

# File net/smtp.rb, line 1001
def check_response(res)
  unless res.success?
    raise res.exception_class, res.message
  end
end

# File net/smtp.rb, line 833
def cram_md5_response(secret, challenge)
  tmp = Digest::MD5.digest(cram_secret(secret, IMASK) + challenge)
  Digest::MD5.hexdigest(cram_secret(secret, OMASK) + tmp)
end

# File net/smtp.rb, line 840
def cram_secret(secret, mask)
  secret = Digest::MD5.digest(secret) if secret.size > CRAM_BUFSIZE
  buf = secret.ljust(CRAM_BUFSIZE, "\0")
  0.upto(buf.size - 1) do |i|
    buf[i] = (buf[i].ord ^ mask).chr
  end
  buf
end

# File net/smtp.rb, line 991
def critical
  return Response.parse('200 dummy reply code') if @error_occurred
  begin
    return yield()
  rescue Exception
    @error_occurred = true
    raise
  end
end

# File net/smtp.rb, line 651
def do_finish
  quit if @socket and not @socket.closed? and not @error_occurred
ensure
  @started = false
  @error_occurred = false
  @socket.close if @socket
  @socket = nil
end

# File net/smtp.rb, line 639
def do_helo(helo_domain)
  res = @esmtp ? ehlo(helo_domain) : helo(helo_domain)
  @capabilities = res.capabilities
rescue SMTPError
  if @esmtp
    @esmtp = false
    @error_occurred = false
    retry
  end
  raise
end

# File net/smtp.rb, line 581
def do_start(helo_domain, user, secret, authtype)
  raise IOError, 'SMTP session already started' if @started
  if user or secret
    check_auth_method(authtype || DEFAULT_AUTH_TYPE)
    check_auth_args user, secret
  end
  s = Timeout.timeout(@open_timeout, Net::OpenTimeout) do
    tcp_socket(@address, @port)
  end
  logging "Connection opened: #{@address}:#{@port}"
  @socket = new_internet_message_io(tls? ? tlsconnect(s, @ssl_context_tls) : s)
  check_response critical { recv_response() }
  do_helo helo_domain
  if ! tls? and (starttls_always? or (capable_starttls? and starttls_auto?))
    unless capable_starttls?
      raise SMTPUnsupportedCommand,
          "STARTTLS is not supported on this server"
    end
    starttls
    @socket = new_internet_message_io(tlsconnect(s, @ssl_context_starttls))
    # helo response may be different after STARTTLS
    do_helo helo_domain
  end
  authenticate user, secret, (authtype || DEFAULT_AUTH_TYPE) if user
  @started = true
ensure
  unless @started
    # authentication failed, cancel connection.
    s.close if s
    @socket = nil
  end
end

# File net/smtp.rb, line 975
def get_response(reqline)
  validate_line reqline
  @socket.writeline reqline
  recv_response()
end

# File net/smtp.rb, line 965
def getok(reqline)
  validate_line reqline
  res = critical {
    @socket.writeline reqline
    recv_response()
  }
  check_response res
  res
end

# File net/smtp.rb, line 1104
def logging(msg)
  @debug_output << msg + "\n" if @debug_output
end

# File net/smtp.rb, line 634
def new_internet_message_io(s)
  InternetMessageIO.new(s, read_timeout: @read_timeout,
                        debug_output: @debug_output)
end

# File net/smtp.rb, line 981
def recv_response
  buf = ''.dup
  while true
    line = @socket.readline
    buf << line << "\n"
    break unless line[3,1] == '-'   # "210-PIPELINING"
  end
  Response.parse(buf)
end

# File net/smtp.rb, line 614
def ssl_socket(socket, context)
  OpenSSL::SSL::SSLSocket.new socket, context
end

# File net/smtp.rb, line 577
def tcp_socket(address, port)
  TCPSocket.open address, port
end

# File net/smtp.rb, line 618
def tlsconnect(s, context)
  verified = false
  s = ssl_socket(s, context)
  logging "TLS connection started"
  s.sync_close = true
  s.hostname = @tls_hostname || @address if s.respond_to? :hostname=
  ssl_socket_connect(s, @open_timeout)
  if context.verify_mode && context.verify_mode != OpenSSL::SSL::VERIFY_NONE
    s.post_connection_check(@tls_hostname || @address)
  end
  verified = true
  s
ensure
  s.close unless verified
end

# File net/smtp.rb, line 958
def validate_line(line)
  # A bare CR or LF is not allowed in RFC5321.
  if /[\r\n]/ =~ line
    raise ArgumentError, "A line must not contain CR or LF"
  end
end