g10/plaintext.c

   1 /* plaintext.c -  process an plaintext packet
   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 "memory.h"
  28 #include "options.h"
  29 #include "packet.h"
  30 #include "ttyio.h"
  31 #include "filter.h"
  32 #include "main.h"
  33 #include "status.h"
  34 #include "i18n.h"
  35
  36
  37 /****************
  38  * Defer the last CR,LF
  39  */
  40 static void
  41 special_md_putc( MD_HANDLE md, int c, int *state )
  42 {
  43     if( c == -1 ) { /* flush */
  44         if( *state == 1 ) {
  45             md_putc(md, '\r');
  46         }
  47
  48         *state = 0;
  49         return;
  50     }
  51   again:
  52     switch( *state ) {
  53       case 0:
  54         if( c == '\r' )
  55             *state = 1;
  56         else
  57             md_putc(md, c );
  58         break;
  59       case 1:
  60         if( c == '\n' )
  61             *state = 2;
  62         else {
  63             md_putc(md, '\r');
  64             *state = 0;
  65             goto again;
  66         }
  67         break;
  68       case 2:
  69         md_putc(md, '\r');
  70         md_putc(md, '\n');
  71         *state = 0;
  72         goto again;
  73       default: BUG();
  74     }
  75 }
  76
  77
  78 /****************
  79  * Handle a plaintext packet.  If MFX is not NULL, update the MDs
  80  * Note: we should use the filter stuff here, but we have to add some
  81  *       easy mimic to set a read limit, so we calculate only the
  82  *       bytes from the plaintext.
  83  */
  84 int
  85 handle_plaintext( PKT_plaintext *pt, md_filter_context_t *mfx,
  86                   int nooutput, int clearsig )
  87 {
  88     char *fname = NULL;
  89     FILE *fp = NULL;
  90     int rc = 0;
  91     int c;
  92     int convert = pt->mode == 't';
  93     int special_state = 0;
  94
  95     /* create the filename as C string */
  96     if( nooutput )
  97         ;
  98     else if( opt.outfile ) {
  99         fname = m_alloc( strlen( opt.outfile ) + 1);
 100         strcpy(fname, opt.outfile );
 101     }
 102     else if( pt->namelen == 8 && !memcmp( pt->name, "_CONSOLE", 8 ) ) {
 103         log_info(_("data not saved; use option \"--output\" to save it\n"));
 104         nooutput = 1;
 105     }
 106     else {
 107         fname = m_alloc( pt->namelen +1 );
 108         memcpy( fname, pt->name, pt->namelen );
 109         fname[pt->namelen] = 0;
 110     }
 111
 112     if( nooutput )
 113         ;
 114     else if( !*fname || (*fname=='-' && !fname[1])) {
 115         /* no filename or "-" given; write to stdout */
 116         fp = stdout;
 117     }
 118     else if( !overwrite_filep( fname ) ) {
 119         rc = G10ERR_CREATE_FILE;
 120         goto leave;
 121     }
 122
 123     if( fp || nooutput )
 124         ;
 125     else if( !(fp = fopen(fname,"wb")) ) {
 126         log_error("Error creating '%s': %s\n", fname, strerror(errno) );
 127         rc = G10ERR_CREATE_FILE;
 128         goto leave;
 129     }
 130
 131     if( pt->len ) {
 132         for( ; pt->len; pt->len-- ) {
 133             if( (c = iobuf_get(pt->buf)) == -1 ) {
 134                 log_error("Problem reading source (%u bytes remaining)\n",
 135                                                          (unsigned)pt->len);
 136                 rc = G10ERR_READ_FILE;
 137                 goto leave;
 138             }
 139             if( mfx->md ) {
 140                 if( convert && clearsig )
 141                     special_md_putc(mfx->md, c, &special_state );
 142                 else
 143                     md_putc(mfx->md, c );
 144             }
 145             if( convert && !clearsig && c == '\r' )
 146                 continue; /* fixme: this hack might be too simple */
 147             if( fp ) {
 148                 if( putc( c, fp ) == EOF ) {
 149                     log_error("Error writing to '%s': %s\n",
 150                                             fname, strerror(errno) );
 151                     rc = G10ERR_WRITE_FILE;
 152                     goto leave;
 153                 }
 154             }
 155         }
 156     }
 157     else {
 158         while( (c = iobuf_get(pt->buf)) != -1 ) {
 159             if( mfx->md ) {
 160                 if( convert && clearsig )
 161                     special_md_putc(mfx->md, c, &special_state  );
 162                 else
 163                     md_putc(mfx->md, c );
 164             }
 165             if( convert && !clearsig && c == '\r' )
 166                 continue; /* fixme: this hack might be too simple */
 167             if( fp ) {
 168                 if( putc( c, fp ) == EOF ) {
 169                     log_error("Error writing to '%s': %s\n",
 170                                                 fname, strerror(errno) );
 171                     rc = G10ERR_WRITE_FILE;
 172                     goto leave;
 173                 }
 174             }
 175         }
 176         iobuf_clear_eof(pt->buf);
 177     }
 178     if( mfx->md && convert && clearsig )
 179         special_md_putc(mfx->md, -1, &special_state  ); /* flush */
 180
 181     if( fp && fp != stdout && fclose(fp) ) {
 182         log_error("Error closing '%s': %s\n", fname, strerror(errno) );
 183         fp = NULL;
 184         rc = G10ERR_WRITE_FILE;
 185         goto leave;
 186     }
 187     fp = NULL;
 188
 189   leave:
 190     if( fp && fp != stdout )
 191         fclose(fp);
 192     m_free(fname);
 193     return rc;
 194 }
 195
 196
 197 /****************
 198  * Ask for the detached datafile and calculate the digest from it.
 199  * INFILE is the name of the input file.
 200  */
 201 int
 202 ask_for_detached_datafile( md_filter_context_t *mfx, const char *inname )
 203 {
 204     char *answer = NULL;
 205     IOBUF fp;
 206     int rc = 0;
 207     int c;
 208
 209     fp = open_sigfile( inname ); /* open default file */
 210     if( !fp && !opt.batch ) {
 211         int any=0;
 212         tty_printf("Detached signature.\n");
 213         do {
 214             m_free(answer);
 215             answer = cpr_get("detached_signature.filename",
 216                            _("Please enter name of data file: "));
 217             cpr_kill_prompt();
 218             if( any && !*answer ) {
 219                 rc = G10ERR_READ_FILE;
 220                 goto leave;
 221             }
 222             fp = iobuf_open(answer);
 223             if( !fp && errno == ENOENT ) {
 224                 tty_printf("No such file, try again or hit enter to quit.\n");
 225                 any++;
 226             }
 227             else if( !fp ) {
 228                 log_error("can't open '%s': %s\n", answer, strerror(errno) );
 229                 rc = G10ERR_READ_FILE;
 230                 goto leave;
 231             }
 232         } while( !fp );
 233     }
 234
 235     if( !fp ) {
 236         if( opt.verbose )
 237             log_info(_("reading stdin ...\n"));
 238         while( (c = getchar()) != EOF ) {
 239             if( mfx->md )
 240                 md_putc(mfx->md, c );
 241         }
 242     }
 243     else {
 244         while( (c = iobuf_get(fp)) != -1 ) {
 245             if( mfx->md )
 246                 md_putc(mfx->md, c );
 247         }
 248         iobuf_close(fp);
 249     }
 250
 251   leave:
 252     m_free(answer);
 253     return rc;
 254 }
 255
 256
 257 static void
 258 do_hash( MD_HANDLE md, IOBUF fp, int textmode )
 259 {
 260     text_filter_context_t tfx;
 261     int c;
 262
 263     if( textmode ) {
 264         memset( &tfx, 0, sizeof tfx);
 265         iobuf_push_filter( fp, text_filter, &tfx );
 266     }
 267     while( (c = iobuf_get(fp)) != -1 )
 268         md_putc(md, c );
 269 }
 270
 271
 272 /****************
 273  * Hash the given files and append the hash to hash context md.
 274  * If FILES is NULL, hash stdin.
 275  */
 276 int
 277 hash_datafiles( MD_HANDLE md, STRLIST files,
 278                 const char *sigfilename, int textmode )
 279 {
 280     IOBUF fp;
 281     STRLIST sl=NULL;
 282
 283     if( !files ) {
 284         /* check whether we can opne the signed material */
 285         fp = open_sigfile( sigfilename );
 286         if( fp ) {
 287             do_hash( md, fp, textmode );
 288             iobuf_close(fp);
 289             return 0;
 290         }
 291         /* no we can't (no sigfile) - read signed stuff from stdin */
 292         add_to_strlist( &sl, "-");
 293     }
 294     else
 295         sl = files;
 296
 297     for( ; sl; sl = sl->next ) {
 298         fp = iobuf_open( sl->d );
 299         if( !fp ) {
 300             log_error(_("can't open signed data '%s'\n"),
 301                                                 print_fname_stdin(sl->d));
 302             if( !files )
 303                 free_strlist(sl);
 304             return G10ERR_OPEN_FILE;
 305         }
 306         do_hash( md, fp, textmode );
 307         iobuf_close(fp);
 308     }
 309
 310     if( !files )
 311         free_strlist(sl);
 312     return 0;
 313 }
 314
 315