[patch] UBIFS: Add cryptographic functionality when a key is passed to the compress / decompress functions

Artem Bityutskiy dedekind1 at gmail.com
Thu Apr 12 10:05:43 EDT 2012


I've pushed this patch (actually folded into the previous one), but
massaged it a bit.

Please, verify.

 Please, see below and
take into account for the future submissions.

1. Keep the subject line short - it should give only rough idea what
patch does.

2. When you re-send, add [PATCH v2] prefix.

3. When you re-send, send full patch with the commit message, not just
a reply with new diff. E.g., if I apply this patch using git am I will
get the following:

----
UBIFS: Add cryptographic functionality when a key is passed to the compress / decompress functions

Without the goto:

Signed-off-by: Joel Reardon <reardonj at inf.ethz.ch>
----

Does the commit message make sense? No. I have to manually copy-paste
you commit message from the previous patch...


On Tue, 2012-04-03 at 13:35 +0200, Joel Reardon wrote:
> Without the goto:
> 
> Signed-off-by: Joel Reardon <reardonj at inf.ethz.ch>
> ---
>  fs/ubifs/compress.c |   74 ++++++++++++++++++++++++++++++++++++++++++++++++---
>  fs/ubifs/ubifs.h    |    8 +++++-
>  2 files changed, 77 insertions(+), 5 deletions(-)
> 
> diff --git a/fs/ubifs/compress.c b/fs/ubifs/compress.c
> index b796b8d..61fe584 100644
> --- a/fs/ubifs/compress.c
> +++ b/fs/ubifs/compress.c
> @@ -27,9 +27,12 @@
>   * decompression.
>   */
> 
> -#include <linux/crypto.h>
>  #include "ubifs.h"
> 
> +#include <linux/crypto.h>
> +#include <linux/scatterlist.h>

No need to re-arrange includes unnecessarily. All .c files in UBIFS
first include linux headers and then UBIFS headers. I do not know if
this is good or bad, but at least consistent. If you want to change
this, change globally.

> +
> +

No need to add extra new lines without a need. In general, do not
make unneeded changes, and look at the style of the current code
and try to follow it. If you want to change it - fine, but change for
entire UBIFS.

>  /* Fake description object for the "none" compressor */
>  static struct ubifs_compressor none_compr = {
>  	.compr_type = UBIFS_COMPR_NONE,
> @@ -75,6 +78,53 @@ static struct ubifs_compressor zlib_compr = {
>  struct ubifs_compressor *ubifs_compressors[UBIFS_COMPR_TYPES_CNT];
> 
>  /**
> + * ubifs_aes_crypt - encrypt / decrypt data.
> + * @str: data to crypt
> + * @len: length of the data
> + * @crypto_key: the cryptographic key to use to crypt the data
> + * @iv: the initialization vector to use
> + *
> + * This function applies aes encryption to the data. It is done in counter
> + * mode, which means that encryption and decryption are the same operation,
> + * i.e., it XORs the same generated bitstream, so it can be used both for
> + * encryption / decryption. The operation is done in-place, so str mutates.
> + * Both crypto_key and iv are valid pointers to a buffer of length
> + * UBIFS_CRYPTO_KEYSIZE.
> + */

I believe this was stupid, but I used kernel-doc @ and % prefixes for
argument and costants even in the comments. So to be consistent, use
@str and %UBIFS_CRYPTO_KEYSIZE.

> +int ubifs_aes_crypt(void *str, int len, u8 *crypto_key, u8 *iv)

Surely you can add 'const' qualifier to 'crypto_key' ?
And it can be static - no users outside compress.c.

> +{
> +	struct crypto_blkcipher *tfm;
> +	struct blkcipher_desc desc;
> +	struct scatterlist sg;
> +	int err = 0;

Unneeded initialization.

> +
> +	tfm = crypto_alloc_blkcipher(UBIFS_CRYPTO_ALGORITHM, 0, 0);
> +	if (IS_ERR(tfm)) {
> +		ubifs_err("failed to load transform for aes: %ld",
> +			  PTR_ERR(tfm));
> +		return err;

You return 0 here, which is wrong. Should be "return PTR_ERR(tfm)".

> +	}
> +
> +	err = crypto_blkcipher_setkey(tfm, crypto_key, UBIFS_CRYPTO_KEYSIZE);
> +	desc.tfm = tfm;
> +	desc.flags = 0;
Why you ingected the above 2 lines here? Initialize 'desc' in one place,
do not spread it all over the function.
> +	if (err) {
> +		ubifs_err("crypto_blkcipher_setkey() failed  flags=%#x",
> +			  crypto_blkcipher_get_flags(tfm));

I tried to not include function names in error messages.

> +		return err;
> +	}
> +
> +	memset(&sg, 0, sizeof(struct scatterlist));
> +	sg_set_buf(&sg, str, len);
> +	desc.info = iv;
> +	err = crypto_blkcipher_encrypt(&desc, &sg, &sg, len);
> +	crypto_free_blkcipher(tfm);
> +	if (err)
> +		return err;
> +	return 0;

Can be just 'return err' instead of the 3 above lines.

>  /**
> @@ -148,8 +205,10 @@ no_compr:
>   * This function decompresses data from buffer @in_buf into buffer @out_buf.
>   * The length of the uncompressed data is returned in @out_len. This functions
>   * returns %0 on success or a negative error code on failure.
> + *
> + * WARNING: this function may modify the contents of in_buf when executing.
>   */
> -int ubifs_decompress(const void *in_buf, int in_len, void *out_buf,
> +int ubifs_decompress(void *in_buf, int in_len, void *out_buf,
>  		     int *out_len, int compr_type, u8 *crypto_key)
>  {
>  	int err;
> @@ -167,6 +226,13 @@ int ubifs_decompress(const void *in_buf, int in_len, void *out_buf,
>  		return -EINVAL;
>  	}
> 
> +	if (crypto_key) {
> +		u8 iv[UBIFS_CRYPTO_KEYSIZE];
> +
> +		memset(iv, 0, UBIFS_CRYPTO_KEYSIZE);
> +		ubifs_aes_crypt(in_buf, in_len, crypto_key, iv);
> +	}

So what's the point of defining and initializing 'iv' outside of
'ubifs_aes_crypt()', 2 times? Why not to move it inside?


> +/* 128 bit key size in bytes for UBIFS */
> +#define UBIFS_CRYPTO_KEYSIZE 16
> +/* AES in counter mode is the encryption algorithm. */

Nit-pick: no dot at the end of one-line comment. I've removed it myself,
just keep in mind.

Anyway, too many things. I've modified your patch and pushed, here is
the diff against this patch - please, take a look and take into account.
There are few changes in the comments which I made to make the shorter
as well, not so important.

I've done the following 2 things as well:

1. Move iv to ubifs_aes_crypt()
2. make crypto_key void. In general, if we do not look inside the
pointer, we make it void, because we do not really care about the type,
since we do not look inside.

Please, verify. Below is the resulting patch. Also in the 'joel' branch.

From 8bd3e978c5034f2e4a4017f2453eaa8271487cfa Mon Sep 17 00:00:00 2001
From: Joel Reardon <joel at clambassador.com>
Date: Thu, 12 Apr 2012 16:29:59 +0300
Subject: [PATCH] UBIFS: impliment encryption in the (de)compress functions

This patch adds a function to perform AES encryption. The compress and
decompress routines use this function if they are called with a non-NULL
key parameter. It uses AES counter mode (where encryption and decryption
are the same function) and performs the operation in place on the data. It
uses a default IV of 0, since each key is only evey used to encrypt one
data item the IV does not matter.

The const qualifier was removed from the decompress routine for the
following reason. Encrypted data is not compressable, so compression is
first applied then the result is encrypted. In the reverse, decryption is
first applied and the result decompressed. This means that either the
input buffer for decompression is used to perform an in-place decryption
before decompression, or a third buffer is added and data is copied around.

This was tested by using a static key as the key parameter where both compress
and decompress were called. Data was written and the file system was unmounted
and the mtd dev was scanned with hexdump to make sure that no plaintext data
was available. The drive was remounted and the data successful read back. The
static key was then removed and replaced with NULL, and the same test was done
except that now the data appeared in plaintext on the raw device.

Signed-off-by: Joel Reardon <reardonj at inf.ethz.ch>
Signed-off-by: Artem Bityutskiy <artem.bityutskiy at linux.intel.com>
---
 fs/ubifs/compress.c |   67 ++++++++++++++++++++++++++++++++++++++++++++++++--
 fs/ubifs/file.c     |    5 ++-
 fs/ubifs/journal.c  |    6 ++--
 fs/ubifs/ubifs.h    |   11 ++++++--
 4 files changed, 78 insertions(+), 11 deletions(-)

diff --git a/fs/ubifs/compress.c b/fs/ubifs/compress.c
index 11e4132..7b57457 100644
--- a/fs/ubifs/compress.c
+++ b/fs/ubifs/compress.c
@@ -28,6 +28,7 @@
  */
 
 #include <linux/crypto.h>
+#include <linux/scatterlist.h>
 #include "ubifs.h"
 
 /* Fake description object for the "none" compressor */
@@ -75,6 +76,53 @@ static struct ubifs_compressor zlib_compr = {
 struct ubifs_compressor *ubifs_compressors[UBIFS_COMPR_TYPES_CNT];
 
 /**
+ * aes_crypt - encrypt / decrypt data.
+ * @str: the data to crypt
+ * @len: length of the data
+ * @crypto_key: the cryptographic key to use to crypt the data
+ *
+ * This function applies AES encryption to the data. It is done in counter
+ * mode, which means that encryption and decryption are the same operation,
+ * i.e., it XORs the same generated bitstream, so it can be used both for
+ * encryption / decryption. Returns zero in case of success and a negative
+ * error code in case of failure.
+ *
+ * WARNING: The operation is done in-place, so @str mutates!
+ */
+static int aes_crypt(void *str, int len, const void *crypto_key)
+{
+	struct crypto_blkcipher *tfm;
+	struct blkcipher_desc desc;
+	struct scatterlist sg;
+	uint8_t iv[UBIFS_CRYPTO_KEYSIZE];
+	int err;
+
+	tfm = crypto_alloc_blkcipher(UBIFS_CRYPTO_ALGORITHM, 0, 0);
+	if (IS_ERR(tfm)) {
+		err = PTR_ERR(tfm);
+		ubifs_err("failed to load transform for aes, error %d", err);
+		return err;
+	}
+
+	err = crypto_blkcipher_setkey(tfm, crypto_key, UBIFS_CRYPTO_KEYSIZE);
+	if (err) {
+		ubifs_err("cannot set the AES key, flags %#x, error %d",
+			  crypto_blkcipher_get_flags(tfm), err);
+		return err;
+	}
+
+	memset(&sg, 0, sizeof(struct scatterlist));
+	sg_set_buf(&sg, str, len);
+	memset(iv, 0, UBIFS_CRYPTO_KEYSIZE);
+	desc.info = iv;
+	desc.tfm = tfm;
+	desc.flags = 0;
+	err = crypto_blkcipher_encrypt(&desc, &sg, &sg, len);
+	crypto_free_blkcipher(tfm);
+	return err;
+}
+
+/**
  * ubifs_compress - compress data.
  * @in_buf: data to compress
  * @in_len: length of the data to compress
@@ -82,6 +130,7 @@ struct ubifs_compressor *ubifs_compressors[UBIFS_COMPR_TYPES_CNT];
  * @out_len: output buffer length is returned here
  * @compr_type: type of compression to use on enter, actually used compression
  *              type on exit
+ * @crypto_key: the encryption key or NULL if no encryption needed
  *
  * This function compresses input buffer @in_buf of length @in_len and stores
  * the result in the output buffer @out_buf and the resulting length in
@@ -93,7 +142,7 @@ struct ubifs_compressor *ubifs_compressors[UBIFS_COMPR_TYPES_CNT];
  * buffer and %UBIFS_COMPR_NONE is returned in @compr_type.
  */
 void ubifs_compress(const void *in_buf, int in_len, void *out_buf, int *out_len,
-		    int *compr_type)
+		    int *compr_type, void *crypto_key)
 {
 	int err;
 	struct ubifs_compressor *compr = ubifs_compressors[*compr_type];
@@ -125,12 +174,18 @@ void ubifs_compress(const void *in_buf, int in_len, void *out_buf, int *out_len,
 	if (in_len - *out_len < UBIFS_MIN_COMPRESS_DIFF)
 		goto no_compr;
 
+	if (crypto_key)
+		aes_crypt(out_buf, *out_len, crypto_key);
+
 	return;
 
 no_compr:
 	memcpy(out_buf, in_buf, in_len);
 	*out_len = in_len;
 	*compr_type = UBIFS_COMPR_NONE;
+
+	if (crypto_key)
+		aes_crypt(out_buf, *out_len, crypto_key);
 }
 
 /**
@@ -140,13 +195,16 @@ no_compr:
  * @out_buf: output buffer where decompressed data should
  * @out_len: output length is returned here
  * @compr_type: type of compression
+ * @crypto_key: the encryption key or %NULL if no encryption needed
  *
  * This function decompresses data from buffer @in_buf into buffer @out_buf.
  * The length of the uncompressed data is returned in @out_len. This functions
  * returns %0 on success or a negative error code on failure.
+ *
+ * WARNING: this function may modify the contents of @in_buf!
  */
-int ubifs_decompress(const void *in_buf, int in_len, void *out_buf,
-		     int *out_len, int compr_type)
+int ubifs_decompress(void *in_buf, int in_len, void *out_buf,
+		     int *out_len, int compr_type, void *crypto_key)
 {
 	int err;
 	struct ubifs_compressor *compr;
@@ -163,6 +221,9 @@ int ubifs_decompress(const void *in_buf, int in_len, void *out_buf,
 		return -EINVAL;
 	}
 
+	if (crypto_key)
+		aes_crypt(in_buf, in_len, crypto_key);
+
 	if (compr_type == UBIFS_COMPR_NONE) {
 		memcpy(out_buf, in_buf, in_len);
 		*out_len = in_len;
diff --git a/fs/ubifs/file.c b/fs/ubifs/file.c
index 5c8f6dc..e8fa837 100644
--- a/fs/ubifs/file.c
+++ b/fs/ubifs/file.c
@@ -80,7 +80,7 @@ static int read_block(struct inode *inode, void *addr, unsigned int block,
 	dlen = le32_to_cpu(dn->ch.len) - UBIFS_DATA_NODE_SZ;
 	out_len = UBIFS_BLOCK_SIZE;
 	err = ubifs_decompress(&dn->data, dlen, addr, &out_len,
-			       le16_to_cpu(dn->compr_type));
+			       le16_to_cpu(dn->compr_type), NULL);
 	if (err || len != out_len)
 		goto dump;
 
@@ -649,7 +649,8 @@ static int populate_page(struct ubifs_info *c, struct page *page,
 			dlen = le32_to_cpu(dn->ch.len) - UBIFS_DATA_NODE_SZ;
 			out_len = UBIFS_BLOCK_SIZE;
 			err = ubifs_decompress(&dn->data, dlen, addr, &out_len,
-					       le16_to_cpu(dn->compr_type));
+					       le16_to_cpu(dn->compr_type),
+					       NULL);
 			if (err || len != out_len)
 				goto out_err;
 
diff --git a/fs/ubifs/journal.c b/fs/ubifs/journal.c
index 137415e..57c4d2f 100644
--- a/fs/ubifs/journal.c
+++ b/fs/ubifs/journal.c
@@ -730,7 +730,7 @@ int ubifs_jnl_write_data(struct ubifs_info *c, const struct inode *inode,
 		compr_type = ui->compr_type;
 
 	out_len = dlen - UBIFS_DATA_NODE_SZ;
-	ubifs_compress(buf, len, &data->data, &out_len, &compr_type);
+	ubifs_compress(buf, len, &data->data, &out_len, &compr_type, NULL);
 	ubifs_assert(out_len <= UBIFS_BLOCK_SIZE);
 
 	dlen = UBIFS_DATA_NODE_SZ + out_len;
@@ -1113,11 +1113,11 @@ static int recomp_data_node(struct ubifs_data_node *dn, int *new_len)
 
 	len = le32_to_cpu(dn->ch.len) - UBIFS_DATA_NODE_SZ;
 	compr_type = le16_to_cpu(dn->compr_type);
-	err = ubifs_decompress(&dn->data, len, buf, &out_len, compr_type);
+	err = ubifs_decompress(&dn->data, len, buf, &out_len, compr_type, NULL);
 	if (err)
 		goto out;
 
-	ubifs_compress(buf, *new_len, &dn->data, &out_len, &compr_type);
+	ubifs_compress(buf, *new_len, &dn->data, &out_len, &compr_type, NULL);
 	ubifs_assert(out_len <= UBIFS_BLOCK_SIZE);
 	dn->compr_type = cpu_to_le16(compr_type);
 	dn->size = cpu_to_le32(*new_len);
diff --git a/fs/ubifs/ubifs.h b/fs/ubifs/ubifs.h
index 7c343e1..ce6d8c2 100644
--- a/fs/ubifs/ubifs.h
+++ b/fs/ubifs/ubifs.h
@@ -160,6 +160,11 @@
 /* Maximum number of data nodes to bulk-read */
 #define UBIFS_MAX_BULK_READ 32
 
+/* 128 bit key size in bytes for UBIFS */
+#define UBIFS_CRYPTO_KEYSIZE 16
+/* AES in counter mode is the encryption algorithm */
+#define UBIFS_CRYPTO_ALGORITHM "ctr(aes)"
+
 /*
  * Lockdep classes for UBIFS inode @ui_mutex.
  */
@@ -1772,9 +1777,9 @@ long ubifs_compat_ioctl(struct file *file, unsigned int cmd, unsigned long arg);
 int __init ubifs_compressors_init(void);
 void ubifs_compressors_exit(void);
 void ubifs_compress(const void *in_buf, int in_len, void *out_buf, int *out_len,
-		    int *compr_type);
-int ubifs_decompress(const void *buf, int len, void *out, int *out_len,
-		     int compr_type);
+		    int *compr_type, void *crypto_key);
+int ubifs_decompress(void *buf, int len, void *out, int *out_len,
+		     int compr_type, void *crypto_key);
 
 #include "debug.h"
 #include "misc.h"
-- 
1.7.9.1

-- 
Best Regards,
Artem Bityutskiy
-------------- next part --------------
A non-text attachment was scrubbed...
Name: signature.asc
Type: application/pgp-signature
Size: 836 bytes
Desc: This is a digitally signed message part
URL: <http://lists.infradead.org/pipermail/linux-mtd/attachments/20120412/e53f46a9/attachment.sig>


More information about the linux-mtd mailing list