apiotics-paho-mqtt 1.0.0

data/lib/paho_mqtt/handler.rb

@@ -0,0 +1,273 @@
+# Copyright (c) 2016-2017 Pierre Goudet <p-goudet@ruby-dev.jp>
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Eclipse Public License v1.0
+# and Eclipse Distribution License v1.0 which accompany this distribution.
+#
+# The Eclipse Public License is available at
+#    https://eclipse.org/org/documents/epl-v10.php.
+# and the Eclipse Distribution License is available at
+#   https://eclipse.org/org/documents/edl-v10.php.
+#
+# Contributors:
+#    Pierre Goudet - initial committer
+
+module PahoMqtt
+  class Handler
+
+    attr_reader :registered_callback
+    attr_accessor :last_ping_resp
+    attr_accessor :clean_session
+
+    def initialize
+      @registered_callback = []
+      @last_ping_resp      = -1
+      @publisher           = nil
+      @subscriber          = nil
+    end
+
+    def config_pubsub(publisher, subscriber)
+      @publisher = publisher
+      @subscriber = subscriber
+    end
+
+    def socket=(socket)
+      @socket = socket
+    end
+
+    def receive_packet
+      result = IO.select([@socket], [], [], SELECT_TIMEOUT) unless @socket.nil? || @socket.closed?
+      unless result.nil?
+        packet = PahoMqtt::Packet::Base.read(@socket)
+        unless packet.nil?
+          if packet.is_a?(PahoMqtt::Packet::Connack)
+            @last_ping_resp = Time.now
+            handle_connack(packet)
+          else
+            handle_packet(packet)
+            @last_ping_resp = Time.now
+          end
+        end
+      end
+    end
+
+    def handle_packet(packet)
+      PahoMqtt.logger.info("New packet #{packet.class} received.") if PahoMqtt.logger?
+      type = packet_type(packet)
+      self.send("handle_#{type}", packet)
+    end
+
+    def register_topic_callback(topic, callback, &block)
+      if topic.nil?
+        PahoMqtt.logger.error("The topics where the callback is trying to be registered have been found nil.") if PahoMqtt.logger?
+        raise ArgumentError
+      end
+      clear_topic_callback(topic)
+      if block_given?
+        @registered_callback.push([topic, block])
+      elsif !(callback.nil?) && callback.is_a?(Proc)
+        @registered_callback.push([topic, callback])
+      end
+      MQTT_ERR_SUCCESS
+    end
+
+    def clear_topic_callback(topic)
+      if topic.nil?
+        PahoMqtt.logger.error("The topics where the callback is trying to be unregistered have been found nil.") if PahoMqtt.logger?
+        raise ArgumentError
+      end
+      @registered_callback.delete_if { |pair| pair.first == topic }
+      MQTT_ERR_SUCCESS
+    end
+
+    def handle_connack(packet)
+      if packet.return_code == 0x00
+        PahoMqtt.logger.debug(packet.return_msg) if PahoMqtt.logger?
+        handle_connack_accepted(packet.session_present)
+      else
+        PahoMqtt.logger.warm(packet.return_msg) if PahoMqtt.logger?
+        MQTT_CS_DISCONNECTED
+      end
+      @on_connack.call(packet) unless @on_connack.nil?
+      MQTT_CS_CONNECTED
+    end
+
+    def handle_connack_accepted(session_flag)
+      clean_session?(session_flag)
+      new_session?(session_flag)
+      old_session?(session_flag)
+    end
+
+    def new_session?(session_flag)
+      if !@clean_session && !session_flag
+        PahoMqtt.logger.debug("New session created for the client.") if PahoMqtt.logger?
+      end
+    end
+
+    def clean_session?(session_flag)
+      if @clean_session && !session_flag
+        PahoMqtt.logger.debug("No previous session found by server, starting a new one.") if PahoMqtt.logger?
+      end
+    end
+
+    def old_session?(session_flag)
+      if !@clean_session && session_flag
+        PahoMqtt.logger.debug("Previous session restored by the server.") if PahoMqtt.logger?
+      end
+    end
+
+    def handle_pingresp(_packet)
+      @last_ping_resp = Time.now
+    end
+
+    def handle_suback(packet)
+      max_qos = packet.return_codes
+      id      = packet.id
+      topics  = []
+      topics  = @subscriber.add_subscription(max_qos, id, topics)
+      unless topics.empty?
+        @on_suback.call(topics) unless @on_suback.nil?
+      end
+    end
+
+    def handle_unsuback(packet)
+      id = packet.id
+      topics = []
+      topics = @subscriber.remove_subscription(id, topics)
+      unless topics.empty?
+        @on_unsuback.call(topics) unless @on_unsuback.nil?
+      end
+    end
+
+    def handle_publish(packet)
+        id = packet.id
+        qos = packet.qos
+        if @publisher.do_publish(qos, id) == MQTT_ERR_SUCCESS
+          @on_message.call(packet) unless @on_message.nil?
+          check_callback(packet)
+        end
+    end
+
+    def handle_puback(packet)
+      id = packet.id
+      if @publisher.do_puback(id) == MQTT_ERR_SUCCESS
+        @on_puback.call(packet) unless @on_puback.nil?
+      end
+    end
+
+    def handle_pubrec(packet)
+      id = packet.id
+      if @publisher.do_pubrec(id) == MQTT_ERR_SUCCESS
+        @on_pubrec.call(packet) unless @on_pubrec.nil?
+      end
+    end
+
+    def handle_pubrel(packet)
+      id = packet.id
+      if @publisher.do_pubrel(id) == MQTT_ERR_SUCCESS
+        @on_pubrel.call(packet) unless @on_pubrel.nil?
+      end
+    end
+
+    def handle_pubcomp(packet)
+      id = packet.id
+      if @publisher.do_pubcomp(id) == MQTT_ERR_SUCCESS
+        @on_pubcomp.call(packet) unless @on_pubcomp.nil?
+      end
+    end
+
+    def on_connack(&block)
+      @on_connack = block if block_given?
+      @on_connack
+    end
+
+    def on_suback(&block)
+      @on_suback = block if block_given?
+      @on_suback
+    end
+
+    def on_unsuback(&block)
+      @on_unsuback = block if block_given?
+      @on_unsuback
+    end
+
+    def on_puback(&block)
+      @on_puback = block if block_given?
+      @on_puback
+    end
+
+    def on_pubrec(&block)
+      @on_pubrec = block if block_given?
+      @on_pubrec
+    end
+
+    def on_pubrel(&block)
+      @on_pubrel = block if block_given?
+      @on_pubrel
+    end
+
+    def on_pubcomp(&block)
+      @on_pubcomp = block if block_given?
+      @on_pubcomp
+    end
+
+    def on_message(&block)
+      @on_message = block if block_given?
+      @on_message
+    end
+
+    def on_connack=(callback)
+      @on_connack = callback if callback.is_a?(Proc)
+    end
+
+    def on_suback=(callback)
+      @on_suback = callback if callback.is_a?(Proc)
+    end
+
+    def on_unsuback=(callback)
+      @on_unsuback = callback if callback.is_a?(Proc)
+    end
+
+    def on_puback=(callback)
+      @on_puback = callback if callback.is_a?(Proc)
+    end
+
+    def on_pubrec=(callback)
+      @on_pubrec = callback if callback.is_a?(Proc)
+    end
+
+    def on_pubrel=(callback)
+      @on_pubrel = callback if callback.is_a?(Proc)
+    end
+
+    def on_pubcomp=(callback)
+      @on_pubcomp = callback if callback.is_a?(Proc)
+    end
+
+    def on_message=(callback)
+      @on_message = callback if callback.is_a?(Proc)
+    end
+
+    def packet_type(packet)
+      type = packet.class
+      if PahoMqtt::PACKET_TYPES[3..13].include?(type)
+        type.to_s.split('::').last.downcase
+      else
+        PahoMqtt.logger.error("Received an unexpeceted packet: #{packet}.") if PahoMqtt.logger?
+         raise PacketException.new('Invalid packet type id')
+      end
+    end
+
+    def check_callback(packet)
+      callbacks = []
+      @registered_callback.each do |reccord|
+        callbacks.push(reccord.last) if PahoMqtt.match_filter(packet.topic, reccord.first)
+      end
+      unless callbacks.empty?
+        callbacks.each do |callback|
+          callback.call(packet)
+        end
+      end
+    end
+  end
+end

data/lib/paho_mqtt/packet/base.rb

@@ -0,0 +1,315 @@
+# encoding: BINARY
+### original file from the ruby-mqtt gem
+### located at https://github.com/njh/ruby-mqtt/blob/master/lib/mqtt/packet.rb
+### Copyright (c) 2009-2013 Nicholas J Humfrey
+
+# Copyright (c) 2016-2017 Pierre Goudet <p-goudet@ruby-dev.jp>
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Eclipse Public License v1.0
+# and Eclipse Distribution License v1.0 which accompany this distribution.
+#
+# The Eclipse Public License is available at
+#    https://eclipse.org/org/documents/epl-v10.php.
+# and the Eclipse Distribution License is available at
+#   https://eclipse.org/org/documents/edl-v10.php.
+#
+# Contributors:
+#    Pierre Goudet - initial committer
+
+module PahoMqtt
+  module Packet
+    # Class representing a MQTT Packet
+    # Performs binary encoding and decoding of headers
+    class Base
+      # The version number of the MQTT protocol to use (default 3.1.0)
+      attr_accessor :version
+
+      # Identifier to link related control packets together
+      attr_accessor :id
+
+      # Array of 4 bits in the fixed header
+      attr_accessor :flags
+
+      # The length of the parsed packet body
+      attr_reader :body_length
+
+      # Default attribute values
+      ATTR_DEFAULTS = {
+        :version     => '3.1.0',
+        :id          => 0,
+        :body_length => nil
+      }
+
+      # Read in a packet from a socket
+      def self.read(socket)
+        # Read in the packet header and create a new packet object
+        packet = create_from_header(
+          read_byte(socket)
+        )
+
+        unless packet.nil?
+          packet.validate_flags
+
+          # Read in the packet length
+          multiplier = 1
+          body_length = 0
+          pos = 1
+          begin
+            digit = read_byte(socket)
+            body_length += ((digit & 0x7F) * multiplier)
+            multiplier *= 0x80
+            pos += 1
+          end while ((digit & 0x80) != 0x00) && pos <= 4
+
+          # Store the expected body length in the packet
+          packet.instance_variable_set('@body_length', body_length)
+
+          # Read in the packet body
+          packet.parse_body(socket.read(body_length))
+        end
+        packet
+      end
+
+      # Parse buffer into new packet object
+      def self.parse(buffer)
+        packet = parse_header(buffer)
+        packet.parse_body(buffer)
+        packet
+      end
+
+      # Parse the header and create a new packet object of the correct type
+      # The header is removed from the buffer passed into this function
+      def self.parse_header(buffer)
+        # Check that the packet is a long as the minimum packet size
+        if buffer.bytesize < 2
+          raise PahoMqtt::PacketFormatException.new(
+                  "Invalid packet: less than 2 bytes long")
+        end
+
+        # Create a new packet object
+        bytes = buffer.unpack("C5")
+        packet = create_from_header(bytes.first)
+        packet.validate_flags
+
+        # Parse the packet length
+        body_length = 0
+        multiplier = 1
+        pos = 1
+        begin
+          if buffer.bytesize <= pos
+            raise PahoMqtt::PacketFormatException.new(
+                    "The packet length header is incomplete")
+          end
+          digit = bytes[pos]
+          body_length += ((digit & 0x7F) * multiplier)
+          multiplier *= 0x80
+          pos += 1
+        end while ((digit & 0x80) != 0x00) && pos <= 4
+
+        # Store the expected body length in the packet
+        packet.instance_variable_set('@body_length', body_length)
+
+        # Delete the fixed header from the raw packet passed in
+        buffer.slice!(0...pos)
+
+        packet
+      end
+
+      # Create a new packet object from the first byte of a MQTT packet
+      def self.create_from_header(byte)
+        unless byte.nil?
+          # Work out the class
+          type_id = ((byte & 0xF0) >> 4)
+          packet_class = PahoMqtt::PACKET_TYPES[type_id]
+          if packet_class.nil?
+            raise PahoMqtt::PacketFormatException.new(
+                    "Invalid packet type identifier: #{type_id}")
+          end
+
+          # Convert the last 4 bits of byte into array of true/false
+          flags = (0..3).map { |i| byte & (2 ** i) != 0 }
+
+          # Create a new packet object
+          packet_class.new(:flags => flags)
+        end
+      end
+
+      # Create a new empty packet
+      def initialize(args={})
+        # We must set flags before the other values
+        @flags = [false, false, false, false]
+        update_attributes(ATTR_DEFAULTS.merge(args))
+      end
+
+      # Set packet attributes from a hash of attribute names and values
+      def update_attributes(attr={})
+        attr.each_pair do |k,v|
+          if v.is_a?(Array) || v.is_a?(Hash)
+            send("#{k}=", v.dup)
+          else
+            send("#{k}=", v)
+          end
+        end
+      end
+
+      # Get the identifer for this packet type
+      def type_id
+        index = PahoMqtt::PACKET_TYPES.index(self.class)
+        if index.nil?
+          raise PahoMqtt::PacketFormatException.new(
+                  "Invalid packet type: #{self.class}")
+        end
+        index
+      end
+
+      # Get the name of the packet type as a string in capitals
+      # (like the MQTT specification uses)
+      #
+      # Example: CONNACK
+      def type_name
+        self.class.name.split('::').last.upcase
+      end
+
+      # Set the protocol version number
+      def version=(arg)
+        @version = arg.to_s
+      end
+
+      # Set the length of the packet body
+      def body_length=(arg)
+        @body_length = arg.to_i
+      end
+
+      # Parse the body (variable header and payload) of a packet
+      def parse_body(buffer)
+        if buffer.bytesize != body_length
+          raise PahoMqtt::PacketFormatException.new(
+                  "Failed to parse packet - input buffer (#{buffer.bytesize}) is not the same as the body length header (#{body_length})")
+        end
+      end
+
+      # Get serialisation of packet's body (variable header and payload)
+      def encode_body
+        '' # No body by default
+      end
+
+      # Serialise the packet
+      def to_s
+        # Encode the fixed header
+        header = [
+          ((type_id.to_i & 0x0F) << 4) |
+          (flags[3] ? 0x8 : 0x0) |
+          (flags[2] ? 0x4 : 0x0) |
+          (flags[1] ? 0x2 : 0x0) |
+          (flags[0] ? 0x1 : 0x0)
+        ]
+
+        # Get the packet's variable header and payload
+        body = self.encode_body
+
+        # Check that that packet isn't too big
+        body_length = body.bytesize
+        if body_length > 268435455
+          raise PahoMqtt::PacketFormatException.new(
+          "Error serialising packet: body is more than 256MB")
+        end
+
+        # Build up the body length field bytes
+        begin
+          digit = (body_length % 128)
+          body_length = (body_length / 128)
+          # if there are more digits to encode, set the top bit of this digit
+          digit |= 0x80 if (body_length > 0)
+          header.push(digit)
+        end while (body_length > 0)
+
+        # Convert header to binary and add on body
+        header.pack('C*') + body
+      end
+
+      # Check that fixed header flags are valid for types that don't use the flags
+      # @private
+      def validate_flags
+        if flags != [false, false, false, false]
+          raise PahoMqtt::PacketFormatException.new(
+          "Invalid flags in #{type_name} packet header")
+        end
+      end
+
+      # Returns a human readable string
+      def inspect
+        "\#<#{self.class}>"
+      end
+
+      protected
+
+      # Encode an array of bytes and return them
+      def encode_bytes(*bytes)
+        bytes.pack('C*')
+      end
+
+      # Encode an array of bits and return them
+      def encode_bits(bits)
+        [bits.map { |b| b ? '1' : '0' }.join].pack('b*')
+      end
+
+      # Encode a 16-bit unsigned integer and return it
+      def encode_short(val)
+        [val.to_i].pack('n')
+      end
+
+      # Encode a UTF-8 string and return it
+      # (preceded by the length of the string)
+      def encode_string(str)
+        str = str.to_s.encode('UTF-8')
+
+        # Force to binary, when assembling the packet
+        str.force_encoding('ASCII-8BIT')
+        encode_short(str.bytesize) + str
+      end
+
+      # Remove a 16-bit unsigned integer from the front of buffer
+      def shift_short(buffer)
+        bytes = buffer.slice!(0..1)
+        bytes.unpack('n').first
+      end
+
+      # Remove one byte from the front of the string
+      def shift_byte(buffer)
+        buffer.slice!(0...1).unpack('C').first
+      end
+
+      # Remove 8 bits from the front of buffer
+      def shift_bits(buffer)
+        buffer.slice!(0...1).unpack('b8').first.split('').map { |b| b == '1' }
+      end
+
+      # Remove n bytes from the front of buffer
+      def shift_data(buffer,bytes)
+        buffer.slice!(0...bytes)
+      end
+
+      # Remove string from the front of buffer
+      def shift_string(buffer)
+        len = shift_short(buffer)
+        str = shift_data(buffer,len)
+        # Strings in MQTT v3.1 are all UTF-8
+        str.force_encoding('UTF-8')
+      end
+
+
+      private
+
+      # Read and unpack a single byte from a socket
+      def self.read_byte(socket)
+        byte = socket.read(1)
+        unless byte.nil?
+          byte.unpack('C').first
+        else
+          nil
+        end
+      end
+    end
+  end
+end

data/lib/paho_mqtt/packet/connack.rb

@@ -0,0 +1,102 @@
+# encoding: BINARY
+### original file from the ruby-mqtt gem
+### located at https://github.com/njh/ruby-mqtt/blob/master/lib/mqtt/packet.rb
+### Copyright (c) 2009-2013 Nicholas J Humfrey
+
+# Copyright (c) 2016-2017 Pierre Goudet <p-goudet@ruby-dev.jp>
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Eclipse Public License v1.0
+# and Eclipse Distribution License v1.0 which accompany this distribution.
+#
+# The Eclipse Public License is available at
+#    https://eclipse.org/org/documents/epl-v10.php.
+# and the Eclipse Distribution License is available at
+#   https://eclipse.org/org/documents/edl-v10.php.
+#
+# Contributors:
+#    Pierre Goudet - initial committer
+
+module PahoMqtt
+  module Packet
+    class Connack < PahoMqtt::Packet::Base
+      # Session Present flag
+      attr_accessor :session_present
+
+      # The return code (defaults to 0 for connection accepted)
+      attr_accessor :return_code
+
+      # Default attribute values
+      ATTR_DEFAULTS = { :return_code => 0x00 }
+
+      # Create a new Client Connect packet
+      def initialize(args={})
+        # We must set flags before other attributes
+        @connack_flags = [false, false, false, false, false, false, false, false]
+        super(ATTR_DEFAULTS.merge(args))
+      end
+
+      # Get the Session Present flag
+      def session_present
+        @connack_flags[0]
+      end
+
+      # Set the Session Present flag
+      def session_present=(arg)
+        if arg.kind_of?(Integer)
+          @connack_flags[0] = (arg == 0x1)
+        else
+          @connack_flags[0] = arg
+        end
+      end
+
+      # Get a string message corresponding to a return code
+      def return_msg
+        case return_code
+        when 0x00
+          "Connection accepted"
+        when 0x01
+          raise LowVersionException
+        when 0x02
+          "Connection refused: client identifier rejected"
+        when 0x03
+          "Connection refused: server unavailable"
+        when 0x04
+          "Connection refused: bad user name or password"
+        when 0x05
+          "Connection refused: not authorised"
+        else
+          "Connection refused: error code #{return_code}"
+        end
+      end
+
+      # Get serialisation of packet's body
+      def encode_body
+        body = ''
+        body += encode_bits(@connack_flags)
+        body += encode_bytes(@return_code.to_i)
+        body
+      end
+
+      # Parse the body (variable header and payload) of a Connect Acknowledgment packet
+      def parse_body(buffer)
+        super(buffer)
+        @connack_flags = shift_bits(buffer)
+        unless @connack_flags[1, 7] == [false, false, false, false, false, false, false]
+          raise PacketFormatException.new(
+                  "Invalid flags in Connack variable header")
+        end
+        @return_code = shift_byte(buffer)
+        unless buffer.empty?
+          raise PacketFormatException.new(
+                  "Extra bytes at end of Connect Acknowledgment packet")
+        end
+      end
+
+      # Returns a human readable string, summarising the properties of the packet
+      def inspect
+        "\#<#{self.class}: 0x%2.2X>" % return_code
+      end
+    end
+  end
+end