babashka/test/babashka/crypto_test.clj

52 lines
2.4 KiB
Clojure

(ns babashka.crypto-test
(:require [babashka.test-utils :as test-utils]
[clojure.edn :as edn]
[clojure.test :refer [deftest is]])
(:import (javax.crypto Mac)
(javax.crypto.spec SecretKeySpec)))
(defn bb [& exprs]
(edn/read-string (apply test-utils/bb nil (map str exprs))))
(defn hmac-sha-256 [key data]
(let [algo "HmacSHA256"
mac (Mac/getInstance algo)]
(.init mac (SecretKeySpec. key algo))
(.doFinal mac (.getBytes data "UTF-8"))))
(deftest hmac-sha-256-test
(let [key-s "some-key"
data "some-data"
expected-sha (String. (.encode (java.util.Base64/getEncoder)
(hmac-sha-256 (.getBytes key-s) data))
"utf-8")]
(prn expected-sha)
(is (= expected-sha (bb '(do (ns net
(:import (javax.crypto Mac)
(javax.crypto.spec SecretKeySpec)))
(defn hmac-sha-256 [key data]
(let [algo "HmacSHA256"
mac (Mac/getInstance algo)]
(.init mac (SecretKeySpec. key algo))
(.doFinal mac (.getBytes data "UTF-8"))))
(let [key-s "some-key"
data "some-data"]
(String. (.encode (java.util.Base64/getEncoder)
(hmac-sha-256 (.getBytes key-s) data))
"utf-8"))))))))
(deftest secretkey-test
(is (= 32 (bb '(do (import 'javax.crypto.SecretKeyFactory)
(import 'javax.crypto.spec.PBEKeySpec)
(defn gen-secret-key
"Generate secret key based on a given token string.
Returns bytes array 256-bit length."
[^String secret-token]
(let [salt (.getBytes "abcde")
factory (SecretKeyFactory/getInstance "PBKDF2WithHmacSHA256")
spec (PBEKeySpec. (.toCharArray secret-token) salt 10000 256)
secret (.generateSecret factory spec)]
(count (.getEncoded secret))))
(gen-secret-key "foo"))))))