cosmos 4.5.0 → 5.0.2.pre.beta2

data/lib/cosmos/interfaces/tcpip_server_interface.rb

@@ -1,12 +1,21 @@
 # encoding: ascii-8bit
 
-# Copyright 2017 Ball Aerospace & Technologies Corp.
+# Copyright 2022 Ball Aerospace & Technologies Corp.
 # All Rights Reserved.
 #
 # This program is free software; you can modify and/or redistribute it
-# under the terms of the GNU General Public License
+# under the terms of the GNU Affero General Public License
 # as published by the Free Software Foundation; version 3 with
 # attribution addendums as found in the LICENSE.txt
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# This program may also be used under the terms of a commercial or
+# enterprise edition license of COSMOS if purchased from the
+# copyright holder
 
 require 'socket'
 require 'thread' # For Mutex
@@ -16,7 +25,6 @@ require 'cosmos/streams/tcpip_socket_stream'
 require 'cosmos/config/config_parser'
 
 module Cosmos
-
   # TCP/IP Server which can both read and write on a single port or two
   # independent ports. A listen thread is setup which waits for client
   # connections. For each connection to the read port, a thread is spawned that
@@ -28,6 +36,7 @@ module Cosmos
     # Data class which stores the interface and associated information
     class InterfaceInfo
       attr_reader :interface, :hostname, :host_ip, :port
+
       def initialize(interface, hostname, host_ip, port)
         @interface = interface
         @hostname = hostname
@@ -125,28 +134,24 @@ module Cosmos
 
       if @write_port
         @write_thread = Thread.new do
-          begin
-            loop do
-              write_thread_body()
-              break if @cancel_threads
-            end
-          rescue Exception => err
-            shutdown_interfaces(@write_interface_infos)
-            Logger.instance.error("#{@name}: Tcpip server write thread unexpectedly died")
-            Logger.instance.error(err.formatted)
+          loop do
+            write_thread_body()
+            break if @cancel_threads
           end
+        rescue Exception => err
+          shutdown_interfaces(@write_interface_infos)
+          Logger.error("#{@name}: Tcpip server write thread unexpectedly died")
+          Logger.error(err.formatted)
         end
         @write_raw_thread = Thread.new do
-          begin
-            loop do
-              write_raw_thread_body()
-              break if @cancel_threads
-            end
-          rescue Exception => err
-            shutdown_interfaces(@write_interface_infos)
-            Logger.instance.error("#{@name}: Tcpip server write raw thread unexpectedly died")
-            Logger.instance.error(err.formatted)
+          loop do
+            write_raw_thread_body()
+            break if @cancel_threads
           end
+        rescue Exception => err
+          shutdown_interfaces(@write_interface_infos)
+          Logger.error("#{@name}: Tcpip server write raw thread unexpectedly died")
+          Logger.error(err.formatted)
         end
       else
         @write_thread = nil
@@ -166,11 +171,9 @@ module Cosmos
       @cancel_threads = true
       @read_queue << nil if @read_queue
       @listen_pipes.each do |pipe|
-        begin
-          pipe.write('.')
-        rescue Exception
-          # Oh well
-        end
+        pipe.write('.')
+      rescue Exception
+        # Oh well
       end
       @listen_pipes.clear
 
@@ -180,11 +183,9 @@ module Cosmos
 
       # Shutdown listen socket(s)
       @listen_sockets.each do |listen_socket|
-        begin
-          Cosmos.close_socket(listen_socket)
-        rescue IOError
-          # Ok may have been closed by the thread
-        end
+        Cosmos.close_socket(listen_socket)
+      rescue IOError
+        # Ok may have been closed by the thread
       end
       @listen_sockets.clear
 
@@ -214,9 +215,12 @@ module Cosmos
     # @return [Packet] Latest packet read from any of the connected clients.
     #   Note this method blocks until data is available.
     def read
-      raise "Interface not connected for read: #{@name}" unless connected? && read_allowed?
+      raise "Interface not connected for read: #{@name}" unless connected?
+      raise "Interface not readable: #{@name}" unless read_allowed?
+
       packet = @read_queue.pop
       return nil unless packet
+
       @read_count += 1
       packet
     end
@@ -224,7 +228,9 @@ module Cosmos
     # @param packet [Packet] Packet to write to all clients connected to the
     #   write port.
     def write(packet)
-      raise "Interface not connected for write: #{@name}" unless connected? && write_allowed?
+      raise "Interface not connected for write: #{@name}" unless connected?
+      raise "Interface not writable: #{@name}" unless write_allowed?
+
       @write_count += 1
       @write_queue << packet.clone
       @write_condition_variable.broadcast
@@ -233,7 +239,9 @@ module Cosmos
     # @param data [String] Data to write to all clients connected to the
     #   write port.
     def write_raw(data)
-      raise "Interface not connected for write_raw: #{@name}" unless connected? && write_raw_allowed?
+      raise "Interface not connected for write_raw: #{@name}" unless connected?
+      raise "Interface not write-rawable: #{@name}" unless write_raw_allowed?
+
       @write_raw_queue << data
       @write_raw_condition_variable.broadcast
       return data
@@ -252,8 +260,8 @@ module Cosmos
     # @return [Integer] The number of connected clients
     def num_clients
       interfaces = []
-      @write_interface_infos.each {|wii| interfaces << wii.interface}
-      @read_interface_infos.each {|rii| interfaces << rii.interface}
+      @write_interface_infos.each { |wii| interfaces << wii.interface }
+      @read_interface_infos.each { |rii| interfaces << rii.interface }
       interfaces.uniq.length
     end
 
@@ -298,10 +306,10 @@ module Cosmos
     def change_raw_logging(method)
       if @raw_logger_pair
         @write_interface_infos.each do |interface_info|
-          interface_info.interface.raw_logger_pair.send(method) if interface_info.interface.raw_logger_pair
+          interface_info.interface.raw_logger_pair.public_send(method) if interface_info.interface.raw_logger_pair
         end
         @read_interface_infos.each do |interface_info|
-          interface_info.interface.raw_logger_pair.send(method) if interface_info.interface.raw_logger_pair
+          interface_info.interface.raw_logger_pair.public_send(method) if interface_info.interface.raw_logger_pair
         end
       end
     end
@@ -338,17 +346,15 @@ module Cosmos
       end
       @listen_sockets << listen_socket
       @listen_threads << Thread.new do
-        begin
-          thread_reader, thread_writer = IO.pipe
-          @listen_pipes << thread_writer
-          loop do
-            listen_thread_body(listen_socket, listen_write, listen_read, thread_reader)
-            break if @cancel_threads
-          end
-        rescue => err
-          Logger.instance.error("#{@name}: Tcpip server listen thread unexpectedly died")
-          Logger.instance.error(err.formatted)
+        thread_reader, thread_writer = IO.pipe
+        @listen_pipes << thread_writer
+        loop do
+          listen_thread_body(listen_socket, listen_write, listen_read, thread_reader)
+          break if @cancel_threads
         end
+      rescue => err
+        Logger.error("#{@name}: Tcpip server listen thread unexpectedly died")
+        Logger.error(err.formatted)
       end
     end
 
@@ -366,16 +372,16 @@ module Cosmos
 
       port, host_ip = Socket.unpack_sockaddr_in(address)
       hostname = ''
-      hostname = Socket.lookup_hostname_from_ip(host_ip) if System.instance.use_dns
-      if System.instance.acl
-        addr = ["AF_INET", 10, "lc630", host_ip.to_s]
-        if not System.instance.acl.allow_addr?(addr)
-          # Reject connection
-          Cosmos.close_socket(socket)
-          Logger.instance.info "#{@name}: Tcpip server rejected connection from #{hostname}(#{host_ip}):#{port}"
-          return
-        end
-      end
+      hostname = Socket.lookup_hostname_from_ip(host_ip)
+      # if System.instance.acl
+      #   addr = ["AF_INET", 10, "lc630", host_ip.to_s]
+      #   if not System.instance.acl.allow_addr?(addr)
+      #     # Reject connection
+      #     Cosmos.close_socket(socket)
+      #     Logger.info "#{@name}: Tcpip server rejected connection from #{hostname}(#{host_ip}):#{port}"
+      #     return
+      #   end
+      # end
 
       # Configure TCP_NODELAY option
       socket.setsockopt(Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1)
@@ -417,7 +423,7 @@ module Cosmos
         end
         start_read_thread(@read_interface_infos[-1])
       end
-      Logger.instance.info "#{@name}: Tcpip server accepted connection from #{hostname}(#{host_ip}):#{port}"
+      Logger.info "#{@name}: Tcpip server accepted connection from #{hostname}(#{host_ip}):#{port}"
     end
 
     def start_read_thread(interface_info)
@@ -427,34 +433,32 @@ module Cosmos
           begin
             read_thread_body(interface_info.interface)
           rescue Exception => err
-            Logger.instance.error "#{@name}: Tcpip server read thread unexpectedly died"
-            Logger.instance.error err.formatted
+            Logger.error "#{@name}: Tcpip server read thread unexpectedly died"
+            Logger.error err.formatted
           end
-          Logger.instance.info "#{@name}: Tcpip server lost read connection to #{interface_info.hostname}(#{interface_info.host_ip}):#{interface_info.port}"
+          Logger.info "#{@name}: Tcpip server lost read connection to #{interface_info.hostname}(#{interface_info.host_ip}):#{interface_info.port}"
           @read_threads.delete(Thread.current)
 
           index_to_delete = nil
           @connection_mutex.synchronize do
-            begin
-              index = 0
-              @read_interface_infos.each do |read_interface_info|
-                if interface_info.interface == read_interface_info.interface
-                  index_to_delete = index
-                  read_interface_info.interface.disconnect
-                  read_interface_info.interface.raw_logger_pair.stop if read_interface_info.interface.raw_logger_pair
-                  break
-                end
-                index += 1
-              end
-            ensure
-              if index_to_delete
-                @read_interface_infos.delete_at(index_to_delete)
+            index = 0
+            @read_interface_infos.each do |read_interface_info|
+              if interface_info.interface == read_interface_info.interface
+                index_to_delete = index
+                read_interface_info.interface.disconnect
+                read_interface_info.interface.raw_logger_pair.stop if read_interface_info.interface.raw_logger_pair
+                break
               end
+              index += 1
+            end
+          ensure
+            if index_to_delete
+              @read_interface_infos.delete_at(index_to_delete)
             end
           end
         rescue Exception => err
-          Logger.instance.error "#{@name}: Tcpip server read thread unexpectedly died"
-          Logger.instance.error err.formatted
+          Logger.error "#{@name}: Tcpip server read thread unexpectedly died"
+          Logger.error err.formatted
         end
       end
     end
@@ -466,6 +470,7 @@ module Cosmos
 
       loop do
         break if @cancel_threads
+
         begin
           packet = @write_queue.pop(true) # non_block to raise ThreadError
           break
@@ -484,6 +489,7 @@ module Cosmos
 
       loop do
         break if @cancel_threads
+
         begin
           data = @write_raw_queue.pop(true) # non_block to raise ThreadError
           break
@@ -500,8 +506,8 @@ module Cosmos
     end
 
     def interface_disconnect(interface_info)
-      Logger.instance.info "#{@name}: Tcpip server lost write connection to "\
-        "#{interface_info.hostname}(#{interface_info.host_ip}):#{interface_info.port}"
+      Logger.info "#{@name}: Tcpip server lost write connection to "\
+                           "#{interface_info.hostname}(#{interface_info.host_ip}):#{interface_info.port}"
       interface_info.interface.disconnect
       interface_info.interface.raw_logger_pair.stop if interface_info.interface.raw_logger_pair
     end
@@ -525,6 +531,7 @@ module Cosmos
           thread_bytes_read = interface_bytes_read
         end
         return if !packet || @cancel_threads
+
         packet = read_thread_hook(packet) # Do work on received packet
         @read_raw_data_time = interface.read_raw_data_time
         @read_raw_data = interface.read_raw_data
@@ -543,30 +550,28 @@ module Cosmos
 
       @connection_mutex.synchronize do
         @write_interface_infos.each do |interface_info|
-          begin
-            if (@write_port != @read_port)
-              # Socket should return EWOULDBLOCK if it is still cleanly connected
-              interface_info.interface.stream.write_socket.recvfrom_nonblock(10)
-            elsif (!interface_info.interface.stream.write_socket.closed?)
-              # Let read thread detect disconnect
-              next
-            end
-            # Client has disconnected (or is invalidly sending data on the socket)
-            Logger.instance.info "#{@name}: Tcpip server lost write connection to #{interface_info.hostname}(#{interface_info.host_ip}):#{interface_info.port}"
-            interface_info.interface.disconnect
-            interface_info.interface.raw_logger_pair.stop if interface_info.interface.raw_logger_pair
-            indexes_to_delete.unshift(index) # Put later indexes at front of array
-          rescue Errno::ECONNRESET, Errno::ECONNABORTED, IOError
-            # Client has disconnected
-            Logger.instance.info "#{@name}: Tcpip server lost write connection to #{interface_info.hostname}(#{interface_info.host_ip}):#{interface_info.port}"
-            interface_info.interface.disconnect
-            interface_info.interface.raw_logger_pair.stop if interface_info.interface.raw_logger_pair
-            indexes_to_delete.unshift(index) # Put later indexes at front of array
-          rescue Errno::EWOULDBLOCK
-            # Client is still cleanly connected as far as we can tell without writing to the socket
-          ensure
-            index += 1
+          if @write_port != @read_port
+            # Socket should return EWOULDBLOCK if it is still cleanly connected
+            interface_info.interface.stream.write_socket.recvfrom_nonblock(10)
+          elsif !interface_info.interface.stream.write_socket.closed?
+            # Let read thread detect disconnect
+            next
           end
+          # Client has disconnected (or is invalidly sending data on the socket)
+          Logger.info "#{@name}: Tcpip server lost write connection to #{interface_info.hostname}(#{interface_info.host_ip}):#{interface_info.port}"
+          interface_info.interface.disconnect
+          interface_info.interface.raw_logger_pair.stop if interface_info.interface.raw_logger_pair
+          indexes_to_delete.unshift(index) # Put later indexes at front of array
+        rescue Errno::ECONNRESET, Errno::ECONNABORTED, IOError
+          # Client has disconnected
+          Logger.info "#{@name}: Tcpip server lost write connection to #{interface_info.hostname}(#{interface_info.host_ip}):#{interface_info.port}"
+          interface_info.interface.disconnect
+          interface_info.interface.raw_logger_pair.stop if interface_info.interface.raw_logger_pair
+          indexes_to_delete.unshift(index) # Put later indexes at front of array
+        rescue Errno::EWOULDBLOCK
+          # Client is still cleanly connected as far as we can tell without writing to the socket
+        ensure
+          index += 1
         end
 
         # Delete any dead sockets
@@ -590,7 +595,7 @@ module Cosmos
           need_disconnect = false
           begin
             interface_bytes_written = interface_info.interface.bytes_written
-            interface_info.interface.send(method, packet_or_data)
+            interface_info.interface.public_send(method, packet_or_data)
             diff = interface_info.interface.bytes_written - interface_bytes_written
             @written_raw_data_time = interface_info.interface.written_raw_data_time
             @written_raw_data = interface_info.interface.written_raw_data
@@ -600,13 +605,13 @@ module Cosmos
             need_disconnect = true
           rescue Exception => err
             if err.message != "Stream not connected for write_raw"
-              Logger.instance.error "#{@name}: Error sending to client: #{err.class} #{err.message}"
+              Logger.error "#{@name}: Error sending to client: #{err.class} #{err.message}"
             end
             need_disconnect = true
           end
 
           if need_disconnect
-            Logger.instance.info "#{@name}: Tcpip server lost write connection to #{interface_info.hostname}(#{interface_info.host_ip}):#{interface_info.port}"
+            Logger.info "#{@name}: Tcpip server lost write connection to #{interface_info.hostname}(#{interface_info.host_ip}):#{interface_info.port}"
             interface_info.interface.disconnect
             interface_info.interface.raw_logger_pair.stop if interface_info.interface.raw_logger_pair
             indexes_to_delete.unshift(index) # Put later indexes at front of array

data/lib/cosmos/interfaces/udp_interface.rb

@@ -1,12 +1,21 @@
 # encoding: ascii-8bit
 
-# Copyright 2017 Ball Aerospace & Technologies Corp.
+# Copyright 2022 Ball Aerospace & Technologies Corp.
 # All Rights Reserved.
 #
 # This program is free software; you can modify and/or redistribute it
-# under the terms of the GNU General Public License
+# under the terms of the GNU Affero General Public License
 # as published by the Free Software Foundation; version 3 with
 # attribution addendums as found in the LICENSE.txt
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# This program may also be used under the terms of a commercial or
+# enterprise edition license of COSMOS if purchased from the
+# copyright holder
 
 require 'cosmos/interfaces/interface'
 require 'cosmos/io/udp_sockets'
@@ -36,7 +45,8 @@ module Cosmos
       ttl = 128, # default for Windows
       write_timeout = 10.0,
       read_timeout = nil,
-      bind_address = '0.0.0.0')
+      bind_address = '0.0.0.0'
+    )
 
       super()
       @hostname = ConfigParser.handle_nil(hostname)
@@ -82,21 +92,24 @@ module Cosmos
           @write_dest_port,
           @hostname,
           @interface_address,
-          @ttl)
+          @ttl
+        )
         @write_socket = @read_socket
       else
         @read_socket = UdpReadSocket.new(
           @read_port,
           @hostname,
           @interface_address,
-          @bind_address) if @read_port
+          @bind_address
+        ) if @read_port
         @write_socket = UdpWriteSocket.new(
           @hostname,
           @write_dest_port,
           @write_src_port,
           @interface_address,
           @ttl,
-          @bind_address) if @write_dest_port
+          @bind_address
+        ) if @write_dest_port
       end
       @thread_sleeper = nil
     end
@@ -128,6 +141,7 @@ module Cosmos
 
     def read
       return super() if @read_port
+
       # Write only interface so stop the thread which calls read
       @thread_sleeper = Sleeper.new
       @thread_sleeper.sleep(1_000_000_000) while connected?
@@ -137,6 +151,7 @@ module Cosmos
     # Reads from the socket if the read_port is defined
     def read_interface
       data = @read_socket.read(@read_timeout)
+      Logger.info "#{@name}: Udp read returned 0 bytes (stream closed)" if data.length <= 0
       read_interface_base(data)
       return data
     rescue IOError # Disconnected

data/lib/cosmos/interfaces.rb

@@ -1,24 +1,44 @@
 # encoding: ascii-8bit
 
-require 'cosmos/interfaces/interface'
-require 'cosmos/interfaces/stream_interface'
-require 'cosmos/interfaces/cmd_tlm_server_interface'
-require 'cosmos/interfaces/serial_interface'
-require 'cosmos/interfaces/simulated_target_interface'
-require 'cosmos/interfaces/tcpip_client_interface'
-require 'cosmos/interfaces/tcpip_server_interface'
-require 'cosmos/interfaces/udp_interface'
-require 'cosmos/interfaces/linc_interface'
-require 'cosmos/interfaces/dart_status_interface'
+# Copyright 2022 Ball Aerospace & Technologies Corp.
+# All Rights Reserved.
+#
+# This program is free software; you can modify and/or redistribute it
+# under the terms of the GNU Affero General Public License
+# as published by the Free Software Foundation; version 3 with
+# attribution addendums as found in the LICENSE.txt
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# This program may also be used under the terms of a commercial or
+# enterprise edition license of COSMOS if purchased from the
+# copyright holder
 
-require 'cosmos/interfaces/protocols/protocol'
-require 'cosmos/interfaces/protocols/burst_protocol'
-require 'cosmos/interfaces/protocols/fixed_protocol'
-require 'cosmos/interfaces/protocols/length_protocol'
-require 'cosmos/interfaces/protocols/preidentified_protocol'
-require 'cosmos/interfaces/protocols/template_protocol'
-require 'cosmos/interfaces/protocols/terminated_protocol'
+module Cosmos
+  autoload(:Interface, 'cosmos/interfaces/interface.rb')
+  autoload(:StreamInterface, 'cosmos/interfaces/stream_interface.rb')
+  autoload(:SerialInterface, 'cosmos/interfaces/serial_interface.rb')
+  autoload(:SimulatedTargetInterface, 'cosmos/interfaces/simulated_target_interface.rb')
+  autoload(:TcpipClientInterface, 'cosmos/interfaces/tcpip_client_interface.rb')
+  autoload(:TcpipServerInterface, 'cosmos/interfaces/tcpip_server_interface.rb')
+  autoload(:UdpInterface, 'cosmos/interfaces/udp_interface.rb')
+  autoload(:LincInterface, 'cosmos/interfaces/linc_interface.rb')
+  autoload(:LincHandshakeCommand, 'cosmos/interfaces/linc_interface.rb')
+  autoload(:LincHandshake, 'cosmos/interfaces/linc_interface.rb')
+  autoload(:DartStatusInterface, 'cosmos/interfaces/dart_status_interface.rb')
 
-require 'cosmos/interfaces/protocols/override_protocol'
-require 'cosmos/interfaces/protocols/crc_protocol'
-require 'cosmos/interfaces/protocols/ignore_packet_protocol'
+  autoload(:Protocol, 'cosmos/interfaces/protocols/protocol.rb')
+  autoload(:BurstProtocol, 'cosmos/interfaces/protocols/burst_protocol.rb')
+  autoload(:FixedProtocol, 'cosmos/interfaces/protocols/fixed_protocol.rb')
+  autoload(:LengthProtocol, 'cosmos/interfaces/protocols/length_protocol.rb')
+  autoload(:PreidentifiedProtocol, 'cosmos/interfaces/protocols/preidentified_protocol.rb')
+  autoload(:TemplateProtocol, 'cosmos/interfaces/protocols/template_protocol.rb')
+  autoload(:TerminatedProtocol, 'cosmos/interfaces/protocols/terminated_protocol.rb')
+
+  autoload(:OverrideProtocol, 'cosmos/interfaces/protocols/override_protocol.rb')
+  autoload(:CrcProtocol, 'cosmos/interfaces/protocols/crc_protocol.rb')
+  autoload(:IgnorePacketProtocol, 'cosmos/interfaces/protocols/ignore_packet_protocol.rb')
+end

data/lib/cosmos/io/buffered_file.rb

@@ -1,12 +1,21 @@
 # encoding: ascii-8bit
 
-# Copyright 2017 Ball Aerospace & Technologies Corp.
+# Copyright 2022 Ball Aerospace & Technologies Corp.
 # All Rights Reserved.
 #
 # This program is free software; you can modify and/or redistribute it
-# under the terms of the GNU General Public License
+# under the terms of the GNU Affero General Public License
 # as published by the Free Software Foundation; version 3 with
 # attribution addendums as found in the LICENSE.txt
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# This program may also be used under the terms of a commercial or
+# enterprise edition license of COSMOS if purchased from the
+# copyright holder
 
 if RUBY_ENGINE == 'ruby' and !ENV['COSMOS_NO_EXT']
   require 'cosmos/ext/buffered_file'
@@ -51,6 +60,7 @@ else
           if @buffer.length <= 0
             return nil
           end
+
           if read_length <= @buffer.length
             result = @buffer[@buffer_index, read_length]
             @buffer_index += read_length
@@ -94,7 +104,6 @@ else
         @buffer_index = 0
         return super(*args)
       end
-
     end
   end
 end

data/lib/cosmos/io/cosmos_snmp.rb

@@ -1,12 +1,21 @@
 # encoding: ascii-8bit
 
-# Copyright 2014 Ball Aerospace & Technologies Corp.
+# Copyright 2022 Ball Aerospace & Technologies Corp.
 # All Rights Reserved.
 #
 # This program is free software; you can modify and/or redistribute it
-# under the terms of the GNU General Public License
+# under the terms of the GNU Affero General Public License
 # as published by the Free Software Foundation; version 3 with
 # attribution addendums as found in the LICENSE.txt
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# This program may also be used under the terms of a commercial or
+# enterprise edition license of COSMOS if purchased from the
+# copyright holder
 
 # This file updates the Ruby SNMP gem for non-blocking sockets
 # to work around potential process lockups.
@@ -22,13 +31,12 @@ module SNMP
     def recv(max_bytes)
       # Implement blocking using recvfrom_nonblock to prevent potential
       # ruby thread deadlock
-      begin
-        data, host_info = @socket.recvfrom_nonblock(max_bytes)
-        return data
-      rescue Errno::EAGAIN, Errno::EWOULDBLOCK
-        IO.fast_select([@socket])
-        retry
-      end
+
+      data, host_info = @socket.recvfrom_nonblock(max_bytes)
+      return data
+    rescue Errno::EAGAIN, Errno::EWOULDBLOCK
+      IO.fast_select([@socket])
+      retry
     end
   end
 

data/lib/cosmos/io/io_multiplexer.rb

@@ -1,19 +1,26 @@
 # encoding: ascii-8bit
 
-# Copyright 2014 Ball Aerospace & Technologies Corp.
+# Copyright 2022 Ball Aerospace & Technologies Corp.
 # All Rights Reserved.
 #
 # This program is free software; you can modify and/or redistribute it
-# under the terms of the GNU General Public License
+# under the terms of the GNU Affero General Public License
 # as published by the Free Software Foundation; version 3 with
 # attribution addendums as found in the LICENSE.txt
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# This program may also be used under the terms of a commercial or
+# enterprise edition license of COSMOS if purchased from the
+# copyright holder
 
 module Cosmos
-
   # Adds IO streams and then defers to the streams when using any of the Ruby
   # output methods such as print, puts, etc.
   class IoMultiplexer
-
     # Create the empty stream array
     def initialize
       @streams = []
@@ -40,11 +47,15 @@ module Cosmos
       result = nil
       @streams.each do |stream|
         if first
-          result = stream.send(method_name, *args)
+          # Fortify Access Specifier Manipulation
+          # We're forwarding only public methods to the stream
+          result = stream.public_send(method_name, *args)
           result = self if result == stream
           first = false
         else
-          stream.send(method_name, *args)
+          # Fortify Access Specifier Manipulation
+          # We're forwarding only public methods to the stream
+          stream.public_send(method_name, *args)
         end
       end
       result
@@ -65,7 +76,5 @@ module Cosmos
     def remove_stream(stream)
       @streams.delete(stream)
     end
-
   end # class IoMultiplexer
-
 end # module Cosmos