Change int types.

This commit is contained in:
Milan Broz
2012-06-07 14:29:43 +02:00
parent 0e79728f86
commit 9fb8b816c5
2 changed files with 42 additions and 76 deletions

View File

@@ -26,7 +26,7 @@
#include "verity.h"
#include "internal.h"
static unsigned get_bits_up(unsigned u)
static unsigned get_bits_up(size_t u)
{
unsigned i = 0;
while ((1 << i) < u)
@@ -34,7 +34,7 @@ static unsigned get_bits_up(unsigned u)
return i;
}
static unsigned get_bits_down(unsigned u)
static unsigned get_bits_down(size_t u)
{
unsigned i = 0;
while ((u >> i) > 1)
@@ -42,17 +42,17 @@ static unsigned get_bits_down(unsigned u)
return i;
}
static int verify_zero(struct crypt_device *cd, FILE *wr, unsigned bytes)
static int verify_zero(struct crypt_device *cd, FILE *wr, size_t bytes)
{
unsigned i;
char block[bytes];
size_t i;
if (fread(block, bytes, 1, wr) != 1)
return -EIO;
for (i = 0; i < bytes; i++)
if (block[i]) {
log_err(cd, "spare area is not zeroed at position %lld\n",
(long long)ftello(wr) - bytes);
log_err(cd, "spare area is not zeroed at position %" PRIu64 "\n",
ftello(wr) - bytes);
return -EPERM;
}
return 0;
@@ -85,28 +85,26 @@ out:
}
static int create_or_verify(struct crypt_device *cd, FILE *rd, FILE *wr,
off_t data_block, int data_block_size,
off_t hash_block, int hash_block_size,
off_t data_block, size_t data_block_size,
off_t hash_block, size_t hash_block_size,
off_t blocks, int version,
const char *hash_name, int verify,
char *calculated_digest, unsigned digest_size,
const char *salt, unsigned salt_size)
char *calculated_digest, size_t digest_size,
const char *salt, size_t salt_size)
{
char left_block[hash_block_size];
char data_buffer[data_block_size];
char read_digest[digest_size];
off_t hash_per_block = 1 << get_bits_down(hash_block_size / digest_size);
size_t hash_per_block = 1 << get_bits_down(hash_block_size / digest_size);
size_t digest_size_full = 1 << get_bits_up(digest_size);
off_t blocks_to_write = (blocks + hash_per_block - 1) / hash_per_block;
unsigned i, left_bytes;
off_t digest_size_full = 1 << get_bits_up(digest_size);
int r;
unsigned long long pos_rd = (unsigned long long)data_block * data_block_size;
unsigned long long pos_wr = (unsigned long long)hash_block * hash_block_size;
size_t left_bytes;
int i, r;
if (fseeko(rd, pos_rd, SEEK_SET))
if (fseeko(rd, data_block * data_block_size, SEEK_SET))
return -EIO;
if (wr && fseeko(wr, pos_wr, SEEK_SET))
if (wr && fseeko(wr, hash_block * hash_block_size, SEEK_SET))
return -EIO;
memset(left_block, 0, hash_block_size);
@@ -131,8 +129,8 @@ static int create_or_verify(struct crypt_device *cd, FILE *rd, FILE *wr,
if (fread(read_digest, digest_size, 1, wr) != 1)
return -EIO;
if (memcmp(read_digest, calculated_digest, digest_size)) {
log_err(cd, "verification failed at position %lld\n",
(long long)ftello(rd) - data_block_size);
log_err(cd, "verification failed at position %" PRIu64 "\n",
ftello(rd) - data_block_size);
return -EPERM;
}
} else {
@@ -172,30 +170,29 @@ static int VERITY_create_or_verify_hash(struct crypt_device *cd,
const char *hash_name,
const char *hash_device,
const char *data_device,
int hash_block_size,
int data_block_size,
size_t hash_block_size,
size_t data_block_size,
off_t data_blocks,
long long hash_position,
off_t hash_position,
char *root_hash,
unsigned digest_size,
size_t digest_size,
const char *salt,
unsigned salt_size)
size_t salt_size)
{
static FILE *data_file = NULL;
static FILE *hash_file = NULL, *hash_file_2;
int i, r;
char calculated_digest[digest_size];
FILE *data_file = NULL;
FILE *hash_file = NULL, *hash_file_2;
off_t hash_level_block[VERITY_MAX_LEVELS];
off_t hash_level_size[VERITY_MAX_LEVELS];
off_t data_file_blocks;
off_t data_file_blocks, s;
size_t hash_per_block, hash_per_block_bits;
uint64_t data_device_size;
unsigned long long hash_per_block, hash_per_block_bits;
unsigned levels;
int levels, i, r;
log_dbg("Userspace hash %s %s, data device %s, data blocks %u, hash device %s, offset %u.",
verify ? "verification" : "creation", hash_name, data_device,
(unsigned)data_blocks, hash_device, (unsigned)hash_position);
log_dbg("Userspace hash %s %s, data device %s, data blocks %" PRIu64
", hash_device %s, offset %" PRIu64 ".",
verify ? "verification" : "creation", hash_name,
data_device, data_blocks, hash_device, hash_position);
if (!data_blocks) {
r = device_size(data_device, &data_device_size);
@@ -216,8 +213,7 @@ static int VERITY_create_or_verify_hash(struct crypt_device *cd,
levels = 0;
if (data_file_blocks) {
while (hash_per_block_bits * levels < 64 &&
(unsigned long long)(data_file_blocks - 1) >>
(hash_per_block_bits * levels))
(data_file_blocks - 1) >> (hash_per_block_bits * levels))
levels++;
}
@@ -227,15 +223,14 @@ static int VERITY_create_or_verify_hash(struct crypt_device *cd,
}
for (i = levels - 1; i >= 0; i--) {
off_t s;
hash_level_block[i] = hash_position;
// verity position of block data_file_blocks at level i
s = data_file_blocks >> (i * hash_per_block_bits);
s = (s + hash_per_block - 1) / hash_per_block;
hash_level_size[i] = s;
if (hash_position + s < hash_position ||
(off_t)(hash_position + s) < 0 ||
(off_t)(hash_position + s) != hash_position + s) {
(hash_position + s) < 0 ||
(hash_position + s) != hash_position + s) {
log_err(cd, "hash device offset overflow\n");
return -EINVAL;
}

View File

@@ -19,7 +19,6 @@
/* TODO:
* - init_by_name()
* - unify units / uint64 etc
* - check translations
* - support device without superblock
* - audit alloc errors / error path
@@ -57,13 +56,11 @@ static int version = 1;
static int data_block_size = 4096;
static int hash_block_size = 4096;
static char *data_blocks_string = NULL;
static long long data_blocks = 0;
static uint64_t data_blocks = 0;
static char *hash_start_string = NULL;
static const char *salt_string = NULL;
static unsigned salt_size = 32;
static off_t superblock_position = 0;
static uint64_t hash_start = 0;
static int opt_verbose = 0;
static int opt_debug = 0;
@@ -142,7 +139,7 @@ static int action_dump(void)
if ((r = crypt_init(&cd, hash_device)))
return r;
params.hash_area_offset = superblock_position;
params.hash_area_offset = hash_start;
r = crypt_load(cd, CRYPT_VERITY, &params);
if (!r)
crypt_dump(cd);
@@ -165,7 +162,7 @@ static int action_activate(int verify)
params.flags |= CRYPT_VERITY_CHECK_HASH;
if (use_superblock) {
params.hash_area_offset = superblock_position;
params.hash_area_offset = hash_start;
r = crypt_load(cd, CRYPT_VERITY, &params);
} else {/*
params.hash_name = hash_algorithm;
@@ -213,7 +210,7 @@ static int action_create(void)
if ((r = crypt_init(&cd, hash_device)))
goto out;
params.hash_name = hash_algorithm;
params.hash_name = hash_algorithm ?: "sha256";
params.data_device = data_device;
if (salt_string) {
@@ -228,7 +225,7 @@ static int action_create(void)
params.data_block_size = data_block_size;
params.hash_block_size = hash_block_size;
params.data_size = data_blocks;
params.hash_area_offset = superblock_position;
params.hash_area_offset = hash_start;
params.version = version;
params.flags = CRYPT_VERITY_CREATE_HASH;
if (!use_superblock)
@@ -310,7 +307,6 @@ int main(int argc, const char **argv)
poptContext popt_context;
int r;
char *end;
static long long hash_start = 0;
crypt_set_log_callback(NULL, _log, NULL);
@@ -381,40 +377,15 @@ int main(int argc, const char **argv)
usage(popt_context, EXIT_FAILURE,
_("Invalid hash device offset."),
poptGetInvocationName(popt_context));
hash_start *= 512;
}
if (hash_start < 0 ||
(unsigned long long)hash_start * 512 / 512 != hash_start ||
(off_t)(hash_start * 512) < 0 ||
(off_t)(hash_start * 512) != hash_start * 512)
usage(popt_context, EXIT_FAILURE,
_("Invalid hash device offset."),
poptGetInvocationName(popt_context));
if (use_superblock)
superblock_position = hash_start * 512;
if (salt_string || !use_superblock) {
if (!salt_string || !strcmp(salt_string, "-"))
salt_string = "";
salt_size = strlen(salt_string) / 2;
}
if (data_block_size < 512 || (data_block_size & (data_block_size - 1)) || data_block_size >= 1U << 31)
usage(popt_context, EXIT_FAILURE, _("Invalid data block size."),
poptGetInvocationName(popt_context));
if (hash_block_size < 512 || (hash_block_size & (hash_block_size - 1)) || hash_block_size >= 1U << 31)
usage(popt_context, EXIT_FAILURE, _("Invalid hash block size."),
poptGetInvocationName(popt_context));
if (data_blocks < 0 || (off_t)data_blocks < 0 || (off_t)data_blocks != data_blocks)
usage(popt_context, EXIT_FAILURE, _("Invalid number of data blocks."),
poptGetInvocationName(popt_context));
if (!hash_algorithm)
hash_algorithm = "sha256";
if (opt_debug) {
opt_verbose = 1;
crypt_set_debug_level(-1);