Last active
May 4, 2020 14:54
-
-
Save jernejstrasner/1d5fa5e2fabda2e729d1 to your computer and use it in GitHub Desktop.
HMAC digest in Swift
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
// Make sure you add #import <CommonCrypto/CommonCrypto.h> to the Xcode bridging header! | |
enum CryptoAlgorithm { | |
case MD5, SHA1, SHA224, SHA256, SHA384, SHA512 | |
var HMACAlgorithm: CCHmacAlgorithm { | |
var result: Int = 0 | |
switch self { | |
case .MD5: result = kCCHmacAlgMD5 | |
case .SHA1: result = kCCHmacAlgSHA1 | |
case .SHA224: result = kCCHmacAlgSHA224 | |
case .SHA256: result = kCCHmacAlgSHA256 | |
case .SHA384: result = kCCHmacAlgSHA384 | |
case .SHA512: result = kCCHmacAlgSHA512 | |
} | |
return CCHmacAlgorithm(result) | |
} | |
var digestLength: Int { | |
var result: Int32 = 0 | |
switch self { | |
case .MD5: result = CC_MD5_DIGEST_LENGTH | |
case .SHA1: result = CC_SHA1_DIGEST_LENGTH | |
case .SHA224: result = CC_SHA224_DIGEST_LENGTH | |
case .SHA256: result = CC_SHA256_DIGEST_LENGTH | |
case .SHA384: result = CC_SHA384_DIGEST_LENGTH | |
case .SHA512: result = CC_SHA512_DIGEST_LENGTH | |
} | |
return Int(result) | |
} | |
} | |
extension String { | |
func hmac(algorithm: CryptoAlgorithm, key: String) -> String { | |
let str = self.cStringUsingEncoding(NSUTF8StringEncoding) | |
let strLen = Int(self.lengthOfBytesUsingEncoding(NSUTF8StringEncoding)) | |
let digestLen = algorithm.digestLength | |
let result = UnsafeMutablePointer<CUnsignedChar>.alloc(digestLen) | |
let keyStr = key.cStringUsingEncoding(NSUTF8StringEncoding) | |
let keyLen = Int(key.lengthOfBytesUsingEncoding(NSUTF8StringEncoding)) | |
CCHmac(algorithm.HMACAlgorithm, keyStr!, keyLen, str!, strLen, result) | |
let digest = stringFromResult(result, length: digestLen) | |
result.dealloc(digestLen) | |
return digest | |
} | |
private func stringFromResult(result: UnsafeMutablePointer<CUnsignedChar>, length: Int) -> String { | |
var hash = NSMutableString() | |
for i in 0..<length { | |
hash.appendFormat("%02x", result[i]) | |
} | |
return String(hash) | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
On Xcode 6.1.1, line 51 should be UnsafeMutablePointer.alloc(digestLen) instead.
Here is a Base64 implementation: https://gist.github.com/joelchen/9a93479d8ad0c68e4f6a