mtproto 0.0.4 → 0.0.6

data/lib/mtproto/tl/new_session_created.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module MTProto
+  module TL
+    class NewSessionCreated
+      CONSTRUCTOR = 0x9ec20908
+
+      attr_reader :first_msg_id, :unique_id, :server_salt
+
+      def self.deserialize(data)
+        offset = 4
+        first_msg_id = data[offset, 8].unpack1('Q<')
+        offset += 8
+
+        unique_id = data[offset, 8].unpack1('Q<')
+        offset += 8
+
+        server_salt = data[offset, 8].unpack1('Q<')
+
+        new(first_msg_id: first_msg_id, unique_id: unique_id, server_salt: server_salt)
+      end
+
+      def initialize(first_msg_id:, unique_id:, server_salt:)
+        @first_msg_id = first_msg_id
+        @unique_id = unique_id
+        @server_salt = server_salt
+      end
+    end
+  end
+end

data/lib/mtproto/tl/p_q_inner_data.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+require_relative 'serializer'
+
+module MTProto
+  module TL
+    class PQInnerData
+      CONSTRUCTOR_DC = 0xa9f55f95
+      CONSTRUCTOR_TEMP_DC = 0x56fddf88
+
+      attr_reader :pq, :p, :q, :nonce, :server_nonce, :new_nonce, :dc, :expires_in
+
+      def initialize(pq:, p:, q:, nonce:, server_nonce:, new_nonce:, dc:, expires_in: nil)
+        @pq = pq
+        @p = p
+        @q = q
+        @nonce = nonce
+        @server_nonce = server_nonce
+        @new_nonce = new_nonce
+        @dc = dc
+        @expires_in = expires_in
+      end
+
+      def serialize
+        constructor = @expires_in ? CONSTRUCTOR_TEMP_DC : CONSTRUCTOR_DC
+
+        data = Serializer.serialize_int(constructor)
+        data += Serializer.serialize_bytes(Serializer.integer_to_bytes(@pq))
+        data += Serializer.serialize_bytes(Serializer.integer_to_bytes(@p))
+        data += Serializer.serialize_bytes(Serializer.integer_to_bytes(@q))
+        data += @nonce
+        data += @server_nonce
+        data += @new_nonce
+
+        data += Serializer.serialize_int(@expires_in) if @expires_in
+
+        data + Serializer.serialize_int(@dc)
+      end
+    end
+  end
+end

data/lib/mtproto/tl/rpc_error.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module MTProto
+  module TL
+    class RpcError
+      CONSTRUCTOR = 0x2144ca19
+
+      attr_reader :error_code, :error_message
+
+      def self.deserialize(data)
+        offset = 4
+        error_code = data[offset, 4].unpack1('l<')
+        offset += 4
+
+        message_length = data[offset].ord
+        offset += 1
+
+        error_message = data[offset, message_length]
+        offset += message_length
+
+        new(error_code: error_code, error_message: error_message)
+      end
+
+      def initialize(error_code:, error_message:)
+        @error_code = error_code
+        @error_message = error_message
+      end
+
+      def to_s
+        "RPC Error #{@error_code}: #{@error_message}"
+      end
+    end
+  end
+end

data/lib/mtproto/tl/sent_code.rb

@@ -0,0 +1,128 @@
+# frozen_string_literal: true
+
+module MTProto
+  module TL
+    class SentCode
+      CONSTRUCTOR = 0x5e002502
+
+      attr_reader :flags, :type, :phone_code_hash, :next_type, :timeout
+
+      def self.deserialize(data)
+        offset = 4
+
+        flags = data[offset, 4].unpack1('L<')
+        offset += 4
+
+        type, type_bytes_read = SentCodeType.deserialize_from(data[offset..])
+        offset += type_bytes_read
+
+        phone_code_hash_length = data[offset].ord
+        offset += 1
+
+        phone_code_hash = data[offset, phone_code_hash_length]
+        offset += phone_code_hash_length
+
+        padding = (4 - ((phone_code_hash_length + 1) % 4)) % 4
+        offset += padding
+
+        next_type = nil
+        if (flags & (1 << 1)) != 0
+          next_type, next_type_bytes_read = CodeType.deserialize_from(data[offset..])
+          offset += next_type_bytes_read
+        end
+
+        timeout = nil
+        if (flags & (1 << 2)) != 0
+          timeout = data[offset, 4].unpack1('L<')
+        end
+
+        new(
+          flags: flags,
+          type: type,
+          phone_code_hash: phone_code_hash,
+          next_type: next_type,
+          timeout: timeout
+        )
+      end
+
+      def initialize(flags:, type:, phone_code_hash:, next_type: nil, timeout: nil)
+        @flags = flags
+        @type = type
+        @phone_code_hash = phone_code_hash
+        @next_type = next_type
+        @timeout = timeout
+      end
+
+      def to_h
+        {
+          phone_code_hash: @phone_code_hash,
+          type: @type,
+          next_type: @next_type,
+          timeout: @timeout
+        }.compact
+      end
+    end
+
+    module SentCodeType
+      def self.deserialize_from(data)
+        constructor = data[0, 4].unpack1('L<')
+        offset = 4
+
+        case constructor
+        when 0x3dbb5986 # auth.sentCodeTypeApp
+          length = data[offset, 4].unpack1('L<')
+          offset += 4
+          [{ _: :sent_code_type_app, length: length }, offset]
+        when 0xc000bba2 # auth.sentCodeTypeSms
+          length = data[offset, 4].unpack1('L<')
+          offset += 4
+          [{ _: :sent_code_type_sms, length: length }, offset]
+        when 0x5353e5a7 # auth.sentCodeTypeCall
+          length = data[offset, 4].unpack1('L<')
+          offset += 4
+          [{ _: :sent_code_type_call, length: length }, offset]
+        when 0xab03c6d9 # auth.sentCodeTypeFlashCall
+          pattern_length = data[offset].ord
+          offset += 1
+          pattern = data[offset, pattern_length]
+          offset += pattern_length
+          padding = (4 - ((pattern_length + 1) % 4)) % 4
+          offset += padding
+          [{ _: :sent_code_type_flash_call, pattern: pattern }, offset]
+        when 0x82006484 # auth.sentCodeTypeMissedCall
+          prefix_length = data[offset].ord
+          offset += 1
+          prefix = data[offset, prefix_length]
+          offset += prefix_length
+          padding = (4 - ((prefix_length + 1) % 4)) % 4
+          offset += padding
+          length = data[offset, 4].unpack1('L<')
+          offset += 4
+          [{ _: :sent_code_type_missed_call, prefix: prefix, length: length }, offset]
+        else
+          raise "Unknown SentCodeType constructor: 0x#{constructor.to_s(16)}"
+        end
+      end
+    end
+
+    module CodeType
+      def self.deserialize_from(data)
+        constructor = data[0, 4].unpack1('L<')
+        offset = 4
+
+        case constructor
+        when 0x72a3158c # codeTypeSms
+          [{ _: :code_type_sms }, offset]
+        when 0x741cd3e3 # codeTypeCall
+          [{ _: :code_type_call }, offset]
+        when 0x226ccefb # codeTypeFlashCall
+          [{ _: :code_type_flash_call }, offset]
+        when 0xd61ad6ee # codeTypeMissedCall
+          [{ _: :code_type_missed_call }, offset]
+        else
+          raise "Unknown CodeType constructor: 0x#{constructor.to_s(16)}"
+        end
+      end
+    end
+  end
+end

data/lib/mtproto/tl/serializer.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+module MTProto
+  module TL
+    module Serializer
+      module_function
+
+      def serialize_int(value)
+        [value].pack('L<')
+      end
+
+      def serialize_long(value)
+        [value].pack('Q<')
+      end
+
+      def serialize_int128(value)
+        value.is_a?(String) ? value : [value].pack('Q<Q<')
+      end
+
+      def serialize_int256(value)
+        value.is_a?(String) ? value : [value].pack('Q<Q<Q<Q<')
+      end
+
+      def serialize_bytes(bytes)
+        length = bytes.bytesize
+
+        if length <= 253
+          [length].pack('C').b + bytes.b + padding(length + 1)
+        else
+          [254].pack('C').b + [length].pack('L<')[0, 3].b + bytes.b + padding(length + 4)
+        end
+      end
+
+      def serialize_string(str)
+        serialize_bytes(str)
+      end
+
+      def padding(current_length)
+        pad_length = (4 - (current_length % 4)) % 4
+        ("\x00" * pad_length).b
+      end
+
+      def integer_to_bytes(int)
+        return "\x00" if int.zero?
+
+        bytes = []
+        while int > 0
+          bytes.unshift(int & 0xff)
+          int >>= 8
+        end
+        bytes.pack('C*')
+      end
+    end
+  end
+end

data/lib/mtproto/tl/server_dh_inner_data.rb

@@ -0,0 +1,85 @@
+# frozen_string_literal: true
+
+module MTProto
+  module TL
+    class ServerDHInnerData
+      CONSTRUCTOR = 0xb5890dba
+
+      attr_reader :nonce, :server_nonce, :g, :dh_prime, :g_a, :server_time
+
+      def self.deserialize(data)
+        constructor = data[0, 4].unpack1('L<')
+        raise "Unexpected constructor: 0x#{constructor.to_s(16)}" unless constructor == CONSTRUCTOR
+
+        offset = 4
+
+        nonce = data[offset, 16]
+        offset += 16
+
+        server_nonce = data[offset, 16]
+        offset += 16
+
+        g = data[offset, 4].unpack1('L<')
+        offset += 4
+
+        dh_prime_length_byte = data[offset].ord
+        offset += 1
+
+        if dh_prime_length_byte == 254
+          length_bytes = data[offset, 3].bytes
+          dh_prime_length = length_bytes[0] | (length_bytes[1] << 8) | (length_bytes[2] << 16)
+          offset += 3
+        else
+          dh_prime_length = dh_prime_length_byte
+        end
+
+        dh_prime = data[offset, dh_prime_length]
+        offset += dh_prime_length
+        offset += padding_length(dh_prime_length_byte == 254 ? dh_prime_length + 4 : dh_prime_length + 1)
+
+        g_a_length_byte = data[offset].ord
+        offset += 1
+
+        if g_a_length_byte == 254
+          length_bytes = data[offset, 3].bytes
+          g_a_length = length_bytes[0] | (length_bytes[1] << 8) | (length_bytes[2] << 16)
+          offset += 3
+        else
+          g_a_length = g_a_length_byte
+        end
+
+        g_a = data[offset, g_a_length]
+        offset += g_a_length
+        offset += padding_length(g_a_length_byte == 254 ? g_a_length + 4 : g_a_length + 1)
+
+        server_time = data[offset, 4].unpack1('L<')
+
+        new(
+          nonce: nonce,
+          server_nonce: server_nonce,
+          g: g,
+          dh_prime: dh_prime,
+          g_a: g_a,
+          server_time: server_time
+        )
+      end
+
+      def initialize(nonce:, server_nonce:, g:, dh_prime:, g_a:, server_time:)
+        @nonce = nonce
+        @server_nonce = server_nonce
+        @g = g
+        @dh_prime = dh_prime
+        @g_a = g_a
+        @server_time = server_time
+      end
+
+      def self.padding_length(length)
+        (4 - (length % 4)) % 4
+      end
+
+      def padding_length(length)
+        self.class.padding_length(length)
+      end
+    end
+  end
+end

data/lib/mtproto/transport/tcp_connection.rb

@@ -17,7 +17,7 @@ module MTProto
         @socket = nil
       end
 
-      def connect
+      def connect!
         return if connected?
 
         @socket = TCPSocket.new(@host, @port)

data/lib/mtproto/version.rb

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

data/lib/mtproto.rb

@@ -1,11 +1,35 @@
 # frozen_string_literal: true
 
 require_relative 'mtproto/version'
+require_relative 'mtproto/errors'
 require_relative 'mtproto/transport/abridged_packet_codec'
 require_relative 'mtproto/transport/tcp_connection'
 require_relative 'mtproto/tl/message'
+require_relative 'mtproto/tl/serializer'
+require_relative 'mtproto/tl/p_q_inner_data'
+require_relative 'mtproto/tl/server_dh_inner_data'
+require_relative 'mtproto/tl/client_dh_inner_data'
+require_relative 'mtproto/tl/bad_msg_notification'
+require_relative 'mtproto/tl/msg_container'
+require_relative 'mtproto/tl/new_session_created'
+require_relative 'mtproto/tl/rpc_error'
+require_relative 'mtproto/tl/gzip_packed'
+require_relative 'mtproto/tl/config'
+require_relative 'mtproto/tl/code_settings'
+require_relative 'mtproto/tl/sent_code'
 require_relative 'mtproto/crypto/rsa_key'
+require_relative 'mtproto/crypto/factorization'
+require_relative 'mtproto/crypto/aes_ige'
+require_relative 'mtproto/crypto/rsa_pad'
+require_relative 'mtproto/crypto/auth_key_helper'
+require_relative 'mtproto/crypto/dh_validator'
+require_relative 'mtproto/crypto/dh_key_exchange'
+require_relative 'mtproto/crypto/message_key'
+require_relative 'mtproto/auth_key_generator'
+require_relative 'mtproto/session'
+require_relative 'mtproto/encrypted_message'
 require_relative 'mtproto/client'
+require_relative 'mtproto/connection'
 
 module MTProto
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mtproto
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.6
 platform: ruby
 authors:
 - Artem Levenkov
@@ -17,15 +17,47 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- ".env.example"
 - ".ruby-version"
 - Rakefile
+- ext/aes_ige/Makefile
+- ext/aes_ige/aes_ige.c
+- ext/aes_ige/extconf.rb
+- ext/factorization/Makefile
+- ext/factorization/extconf.rb
+- ext/factorization/factorization.c
 - lib/mtproto.rb
+- lib/mtproto/auth_key_generator.rb
 - lib/mtproto/client.rb
+- lib/mtproto/connection.rb
+- lib/mtproto/crypto/aes_ige.rb
+- lib/mtproto/crypto/auth_key_helper.rb
+- lib/mtproto/crypto/dh_key_exchange.rb
+- lib/mtproto/crypto/dh_validator.rb
+- lib/mtproto/crypto/factorization.rb
+- lib/mtproto/crypto/message_key.rb
 - lib/mtproto/crypto/rsa_key.rb
+- lib/mtproto/crypto/rsa_pad.rb
+- lib/mtproto/encrypted_message.rb
+- lib/mtproto/errors.rb
+- lib/mtproto/session.rb
+- lib/mtproto/tl/bad_msg_notification.rb
+- lib/mtproto/tl/client_dh_inner_data.rb
+- lib/mtproto/tl/code_settings.rb
+- lib/mtproto/tl/config.rb
+- lib/mtproto/tl/gzip_packed.rb
 - lib/mtproto/tl/message.rb
+- lib/mtproto/tl/msg_container.rb
+- lib/mtproto/tl/new_session_created.rb
+- lib/mtproto/tl/p_q_inner_data.rb
+- lib/mtproto/tl/rpc_error.rb
+- lib/mtproto/tl/sent_code.rb
+- lib/mtproto/tl/serializer.rb
+- lib/mtproto/tl/server_dh_inner_data.rb
 - lib/mtproto/transport/abridged_packet_codec.rb
 - lib/mtproto/transport/tcp_connection.rb
 - lib/mtproto/version.rb
+- tmp/.keep
 homepage: https://github.com/alev-pro/mtproto-ruby
 licenses:
 - MIT