poke-go-api 0.0.8 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f8a4ee180cf34ffc94cebb3b8bb6528b17d19382
-  data.tar.gz: b4fd14e947d0427c1ef50784525d3086b6e8fa8e
+  metadata.gz: 7d4e06820ff24205fd6343be9abac2400d08b2e3
+  data.tar.gz: 1f3294582db49573f3805226e6515d6b42c94d37
 SHA512:
-  metadata.gz: a9a59b5cc9518cdf34390f1128771490bb382e67d57ae55c69235f4859043f2bb8074195375164a9a628e80b29b36e8ebc26dc1988bd3f148e4873c0a14959bd
-  data.tar.gz: ef2d410e34aafa7fb3d1b5f3f85b83e82b22bdf5823177f218b7c84d3f1e5b773f389e4be747c5779944768408a2575acfcbcfad873b5a75795fa09633f6071c
+  metadata.gz: 73b3a3e1c09ffdaca9bfbb14d5f08023dafc78ee4dc0b5339b673ad99e0d91ad401e0957fdd0701846747b25c98c01b78dd7d790d5634c0faf39c3eac6f0125c
+  data.tar.gz: d8b0b6f96f603393cd721172ca9308f4de3198605910c8d3b54b64ff892f6dfcf901811aa433d1f6790ff43667afe1537a3d45ce6ff59b90d322d808a3f47d12

data/lib/poke-api.rb

@@ -1,4 +1,4 @@
-# Load Poke-API related classes
+# Load main Poke::API related classes
 require 'poke-api/logging'
 require 'poke-api/client'
 require 'poke-api/helpers'
@@ -8,7 +8,15 @@ require 'poke-api/errors'
 require 'poke-api/version'
 require 'poke-api/auth/ptc'
 require 'poke-api/auth/google'
+require 'poke-api/auth/ticket'
+
+# Load Geometry libraries (native Ruby)
+require 'poke-api/geometry/s2_base'
+require 'poke-api/geometry/s2_cell_id'
+require 'poke-api/geometry/s2_lat_lon'
+require 'poke-api/geometry/s2_point'
 
 # Load Google Generated POGOProtos
+require 'poke-api/POGOProtos/Signature'
 require 'poke-api/POGOProtos/Networking/Envelopes/RequestEnvelope'
 require 'poke-api/POGOProtos/Networking/Envelopes/ResponseEnvelope'

data/lib/poke-api/POGOProtos/Signature.rb

@@ -0,0 +1,94 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: Signature.proto
+
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_message "Signature" do
+    optional :timestamp_since_start, :uint64, 2
+    repeated :location_fix, :message, 4, "Signature.LocationFix"
+    optional :gps_info, :message, 5, "Signature.AndroidGpsInfo"
+    optional :sensor_info, :message, 7, "Signature.SensorInfo"
+    optional :device_info, :message, 8, "Signature.DeviceInfo"
+    optional :activity_status, :message, 9, "Signature.ActivityStatus"
+    optional :location_hash1, :uint32, 10
+    optional :location_hash2, :uint32, 20
+    optional :unk22, :bytes, 22
+    optional :timestamp, :uint64, 23
+    repeated :request_hash, :uint64, 24
+  end
+  add_message "Signature.LocationFix" do
+    optional :provider, :string, 1
+    optional :timestamp_since_start, :uint64, 2
+    optional :latitude, :float, 13
+    optional :longitude, :float, 14
+    optional :horizontal_accuracy, :float, 20
+    optional :altitude, :float, 21
+    optional :vertical_accuracy, :float, 22
+    optional :provider_status, :uint64, 26
+    optional :floor, :uint32, 27
+    optional :location_type, :uint64, 28
+  end
+  add_message "Signature.AndroidGpsInfo" do
+    optional :time_to_fix, :uint64, 1
+    repeated :satellites_prn, :int32, 2
+    repeated :snr, :float, 3
+    repeated :azimuth, :float, 4
+    repeated :elevation, :float, 5
+    repeated :has_almanac, :bool, 6
+    repeated :has_ephemeris, :bool, 7
+    repeated :used_in_fix, :bool, 8
+  end
+  add_message "Signature.SensorInfo" do
+    optional :timestamp_snapshot, :uint64, 1
+    optional :magnetometer_x, :double, 3
+    optional :magnetometer_y, :double, 4
+    optional :magnetometer_z, :double, 5
+    optional :angle_normalized_x, :double, 6
+    optional :angle_normalized_y, :double, 7
+    optional :angle_normalized_z, :double, 8
+    optional :accel_raw_x, :double, 10
+    optional :accel_raw_y, :double, 11
+    optional :accel_raw_z, :double, 12
+    optional :gyroscope_raw_x, :double, 13
+    optional :gyroscope_raw_y, :double, 14
+    optional :gyroscope_raw_z, :double, 15
+    optional :accel_normalized_x, :double, 16
+    optional :accel_normalized_y, :double, 17
+    optional :accel_normalized_z, :double, 18
+    optional :accelerometer_axes, :uint64, 19
+  end
+  add_message "Signature.DeviceInfo" do
+    optional :device_id, :string, 1
+    optional :android_board_name, :string, 2
+    optional :android_bootloader, :string, 3
+    optional :device_brand, :string, 4
+    optional :device_model, :string, 5
+    optional :device_model_identifier, :string, 6
+    optional :device_model_boot, :string, 7
+    optional :hardware_manufacturer, :string, 8
+    optional :hardware_model, :string, 9
+    optional :firmware_brand, :string, 10
+    optional :firmware_tags, :string, 12
+    optional :firmware_type, :string, 13
+    optional :firmware_fingerprint, :string, 14
+  end
+  add_message "Signature.ActivityStatus" do
+    optional :start_time_ms, :uint64, 1
+    optional :unknown_status, :bool, 2
+    optional :walking, :bool, 3
+    optional :running, :bool, 4
+    optional :stationary, :bool, 5
+    optional :automotive, :bool, 6
+    optional :tilting, :bool, 7
+    optional :cycling, :bool, 8
+    optional :status, :bytes, 9
+  end
+end
+
+Signature = Google::Protobuf::DescriptorPool.generated_pool.lookup("Signature").msgclass
+Signature::LocationFix = Google::Protobuf::DescriptorPool.generated_pool.lookup("Signature.LocationFix").msgclass
+Signature::AndroidGpsInfo = Google::Protobuf::DescriptorPool.generated_pool.lookup("Signature.AndroidGpsInfo").msgclass
+Signature::SensorInfo = Google::Protobuf::DescriptorPool.generated_pool.lookup("Signature.SensorInfo").msgclass
+Signature::DeviceInfo = Google::Protobuf::DescriptorPool.generated_pool.lookup("Signature.DeviceInfo").msgclass
+Signature::ActivityStatus = Google::Protobuf::DescriptorPool.generated_pool.lookup("Signature.ActivityStatus").msgclass

data/lib/poke-api/POGOProtosSource/Signature.proto

@@ -0,0 +1,100 @@
+syntax = "proto3";
+
+message Signature {
+
+    message LocationFix {
+        string provider = 1; // "network", "gps", "fused", possibly others
+        uint64 timestamp_since_start = 2; // in ms
+        float latitude = 13;
+        float longitude = 14;
+
+        // ??? shows up in struct, dunno where these go
+        // float device_speed;
+        // float device_course;
+        float horizontal_accuracy = 20; // iOS only? (range seems to be -1 to +1)
+        float altitude = 21;
+        float vertical_accuracy = 22; // iOS only? (range seems to be ~10-12)
+        uint64 provider_status = 26; // Usually 3 (possibly GPS status: 1 = no fix, 2 = acquiring/inaccurate, 3 = fix acquired)
+                           // On iOS there are some LocationFixes with unk26=1 and everything else empty
+        uint32 floor = 27; // No idea what this is, seems to be optional
+        uint64 location_type = 28; // Always 1 (if there is data at all)
+    }
+
+    // don't really care about this since we're not using it
+    message AndroidGpsInfo {
+        uint64 time_to_fix = 1;
+        repeated int32 satellites_prn = 2;
+        repeated float snr = 3;
+        repeated float azimuth = 4;
+        repeated float elevation = 5; 
+        repeated bool has_almanac = 6;
+        repeated bool has_ephemeris = 7;
+        repeated bool used_in_fix = 8;
+    }
+
+    message SensorInfo {
+        uint64 timestamp_snapshot = 1; // in ms
+        double magnetometer_x = 3;
+        double magnetometer_y = 4;
+        double magnetometer_z = 5;
+        double angle_normalized_x = 6;
+        double angle_normalized_y = 7;
+        double angle_normalized_z = 8;
+        double accel_raw_x = 10;
+        double accel_raw_y = 11;
+        double accel_raw_z = 12;
+        double gyroscope_raw_x = 13;
+        double gyroscope_raw_y = 14;
+        double gyroscope_raw_z = 15;
+        double accel_normalized_x = 16;
+        double accel_normalized_y = 17;
+        double accel_normalized_z = 18;
+        uint64 accelerometer_axes = 19; // Always 3
+    }
+
+    message DeviceInfo {
+        string device_id = 1; // Hex string
+        string android_board_name = 2;
+        string android_bootloader = 3;
+        string device_brand = 4; // On Android: product.brand
+        string device_model = 5; // On Android: product.device
+        string device_model_identifier = 6; // Android only, build.display.id
+        string device_model_boot = 7;  // On Android: boot.hardware
+        string hardware_manufacturer = 8; // On Android: product.manufacturer
+        string hardware_model = 9;  // On Android: product.model
+        string firmware_brand = 10; // On Android: product.name, on iOS: "iPhone OS"
+        string firmware_tags = 12; // Android only, build.tags
+        string firmware_type = 13; // On Android: build.type, on iOS instead: iOS version
+        string firmware_fingerprint = 14; // Android only, build.fingerprint
+    }
+
+    message ActivityStatus {
+        // all of these had 1 as their value
+        uint64 start_time_ms = 1;
+        bool unknown_status = 2;
+        bool walking = 3;
+        bool running = 4;
+        bool stationary = 5;
+        bool automotive = 6;
+        bool tilting = 7;
+        bool cycling = 8;
+        bytes status = 9;
+    }
+
+    uint64 timestamp_since_start = 2; // in ms
+    repeated LocationFix location_fix = 4;
+    AndroidGpsInfo gps_info = 5;
+    SensorInfo sensor_info = 7;
+    DeviceInfo device_info = 8;
+    ActivityStatus activity_status = 9;
+    uint32 location_hash1 = 10; // Location1 hashed based on the auth_token - xxHash32
+    uint32 location_hash2 = 20; // Location2 hashed based on the auth_token - xxHash32
+    bytes unk22 = 22; // possibly replay check. Generation unknown but pointed to by 0001B8614
+    uint64 timestamp = 23; // epoch timestamp in ms
+    repeated uint64 request_hash = 24; // hashes of each request message in a hashArray - xxhash64
+
+    // Addresses for the corresponding hash functions:
+    //    xxHash32              00054D28
+    //    xxhash64              000546C8 - Feeds into 00053D40
+
+}

data/lib/poke-api/auth/google.rb

@@ -7,11 +7,11 @@ module Poke
         include Logging
         attr_reader :access_token, :provider
 
-        GOOGLE_LOGIN_ANDROID_ID = '9774d56d682e549c'
+        GOOGLE_LOGIN_ANDROID_ID = '9774d56d682e549c'.freeze
         GOOGLE_LOGIN_SERVICE    = 'audience:server:client_id:848232511240-7so421jotr' \
-                                  '2609rmqakceuu1luuq0ptb.apps.googleusercontent.com'
-        GOOGLE_LOGIN_APP        = 'com.nianticlabs.pokemongo'
-        GOOGLE_LOGIN_CLIENT_SIG = '321187995bc7cdc2b5fc91b11a96e2baa8602c62'
+                                  '2609rmqakceuu1luuq0ptb.apps.googleusercontent.com'.freeze
+        GOOGLE_LOGIN_APP        = 'com.nianticlabs.pokemongo'.freeze
+        GOOGLE_LOGIN_CLIENT_SIG = '321187995bc7cdc2b5fc91b11a96e2baa8602c62'.freeze
 
         def initialize(username, password)
           @username = username
@@ -40,7 +40,7 @@ module Poke
           response = yield
 
           if response['Error'] == 'NeedsBrowser'
-            raise Errors::GoogleTwoFactorAuthenticationFailure.new(response)
+            raise Errors::GoogleTwoFactorAuthenticationFailure
           else
             unless response['Token'] || response['Auth']
               raise Errors::GoogleAuthenticationFailure.new(method, response)

data/lib/poke-api/auth/ptc.rb

@@ -9,9 +9,9 @@ module Poke
         attr_reader :access_token, :provider
 
         PTC_LOGIN_URL = 'https://sso.pokemon.com/sso/login?service=https%3A%2F%2Fsso.pokemon.com' \
-                        '%2Fsso%2Foauth2.0%2FcallbackAuthorize'
-        PTC_LOGIN_OAUTH = 'https://sso.pokemon.com/sso/oauth2.0/accessToken'
-        PTC_LOGIN_CLIENT_SECRET = 'w8ScCUXJQc6kXKw8FiOhd8Fixzht18Dq3PEVkUCP5ZPxtgyWsbTvWHFLm2wNY0JR'
+                        '%2Fsso%2Foauth2.0%2FcallbackAuthorize'.freeze
+        PTC_LOGIN_OAUTH = 'https://sso.pokemon.com/sso/oauth2.0/accessToken'.freeze
+        PTC_LOGIN_CLIENT_SECRET = 'w8ScCUXJQc6kXKw8FiOhd8Fixzht18Dq3PEVkUCP5ZPxtgyWsbTvWHFLm2wNY0JR'.freeze
 
         def initialize(username, password)
           @username = username

data/lib/poke-api/auth/ticket.rb

@@ -0,0 +1,51 @@
+module Poke
+  module API
+    module Auth
+      class Ticket
+        attr_reader :start, :ends, :expire
+
+        def initialize
+          @expire = nil
+          @start  = nil
+          @ends   = nil
+        end
+
+        def has_ticket?
+          return true if @start && @ends && @expire
+          false
+        end
+
+        def set_ticket(auth)
+          @expire = auth[:expire_timestamp_ms]
+          @start  = auth[:start]
+          @ends   = auth[:end]
+        end
+
+        def is_new_ticket?(new_ticket_time)
+          return true unless @expire && new_ticket_time > @expire
+          false
+        end
+
+        def get_ticket
+          return false unless check_ticket
+
+          true
+        end
+
+        private
+
+        def check_ticket
+          return false unless has_ticket?
+
+          now_ms = (Time.now.to_f * 1000).to_i
+          return true if now_ms < (@expire - 10000)
+
+          @expire = nil
+          @start  = nil
+          @ends   = nil
+          false
+        end
+      end
+    end
+  end
+end

data/lib/poke-api/client.rb

@@ -2,7 +2,7 @@ module Poke
   module API
     class Client
       include Logging
-      attr_accessor :lat, :lng, :alt, :endpoint
+      attr_accessor :lat, :lng, :alt, :endpoint, :ticket, :sig_path
 
       def initialize
         @auth     = nil
@@ -10,7 +10,9 @@ module Poke
         @reqs     = []
         @lat      = 0
         @lng      = 0
-        @alt      = 0
+        @alt      = 8
+        @ticket   = Auth::Ticket.new
+        @sig_path = nil
       end
 
       def login(username, password, provider)
@@ -25,12 +27,14 @@ module Poke
           raise Errors::LoginFailure.new(provider, ex)
         end
 
+        get_hatched_eggs
+        call
         logger.info "[+] Login with #{provider} Successful"
-        fetch_endpoint
       end
 
       def call
         raise Errors::LoginRequired unless @auth
+        raise Errors::NoRequests if @reqs.empty?
         req = RequestBuilder.new(@auth, [@lat, @lng, @alt], @endpoint)
 
         begin
@@ -45,6 +49,15 @@ module Poke
         resp
       end
 
+      def activate_signature(file_path)
+        if File.exist?(path)
+          logger.info "[+] File #{file_path} has been set for signature generation"
+          @sig_path = path
+        else
+          raise Errors::InvalidSignatureFilePath, file_path
+        end
+      end
+
       def store_location(loc)
         pos = Poke::API::Helpers.get_position(loc).first
         logger.info "[+] Given location: #{pos.address}"
@@ -56,8 +69,7 @@ module Poke
 
       def store_lat_lng(lat, lng)
         logger.info "[+] Lat/Long: #{lat}, #{lng}"
-        @lat = lat
-        @lng = lng
+        @lat, @lng = lat, lng
       end
 
       def inspect
@@ -67,16 +79,6 @@ module Poke
 
       private
 
-      def fetch_endpoint
-        get_player
-        get_hatched_eggs
-        get_inventory
-        check_awarded_badges
-        download_settings(hash: '4a2e9bc330dae60e7b74fc85b98868ab4700802e')
-
-        call
-      end
-
       def method_missing(method, *args)
         POGOProtos::Networking::Requests::RequestType.const_get(method.upcase)
         @reqs << (args.empty? ? method.to_sym.upcase : { method.to_sym.upcase => args.first })

data/lib/poke-api/errors.rb

@@ -48,6 +48,24 @@ module Poke
           super("Unable to fetch endpoint, please try to login again.")
         end
       end
+
+      class NoRequests < StandardError
+        def initialize
+          super("Can not call without any requests set")
+        end
+      end
+
+      class InvalidSignatureFilePath < StandardError
+        def initialize(file_path)
+          super("Could not find file #{file_path} for signature generation")
+        end
+      end
+
+      class InvalidLevel < StandardError
+        def initialize(level)
+          super("Level #{level} is invalid, must be between 0 and 30.")
+        end
+      end
     end
   end
 end

data/lib/poke-api/geometry/s2_base.rb

@@ -0,0 +1,50 @@
+module Poke
+  module API
+    module Geometry
+      module S2Base
+        LINEAR_PROJECTION    = 0
+        TAN_PROJECTION       = 1
+        QUADRATIC_PROJECTION = 2
+
+        MAX_LEVEL   = 30
+        NUM_FACES   = 6
+        POS_BITS    = 2 * MAX_LEVEL + 1
+        MAX_SIZE    = 1 << MAX_LEVEL
+        SWAP_MASK   = 0x01
+        INVERT_MASK = 0x02
+        LOOKUP_BITS = 4
+        POS_TO_OR   = [SWAP_MASK, 0, 0, INVERT_MASK | SWAP_MASK].freeze
+        POS_TO_IJ   = [[0, 1, 3, 2],
+                       [0, 2, 3, 1],
+                       [3, 2, 0, 1],
+                       [3, 1, 0, 2]].freeze
+
+        LOOKUP_POS = [nil] * (1 << (2 * LOOKUP_BITS + 2))
+        LOOKUP_IJ = [nil] * (1 << (2 * LOOKUP_BITS + 2))
+
+        def self.lookup_cells(level, i, j, orig_orientation, pos, orientation)
+          return lookup_bits(i, j, orig_orientation, pos, orientation) if level == LOOKUP_BITS
+
+          r = POS_TO_IJ[orientation]
+          4.times do |index|
+            lookup_cells(
+              level + 1, (i << 1) + (r[index] >> 1), (j << 1) + (r[index] & 1),
+              orig_orientation, (pos << 2) + index, orientation ^ POS_TO_OR[index]
+            )
+          end
+        end
+
+        def self.lookup_bits(i, j, orig_orientation, pos, orientation)
+          ij = (i << LOOKUP_BITS) + j
+          LOOKUP_POS[(ij << 2) + orig_orientation] = (pos << 2) + orientation
+          LOOKUP_IJ[(pos << 2) + orig_orientation] = (ij << 2) + orientation
+        end
+
+        lookup_cells(0, 0, 0, 0, 0, 0)
+        lookup_cells(0, 0, 0, SWAP_MASK, 0, SWAP_MASK)
+        lookup_cells(0, 0, 0, INVERT_MASK, 0, INVERT_MASK)
+        lookup_cells(0, 0, 0, SWAP_MASK | INVERT_MASK, 0, SWAP_MASK | INVERT_MASK)
+      end
+    end
+  end
+end

data/lib/poke-api/geometry/s2_cell_id.rb

@@ -0,0 +1,146 @@
+module Poke
+  module API
+    module Geometry
+      class S2CellId
+        include S2Base
+        attr_reader :id
+
+        def initialize(id)
+          @id = id
+        end
+
+        def parent(level)
+          new_lsb = lsb_for_level(level)
+          s2 = S2CellId.new((@id & -new_lsb) | new_lsb)
+
+          raise Errors::InvalidLevel, level unless valid?(s2.id)
+          s2
+        end
+
+        def prev
+          S2CellId.new(@id - (lsb << 1))
+        end
+
+        def next
+          S2CellId.new(@id + (lsb << 1))
+        end
+
+        def level
+          return MAX_LEVEL if leaf?
+
+          x = (@id & 0xffffffff)
+          level = -1
+
+          if x != 0
+            level += 16
+          else
+            x = ((@id >> 32) & 0xffffffff)
+          end
+
+          x &= -x
+
+          level += 8 unless (x & 0x00005555).zero?
+          level += 4 unless (x & 0x00550055).zero?
+          level += 2 unless (x & 0x05050505).zero?
+          level += 1 unless (x & 0x11111111).zero?
+          level
+        end
+
+        def self.from_point(p)
+          face, u, v = xyz_to_face_uv(p)
+          i = st_to_ij(uv_to_st(u))
+          j = st_to_ij(uv_to_st(v))
+
+          S2CellId.new(from_face_ij(face, i, j))
+        end
+
+        def self.from_face_ij(face, i, j)
+          n = face << (POS_BITS - 1)
+          bits = face & SWAP_MASK
+
+          7.downto(0).each do |k|
+            mask = (1 << LOOKUP_BITS) - 1
+            bits += (((i >> (k * LOOKUP_BITS)) & mask) << (LOOKUP_BITS + 2))
+            bits += (((j >> (k * LOOKUP_BITS)) & mask) << 2)
+            bits = LOOKUP_POS[bits]
+            n |= (bits >> 2) << (k * 2 * LOOKUP_BITS)
+            bits &= (SWAP_MASK | INVERT_MASK)
+          end
+
+          @id = (n * 2 + 1)
+        end
+
+        def self.xyz_to_face_uv(p)
+          face = p.largest_abs_component
+
+          pface = case face
+                  when 0 then p.x
+                  when 1 then p.y
+                  else p.z
+                  end
+
+          face += 3 if pface < 0
+
+          u, v = valid_face_xyz_to_uv(face, p)
+          [face, u, v]
+        end
+
+        def self.uv_to_st(u)
+          return 0.5 * Math.sqrt(1 + 3 * u) if u >= 0
+          1 - 0.5 * Math.sqrt(1 - 3 * u)
+        end
+
+        def self.st_to_ij(s)
+          [0, [MAX_SIZE - 1, Integer((MAX_SIZE * s).floor)].min].max
+        end
+
+        def self.valid_face_xyz_to_uv(face, p)
+          raise unless p.dot_prod(face_uv_to_xyz(face, 0, 0)) > 0
+
+          case face
+          when 0 then [p.y / p.x, p.z / p.x]
+          when 1 then [-p.x / p.y, p.z / p.y]
+          when 2 then [-p.x / p.z, -p.y / p.z]
+          when 3 then [p.z / p.x, p.y / p.x]
+          when 4 then [p.z / p.y, -p.x / p.y]
+          else [-p.y / p.z, -p.x / p.z]
+          end
+        end
+
+        def self.face_uv_to_xyz(face, u, v)
+          case face
+          when 0 then S2Point.new(1, u, v)
+          when 1 then S2Point.new(-u, 1, v)
+          when 2 then S2Point.new(-u, -v, 1)
+          when 3 then S2Point.new(-1, -v, -u)
+          when 4 then S2Point.new(v, -1, -u)
+          else S2Point.new(v, u, -1)
+          end
+        end
+
+        private_class_method :face_uv_to_xyz, :valid_face_xyz_to_uv, :uv_to_st,
+                             :st_to_ij, :xyz_to_face_uv, :from_face_ij
+
+        private
+
+        def leaf?
+          @id & 1 != 0
+        end
+
+        def valid?(s2_id)
+          face = s2_id >> POS_BITS
+          lsb = s2_id & -s2_id
+          (face < NUM_FACES) && (lsb & 0x1555555555555555) != 0
+        end
+
+        def lsb
+          @id & -@id
+        end
+
+        def lsb_for_level(level)
+          1 << (2 * (MAX_LEVEL - level))
+        end
+      end
+    end
+  end
+end

data/lib/poke-api/geometry/s2_lat_lon.rb

@@ -0,0 +1,19 @@
+module Poke
+  module API
+    module Geometry
+      class S2LatLon
+        def initialize(lat_degrees, lon_degrees)
+          @lat = lat_degrees * Math::PI / 180
+          @lon = lon_degrees * Math::PI / 180
+        end
+
+        def to_point
+          phi    = @lat
+          theta  = @lon
+          cosphi = Math.cos(phi)
+          S2Point.new(Math.cos(theta) * cosphi, Math.sin(theta) * cosphi, Math.sin(phi))
+        end
+      end
+    end
+  end
+end

data/lib/poke-api/geometry/s2_point.rb

@@ -0,0 +1,33 @@
+module Poke
+  module API
+    module Geometry
+      class S2Point
+        attr_reader :x, :y, :z
+
+        def initialize(x, y, z)
+          @x = x
+          @y = y
+          @z = z
+        end
+
+        def abs
+          [@x.abs, @y.abs, @z.abs]
+        end
+
+        def largest_abs_component
+          temp = abs
+
+          if temp[0] > temp[1]
+            temp[0] > temp[2] ? 0 : 2
+          else
+            temp[1] > temp[2] ? 1 : 2
+          end
+        end
+
+        def dot_prod(o)
+          @x * o.x + @y * o.y + @z * o.z
+        end
+      end
+    end
+  end
+end

data/lib/poke-api/helpers.rb

@@ -1,4 +1,5 @@
 require 'geocoder'
+require 'ruby-xxhash'
 
 module Poke
   module API
@@ -11,6 +12,29 @@ module Poke
         def camel_case_lower(sym)
           sym.to_s.split('_').collect(&:capitalize).join
         end
+
+        def generate_location_one(auth_ticket, pos)
+          first_hash = XXhash.xxh32(auth_ticket, 0x1B845238)
+          location_bytes = d2h(pos[0]) + d2h(pos[1]) + d2h(pos[2])
+
+          XXhash.xxh32(location_bytes, first_hash)
+        end
+
+        def generate_location_two(pos)
+          location_bytes = d2h(pos[0]) + d2h(pos[1]) + d2h(pos[2])
+
+          XXhash.xxh32(location_bytes, seed=0x1B845238)
+        end 
+
+        def generate_request(auth_ticket, request)
+          first_hash = XXhash.xxh64(auth_ticket, 0x1B845238)
+
+          XXhash.xxh64(request, first_hash)
+        end
+
+        def d2h(input)
+          [input].pack('d').unpack('Q').first.to_s(16).split.pack('H*')
+        end
       end
     end
   end

data/lib/poke-api/request_builder.rb

@@ -1,6 +1,10 @@
+require 'ffi'
+require 'securerandom'
+
 module Poke
   module API
     class RequestBuilder
+      extend FFI::Library
       include Logging
 
       def initialize(auth, pos, endpoint)
@@ -9,11 +13,14 @@ module Poke
         @endpoint     = endpoint
         @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)
         logger.debug '[+] Creating new request'
-        request_proto = build_main_request(reqs)
+        request_proto = build_main_request(reqs, client)
+        logger.debug "[+] Generated RPC protobuf encoded request \r\n#{request_proto.inspect}"
 
         logger.info '[+] Executing RPC request'
         resp = execute_rpc_request(request_proto)
@@ -26,7 +33,7 @@ module Poke
 
       private
 
-      def build_main_request(sub_reqs)
+      def build_main_request(sub_reqs, client)
         request_envelope = POGOProtos::Networking::Envelopes::RequestEnvelope
         req = request_envelope.new(
           status_code: 2,
@@ -35,15 +42,85 @@ module Poke
         )
         req.latitude, req.longitude, req.altitude = @position
 
-        token = request_envelope::AuthInfo::JWT.new(contents: @access_token, unknown2: 59)
-        req.auth_info = request_envelope::AuthInfo.new(provider: @provider, token: token)
-
         build_sub_request(req, sub_reqs)
+        set_authentication(req, client, request_envelope)
 
         logger.debug "[+] Generated RPC protobuf request \r\n#{req.inspect}"
         req.to_proto
       end
 
+      def set_authentication(req, client, request_envelope)
+        if client.ticket.get_ticket
+          logger.info '[+] Using Auth ticket'
+
+          req.auth_ticket = POGOProtos::Networking::Envelopes::AuthTicket.new(
+            start: client.ticket.start, end: client.ticket.ends,
+            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
+        else
+          logger.info '[+] Using Provider token'
+          token = request_envelope::AuthInfo::JWT.new(contents: @access_token, unknown2: 59)
+          req.auth_info = request_envelope::AuthInfo.new(provider: @provider, token: token)
+        end
+      end
+
+      def set_signature(req, _request_envelope)
+        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),
+          timestamp: (Time.now.to_f * 1000).to_i,
+          timestamp_since_start: (Time.now.to_f * 1000).to_i - @start_time
+        )
+
+        req.requests.each do |r|
+          signature.request_hash << Helpers.generate_request(req.auth_ticket.to_proto, r.to_proto)
+        end
+
+        unknown6 = POGOProtos::Networking::Envelopes::Unknown6
+
+        req.unknown6 = unknown6.new(
+          unknown1: 6,
+          unknown2: unknown6::Unknown2.new(unknown1: 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/response.rb

@@ -16,6 +16,9 @@ module Poke
         @response = POGOProtos::Networking::Envelopes::ResponseEnvelope.decode(@response)
         logger.debug "[+] Decoded RPC response \r\n#{@response.inspect}"
 
+        store_ticket(client)
+        store_endpoint(client)
+
         logger.info '[+] Decoding Sub RPC responses'
         decoded_resp = parse_rpc_fields(decode_sub_responses)
 
@@ -24,7 +27,6 @@ module Poke
           parse_rpc_fields(decoded_resp)
         end
 
-        store_endpoint(client)
         decoded_resp.merge!(status_code: @response.status_code,
                             api_url: @response.api_url, error: @response.error)
 
@@ -34,6 +36,17 @@ module Poke
 
       private
 
+      def store_ticket(client)
+        return unless @response.auth_ticket
+
+        auth = @response.auth_ticket.to_hash
+
+        if client.ticket.is_new_ticket?(auth[:expire_timestamp_ms])
+          logger.debug "[+] Storing Auth Ticket\r\n#{auth}"
+          client.ticket.set_ticket(auth)
+        end
+      end
+
       def store_endpoint(client)
         logger.debug "[+] Current Endpoint #{client.endpoint}"
 
@@ -43,7 +56,7 @@ module Poke
 
         return if @response.api_url.empty?
 
-        logger.debug "[+] Setting Endpoint to #{@response.api_url}"
+        logger.debug "[+] Setting Endpoint to https://#{@response.api_url}/rpc"
         client.endpoint = "https://#{@response.api_url}/rpc"
       end
 

data/lib/poke-api/testing.rb

@@ -1,24 +1,37 @@
 # Load Poke-API related classes
 require 'pp'
-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 'auth/ptc'
-require_relative 'auth/google'
+require 'poke-api'
 
-# Load Google Generated POGOProtos
-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
+  s2_cell  = Poke::API::Geometry::S2CellId.from_point(s2_point).parent(15)
 
+  next_cell = s2_cell.next
+  prev_cell = s2_cell.prev
+
+  radius.times.reduce([s2_cell.id]) do |acc, el|
+    acc += [next_cell.id, prev_cell.id]
+    next_cell = next_cell.next
+    prev_cell = prev_cell.prev
+    acc
+  end.sort
+end
+
+#Poke::API::Logging.log_level = :DEBUG
 client = Poke::API::Client.new
-Poke::API::Logging.log_level = :DEBUG
-#client.store_lat_lng(55.864237, -4.251806)
+client.activate_signature('encrypt64bit.dll')
+
+client.store_location('Glasgow')
 client.login('srn720', 'illi0wnz', 'ptc')
-#client.get_player
-#resp = client.call
-#pp resp
+
+#Get cells
+cell_ids = get_cells(client.lat, client.lng)
+
+# Construct map objects call
+client.get_map_objects(
+  latitude: client.lat,
+  longitude: client.lng,
+  since_timestamp_ms: [0] * cell_ids.length,
+  cell_id: cell_ids
+)
+p client.call

data/lib/poke-api/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: poke-go-api
 version: !ruby/object:Gem::Version
-  version: 0.0.8
+  version: 0.1.0
 platform: ruby
 authors:
 - Nabeel Amjad
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-08-03 00:00:00.000000000 Z
+date: 2016-08-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: httpclient
@@ -66,6 +66,34 @@ dependencies:
     - - '='
       - !ruby/object:Gem::Version
         version: 0.1.2
+- !ruby/object:Gem::Dependency
+  name: ruby-xxHash
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.4.0.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !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: []
@@ -294,6 +322,7 @@ files:
 - lib/poke-api/POGOProtos/Settings/Master/PokemonSettings.rb
 - lib/poke-api/POGOProtos/Settings/Master/PokemonUpgradeSettings.rb
 - lib/poke-api/POGOProtos/Settings/Master/TypeEffectiveSettings.rb
+- lib/poke-api/POGOProtos/Signature.rb
 - lib/poke-api/POGOProtosSource/Data/AssetDigestEntry.proto
 - lib/poke-api/POGOProtosSource/Data/Battle/BattleAction.proto
 - lib/poke-api/POGOProtosSource/Data/Battle/BattleActionType.proto
@@ -518,10 +547,16 @@ files:
 - lib/poke-api/POGOProtosSource/Settings/Master/PokemonSettings.proto
 - lib/poke-api/POGOProtosSource/Settings/Master/PokemonUpgradeSettings.proto
 - lib/poke-api/POGOProtosSource/Settings/Master/TypeEffectiveSettings.proto
+- lib/poke-api/POGOProtosSource/Signature.proto
 - lib/poke-api/auth/google.rb
 - lib/poke-api/auth/ptc.rb
+- lib/poke-api/auth/ticket.rb
 - lib/poke-api/client.rb
 - lib/poke-api/errors.rb
+- lib/poke-api/geometry/s2_base.rb
+- lib/poke-api/geometry/s2_cell_id.rb
+- lib/poke-api/geometry/s2_lat_lon.rb
+- lib/poke-api/geometry/s2_point.rb
 - lib/poke-api/helpers.rb
 - lib/poke-api/logging.rb
 - lib/poke-api/request_builder.rb