From bd6f2159931b5877922efed11f7ea9c54b172379 Mon Sep 17 00:00:00 2001
From: JP Appel <jeanpierre.appel01@gmail.com>
Date: Tue, 15 Oct 2024 23:56:50 -0400
Subject: Add authentication middleware

---
 db/auth.go | 80 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 80 insertions(+)
 create mode 100644 db/auth.go

(limited to 'db/auth.go')

diff --git a/db/auth.go b/db/auth.go
new file mode 100644
index 0000000..e2bac38
--- /dev/null
+++ b/db/auth.go
@@ -0,0 +1,80 @@
+package db
+
+import (
+	"crypto/subtle"
+	"database/sql"
+)
+
+type AuthProvider interface {
+	SaltedHash(username string) [32]byte
+	Salt(username string) [16]byte
+	UserExists(username string) bool
+}
+
+type DBAuthProvider struct {
+	db *sql.DB
+}
+
+func NewDBAuthProvider(db *sql.DB) AuthProvider {
+	provider := new(DBAuthProvider)
+	provider.db = db
+
+	return provider
+}
+
+func (dbAuth DBAuthProvider) Close() {
+	dbAuth.db.Close()
+}
+
+func (dbAuth DBAuthProvider) SaltedHash(username string) [32]byte {
+	row := dbAuth.db.QueryRow(`
+    SELECT saltedhash FROM users WHERE username = ?
+    `, username)
+
+	buf := make([]byte, 32)
+	should_copy := 1
+
+	saltedHash := [32]byte{0}
+
+	err := row.Scan(&buf)
+	if err != nil {
+		should_copy = 0
+	}
+
+	subtle.ConstantTimeCopy(should_copy, saltedHash[:], buf[:32])
+
+	return saltedHash
+}
+
+func (dbAuth DBAuthProvider) Salt(username string) [16]byte {
+	row := dbAuth.db.QueryRow(`
+    SELECT salt FROM users WHERE username = ?
+    `, username)
+
+	buf := make([]byte, 16)
+	should_copy := 1
+
+	salt := [16]byte{0}
+
+	err := row.Scan(&buf)
+	if err != nil {
+		should_copy = 0
+	}
+
+	subtle.ConstantTimeCopy(should_copy, salt[:], buf[:16])
+
+	return salt
+}
+
+func (dbAuth DBAuthProvider) UserExists(username string) bool {
+	row := dbAuth.db.QueryRow(`
+    SELECT 1 FROM users WHERE username = ?
+    `, username)
+
+	err := row.Scan(new(int))
+	if err != nil {
+		return false
+	}
+
+	return true
+}
-- 
cgit v1.2.3