auth.hpp

Go to the documentation of this file.

#pragma once
 
#include <random>
#include <string>
#include <expected>
#include <format>
#include <regex>
 
#include <absl/container/flat_hash_map.h>
#include <spdlog/spdlog.h>
#include <secp256k1.h>
#include <secp256k1_recovery.h>
#include <jwt-cpp/jwt.h>
 
#include "async.hpp"
#include "keccak256.hpp"
#include "chain.hpp"
 
namespace dcn::auth
{
    struct AuthError
    {
        enum class Kind : std::uint8_t
        {
            UNKNOWN = 0,
 
            MISSING_TOKEN,
            INVALID_TOKEN,
 
            INVALID_SIGNATURE,
 
            INVALID_NONCE,
 
            INVALID_ADDRESS
        }
        kind = Kind::UNKNOWN;
 
        std::string message = "";
    };
 
    class AuthManager
    {
        public:
            AuthManager() = delete;
            AuthManager(asio::io_context & io_context);
 
            AuthManager(const AuthManager&) = delete;
            AuthManager& operator=(const AuthManager&) = delete;
            
            ~AuthManager() = default;
 
            asio::awaitable<std::string> generateNonce(const chain::Address & address);
 
            asio::awaitable<bool> verifyNonce(const chain::Address & address, const std::string & nonce);
 
            asio::awaitable<bool> verifySignature(const chain::Address & address, const std::string& signature, const std::string& message);
 
            asio::awaitable<std::string> generateAccessToken(const chain::Address & address);
 
            asio::awaitable<std::expected<chain::Address, AuthError>> verifyAccessToken(std::string token) const;
 
            asio::awaitable<bool> compareAccessToken(const chain::Address & address, std::string token) const;
 
            asio::awaitable<void> invalidateAccessToken(const chain::Address & address);
 
        private:
            asio::strand<asio::io_context::executor_type> _strand;
 
            const std::string _SECRET; // !!! TODO !!! use secure secret in production
 
            static std::random_device _rng;
 
            std::uniform_int_distribution<int> _dist;
            absl::flat_hash_map<chain::Address, std::string> _nonces;
 
            absl::flat_hash_map<chain::Address, std::string> _access_tokens;
    };
}
 
template <>
struct std::formatter<dcn::auth::AuthError::Kind> : std::formatter<std::string> {
    auto format(const dcn::auth::AuthError::Kind & err, format_context& ctx) const {
        switch(err)
        {
            case dcn::auth::AuthError::Kind::MISSING_TOKEN : return formatter<string>::format("Missing token", ctx);
            case dcn::auth::AuthError::Kind::INVALID_TOKEN : return formatter<string>::format("Invalid token", ctx);
            case dcn::auth::AuthError::Kind::INVALID_SIGNATURE : return formatter<string>::format("Invalid signature", ctx);
            case dcn::auth::AuthError::Kind::INVALID_NONCE : return formatter<string>::format("Invalid nonce", ctx);
            case dcn::auth::AuthError::Kind::INVALID_ADDRESS : return formatter<string>::format("Invalid address", ctx);
 
            default:  return formatter<string>::format("Unknown", ctx);
        }
        return formatter<string>::format("", ctx);
    }
};