Style cleanup

Author: jhart-r7

lib/msf/core/auxiliary/mqtt.rb

@@ -5,8 +5,6 @@
 
 module Msf
   module Auxiliary::MQTT
-      include Exploit::Remote::Tcp
-
       def initialize(info = {})
         super
 
@@ -25,7 +23,7 @@ def initialize(info = {})
           ]
         )
 
-        register_autofilter_ports([Rex::Proto::MQTT::DEFAULT_PORT])
+        register_autofilter_ports([Rex::Proto::MQTT::DEFAULT_PORT, Rex::Proto::MQTT::DEFAULT_PORT])
       end
 
       def setup
@@ -44,9 +42,10 @@ def client_id
         datastore['CLIENT_ID'] || Rex::Text.rand_text_alpha(1 + rand(10))
       end
 
+      # creates a new mqtt client for use against the connected socket
       def mqtt_client
         client_opts = {
-          client_id: client_id().to_s,
+          client_id: client_id.to_s,
           username: datastore['USERNAME'],
           password: datastore['PASSWORD'],
           read_timeout: read_timeout
@@ -59,7 +58,7 @@ def mqtt_connect(client)
       end
 
       def mqtt_connect?(client)
-        mqtt_connect(client).return_code == 0
+        client.connect?
       end
 
       def mqtt_disconnect(client)

lib/rex/proto/mqtt/client.rb

@@ -29,7 +29,7 @@ def connect
         end
 
         def connect?
-          connect.return_code == 0
+          connect.return_code.zero?
         end
 
         def disconnect