bmedia-casserver 1.1.1

data/lib/casserver/cas.rb

@@ -0,0 +1,323 @@
+require 'uri'
+require 'net/https'
+
+require 'casserver/model'
+
+# Encapsulates CAS functionality. This module is meant to be included in
+# the CASServer::Controllers module.
+module CASServer::CAS
+
+  include CASServer::Model
+
+  def generate_login_ticket
+    # 3.5 (login ticket)
+    lt = LoginTicket.new
+    lt.ticket = "LT-" + CASServer::Utils.random_string
+
+    lt.client_hostname = @env['HTTP_X_FORWARDED_FOR'] || @env['REMOTE_HOST'] || @env['REMOTE_ADDR']
+    lt.save!
+    $LOG.debug("Generated login ticket '#{lt.ticket}' for client" +
+      " at '#{lt.client_hostname}'")
+    lt
+  end
+
+  # Creates a TicketGrantingTicket for the given username. This is done when the user logs in
+  # for the first time to establish their SSO session (after their credentials have been validated).
+  #
+  # The optional 'extra_attributes' parameter takes a hash of additional attributes
+  # that will be sent along with the username in the CAS response to subsequent
+  # validation requests from clients.
+  def generate_ticket_granting_ticket(username, extra_attributes = {})
+    # 3.6 (ticket granting cookie/ticket)
+    tgt = TicketGrantingTicket.new
+    tgt.ticket = "TGC-" + CASServer::Utils.random_string
+    tgt.username = username
+    tgt.extra_attributes = extra_attributes
+    tgt.client_hostname = @env['HTTP_X_FORWARDED_FOR'] || @env['REMOTE_HOST'] || @env['REMOTE_ADDR']
+    tgt.save!
+    $LOG.debug("Generated ticket granting ticket '#{tgt.ticket}' for user" +
+      " '#{tgt.username}' at '#{tgt.client_hostname}'" +
+      (extra_attributes.blank? ? "" : " with extra attributes #{extra_attributes.inspect}"))
+    tgt
+  end
+
+  def generate_service_ticket(service, username, tgt)
+    # 3.1 (service ticket)
+    st = ServiceTicket.new
+    st.ticket = "ST-" + CASServer::Utils.random_string
+    st.service = service
+    st.username = username
+    st.granted_by_tgt_id = tgt.id
+    st.client_hostname = @env['HTTP_X_FORWARDED_FOR'] || @env['REMOTE_HOST'] || @env['REMOTE_ADDR']
+    st.save!
+    $LOG.debug("Generated service ticket '#{st.ticket}' for service '#{st.service}'" +
+      " for user '#{st.username}' at '#{st.client_hostname}'")
+    st
+  end
+
+  def generate_proxy_ticket(target_service, pgt)
+    # 3.2 (proxy ticket)
+    pt = ProxyTicket.new
+    pt.ticket = "PT-" + CASServer::Utils.random_string
+    pt.service = target_service
+    pt.username = pgt.service_ticket.username
+    pt.granted_by_pgt_id = pgt.id
+    pt.granted_by_tgt_id = pgt.service_ticket.granted_by_tgt.id
+    pt.client_hostname = @env['HTTP_X_FORWARDED_FOR'] || @env['REMOTE_HOST'] || @env['REMOTE_ADDR']
+    pt.save!
+    $LOG.debug("Generated proxy ticket '#{pt.ticket}' for target service '#{pt.service}'" +
+      " for user '#{pt.username}' at '#{pt.client_hostname}' using proxy-granting" +
+      " ticket '#{pgt.ticket}'")
+    pt
+  end
+
+  def generate_proxy_granting_ticket(pgt_url, st)
+    uri = URI.parse(pgt_url)
+    https = Net::HTTP.new(uri.host,uri.port)
+    https.use_ssl = true
+
+    # Here's what's going on here:
+    #
+    #   1. We generate a ProxyGrantingTicket (but don't store it in the database just yet)
+    #   2. Deposit the PGT and it's associated IOU at the proxy callback URL.
+    #   3. If the proxy callback URL responds with HTTP code 200, store the PGT and return it;
+    #      otherwise don't save it and return nothing.
+    #
+    https.start do |conn|
+      path = uri.path.empty? ? '/' : uri.path
+      path += '?' + uri.query unless (uri.query.nil? || uri.query.empty?)
+      
+      pgt = ProxyGrantingTicket.new
+      pgt.ticket = "PGT-" + CASServer::Utils.random_string(60)
+      pgt.iou = "PGTIOU-" + CASServer::Utils.random_string(57)
+      pgt.service_ticket_id = st.id
+      pgt.client_hostname = @env['HTTP_X_FORWARDED_FOR'] || @env['REMOTE_HOST'] || @env['REMOTE_ADDR']
+
+      # FIXME: The CAS protocol spec says to use 'pgt' as the parameter, but in practice
+      #         the JA-SIG and Yale server implementations use pgtId. We'll go with the
+      #         in-practice standard.
+      path += (uri.query.nil? || uri.query.empty? ? '?' : '&') + "pgtId=#{pgt.ticket}&pgtIou=#{pgt.iou}"
+
+      response = conn.request_get(path)
+      # TODO: follow redirects... 2.5.4 says that redirects MAY be followed
+      # NOTE: The following response codes are valid according to the JA-SIG implementation even without following redirects
+      
+      if %w(200 202 301 302 304).include?(response.code)
+        # 3.4 (proxy-granting ticket IOU)
+        pgt.save!
+        $LOG.debug "PGT generated for pgt_url '#{pgt_url}': #{pgt.inspect}"
+        pgt
+      else
+        $LOG.warn "PGT callback server responded with a bad result code '#{response.code}'. PGT will not be stored."
+        nil
+      end
+    end
+  end
+
+  def validate_login_ticket(ticket)
+    $LOG.debug("Validating login ticket '#{ticket}'")
+
+    success = false
+    if ticket.nil?
+      error = t.error.no_login_ticket
+      $LOG.warn "Missing login ticket."
+    elsif lt = LoginTicket.find_by_ticket(ticket)
+      if lt.consumed?
+        error = t.error.login_ticket_already_used
+        $LOG.warn "Login ticket '#{ticket}' previously used up"
+      elsif Time.now - lt.created_on < settings.config[:maximum_unused_login_ticket_lifetime]
+        $LOG.info "Login ticket '#{ticket}' successfully validated"
+      else
+        error = t.error.login_timeout
+        $LOG.warn "Expired login ticket '#{ticket}'"
+      end
+    else
+      error = t.error.invalid_login_ticket
+      $LOG.warn "Invalid login ticket '#{ticket}'"
+    end
+
+    lt.consume! if lt
+
+    error
+  end
+
+  def validate_ticket_granting_ticket(ticket)
+    $LOG.debug("Validating ticket granting ticket '#{ticket}'")
+
+    if ticket.nil?
+      error = "No ticket granting ticket given."
+      $LOG.debug error
+    elsif tgt = TicketGrantingTicket.find_by_ticket(ticket)
+      if settings.config[:maximum_session_lifetime] && Time.now - tgt.created_on > settings.config[:maximum_session_lifetime]
+	tgt.destroy
+        error = "Your session has expired. Please log in again."
+        $LOG.info "Ticket granting ticket '#{ticket}' for user '#{tgt.username}' expired."
+      else
+        $LOG.info "Ticket granting ticket '#{ticket}' for user '#{tgt.username}' successfully validated."
+      end
+    else
+      error = "Invalid ticket granting ticket '#{ticket}' (no matching ticket found in the database)."
+      $LOG.warn(error)
+    end
+
+    [tgt, error]
+  end
+
+  def validate_service_ticket(service, ticket, allow_proxy_tickets = false)
+    $LOG.debug "Validating service/proxy ticket '#{ticket}' for service '#{service}'"
+
+    if service.nil? or ticket.nil?
+      error = Error.new(:INVALID_REQUEST, "Ticket or service parameter was missing in the request.")
+      $LOG.warn "#{error.code} - #{error.message}"
+    elsif st = ServiceTicket.find_by_ticket(ticket)
+      if st.consumed?
+        error = Error.new(:INVALID_TICKET, "Ticket '#{ticket}' has already been used up.")
+        $LOG.warn "#{error.code} - #{error.message}"
+      elsif st.kind_of?(CASServer::Model::ProxyTicket) && !allow_proxy_tickets
+        error = Error.new(:INVALID_TICKET, "Ticket '#{ticket}' is a proxy ticket, but only service tickets are allowed here.")
+        $LOG.warn "#{error.code} - #{error.message}"
+      elsif Time.now - st.created_on > settings.config[:maximum_unused_service_ticket_lifetime]
+        error = Error.new(:INVALID_TICKET, "Ticket '#{ticket}' has expired.")
+        $LOG.warn "Ticket '#{ticket}' has expired."
+      elsif !st.matches_service? service
+        error = Error.new(:INVALID_SERVICE, "The ticket '#{ticket}' belonging to user '#{st.username}' is valid,"+
+          " but the requested service '#{service}' does not match the service '#{st.service}' associated with this ticket.")
+        $LOG.warn "#{error.code} - #{error.message}"
+      else
+        $LOG.info("Ticket '#{ticket}' for service '#{service}' for user '#{st.username}' successfully validated.")
+      end
+    else
+      error = Error.new(:INVALID_TICKET, "Ticket '#{ticket}' not recognized.")
+      $LOG.warn("#{error.code} - #{error.message}")
+    end
+
+    if st
+      st.consume!
+    end
+
+
+    [st, error]
+  end
+
+  def validate_proxy_ticket(service, ticket)
+    pt, error = validate_service_ticket(service, ticket, true)
+
+    if pt.kind_of?(CASServer::Model::ProxyTicket) && !error
+      if not pt.granted_by_pgt
+        error = Error.new(:INTERNAL_ERROR, "Proxy ticket '#{pt}' belonging to user '#{pt.username}' is not associated with a proxy granting ticket.")
+      elsif not pt.granted_by_pgt.service_ticket
+        error = Error.new(:INTERNAL_ERROR, "Proxy granting ticket '#{pt.granted_by_pgt}'"+
+          " (associated with proxy ticket '#{pt}' and belonging to user '#{pt.username}' is not associated with a service ticket.")
+      end
+    end
+
+    [pt, error]
+  end
+
+  def validate_proxy_granting_ticket(ticket)
+    if ticket.nil?
+      error = Error.new(:INVALID_REQUEST, "pgt parameter was missing in the request.")
+      $LOG.warn("#{error.code} - #{error.message}")
+    elsif pgt = ProxyGrantingTicket.find_by_ticket(ticket)
+      if pgt.service_ticket
+        $LOG.info("Proxy granting ticket '#{ticket}' belonging to user '#{pgt.service_ticket.username}' successfully validated.")
+      else
+        error = Error.new(:INTERNAL_ERROR, "Proxy granting ticket '#{ticket}' is not associated with a service ticket.")
+        $LOG.error("#{error.code} - #{error.message}")
+      end
+    else
+      error = Error.new(:BAD_PGT, "Invalid proxy granting ticket '#{ticket}' (no matching ticket found in the database).")
+      $LOG.warn("#{error.code} - #{error.message}")
+    end
+
+    [pgt, error]
+  end
+
+  # Takes an existing ServiceTicket object (presumably pulled from the database)
+  # and sends a POST with logout information to the service that the ticket
+  # was generated for.
+  #
+  # This makes possible the "single sign-out" functionality added in CAS 3.1.
+  # See http://www.ja-sig.org/wiki/display/CASUM/Single+Sign+Out
+  def send_logout_notification_for_service_ticket(st)
+    uri = URI.parse(st.service)
+    uri.path = '/' if uri.path.empty?
+    time = Time.now
+    rand = CASServer::Utils.random_string
+    path = uri.path
+    req = Net::HTTP::Post.new(path)
+    req.set_form_data('logoutRequest' => %{<samlp:LogoutRequest ID="#{rand}" Version="2.0" IssueInstant="#{time.rfc2822}">
+ <saml:NameID></saml:NameID>
+ <samlp:SessionIndex>#{st.ticket}</samlp:SessionIndex>
+ </samlp:LogoutRequest>})
+ 
+    begin
+      http = Net::HTTP.new(uri.host, uri.port)
+      http.use_ssl = true if uri.scheme =='https'
+      
+      http.start do |conn|
+        response = conn.request(req)
+        if response.kind_of? Net::HTTPSuccess
+          $LOG.info "Logout notification successfully posted to #{st.service.inspect}."
+          return true
+        else
+          $LOG.error "Service #{st.service.inspect} responed to logout notification with code '#{response.code}'!"
+          return false
+        end
+      end
+    rescue Exception => e
+      $LOG.error "Failed to send logout notification to service #{st.service.inspect} due to #{e}"
+      return false
+    end
+  end
+
+  def service_uri_with_ticket(service, st)
+    raise ArgumentError, "Second argument must be a ServiceTicket!" unless st.kind_of? CASServer::Model::ServiceTicket
+
+    # This will choke with a URI::InvalidURIError if service URI is not properly URI-escaped...
+    # This exception is handled further upstream (i.e. in the controller).
+    service_uri = URI.parse(service)
+
+    if service.include? "?"
+      if service_uri.query.empty?
+        query_separator = ""
+      else
+        query_separator = "&"
+      end
+    else
+      query_separator = "?"
+    end
+
+    service_with_ticket = service + query_separator + "ticket=" + st.ticket
+    service_with_ticket
+  end
+
+  # Strips CAS-related parameters from a service URL and normalizes it,
+  # removing trailing / and ?. Also converts any spaces to +.
+  #
+  # For example, "http://google.com?ticket=12345" will be returned as
+  # "http://google.com". Also, "http://google.com/" would be returned as
+  # "http://google.com".
+  #
+  # Note that only the first occurance of each CAS-related parameter is
+  # removed, so that "http://google.com?ticket=12345&ticket=abcd" would be
+  # returned as "http://google.com?ticket=abcd".
+  def clean_service_url(dirty_service)
+    return dirty_service if dirty_service.blank?
+    clean_service = dirty_service.dup
+    ['service', 'ticket', 'gateway', 'renew'].each do |p|
+      clean_service.sub!(Regexp.new("&?#{p}=[^&]*"), '')
+    end
+
+    clean_service.gsub!(/[\/\?&]$/, '') # remove trailing ?, /, or &
+    clean_service.gsub!('?&', '?')
+    clean_service.gsub!(' ', '+')
+
+    $LOG.debug("Cleaned dirty service URL #{dirty_service.inspect} to #{clean_service.inspect}") if
+      dirty_service != clean_service
+
+    return clean_service
+  end
+  module_function :clean_service_url
+
+end

data/lib/casserver/localization.rb

@@ -0,0 +1,13 @@
+require 'sinatra/r18n'
+
+module CASServer
+  module Localization
+    def self.included(mod)
+      mod.module_eval do
+        register Sinatra::R18n
+        set :default_locale, 'en'
+        set :translations, File.dirname(__FILE__) + "/../../locales"
+      end
+    end
+  end
+end

data/lib/casserver/model.rb

@@ -0,0 +1,270 @@
+require 'active_record'
+require 'active_record/base'
+
+module CASServer::Model
+
+  module Consumable
+    def consume!
+      self.consumed = Time.now
+      self.save!
+    end
+
+    def self.included(mod)
+      mod.extend(ClassMethods)
+    end
+
+    module ClassMethods
+      def cleanup(max_lifetime, max_unconsumed_lifetime)
+        transaction do
+          conditions = ["created_on < ? OR (consumed IS NULL AND created_on < ?)",
+                          Time.now - max_lifetime,
+                          Time.now - max_unconsumed_lifetime]
+                        
+          expired_tickets_count = count(:conditions => conditions)
+
+          $LOG.debug("Destroying #{expired_tickets_count} expired #{self.name.demodulize}"+
+            "#{'s' if expired_tickets_count > 1}.") if expired_tickets_count > 0
+
+          destroy_all(conditions)
+        end
+      end
+    end
+  end
+
+  class Base < ActiveRecord::Base
+  end
+
+  class Ticket < Base
+    def to_s
+      ticket
+    end
+
+    def self.cleanup(max_lifetime)
+      transaction do
+        conditions = ["created_on < ?", Time.now - max_lifetime]
+        expired_tickets_count = count(:conditions => conditions)
+
+        $LOG.debug("Destroying #{expired_tickets_count} expired #{self.name.demodulize}"+
+          "#{'s' if expired_tickets_count > 1}.") if expired_tickets_count > 0
+
+        destroy_all(conditions)
+      end
+    end
+  end
+
+  class LoginTicket < Ticket
+    set_table_name 'casserver_lt'
+    include Consumable
+  end
+
+  class ServiceTicket < Ticket
+    set_table_name 'casserver_st'
+    include Consumable
+
+    belongs_to :granted_by_tgt,
+      :class_name => 'CASServer::Model::TicketGrantingTicket',
+      :foreign_key => :granted_by_tgt_id
+    has_one :proxy_granting_ticket,
+      :foreign_key => :created_by_st_id
+
+    def matches_service?(service)
+      CASServer::CAS.clean_service_url(self.service) ==
+        CASServer::CAS.clean_service_url(service)
+    end
+  end
+
+  class ProxyTicket < ServiceTicket
+    belongs_to :granted_by_pgt,
+      :class_name => 'CASServer::Model::ProxyGrantingTicket',
+      :foreign_key => :granted_by_pgt_id
+  end
+
+  class TicketGrantingTicket < Ticket
+    set_table_name 'casserver_tgt'
+
+    serialize :extra_attributes
+
+    has_many :granted_service_tickets,
+      :class_name => 'CASServer::Model::ServiceTicket',
+      :foreign_key => :granted_by_tgt_id
+  end
+
+  class ProxyGrantingTicket < Ticket
+    set_table_name 'casserver_pgt'
+    belongs_to :service_ticket
+    has_many :granted_proxy_tickets,
+      :class_name => 'CASServer::Model::ProxyTicket',
+      :foreign_key => :granted_by_pgt_id
+  end
+
+  class Error
+    attr_reader :code, :message
+
+    def initialize(code, message)
+      @code = code
+      @message = message
+    end
+
+    def to_s
+      message
+    end
+  end
+
+#  class CreateCASServer < V 0.1
+#    def self.up
+#      if ActiveRecord::Base.connection.table_alias_length > 30
+#        $LOG.info("Creating database with long table names...")
+#
+#        create_table :casserver_login_tickets, :force => true do |t|
+#          t.column :ticket,     :string,   :null => false
+#          t.column :created_on, :timestamp, :null => false
+#          t.column :consumed,   :datetime, :null => true
+#          t.column :client_hostname, :string, :null => false
+#        end
+#
+#        create_table :casserver_service_tickets, :force => true do |t|
+#          t.column :ticket,     :string,    :null => false
+#          t.column :service,    :string,    :null => false
+#          t.column :created_on, :timestamp, :null => false
+#          t.column :consumed,   :datetime, :null => true
+#          t.column :client_hostname, :string, :null => false
+#          t.column :username,   :string,  :null => false
+#          t.column :type,       :string,   :null => false
+#          t.column :proxy_granting_ticket_id, :integer, :null => true
+#        end
+#
+#        create_table :casserver_ticket_granting_tickets, :force => true do |t|
+#          t.column :ticket,     :string,    :null => false
+#          t.column :created_on, :timestamp, :null => false
+#          t.column :client_hostname, :string, :null => false
+#          t.column :username,   :string,    :null => false
+#        end
+#
+#        create_table :casserver_proxy_granting_tickets, :force => true do |t|
+#          t.column :ticket,     :string,    :null => false
+#          t.column :created_on, :timestamp, :null => false
+#          t.column :client_hostname, :string, :null => false
+#          t.column :iou,        :string,    :null => false
+#          t.column :service_ticket_id, :integer, :null => false
+#        end
+#      end
+#    end
+#
+#    def self.down
+#      if ActiveRecord::Base.connection.table_alias_length > 30
+#        drop_table :casserver_proxy_granting_tickets
+#        drop_table :casserver_ticket_granting_tickets
+#        drop_table :casserver_service_tickets
+#        drop_table :casserver_login_tickets
+#      end
+#    end
+#  end
+#
+#  # Oracle table names cannot exceed 30 chars...
+#  # See http://code.google.com/p/rubycas-server/issues/detail?id=15
+#  class ShortenTableNames < V 0.5
+#    def self.up
+#      if ActiveRecord::Base.connection.table_alias_length > 30
+#        $LOG.info("Shortening table names")
+#        rename_table :casserver_login_tickets, :casserver_lt
+#        rename_table :casserver_service_tickets, :casserver_st
+#        rename_table :casserver_ticket_granting_tickets, :casserver_tgt
+#        rename_table :casserver_proxy_granting_tickets, :casserver_pgt
+#      else
+#        create_table :casserver_lt, :force => true do |t|
+#          t.column :ticket,     :string,   :null => false
+#          t.column :created_on, :timestamp, :null => false
+#          t.column :consumed,   :datetime, :null => true
+#          t.column :client_hostname, :string, :null => false
+#        end
+#
+#        create_table :casserver_st, :force => true do |t|
+#          t.column :ticket,     :string,    :null => false
+#          t.column :service,    :string,    :null => false
+#          t.column :created_on, :timestamp, :null => false
+#          t.column :consumed,   :datetime, :null => true
+#          t.column :client_hostname, :string, :null => false
+#          t.column :username,   :string,  :null => false
+#          t.column :type,       :string,   :null => false
+#          t.column :proxy_granting_ticket_id, :integer, :null => true
+#        end
+#
+#        create_table :casserver_tgt, :force => true do |t|
+#          t.column :ticket,     :string,    :null => false
+#          t.column :created_on, :timestamp, :null => false
+#          t.column :client_hostname, :string, :null => false
+#          t.column :username,   :string,    :null => false
+#        end
+#
+#        create_table :casserver_pgt, :force => true do |t|
+#          t.column :ticket,     :string,    :null => false
+#          t.column :created_on, :timestamp, :null => false
+#          t.column :client_hostname, :string, :null => false
+#          t.column :iou,        :string,    :null => false
+#          t.column :service_ticket_id, :integer, :null => false
+#        end
+#      end
+#    end
+#
+#    def self.down
+#      if ActiveRecord::Base.connection.table_alias_length > 30
+#        rename_table :casserver_lt, :cassserver_login_tickets
+#        rename_table :casserver_st, :casserver_service_tickets
+#        rename_table :casserver_tgt, :casserver_ticket_granting_tickets
+#        rename_table :casserver_pgt, :casserver_proxy_granting_tickets
+#      else
+#        drop_table :casserver_pgt
+#        drop_table :casserver_tgt
+#        drop_table :casserver_st
+#        drop_table :casserver_lt
+#      end
+#    end
+#  end
+#
+#  class AddTgtToSt < V 0.7
+#    def self.up
+#      add_column :casserver_st, :tgt_id, :integer, :null => true
+#    end
+#
+#    def self.down
+#      remove_column :casserver_st, :tgt_id, :integer
+#    end
+#  end
+#
+#  class ChangeServiceToText < V 0.71
+#    def self.up
+#      # using change_column to change the column type from :string to :text
+#      # doesn't seem to work, at least under MySQL, so we drop and re-create
+#      # the column instead
+#      remove_column :casserver_st, :service
+#      say "WARNING: All existing service tickets are being deleted."
+#      add_column :casserver_st, :service, :text
+#    end
+#
+#    def self.down
+#      change_column :casserver_st, :service, :string
+#    end
+#  end
+#
+#  class AddExtraAttributes < V 0.72
+#    def self.up
+#      add_column :casserver_tgt, :extra_attributes, :text
+#    end
+#
+#    def self.down
+#      remove_column :casserver_tgt, :extra_attributes
+#    end
+#  end
+#
+#  class RenamePgtForeignKeys < V 0.80
+#    def self.up
+#      rename_column :casserver_st,  :proxy_granting_ticket_id,  :granted_by_pgt_id
+#      rename_column :casserver_st,  :tgt_id,                    :granted_by_tgt_id
+#    end
+#
+#    def self.down
+#      rename_column :casserver_st,  :granted_by_pgt_id,         :proxy_granting_ticket_id
+#      rename_column :casserver_st,  :granted_by_tgt_id,         :tgt_id
+#    end
+#  end
+end