Compare commits

...

15 Commits

15 changed files with 376 additions and 128 deletions

View File

@ -8,8 +8,10 @@ ADMIN_NAME="admin"
ADMIN_EMAIL="admin@test.net"
ADMIN_PASSWORD="secret"
JWT_PRIVATE_KEY="ecdsa"
JWT_PUBLIC_KEY="ecdsa"
JWT_PRIVATE_KEY="rsa"
JWT_PUBLIC_KEY="rsa"
JWT_KID="my-rsa-key-1"
MINIO_ENDPOINT="localhost:9000"
MINIO_ACCESS_KEY=""

View File

@ -38,10 +38,10 @@ func (s *APIServer) Run() error {
// staticDir := http.Dir(filepath.Join(workDir, "static"))
// FileServer(router, "/static", staticDir)
oauthHandler := oauth.NewOAuthHandler()
oauthHandler := oauth.NewOAuthHandler(s.repo)
router.Route("/api/v1", func(r chi.Router) {
r.Use(imiddleware.WithSkipper(imiddleware.AuthMiddleware, "/api/v1/login", "/api/v1/register", "/api/v1/oauth"))
r.Use(imiddleware.WithSkipper(imiddleware.AuthMiddleware, "/api/v1/login", "/api/v1/register", "/api/v1/oauth/token"))
userHandler := user.NewUserHandler(s.repo, s.storage)
userHandler.RegisterRoutes(r)
@ -52,7 +52,8 @@ func (s *APIServer) Run() error {
oauthHandler.RegisterRoutes(r)
})
router.Get("/.well-known/openid-configuration", oauthHandler.Metadata)
router.Get("/.well-known/jwks.json", auth.WriteJWKS)
router.Get("/.well-known/openid-configuration", auth.OpenIdConfiguration)
router.Get("/*", func(w http.ResponseWriter, r *http.Request) {
path := "./dist" + r.URL.Path

View File

@ -1,52 +1,105 @@
package auth
import (
"crypto/ecdsa"
"crypto/rsa"
"crypto/x509"
"encoding/base64"
"encoding/json"
"fmt"
"net/http"
"os"
"gitea.local/admin/hspguard/internal/types"
"gitea.local/admin/hspguard/internal/web"
"github.com/golang-jwt/jwt/v5"
)
func parseBase64PrivateKey(envVar string) (*ecdsa.PrivateKey, error) {
b64 := os.Getenv(envVar)
if b64 == "" {
return nil, fmt.Errorf("env var %s is empty", envVar)
}
func WriteJWKS(w http.ResponseWriter, r *http.Request) {
pubKey, err := parseBase64PublicKey("JWT_PUBLIC_KEY")
if err != nil {
web.Error(w, "failed to parse public key", http.StatusInternalServerError)
}
decoded, err := base64.StdEncoding.DecodeString(b64)
if err != nil {
return nil, fmt.Errorf("failed to decode base64 key: %v", err)
}
n := base64.RawURLEncoding.EncodeToString(pubKey.N.Bytes())
e := base64.RawURLEncoding.EncodeToString([]byte{1, 0, 1}) // 65537 = 0x010001
return x509.ParseECPrivateKey(decoded)
jwks := map[string]interface{}{
"keys": []map[string]string{
{
"kty": "RSA",
"kid": "my-rsa-key-1",
"use": "sig",
"alg": "RS256",
"n": n,
"e": e,
},
},
}
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(jwks)
}
func parseBase64PublicKey(envVar string) (*ecdsa.PublicKey, error) {
b64 := os.Getenv(envVar)
if b64 == "" {
return nil, fmt.Errorf("env var %s is empty", envVar)
}
func OpenIdConfiguration(w http.ResponseWriter, r *http.Request) {
type Response struct {
TokenEndpoint string `json:"token_endpoint"`
AuthorizationEndpoint string `json:"authorization_endpoint"`
JwksURI string `json:"jwks_uri"`
Issuer string `json:"issuer"`
EndSessionEndpoint string `json:"end_session_endpoint"`
}
decoded, err := base64.StdEncoding.DecodeString(b64)
if err != nil {
return nil, fmt.Errorf("failed to decode base64 key: %v", err)
}
w.Header().Set("Content-Type", "application/json")
pubInterface, err := x509.ParsePKIXPublicKey(decoded)
if err != nil {
return nil, fmt.Errorf("failed to parse public key: %v", err)
}
encoder := json.NewEncoder(w)
if err := encoder.Encode(Response{
TokenEndpoint: "https://cb5f-2a00-10-5b00-c801-e955-5c68-63d0-b777.ngrok-free.app/api/v1/oauth/token",
AuthorizationEndpoint: "https://cb5f-2a00-10-5b00-c801-e955-5c68-63d0-b777.ngrok-free.app/authorize",
JwksURI: "https://cb5f-2a00-10-5b00-c801-e955-5c68-63d0-b777.ngrok-free.app/.well-known/jwks.json",
Issuer: "https://cb5f-2a00-10-5b00-c801-e955-5c68-63d0-b777.ngrok-free.app",
EndSessionEndpoint: "https://cb5f-2a00-10-5b00-c801-e955-5c68-63d0-b777.ngrok-free.app/api/v1/oauth/logout",
}); err != nil {
web.Error(w, "failed to encode response", http.StatusInternalServerError)
}
}
pubKey, ok := pubInterface.(*ecdsa.PublicKey)
if !ok {
return nil, fmt.Errorf("not an ECDSA public key")
}
func parseBase64PrivateKey(envVar string) (*rsa.PrivateKey, error) {
b64 := os.Getenv(envVar)
if b64 == "" {
return nil, fmt.Errorf("env var %s is empty", envVar)
}
return pubKey, nil
decoded, err := base64.StdEncoding.DecodeString(b64)
if err != nil {
return nil, fmt.Errorf("failed to decode base64 key: %v", err)
}
key, err := x509.ParsePKCS8PrivateKey(decoded)
return key.(*rsa.PrivateKey), err
}
func parseBase64PublicKey(envVar string) (*rsa.PublicKey, error) {
b64 := os.Getenv(envVar)
if b64 == "" {
return nil, fmt.Errorf("env var %s is empty", envVar)
}
decoded, err := base64.StdEncoding.DecodeString(b64)
if err != nil {
return nil, fmt.Errorf("failed to decode base64 key: %v", err)
}
pubInterface, err := x509.ParsePKIXPublicKey(decoded)
if err != nil {
return nil, fmt.Errorf("failed to parse public key: %v", err)
}
pubKey, ok := pubInterface.(*rsa.PublicKey)
if !ok {
return nil, fmt.Errorf("not an RSA public key")
}
return pubKey, nil
}
func SignJwtToken(claims jwt.Claims) (string, error) {
@ -55,7 +108,10 @@ func SignJwtToken(claims jwt.Claims) (string, error) {
return "", err
}
token := jwt.NewWithClaims(jwt.SigningMethodES256, claims)
token := jwt.NewWithClaims(jwt.SigningMethodRS256, claims)
token.Header["kid"] = "my-rsa-key-1"
s, err := token.SignedString(privateKey)
if err != nil {
return "", err
@ -72,7 +128,7 @@ func VerifyToken(token string) (*jwt.Token, *types.UserClaims, error) {
claims := &types.UserClaims{}
parsed, err := jwt.ParseWithClaims(token, claims, func(t *jwt.Token) (any, error) {
if _, ok := t.Method.(*jwt.SigningMethodECDSA); !ok {
if _, ok := t.Method.(*jwt.SigningMethodRSA); !ok {
return nil, fmt.Errorf("unexpected signing method: %v", t.Header["alg"])
}
return publicKey, nil
@ -88,4 +144,3 @@ func VerifyToken(token string) (*jwt.Token, *types.UserClaims, error) {
return parsed, claims, nil
}

View File

@ -132,6 +132,8 @@ func (h *AuthHandler) login(w http.ResponseWriter, r *http.Request) {
// Avatar
}
w.Header().Set("Content-Type", "application/json")
if err := encoder.Encode(Response{
AccessToken: accessToken,
RefreshToken: refreshToken,

View File

@ -1,49 +1,210 @@
package oauth
import (
"encoding/base64"
"encoding/json"
"fmt"
"log"
"net/http"
"strings"
"time"
"gitea.local/admin/hspguard/internal/auth"
"gitea.local/admin/hspguard/internal/repository"
"gitea.local/admin/hspguard/internal/types"
"gitea.local/admin/hspguard/internal/util"
"gitea.local/admin/hspguard/internal/web"
"github.com/go-chi/chi/v5"
"github.com/golang-jwt/jwt/v5"
"github.com/google/uuid"
)
type OAuthHandler struct{}
type OAuthHandler struct {
repo *repository.Queries
}
func NewOAuthHandler() *OAuthHandler {
return &OAuthHandler{}
func NewOAuthHandler(repo *repository.Queries) *OAuthHandler {
return &OAuthHandler{
repo,
}
}
func (h *OAuthHandler) RegisterRoutes(r chi.Router) {
r.Get("/oauth/authorize", h.authorizeEndpoint)
r.Get("/oauth/token", h.tokenEndpoint)
r.Post("/oauth/token", h.tokenEndpoint)
r.Post("/oauth/code", h.getAuthCode)
}
func (h *OAuthHandler) tokenEndpoint(w http.ResponseWriter, r *http.Request) {
log.Println("[OAUTH] New request to token endpoint")
w.WriteHeader(http.StatusOK)
w.Write([]byte("OK"))
}
func (h *OAuthHandler) getAuthCode(w http.ResponseWriter, r *http.Request) {
userId, ok := util.GetRequestUserId(r.Context())
if !ok {
web.Error(w, "failed to get user id from auth session", http.StatusInternalServerError)
return
}
func (h *OAuthHandler) authorizeEndpoint(w http.ResponseWriter, r *http.Request) {
log.Println("[OAUTH] New request to authorize endpoint")
w.WriteHeader(http.StatusOK)
w.Write([]byte("OK"))
}
user, err := h.repo.FindUserId(r.Context(), uuid.MustParse(userId))
if err != nil {
web.Error(w, "user with provided id does not exist", http.StatusUnauthorized)
return
}
type Request struct {
Nonce string `json:"nonce"`
}
var req Request
decoder := json.NewDecoder(r.Body)
if err := decoder.Decode(&req); err != nil {
web.Error(w, "nonce field is required in request", http.StatusBadRequest)
return
}
// TODO: Create real authorization code
func (h *OAuthHandler) Metadata(w http.ResponseWriter, r *http.Request) {
type Response struct {
TokenEndpoint string `json:"token_endpoint"`
AuthEndpoint string `json:"authorization_endpoint"`
Code string `json:"code"`
}
encoder := json.NewEncoder(w)
w.Header().Set("Content-Type", "application/json")
if err := encoder.Encode(Response{
TokenEndpoint: "http://192.168.178.21:3001/api/v1/oauth/token",
AuthEndpoint: "http://192.168.178.21:5173/authorize",
Code: fmt.Sprintf("%s,%s", user.ID.String(), req.Nonce),
}); err != nil {
web.Error(w, "failed to encode response", http.StatusInternalServerError)
}
}
func (h *OAuthHandler) tokenEndpoint(w http.ResponseWriter, r *http.Request) {
log.Println("[OAUTH] New request to token endpoint")
authHeader := r.Header.Get("Authorization")
if authHeader == "" || !strings.HasPrefix(authHeader, "Basic ") {
w.Header().Set("WWW-Authenticate", `Basic realm="Restricted"`)
http.Error(w, "Unauthorized", http.StatusUnauthorized)
return
}
// Decode credentials
payload, err := base64.StdEncoding.DecodeString(strings.TrimPrefix(authHeader, "Basic "))
if err != nil {
http.Error(w, "Invalid auth encoding", http.StatusBadRequest)
return
}
var clientId string
var clientSecret string
parts := strings.SplitN(string(payload), ":", 2)
if len(parts) != 2 {
http.Error(w, "Unauthorized", http.StatusForbidden)
return
}
clientId = parts[0]
clientSecret = parts[1]
log.Printf("Some client is trying to exchange code with id: %s and secret: %s\n", clientId, clientSecret)
// Parse the form data
err = r.ParseForm()
if err != nil {
http.Error(w, "Failed to parse form", http.StatusBadRequest)
return
}
grantType := r.FormValue("grant_type")
redirectUri := r.FormValue("redirect_uri")
log.Printf("Redirect URI is %s\n", redirectUri)
switch grantType {
case "authorization_code":
code := r.FormValue("code")
fmt.Printf("Code received: %s\n", code)
// TODO: Verify code from another db table
nonce := strings.Split(code, ",")[1]
userId := strings.Split(code, ",")[0]
user, err := h.repo.FindUserId(r.Context(), uuid.MustParse(userId))
if err != nil {
web.Error(w, "requested user not found", http.StatusNotFound)
return
}
claims := types.ApiClaims{
Email: user.Email,
// TODO:
EmailVerified: true,
Name: user.FullName,
Picture: user.ProfilePicture.String,
Nonce: nonce,
Roles: []string{"user", "admin"},
RegisteredClaims: jwt.RegisteredClaims{
Issuer: "https://cb5f-2a00-10-5b00-c801-e955-5c68-63d0-b777.ngrok-free.app",
// TODO: use dedicated API id that is in local DB and bind to user there
Subject: user.ID.String(),
Audience: jwt.ClaimStrings{clientId},
IssuedAt: jwt.NewNumericDate(time.Now()),
ExpiresAt: jwt.NewNumericDate(time.Now().Add(15 * time.Minute)),
},
}
idToken, err := auth.SignJwtToken(claims)
if err != nil {
web.Error(w, "failed to sign id token", http.StatusInternalServerError)
return
}
type Response struct {
IdToken string `json:"id_token"`
TokenType string `json:"token_type"`
AccessToken string `json:"access_token"`
Email string `json:"email"`
// TODO: add expires_in, refresh_token, scope (RFC 8693 $2)
}
response := Response{
IdToken: idToken,
TokenType: "Bearer",
// FIXME:
AccessToken: idToken,
Email: user.Email,
}
log.Printf("sending following response: %#v\n", response)
w.Header().Set("Content-Type", "application/json")
encoder := json.NewEncoder(w)
if err := encoder.Encode(response); err != nil {
web.Error(w, "failed to encode response", http.StatusInternalServerError)
}
default:
web.Error(w, "unsupported grant type", http.StatusBadRequest)
}
}
// func (h *OAuthHandler) Metadata(w http.ResponseWriter, r *http.Request) {
// type Response struct {
// TokenEndpoint string `json:"token_endpoint"`
// AuthEndpoint string `json:"authorization_endpoint"`
// Issuer string `json:"issuer"`
// JwksUri string `json:"jwks_uri"`
// }
// encoder := json.NewEncoder(w)
// if err := encoder.Encode(Response{
// TokenEndpoint: "http://192.168.178.21:3001/api/v1/oauth/token",
// AuthEndpoint: "https://147f-2a00-10-5b00-c801-4882-7ef0-5e68-d360.ngrok-free.app/authorize",
// JwksUri: "http://192.168.178.21:3001/.well-known/jwks.json",
// Issuer: "http://192.168.178.21:3001",
// }); err != nil {
// web.Error(w, "failed to encode response", http.StatusInternalServerError)
// }
// }

View File

@ -8,12 +8,12 @@ type UserClaims struct {
}
type ApiClaims struct {
UserID string `json:"user_id"`
// Permissions are guard's defined permissions
// Examples:
// 1. User MetaData (specifically some fields like email, profile picture and name)
// 2. Actions on User, e.g. home permissions fetching, notifications emitting
Permissions []string `json:"permissions"`
// Subject is an API ID defined in guard's DB after registration
Email string `json:"email"`
EmailVerified bool `json:"email_verified"`
Name string `json:"name"`
Picture string `json:"picture"`
Nonce string `json:"nonce"`
Roles []string `json:"roles"`
// TODO: add given_name, family_name, locale...
jwt.RegisteredClaims
}

View File

@ -1,16 +1,17 @@
# Generate private key
openssl ecparam -genkey -name prime256v1 -noout -out ec256-private.pem
# Generate 2048-bit RSA private key (suppress output)
openssl genpkey -algorithm RSA -out rsa-private.pem -pkeyopt rsa_keygen_bits:2048 *> $null
# Extract public key
openssl ec -in ec256-private.pem -pubout -out ec256-public.pem
# Extract the public key from the private key (suppress output)
openssl rsa -in rsa-private.pem -pubout -out rsa-public.pem *> $null
Write-Host ""
Write-Host "Private Key (DER base64):"
openssl ec -in ec256-private.pem -outform DER | openssl base64 -A
Write-Host ""
Write-Host "--------------------------------"
Write-Host ""
# Base64 encode private key (DER format, for JWT_PRIVATE_KEY)
Write-Host -NoNewline 'JWT_PRIVATE_KEY="'
openssl pkcs8 -topk8 -nocrypt -in rsa-private.pem -outform DER 2>$null | openssl base64 -A
Write-Host '"'
Write-Host "Public Key (DER base64):"
openssl ec -in ec256-private.pem -pubout -outform DER | openssl base64 -A
# Base64 encode public key (DER format, for JWT_PUBLIC_KEY)
Write-Host -NoNewline 'JWT_PUBLIC_KEY="'
openssl rsa -in rsa-private.pem -pubout -outform DER 2>$null | openssl base64 -A
Write-Host '"'

View File

@ -1,26 +1,19 @@
#!/bin/bash
# Generate private key
# openssl ecparam -genkey -name prime256v1 -noout -out ec256-private.pem
# openssl ec -in ec256-private.pem -outform DER | base64 -w 0
# Generate 2048-bit RSA private key (suppress all output)
openssl genpkey -algorithm RSA -out rsa-private.pem -pkeyopt rsa_keygen_bits:2048 >/dev/null 2>&1
# Extract public key
# openssl ec -in ec256-private.pem -pubout -out ec256-public.pem
# openssl ec -in ec256-private.pem -pubout -outform DER | base64 -w 0
# Generate private key
openssl ecparam -genkey -name prime256v1 -noout -out ec256-private.pem
# Extract public key
openssl ec -in ec256-private.pem -pubout -out ec256-public.pem
# Extract the public key from the private key (suppress all output)
openssl rsa -in rsa-private.pem -pubout -out rsa-public.pem >/dev/null 2>&1
echo ""
echo "Private Key (DER base64):"
openssl ec -in ec256-private.pem -outform DER | base64 -w 0
echo "
--------------------------------"
# Base64 encode private key (for JWT_PRIVATE_KEY)
echo -n 'JWT_PRIVATE_KEY="'
openssl pkcs8 -topk8 -nocrypt -in rsa-private.pem -outform DER 2>/dev/null | base64 -w 0
echo '"'
echo ""
echo "Public Key (DER base64):"
openssl ec -in ec256-private.pem -pubout -outform DER | base64 -w 0
# Base64 encode public key (for JWT_PUBLIC_KEY)
echo -n 'JWT_PUBLIC_KEY="'
openssl rsa -in rsa-private.pem -pubout -outform DER 2>/dev/null | base64 -w 0
echo '"'

View File

@ -5,7 +5,7 @@
"type": "module",
"scripts": {
"dev": "vite",
"build": "tsc -b && vite build --watch",
"build": "tsc -b && vite build",
"lint": "eslint .",
"preview": "vite preview"
},

25
web/src/api/code.ts Normal file
View File

@ -0,0 +1,25 @@
import { handleApiError } from ".";
export interface CodeResponse {
code: string;
}
export const codeApi = async (accessToken: string, nonce: string) => {
const response = await fetch("/api/v1/oauth/code", {
method: "POST",
headers: {
"Content-Type": "application/json",
Authorization: `Bearer ${accessToken}`,
},
body: JSON.stringify({
nonce,
}),
});
if (response.status !== 200 && response.status !== 201)
throw await handleApiError(response);
const data: CodeResponse = await response.json();
return data;
};

View File

@ -13,6 +13,7 @@ interface OAuthContextValues {
setScope: (scopes: string[]) => void;
setState: (state: string) => void;
setNonce: (nonce: string) => void;
selectSession: (token: string) => Promise<void>;
}
export const OAuthContext = createContext<OAuthContextValues>({
@ -28,6 +29,7 @@ export const OAuthContext = createContext<OAuthContextValues>({
setScope: () => {},
setState: () => {},
setNonce: () => {},
selectSession: async () => {},
});
export const useOAuthContext = () => useContext(OAuthContext);

View File

@ -1,5 +1,6 @@
import { useState, type FC, type ReactNode } from "react";
import { useCallback, useState, type FC, type ReactNode } from "react";
import { OAuthContext } from ".";
import { codeApi } from "@/api/code";
interface IOAuthProvider {
children: ReactNode;
@ -13,6 +14,24 @@ export const OAuthProvider: FC<IOAuthProvider> = ({ children }) => {
const [state, setState] = useState("");
const [nonce, setNonce] = useState("");
const selectSession = useCallback(
async (token: string) => {
if (active && redirectURI) {
const codeResponse = await codeApi(token, nonce);
console.log("gen code:", { codeResponse });
const params = new URLSearchParams({
code: codeResponse.code,
state,
});
window.location.replace(`${redirectURI}?${params.toString()}`);
}
},
[active, nonce, redirectURI, state]
);
return (
<OAuthContext.Provider
value={{
@ -28,6 +47,7 @@ export const OAuthProvider: FC<IOAuthProvider> = ({ children }) => {
setScope,
setState,
setNonce,
selectSession,
}}
>
{children}

View File

@ -1,7 +1,8 @@
import { useDbContext } from "@/context/db";
import { useOAuthContext } from "@/context/oauth";
import { type LocalAccount, useAccountRepo } from "@/repository/account";
import { CirclePlus, User } from "lucide-react";
import { useEffect, useState, type FC } from "react";
import { useCallback, useEffect, useState, type FC } from "react";
import { Link } from "react-router-dom";
const AccountList: FC = () => {
@ -10,6 +11,15 @@ const AccountList: FC = () => {
const repo = useAccountRepo();
const { connected } = useDbContext();
const oauth = useOAuthContext();
const handleAccountSelect = useCallback(
(account: LocalAccount) => {
oauth.selectSession(account.access);
},
[oauth]
);
useEffect(() => {
if (connected) repo.loadAll().then(setAccounts);
// eslint-disable-next-line react-hooks/exhaustive-deps
@ -46,6 +56,7 @@ const AccountList: FC = () => {
{accounts.map((account) => (
<div
key={account.accountId}
onClick={() => handleAccountSelect(account)}
className="flex flex-row items-center p-4 border-gray-200 dark:border-gray-700/65 border-b border-r-0 border-l-0 select-none cursor-pointer hover:bg-gray-50/50 dark:hover:bg-gray-800/10 transition-colors mb-0"
>
<div>

View File

@ -9,6 +9,7 @@ import { useForm, type SubmitHandler } from "react-hook-form";
import { useCallback, useState } from "react";
import { loginApi } from "@/api/login";
import { useAccountRepo } from "@/repository/account";
import { useOAuthContext } from "@/context/oauth";
interface LoginForm {
email: string;
@ -27,6 +28,8 @@ export default function LoginPage() {
const [error, setError] = useState("");
const [success, setSuccess] = useState("");
const oauth = useOAuthContext();
const repo = useAccountRepo();
const onSubmit: SubmitHandler<LoginForm> = useCallback(
@ -56,6 +59,8 @@ export default function LoginPage() {
setSuccess("You have successfully logged in");
reset();
oauth.selectSession(response.access);
} catch (err: any) {
console.log(err);
setError(
@ -66,7 +71,7 @@ export default function LoginPage() {
setLoading(false);
}
},
[repo, reset]
[oauth, repo, reset]
);
return (

View File

@ -1,30 +0,0 @@
import { deriveDeviceKey, getDeviceId } from "./deviceId";
import { encryptToken } from "./token";
const storeTokensForAccount = async (
accountId: string,
accessToken: string,
refreshToken: string
) => {
const deviceKeyId = await getDeviceId();
const key = await deriveDeviceKey(deviceKeyId);
const access = await encryptToken(accessToken, key);
const refresh = await encryptToken(refreshToken, key);
const entry = {
accountId,
label: `Account for ${accountId}`,
access: {
data: Array.from(new Uint8Array(access.cipherText)),
iv: Array.from(access.iv),
},
refresh: {
data: Array.from(new Uint8Array(refresh.cipherText)),
iv: Array.from(refresh.iv),
},
updatedAt: new Date().toISOString(),
};
// Save this `entry` in IndexedDB (or use a localforage wrapper)
};