Fix protected method call & init_ui

sjanusz-r7 · Sep 21, 2023 · 299a0ac · 299a0ac
1 parent f2020ce
commit 299a0ac
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/lib/msf/base/sessions/postgresql.rb b/lib/msf/base/sessions/postgresql.rb
@@ -62,20 +62,16 @@ def desc
     'PostgreSQL'
   end
 
-  protected
-
   ##
   # :category: Msf::Session::Interactive implementors
   #
   # Initializes the console's I/O handles.
   #
   def init_ui(input, output)
-    self.user_input = input
-    self.user_output = output
-    self.console.init_ui(self.user_input, self.user_output)
-    self.console.set_log_source(self.log_source)
+    super(input, output)
 
-    super
+    self.console.init_ui(input, output)
+    self.console.set_log_source(self.log_source)
   end
 
   ##
@@ -88,6 +84,8 @@ def reset_ui
     self.console.reset_ui
   end
 
+  protected
+
   ##
   # :category: Msf::Session::Interactive implementors
   #