(CkPython) Verify Okta Access Token Locally

This example demonstrates how to validate an Okta access token using Chilkat's JWT class.

For more information, see https://developer.okta.com/docs/guides/validate-access-tokens/overview/#what-to-check-when-validating-an-access-token

Chilkat Python Downloads install with pip pip3 install chilkat or download... Python Module for Windows, MacOS, Linux, Alpine Linux, Solaris

import sys
import chilkat

# This example assumes the Chilkat API to have been previously unlocked.
# See Global Unlock Sample for sample code.

# This example begins with two JSON files:
# 
# 1. The access token obtained from Okta as shown in one fo these examples:  
#    Get Okta Token using Resource Owner Password Flow
# 
# 2. The Okta web keys obtained by this example:  Get Okta Web Keys
# 
# 

# Load the access token to be verified.
# It contains JSON that looks like this:
# {
#   "access_token": "eyJraWQiOiJhb ... O_eVu-kBp6g",
#   "token_type": "Bearer",
#   "expires_in": 3600,
#   "scope": "openid",
#   "id_token": "eyJraWQi ... FrL9WOuwbQtUg"
# }
# This example verifies the access_token.  (The id_token is verified in this example:  Verify Okta ID Token

jsonToken = chilkat.CkJsonObject()
success = jsonToken.LoadFile("qa_data/tokens/okta_access_token.json")

# Load the public keys (Okta web keys), one of which is needed to validate.
# The web keys JSON looks like this:
# {
#   "keys": [
#     {
#       "kty": "RSA",
#       "alg": "RS256",
#       "kid": "anSaRDPfWGOSCVNZEIZB9quCbNsdsvl5uWGBzxbudWQ",
#       "use": "sig",
#       "e": "AQAB",
#       "n": "jT8uAgd5w ... euLB1HaVw"
#     },
#     {
# 	...
#     }
#   ]
# }

jsonWebKeys = chilkat.CkJsonObject()
success = jsonWebKeys.LoadFile("qa_data/tokens/okta_web_keys.json")

# ------------------------
# Step 1: Get the JOSE header from the JWT.  The JOSE header contains JSON.  One of the JSON members will be the key ID "kid" which identifies the web key to be used for validation.
# 
jwt = chilkat.CkJwt()
accessToken = jsonToken.stringOf("access_token")
joseHeader = jwt.getHeader(accessToken)

print(joseHeader)
# The joseHeader contains this:   {"kid":"anSaRDPfWGOSCVNZEIZB9quCbNsdsvl5uWGBzxbudWQ","alg":"RS256"}

json = chilkat.CkJsonObject()
json.Load(joseHeader)
kid = json.stringOf("kid")
print("kid to find: " + kid)

# ------------------------
# Step 2: Find the key with the same "kid" in the Okta web keys.

sbKid = chilkat.CkStringBuilder()
e = ""
n = ""

i = 0
count_i = jsonWebKeys.SizeOfArray("keys")
bFound = False
iMatch = 0
while (bFound == False) and (i < count_i) :
    jsonWebKeys.put_I(i)
    sbKid.Clear()
    jsonWebKeys.StringOfSb("keys[i].kid",sbKid)
    print("checking kid: " + sbKid.getAsString())

    if (sbKid.ContentsEqual(kid,True) == True):
        e = jsonWebKeys.stringOf("keys[i].e")
        n = jsonWebKeys.stringOf("keys[i].n")
        # Exit the loop. 
        print("Found matching kid.")
        iMatch = i
        bFound = True

    i = i + 1

if (bFound == False):
    print("No matching key ID found.")
    sys.exit()

print("Matching key:")
print("  exponent = " + e)
print("  modulus = " + n)

# ------------------------
# Step 3: Load the RSA modulus and exponent into a Chilkat public key object.
pubkey = chilkat.CkPublicKey()

# Get the matching JSON key from the array of keys.
jsonWebKeys.put_I(iMatch)
# jsonWebKey is a CkJsonObject
jsonWebKey = jsonWebKeys.ObjectOf("keys[i]")
success = pubkey.LoadFromString(jsonWebKey.emit())
if (success == False):
    print("Failed to load JSON web key.")
    print(jsonWebKey.emit())
    print(pubkey.lastErrorText())

    sys.exit()

print("successfully loaded web key.")

# OK.. we have the desired JSON web key loaded into our public key object.
# Now we can verify the access token.

# ------------------------
# Step 4: Verify the access token.
bVerified = jwt.VerifyJwtPk(accessToken,pubkey)
if (bVerified == True):
    print("The access token is valid.")
else:
    print("The access token is NOT valid.")