| /* LibTomCrypt, modular cryptographic library -- Tom St Denis |
| * |
| * LibTomCrypt is a library that provides various cryptographic |
| * algorithms in a highly modular and flexible manner. |
| * |
| * The library is free for all purposes without any express |
| * guarantee it works. |
| * |
| * Tom St Denis, tomstdenis@iahu.ca, http://libtomcrypt.org |
| */ |
| #include "mycrypt.h" |
| |
| /* The Mask Generation Function (MGF1) for PKCS #1 -- Tom St Denis */ |
| |
| #ifdef PKCS_1 |
| |
| int pkcs_1_mgf1(const unsigned char *seed, unsigned long seedlen, |
| int hash_idx, |
| unsigned char *mask, unsigned long masklen) |
| { |
| unsigned long hLen, counter, x; |
| int err; |
| hash_state *md; |
| unsigned char *buf; |
| |
| _ARGCHK(seed != NULL); |
| _ARGCHK(mask != NULL); |
| |
| /* ensure valid hash */ |
| if ((err = hash_is_valid(hash_idx)) != CRYPT_OK) { |
| return err; |
| } |
| |
| /* get hash output size */ |
| hLen = hash_descriptor[hash_idx].hashsize; |
| |
| /* allocate memory */ |
| md = XMALLOC(sizeof(hash_state)); |
| buf = XMALLOC(hLen); |
| if (md == NULL || buf == NULL) { |
| if (md != NULL) { |
| XFREE(md); |
| } |
| if (buf != NULL) { |
| XFREE(buf); |
| } |
| return CRYPT_MEM; |
| } |
| |
| /* start counter */ |
| counter = 0; |
| |
| while (masklen > 0) { |
| /* handle counter */ |
| STORE32H(counter, buf); |
| ++counter; |
| |
| /* get hash of seed || counter */ |
| if ((err = hash_descriptor[hash_idx].init(md)) != CRYPT_OK) { |
| goto __ERR; |
| } |
| if ((err = hash_descriptor[hash_idx].process(md, seed, seedlen)) != CRYPT_OK) { |
| goto __ERR; |
| } |
| if ((err = hash_descriptor[hash_idx].process(md, buf, 4)) != CRYPT_OK) { |
| goto __ERR; |
| } |
| if ((err = hash_descriptor[hash_idx].done(md, buf)) != CRYPT_OK) { |
| goto __ERR; |
| } |
| |
| /* store it */ |
| for (x = 0; x < hLen && masklen > 0; x++, masklen--) { |
| *mask++ = buf[x]; |
| } |
| } |
| |
| err = CRYPT_OK; |
| __ERR: |
| #ifdef CLEAN_STACK |
| zeromem(buf, hLen); |
| zeromem(md, sizeof(hash_state)); |
| #endif |
| |
| XFREE(buf); |
| XFREE(md); |
| |
| return err; |
| } |
| |
| #endif /* PKCS_1 */ |