mtproto 0.0.1 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c498fa937d08bb1896de2e69cfa96130b5d4fb6b1dae44761f8251ef426a03b5
-  data.tar.gz: ffc71857dcdf59248a9fd41750110a632db075558a8362780439fba2fc751d56
+  metadata.gz: fa11aaca7c34af33346fa59458bf300211785023ff833336738b0988deeafc32
+  data.tar.gz: 417cef055fd9d042956880f8e0214a10669759d7881e3af67e15e3bd2018de6c
 SHA512:
-  metadata.gz: 88b02a01b90bbfe65077bf9b0318ad95559a3933cd735f4e9bb11eee254864479f20cb15edb52b3f3ba6b972e779f4367318742b2d165498161779628ec64643
-  data.tar.gz: 30c8ea708f0b32d17c0a621cb7251a7eb1ea34450ceb3036340bd9989a86dc294aedab892030b72def482e3690fc16669457233b60471d2c4a6eb325d15b1b4f
+  metadata.gz: 630dadd58d377ed2342abddf659385e8864fbfca3e6c4e1e9fa5a861b29a7a4fcb7628f9e3c7341c716588c5b37c18944d14e954cf0304ced52fa8482dcf859b
+  data.tar.gz: 796286ff4a5b9b9a42d7b0c14616040f3b8e0b310c2afece12943f49c14622d1f82f5b4479ef23b47fa6cabd3216b95aa16d0eb1ba4459cddab039042b94dd2a

data/lib/mtproto/transport/abridged_packet_codec.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+module MTProto
+  module Transport
+    class AbridgedPacketCodec
+      TAG = "\xef".b
+      OBFUSCATE_TAG = "\xef\xef\xef\xef".b
+
+      def encode_packet(data)
+        length = data.bytesize >> 2
+
+        length_prefix = if length < 127
+          [length].pack('C')
+        else
+          "\x7f".b + [length].pack('L<')[0, 3]
+        end
+
+        length_prefix + data
+      end
+
+      def decode_packet(data)
+        length = data.unpack1('C')
+        offset = 1
+
+        if length >= 127
+          length = (data[1, 3] + "\x00").unpack1('L<')
+          offset = 4
+        end
+
+        actual_length = length << 2
+        data[offset, actual_length]
+      end
+    end
+  end
+end

data/lib/mtproto/transport/tcp_connection.rb

@@ -0,0 +1,91 @@
+# frozen_string_literal: true
+
+require 'socket'
+require 'timeout'
+
+module MTProto
+  module Transport
+    class ConnectionError < StandardError; end
+
+    class TCPConnection
+      attr_reader :host, :port, :codec
+
+      def initialize(host, port, codec)
+        @host = host
+        @port = port
+        @codec = codec
+        @socket = nil
+      end
+
+      def connect
+        return if connected?
+
+        @socket = TCPSocket.new(@host, @port)
+
+        send_init_tag if @codec.class.const_defined?(:TAG)
+      end
+
+      def connected?
+        !@socket.nil? && !@socket.closed?
+      end
+
+      def send(data)
+        raise ConnectionError, 'Not connected' unless connected?
+
+        encoded = @codec.encode_packet(data)
+        @socket.write(encoded)
+      end
+
+      def recv(timeout: 60)
+        raise ConnectionError, 'Not connected' unless connected?
+
+        Timeout.timeout(timeout) do
+          read_packet
+        end
+      rescue Timeout::Error
+        raise ConnectionError, 'Receive timeout'
+      end
+
+      def close
+        return unless @socket
+
+        @socket.close
+      rescue StandardError
+        nil
+      ensure
+        @socket = nil
+      end
+
+      private
+
+      def send_init_tag
+        tag = @codec.class.const_get(:TAG)
+        @socket.write(tag) if tag
+      end
+
+      def read_packet
+        first_byte = read_exactly(1)
+        length = first_byte.unpack1('C')
+
+        if length >= 127
+          length_bytes = read_exactly(3)
+          length = (length_bytes + "\x00").unpack1('L<')
+        end
+
+        actual_length = length << 2
+        read_exactly(actual_length)
+      end
+
+      def read_exactly(bytes_needed)
+        result = ''.b
+        while result.bytesize < bytes_needed
+          chunk = @socket.read(bytes_needed - result.bytesize)
+          raise ConnectionError, 'EOF while reading' if chunk.nil? || chunk.empty?
+
+          result += chunk
+        end
+        result
+      end
+    end
+  end
+end

data/lib/mtproto/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module MTProto
-  VERSION = '0.0.1'
-end
+  VERSION = '0.0.3'
+end

data/lib/mtproto.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+require_relative 'mtproto/version'
+require_relative 'mtproto/transport/abridged_packet_codec'
+require_relative 'mtproto/transport/tcp_connection'
+
+module MTProto
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mtproto
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.3
 platform: ruby
 authors:
 - Artem Levenkov
@@ -19,6 +19,9 @@ extra_rdoc_files: []
 files:
 - ".ruby-version"
 - Rakefile
+- lib/mtproto.rb
+- lib/mtproto/transport/abridged_packet_codec.rb
+- lib/mtproto/transport/tcp_connection.rb
 - lib/mtproto/version.rb
 homepage: https://github.com/alev-pro/mtproto-ruby
 licenses: