Mercurial > dropbear

comparison libtomcrypt/demos/encrypt.c @ 302:973fccb59ea4 ucc-axis-hack

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
propagate from branch 'au.asn.ucc.matt.dropbear' (head 11034278bd1917bebcbdc69cf53b1891ce9db121) to branch 'au.asn.ucc.matt.dropbear.ucc-axis-hack' (head 10a1f614fec73d0820c3f61160d9db409b9beb46)
author Matt Johnston <matt@ucc.asn.au>
date Sat, 25 Mar 2006 12:59:58 +0000
parents 1b9e69c058d2
children 0cbe8f6dbf9e
comparison
equal deleted inserted replaced
299:740e782679be 302:973fccb59ea4
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 $ */