Mercurial > dropbear
annotate demos/encrypt.c @ 230:809b681a9af5 libtomcrypt
- check for generic AES/TWOFISH settings from Dropbear, not AES128 etc
| author | Matt Johnston <matt@ucc.asn.au> |
|---|---|
| date | Tue, 30 Aug 2005 17:40:30 +0000 |
| parents | 39d5d58461d6 |
| children |
| rev | line source |
|---|---|
| 3 | 1 /* encrypt V1.1 Fri Oct 18 04:28:03 NZDT 2002 */ |
| 2 /* File de/encryption, using libtomcrypt */ | |
| 3 /* Written by Daniel Richards <[email protected]> */ | |
| 4 /* Help from Tom St Denis with various bits */ | |
| 5 /* This code is public domain, no rights reserved. */ | |
| 6 /* Encrypts by default, -d flag enables decryption */ | |
| 7 /* ie: ./encrypt blowfish story.txt story.ct */ | |
| 8 /* ./encrypt -d blowfish story.ct story.pt */ | |
| 9 | |
|
191
1c15b283127b
Import of libtomcrypt 1.02 with manual path rename rearrangement etc
Matt Johnston <matt@ucc.asn.au>
parents:
3
diff
changeset
|
10 #include <tomcrypt.h> |
| 3 | 11 |
| 12 int errno; | |
| 13 | |
| 14 int usage(char *name) | |
| 15 { | |
| 16 int x; | |
| 17 | |
| 18 printf("Usage: %s [-d](ecrypt) cipher infile outfile\nCiphers:\n", name); | |
| 19 for (x = 0; cipher_descriptor[x].name != NULL; x++) { | |
| 20 printf("%s\n",cipher_descriptor[x].name); | |
| 21 } | |
| 22 exit(1); | |
| 23 } | |
| 24 | |
| 25 void register_algs(void) | |
| 26 { | |
| 27 int x; | |
| 28 | |
| 29 #ifdef RIJNDAEL | |
| 30 register_cipher (&aes_desc); | |
| 31 #endif | |
| 32 #ifdef BLOWFISH | |
| 33 register_cipher (&blowfish_desc); | |
| 34 #endif | |
| 35 #ifdef XTEA | |
| 36 register_cipher (&xtea_desc); | |
| 37 #endif | |
| 38 #ifdef RC5 | |
| 39 register_cipher (&rc5_desc); | |
| 40 #endif | |
| 41 #ifdef RC6 | |
| 42 register_cipher (&rc6_desc); | |
| 43 #endif | |
| 44 #ifdef SAFERP | |
| 45 register_cipher (&saferp_desc); | |
| 46 #endif | |
| 47 #ifdef TWOFISH | |
| 48 register_cipher (&twofish_desc); | |
| 49 #endif | |
| 50 #ifdef SAFER | |
| 51 register_cipher (&safer_k64_desc); | |
| 52 register_cipher (&safer_sk64_desc); | |
| 53 register_cipher (&safer_k128_desc); | |
| 54 register_cipher (&safer_sk128_desc); | |
| 55 #endif | |
| 56 #ifdef RC2 | |
| 57 register_cipher (&rc2_desc); | |
| 58 #endif | |
| 59 #ifdef DES | |
| 60 register_cipher (&des_desc); | |
| 61 register_cipher (&des3_desc); | |
| 62 #endif | |
| 63 #ifdef CAST5 | |
| 64 register_cipher (&cast5_desc); | |
| 65 #endif | |
| 66 #ifdef NOEKEON | |
| 67 register_cipher (&noekeon_desc); | |
| 68 #endif | |
| 69 #ifdef SKIPJACK | |
| 70 register_cipher (&skipjack_desc); | |
| 71 #endif | |
|
191
1c15b283127b
Import of libtomcrypt 1.02 with manual path rename rearrangement etc
Matt Johnston <matt@ucc.asn.au>
parents:
3
diff
changeset
|
72 #ifdef KHAZAD |
|
1c15b283127b
Import of libtomcrypt 1.02 with manual path rename rearrangement etc
Matt Johnston <matt@ucc.asn.au>
parents:
3
diff
changeset
|
73 register_cipher (&khazad_desc); |
|
1c15b283127b
Import of libtomcrypt 1.02 with manual path rename rearrangement etc
Matt Johnston <matt@ucc.asn.au>
parents:
3
diff
changeset
|
74 #endif |
|
1c15b283127b
Import of libtomcrypt 1.02 with manual path rename rearrangement etc
Matt Johnston <matt@ucc.asn.au>
parents:
3
diff
changeset
|
75 #ifdef ANUBIS |
|
1c15b283127b
Import of libtomcrypt 1.02 with manual path rename rearrangement etc
Matt Johnston <matt@ucc.asn.au>
parents:
3
diff
changeset
|
76 register_cipher (&anubis_desc); |
|
1c15b283127b
Import of libtomcrypt 1.02 with manual path rename rearrangement etc
Matt Johnston <matt@ucc.asn.au>
parents:
3
diff
changeset
|
77 #endif |
| 3 | 78 |
| 79 if (register_hash(&sha256_desc) == -1) { | |
| 80 printf("Error registering SHA256\n"); | |
| 81 exit(-1); | |
| 82 } | |
| 83 | |
| 84 if (register_prng(&yarrow_desc) == -1) { | |
| 85 printf("Error registering yarrow PRNG\n"); | |
| 86 exit(-1); | |
| 87 } | |
| 88 | |
| 89 if (register_prng(&sprng_desc) == -1) { | |
| 90 printf("Error registering sprng PRNG\n"); | |
| 91 exit(-1); | |
| 92 } | |
| 93 } | |
| 94 | |
| 95 int main(int argc, char *argv[]) | |
| 96 { | |
| 97 unsigned char plaintext[512],ciphertext[512]; | |
| 98 unsigned char tmpkey[512], key[MAXBLOCKSIZE], IV[MAXBLOCKSIZE]; | |
| 99 unsigned char inbuf[512]; /* i/o block size */ | |
| 100 unsigned long outlen, y, ivsize, x, decrypt; | |
| 101 symmetric_CTR ctr; | |
| 102 int cipher_idx, hash_idx, ks; | |
| 103 char *infile, *outfile, *cipher; | |
| 104 prng_state prng; | |
| 105 FILE *fdin, *fdout; | |
| 106 | |
| 107 /* register algs, so they can be printed */ | |
| 108 register_algs(); | |
| 109 | |
| 110 if (argc < 4) { | |
| 111 return usage(argv[0]); | |
| 112 } | |
| 113 | |
| 114 if (!strcmp(argv[1], "-d")) { | |
| 115 decrypt = 1; | |
| 116 cipher = argv[2]; | |
| 117 infile = argv[3]; | |
| 118 outfile = argv[4]; | |
| 119 } else { | |
| 120 decrypt = 0; | |
| 121 cipher = argv[1]; | |
| 122 infile = argv[2]; | |
| 123 outfile = argv[3]; | |
| 124 } | |
| 125 | |
| 126 /* file handles setup */ | |
| 127 fdin = fopen(infile,"rb"); | |
| 128 if (fdin == NULL) { | |
| 129 perror("Can't open input for reading"); | |
| 130 exit(-1); | |
| 131 } | |
| 132 | |
| 133 fdout = fopen(outfile,"wb"); | |
| 134 if (fdout == NULL) { | |
| 135 perror("Can't open output for writing"); | |
| 136 exit(-1); | |
| 137 } | |
| 138 | |
| 139 cipher_idx = find_cipher(cipher); | |
| 140 if (cipher_idx == -1) { | |
| 141 printf("Invalid cipher entered on command line.\n"); | |
| 142 exit(-1); | |
| 143 } | |
| 144 | |
| 145 hash_idx = find_hash("sha256"); | |
| 146 if (hash_idx == -1) { | |
| 147 printf("SHA256 not found...?\n"); | |
| 148 exit(-1); | |
| 149 } | |
| 150 | |
| 151 ivsize = cipher_descriptor[cipher_idx].block_length; | |
| 152 ks = hash_descriptor[hash_idx].hashsize; | |
| 153 if (cipher_descriptor[cipher_idx].keysize(&ks) != CRYPT_OK) { | |
| 154 printf("Invalid keysize???\n"); | |
| 155 exit(-1); | |
| 156 } | |
| 157 | |
| 158 printf("\nEnter key: "); | |
| 159 fgets((char *)tmpkey,sizeof(tmpkey), stdin); | |
| 160 outlen = sizeof(key); | |
| 161 if ((errno = hash_memory(hash_idx,tmpkey,strlen((char *)tmpkey),key,&outlen)) != CRYPT_OK) { | |
| 162 printf("Error hashing key: %s\n", error_to_string(errno)); | |
| 163 exit(-1); | |
| 164 } | |
| 165 | |
| 166 if (decrypt) { | |
| 167 /* Need to read in IV */ | |
| 168 if (fread(IV,1,ivsize,fdin) != ivsize) { | |
| 169 printf("Error reading IV from input.\n"); | |
| 170 exit(-1); | |
| 171 } | |
| 172 | |
| 173 if ((errno = ctr_start(cipher_idx,IV,key,ks,0,&ctr)) != CRYPT_OK) { | |
| 174 printf("ctr_start error: %s\n",error_to_string(errno)); | |
| 175 exit(-1); | |
| 176 } | |
| 177 | |
| 178 /* IV done */ | |
| 179 do { | |
| 180 y = fread(inbuf,1,sizeof(inbuf),fdin); | |
| 181 | |
| 182 if ((errno = ctr_decrypt(inbuf,plaintext,y,&ctr)) != CRYPT_OK) { | |
| 183 printf("ctr_decrypt error: %s\n", error_to_string(errno)); | |
| 184 exit(-1); | |
| 185 } | |
| 186 | |
| 187 if (fwrite(plaintext,1,y,fdout) != y) { | |
| 188 printf("Error writing to file.\n"); | |
| 189 exit(-1); | |
| 190 } | |
| 191 } while (y == sizeof(inbuf)); | |
| 192 fclose(fdin); | |
| 193 fclose(fdout); | |
| 194 | |
| 195 } else { /* encrypt */ | |
| 196 /* Setup yarrow for random bytes for IV */ | |
| 197 | |
| 198 if ((errno = rng_make_prng(128, find_prng("yarrow"), &prng, NULL)) != CRYPT_OK) { | |
| 199 printf("Error setting up PRNG, %s\n", error_to_string(errno)); | |
| 200 } | |
| 201 | |
| 202 /* You can use rng_get_bytes on platforms that support it */ | |
| 203 /* x = rng_get_bytes(IV,ivsize,NULL);*/ | |
| 204 x = yarrow_read(IV,ivsize,&prng); | |
| 205 if (x != ivsize) { | |
| 206 printf("Error reading PRNG for IV required.\n"); | |
| 207 exit(-1); | |
| 208 } | |
| 209 | |
| 210 if (fwrite(IV,1,ivsize,fdout) != ivsize) { | |
| 211 printf("Error writing IV to output.\n"); | |
| 212 exit(-1); | |
| 213 } | |
| 214 | |
| 215 if ((errno = ctr_start(cipher_idx,IV,key,ks,0,&ctr)) != CRYPT_OK) { | |
| 216 printf("ctr_start error: %s\n",error_to_string(errno)); | |
| 217 exit(-1); | |
| 218 } | |
| 219 | |
| 220 do { | |
| 221 y = fread(inbuf,1,sizeof(inbuf),fdin); | |
| 222 | |
| 223 if ((errno = ctr_encrypt(inbuf,ciphertext,y,&ctr)) != CRYPT_OK) { | |
| 224 printf("ctr_encrypt error: %s\n", error_to_string(errno)); | |
| 225 exit(-1); | |
| 226 } | |
| 227 | |
| 228 if (fwrite(ciphertext,1,y,fdout) != y) { | |
| 229 printf("Error writing to output.\n"); | |
| 230 exit(-1); | |
| 231 } | |
| 232 } while (y == sizeof(inbuf)); | |
| 233 fclose(fdout); | |
| 234 fclose(fdin); | |
| 235 } | |
| 236 return 0; | |
| 237 } | |
| 209 | 238 |
| 239 /* $Source: /cvs/libtom/libtomcrypt/demos/encrypt.c,v $ */ | |
| 240 /* $Revision: 1.2 $ */ | |
| 241 /* $Date: 2005/05/05 14:35:56 $ */ |