5896c3f32e3595a07d14556c3276ea6aec4e7a51
[gnupg.git] / g10 / textfilter.c
1 /* textfilter.c
2  *      Copyright (C) 1998,1999 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 <assert.h>
27
28 #include "errors.h"
29 #include "iobuf.h"
30 #include "memory.h"
31 #include "util.h"
32 #include "filter.h"
33 #include "i18n.h"
34
35
36 #define MAX_LINELEN 19995 /* a little bit smaller than in armor.c */
37                           /* to make sure that a warning is displayed while */
38                           /* creating a message */
39
40 unsigned
41 len_without_trailing_ws( byte *line, unsigned len )
42 {
43     byte *p, *mark;
44     unsigned n;
45
46     for(mark=NULL, p=line, n=0; n < len; n++, p++ ) {
47         if( strchr(" \t\r\n", *p ) ) {
48             if( !mark )
49                 mark = p;
50         }
51         else
52             mark = NULL;
53     }
54
55     return mark? (mark - line) : len;
56 }
57
58
59
60
61 static int
62 standard( text_filter_context_t *tfx, IOBUF a,
63           byte *buf, size_t size, size_t *ret_len)
64 {
65     int rc=0;
66     size_t len = 0;
67     unsigned maxlen;
68
69     assert( size > 10 );
70     size -= 2;  /* reserve 2 bytes to append CR,LF */
71     while( !rc && len < size ) {
72         int lf_seen;
73
74         while( len < size && tfx->buffer_pos < tfx->buffer_len )
75             buf[len++] = tfx->buffer[tfx->buffer_pos++];
76         if( len >= size )
77             continue;
78
79         /* read the next line */
80         maxlen = MAX_LINELEN;
81         tfx->buffer_pos = 0;
82         tfx->buffer_len = iobuf_read_line( a, &tfx->buffer,
83                                            &tfx->buffer_size, &maxlen );
84         if( !maxlen )
85             tfx->truncated++;
86         if( !tfx->buffer_len ) {
87             if( !len )
88                 rc = -1; /* eof */
89             break;
90         }
91         lf_seen = tfx->buffer[tfx->buffer_len-1] == '\n';
92         tfx->buffer_len = trim_trailing_ws( tfx->buffer, tfx->buffer_len );
93         if( lf_seen ) {
94             tfx->buffer[tfx->buffer_len++] = '\r';
95             tfx->buffer[tfx->buffer_len++] = '\n';
96         }
97     }
98     *ret_len = len;
99     return rc;
100 }
101
102
103
104
105 /****************
106  * The filter is used to make canonical text: Lines are terminated by
107  * CR, LF, trailing white spaces are removed.
108  */
109 int
110 text_filter( void *opaque, int control,
111              IOBUF a, byte *buf, size_t *ret_len)
112 {
113     size_t size = *ret_len;
114     text_filter_context_t *tfx = opaque;
115     int rc=0;
116
117     if( control == IOBUFCTRL_UNDERFLOW ) {
118         rc = standard( tfx, a, buf, size, ret_len );
119     }
120     else if( control == IOBUFCTRL_FREE ) {
121         if( tfx->truncated )
122             log_error(_("can't handle text lines longer than %d characters\n"),
123                         MAX_LINELEN );
124         m_free( tfx->buffer );
125         tfx->buffer = NULL;
126     }
127     else if( control == IOBUFCTRL_DESC )
128         *(char**)buf = "text_filter";
129     return rc;
130 }
131
132
133 /****************
134  * Copy data from INP to OUT and do some escaping if requested.
135  * md is updated as required by rfc2440
136  */
137 int
138 copy_clearsig_text( IOBUF out, IOBUF inp, MD_HANDLE md,
139                                int escape_dash, int escape_from )
140 {
141     unsigned maxlen;
142     byte *buffer = NULL;    /* malloced buffer */
143     unsigned bufsize;       /* and size of this buffer */
144     unsigned n;
145     int truncated = 0;
146     int pending_lf = 0;
147
148     if( !escape_dash )
149         escape_from = 0;
150
151     for(;;) {
152         maxlen = MAX_LINELEN;
153         n = iobuf_read_line( inp, &buffer, &bufsize, &maxlen );
154         if( !maxlen )
155             truncated++;
156
157         if( !n )
158             break; /* read_line has returned eof */
159
160         /* update the message digest */
161         if( escape_dash ) {
162             if( pending_lf ) {
163                 md_putc( md, '\r' );
164                 md_putc( md, '\n' );
165             }
166             md_write( md, buffer, len_without_trailing_ws( buffer, n ) );
167         }
168         else
169             md_write( md, buffer, n );
170         pending_lf = buffer[n-1] == '\n';
171
172         /* write the output */
173         if(    ( escape_dash && *buffer == '-')
174             || ( escape_from && n > 4 && !memcmp(buffer, "From ", 5 ) ) ) {
175             iobuf_put( out, '-' );
176             iobuf_put( out, ' ' );
177         }
178         iobuf_write( out, buffer, n );
179     }
180
181     /* at eof */
182     if( !pending_lf ) { /* make sure that the file ends with a LF */
183         iobuf_put( out, '\n');
184         if( !escape_dash )
185             md_putc( md, '\n' );
186     }
187
188     if( truncated )
189         log_info(_("input line longer than %d characters\n"), MAX_LINELEN );
190
191     return 0; /* okay */
192 }
193