started with trust stuff
[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 <errno.h>
25 #include "util.h"
26 #include "cipher.h"
27 #include "errors.h"
28
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     if( algo )
43         md_enable( hd, algo );
44     return hd;
45 }
46
47 void
48 md_enable( MD_HANDLE h, int algo )
49 {
50     if( algo == DIGEST_ALGO_MD5 ) {
51         md5_init( &h->md5 );
52         h->use_md5 = 1;
53     }
54     else if( algo == DIGEST_ALGO_RMD160 ) {
55         rmd160_init( &h->rmd160 );
56         h->use_rmd160 = 1;
57     }
58     else if( algo == DIGEST_ALGO_SHA1 ) {
59         sha1_init( &h->sha1 );
60         h->use_sha1 = 1;
61     }
62     else
63         log_bug("md_enable(%d)", algo );
64 }
65
66
67 MD_HANDLE
68 md_copy( MD_HANDLE a )
69 {
70     MD_HANDLE b;
71
72     b = m_is_secure(a)? m_alloc_secure( sizeof *b )
73                       : m_alloc( sizeof *b );
74     memcpy( b, a, sizeof *a );
75     return b;
76 }
77
78
79 void
80 md_close(MD_HANDLE a)
81 {
82     if( !a )
83         return;
84     m_free(a);
85 }
86
87
88 void
89 md_write( MD_HANDLE a, byte *inbuf, size_t inlen)
90 {
91     if( a->use_rmd160 ) {
92         rmd160_write( &a->rmd160, a->buffer, a->bufcount );
93         rmd160_write( &a->rmd160, inbuf, inlen  );
94     }
95     if( a->use_sha1 ) {
96         sha1_write( &a->sha1, a->buffer, a->bufcount );
97         sha1_write( &a->sha1, inbuf, inlen  );
98     }
99     if( a->use_md5 ) {
100         md5_write( &a->md5, a->buffer, a->bufcount );
101         md5_write( &a->md5, inbuf, inlen  );
102     }
103     a->bufcount = 0;
104 }
105
106
107
108 void
109 md_final(MD_HANDLE a)
110 {
111     if( a->bufcount )
112         md_write( a, NULL, 0 );
113     if( a->use_rmd160 ) {
114         byte *p;
115         rmd160_final( &a->rmd160 );
116         p = rmd160_read( &a->rmd160 );
117     }
118     if( a->use_sha1 )
119         sha1_final( &a->sha1 );
120     if( a->use_md5 )
121         md5_final( &a->md5 );
122 }
123
124
125 /****************
126  * if ALGO is null get the digest for the used algo (which should be only one)
127  */
128 byte *
129 md_read( MD_HANDLE a, int algo )
130 {
131     if( !algo ) {
132         if( a->use_rmd160 )
133             return rmd160_read( &a->rmd160 );
134         if( a->use_sha1 )
135             return sha1_read( &a->sha1 );
136         if( a->use_md5 )
137             return md5_read( &a->md5 );
138     }
139     else {
140         if( algo == DIGEST_ALGO_RMD160 )
141             return rmd160_read( &a->rmd160 );
142         if( algo == DIGEST_ALGO_SHA1 )
143             return sha1_read( &a->sha1 );
144         if( algo == DIGEST_ALGO_MD5 )
145             return md5_read( &a->md5 );
146     }
147     log_bug(NULL);
148 }
149
150 int
151 md_get_algo( MD_HANDLE a )
152 {
153     if( a->use_rmd160 )
154         return DIGEST_ALGO_RMD160;
155     if( a->use_sha1 )
156         return DIGEST_ALGO_SHA1;
157     if( a->use_md5 )
158         return DIGEST_ALGO_MD5;
159     return 0;
160 }
161