web-base/Core/Objects/TwoFactor/KeyBasedTwoFactorToken.class.php

<?php

namespace Core\Objects\TwoFactor;

use Core\Driver\SQL\SQL;
use Cose\Algorithm\Signature\ECDSA\ECSignature;
use Core\Objects\DatabaseEntity\TwoFactorToken;

class KeyBasedTwoFactorToken extends TwoFactorToken {

  const TYPE = "fido";

  private ?string $challenge;
  private ?string $credentialId;
  private ?PublicKey $publicKey;

  public function __construct(string $challenge) {
    parent::__construct(self::TYPE);
    $this->challenge = $challenge;
  }

  protected function readData(string $data) {
    if (!$this->isConfirmed()) {
      $this->challenge = base64_decode($data);
      $this->credentialId = null;
      $this->publicKey = null;
    } else {
      $jsonData = json_decode($data, true);
      $this->challenge = base64_decode($_SESSION["challenge"] ?? "");
      $this->credentialId = base64_decode($jsonData["credentialID"]);
      $this->publicKey = PublicKey::fromJson($jsonData["publicKey"]);
    }
  }

  public function getData(): string {
    if ($this->isConfirmed()) {
      return base64_encode($this->challenge);
    } else {
      return json_encode([
        "credentialId" => $this->credentialId,
        "publicKey" => $this->publicKey->jsonSerialize()
      ]);
    }
  }

  public function confirmKeyBased(SQL $sql, string $credentialId, PublicKey $publicKey): bool {
    $this->credentialId = $credentialId;
    $this->publicKey = $publicKey;
    return parent::confirm($sql);
  }


  public function getPublicKey(): ?PublicKey {
    return $this->publicKey;
  }

  public function getCredentialId(): ?string {
    return $this->credentialId;
  }

  public function jsonSerialize(?array $propertyNames = null): array {
    $jsonData = parent::jsonSerialize();

    if (!empty($this->challenge) && !$this->isAuthenticated() && in_array("challenge", $propertyNames)) {
      $jsonData["challenge"] = base64_encode($this->challenge);
    }

    if (!empty($this->credentialId) && in_array("credentialID", $propertyNames)) {
      $jsonData["credentialID"] = base64_encode($this->credentialId);
    }

    return $jsonData;
  }

  // TODO: algorithms, hardcoded values, ...
  public function verify(string $signature, string $data): bool {
    switch ($this->publicKey->getUsedAlgorithm()) {
      case -7: // EC2

        if (strlen($signature) !== 64) {
          $signature = \Cose\Algorithm\Signature\ECDSA\ECSignature::fromAsn1($signature, 64);
        }

        $coseKey = new \Cose\Key\Key($this->publicKey->getNormalizedData());
        $ec2key = new \Cose\Key\Ec2Key($coseKey->getData());
        $publicKey = $ec2key->toPublic();
        $signature = ECSignature::toAsn1($signature, 64);
        return openssl_verify($data, $signature, $publicKey->asPEM(), "sha256") === 1;
      default:
        // Not implemented :(
        return false;
    }
  }
}