bug fixes
[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 /*static FILE *dumpfp;*/
32
33 /****************
34  * Open a message digest handle for use with algorithm ALGO.
35  * More algorithms may be added by md_enable(). The initial algorithm
36  * may be 0.
37  */
38 MD_HANDLE
39 md_open( int algo, int secure )
40 {
41     MD_HANDLE hd;
42
43    #if 0
44     if( !dumpfp )
45         dumpfp = fopen("md.out", "w");
46     if( !dumpfp )
47         BUG();
48     { int i; for(i=0; i < 16; i++ ) putc('\xff', dumpfp ); }
49    #endif
50     hd = secure ? m_alloc_secure_clear( sizeof *hd )
51                 : m_alloc_clear( sizeof *hd );
52     hd->secure = secure;
53     if( algo )
54         md_enable( hd, algo );
55     return hd;
56 }
57
58 void
59 md_enable( MD_HANDLE h, int algo )
60 {
61     if( algo == DIGEST_ALGO_MD5 ) {
62         md5_init( &h->md5 );
63         h->use_md5 = 1;
64     }
65     else if( algo == DIGEST_ALGO_RMD160 ) {
66         rmd160_init( &h->rmd160 );
67         h->use_rmd160 = 1;
68     }
69     else if( algo == DIGEST_ALGO_SHA1 ) {
70         sha1_init( &h->sha1 );
71         h->use_sha1 = 1;
72     }
73     else
74         log_bug("md_enable(%d)", algo );
75 }
76
77
78 MD_HANDLE
79 md_copy( MD_HANDLE a )
80 {
81     MD_HANDLE b;
82
83     /*{ int i; for(i=0; i < 16; i++ ) putc('\xee', dumpfp ); }*/
84     b = a->secure ? m_alloc_secure( sizeof *b )
85                   : m_alloc( sizeof *b );
86     memcpy( b, a, sizeof *a );
87     return b;
88 }
89
90
91 void
92 md_close(MD_HANDLE a)
93 {
94     if( !a )
95         return;
96     m_free(a);
97 }
98
99
100 void
101 md_write( MD_HANDLE a, byte *inbuf, size_t inlen)
102 {
103   /*  if( a->bufcount && fwrite(a->buffer, a->bufcount, 1, dumpfp ) != 1 )
104         BUG();
105     if( inlen && fwrite(inbuf, inlen, 1, dumpfp ) != 1 )
106         BUG(); */
107     if( a->use_rmd160 ) {
108         rmd160_write( &a->rmd160, a->buffer, a->bufcount );
109         rmd160_write( &a->rmd160, inbuf, inlen  );
110     }
111     if( a->use_sha1 ) {
112         sha1_write( &a->sha1, a->buffer, a->bufcount );
113         sha1_write( &a->sha1, inbuf, inlen  );
114     }
115     if( a->use_md5 ) {
116         md5_write( &a->md5, a->buffer, a->bufcount );
117         md5_write( &a->md5, inbuf, inlen  );
118     }
119     a->bufcount = 0;
120 }
121
122
123
124 void
125 md_final(MD_HANDLE a)
126 {
127     if( a->bufcount )
128         md_write( a, NULL, 0 );
129     /*{ int i; for(i=0; i < 16; i++ ) putc('\xcc', dumpfp ); }*/
130     if( a->use_rmd160 ) {
131         byte *p;
132         rmd160_final( &a->rmd160 );
133         p = rmd160_read( &a->rmd160 );
134     }
135     if( a->use_sha1 )
136         sha1_final( &a->sha1 );
137     if( a->use_md5 )
138         md5_final( &a->md5 );
139 }
140
141
142 /****************
143  * if ALGO is null get the digest for the used algo (which should be only one)
144  */
145 byte *
146 md_read( MD_HANDLE a, int algo )
147 {
148     if( !algo ) {
149         if( a->use_rmd160 )
150             return rmd160_read( &a->rmd160 );
151         if( a->use_sha1 )
152             return sha1_read( &a->sha1 );
153         if( a->use_md5 )
154             return md5_read( &a->md5 );
155     }
156     else {
157         if( algo == DIGEST_ALGO_RMD160 )
158             return rmd160_read( &a->rmd160 );
159         if( algo == DIGEST_ALGO_SHA1 )
160             return sha1_read( &a->sha1 );
161         if( algo == DIGEST_ALGO_MD5 )
162             return md5_read( &a->md5 );
163     }
164     BUG();
165 }
166
167 int
168 md_get_algo( MD_HANDLE a )
169 {
170     if( a->use_rmd160 )
171         return DIGEST_ALGO_RMD160;
172     if( a->use_sha1 )
173         return DIGEST_ALGO_SHA1;
174     if( a->use_md5 )
175         return DIGEST_ALGO_MD5;
176     return 0;
177 }
178