New configure option --disable-endian-check.
[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   return 0;
49 }
50
51 #define FOO_BLOCKSIZE 16
52
53 void
54 foo_encrypt (void *c, unsigned char *outbuf, const unsigned char *inbuf)
55 {
56   int i;
57
58   for (i = 0; i < FOO_BLOCKSIZE; i++)
59     outbuf[i] = inbuf[i] ^ 0x42;
60 }
61
62 void
63 foo_decrypt (void *c, unsigned char *outbuf, const unsigned char *inbuf)
64 {
65   int i;
66
67   for (i = 0; i < FOO_BLOCKSIZE; i++)
68     outbuf[i] = inbuf[i] ^ 0x42;
69 }
70
71 gcry_cipher_spec_t cipher_spec_foo =
72   {
73     "FOO", NULL, NULL, 16, 0, 0,
74     foo_setkey, foo_encrypt, foo_decrypt,
75     NULL, NULL,
76   };
77
78 int
79 check_list (int algorithm)
80 {
81   gcry_error_t err = GPG_ERR_NO_ERROR;
82   int *list, list_length;
83   int i, ret = 0;
84
85   err = gcry_cipher_list (NULL, &list_length);
86   assert (! err);
87   list = malloc (sizeof (int) * list_length);
88   assert (list);
89   err = gcry_cipher_list (list, &list_length);
90   
91   for (i = 0; i < list_length && (! ret); i++)
92     if (list[i] == algorithm)
93       ret = 1;
94
95   return ret;
96 }
97
98 void
99 check_run (void)
100 {
101   int err, algorithm;
102   gcry_cipher_hd_t h;
103   char plain[16] = "Heil Discordia!";
104   char encrypted[16], decrypted[16];
105   gcry_module_t module;
106   int ret;
107
108   err = gcry_cipher_register (&cipher_spec_foo, &algorithm, &module);
109   assert (! err);
110
111   err = gcry_cipher_open (&h, algorithm, GCRY_CIPHER_MODE_CBC, 0);
112   assert (! err);
113
114   err = gcry_cipher_encrypt (h,
115                              (unsigned char *) encrypted, sizeof (encrypted),
116                              (unsigned char *) plain, sizeof (plain));
117   assert (! err);
118   assert (memcmp ((void *) plain, (void *) encrypted, sizeof (plain)));
119
120   err = gcry_cipher_reset (h);
121   assert (! err);
122
123   err = gcry_cipher_decrypt (h,
124                              (unsigned char *) decrypted, sizeof (decrypted),
125                              (unsigned char *) encrypted, sizeof (encrypted));
126   assert (! err);
127   assert (! memcmp ((void *) plain, (void *) decrypted, sizeof (plain)));
128
129   ret = check_list (algorithm);
130   assert (ret);
131
132   gcry_cipher_close (h);
133
134   gcry_cipher_unregister (module);
135
136   ret = check_list (algorithm);
137   assert (! ret);
138 }
139
140 int
141 main (int argc, char **argv)
142 {
143   int debug = 0;
144   int i = 1;
145
146   if (argc > 1 && !strcmp (argv[1], "--verbose"))
147     verbose = 1;
148   else if (argc > 1 && !strcmp (argv[1], "--debug"))
149     verbose = debug = 1;
150
151   gcry_control (GCRYCTL_DISABLE_SECMEM, 0);
152   if (!gcry_check_version (GCRYPT_VERSION))
153     die ("version mismatch\n");
154   gcry_control (GCRYCTL_INITIALIZATION_FINISHED, 0);
155   if (debug)
156     gcry_control (GCRYCTL_SET_DEBUG_FLAGS, 1u , 0);
157
158   for (; i > 0; i--)
159     check_run ();
160   
161   return 0;
162 }