ruflet_server 0.0.6 → 0.0.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2f889edb097984732cc8ccc1927674aa4f4b6fbd0b218b17cf352365100f4e9c
-  data.tar.gz: 37707b6486c6e6fb50ac40011ab3c80a6d667150957f3d07daf6aaeb649236e5
+  metadata.gz: f0ec355d06e9fb730a851fa6b085bc97b74bd1231aa682bfc03a93ed3078f762
+  data.tar.gz: 2c3af4192affd015f59bfc0a49dc4fe3af264f1da38f5914c78af3d8602ace57
 SHA512:
-  metadata.gz: ccb7dcf7dcbc3ab808b8bc300ffbc1da83b687256c9f892b664ce7b5e815b434c5b6a022c968b10fba3dc88ee5ee18ee7f1aed54ba55b9ea939e8f4804b04f69
-  data.tar.gz: 1d4cf5d940ab17a9681da6aa853303df9fbf4ee83d90264e7bbfd654c65757dd6144bc9aa46c91ffc1422b7981904048351f7f211c9d7d8501d4a91bf0595aa1
+  metadata.gz: 250363621bf1bfaefb04a2a2a89f30336590272250239c21061d96c60cac1ae5464f5a5442a42fab0f415bea3085a5921be8bb073e59d7d968b5bc537635eac9
+  data.tar.gz: 14d25116e02035b24646f73577aa3780b5f990473191725636324d26c53cccd3a65da53d84db3d2534a4a60450aac5ce7c79799f70b44703db271819c8f8f687

data/lib/ruflet/server/web_socket_connection.rb

@@ -2,6 +2,9 @@
 
 module Ruflet
   class WebSocketConnection
+    # Ruflet control messages are small; anything much larger is invalid or hostile.
+    MAX_FRAME_PAYLOAD_BYTES = 16 * 1024 * 1024
+
     def initialize(socket)
       @socket = socket
       @write_mutex = Mutex.new
@@ -70,10 +73,20 @@ module Ruflet
       masked = (b2 & 0x80) != 0
       payload_len = b2 & 0x7f
 
-      payload_len = read_exact(2).unpack1("n") if payload_len == 126
-      payload_len = read_exact(8).unpack1("Q>") if payload_len == 127
+      if payload_len == 126
+        ext = read_exact(2)
+        return nil if ext.nil?
+        payload_len = ext.unpack1("n")
+      elsif payload_len == 127
+        ext = read_exact(8)
+        return nil if ext.nil?
+        payload_len = ext.unpack1("Q>")
+      end
+
+      return nil if payload_len.negative? || payload_len > MAX_FRAME_PAYLOAD_BYTES
 
       masking_key = masked ? read_exact(4) : nil
+      return nil if masked && masking_key.nil?
       payload = payload_len.zero? ? "".b : read_exact(payload_len)
       return nil if payload.nil?
 
@@ -112,6 +125,9 @@ module Ruflet
     end
 
     def read_exact(length)
+      return nil unless length.is_a?(Integer)
+      return nil if length.negative? || length > MAX_FRAME_PAYLOAD_BYTES
+
       chunk = +""
       chunk.force_encoding(Encoding::BINARY)
 
@@ -123,6 +139,8 @@ module Ruflet
       end
 
       chunk
+    rescue IOError, SystemCallError
+      nil
     end
   end
 end

data/lib/ruflet/server/wire_codec.rb

@@ -128,10 +128,23 @@ module Ruflet
         when 0xd9 then reader.read_string(reader.read_u8)
         when 0xda then reader.read_string(reader.read_u16)
         when 0xdb then reader.read_string(reader.read_u32)
+        when 0xc4 then reader.read_binary(reader.read_u8)
+        when 0xc5 then reader.read_binary(reader.read_u16)
+        when 0xc6 then reader.read_binary(reader.read_u32)
         when 0xdc then read_array(reader, reader.read_u16)
         when 0xdd then read_array(reader, reader.read_u32)
         when 0xde then read_map(reader, reader.read_u16)
         when 0xdf then read_map(reader, reader.read_u32)
+        when 0xd4
+          read_ext(reader, 1)
+        when 0xd5
+          read_ext(reader, 2)
+        when 0xd6
+          read_ext(reader, 4)
+        when 0xd7
+          read_ext(reader, 8)
+        when 0xd8
+          read_ext(reader, 16)
         when 0xc7
           read_ext(reader, reader.read_u8)
         when 0xc8
@@ -231,6 +244,10 @@ module Ruflet
       def read_string(size)
         read_exact(size).force_encoding("UTF-8")
       end
+
+      def read_binary(size)
+        read_exact(size)
+      end
     end
   end
 end

data/lib/ruflet/server.rb

@@ -96,6 +96,33 @@ module Ruflet
       end
     end
 
+    def reload_app!
+      snapshots = @sessions_mutex.synchronize { @sessions.to_a }
+
+      snapshots.each do |session_key, current_page|
+        ws = @connections_mutex.synchronize { @connections[session_key] }
+        next unless ws
+
+        refreshed_page = Page.new(
+          session_id: current_page.session_id,
+          client_details: current_page.client_details,
+          sender: lambda do |action, payload|
+            send_message(ws, action, payload)
+          end
+        )
+        refreshed_page.title = "Ruflet App"
+
+        @sessions_mutex.synchronize do
+          @sessions[session_key] = refreshed_page
+        end
+
+        @app_block.call(refreshed_page)
+        refreshed_page.update
+      rescue StandardError => e
+        warn "reload error: #{e.class}: #{e.message}"
+      end
+    end
+
     private
 
     def trap_stop_signals
@@ -159,6 +186,8 @@ module Ruflet
       ws = nil
       begin
         path, headers = read_http_upgrade_request(socket)
+        return if path.nil?
+
         if websocket_upgrade_request?(path, headers)
           send_handshake_response(socket, headers["sec-websocket-key"])
           ws = Ruflet::WebSocketConnection.new(socket)
@@ -167,6 +196,8 @@ module Ruflet
           handle_http_request(socket, path)
         end
       rescue StandardError => e
+        return if disconnect_error?(e)
+
         warn "server error: #{e.class}: #{e.message}"
         warn e.backtrace.join("\n") if e.backtrace
         send_message(ws, Protocol::ACTIONS[:session_crashed], { "message" => e.message.to_s.dup.force_encoding("UTF-8") }) if ws
@@ -182,6 +213,8 @@ module Ruflet
         handle_message(ws, raw)
       end
     rescue StandardError => e
+      return if disconnect_error?(e)
+
       warn "server error: #{e.class}: #{e.message}"
       warn e.backtrace.join("\n") if e.backtrace
       send_message(ws, Protocol::ACTIONS[:session_crashed], { "message" => e.message.to_s.dup.force_encoding("UTF-8") })
@@ -197,10 +230,12 @@ module Ruflet
 
     def read_http_upgrade_request(socket)
       request_line = socket.gets("\r\n")
-      raise "Invalid HTTP request" if request_line.nil?
+      return [nil, {}] if request_line.nil?
+      return [nil, {}] unless request_line.include?(" ")
 
       method, path, _version = request_line.strip.split(" ", 3)
-      raise "Unsupported HTTP method: #{method}" unless method == "GET"
+      return [nil, {}] unless method == "GET"
+      return [nil, {}] if path.to_s.empty?
 
       headers = {}
       loop do
@@ -438,9 +473,9 @@ module Ruflet
       raise
     end
 
-    def on_invoke_control_method(_ws, _payload)
-      # Client response to invoke_control_method; no server-side handling yet.
-      nil
+    def on_invoke_control_method(ws, payload)
+      page = fetch_page(ws)
+      page.handle_invoke_method_result(payload)
     end
 
     def on_control_event(ws, payload)
@@ -482,10 +517,30 @@ module Ruflet
     end
 
     def send_message(ws, action, payload)
+      return if ws.nil? || ws.closed?
+
       message = [action, payload]
       ws.send_binary(Ruflet::WireCodec.pack(message))
     rescue StandardError => e
-      warn "send error: #{e.class}: #{e.message}"
+      unless disconnect_error?(e)
+        warn "send error: #{e.class}: #{e.message}"
+      end
+      remove_session(ws)
+      unregister_connection(ws)
+      ws&.close
+    end
+
+    def disconnect_error?(error)
+      return true if error.is_a?(IOError)
+      return true if error.is_a?(Errno::EPIPE)
+      return true if error.is_a?(Errno::ECONNRESET)
+      return true if error.is_a?(Errno::ECONNABORTED)
+      return true if error.is_a?(Errno::ENOTCONN)
+      return true if error.is_a?(Errno::ESHUTDOWN)
+      return true if error.is_a?(Errno::EBADF)
+      return true if error.is_a?(Errno::EINVAL)
+
+      false
     end
 
     def pseudo_uuid

data/lib/ruflet/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Ruflet
-  VERSION = "0.0.6" unless const_defined?(:VERSION)
+  VERSION = "0.0.8" unless const_defined?(:VERSION)
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruflet_server
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.8
 platform: ruby
 authors:
 - AdamMusa
@@ -15,14 +15,14 @@ dependencies:
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 0.0.6
+        version: 0.0.8
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 0.0.6
+        version: 0.0.8
 description: Ruflet WebSocket server runtime compatible with Flet protocol.
 email:
 - adammusa2222@gmail.com