g10/plaintext.c

   1 /* plaintext.c -  process an plaintext packet
   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 "memory.h"
  28 #include "options.h"
  29 #include "packet.h"
  30 #include "ttyio.h"
  31 #include "filter.h"
  32
  33
  34 /****************
  35  * Handle a plaintext packet.  If MFX is not NULL, update the MDs
  36  * Note: we should use the filter stuff here, but we have to add some
  37  *       easy mimic to set a read limit, so we calculate only the
  38  *       bytes from the plaintext.
  39  */
  40 int
  41 handle_plaintext( PKT_plaintext *pt, md_filter_context_t *mfx )
  42 {
  43     char *fname;
  44     FILE *fp = NULL;
  45     int rc = 0;
  46     int c;
  47
  48     /* create the filename as C string */
  49     if( opt.outfile ) {
  50         fname = m_alloc( strlen( opt.outfile ) + 1);
  51         strcpy(fname, opt.outfile );
  52     }
  53     else {
  54         fname = m_alloc( pt->namelen +1 );
  55         memcpy( fname, pt->name, pt->namelen );
  56         fname[pt->namelen] = 0;
  57     }
  58
  59     if( !*fname ) { /* no filename given */
  60         if( opt.outfile_is_stdout )
  61             fp = stdout;
  62         else {
  63             log_error("no outputfile given\n");
  64             goto leave;
  65         }
  66     }
  67     else if( overwrite_filep( fname ) )
  68         goto leave;
  69
  70     if( fp )
  71         ;
  72     else if( !(fp = fopen(fname,"wb")) ) {
  73         log_error("Error creating '%s': %s\n", fname, strerror(errno) );
  74         rc = G10ERR_WRITE_FILE;
  75         goto leave;
  76     }
  77
  78     if( pt->len ) {
  79         for( ; pt->len; pt->len-- ) {
  80             if( (c = iobuf_get(pt->buf)) == -1 ) {
  81                 log_error("Problem reading source\n");
  82                 rc = G10ERR_READ_FILE;
  83                 goto leave;
  84             }
  85             if( mfx->rmd160 )
  86                 rmd160_putchar(mfx->rmd160, c );
  87             if( mfx->md5 )
  88                 md5_putchar(mfx->md5, c );
  89             if( putc( c, fp ) == EOF ) {
  90                 log_error("Error writing to '%s': %s\n", fname, strerror(errno) );
  91                 rc = G10ERR_WRITE_FILE;
  92                 goto leave;
  93             }
  94         }
  95     }
  96     else {
  97         while( (c = iobuf_get(pt->buf)) != -1 ) {
  98             if( mfx->rmd160 )
  99                 rmd160_putchar(mfx->rmd160, c );
 100             if( mfx->md5 )
 101                 md5_putchar(mfx->md5, c );
 102             if( putc( c, fp ) == EOF ) {
 103                 log_error("Error writing to '%s': %s\n",
 104                                             fname, strerror(errno) );
 105                 rc = G10ERR_WRITE_FILE;
 106                 goto leave;
 107             }
 108         }
 109         iobuf_clear_eof(pt->buf);
 110     }
 111
 112     if( fp && fp != stdout && fclose(fp) ) {
 113         log_error("Error closing '%s': %s\n", fname, strerror(errno) );
 114         fp = NULL;
 115         rc = G10ERR_WRITE_FILE;
 116         goto leave;
 117     }
 118     fp = NULL;
 119
 120   leave:
 121     if( fp && fp != stdout )
 122         fclose(fp);
 123     m_free(fname);
 124     return rc;
 125 }
 126