poke-go-api 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 7d4e06820ff24205fd6343be9abac2400d08b2e3
-  data.tar.gz: 1f3294582db49573f3805226e6515d6b42c94d37
+  metadata.gz: 44e400f988fa8c9fac5ce45524b7479e1abc6122
+  data.tar.gz: 747e7c6b71e2f2736d50eacdb8b9f535c7a9d8f0
 SHA512:
-  metadata.gz: 73b3a3e1c09ffdaca9bfbb14d5f08023dafc78ee4dc0b5339b673ad99e0d91ad401e0957fdd0701846747b25c98c01b78dd7d790d5634c0faf39c3eac6f0125c
-  data.tar.gz: d8b0b6f96f603393cd721172ca9308f4de3198605910c8d3b54b64ff892f6dfcf901811aa433d1f6790ff43667afe1537a3d45ce6ff59b90d322d808a3f47d12
+  metadata.gz: 9293d55c4802524510183d77a3e2666e9522a74b4d50a58e519048a4773b37dd404b6a6fbe7382949ddc488bae6278da755386df2152be9475b885c1f3382600
+  data.tar.gz: 9d290770cc439b156b6337b3a609b29e99718a6cbbe08e84ab4fae1fd7575611d2dd020b8f83a3dc78c92b339d195b849e0dd5cee127d3bc00b4b1fb9c8e34e0

data/lib/poke-api.rb

@@ -6,6 +6,7 @@ require 'poke-api/request_builder'
 require 'poke-api/response'
 require 'poke-api/errors'
 require 'poke-api/version'
+require 'poke-api/signature'
 require 'poke-api/auth/ptc'
 require 'poke-api/auth/google'
 require 'poke-api/auth/ticket'

data/lib/poke-api/client.rb

@@ -2,17 +2,19 @@ module Poke
   module API
     class Client
       include Logging
-      attr_accessor :lat, :lng, :alt, :endpoint, :ticket, :sig_path
+      attr_accessor :lat, :lng, :alt, :endpoint, :ticket, :sig_loaded
+      attr_reader :sig_path
 
       def initialize
-        @auth     = nil
-        @endpoint = 'https://pgorelease.nianticlabs.com/plfe/rpc'
-        @reqs     = []
-        @lat      = 0
-        @lng      = 0
-        @alt      = 8
-        @ticket   = Auth::Ticket.new
-        @sig_path = nil
+        @auth       = nil
+        @endpoint   = 'https://pgorelease.nianticlabs.com/plfe/rpc'
+        @reqs       = []
+        @lat        = 0
+        @lng        = 0
+        @alt        = 8
+        @ticket     = Auth::Ticket.new
+        @sig_path   = nil
+        @sig_loaded = false
       end
 
       def login(username, password, provider)
@@ -50,9 +52,9 @@ module Poke
       end
 
       def activate_signature(file_path)
-        if File.exist?(path)
+        if File.exist?(file_path)
           logger.info "[+] File #{file_path} has been set for signature generation"
-          @sig_path = path
+          @sig_path = file_path
         else
           raise Errors::InvalidSignatureFilePath, file_path
         end

data/lib/poke-api/request_builder.rb

@@ -1,10 +1,8 @@
-require 'ffi'
 require 'securerandom'
 
 module Poke
   module API
     class RequestBuilder
-      extend FFI::Library
       include Logging
 
       def initialize(auth, pos, endpoint)
@@ -14,7 +12,6 @@ module Poke
         @position     = pos
         @client       = HTTPClient.new(agent_name: 'PokeAPI/0.0.1')
         @start_time   = (Time.now.to_f * 1000).to_i
-        @ffi_enabled  = false
       end
 
       def request(reqs, client)
@@ -58,8 +55,8 @@ module Poke
             expire_timestamp_ms: client.ticket.expire
           )
 
-          load_library(client.sig_path) if client.sig_path && !@ffi_enabled
-          set_signature(req, request_envelope) if @ffi_enabled
+          Signature.load_signature(client) if client.sig_path
+          set_signature(req, request_envelope) if client.sig_loaded
         else
           logger.info '[+] Using Provider token'
           token = request_envelope::AuthInfo::JWT.new(contents: @access_token, unknown2: 59)
@@ -68,7 +65,7 @@ module Poke
       end
 
       def set_signature(req, _request_envelope)
-        signature = Signature.new(
+        signature = ::Signature.new(
           location_hash1: Helpers.generate_location_one(req.auth_ticket.to_proto, @position),
           location_hash2: Helpers.generate_location_two(@position),
           unk22: SecureRandom.random_bytes(32),
@@ -80,47 +77,18 @@ module Poke
           signature.request_hash << Helpers.generate_request(req.auth_ticket.to_proto, r.to_proto)
         end
 
-        unknown6 = POGOProtos::Networking::Envelopes::Unknown6
+        unk6 = POGOProtos::Networking::Envelopes::Unknown6
 
-        req.unknown6 = unknown6.new(
+        logger.info '[+] Generating Signature'
+
+        req.unknown6 = unk6.new(
           unknown1: 6,
-          unknown2: unknown6::Unknown2.new(unknown1: generate_signature(signature.to_proto))
+          unknown2: unk6::Unknown2.new(unknown1: Signature.generate_signature(signature.to_proto))
         )
 
         logger.info '[+] Setting Signature'
       end
 
-      def generate_signature(signature)
-        attempts = 5
-
-        begin
-          logger.info '[+] Generating Signature'
-          output_size = FFI::MemoryPointer.new(:size_t)
-
-          iv = SecureRandom.random_bytes(32)
-          RequestBuilder.encrypt(signature, signature.length, iv, 32, nil, output_size)
-
-          output = FFI::MemoryPointer.new(output_size.read_int)
-          RequestBuilder.encrypt(signature, signature.length, iv, 32, output, output_size)
-        rescue StandardError => ex
-          raise ex if (attempts -= 1) == 0
-          retry
-        end
-
-        output.read_string
-      end
-
-      def load_library(path)
-        RequestBuilder.ffi_lib(path)
-        RequestBuilder.attach_function(
-          :encrypt,
-          [:string, :size_t, :string, :size_t, :pointer, :pointer],
-          :int
-        )
-
-        @ffi_enabled = true
-      end
-
       def build_sub_request(req, sub_reqs)
         sub_reqs.each do |sub_req|
           if sub_req.is_a?(Symbol)

data/lib/poke-api/signature.rb

@@ -0,0 +1,29 @@
+module Poke
+  module API
+    module Signature
+      extend Fiddle::Importer
+      extend Logging
+
+      def self.load_signature(client)
+        dlload client.sig_path
+        extern "int encrypt(const unsigned char *input, size_t input_size," \
+               " const unsigned char* iv, size_t iv_size, unsigned char* " \
+               "output, size_t * output_size)"
+        logger.info '[+] Loaded Signature module'
+        client.sig_loaded = true
+      end
+
+      def self.generate_signature(signature)
+        output_size = Fiddle::Pointer.malloc(1)
+
+        iv = SecureRandom.random_bytes(32)
+        encrypt(signature, signature.length, iv, 32, nil, output_size)
+
+        output = Fiddle::Pointer.malloc(288)
+        encrypt(signature, signature.length, iv, 32, output, output_size)
+
+        output.to_str
+      end
+    end
+  end
+end

data/lib/poke-api/testing.rb

@@ -1,6 +1,29 @@
 # Load Poke-API related classes
 require 'pp'
-require 'poke-api'
+require 'pry'
+require_relative 'logging'
+require_relative 'client'
+require_relative 'helpers'
+require_relative 'request_builder'
+require_relative 'response'
+require_relative 'errors'
+require_relative 'version'
+require_relative 'signature'
+require_relative 'auth/ptc'
+require_relative 'auth/google'
+require_relative 'auth/ticket'
+
+# Load Geometry libraries (native Ruby)
+require_relative 'geometry/s2_base'
+require_relative 'geometry/s2_cell_id'
+require_relative 'geometry/s2_lat_lon'
+require_relative 'geometry/s2_point'
+
+# Load Google Generated POGOProtos
+require_relative 'POGOProtos/Signature'
+require_relative 'POGOProtos/Networking/Envelopes/RequestEnvelope'
+require_relative 'POGOProtos/Networking/Envelopes/ResponseEnvelope'
+
 
 def get_cells(lat, lng, radius = 10)
   s2_point = Poke::API::Geometry::S2LatLon.new(lat, lng).to_point

data/lib/poke-api/version.rb

@@ -1,5 +1,5 @@
 module Poke
   module API
-    VERSION = '0.1.0'.freeze
+    VERSION = '0.1.1'.freeze
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: poke-go-api
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Nabeel Amjad
@@ -80,20 +80,6 @@ dependencies:
     - - '='
       - !ruby/object:Gem::Version
         version: 0.4.0.1
-- !ruby/object:Gem::Dependency
-  name: ffi
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - '='
-      - !ruby/object:Gem::Version
-        version: 1.9.14
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - '='
-      - !ruby/object:Gem::Version
-        version: 1.9.14
 description: An API wrapper for Pokemon GO
 email: 
 executables: []
@@ -561,6 +547,7 @@ files:
 - lib/poke-api/logging.rb
 - lib/poke-api/request_builder.rb
 - lib/poke-api/response.rb
+- lib/poke-api/signature.rb
 - lib/poke-api/testing.rb
 - lib/poke-api/version.rb
 homepage: https://github.com/nabeelamjad/poke-api