#include "os.h"
#include <libsec.h>

/* rfc2898 */
void
pbkdf2_x(p, plen, s, slen, rounds, d, dlen, x, xlen)
       uchar *p, *s, *d;
       ulong plen, slen, dlen, rounds;
       DigestState* (*x)(uchar*, ulong, uchar*, ulong, uchar*, DigestState*);
       int xlen;
{
       uchar block[256], tmp[256];
       ulong i, j, k, n;
       DigestState *ds;

       assert(xlen <= sizeof(tmp));

       for(i = 1; dlen > 0; i++, d += n, dlen -= n){
               tmp[3] = i;
               tmp[2] = i >> 8;
               tmp[1] = i >> 16;
               tmp[0] = i >> 24;
               ds = (*x)(s, slen, p, plen, nil, nil);
               (*x)(tmp, 4, p, plen, block, ds);
               memmove(tmp, block, xlen);
               for(j = 1; j < rounds; j++){
                       (*x)(tmp, xlen, p, plen, tmp, nil);
                       for(k=0; k<xlen; k++)
                               block[k] ^= tmp[k];
               }
               n = dlen > xlen ? xlen : dlen;
               memmove(d, block, n);
       }
}