some cleanups
[libgcrypt.git] / cipher / md.c
1 /* md.c  -  message digest dispatcher
2  *      Copyright (C) 1998 Free Software Foundation, Inc.
3  *
4  * This file is part of GNUPG.
5  *
6  * GNUPG 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  * GNUPG 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  * Open a message digest handle for use with algorithm ALGO.
32  * More algorithms may be added by md_enable(). The initial algorithm
33  * may be 0.
34  */
35 MD_HANDLE
36 md_open( int algo, int secure )
37 {
38     MD_HANDLE hd;
39
40     hd = secure ? m_alloc_secure_clear( sizeof *hd )
41                 : m_alloc_clear( sizeof *hd );
42     hd->secure = secure;
43     if( algo )
44         md_enable( hd, algo );
45     fast_random_poll();
46     return hd;
47 }
48
49 void
50 md_enable( MD_HANDLE h, int algo )
51 {
52     if( algo == DIGEST_ALGO_MD5 ) {
53         if( !h->use_md5 )
54             md5_init( &h->md5 );
55         h->use_md5 = 1;
56     }
57     else if( algo == DIGEST_ALGO_RMD160 ) {
58         if( !h->use_rmd160 )
59             rmd160_init( &h->rmd160 );
60         h->use_rmd160 = 1;
61     }
62     else if( algo == DIGEST_ALGO_SHA1 ) {
63         if( !h->use_sha1 )
64             sha1_init( &h->sha1 );
65         h->use_sha1 = 1;
66     }
67     else
68         log_bug("md_enable(%d)", algo );
69 }
70
71
72 MD_HANDLE
73 md_copy( MD_HANDLE a )
74 {
75     MD_HANDLE b;
76
77     b = a->secure ? m_alloc_secure( sizeof *b )
78                   : m_alloc( sizeof *b );
79     memcpy( b, a, sizeof *a );
80     return b;
81 }
82
83
84 void
85 md_close(MD_HANDLE a)
86 {
87     if( !a )
88         return;
89     m_free(a);
90 }
91
92
93 void
94 md_write( MD_HANDLE a, byte *inbuf, size_t inlen)
95 {
96     if( a->debug ) {
97         if( a->bufcount && fwrite(a->buffer, a->bufcount, 1, a->debug ) != 1 )
98             BUG();
99         if( inlen && fwrite(inbuf, inlen, 1, a->debug ) != 1 )
100             BUG();
101     }
102     if( a->use_rmd160 ) {
103         rmd160_write( &a->rmd160, a->buffer, a->bufcount );
104         rmd160_write( &a->rmd160, inbuf, inlen  );
105     }
106     if( a->use_sha1 ) {
107         sha1_write( &a->sha1, a->buffer, a->bufcount );
108         sha1_write( &a->sha1, inbuf, inlen  );
109     }
110     if( a->use_md5 ) {
111         md5_write( &a->md5, a->buffer, a->bufcount );
112         md5_write( &a->md5, inbuf, inlen  );
113     }
114     a->bufcount = 0;
115 }
116
117
118
119 void
120 md_final(MD_HANDLE a)
121 {
122     if( a->bufcount )
123         md_write( a, NULL, 0 );
124     if( a->use_rmd160 ) {
125         byte *p;
126         rmd160_final( &a->rmd160 );
127         p = rmd160_read( &a->rmd160 );
128     }
129     if( a->use_sha1 )
130         sha1_final( &a->sha1 );
131     if( a->use_md5 )
132         md5_final( &a->md5 );
133 }
134
135
136 /****************
137  * if ALGO is null get the digest for the used algo (which should be only one)
138  */
139 byte *
140 md_read( MD_HANDLE a, int algo )
141 {
142     if( !algo ) {
143         if( a->use_rmd160 )
144             return rmd160_read( &a->rmd160 );
145         if( a->use_sha1 )
146             return sha1_read( &a->sha1 );
147         if( a->use_md5 )
148             return md5_read( &a->md5 );
149     }
150     else {
151         if( algo == DIGEST_ALGO_RMD160 )
152             return rmd160_read( &a->rmd160 );
153         if( algo == DIGEST_ALGO_SHA1 )
154             return sha1_read( &a->sha1 );
155         if( algo == DIGEST_ALGO_MD5 )
156             return md5_read( &a->md5 );
157     }
158     BUG();
159 }
160
161 int
162 md_get_algo( MD_HANDLE a )
163 {
164     if( a->use_rmd160 )
165         return DIGEST_ALGO_RMD160;
166     if( a->use_sha1 )
167         return DIGEST_ALGO_SHA1;
168     if( a->use_md5 )
169         return DIGEST_ALGO_MD5;
170     return 0;
171 }
172
173 /****************
174  * Return the length of the digest
175  */
176 int
177 md_digest_length( int algo )
178 {
179     switch( algo ) {
180       case DIGEST_ALGO_RMD160:
181       case DIGEST_ALGO_SHA1:
182         return 20;
183       default:
184         return 16;
185     }
186 }
187
188
189 const byte *
190 md_asn_oid( int algo, size_t *asnlen, size_t *mdlen )
191 {
192     size_t alen, mlen;
193     byte *p;
194
195     if( algo == DIGEST_ALGO_MD5 ) {
196         static byte asn[18] = /* Object ID is 1.2.840.113549.2.5 */
197                     { 0x30, 0x20, 0x30, 0x0c, 0x06, 0x08, 0x2a, 0x86,0x48,
198                       0x86, 0xf7, 0x0d, 0x02, 0x05, 0x05, 0x00, 0x04, 0x10 };
199         mlen = 16; alen = DIM(asn); p = asn;
200     }
201     else if( algo == DIGEST_ALGO_RMD160 ) {
202         static byte asn[15] = /* Object ID is 1.3.36.3.2.1 */
203           { 0x30, 0x21, 0x30, 0x09, 0x06, 0x05, 0x2b, 0x24, 0x03,
204             0x02, 0x01, 0x05, 0x00, 0x04, 0x14 };
205         mlen = 20; alen = DIM(asn); p = asn;
206     }
207     else if( algo == DIGEST_ALGO_SHA1 ) {
208         static byte asn[15] = /* Objet ID is 1.3.14.3.2.26 */
209                     { 0x30, 0x21, 0x30, 0x09, 0x06, 0x05, 0x2b, 0x0e, 0x03,
210                       0x02, 0x1a, 0x05, 0x00, 0x04, 0x14 };
211         mlen = 20; alen = DIM(asn); p = asn;
212     }
213     else
214         log_bug("md_asn_oid(%d)", algo );
215
216     if( asnlen )
217         *asnlen = alen;
218     if( mdlen )
219         *mdlen = mlen;
220     return p;
221 }
222
223