221cf7199f65399e70a8d046921043792373a2ce
[libgcrypt.git] / cipher / md.c
1 /* md.c  -  message digest dispatcher
2  *      Copyright (c) 1997 by Werner Koch (dd9jn)
3  *
4  * This file is part of G10.
5  *
6  * G10 is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * G10 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 General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * 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 <config.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <string.h>
25 #include <errno.h>
26 #include "util.h"
27 #include "cipher.h"
28 #include "errors.h"
29
30
31 /****************
32  * Open a message digest handle for use with algorithm ALGO.
33  * More algorithms may be added by md_enable(). The initial algorithm
34  * may be 0.
35  */
36 MD_HANDLE
37 md_open( int algo, int secure )
38 {
39     MD_HANDLE hd;
40
41     hd = secure ? m_alloc_secure_clear( sizeof *hd )
42                 : m_alloc_clear( sizeof *hd );
43     if( algo )
44         md_enable( hd, algo );
45     return hd;
46 }
47
48 void
49 md_enable( MD_HANDLE h, int algo )
50 {
51     if( algo == DIGEST_ALGO_MD5 ) {
52         md5_init( &h->md5 );
53         h->use_md5 = 1;
54     }
55     else if( algo == DIGEST_ALGO_RMD160 ) {
56         rmd160_init( &h->rmd160 );
57         h->use_rmd160 = 1;
58     }
59     else if( algo == DIGEST_ALGO_SHA1 ) {
60         sha1_init( &h->sha1 );
61         h->use_sha1 = 1;
62     }
63     else
64         log_bug("md_enable(%d)", algo );
65 }
66
67
68 MD_HANDLE
69 md_copy( MD_HANDLE a )
70 {
71     MD_HANDLE b;
72
73     b = m_is_secure(a)? m_alloc_secure( sizeof *b )
74                       : m_alloc( sizeof *b );
75     memcpy( b, a, sizeof *a );
76     return b;
77 }
78
79
80 void
81 md_close(MD_HANDLE a)
82 {
83     if( !a )
84         return;
85     m_free(a);
86 }
87
88
89 void
90 md_write( MD_HANDLE a, byte *inbuf, size_t inlen)
91 {
92     if( a->use_rmd160 ) {
93         rmd160_write( &a->rmd160, a->buffer, a->bufcount );
94         rmd160_write( &a->rmd160, inbuf, inlen  );
95     }
96     if( a->use_sha1 ) {
97         sha1_write( &a->sha1, a->buffer, a->bufcount );
98         sha1_write( &a->sha1, inbuf, inlen  );
99     }
100     if( a->use_md5 ) {
101         md5_write( &a->md5, a->buffer, a->bufcount );
102         md5_write( &a->md5, inbuf, inlen  );
103     }
104     a->bufcount = 0;
105 }
106
107
108
109 void
110 md_final(MD_HANDLE a)
111 {
112     if( a->bufcount )
113         md_write( a, NULL, 0 );
114     if( a->use_rmd160 ) {
115         byte *p;
116         rmd160_final( &a->rmd160 );
117         p = rmd160_read( &a->rmd160 );
118     }
119     if( a->use_sha1 )
120         sha1_final( &a->sha1 );
121     if( a->use_md5 )
122         md5_final( &a->md5 );
123 }
124
125
126 /****************
127  * if ALGO is null get the digest for the used algo (which should be only one)
128  */
129 byte *
130 md_read( MD_HANDLE a, int algo )
131 {
132     if( !algo ) {
133         if( a->use_rmd160 )
134             return rmd160_read( &a->rmd160 );
135         if( a->use_sha1 )
136             return sha1_read( &a->sha1 );
137         if( a->use_md5 )
138             return md5_read( &a->md5 );
139     }
140     else {
141         if( algo == DIGEST_ALGO_RMD160 )
142             return rmd160_read( &a->rmd160 );
143         if( algo == DIGEST_ALGO_SHA1 )
144             return sha1_read( &a->sha1 );
145         if( algo == DIGEST_ALGO_MD5 )
146             return md5_read( &a->md5 );
147     }
148     BUG();
149 }
150
151 int
152 md_get_algo( MD_HANDLE a )
153 {
154     if( a->use_rmd160 )
155         return DIGEST_ALGO_RMD160;
156     if( a->use_sha1 )
157         return DIGEST_ALGO_SHA1;
158     if( a->use_md5 )
159         return DIGEST_ALGO_MD5;
160     return 0;
161 }
162