CNSA-276-FP/credentailsex.py

# Copyright (c) 2018 Yubico AB
# All rights reserved.
#
#   Redistribution and use in source and binary forms, with or
#   without modification, are permitted provided that the following
#   conditions are met:
#
#    1. Redistributions of source code must retain the above copyright
#       notice, this list of conditions and the following disclaimer.
#    2. Redistributions in binary form must reproduce the above
#       copyright notice, this list of conditions and the following
#       disclaimer in the documentation and/or other materials provided
#       with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
# FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
# COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
# BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.

"""
Connects to the first FIDO device found (starts from USB, then looks into NFC),
creates a new credential for it, and authenticates the credential.
This works with both FIDO 2.0 devices as well as with U2F devices.
On Windows, the native WebAuthn API will be used.
"""
import time

import fido2.webauthn
from fido2.hid import CtapHidDevice
from fido2.client import Fido2Client, WindowsClient, UserInteraction
from fido2.server import Fido2Server
from getpass import getpass
import sys
import ctypes
from fido2.ctap2 import Ctap2

REGISTER = False
gotAaguid = None

while True:
    while True:
        dev = next(CtapHidDevice.list_devices(), None)

        if dev:
            ctap2 = Ctap2(dev)

            info = ctap2.get_info()
            gotAaguid = info.aaguid
            print(str(gotAaguid))
            break

        print("Waiting for device...")
        time.sleep(0.1)

    # Handle user interaction
    class CliInteraction(UserInteraction):
        def prompt_up(self):
            print("\nTouch your authenticator device now...\n")

        def request_pin(self, permissions, rd_id):
            return getpass("Enter PIN: ")

        def request_uv(self, permissions, rd_id):
            print("User Verification required.")
            return True
    if REGISTER:

        uv = "discouraged"

        if WindowsClient.is_available() and not ctypes.windll.shell32.IsUserAnAdmin():
            # Use the Windows WebAuthn API if available, and we're not running as admin
            client = WindowsClient("https://example.com")
        else:
            # Locate a device
            dev = next(CtapHidDevice.list_devices(), None)
            if dev is not None:
                print("Use USB HID channel.")
            else:
                try:
                    from fido2.pcsc import CtapPcscDevice

                    dev = next(CtapPcscDevice.list_devices(), None)
                    print("Use NFC channel.")
                except Exception as e:
                    print("NFC channel search error:", e)

            if not dev:
                print("No FIDO device found")
                sys.exit(1)

            # Set up a FIDO 2 client using the origin https://example.com
            client = Fido2Client(dev, "https://example.com", user_interaction=CliInteraction())

            # Prefer UV if supported and configured
            if client.info.options.get("uv") or client.info.options.get("pinUvAuthToken"):
                uv = "preferred"
                print("Authenticator supports User Verification")


        server = Fido2Server({"id": "example.com", "name": "Example RP"}, attestation="direct")

        user = {"id": b"user_id", "name": "A. User"}


        # Prepare parameters for makeCredential
        create_options, state = server.register_begin(
            user, user_verification=uv, authenticator_attachment="cross-platform"
        )

        # Create a credential
        result = client.make_credential(create_options["publicKey"])

        # Complete registration
        auth_data = server.register_complete(
            state, result.client_data, result.attestation_object
        )
        credentials = [auth_data.credential_data]

        AAGUID = auth_data.credential_data.aaguid
        cred_id = auth_data.credential_data.credential_id
        pk_algo = auth_data.credential_data.public_key.ALGORITHM
        pk_1 = auth_data.credential_data.public_key.get(1)
        pk_3 = auth_data.credential_data.public_key.get(3)
        pk__1 = auth_data.credential_data.public_key.get(-1)
        pk__2 = auth_data.credential_data.public_key.get(-2)
        pk__3 = auth_data.credential_data.public_key.get(-3)


        import psycopg2

        # Replace these variables with your connection parameters
        dbname = "CNSA-276-FP-DAS"
        user = "FP-DEV-USER"
        password = "purchase-immortal-prescribe-repave-detention-seizing-candied-antiques-episode-list"
        host = "postgres.eggtech.net"


        # Function to connect to the PostgreSQL database and insert data
        def insert_data(aaguid, cred_id, pk_algo, pk_1, pk_3, pk__1, pk__2, pk__3):
            conn = None
            try:
                # Connect to the PostgreSQL server
                conn = psycopg2.connect(dbname=dbname, user=user, password=password, host=host)
                cur = conn.cursor()
                # Define the SQL query for inserting data
                insert_query = '''
                INSERT INTO credential_data ("AAGUID", "credential_id", "pk_algo", "pk_1", "pk_3", "pk_neg1", "pk_neg2", "pk_neg3")
                VALUES (%s, %s, %s, %s, %s, %s, %s, %s);
                '''

                # Execute the SQL query
                cur.execute(insert_query, (aaguid, cred_id, pk_algo, pk_1, pk_3, pk__1, pk__2, pk__3))

                # Commit the changes
                conn.commit()

                # Close communication with the database
                cur.close()
            except (Exception, psycopg2.DatabaseError) as error:
                print(error)
            finally:
                if conn is not None:
                    conn.close()

        # Assuming auth_data is defined and you've extracted your data as you've shown above
        insert_data(
            AAGUID,
            cred_id,
            pk_algo,
            pk_1,
            pk_3,
            pk__1,
            pk__2,
            pk__3
        )



        print("New credential created!")

        print("CLIENT DATA:", result.client_data)
        print("ATTESTATION OBJECT:", result.attestation_object)
        print()
        print("CREDENTIAL DATA:", auth_data.credential_data)

    else:
        server = Fido2Server({"id": "example.com", "name": "Example RP"}, attestation="direct")
        uv = "discouraged"

        # Set up a FIDO 2 client using the origin https://example.com
        client = Fido2Client(dev, "https://example.com", user_interaction=CliInteraction())

        import psycopg2

        # Replace these variables with your connection parameters
        dbname = "CNSA-276-FP-DAS"
        user = "FP-DEV-USER"
        password = "purchase-immortal-prescribe-repave-detention-seizing-candied-antiques-episode-list"
        host = "postgres.eggtech.net"

        # Define the SQL query for retrieving data
        select_query = '''
        SELECT "AAGUID", credential_id, pk_algo, pk_1, pk_3, pk_neg1, pk_neg2, pk_neg3
        FROM credential_data
        WHERE "AAGUID" = %s;
        '''


        # Function to connect to the PostgreSQL database and retrieve data
        def fetch_data(aaguid):
            conn = None
            try:
                # Connect to the PostgreSQL server
                conn = psycopg2.connect(dbname=dbname, user=user, password=password, host=host)
                cur = conn.cursor()

                # Execute the SQL query
                cur.execute(select_query, (gotAaguid,))

                # Fetch the results
                result = cur.fetchone()
                if result:
                    data = {
                        "AAGUID": result[0],
                        "credential_id": result[1],
                        "pk_algo": result[2],
                        "pk_1": result[3],
                        "pk_3": result[4],
                        "pk_neg1": result[5],
                        "pk_neg2": result[6],
                        "pk_neg3": result[7]
                    }
                    return data
                else:
                    print("No data found for AAGUID:", aaguid)
                    return None

                # Close communication with the database
                cur.close()
            except (Exception, psycopg2.DatabaseError) as error:
                print(error)
            finally:
                if conn is not None:
                    conn.close()


        # Example usage: Fetch data for a specific AAGUID and store in variables
        aaguid_data = fetch_data("your_specific_aaguid_here")

        if aaguid_data:
            # Store each piece of data into a separate variable
            testa = aaguid_data["AAGUID"]
            cred_id = aaguid_data["credential_id"]
            pk_algo = aaguid_data["pk_algo"]
            pk_1 = aaguid_data["pk_1"]
            pk_3 = aaguid_data["pk_3"]
            pk_neg1 = aaguid_data["pk_neg1"]
            pk_neg2 = aaguid_data["pk_neg2"]
            pk_neg3 = aaguid_data["pk_neg3"]
            auth_data = fido2.webauthn.AttestedCredentialData(testa, cred_id, pk_algo)
            # auth_data.aaguid = AAGUID
            # auth_data.credential_id = cred_id
            # auth_data.public_key.ALGORITHM= pk_algo
            # auth_data.public_key.setdefault(1, pk_1)
            # auth_data.public_key.setdefault(3, pk_3)
            # auth_data.public_key.setdefault(-1, pk__1)
            # auth_data.public_key.setdefault(-2, pk__2)
            # auth_data.public_key.setdefault(-3,pk__3)

        credentials = [auth_data]


        # Prepare parameters for getAssertion
        request_options, state = server.authenticate_begin(credentials, user_verification=uv)

        # Authenticate the credential
        result = client.get_assertion(request_options["publicKey"])

        # Only one cred in allowCredentials, only one response.
        result = result.get_response(0)

        # Complete authenticator
        server.authenticate_complete(
            state,

            credentials,
            result.credential_id,
            result.client_data,
            result.authenticator_data,
            result.signature,
        )

        print("Credential authenticated!")

        print("CLIENT DATA:", result.client_data)
        print()
        print("AUTH DATA:", result.authenticator_data)