2003-06-18 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 gpg_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", 0, 16, 0, 0,
71     foo_setkey, foo_encrypt, foo_decrypt,
72     NULL, NULL,
73   };
74
75 void
76 check_run (void)
77 {
78   int err, id;
79   gcry_cipher_hd_t h;
80   char plain[16] = "Heil Discordia!";
81   char encrypted[16], decrypted[16];
82   gcry_module_t *module;
83
84   err = gcry_cipher_register (&cipher_spec_foo, &module);
85   assert (! err);
86   id = cipher_spec_foo.id;
87
88   err = gcry_cipher_open (&h, id, GCRY_CIPHER_MODE_CBC, 0);
89   assert (! err);
90
91   err = gcry_cipher_encrypt (h,
92                              (unsigned char *) encrypted, sizeof (encrypted),
93                              (unsigned char *) plain, sizeof (plain));
94   assert (! err);
95   assert (memcmp ((void *) plain, (void *) encrypted, sizeof (plain)));
96
97   err = gcry_cipher_reset (h);
98   assert (! err);
99
100   err = gcry_cipher_decrypt (h,
101                              (unsigned char *) decrypted, sizeof (decrypted),
102                              (unsigned char *) encrypted, sizeof (encrypted));
103   assert (! err);
104   assert (! memcmp ((void *) plain, (void *) decrypted, sizeof (plain)));
105
106   gcry_cipher_close (h);
107
108   gcry_cipher_unregister (module);
109 }
110
111 int
112 main (int argc, char **argv)
113 {
114   int debug = 0;
115   int i = 1;
116
117   if (argc > 1 && !strcmp (argv[1], "--verbose"))
118     verbose = 1;
119   else if (argc > 1 && !strcmp (argv[1], "--debug"))
120     verbose = debug = 1;
121
122   gcry_control (GCRYCTL_DISABLE_SECMEM, 0);
123   if (!gcry_check_version (GCRYPT_VERSION))
124     die ("version mismatch\n");
125   gcry_control (GCRYCTL_INITIALIZATION_FINISHED, 0);
126   if (debug)
127     gcry_control (GCRYCTL_SET_DEBUG_FLAGS, 1u , 0);
128
129   for (; i > 0; i--)
130     check_run ();
131   
132   return 0;
133 }