test-auth.py

import json
import sys
import urllib.parse
from lib.NessAuth import NessAuth
from Crypto.Hash import MD5


class AuthTester:

    data_user = {}
    data_node = {}

    def __init__(self):
        return

    def loadUser(self, username: str):
        filename = username + ".key.json"
        f = open("out/keys/user/" + filename, "r")
        return json.loads(f.read())

    def loadNode(self, url: str):
        filename = urllib.parse.quote_plus(url) + ".key.json"
        f = open("out/keys/node/" + filename, "r")
        return json.loads(f.read())

    def test(self, username: str, node_url: str):
        ness_auth = NessAuth()
        user = self.loadUser(username)
        node = self.loadNode(node_url)
        url = node_url + "/node/test/auth"

        user_private_key = user['keys']["private"][user['keys']['current']]

        result = ness_auth.get_by_auth_id(url, user_private_key, node_url, node["nonce"], username, username, user["nonce"])

        if result['result'] == 'error':
            print(" ~~~ TEST #1 Auth ID FAILED ~~~ ")
            print(result['error'])
        else:
            print(" *** TEST #1 Auth ID OK !!! *** ")
            print(result['message'])

        test_string = 'The state calls its own violence law, but that of the individual, crime.'

        result = ness_auth.get_by_two_way_encryption(url, test_string, node['public'], user_private_key, username)

        if result['result'] == 'error':
            print(" ~~~ TEST #2 Two way encryption FAILED ~~~ ")
            print(result['error'])
        else:
            if ness_auth.verify_two_way_result(node['verify'], result):
                print(" *** TEST #2 Two way encryption OK !!! *** ")
                print(ness_auth.decrypt_two_way_result(result, user_private_key))
            else:
                print(" ~~~ TEST #2 Two way encryption FAILED ~~~ ")
                print(" Verifying signature failed ")

                return False

        url = node_url + "/node/joined"

        result = ness_auth.get_by_two_way_encryption(url, 'test', node['public'], user_private_key, username)

        if result['result'] == 'error':
            print(" ~~~ TEST #3 Registration check FAILED ~~~ ")
            print(result['error'])
        else:
            if ness_auth.verify_two_way_result(node['verify'], result):
                print(" *** TEST #3 Registration check OK !!! *** ")
                result = ness_auth.decrypt_two_way_result(result, user_private_key)
                data = json.loads(result)

                if data['joined']:
                    shadowname = data['shadowname']
                    print(" *** The user:" + username + " is joined with shadowname:" + shadowname + " OK !!! ***")
                else:
                    print("The user:" + username + " is not joined yet.")

                    url = node_url + "/node/join"

                    result = ness_auth.get_by_two_way_encryption(url, test_string, node['public'], user_private_key, username)

                    if result['result'] == 'error':
                        print(" ~~~ TEST #3.1 Registration FAILED ~~~ ")
                        print(result['error'])
                    else:
                        if ness_auth.verify_two_way_result(node['verify'], result):
                            print(" *** TEST #3.1 Registration OK *** ")
                            result = ness_auth.decrypt_two_way_result(result, user_private_key)
                            data = json.loads(result)
                            shadowname = data['shadowname']
                            addr = data['address']
                            print("User registered with shadowname:" + shadowname + " and addr:" + addr)
                        else:
                            print(" ~~~ TEST #3.1 Registration FAILED ~~~ ")
                            print(" Verifying signature failed ")


                url = node_url + "/node/test/auth-shadow"

                result = ness_auth.get_by_auth_id(url, user_private_key, node_url, node["nonce"], username, shadowname, user["nonce"])

                if result['result'] == 'error':
                    print(" ~~~ TEST #4 Auth ID with shadowname FAILED ~~~ ")
                    print(result['error'])
                else:
                    print(" *** TEST #4 Auth ID with shadowname OK !!! *** ")
                    print(result['message'])

                test_string = 'Whoever knows how to take, to defend, the thing, to him belongs property'

                result = ness_auth.get_by_two_way_encryption(url, test_string, node['public'], user_private_key, shadowname)

                if result['result'] == 'error':
                    print(" ~~~ TEST #5 Two way encryption FAILED ~~~ ")
                    print(result['error'])
                else:
                    if ness_auth.verify_two_way_result(node['verify'], result):
                        print(" *** TEST #5 Two way encryption OK !!! *** ")
                        print(ness_auth.decrypt_two_way_result(result, user_private_key))
                    else:
                        print(" ~~~ TEST #5 Two way encryption FAILED ~~~ ")
                        print(" Verifying signature failed ")

            else:
                print(" ~~~ TEST #3 Registration check FAILED ~~~ ")
                print(" Verifying signature failed ")

        url = node_url + "/node/test/auth-shadow"

        return True


print('Test authentication')

if len(sys.argv) == 3:
    tester = AuthTester()
    tester.test(sys.argv[1], sys.argv[2])
else:
    print('Usage: python test-auth.py <username> <node URL>')