A lot of cleanups as well as minor API changes.
[libgcrypt.git] / tests / register.c
1 /* register.c - Test for registering of additional cipher modules.
2  *      Copyright (C) 2001, 2002, 2003, 2005 Free Software Foundation, Inc.
3  *
4  * This file is part of Libgcrypt.
5  *
6  * Libgcrypt is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU Lesser General Public License as
8  * published by the Free Software Foundation; either version 2.1 of
9  * the License, or (at your option) any later version.
10  *
11  * Libgcrypt is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this program; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
19  */
20
21 #ifdef HAVE_CONFIG_H
22 #include <config.h>
23 #endif
24 #include <stdarg.h>
25 #include <stdio.h>
26 #include <stdlib.h>
27 #include <string.h>
28 #include <assert.h>
29
30 #include "../src/gcrypt.h"
31
32 static int verbose;
33
34 static void
35 die (const char *format, ...)
36 {
37   va_list arg_ptr ;
38
39   va_start( arg_ptr, format ) ;
40   vfprintf (stderr, format, arg_ptr );
41   va_end(arg_ptr);
42   exit (1);
43 }
44
45 gcry_err_code_t
46 foo_setkey (void *c, const unsigned char *key, unsigned keylen)
47 {
48   (void)c;
49   (void)key;
50   (void)keylen;
51
52   return 0;
53 }
54
55 #define FOO_BLOCKSIZE 16
56
57 void
58 foo_encrypt (void *c, unsigned char *outbuf, const unsigned char *inbuf)
59 {
60   int i;
61
62   (void)c;
63
64   for (i = 0; i < FOO_BLOCKSIZE; i++)
65     outbuf[i] = inbuf[i] ^ 0x42;
66 }
67
68 void
69 foo_decrypt (void *c, unsigned char *outbuf, const unsigned char *inbuf)
70 {
71   int i;
72
73   (void)c;
74
75   for (i = 0; i < FOO_BLOCKSIZE; i++)
76     outbuf[i] = inbuf[i] ^ 0x42;
77 }
78
79 gcry_cipher_spec_t cipher_spec_foo =
80   {
81     "FOO", NULL, NULL, 16, 0, 0,
82     foo_setkey, foo_encrypt, foo_decrypt,
83     NULL, NULL,
84   };
85
86 int
87 check_list (int algorithm)
88 {
89   gcry_error_t err = GPG_ERR_NO_ERROR;
90   int *list, list_length;
91   int i, ret = 0;
92
93   err = gcry_cipher_list (NULL, &list_length);
94   assert (! err);
95   list = malloc (sizeof (int) * list_length);
96   assert (list);
97   err = gcry_cipher_list (list, &list_length);
98   
99   for (i = 0; i < list_length && (! ret); i++)
100     if (list[i] == algorithm)
101       ret = 1;
102
103   return ret;
104 }
105
106 void
107 check_run (void)
108 {
109   int err, algorithm;
110   gcry_cipher_hd_t h;
111   char plain[16] = "Heil Discordia!";
112   char encrypted[16], decrypted[16];
113   gcry_module_t module;
114   int ret;
115
116   err = gcry_cipher_register (&cipher_spec_foo, &algorithm, &module);
117   assert (! err);
118
119   err = gcry_cipher_open (&h, algorithm, GCRY_CIPHER_MODE_CBC, 0);
120   assert (! err);
121
122   err = gcry_cipher_encrypt (h,
123                              (unsigned char *) encrypted, sizeof (encrypted),
124                              (unsigned char *) plain, sizeof (plain));
125   assert (! err);
126   assert (memcmp ((void *) plain, (void *) encrypted, sizeof (plain)));
127
128   err = gcry_cipher_reset (h);
129   assert (! err);
130
131   err = gcry_cipher_decrypt (h,
132                              (unsigned char *) decrypted, sizeof (decrypted),
133                              (unsigned char *) encrypted, sizeof (encrypted));
134   assert (! err);
135   assert (! memcmp ((void *) plain, (void *) decrypted, sizeof (plain)));
136
137   ret = check_list (algorithm);
138   assert (ret);
139
140   gcry_cipher_close (h);
141
142   gcry_cipher_unregister (module);
143
144   ret = check_list (algorithm);
145   assert (! ret);
146 }
147
148 int
149 main (int argc, char **argv)
150 {
151   int debug = 0;
152   int i = 1;
153
154   if (argc > 1 && !strcmp (argv[1], "--verbose"))
155     verbose = 1;
156   else if (argc > 1 && !strcmp (argv[1], "--debug"))
157     verbose = debug = 1;
158
159   gcry_control (GCRYCTL_DISABLE_SECMEM, 0);
160   if (!gcry_check_version (GCRYPT_VERSION))
161     die ("version mismatch\n");
162   gcry_control (GCRYCTL_INITIALIZATION_FINISHED, 0);
163   if (debug)
164     gcry_control (GCRYCTL_SET_DEBUG_FLAGS, 1u , 0);
165
166   for (; i > 0; i--)
167     check_run ();
168   
169   return 0;
170 }