JWT.h

/*
╔══════════════════════════════════════════════════════════════════════════════╗
║                       SIMPLY POWERFUL TOOLKIT (SPTK)                         ║
╟──────────────────────────────────────────────────────────────────────────────╢
║  copyright            © 1999-2022 Alexey Parshin. All rights reserved.       ║
║  email                alexeyp@gmail.com                                      ║
╚══════════════════════════════════════════════════════════════════════════════╝
┌──────────────────────────────────────────────────────────────────────────────┐
│   This library is free software; you can redistribute it and/or modify it    │
│   under the terms of the GNU Library General Public License as published by  │
│   the Free Software Foundation; either version 2 of the License, or (at your │
│   option) any later version.                                                 │
│                                                                              │
│   This library is distributed in the hope that it will be useful, but        │
│   WITHOUT ANY WARRANTY; without even the implied warranty of                 │
│   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Library   │
│   General Public License for more details.                                   │
│                                                                              │
│   You should have received a copy of the GNU Library General Public License  │
│   along with this library; if not, write to the Free Software Foundation,    │
│   Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.               │
│                                                                              │
│   Please report all bugs and problems to alexeyp@gmail.com.                  │
└──────────────────────────────────────────────────────────────────────────────┘
┌──────────────────────────────────────────────────────────────────────────────┐
│   The code in this module is based JWT C Library, developed by Ben Collins.  │
│   Please see http://github.com/benmcollins/libjwt for more information.      │
└──────────────────────────────────────────────────────────────────────────────┘
*/
 
#pragma once
 
#include <sptk5/cutils>
#include <sptk5/xdoc/Document.h>
#include <sstream>
 
namespace sptk {
 
class SP_EXPORT JWT
{
public:
    enum class Algorithm : uint8_t
    {
        NONE = 0,
        HS256,
        HS384,
        HS512,
        RS256,
        RS384,
        RS512,
        ES256,
        ES384,
        ES512,
        INVAL
    };
 
    Algorithm alg {Algorithm::NONE}; 
    String key;                      
    xdoc::Document grants;           
 
    JWT() = default;
 
    const Variant& get(const String& name) const
    {
        return grants.root()->findOrCreate(name)->getValue();
    }
 
    void set(const String& name, const Variant& data) const
    {
        grants.root()->set(name, data);
    }
 
    /*
         * Get signature encryption algorithm
         * @return signature encryption algorithm
         */
    Algorithm get_alg() const;
 
    void set_alg(Algorithm _alg, const String& _key);
 
    static const char* alg_str(Algorithm _alg);
 
    static Algorithm str_alg(const char* alg);
 
    void sign(Buffer& token, const char* str) const;
 
    void encode(std::ostream& out) const;
 
    void decode(const char* token, const String& _key = "");
 
    void exportTo(std::ostream& output, bool pretty) const;
 
    static String get_js_string(const xdoc::SNode& node, const String& key, bool* found = nullptr);
 
    static long get_js_int(const xdoc::SNode& node, const String& key, bool* found = nullptr);
 
    static bool get_js_bool(const xdoc::SNode& node, const String& key, bool* found = nullptr);
 
    void write_head(std::ostream& output, bool pretty) const;
 
    void write_body(std::ostream& output, bool pretty) const;
 
    void verify(const Buffer& head, const Buffer& sig) const;
 
    void sign_sha_hmac(Buffer& out, const char* str) const;
 
    void verify_sha_hmac(const char* head, const char* sig) const;
 
    void sign_sha_pem(Buffer& out, const char* str) const;
 
    void verify_sha_pem(const char* head, const char* sig_b64) const;
 
private:
    static xdoc::SNode find_grant(const xdoc::SNode& node, const String& key);
};
 
void jwt_base64uri_encode(Buffer& buffer);
 
void jwt_b64_decode(Buffer& destination, const char* src);
 
} // namespace sptk