rpush 7.0.1 → 9.0.0

data/lib/rpush/daemon/apns.rb

@@ -1,17 +0,0 @@
-module Rpush
-  module Daemon
-    module Apns
-      extend ServiceConfigMethods
-
-      HOSTS = {
-        production: ['gateway.push.apple.com', 2195],
-        development: ['gateway.sandbox.push.apple.com', 2195], # deprecated
-        sandbox: ['gateway.sandbox.push.apple.com', 2195]
-      }
-
-      batch_deliveries true
-      dispatcher :apns_tcp, host: proc { |app| HOSTS[app.environment.to_sym] }
-      loops Rpush::Daemon::Apns::FeedbackReceiver, if: -> { Rpush.config.apns.feedback_receiver.enabled && !Rpush.config.push }
-    end
-  end
-end

data/lib/rpush/daemon/dispatcher/apns_tcp.rb

@@ -1,152 +0,0 @@
-module Rpush
-  module Daemon
-    module Dispatcher
-      class ApnsTcp < Rpush::Daemon::Dispatcher::Tcp
-        include Loggable
-        include Reflectable
-
-        SELECT_TIMEOUT = 10
-        ERROR_TUPLE_BYTES = 6
-        APNS_ERRORS = {
-          1 => 'Processing error',
-          2 => 'Missing device token',
-          3 => 'Missing topic',
-          4 => 'Missing payload',
-          5 => 'Missing token size',
-          6 => 'Missing topic size',
-          7 => 'Missing payload size',
-          8 => 'Invalid device token',
-          10 => 'APNs closed connection (possible maintenance)',
-          255 => 'None (unknown error)'
-        }
-
-        def initialize(*args)
-          super
-          @dispatch_mutex = Mutex.new
-          @stop_error_receiver = false
-          @connection.on_connect { start_error_receiver }
-        end
-
-        def dispatch(payload)
-          @dispatch_mutex.synchronize do
-            @delivery_class.new(@app, @connection, payload.batch).perform
-            record_batch(payload.batch)
-          end
-        end
-
-        def cleanup
-          if Rpush.config.push
-            # In push mode only a single batch is sent, followed by immediate shutdown.
-            # Allow the error receiver time to handle any errors.
-            @reconnect_disabled = true
-            sleep 1
-          end
-
-          @stop_error_receiver = true
-          super
-          @error_receiver_thread.join if @error_receiver_thread
-        rescue StandardError => e
-          log_error(e)
-          reflect(:error, e)
-        ensure
-          @error_receiver_thread = nil
-        end
-
-        private
-
-        def start_error_receiver
-          @error_receiver_thread = Thread.new do
-            check_for_error until @stop_error_receiver
-            Rpush::Daemon.store.release_connection
-          end
-        end
-
-        def delivered_buffer
-          @delivered_buffer ||= RingBuffer.new(Rpush.config.batch_size * 10)
-        end
-
-        def record_batch(batch)
-          batch.each_delivered do |notification|
-            delivered_buffer << notification.id
-          end
-        end
-
-        def check_for_error
-          begin
-            # On Linux, select returns nil from a dropped connection.
-            # On OS X, Errno::EBADF is raised following a Errno::EADDRNOTAVAIL from the write call.
-            return unless @connection.select(SELECT_TIMEOUT)
-            tuple = @connection.read(ERROR_TUPLE_BYTES)
-          rescue *TcpConnection::TCP_ERRORS
-            reconnect unless @stop_error_receiver
-            return
-          end
-
-          @dispatch_mutex.synchronize { handle_error_response(tuple) }
-        rescue StandardError => e
-          log_error(e)
-        end
-
-        def handle_error_response(tuple)
-          if tuple
-            _, code, notification_id = tuple.unpack('ccN')
-            handle_error(code, notification_id)
-          else
-            handle_disconnect
-          end
-
-          if Rpush.config.push
-            # Only attempt to handle a single error in Push mode.
-            @stop_error_receiver = true
-            return
-          end
-
-          reconnect
-        ensure
-          delivered_buffer.clear
-        end
-
-        def reconnect
-          return if @reconnect_disabled
-          log_error("Lost connection to #{@connection.host}:#{@connection.port}, reconnecting...")
-          @connection.reconnect_with_rescue
-        end
-
-        def handle_disconnect
-          log_error("The APNs disconnected before any notifications could be delivered. This usually indicates you are using an invalid certificate.") if delivered_buffer.size == 0
-        end
-
-        def handle_error(code, notification_id)
-          notification_id = Rpush::Daemon.store.translate_integer_notification_id(notification_id)
-          failed_pos = delivered_buffer.index(notification_id)
-          description = description_for_code(code)
-          log_error("Notification #{notification_id} failed with error: " + description)
-          Rpush::Daemon.store.mark_ids_failed([notification_id], code, description, Time.now)
-          reflect(:notification_id_failed, @app, notification_id, code, description)
-
-          if failed_pos
-            retry_ids = delivered_buffer[(failed_pos + 1)..-1]
-            retry_notification_ids(retry_ids, notification_id)
-          elsif delivered_buffer.size > 0
-            log_error("Delivery sequence unknown for notifications following #{notification_id}.")
-          end
-        end
-
-        def description_for_code(code)
-          APNS_ERRORS[code.to_i] ? "#{APNS_ERRORS[code.to_i]} (#{code})" : "Unknown error code #{code.inspect}. Possible Rpush bug?"
-        end
-
-        def retry_notification_ids(ids, notification_id)
-          return if ids.size == 0
-
-          now = Time.now
-          Rpush::Daemon.store.mark_ids_retryable(ids, now)
-          notifications_str = 'Notification'
-          notifications_str += 's' if ids.size > 1
-          log_warn("#{notifications_str} #{ids.join(', ')} will be retried due to the failure of notification #{notification_id}.")
-          ids.each { |id| reflect(:notification_id_will_retry, @app, id, now) }
-        end
-      end
-    end
-  end
-end

data/lib/rpush/daemon/dispatcher/tcp.rb

@@ -1,22 +0,0 @@
-module Rpush
-  module Daemon
-    module Dispatcher
-      class Tcp
-        def initialize(app, delivery_class, options = {})
-          @app = app
-          @delivery_class = delivery_class
-          @host, @port = options[:host].call(@app)
-          @connection = Rpush::Daemon::TcpConnection.new(@app, @host, @port)
-        end
-
-        def dispatch(payload)
-          @delivery_class.new(@app, @connection, payload.notification, payload.batch).perform
-        end
-
-        def cleanup
-          @connection.close if @connection
-        end
-      end
-    end
-  end
-end

data/lib/rpush/daemon/gcm/delivery.rb

@@ -1,241 +0,0 @@
-module Rpush
-  module Daemon
-    module Gcm
-      # https://firebase.google.com/docs/cloud-messaging/server
-      class Delivery < Rpush::Daemon::Delivery
-        include MultiJsonHelper
-
-        host = 'https://fcm.googleapis.com'
-        FCM_URI = URI.parse("#{host}/fcm/send")
-        UNAVAILABLE_STATES = %w(Unavailable BadGateway InternalServerError)
-        INVALID_REGISTRATION_ID_STATES = %w(InvalidRegistration MismatchSenderId NotRegistered InvalidPackageName)
-
-        def initialize(app, http, notification, batch)
-          @app = app
-          @http = http
-          @notification = notification
-          @batch = batch
-        end
-
-        def perform
-          handle_response(do_post)
-        rescue SocketError => error
-          mark_retryable(@notification, Time.now + 10.seconds, error)
-          raise
-        rescue StandardError => error
-          mark_failed(error)
-          raise
-        ensure
-          @batch.notification_processed
-        end
-
-        protected
-
-        def handle_response(response)
-          case response.code.to_i
-          when 200
-            ok(response)
-          when 400
-            bad_request
-          when 401
-            unauthorized
-          when 500
-            internal_server_error(response)
-          when 502
-            bad_gateway(response)
-          when 503
-            service_unavailable(response)
-          when 500..599
-            other_5xx_error(response)
-          else
-            fail Rpush::DeliveryError.new(response.code.to_i, @notification.id, Rpush::Daemon::HTTP_STATUS_CODES[response.code.to_i])
-          end
-        end
-
-        def ok(response)
-          results = process_response(response)
-          handle_successes(results.successes)
-
-          if results.failures.any?
-            handle_failures(results.failures, response)
-          else
-            mark_delivered
-            log_info("#{@notification.id} sent to #{@notification.registration_ids.join(', ')}")
-          end
-        end
-
-        def process_response(response)
-          body = multi_json_load(response.body)
-          results = Results.new(body['results'], @notification.registration_ids)
-          results.process(invalid: INVALID_REGISTRATION_ID_STATES, unavailable: UNAVAILABLE_STATES)
-          results
-        end
-
-        def handle_successes(successes)
-          successes.each do |result|
-            reflect(:gcm_delivered_to_recipient, @notification, result[:registration_id])
-            next unless result.key?(:canonical_id)
-            reflect(:gcm_canonical_id, result[:registration_id], result[:canonical_id])
-          end
-        end
-
-        def handle_failures(failures, response)
-          if failures[:unavailable].count == @notification.registration_ids.count
-            retry_delivery(@notification, response)
-            log_warn("All recipients unavailable. #{retry_message}")
-          else
-            if failures[:unavailable].any?
-              unavailable_idxs = failures[:unavailable].map { |result| result[:index] }
-              new_notification = create_new_notification(response, unavailable_idxs)
-              failures.description += " #{unavailable_idxs.join(', ')} will be retried as notification #{new_notification.id}."
-            end
-            handle_errors(failures)
-            fail Rpush::DeliveryError.new(nil, @notification.id, failures.description)
-          end
-        end
-
-        def handle_errors(failures)
-          failures.each do |result|
-            reflect(:gcm_failed_to_recipient, @notification, result[:error], result[:registration_id])
-          end
-          failures[:invalid].each do |result|
-            reflect(:gcm_invalid_registration_id, @app, result[:error], result[:registration_id])
-          end
-        end
-
-        def create_new_notification(response, unavailable_idxs)
-          attrs = { 'app_id' => @notification.app_id, 'collapse_key' => @notification.collapse_key, 'delay_while_idle' => @notification.delay_while_idle }
-          registration_ids = @notification.registration_ids.values_at(*unavailable_idxs)
-          Rpush::Daemon.store.create_gcm_notification(attrs, @notification.data,
-                                                      registration_ids, deliver_after_header(response), @app)
-        end
-
-        def bad_request
-          fail Rpush::DeliveryError.new(400, @notification.id, 'GCM failed to parse the JSON request. Possibly an Rpush bug, please open an issue.')
-        end
-
-        def unauthorized
-          fail Rpush::DeliveryError.new(401, @notification.id, 'Unauthorized, check your App auth_key.')
-        end
-
-        def internal_server_error(response)
-          retry_delivery(@notification, response)
-          log_warn("GCM responded with an Internal Error. " + retry_message)
-        end
-
-        def bad_gateway(response)
-          retry_delivery(@notification, response)
-          log_warn("GCM responded with a Bad Gateway Error. " + retry_message)
-        end
-
-        def service_unavailable(response)
-          retry_delivery(@notification, response)
-          log_warn("GCM responded with an Service Unavailable Error. " + retry_message)
-        end
-
-        def other_5xx_error(response)
-          retry_delivery(@notification, response)
-          log_warn("GCM responded with a 5xx Error. " + retry_message)
-        end
-
-        def deliver_after_header(response)
-          Rpush::Daemon::RetryHeaderParser.parse(response.header['retry-after'])
-        end
-
-        def retry_delivery(notification, response)
-          time = deliver_after_header(response)
-          if time
-            mark_retryable(notification, time)
-          else
-            mark_retryable_exponential(notification)
-          end
-        end
-
-        def retry_message
-          "Notification #{@notification.id} will be retried after #{@notification.deliver_after.strftime('%Y-%m-%d %H:%M:%S')} (retry #{@notification.retries})."
-        end
-
-        def do_post
-          post = Net::HTTP::Post.new(FCM_URI.path, 'Content-Type'  => 'application/json',
-                                                   'Authorization' => "key=#{@app.auth_key}")
-          post.body = @notification.as_json.to_json
-          @http.request(FCM_URI, post)
-        end
-      end
-
-      class Results
-        attr_reader :successes, :failures
-
-        def initialize(results_data, registration_ids)
-          @results_data = results_data
-          @registration_ids = registration_ids
-        end
-
-        def process(failure_partitions = {}) # rubocop:disable Metrics/AbcSize
-          @successes = []
-          @failures = Failures.new
-          failure_partitions.each_key do |category|
-            failures[category] = []
-          end
-
-          @results_data.each_with_index do |result, index|
-            entry = {
-              registration_id: @registration_ids[index],
-              index: index
-            }
-            if result['message_id']
-              entry[:canonical_id] = result['registration_id'] if result['registration_id'].present?
-              successes << entry
-            elsif result['error']
-              entry[:error] = result['error']
-              failures << entry
-              failure_partitions.each do |category, error_states|
-                failures[category] << entry if error_states.include?(result['error'])
-              end
-            end
-          end
-          failures.all_failed = failures.count == @registration_ids.count
-        end
-      end
-
-      class Failures < Hash
-        include Enumerable
-        attr_writer :all_failed, :description
-
-        def initialize
-          super[:all] = []
-        end
-
-        def each
-          self[:all].each { |x| yield x }
-        end
-
-        def <<(item)
-          self[:all] << item
-        end
-
-        def description
-          @description ||= describe
-        end
-
-        def any?
-          self[:all].any?
-        end
-
-        private
-
-        def describe
-          if @all_failed
-            error_description = "Failed to deliver to all recipients."
-          else
-            index_list = map { |item| item[:index] }
-            error_description = "Failed to deliver to recipients #{index_list.join(', ')}."
-          end
-
-          error_list = map { |item| item[:error] }
-          error_description + " Errors: #{error_list.join(', ')}."
-        end
-      end
-    end
-  end
-end

data/lib/rpush/daemon/tcp_connection.rb

@@ -1,190 +0,0 @@
-module Rpush
-  module Daemon
-    class TcpConnectionError < StandardError; end
-
-    class TcpConnection
-      include Reflectable
-      include Loggable
-
-      OSX_TCP_KEEPALIVE = 0x10 # Defined in <netinet/tcp.h>
-      KEEPALIVE_INTERVAL = 5
-      KEEPALIVE_IDLE = 5
-      KEEPALIVE_MAX_FAIL_PROBES = 1
-      TCP_ERRORS = [SystemCallError, OpenSSL::OpenSSLError, IOError]
-
-      attr_accessor :last_touch
-      attr_reader :host, :port
-
-      def self.idle_period
-        30.minutes
-      end
-
-      def initialize(app, host, port)
-        @app = app
-        @host = host
-        @port = port
-        @certificate = app.certificate
-        @password = app.password
-        @connected = false
-        @connection_callbacks = []
-        touch
-      end
-
-      def on_connect(&blk)
-        raise 'already connected' if @connected
-        @connection_callbacks << blk
-      end
-
-      def connect
-        @ssl_context = setup_ssl_context
-        @tcp_socket, @ssl_socket = connect_socket
-        @connected = true
-
-        @connection_callbacks.each do |blk|
-          begin
-            blk.call
-          rescue StandardError => e
-            log_error(e)
-          end
-        end
-
-        @connection_callbacks.clear
-      end
-
-      def close
-        @ssl_socket.close if @ssl_socket
-        @tcp_socket.close if @tcp_socket
-      rescue IOError # rubocop:disable HandleExceptions
-      end
-
-      def read(num_bytes)
-        @ssl_socket.read(num_bytes) if @ssl_socket
-      end
-
-      def select(timeout)
-        IO.select([@ssl_socket], nil, nil, timeout) if @ssl_socket
-      end
-
-      def write(data)
-        connect unless @connected
-        reconnect_idle if idle_period_exceeded?
-
-        retry_count = 0
-
-        begin
-          write_data(data)
-        rescue *TCP_ERRORS => e
-          retry_count += 1
-
-          if retry_count == 1
-            log_error("Lost connection to #{@host}:#{@port} (#{e.class.name}, #{e.message}), reconnecting...")
-            reflect(:tcp_connection_lost, @app, e)
-          end
-
-          if retry_count <= 3
-            reconnect_with_rescue
-            sleep 1
-            retry
-          else
-            raise TcpConnectionError, "#{@app.name} tried #{retry_count - 1} times to reconnect but failed (#{e.class.name}, #{e.message})."
-          end
-        end
-      end
-
-      def reconnect_with_rescue
-        reconnect
-      rescue StandardError => e
-        log_error(e)
-      end
-
-      def reconnect
-        close
-        @tcp_socket, @ssl_socket = connect_socket
-      end
-
-      protected
-
-      def reconnect_idle
-        log_info("Idle period exceeded, reconnecting...")
-        reconnect
-      end
-
-      def idle_period_exceeded?
-        Time.now - last_touch > self.class.idle_period
-      end
-
-      def write_data(data)
-        @ssl_socket.write(data)
-        @ssl_socket.flush
-        touch
-      end
-
-      def touch
-        self.last_touch = Time.now
-      end
-
-      def setup_ssl_context
-        ssl_context = OpenSSL::SSL::SSLContext.new
-        ssl_context.key = OpenSSL::PKey::RSA.new(@certificate, @password)
-        ssl_context.cert = OpenSSL::X509::Certificate.new(@certificate)
-        ssl_context
-      end
-
-      def connect_socket
-        touch
-        check_certificate_expiration
-
-        tcp_socket = TCPSocket.new(@host, @port)
-        tcp_socket.setsockopt(Socket::SOL_SOCKET, Socket::SO_KEEPALIVE, true)
-        tcp_socket.setsockopt(Socket::IPPROTO_TCP, Socket::TCP_NODELAY, true)
-
-        # Linux
-        if [:SOL_TCP, :TCP_KEEPIDLE, :TCP_KEEPINTVL, :TCP_KEEPCNT].all? { |c| Socket.const_defined?(c) }
-          tcp_socket.setsockopt(Socket::SOL_TCP, Socket::TCP_KEEPIDLE, KEEPALIVE_IDLE)
-          tcp_socket.setsockopt(Socket::SOL_TCP, Socket::TCP_KEEPINTVL, KEEPALIVE_INTERVAL)
-          tcp_socket.setsockopt(Socket::SOL_TCP, Socket::TCP_KEEPCNT, KEEPALIVE_MAX_FAIL_PROBES)
-        end
-
-        # OSX
-        if RUBY_PLATFORM =~ /darwin/
-          tcp_socket.setsockopt(Socket::IPPROTO_TCP, OSX_TCP_KEEPALIVE, KEEPALIVE_IDLE)
-        end
-
-        ssl_socket = OpenSSL::SSL::SSLSocket.new(tcp_socket, @ssl_context)
-        ssl_socket.sync = true
-        ssl_socket.connect
-        [tcp_socket, ssl_socket]
-      rescue *TCP_ERRORS => error
-        if error.message =~ /certificate revoked/i
-          log_error('Certificate has been revoked.')
-          reflect(:ssl_certificate_revoked, @app, error)
-        end
-        raise TcpConnectionError, "#{error.class.name}, #{error.message}"
-      end
-
-      def check_certificate_expiration
-        cert = @ssl_context.cert
-        if certificate_expired?
-          log_error(certificate_msg('expired'))
-          raise Rpush::CertificateExpiredError.new(@app, cert.not_after)
-        elsif certificate_expires_soon?
-          log_warn(certificate_msg('will expire'))
-          reflect(:ssl_certificate_will_expire, @app, cert.not_after)
-        end
-      end
-
-      def certificate_msg(msg)
-        time = @ssl_context.cert.not_after.utc.strftime('%Y-%m-%d %H:%M:%S UTC')
-        "Certificate #{msg} at #{time}."
-      end
-
-      def certificate_expired?
-        @ssl_context.cert.not_after && @ssl_context.cert.not_after.utc < Time.now.utc
-      end
-
-      def certificate_expires_soon?
-        @ssl_context.cert.not_after && @ssl_context.cert.not_after.utc < (Time.now + 1.month).utc
-      end
-    end
-  end
-end