DEV Community

Matheus Cardoso
Matheus Cardoso

Posted on • Edited on • Originally published at Medium

CryptoKit Basics: End-to-End Encryption

Security can be a major concern for companies and developers building applications, especially in the medical field where data breaches are severely penalized. This is where end-to-end encryption comes into play.

In this guide, you'll learn how to implement a basic end-to-end encryption flow in your Swift application for iOS, macOS, other Apple things, and even Linux, using Apple's own CryptoKit framework.

What is CryptoKit?

CryptoKit is a new Swift framework that makes it easier and safer than ever to perform cryptographic operations, whether you simply need to compute a hash or are implementing a more advanced authentication protocol.

What is end-to-end encryption?

End-to-end encryption is a system of communication where the only people who can read the messages are the people communicating. No eavesdropper can access the cryptographic keys needed to decrypt the conversation—not even a company that runs the messaging service.

What you need

CryptoKit is available on the following platforms:

  • iOS 13.0+
  • macOS 10.15+
  • Mac Catalyst 13.0+
  • tvOS 13.0+
  • watchOS 6.0+
  • Linux (as Swift Crypto)

Step 1: Generating key pairs

Cryptographic key pairs are central to end-to-end encryption: A public key is what you use to encrypt data for someone, and a private key is what you use to decrypt data that was encrypted for you. Each user in your application should have a key pair, with their public key available
in a trusted service for other users to fetch, and their private keys stored securely on their device. The trusted service can be written in Swift using Vapor, or another language and framework of your choice.

We will first generate a private key, then extract the associated public key from it, which will be sent to the trusted service. In this guide we'll use the Curve25519 algorithms, but the others should work similarly.

import CryptoKit

// generate key pair
let privateKey = Curve25519.KeyAgreement.PrivateKey()
let publicKey = privateKey.publicKey

// publish public key in trusted service
TrustedService.publishKey(publicKey, for: myIdentity)
Enter fullscreen mode Exit fullscreen mode

The public key has a var rawRepresentation: Data property which can be used to serialize it into the payload for the trusted service.

Step 2: Encrypting data

To encrypt data for a user (recipient), first you need to fetch their public key from the trusted service.

let recipientPublicKey = TrustedService.fetchPublicKey(of: recipientIdentity)
Enter fullscreen mode Exit fullscreen mode

The initializer Curve25519.KeyAgreement.PublicKey(rawRepresentation: Data) can be used to deserialize the public key coming from the trusted service.

Step 2.1: Deriving a symmetric key

Public keys can't be used to encrypt data directly. They're used by the two parties communicating to agree on a symmetric key for encryption, via a Diffie-Hellmann key agreement. To do this, we will use the sender's private key and the recipient's public key to generate a shared secret, from which we can derive the symmetric key using the HKDF key derivation function.

let sharedSecret = try! privateKey.sharedSecretFromKeyAgreement(with: recipientPublicKey)
let symmetricKey = sharedSecret.hkdfDerivedSymmetricKey(using: SHA256.self,
                                                        salt: protocolSalt,
                                                        sharedInfo: Data(),
                                                        outputByteCount: 32)
Enter fullscreen mode Exit fullscreen mode

The protocol salt is a value that alters the outcome of the symmetric key derivation. Choose one that will remain constant for your use case, for example: "My Key Agreement Salt".data(using: .utf8)!

Step 2.2: Encrypting the data with the symmetric key

Now, we can finally use that symmetric key to perform the encryption. This job can be done by one of the ciphers CryptoKit supports. In this guide, we'll use ChaChaPoly, which can be three times faster than AES in mobile devices, according to Adam Langley and other researchers.

let sensitiveMessage = "The result of your test is positive".data(using: .utf8)!
let encryptedData = try! ChaChaPoly.seal(sensitiveMessage, using: symmetricKey).combined
Enter fullscreen mode Exit fullscreen mode

The encryptedData can now be safely sent to our recipient.

Step 3: Decrypting data

To decrypt the data received, the recipient will need to derive the same symmetric key used to encrypt the data. But before it can be derived, we need the sender's public key.

let senderPublicKey = TrustedService.fetchPublicKey(of: senderIdentity)
Enter fullscreen mode Exit fullscreen mode

Step 3.1: Deriving the symmetric key

To derive the symmetric key, we will perform the same process we performed in step 2.1, except now we will use the recipient's private key and the sender's public key. This will allows us to regenerate the shared secret, which can be used for the derivation of the same symmetric key.

let sharedSecret = try! privateKey.sharedSecretFromKeyAgreement(with: senderPublicKey)
let symmetricKey = sharedSecret.hkdfDerivedSymmetricKey(using: SHA256.self,
                                                        salt: protocolSalt,
                                                        sharedInfo: Data(),
                                                        outputByteCount: 32)
Enter fullscreen mode Exit fullscreen mode

We just shared a symmetric key between users without it ever existing outside their devices! 🤯

Step 3.2: Decrypting the data with the symmetric key

Now we can use the symmetric key to decrypt the data.

let sealedBox = try! ChaChaPoly.SealedBox(combined: encryptedData)
let decryptedData = try! ChaChaPoly.open(sealedBox, using: symmetricKey)

let sensitiveMessage = String(data: decryptedData, encoding: .utf8)
print(sensitiveMessage) // "The result of your test is positive"
Enter fullscreen mode Exit fullscreen mode

End-to-end encryption achieved!

Stay tuned

The process described in this guide guarantees one thing: Encryption. This means the data encrypted for a user can only be decrypted by that user.

Not guaranteed: Authentication and Integrity. This means that you cannot know for sure that the encrypted data came from someone in particular and that it was not modified in transit. As such, you are vulnerable to some forms of man-in-the-middle attacks.

Also not guaranteed: Forward Secrecy. This means if a private key is compromised, all data encrypted for the key's owner can be decrypted until they start using a new key. Schemes that provide Forward Secrecy have single-use keys, which, if stolen or cracked, will only compromise a subset of the data.

To guarantee Authentication and Integrity, we'll look into CryptoKit's capabilities of creating and verifying signatures in a future article. Stay tuned!

Extras

Swift Playground

I uploaded a Swift Playground to GitHub that follows roughly the same flow described in this article, minus the TrustedService. Here's a link to it. Make sure to give it a 🌟.

Storing private keys securely on the device

Apple recommends using the device's keychain. It's explained in detail in this guide.

What is Swift Crypto?

Swift Crypto is an open-source implementation of a substantial portion of the API of Apple CryptoKit suitable for use on Linux platforms. It enables cross-platform or server applications with the advantages of CryptoKit.

Found any mistakes?

Though I have some industry experience with end-to-end encryption, I don't call myself an expert. Please do reach out about it at matheus@cardo.so or DM me via twitter at @cardosodev.

Disclaimer: This article, or series of articles, is strictly educational and I don't recommend using it without professional consultation.

Top comments (2)

Collapse
 
harishios profile image
harishios

Hi,

Great Tutorial. Can I use CryptoKit framework in Objective C?

I tried to import but I couldn't.

Collapse
 
tallinn1960 profile image
Olaf Schlüter

No, you can't just by importing it (what did you try to #import at all as there isn't any .h for CryptoKit?). CryptoKit is Swift-only which may be a writing on the wall to abandon Objective-C for any new project.

What you can do is to write a Swift Wrapper class:

import Foundation
import CryptoKit

@objc class Crypto: NSObject  {    
    @objc static func SHA256hash(data: Data) -> Data {
        let digest = SHA256.hash(data: data)
        return Data(digest);
    }
}
Enter fullscreen mode Exit fullscreen mode

and call it from Objective C by:

NSData *nsdata = [Crypto SHA256hashWithData: input]; // input is of type NSData *
Enter fullscreen mode Exit fullscreen mode

Note, that to make any Swift class and its methods within your project annotated with @objc visible in Objective-C you need to do

#import "<project-name>-Swift.h"
Enter fullscreen mode Exit fullscreen mode

with <project-name> being the name of your Xcode project. That header file is auto-generated by Xcode. You won't see it anywhere but it is there.

It is fortunate that all datatypes used by CryptoKit follow the DataProtocol and are as such convertible to and from Data. You may use the Data/NSData combo as the universal data exchange type. Most important aside from data being processed are public and private keys.