mtproto 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fa11aaca7c34af33346fa59458bf300211785023ff833336738b0988deeafc32
-  data.tar.gz: 417cef055fd9d042956880f8e0214a10669759d7881e3af67e15e3bd2018de6c
+  metadata.gz: 414d6f7ed0a5c1c20c5108844469daafff2af7d2c350c8370b2eb6cec73068fe
+  data.tar.gz: d7c8e3397861bdc4f9d3c695ba884a672698f39c051c06271e5e39dcdf0aa7bd
 SHA512:
-  metadata.gz: 630dadd58d377ed2342abddf659385e8864fbfca3e6c4e1e9fa5a861b29a7a4fcb7628f9e3c7341c716588c5b37c18944d14e954cf0304ced52fa8482dcf859b
-  data.tar.gz: 796286ff4a5b9b9a42d7b0c14616040f3b8e0b310c2afece12943f49c14622d1f82f5b4479ef23b47fa6cabd3216b95aa16d0eb1ba4459cddab039042b94dd2a
+  metadata.gz: 2ccbe16eaf853697c6267ebe6b84e8593c071a66d95de9b1a1be1cafb0e33317400fb52fe85ef7aec43da2aa69bbc3f4cbc55af25928116711518925f4ed7897
+  data.tar.gz: 9658e4c8bd82fed68fd43f40e35efb1a57742170d27aa5f6a641528628e6da0013dfa9ddb632f42cb6e5418cd82c3006b8a4624294ebc40567c08d636aeae96b

data/lib/mtproto/client.rb

@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+require 'securerandom'
+require_relative 'transport/tcp_connection'
+require_relative 'transport/abridged_packet_codec'
+require_relative 'tl/message'
+
+module MTProto
+  class Client
+    DC_ADDRESSES = {
+      1 => ['149.154.175.50', 443],
+      2 => ['149.154.167.51', 443],
+      3 => ['149.154.175.100', 443],
+      4 => ['149.154.167.91', 443],
+      5 => ['91.108.56.130', 443]
+    }.freeze
+
+    attr_reader :connection
+
+    def initialize(dc_id: 2)
+      host, port = DC_ADDRESSES[dc_id]
+      raise ArgumentError, "Unknown DC ID: #{dc_id}" unless host
+
+      codec = Transport::AbridgedPacketCodec.new
+      @connection = Transport::TCPConnection.new(host, port, codec)
+    end
+
+    def connect
+      @connection.connect
+    end
+
+    def disconnect
+      @connection.close
+    end
+
+    def req_pq_multi
+      nonce = SecureRandom.random_bytes(16)
+
+      message = TL::Message.req_pq_multi(nonce)
+      payload = message.serialize
+
+      @connection.send(payload)
+
+      response_data = @connection.recv(timeout: 10)
+      response_message = TL::Message.deserialize(response_data)
+
+      res_pq = response_message.parse_res_pq
+
+      raise 'Nonce mismatch!' unless res_pq[:nonce] == nonce
+
+      rsa_key = Crypto::RSAKey.find_by_fingerprint(res_pq[:fingerprints])
+      raise 'No matching RSA key found!' unless rsa_key
+
+      res_pq[:rsa_key] = rsa_key
+      res_pq
+    end
+  end
+end

data/lib/mtproto/crypto/rsa_key.rb

@@ -0,0 +1,66 @@
+# frozen_string_literal: true
+
+require 'openssl'
+require 'digest'
+
+module MTProto
+  module Crypto
+    class RSAKey
+      TELEGRAM_KEY = <<~PEM
+        -----BEGIN RSA PUBLIC KEY-----
+        MIIBCgKCAQEA6LszBcC1LGzyr992NzE0ieY+BSaOW622Aa9Bd4ZHLl+TuFQ4lo4g
+        5nKaMBwK/BIb9xUfg0Q29/2mgIR6Zr9krM7HjuIcCzFvDtr+L0GQjae9H0pRB2OO
+        62cECs5HKhT5DZ98K33vmWiLowc621dQuwKWSQKjWf50XYFw42h21P2KXUGyp2y/
+        +aEyZ+uVgLLQbRA1dEjSDZ2iGRy12Mk5gpYc397aYp438fsJoHIgJ2lgMv5h7WY9
+        t6N/byY9Nw9p21Og3AoXSL2q/2IJ1WRUhebgAdGVMlV1fkuOQoEzR7EdpqtQD9Cs
+        5+bfo3Nhmcyvk5ftB0WkJ9z6bNZ7yxrP8wIDAQAB
+        -----END RSA PUBLIC KEY-----
+      PEM
+
+      attr_reader :key, :fingerprint
+
+      def initialize(pem_string)
+        @key = OpenSSL::PKey::RSA.new(pem_string)
+        @fingerprint = calculate_fingerprint
+      end
+
+      def self.telegram_key
+        @telegram_key ||= new(TELEGRAM_KEY)
+      end
+
+      def self.find_by_fingerprint(fingerprints)
+        telegram_key if fingerprints.include?(telegram_key.fingerprint)
+      end
+
+      private
+
+      def calculate_fingerprint
+        n_bytes = @key.n.to_s(2).unpack1('B*').scan(/.{8}/).map { |b| b.to_i(2) }.pack('C*')
+        e_bytes = @key.e.to_s(2).unpack1('B*').scan(/.{8}/).map { |b| b.to_i(2) }.pack('C*')
+
+        n_tl = serialize_bytes(n_bytes)
+        e_tl = serialize_bytes(e_bytes)
+
+        data = n_tl + e_tl
+        sha1 = Digest::SHA1.digest(data)
+
+        sha1[-8..].unpack1('Q<')
+      end
+
+      def serialize_bytes(bytes)
+        length = bytes.bytesize
+
+        if length <= 253
+          [length].pack('C') + bytes + padding(length + 1)
+        else
+          [254].pack('C') + [length].pack('L<')[0, 3] + bytes + padding(length + 4)
+        end
+      end
+
+      def padding(current_length)
+        pad_length = (4 - (current_length % 4)) % 4
+        "\x00" * pad_length
+      end
+    end
+  end
+end

data/lib/mtproto/tl/message.rb

@@ -0,0 +1,106 @@
+# frozen_string_literal: true
+
+module MTProto
+  module TL
+    class Message
+      CONSTRUCTOR_REQ_PQ_MULTI = 0xbe7e8ef1
+      CONSTRUCTOR_RES_PQ = 0x05162463
+
+      attr_reader :auth_key_id, :msg_id, :body
+
+      def initialize(auth_key_id: 0, msg_id: nil, body: '')
+        @auth_key_id = auth_key_id
+        @msg_id = msg_id || generate_msg_id
+        @body = body
+      end
+
+      def serialize
+        auth_key_id_bytes = [@auth_key_id].pack('Q<')
+        msg_id_bytes = [@msg_id].pack('Q<')
+        body_length = [@body.bytesize].pack('L<')
+
+        auth_key_id_bytes + msg_id_bytes + body_length + @body
+      end
+
+      def self.req_pq_multi(nonce)
+        raise ArgumentError, 'Nonce must be 16 bytes' unless nonce.bytesize == 16
+
+        constructor = [CONSTRUCTOR_REQ_PQ_MULTI].pack('L<')
+        body = constructor + nonce
+
+        new(auth_key_id: 0, body: body)
+      end
+
+      def self.deserialize(data)
+        auth_key_id = data[0, 8].unpack1('Q<')
+        msg_id = data[8, 8].unpack1('Q<')
+        body_length = data[16, 4].unpack1('L<')
+        body = data[20, body_length]
+
+        new(auth_key_id: auth_key_id, msg_id: msg_id, body: body)
+      end
+
+      def parse_res_pq
+        constructor = @body[0, 4].unpack1('L<')
+        raise "Unexpected constructor: 0x#{constructor.to_s(16)}" unless constructor == CONSTRUCTOR_RES_PQ
+
+        offset = 4
+
+        nonce = @body[offset, 16]
+        offset += 16
+
+        server_nonce = @body[offset, 16]
+        offset += 16
+
+        pq_length_byte = @body[offset].unpack1('C')
+        offset += 1
+
+        pq_length = if pq_length_byte == 254
+          @body[offset, 3].unpack1('L<') & 0xffffff
+          offset += 3
+        else
+          pq_length_byte
+        end
+
+        pq = @body[offset, pq_length]
+        offset += pq_length
+        offset += padding_length(pq_length + 1)
+
+        vector_constructor = @body[offset, 4].unpack1('L<')
+        offset += 4
+        raise 'Expected vector constructor' unless vector_constructor == 0x1cb5c415
+
+        fingerprints_count = @body[offset, 4].unpack1('L<')
+        offset += 4
+
+        fingerprints = []
+        fingerprints_count.times do
+          fingerprints << @body[offset, 8].unpack1('Q<')
+          offset += 8
+        end
+
+        {
+          nonce: nonce,
+          server_nonce: server_nonce,
+          pq: pq,
+          fingerprints: fingerprints
+        }
+      end
+
+      private
+
+      def generate_msg_id
+        time_ns = (Time.now.to_f * 1_000_000_000).to_i
+        (time_ns / 4) * 4
+      end
+
+      def padding_length(length)
+        (4 - (length % 4)) % 4
+      end
+
+      def self.padding_length(length)
+        (4 - (length % 4)) % 4
+      end
+    end
+  end
+end

data/lib/mtproto/version.rb

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

data/lib/mtproto.rb

@@ -3,6 +3,9 @@
 require_relative 'mtproto/version'
 require_relative 'mtproto/transport/abridged_packet_codec'
 require_relative 'mtproto/transport/tcp_connection'
+require_relative 'mtproto/tl/message'
+require_relative 'mtproto/crypto/rsa_key'
+require_relative 'mtproto/client'
 
 module MTProto
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mtproto
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Artem Levenkov
@@ -20,6 +20,9 @@ files:
 - ".ruby-version"
 - Rakefile
 - lib/mtproto.rb
+- lib/mtproto/client.rb
+- lib/mtproto/crypto/rsa_key.rb
+- lib/mtproto/tl/message.rb
 - lib/mtproto/transport/abridged_packet_codec.rb
 - lib/mtproto/transport/tcp_connection.rb
 - lib/mtproto/version.rb