You've already forked authenticator-rs
mirror of
https://github.com/librekeys/authenticator-rs.git
synced 2026-04-14 08:47:57 -07:00
J.C. Jones
132 lines
3.8 KiB
Python
132 lines
3.8 KiB
Python
from rich.console import Console
|
|
from rich.logging import RichHandler
|
|
|
|
import argparse
|
|
import logging
|
|
import requests
|
|
|
|
console = Console()
|
|
log = logging.getLogger("webdriver-driver")
|
|
|
|
parser = argparse.ArgumentParser()
|
|
subparsers = parser.add_subparsers(help="sub-command help")
|
|
|
|
parser.add_argument(
|
|
"--verbose", "-v", help="Be more verbose", action="count", default=0
|
|
)
|
|
parser.add_argument(
|
|
"--url",
|
|
default="http://localhost:8080/webauthn/authenticator",
|
|
help="webdriver url",
|
|
)
|
|
|
|
|
|
def device_add(args):
|
|
data = {
|
|
"protocol": args.protocol,
|
|
"transport": args.transport,
|
|
"hasResidentKey": args.residentkey in ["true", "yes"],
|
|
"isUserConsenting": args.consent in ["true", "yes"],
|
|
"hasUserVerification": args.uv in ["available", "verified"],
|
|
"isUserVerified": args.uv in ["verified"],
|
|
}
|
|
console.print("Adding new device: ", data)
|
|
rsp = requests.post(args.url, json=data)
|
|
console.print("Device ID: ", rsp.text)
|
|
|
|
|
|
parser_add = subparsers.add_parser("add", help="Add a device")
|
|
parser_add.set_defaults(func=device_add)
|
|
parser_add.add_argument(
|
|
"--consent",
|
|
choices=["yes", "no", "true", "false"],
|
|
default="true",
|
|
help="consent automatically",
|
|
)
|
|
parser_add.add_argument(
|
|
"--residentkey",
|
|
choices=["yes", "no", "true", "false"],
|
|
default="no",
|
|
help="indicate a resident key",
|
|
)
|
|
parser_add.add_argument(
|
|
"--uv",
|
|
choices=["no", "available", "verified"],
|
|
default="no",
|
|
help="indicate user verification",
|
|
)
|
|
parser_add.add_argument(
|
|
"--protocol", choices=["ctap1/u2f", "ctap2"], default="ctap1/u2f", help="protocol"
|
|
)
|
|
parser_add.add_argument("--transport", default="usb", help="transport type(s)")
|
|
|
|
|
|
def device_delete(args):
|
|
rsp = requests.delete(f"{args.url}/{args.id}")
|
|
console.print(rsp.text)
|
|
|
|
|
|
parser_delete = subparsers.add_parser("delete", help="Delete a device")
|
|
parser_delete.set_defaults(func=device_delete)
|
|
parser_delete.add_argument("id", type=int, help="device ID to delete")
|
|
|
|
|
|
def credential_add(args):
|
|
data = {
|
|
"credentialId": args.credentialId,
|
|
"isResidentCredential": args.isResidentCredential in ["true", "yes"],
|
|
"rpId": args.rpId,
|
|
"privateKey": args.privateKey,
|
|
"signCount": args.signCount,
|
|
}
|
|
if args.userHandle:
|
|
data["userHandle"] = (args.userHandle,)
|
|
|
|
console.print(f"Adding new credential to device {args.id}: ", data)
|
|
rsp = requests.post(f"{args.url}/{args.id}/credential", json=data)
|
|
console.print(rsp.text)
|
|
|
|
|
|
parser_credential_add = subparsers.add_parser("addcred", help="Add a credential")
|
|
parser_credential_add.set_defaults(func=credential_add)
|
|
parser_credential_add.add_argument(
|
|
"--id", required=True, type=int, help="device ID to use"
|
|
)
|
|
parser_credential_add.add_argument(
|
|
"--credentialId", required=True, help="base64url-encoded credential ID"
|
|
)
|
|
parser_credential_add.add_argument(
|
|
"--isResidentCredential",
|
|
choices=["yes", "no", "true", "false"],
|
|
default="no",
|
|
help="indicate a resident key",
|
|
)
|
|
parser_credential_add.add_argument("--rpId", required=True, help="RP id (hostname)")
|
|
parser_credential_add.add_argument(
|
|
"--privateKey", required=True, help="base64url-encoded private key per RFC 5958"
|
|
)
|
|
parser_credential_add.add_argument("--userHandle", help="base64url-encoded user handle")
|
|
parser_credential_add.add_argument(
|
|
"--signCount", default=0, type=int, help="initial signature counter"
|
|
)
|
|
|
|
|
|
def main():
|
|
args = parser.parse_args()
|
|
|
|
loglevel = logging.INFO
|
|
if args.verbose > 0:
|
|
loglevel = logging.DEBUG
|
|
logging.basicConfig(
|
|
level=loglevel, format="%(message)s", datefmt="[%X]", handlers=[RichHandler()]
|
|
)
|
|
|
|
try:
|
|
args.func(args)
|
|
except requests.exceptions.ConnectionError as ce:
|
|
log.error(f"Connection refused to {args.url}: {ce}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|