volumekey.c 2.24 KB
Newer Older
1 2 3 4
/*
 * cryptsetup volume key implementation
 *
 * Copyright (C) 2004-2006, Clemens Fruhwirth <clemens@endorphin.org>
Milan Broz's avatar
Milan Broz committed
5
 * Copyright (C) 2010-2018, Red Hat, Inc. All rights reserved.
6 7 8
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
9 10
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
11 12 13 14 15 16 17 18
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
19
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
20 21 22
 */

#include <string.h>
23
#include <stdint.h>
24 25 26 27
#include <stdlib.h>

#include "internal.h"

28
struct volume_key *crypt_alloc_volume_key(size_t keylength, const char *key)
29
{
30 31
	struct volume_key *vk;

32
	if (keylength > (SIZE_MAX - sizeof(*vk)))
33
		return NULL;
34

35
	vk = malloc(sizeof(*vk) + keylength);
36 37 38
	if (!vk)
		return NULL;

39
	vk->key_description = NULL;
40
	vk->keylength = keylength;
41 42 43 44 45 46 47 48

	/* keylength 0 is valid => no key */
	if (vk->keylength) {
		if (key)
			memcpy(&vk->key, key, keylength);
		else
			crypt_memzero(&vk->key, keylength);
	}
49 50 51 52

	return vk;
}

53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68
void crypt_volume_key_set_description(struct volume_key *vk, const char *key_description)
{
	if (vk) {
		free(CONST_CAST(void*)vk->key_description);
		vk->key_description = key_description;
	}
}

const char *crypt_volume_key_get_description(const struct volume_key *vk)
{
	if (!vk)
		return NULL;

	return vk->key_description;
}

69 70 71
void crypt_free_volume_key(struct volume_key *vk)
{
	if (vk) {
72
		crypt_memzero(vk->key, vk->keylength);
73
		vk->keylength = 0;
74
		free(CONST_CAST(void*)vk->key_description);
75 76 77 78
		free(vk);
	}
}

79
struct volume_key *crypt_generate_volume_key(struct crypt_device *cd, size_t keylength)
80 81 82 83 84 85 86 87
{
	int r;
	struct volume_key *vk;

	vk = crypt_alloc_volume_key(keylength, NULL);
	if (!vk)
		return NULL;

88
	r = crypt_random_get(cd, vk->key, keylength, CRYPT_RND_KEY);
89 90 91 92 93 94
	if(r < 0) {
		crypt_free_volume_key(vk);
		return NULL;
	}
	return vk;
}