Array f_mcrypt_enc_get_supported_key_sizes(CObjRef td) {
int count = 0;
int *key_sizes =
mcrypt_enc_get_supported_key_sizes(td.getTyped<MCrypt>()->m_td, &count);
Array ret = Array::Create();
for (int i = 0; i < count; i++) {
ret.append(key_sizes[i]);
}
mcrypt_free(key_sizes);
return ret;
}
Array HHVM_FUNCTION(mcrypt_enc_get_supported_key_sizes, const Resource& td) {
int count = 0;
int *key_sizes =
mcrypt_enc_get_supported_key_sizes(cast<MCrypt>(td)->m_td, &count);
Array ret = Array::Create();
for (int i = 0; i < count; i++) {
ret.append(key_sizes[i]);
}
mcrypt_free(key_sizes);
return ret;
}
/*
* list_key_sizes(td)
*
* Returns a pointer to an array of integer key sizes for the mcrypt
* handle. This returns the actual list, as opposed to
* mcrypt_enc_get_supported_key_sizes which sometimes just returns a
* formula to create the list.
*
* The list is terminated with a zero.
*
*/
int *list_key_sizes(MCRYPT td) {
int *list;
int *key_sizes, n_key_sizes, i;
key_sizes = mcrypt_enc_get_supported_key_sizes(td, &n_key_sizes);
if (!key_sizes && !n_key_sizes) {
int max_size = mcrypt_enc_get_key_size(td);
list = malloc( (max_size + 1) * sizeof(int) );
if (!list) {
fprintf(stderr, "Out of memory\n");
return NULL;
}
for (i = 0; i < max_size; i++) {
list[i] = i;
}
list[max_size] = 0;
}
else {
list = malloc( (n_key_sizes + 1) * sizeof(int) );
if (!list) {
fprintf(stderr, "Out of memory\n");
return NULL;
}
for (i = 0; i < n_key_sizes; i++) {
list[i] = key_sizes[i];
}
list[n_key_sizes] = 0;
free(key_sizes);
}
return list;
}
static Variant php_mcrypt_do_crypt(CStrRef cipher, CStrRef key, CStrRef data,
CStrRef mode, CStrRef iv, bool dencrypt) {
MCRYPT td = mcrypt_module_open((char*)cipher.data(),
(char*)MCG(algorithms_dir).data(),
(char*)mode.data(),
(char*)MCG(modes_dir).data());
if (td == MCRYPT_FAILED) {
raise_warning(MCRYPT_OPEN_MODULE_FAILED);
return false;
}
/* Checking for key-length */
int max_key_length = mcrypt_enc_get_key_size(td);
if (key.size() > max_key_length) {
raise_warning("Size of key is too large for this algorithm");
}
int count;
int *key_length_sizes = mcrypt_enc_get_supported_key_sizes(td, &count);
int use_key_length;
char *key_s = NULL;
if (count == 0 && key_length_sizes == NULL) { // all lengths 1 - k_l_s = OK
use_key_length = key.size();
key_s = (char*)malloc(use_key_length);
memcpy(key_s, key.data(), use_key_length);
} else if (count == 1) { /* only m_k_l = OK */
key_s = (char*)malloc(key_length_sizes[0]);
memset(key_s, 0, key_length_sizes[0]);
memcpy(key_s, key.data(), MIN(key.size(), key_length_sizes[0]));
use_key_length = key_length_sizes[0];
} else { /* dertermine smallest supported key > length of requested key */
use_key_length = max_key_length; /* start with max key length */
for (int i = 0; i < count; i++) {
if (key_length_sizes[i] >= key.size() &&
key_length_sizes[i] < use_key_length) {
use_key_length = key_length_sizes[i];
}
}
key_s = (char*)malloc(use_key_length);
memset(key_s, 0, use_key_length);
memcpy(key_s, key.data(), MIN(key.size(), use_key_length));
}
mcrypt_free(key_length_sizes);
/* Check IV */
char *iv_s = NULL;
int iv_size = mcrypt_enc_get_iv_size(td);
/* IV is required */
if (mcrypt_enc_mode_has_iv(td) == 1) {
if (!iv.empty()) {
if (iv_size != iv.size()) {
raise_warning("The IV parameter must be as long as the blocksize");
} else {
iv_s = (char*)malloc(iv_size + 1);
memcpy(iv_s, iv.data(), iv_size);
}
} else {
raise_warning("Attempt to use an empty IV, which is NOT recommended");
iv_s = (char*)malloc(iv_size + 1);
memset(iv_s, 0, iv_size + 1);
}
}
int block_size;
unsigned long int data_size;
String s;
char *data_s;
/* Check blocksize */
if (mcrypt_enc_is_block_mode(td) == 1) { /* It's a block algorithm */
block_size = mcrypt_enc_get_block_size(td);
data_size = (((data.size() - 1) / block_size) + 1) * block_size;
s = String(data_size, ReserveString);
data_s = (char*)s.mutableSlice().ptr;
memset(data_s, 0, data_size);
memcpy(data_s, data.data(), data.size());
} else { /* It's not a block algorithm */
data_size = data.size();
s = String(data_size, ReserveString);
data_s = (char*)s.mutableSlice().ptr;
memcpy(data_s, data.data(), data.size());
}
if (mcrypt_generic_init(td, key_s, use_key_length, iv_s) < 0) {
raise_warning("Mcrypt initialisation failed");
return false;
}
if (dencrypt) {
mdecrypt_generic(td, data_s, data_size);
} else {
mcrypt_generic(td, data_s, data_size);
}
/* freeing vars */
mcrypt_generic_end(td);
if (key_s != NULL) {
free(key_s);
}
if (iv_s != NULL) {
free(iv_s);
}
return s.setSize(data_size);
}
static int internal_init_mcrypt(MCRYPT td, const void *key, int lenofkey, const void *IV)
{
int *sizes = NULL;
int num_of_sizes, i, ok = 0;
int key_size = mcrypt_enc_get_key_size(td);
if (lenofkey > key_size || lenofkey==0) {
return MCRYPT_KEY_LEN_ERROR; /* error */
}
sizes = mcrypt_enc_get_supported_key_sizes(td, &num_of_sizes);
if (sizes != NULL) {
for (i = 0; i < num_of_sizes; i++) {
if (lenofkey == sizes[i]) {
ok = 1;
break;
}
}
} else { /* sizes==NULL */
if (num_of_sizes == 0
&& lenofkey <= mcrypt_enc_get_key_size(td))
ok = 1;
}
if (ok == 0) { /* not supported key size */
key_size = mcrypt_enc_get_key_size(td);
if (sizes != NULL) {
for (i = 0; i < num_of_sizes; i++) {
if (lenofkey <= sizes[i]) {
key_size = sizes[i];
break;
}
}
} else { /* well every key size is supported! */
key_size = lenofkey;
}
} else {
key_size = lenofkey;
}
free(sizes);
td->keyword_given = mxcalloc(1, mcrypt_enc_get_key_size(td));
if (td->keyword_given==NULL) return MCRYPT_MEMORY_ALLOCATION_ERROR;
memmove(td->keyword_given, key, lenofkey);
i = mcrypt_get_size(td);
td->akey = mxcalloc(1, i);
if (td->akey==NULL) {
free(td->keyword_given);
return MCRYPT_MEMORY_ALLOCATION_ERROR;
}
i = mcrypt_mode_get_size(td);
if (i > 0) {
td->abuf = mxcalloc(1, i);
if (td->abuf==NULL) {
free(td->keyword_given);
free(td->akey);
return MCRYPT_MEMORY_ALLOCATION_ERROR;
}
}
ok = init_mcrypt(td, td->abuf, key, key_size, IV);
if (ok!=0) {
free(td->keyword_given);
free(td->akey);
free(td->abuf);
return MCRYPT_UNKNOWN_ERROR; /* algorithm error */
}
ok = mcrypt_set_key(td,
(void *) td->akey,
(void *) td->keyword_given,
key_size, IV, IV!=NULL ? mcrypt_enc_get_iv_size(td) : 0);
if (ok!=0) {
internal_end_mcrypt(td);
return MCRYPT_UNKNOWN_ERROR; /* algorithm error */
}
return 0;
}
