84 lines
2.1 KiB
C
84 lines
2.1 KiB
C
#include "mycrypt.h"
|
|
|
|
#ifdef CTR
|
|
|
|
int ctr_start(int cipher, const unsigned char *count, const unsigned char *key, int keylen,
|
|
int num_rounds, symmetric_CTR *ctr)
|
|
{
|
|
int x, err;
|
|
|
|
_ARGCHK(count != NULL);
|
|
_ARGCHK(key != NULL);
|
|
_ARGCHK(ctr != NULL);
|
|
|
|
/* bad param? */
|
|
if ((err = cipher_is_valid(cipher)) != CRYPT_OK) {
|
|
return err;
|
|
}
|
|
|
|
/* setup cipher */
|
|
if ((err = cipher_descriptor[cipher].setup(key, keylen, num_rounds, &ctr->key)) != CRYPT_OK) {
|
|
return err;
|
|
}
|
|
|
|
/* copy ctr */
|
|
ctr->blocklen = cipher_descriptor[cipher].block_length;
|
|
ctr->cipher = cipher;
|
|
ctr->padlen = 0;
|
|
for (x = 0; x < ctr->blocklen; x++) {
|
|
ctr->ctr[x] = count[x];
|
|
}
|
|
cipher_descriptor[ctr->cipher].ecb_encrypt(ctr->ctr, ctr->pad, &ctr->key);
|
|
return CRYPT_OK;
|
|
}
|
|
|
|
int ctr_encrypt(const unsigned char *pt, unsigned char *ct, unsigned long len, symmetric_CTR *ctr)
|
|
{
|
|
int x, err;
|
|
|
|
_ARGCHK(pt != NULL);
|
|
_ARGCHK(ct != NULL);
|
|
_ARGCHK(ctr != NULL);
|
|
|
|
if ((err = cipher_is_valid(ctr->cipher)) != CRYPT_OK) {
|
|
return err;
|
|
}
|
|
|
|
/* is blocklen/padlen valid? */
|
|
if (ctr->blocklen < 0 || ctr->blocklen > (int)sizeof(ctr->ctr) ||
|
|
ctr->padlen < 0 || ctr->padlen > (int)sizeof(ctr->pad)) {
|
|
return CRYPT_INVALID_ARG;
|
|
}
|
|
|
|
while (len-- > 0) {
|
|
/* is the pad empty? */
|
|
if (ctr->padlen == ctr->blocklen) {
|
|
/* increment counter */
|
|
for (x = 0; x < ctr->blocklen; x++) {
|
|
ctr->ctr[x] = (ctr->ctr[x] + (unsigned char)1) & (unsigned char)255;
|
|
if (ctr->ctr[x] != (unsigned char)0) {
|
|
break;
|
|
}
|
|
}
|
|
|
|
/* encrypt it */
|
|
cipher_descriptor[ctr->cipher].ecb_encrypt(ctr->ctr, ctr->pad, &ctr->key);
|
|
ctr->padlen = 0;
|
|
}
|
|
*ct++ = *pt++ ^ ctr->pad[ctr->padlen++];
|
|
}
|
|
return CRYPT_OK;
|
|
}
|
|
|
|
int ctr_decrypt(const unsigned char *ct, unsigned char *pt, unsigned long len, symmetric_CTR *ctr)
|
|
{
|
|
_ARGCHK(pt != NULL);
|
|
_ARGCHK(ct != NULL);
|
|
_ARGCHK(ctr != NULL);
|
|
|
|
return ctr_encrypt(ct, pt, len, ctr);
|
|
}
|
|
|
|
#endif
|
|
|