Mercurial > dropbear

comparison demos/encrypt.c @ 3:7faae8f46238 libtomcrypt-orig

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Branch renaming
author Matt Johnston <matt@ucc.asn.au>
date Mon, 31 May 2004 18:25:41 +0000
parents
children 1c15b283127b
comparison
equal deleted inserted replaced
-1:000000000000 3:7faae8f46238
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 <mycrypt.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
73 if (register_hash(&sha256_desc) == -1) {
74 printf("Error registering SHA256\n");
75 exit(-1);
76 }
77
78 if (register_prng(&yarrow_desc) == -1) {
79 printf("Error registering yarrow PRNG\n");
80 exit(-1);
81 }
82
83 if (register_prng(&sprng_desc) == -1) {
84 printf("Error registering sprng PRNG\n");
85 exit(-1);
86 }
87 }
88
89 int main(int argc, char *argv[])
90 {
91 unsigned char plaintext[512],ciphertext[512];
92 unsigned char tmpkey[512], key[MAXBLOCKSIZE], IV[MAXBLOCKSIZE];
93 unsigned char inbuf[512]; /* i/o block size */
94 unsigned long outlen, y, ivsize, x, decrypt;
95 symmetric_CTR ctr;
96 int cipher_idx, hash_idx, ks;
97 char *infile, *outfile, *cipher;
98 prng_state prng;
99 FILE *fdin, *fdout;
100
101 /* register algs, so they can be printed */
102 register_algs();
103
104 if (argc < 4) {
105 return usage(argv[0]);
106 }
107
108 if (!strcmp(argv[1], "-d")) {
109 decrypt = 1;
110 cipher = argv[2];
111 infile = argv[3];
112 outfile = argv[4];
113 } else {
114 decrypt = 0;
115 cipher = argv[1];
116 infile = argv[2];
117 outfile = argv[3];
118 }
119
120 /* file handles setup */
121 fdin = fopen(infile,"rb");
122 if (fdin == NULL) {
123 perror("Can't open input for reading");
124 exit(-1);
125 }
126
127 fdout = fopen(outfile,"wb");
128 if (fdout == NULL) {
129 perror("Can't open output for writing");
130 exit(-1);
131 }
132
133 cipher_idx = find_cipher(cipher);
134 if (cipher_idx == -1) {
135 printf("Invalid cipher entered on command line.\n");
136 exit(-1);
137 }
138
139 hash_idx = find_hash("sha256");
140 if (hash_idx == -1) {
141 printf("SHA256 not found...?\n");
142 exit(-1);
143 }
144
145 ivsize = cipher_descriptor[cipher_idx].block_length;
146 ks = hash_descriptor[hash_idx].hashsize;
147 if (cipher_descriptor[cipher_idx].keysize(&ks) != CRYPT_OK) {
148 printf("Invalid keysize???\n");
149 exit(-1);
150 }
151
152 printf("\nEnter key: ");
153 fgets((char *)tmpkey,sizeof(tmpkey), stdin);
154 outlen = sizeof(key);
155 if ((errno = hash_memory(hash_idx,tmpkey,strlen((char *)tmpkey),key,&outlen)) != CRYPT_OK) {
156 printf("Error hashing key: %s\n", error_to_string(errno));
157 exit(-1);
158 }
159
160 if (decrypt) {
161 /* Need to read in IV */
162 if (fread(IV,1,ivsize,fdin) != ivsize) {
163 printf("Error reading IV from input.\n");
164 exit(-1);
165 }
166
167 if ((errno = ctr_start(cipher_idx,IV,key,ks,0,&ctr)) != CRYPT_OK) {
168 printf("ctr_start error: %s\n",error_to_string(errno));
169 exit(-1);
170 }
171
172 /* IV done */
173 do {
174 y = fread(inbuf,1,sizeof(inbuf),fdin);
175
176 if ((errno = ctr_decrypt(inbuf,plaintext,y,&ctr)) != CRYPT_OK) {
177 printf("ctr_decrypt error: %s\n", error_to_string(errno));
178 exit(-1);
179 }
180
181 if (fwrite(plaintext,1,y,fdout) != y) {
182 printf("Error writing to file.\n");
183 exit(-1);
184 }
185 } while (y == sizeof(inbuf));
186 fclose(fdin);
187 fclose(fdout);
188
189 } else { /* encrypt */
190 /* Setup yarrow for random bytes for IV */
191
192 if ((errno = rng_make_prng(128, find_prng("yarrow"), &prng, NULL)) != CRYPT_OK) {
193 printf("Error setting up PRNG, %s\n", error_to_string(errno));
194 }
195
196 /* You can use rng_get_bytes on platforms that support it */
197 /* x = rng_get_bytes(IV,ivsize,NULL);*/
198 x = yarrow_read(IV,ivsize,&prng);
199 if (x != ivsize) {
200 printf("Error reading PRNG for IV required.\n");
201 exit(-1);
202 }
203
204 if (fwrite(IV,1,ivsize,fdout) != ivsize) {
205 printf("Error writing IV to output.\n");
206 exit(-1);
207 }
208
209 if ((errno = ctr_start(cipher_idx,IV,key,ks,0,&ctr)) != CRYPT_OK) {
210 printf("ctr_start error: %s\n",error_to_string(errno));
211 exit(-1);
212 }
213
214 do {
215 y = fread(inbuf,1,sizeof(inbuf),fdin);
216
217 if ((errno = ctr_encrypt(inbuf,ciphertext,y,&ctr)) != CRYPT_OK) {
218 printf("ctr_encrypt error: %s\n", error_to_string(errno));
219 exit(-1);
220 }
221
222 if (fwrite(ciphertext,1,y,fdout) != y) {
223 printf("Error writing to output.\n");
224 exit(-1);
225 }
226 } while (y == sizeof(inbuf));
227 fclose(fdout);
228 fclose(fdin);
229 }
230 return 0;
231 }