Merge pull request #1 from elliottcarlson/master

Add Google Auth

Author: Yuu Kashima

README.md

@@ -6,20 +6,34 @@
 
 ## Demo
 
-    $ python2 main.py --debug -u tejado -p 1234 --location "New York, Washington Square"
-    [!] DEBUG mode on
+    $ python main.py -a "ptc" -u "username" -p "password" -l "New York, Washington Square"
     [!] Your given location: Washington Square, Greenwich, NY 12834, USA
     [!] lat/long/alt: 43.0909305 -73.4989367 0.0
-    [!] login for: tejado
-    [+] RPC Session Token: TGT-899360-gFKDueEjBcKX4G ...
-    [+] Received API endpoint: https://pgorelease.nianticlabs.com/plfe/401/rpc
+    [!] PTC login for: sublimnl
+    [+] RPC Session Token: TGT-842594-vsfLBEELnSrF ...
+    [+] Received API endpoint: https://pgorelease.nianticlabs.com/plfe/94/rpc
     [+] Login successful
-    [+] Username: tejado
-    [+] You are playing Pokemon Go since: 2016-07-13 08:10:20
+    [+] Username: username
+    [+] You are playing Pokemon Go since: 2016-07-14 00:05:32
     [+] Poke Storage: 250
     [+] Item Storage: 350
     [+] POKECOIN: 0
-    [+] STARDUST: 600
+    [+] STARDUST: 300
+
+    $ python main.py -a "google" -u "[email protected]" -p "password" -l "New York, Washington Square"
+    [!] Your given location: Washington Square, Greenwich, NY 12834, USA
+    [!] lat/long/alt: 43.0909305 -73.4989367 0.0
+    [!] Google login for: [email protected]
+    [+] RPC Session Token: eyJhbGciOiJSUzI1NiIsImt ...
+    [+] Received API endpoint: https://pgorelease.nianticlabs.com/plfe/490/rpc
+    [+] Login successful
+    [+] Username: <nickname>
+    [+] You are playing Pokemon Go since: 2016-07-12 20:59:39
+    [+] Poke Storage: 250
+    [+] Item Storage: 350
+    [+] POKECOIN: 0
+    [+] STARDUST: 100
+
 
 ## Credits
 Thanks a lot to [Mila432](https://github.com/Mila432/Pokemon_Go_API) !  

main.py

@@ -1,10 +1,12 @@
+#!/usr/bin/env python
 import requests
 import re
 import struct
 import json
 import argparse
 import pokemon_pb2
 
+from gpsoauth import perform_master_login, perform_oauth
 from datetime import datetime
 from geopy.geocoders import GoogleV3
 from requests.packages.urllib3.exceptions import InsecureRequestWarning
@@ -26,6 +28,11 @@
 COORDS_LONGITUDE = 0
 COORDS_ALTITUDE = 0
 
+ANDROID_ID = '9774d56d682e549c'
+SERVICE= 'audience:server:client_id:848232511240-7so421jotr2609rmqakceuu1luuq0ptb.apps.googleusercontent.com'
+APP = 'com.nianticlabs.pokemongo'
+CLIENT_SIG = '321187995bc7cdc2b5fc91b11a96e2baa8602c62'
+
 def f2i(float):
   return struct.unpack('<Q', struct.pack('<d', float))[0]
 
@@ -52,7 +59,7 @@ def set_location_coords(lat, long, alt):
 def get_location_coords():
     return (COORDS_LATITUDE, COORDS_LONGITUDE, COORDS_ALTITUDE)
 
-def api_req(api_endpoint, access_token, req):
+def api_req(service, api_endpoint, access_token, req):
     try:
         p_req = pokemon_pb2.RequestEnvelop()
         p_req.unknown1 = 2
@@ -63,7 +70,7 @@ def api_req(api_endpoint, access_token, req):
         p_req.latitude, p_req.longitude, p_req.altitude = get_location_coords()
 
         p_req.unknown12 = 989
-        p_req.auth.provider = 'ptc'
+        p_req.auth.provider = service
         p_req.auth.token.contents = access_token
         p_req.auth.token.unknown13 = 59
         protobuf = p_req.SerializeToString()
@@ -79,7 +86,7 @@ def api_req(api_endpoint, access_token, req):
         return None
 
 
-def get_api_endpoint(access_token):
+def get_api_endpoint(service, access_token):
     req = pokemon_pb2.RequestEnvelop()
 
     req1 = req.requests.add()
@@ -94,25 +101,33 @@ def get_api_endpoint(access_token):
     req5.type = 5
     req5.message.unknown4 = "4a2e9bc330dae60e7b74fc85b98868ab4700802e"
 
-    p_ret = api_req(API_URL, access_token, req.requests)
+    p_ret = api_req(service, API_URL, access_token, req.requests)
 
     try:
         return ('https://%s/rpc' % p_ret.api_url)
     except:
         return None
 
 
-def get_profile(api_endpoint, access_token):
+def get_profile(service, api_endpoint, access_token):
     req = pokemon_pb2.RequestEnvelop()
 
     req1 = req.requests.add()
     req1.type = 2
 
-    return api_req(api_endpoint, access_token, req.requests)
+    return api_req(service, api_endpoint, access_token, req.requests)
+
+
+def login_google(username, password):
+    print('[!] Google login for: {}'.format(username))
+    r1 = perform_master_login(username, password, ANDROID_ID)
+    r2 = perform_oauth(username, r1.get('Token', ''), ANDROID_ID, SERVICE, APP,
+        CLIENT_SIG)
 
+    return r2.get('Auth') # access token
 
 def login_ptc(username, password):
-    print('[!] login for: {}'.format(username))
+    print('[!] PTC login for: {}'.format(username))
     head = {'User-Agent': 'niantic'}
     r = SESSION.get(LOGIN_URL, headers=head)
     jdata = json.loads(r.content)
@@ -149,14 +164,20 @@ def login_ptc(username, password):
 
 def main():
     parser = argparse.ArgumentParser()
-    parser.add_argument("-u", "--username", help="PTC Username", required=True)
-    parser.add_argument("-p", "--password", help="PTC Password", required=True)
+    parser.add_argument("-a", "--auth_service", help="Auth Service",
+        required=True)
+    parser.add_argument("-u", "--username", help="Username", required=True)
+    parser.add_argument("-p", "--password", help="Password", required=True)
     parser.add_argument("-l", "--location", help="Location", required=True)
     parser.add_argument("-d", "--debug", help="Debug Mode", action='store_true')
     parser.add_argument("-s", "--client_secret", help="PTC Client Secret")
     parser.set_defaults(DEBUG=True)
     args = parser.parse_args()
 
+    if args.auth_service not in ['ptc', 'google']:
+      print('[!] Invalid Auth service specified')
+      return
+
     if args.debug:
         global DEBUG
         DEBUG = True
@@ -168,19 +189,23 @@ def main():
 
     set_location(args.location)
 
-    access_token = login_ptc(args.username, args.password)
+    if args.auth_service == 'ptc':
+        access_token = login_ptc(args.username, args.password)
+    else:
+        access_token = login_google(args.username, args.password)
+
     if access_token is None:
         print('[-] Wrong username/password')
         return
     print('[+] RPC Session Token: {} ...'.format(access_token[:25]))
 
-    api_endpoint = get_api_endpoint(access_token)
+    api_endpoint = get_api_endpoint(args.auth_service, access_token)
     if api_endpoint is None:
         print('[-] RPC server offline')
         return
     print('[+] Received API endpoint: {}'.format(api_endpoint))
 
-    profile = get_profile(api_endpoint, access_token)
+    profile = get_profile(args.auth_service, api_endpoint, access_token)
     if profile is not None:
         print('[+] Login successful')
 

requirements.txt

@@ -1,3 +1,15 @@
+cffi==1.7.0
+cryptography==1.4
+enum34==1.1.6
 geopy==1.11.0
+gpsoauth==0.3.0
+idna==2.1
+ipaddress==1.0.16
+ndg-httpsclient==0.4.1
 protobuf==2.6.1
+pyasn1==0.1.9
+pycparser==2.14
+pycryptodomex==3.4.2
+pyOpenSSL==16.0.0
 requests==2.10.0
+six==1.10.0