#include <iostream>
#include <fstream>
#include <sys/stat.h>
#include <openssl/rsa.h>
#include <openssl/pem.h>
extern "C" {
#include "openssh/digest.h"
#include "openssh/key.h"
}
#include <givaro/givintrsa.h>
#include <givaro/givtimer.h>
template<class RandIter=GivRandom>
struct Givaro_keygen {
Integer& iqmp,
long size,
unsigned long seed) {
IRD.keys_gen(gen, (size>>1)+1, (size>>1)-1, n, e, d, p, q);
if ( IRD.islt(d,IRD.
zero) ) IRD.
addin(d,phim);
if ( IRD.islt(iqmp,IRD.
zero) ) IRD.
addin(iqmp,p);
}
};
BIGNUM* Integer2BN(BIGNUM * n,
const Integer& a) {
std::string str(a);
BN_dec2bn(&n,str.c_str());
return n;
}
int mymain(FILE* fileout, FILE* filepub, long s, unsigned long seed) {
RSA *rsa= new RSA();
Integer in, ie, id, ip, iq, idmp1, idmq1, iiqmp;
Givaro_keygen<>()(in,ie,id,ip,iq,idmp1,idmq1,iiqmp, s, seed);
std::cerr << tim << std::endl;
rsa->n = BN_new(); Integer2BN(rsa->n, in);
rsa->e = BN_new(); Integer2BN(rsa->e, ie);
rsa->d = BN_new(); Integer2BN(rsa->d, id);
rsa->p = BN_new(); Integer2BN(rsa->p, ip);
rsa->q = BN_new(); Integer2BN(rsa->q, iq);
rsa->dmp1 = BN_new(); Integer2BN(rsa->dmp1, idmp1);
rsa->dmq1 = BN_new(); Integer2BN(rsa->dmq1, idmq1);
rsa->iqmp = BN_new(); Integer2BN(rsa->iqmp, iiqmp);
rsakey.type=KEY_RSA;
rsakey.rsa = rsa;
std::cerr << "key's randomart: \n" << sshkey_fingerprint(&rsakey, SSH_FP_HASH_DEFAULT, SSH_FP_RANDOMART) << std::endl;
PEM_write_RSAPrivateKey(fileout,rsa,NULL,NULL,0,NULL,NULL);
key_write(&rsakey, filepub);
fprintf(filepub," givaro\n");
return 0;
}
unsigned long seedfromfile(char * filename) {
std::ifstream filrand(filename);
unsigned long seed=0;
for(unsigned int i=0; i<sizeof(unsigned long); ++i) {
unsigned char t; filrand >> t;
seed <<= 8;
seed |= t;
}
std::cerr << "Generated seed: " << seed << ", using " << filename << std::endl;
return seed;
}
void usage() {
std::cerr << "Usage: givaro-ssh-keygen [-b bits] [-f private-key-file] [-p public-key-file] [-r randomness-file]" << std::endl;
}
int main(int argc, char** argv)
{
if (argc > 10) {
usage(); return 0;
}
long s = 4096;
unsigned long seed = 0;
long files = 0;
std::string filprivname, filpubname;
for (long i = 1; i < argc; i++) {
if (argv[i][0] == '-') {
switch(argv[i][1]) {
case 'h':; case 'H': {
usage(); return 0;
}
case 'b':; case 'B': {
s = atoi(argv[++i]);
break;
}
case 'f':; case 'F': {
filprivname = std::string(argv[++i]);
++files;
break;
}
case 'p':; case 'P': {
filpubname = std::string(argv[++i]);
++files;
break;
}
case 'r':; case 'R': {
seed = seedfromfile(argv[++i]);
break;
}
}
}
}
if (files > 1) {
FILE * filpriv;
filpriv = fopen(filprivname.c_str(),"w");
if (argc>3) {
FILE * filpub ;
filpub = fopen(filpubname.c_str(),"w");
mymain(filpriv,filpub,s,seed);
fclose(filpub);
} else
mymain(filpriv,stdout,s,seed);
fclose(filpriv);
chmod(filprivname.c_str(),(S_IRUSR|S_IWUSR));
} else
mymain(stdout,stdout,s,seed);
return 0;
}