if (key->schedule != NULL)
return 0;
ALLOC(key->schedule, 1);
- if(key->schedule == NULL) {
- krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
- return ENOMEM;
- }
+ if (key->schedule == NULL)
+ return krb5_enomem(context);
ret = krb5_data_alloc(key->schedule, kt->schedule_size);
if(ret) {
free(key->schedule);
size_t i;
*key = _new_derived_key(crypto, 0xff/* KRB5_KU_RFC1510_VARIANT */);
- if(*key == NULL) {
- krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
- return ENOMEM;
- }
+ if (*key == NULL)
+ return krb5_enomem(context);
ret = krb5_copy_keyblock(context, crypto->key.key, &(*key)->key);
if(ret)
return ret;
return KRB5_PROG_ETYPE_NOSUPP;
}
*string = strdup(e->name);
- if(*string == NULL) {
- krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
- return ENOMEM;
- }
+ if (*string == NULL)
+ return krb5_enomem(context);
return 0;
}
block_sz = (sz + et->padsize - 1) &~ (et->padsize - 1); /* pad */
total_sz = block_sz + checksum_sz;
p = calloc(1, total_sz);
- if(p == NULL) {
- krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
- return ENOMEM;
- }
+ if (p == NULL)
+ return krb5_enomem(context);
q = p;
krb5_generate_random_block(q, et->confoundersize); /* XXX */
sz = et->confoundersize + checksum_sz + len;
block_sz = (sz + et->padsize - 1) &~ (et->padsize - 1); /* pad */
p = calloc(1, block_sz);
- if(p == NULL) {
- krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
- return ENOMEM;
- }
+ if (p == NULL)
+ return krb5_enomem(context);
q = p;
krb5_generate_random_block(q, et->confoundersize); /* XXX */
krb5_error_code ret;
tmp = malloc (sz);
- if (tmp == NULL) {
- krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
- return ENOMEM;
- }
+ if (tmp == NULL)
+ return krb5_enomem(context);
p = tmp;
memset (p, 0, cksum_sz);
p += cksum_sz;
}
p = malloc(len);
- if(len != 0 && p == NULL) {
- krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
- return ENOMEM;
- }
+ if (len != 0 && p == NULL)
+ return krb5_enomem(context);
memcpy(p, data, len);
len -= checksum_sz;
result->data = realloc(p, l);
if(result->data == NULL && l != 0) {
free(p);
- krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
- return ENOMEM;
+ return krb5_enomem(context);
}
result->length = l;
return 0;
}
p = malloc(len);
- if(len != 0 && p == NULL) {
- krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
- return ENOMEM;
- }
+ if (len != 0 && p == NULL)
+ return krb5_enomem(context);
memcpy(p, data, len);
ret = _key_schedule(context, &crypto->key);
result->data = realloc(p, l);
if(result->data == NULL && l != 0) {
free(p);
- krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
- return ENOMEM;
+ return krb5_enomem(context);
}
result->length = l;
return 0;
}
p = malloc (len);
- if (p == NULL) {
- krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
- return ENOMEM;
- }
+ if (p == NULL)
+ return krb5_enomem(context);
memcpy(p, data, len);
ret = (*et->encrypt)(context, &crypto->key, p, len, FALSE, usage, ivec);
result->data = realloc(p, sz);
if(result->data == NULL && sz != 0) {
free(p);
- krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
- return ENOMEM;
+ return krb5_enomem(context);
}
result->length = sz;
return 0;
nblocks = (kt->bits + et->blocksize * 8 - 1) / (et->blocksize * 8);
k = malloc(nblocks * et->blocksize);
if(k == NULL) {
- ret = ENOMEM;
- krb5_set_error_message(context, ret, N_("malloc: out of memory", ""));
+ ret = krb5_enomem(context);
goto out;
}
ret = _krb5_n_fold(constant, len, k, et->blocksize);
if (ret) {
- krb5_set_error_message(context, ret, N_("malloc: out of memory", ""));
+ krb5_enomem(context);
goto out;
}
size_t res_len = (kt->bits + 7) / 8;
if(len != 0 && c == NULL) {
- ret = ENOMEM;
- krb5_set_error_message(context, ret, N_("malloc: out of memory", ""));
+ ret = krb5_enomem(context);
goto out;
}
memcpy(c, constant, len);
k = malloc(res_len);
if(res_len != 0 && k == NULL) {
free(c);
- ret = ENOMEM;
- krb5_set_error_message(context, ret, N_("malloc: out of memory", ""));
+ ret = krb5_enomem(context);
goto out;
}
ret = _krb5_n_fold(c, len, k, res_len);
free(c);
if (ret) {
- krb5_set_error_message(context, ret, N_("malloc: out of memory", ""));
+ krb5_enomem(context);
goto out;
}
}
return 0;
}
d = _new_derived_key(crypto, usage);
- if(d == NULL) {
- krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
- return ENOMEM;
- }
+ if (d == NULL)
+ return krb5_enomem(context);
krb5_copy_keyblock(context, crypto->key.key, &d->key);
_krb5_put_int(constant, usage, 5);
_krb5_derive_key(context, crypto->et, d, constant, sizeof(constant));
{
krb5_error_code ret;
ALLOC(*crypto, 1);
- if(*crypto == NULL) {
- krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
- return ENOMEM;
- }
+ if (*crypto == NULL)
+ return krb5_enomem(context);
if(etype == (krb5_enctype)ETYPE_NULL)
etype = key->keytype;
(*crypto)->et = _krb5_find_enctype(etype);
krb5_data_free(&input2);
if (ret)
krb5_data_free(output);
- return 0;
+ return ret;
}
/**
size_t i, keysize;
memset(res, 0, sizeof(*res));
+ krb5_data_zero(&os1);
+ krb5_data_zero(&os2);
ret = krb5_enctype_keysize(context, enctype, &keysize);
if (ret)
}
ret = malloc(n * sizeof(*ret));
- if (ret == NULL && n != 0) {
- krb5_set_error_message(context, ENOMEM, "malloc: out of memory");
- return ENOMEM;
- }
+ if (ret == NULL && n != 0)
+ return krb5_enomem(context);
n = 0;
for (i = _krb5_num_etypes - 1; i >= 0; --i) {
if (_krb5_etypes[i]->keytype->type == keytype
#include "krb5_locl.h"
-static krb5_error_code
-rr13(unsigned char *buf, size_t len)
+static void
+rr13(uint8_t *dst1, uint8_t *dst2, uint8_t *src, size_t len)
{
- unsigned char *tmp;
int bytes = (len + 7) / 8;
int i;
- if(len == 0)
- return 0;
- {
- const int bits = 13 % len;
- const int lbit = len % 8;
-
- tmp = malloc(bytes);
- if (tmp == NULL)
- return ENOMEM;
- memcpy(tmp, buf, bytes);
- if(lbit) {
- /* pad final byte with inital bits */
- tmp[bytes - 1] &= 0xff << (8 - lbit);
- for(i = lbit; i < 8; i += len)
- tmp[bytes - 1] |= buf[0] >> i;
- }
- for(i = 0; i < bytes; i++) {
- int bb;
- int b1, s1, b2, s2;
- /* calculate first bit position of this byte */
- bb = 8 * i - bits;
- while(bb < 0)
- bb += len;
- /* byte offset and shift count */
- b1 = bb / 8;
- s1 = bb % 8;
-
- if(bb + 8 > bytes * 8)
- /* watch for wraparound */
- s2 = (len + 8 - s1) % 8;
- else
- s2 = 8 - s1;
- b2 = (b1 + 1) % bytes;
- buf[i] = (tmp[b1] << s1) | (tmp[b2] >> s2);
- }
- free(tmp);
+ const int bits = 13 % len;
+
+ for (i = 0; i < bytes; i++) {
+ int bb;
+ int b1, s1, b2, s2;
+ /* calculate first bit position of this byte */
+ bb = 8 * i - bits;
+ while(bb < 0)
+ bb += len;
+ /* byte offset and shift count */
+ b1 = bb / 8;
+ s1 = bb % 8;
+
+ if (bb + 8 > bytes * 8)
+ /* watch for wraparound */
+ s2 = (len + 8 - s1) % 8;
+ else
+ s2 = 8 - s1;
+ b2 = (b1 + 1) % bytes;
+ dst1[i] = (src[b1] << s1) | (src[b2] >> s2);
+ dst2[i] = dst1[i];
}
- return 0;
+
+ return;
}
-/* Add `b' to `a', both being one's complement numbers. */
+/*
+ * Add `b' to `a', both being one's complement numbers.
+ * This function assumes that inputs *a, *b are aligned
+ * to 4 bytes.
+ */
static void
-add1(unsigned char *a, unsigned char *b, size_t len)
+add1(uint8_t *a, uint8_t *b, size_t len)
{
int i;
int carry = 0;
- for(i = len - 1; i >= 0; i--){
- int x = a[i] + b[i] + carry;
+ uint32_t x;
+ uint32_t left, right;
+
+ for (i = len - 1; (i+1) % 4; i--) {
+ x = a[i] + b[i] + carry;
carry = x > 0xff;
a[i] = x & 0xff;
}
- for(i = len - 1; carry && i >= 0; i--){
- int x = a[i] + carry;
+
+ for (i = len / 4 - 1; i >= 0; i--) {
+ left = ntohl(((uint32_t *)a)[i]);
+ right = ntohl(((uint32_t *)b)[i]);
+ x = left + right + carry;
+ carry = x < left || x < right;
+ ((uint32_t *)a)[i] = x;
+ }
+
+ for (i = len - 1; (i+1) % 4; i--) {
+ x = a[i] + carry;
carry = x > 0xff;
a[i] = x & 0xff;
}
+
+ for (i = len / 4 - 1; carry && i >= 0; i--) {
+ left = ((uint32_t *)a)[i];
+ x = left + carry;
+ carry = x < left;
+ ((uint32_t *)a)[i] = x;
+ }
+
+ for (i = len / 4 - 1; i >=0; i--)
+ ((uint32_t *)a)[i] = htonl(((uint32_t *)a)[i]);
}
KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
{
/* if len < size we need at most N * len bytes, ie < 2 * size;
if len > size we need at most 2 * len */
- krb5_error_code ret = 0;
size_t maxlen = 2 * max(size, len);
size_t l = 0;
- unsigned char *tmp = malloc(maxlen);
- unsigned char *buf = malloc(len);
+ uint8_t *tmp;
+ uint8_t *tmpbuf;
+ uint8_t *buf1;
+ uint8_t *buf2;
- if (tmp == NULL || buf == NULL) {
- ret = ENOMEM;
- goto out;
- }
+ tmp = malloc(maxlen + 2 * len);
+ if (tmp == NULL)
+ return ENOMEM;
+
+ buf1 = tmp + maxlen;
+ buf2 = tmp + maxlen + len;
- memcpy(buf, str, len);
memset(key, 0, size);
+ memcpy(buf1, str, len);
+ memcpy(tmp, buf1, len);
do {
- memcpy(tmp + l, buf, len);
l += len;
- ret = rr13(buf, len * 8);
- if (ret)
- goto out;
while(l >= size) {
add1(key, tmp, size);
l -= size;
break;
memmove(tmp, tmp + size, l);
}
+ rr13(tmp + l, buf2, buf1, len * 8);
+ tmpbuf = buf1;
+ buf1 = buf2;
+ buf2 = tmpbuf;
} while(l != 0);
-out:
- if (buf) {
- memset(buf, 0, len);
- free(buf);
- }
- if (tmp) {
- memset(tmp, 0, maxlen);
- free(tmp);
- }
- return ret;
+
+ memset(tmp, 0, maxlen + 2 * len);
+ free(tmp);
+ return 0;
}