Mercurial > dropbear

comparison libtomcrypt/demos/encrypt.c @ 285:1b9e69c058d2

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
propagate from branch 'au.asn.ucc.matt.ltc.dropbear' (head 20dccfc09627970a312d77fb41dc2970b62689c3) to branch 'au.asn.ucc.matt.dropbear' (head fdf4a7a3b97ae5046139915de7e40399cceb2c01)
author Matt Johnston <matt@ucc.asn.au>
date Wed, 08 Mar 2006 13:23:58 +0000
parents
children 0cbe8f6dbf9e
comparison
equal deleted inserted replaced
281:997e6f7dc01e 285:1b9e69c058d2
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
10 #include <tomcrypt.h>
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
72 #ifdef KHAZAD
73 register_cipher (&khazad_desc);
74 #endif
75 #ifdef ANUBIS
76 register_cipher (&anubis_desc);
77 #endif
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 }
238
239 /* $Source: /cvs/libtom/libtomcrypt/demos/encrypt.c,v $ */
240 /* $Revision: 1.2 $ */
241 /* $Date: 2005/05/05 14:35:56 $ */