2005-04-11 Moritz Schulte <moritz@g10code.com>
[libgcrypt.git] / tests / register.c
1 /* register.c - Test for registering of additional cipher modules.
2  *      Copyright (C) 2001, 2002, 2003 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 #include <stdarg.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <string.h>
25 #include <assert.h>
26
27 #include "../src/gcrypt.h"
28
29 static int verbose;
30
31 static void
32 die (const char *format, ...)
33 {
34   va_list arg_ptr ;
35
36   va_start( arg_ptr, format ) ;
37   vfprintf (stderr, format, arg_ptr );
38   va_end(arg_ptr);
39   exit (1);
40 }
41
42 gcry_err_code_t
43 foo_setkey (void *c, const unsigned char *key, unsigned keylen)
44 {
45   return 0;
46 }
47
48 #define FOO_BLOCKSIZE 16
49
50 void
51 foo_encrypt (void *c, unsigned char *outbuf, const unsigned char *inbuf)
52 {
53   int i;
54
55   for (i = 0; i < FOO_BLOCKSIZE; i++)
56     outbuf[i] = inbuf[i] ^ 0x42;
57 }
58
59 void
60 foo_decrypt (void *c, unsigned char *outbuf, const unsigned char *inbuf)
61 {
62   int i;
63
64   for (i = 0; i < FOO_BLOCKSIZE; i++)
65     outbuf[i] = inbuf[i] ^ 0x42;
66 }
67
68 gcry_cipher_spec_t cipher_spec_foo =
69   {
70     "FOO", NULL, NULL, 16, 0, 0,
71     foo_setkey, foo_encrypt, foo_decrypt,
72     NULL, NULL,
73   };
74
75 int
76 check_list (int algorithm)
77 {
78   gcry_error_t err = GPG_ERR_NO_ERROR;
79   int *list, list_length;
80   int i, ret = 0;
81
82   err = gcry_cipher_list (NULL, &list_length);
83   assert (! err);
84   list = malloc (sizeof (int) * list_length);
85   assert (list);
86   err = gcry_cipher_list (list, &list_length);
87   
88   for (i = 0; i < list_length && (! ret); i++)
89     if (list[i] == algorithm)
90       ret = 1;
91
92   return ret;
93 }
94
95 void
96 check_run (void)
97 {
98   int err, algorithm;
99   gcry_cipher_hd_t h;
100   char plain[16] = "Heil Discordia!";
101   char encrypted[16], decrypted[16];
102   gcry_module_t module;
103   int ret;
104
105   err = gcry_cipher_register (&cipher_spec_foo, &algorithm, &module);
106   assert (! err);
107
108   err = gcry_cipher_open (&h, algorithm, GCRY_CIPHER_MODE_CBC, 0);
109   assert (! err);
110
111   err = gcry_cipher_encrypt (h,
112                              (unsigned char *) encrypted, sizeof (encrypted),
113                              (unsigned char *) plain, sizeof (plain));
114   assert (! err);
115   assert (memcmp ((void *) plain, (void *) encrypted, sizeof (plain)));
116
117   err = gcry_cipher_reset (h);
118   assert (! err);
119
120   err = gcry_cipher_decrypt (h,
121                              (unsigned char *) decrypted, sizeof (decrypted),
122                              (unsigned char *) encrypted, sizeof (encrypted));
123   assert (! err);
124   assert (! memcmp ((void *) plain, (void *) decrypted, sizeof (plain)));
125
126   ret = check_list (algorithm);
127   assert (ret);
128
129   gcry_cipher_close (h);
130
131   gcry_cipher_unregister (module);
132
133   ret = check_list (algorithm);
134   assert (! ret);
135 }
136
137 int
138 main (int argc, char **argv)
139 {
140   int debug = 0;
141   int i = 1;
142
143   if (argc > 1 && !strcmp (argv[1], "--verbose"))
144     verbose = 1;
145   else if (argc > 1 && !strcmp (argv[1], "--debug"))
146     verbose = debug = 1;
147
148   gcry_control (GCRYCTL_DISABLE_SECMEM, 0);
149   if (!gcry_check_version (GCRYPT_VERSION))
150     die ("version mismatch\n");
151   gcry_control (GCRYCTL_INITIALIZATION_FINISHED, 0);
152   if (debug)
153     gcry_control (GCRYCTL_SET_DEBUG_FLAGS, 1u , 0);
154
155   for (; i > 0; i--)
156     check_run ();
157   
158   return 0;
159 }