Minor fixes
[gnupg.git] / g10 / openfile.c
1 /* openfile.c
2  * Copyright (C) 1998, 1999, 2000, 2001, 2003 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 <assert.h>
26 #include <errno.h>
27 #include <sys/types.h>
28 #include <sys/stat.h>
29 #include <fcntl.h>
30 #include <unistd.h>
31 #include "util.h"
32 #include "memory.h"
33 #include "ttyio.h"
34 #include "options.h"
35 #include "main.h"
36 #include "status.h"
37 #include "i18n.h"
38
39 #ifdef USE_ONLY_8DOT3
40 #define SKELEXT ".skl"
41 #else
42 #define SKELEXT EXTSEP_S "skel"
43 #endif
44
45 #if defined (HAVE_DRIVE_LETTERS) || defined (__riscos__)
46 #define CMP_FILENAME(a,b) ascii_strcasecmp( (a), (b) )
47 #else
48 #define CMP_FILENAME(a,b) strcmp( (a), (b) )
49 #endif
50
51 #ifdef MKDIR_TAKES_ONE_ARG
52 #undef mkdir
53 #define mkdir(a,b) mkdir(a)
54 #endif
55
56 /* FIXME:  Implement opt.interactive. */
57
58 /****************
59  * Check whether FNAME exists and ask if it's okay to overwrite an
60  * existing one.
61  * Returns: True: it's okay to overwrite or the file does not exist
62  *          False: Do not overwrite
63  */
64 int
65 overwrite_filep( const char *fname )
66 {
67     if( iobuf_is_pipe_filename (fname) )
68         return 1; /* Writing to stdout is always okay */
69
70     if( access( fname, F_OK ) )
71         return 1; /* does not exist */
72
73 #ifndef HAVE_DOSISH_SYSTEM
74     if ( !strcmp ( fname, "/dev/null" ) )
75         return 1; /* does not do any harm */
76 #endif
77
78     /* fixme: add some backup stuff in case of overwrite */
79     if( opt.answer_yes )
80         return 1;
81     if( opt.answer_no || opt.batch )
82         return 0;  /* do not overwrite */
83
84     tty_printf(_("File `%s' exists. "), fname);
85     if( cpr_get_answer_is_yes("openfile.overwrite.okay",
86                                _("Overwrite? (y/N) ")) )
87         return 1;
88     return 0;
89 }
90
91
92 /****************
93  * Strip know extensions from iname and return a newly allocated
94  * filename.  Return NULL if we can't do that.
95  */
96 char *
97 make_outfile_name( const char *iname )
98 {
99     size_t n;
100
101     if ( iobuf_is_pipe_filename (iname) )
102         return m_strdup("-");
103
104     n = strlen(iname);
105     if( n > 4 && (    !CMP_FILENAME(iname+n-4, EXTSEP_S "gpg")
106                    || !CMP_FILENAME(iname+n-4, EXTSEP_S "pgp")
107                    || !CMP_FILENAME(iname+n-4, EXTSEP_S "sig")
108                    || !CMP_FILENAME(iname+n-4, EXTSEP_S "asc") ) ) {
109         char *buf = m_strdup( iname );
110         buf[n-4] = 0;
111         return buf;
112     }
113     else if( n > 5 && !CMP_FILENAME(iname+n-5, EXTSEP_S "sign") ) {
114         char *buf = m_strdup( iname );
115         buf[n-5] = 0;
116         return buf;
117     }
118
119     log_info(_("%s: unknown suffix\n"), iname );
120     return NULL;
121 }
122
123
124 /****************
125  * Ask for a outputfilename and use the given one as default.
126  * Return NULL if no file has been given or it is not possible to
127  * ask the user.
128  */
129 char *
130 ask_outfile_name( const char *name, size_t namelen )
131 {
132     size_t n;
133     const char *s;
134     char *prompt;
135     char *fname;
136     char *defname;
137
138     if( opt.batch )
139         return NULL;
140
141     s = _("Enter new filename");
142
143     n = strlen(s) + namelen + 10;
144     defname = name && namelen? make_printable_string( name, namelen, 0): NULL;
145     prompt = m_alloc(n);
146     if( defname )
147         sprintf(prompt, "%s [%s]: ", s, defname );
148     else
149         sprintf(prompt, "%s: ", s );
150     fname = cpr_get("openfile.askoutname", prompt );
151     cpr_kill_prompt();
152     m_free(prompt);
153     if( !*fname ) {
154         m_free( fname ); fname = NULL;
155         fname = defname; defname = NULL;
156     }
157     m_free(defname);
158     if (fname)
159         trim_spaces (fname);
160     return fname;
161 }
162
163
164 /****************
165  * Make an output filename for the inputfile INAME.
166  * Returns an IOBUF and an errorcode
167  * Mode 0 = use ".gpg"
168  *      1 = use ".asc"
169  *      2 = use ".sig"
170  */
171 int
172 open_outfile( const char *iname, int mode, IOBUF *a )
173 {
174   int rc = 0;
175
176   *a = NULL;
177   if( iobuf_is_pipe_filename (iname) && !opt.outfile ) {
178     *a = iobuf_create(NULL);
179     if( !*a ) {
180       log_error(_("can't open `%s': %s\n"), "[stdout]", strerror(errno) );
181       rc = G10ERR_CREATE_FILE;
182     }
183     else if( opt.verbose )
184       log_info(_("writing to stdout\n"));
185   }
186   else {
187     char *buf = NULL;
188     const char *name;
189     
190     if( opt.dry_run )
191       name = "/dev/null";
192     else if( opt.outfile )
193       name = opt.outfile;
194     else {
195 #ifdef USE_ONLY_8DOT3
196       if (opt.mangle_dos_filenames)
197         {
198           /* It is quite common DOS system to have only one dot in a
199            * a filename So if we have something like this, we simple
200            * replace the suffix execpt in cases where the suffix is
201            * larger than 3 characters and not the same as.
202            * We should really map the filenames to 8.3 but this tends to
203            * be more complicated and is probaly a duty of the filesystem
204            */
205           char *dot;
206           const char *newsfx = mode==1 ? ".asc" :
207                                mode==2 ? ".sig" : ".gpg";
208           
209           buf = m_alloc(strlen(iname)+4+1);
210           strcpy(buf,iname);
211           dot = strchr(buf, '.' );
212           if ( dot && dot > buf && dot[1] && strlen(dot) <= 4
213                                   && CMP_FILENAME(newsfx, dot) )
214             {
215               strcpy(dot, newsfx );
216             }
217           else if ( dot && !dot[1] ) /* don't duplicate a dot */
218             strcpy( dot, newsfx+1 );
219           else
220             strcat ( buf, newsfx );
221         }
222       if (!buf)
223 #endif /* USE_ONLY_8DOT3 */
224         {
225           buf = m_alloc(strlen(iname)+4+1);
226           strcpy(stpcpy(buf,iname), mode==1 ? EXTSEP_S "asc" :
227                                    mode==2 ? EXTSEP_S "sig" : EXTSEP_S "gpg");
228         }
229       name = buf;
230     }
231
232     rc = 0;
233     while( !overwrite_filep (name) )
234       {
235         char *tmp = ask_outfile_name (NULL, 0);
236         if ( !tmp || !*tmp )
237           {
238             m_free (tmp);
239             rc = G10ERR_FILE_EXISTS;
240             break;
241           }
242         m_free (buf);
243         name = buf = tmp;
244       }
245     
246     if( !rc )
247       {
248         if (is_secured_filename (name) )
249           {
250             *a = NULL;
251             errno = EPERM;
252           }
253         else
254           *a = iobuf_create( name );
255         if( !*a )
256           {
257             log_error(_("can't create `%s': %s\n"), name, strerror(errno) );
258             rc = G10ERR_CREATE_FILE;
259           }
260         else if( opt.verbose )
261           log_info(_("writing to `%s'\n"), name );
262       }
263     m_free(buf);
264   }
265
266   return rc;
267 }
268
269
270 /****************
271  * Try to open a file without the extension ".sig" or ".asc"
272  * Return NULL if such a file is not available.
273  */
274 IOBUF
275 open_sigfile( const char *iname, progress_filter_context_t *pfx )
276 {
277     IOBUF a = NULL;
278     size_t len;
279
280     if( !iobuf_is_pipe_filename (iname) ) {
281         len = strlen(iname);
282         if( len > 4 && ( !strcmp(iname + len - 4, EXTSEP_S "sig")
283                         || ( len > 5 && !strcmp(iname + len - 5, EXTSEP_S "sign") )
284                         || !strcmp(iname + len - 4, EXTSEP_S "asc")) ) {
285             char *buf;
286             buf = m_strdup(iname);
287             buf[len-(buf[len-1]=='n'?5:4)] = 0 ;
288             a = iobuf_open( buf );
289             if (a && is_secured_file (iobuf_get_fd (a)))
290               {
291                 iobuf_close (a);
292                 a = NULL;
293                 errno = EPERM;
294               }
295             if( a && opt.verbose )
296                 log_info(_("assuming signed data in `%s'\n"), buf );
297             if (a && pfx)
298               handle_progress (pfx, a, buf);
299             m_free(buf);
300         }
301     }
302     return a;
303 }
304
305 /****************
306  * Copy the option file skeleton to the given directory.
307  */
308 static void
309 copy_options_file( const char *destdir )
310 {
311     const char *datadir = GNUPG_DATADIR;
312     char *fname;
313     FILE *src, *dst;
314     int linefeeds=0;
315     int c;
316     mode_t oldmask;
317     int esc = 0;
318     int any_option = 0;
319
320     if( opt.dry_run )
321         return;
322
323     fname = m_alloc( strlen(datadir) + strlen(destdir) + 15 );
324     strcpy(stpcpy(fname, datadir), DIRSEP_S "options" SKELEXT );
325     src = fopen( fname, "r" );
326     if (src && is_secured_file (fileno (src)))
327       {
328         fclose (src);
329         src = NULL;
330         errno = EPERM;
331       }
332     if( !src ) {
333         log_error(_("can't open `%s': %s\n"), fname, strerror(errno) );
334         m_free(fname);
335         return;
336     }
337     strcpy(stpcpy(fname, destdir), DIRSEP_S "gpg" EXTSEP_S "conf" );
338     oldmask=umask(077);
339     if ( is_secured_filename (fname) )
340       {
341         dst = NULL;
342         errno = EPERM;
343       }
344     else
345       dst = fopen( fname, "w" );
346     umask(oldmask);
347     if( !dst ) {
348         log_error(_("can't create `%s': %s\n"), fname, strerror(errno) );
349         fclose( src );
350         m_free(fname);
351         return;
352     }
353
354     while( (c=getc(src)) != EOF ) {
355         if( linefeeds < 3 ) {
356             if( c == '\n' )
357                 linefeeds++;
358         }
359         else {
360             putc( c, dst );
361             if (c== '\n')
362                 esc = 1;
363             else if (esc == 1) {
364                 if (c == ' ' || c == '\t')
365                     ;
366                 else if (c == '#')
367                     esc = 2;
368                 else 
369                     any_option = 1;
370             }
371         }
372     }
373     fclose( dst );
374     fclose( src );
375     log_info(_("new configuration file `%s' created\n"), fname );
376     if (any_option)
377         log_info (_("WARNING: options in `%s'"
378                     " are not yet active during this run\n"),
379                   fname);
380     m_free(fname);
381 }
382
383
384 void
385 try_make_homedir( const char *fname )
386 {
387     const char *defhome = GNUPG_HOMEDIR;
388
389     /* Create the directory only if the supplied directory name
390      * is the same as the default one.  This way we avoid to create
391      * arbitrary directories when a non-default homedirectory is used.
392      * To cope with HOME, we do compare only the suffix if we see that
393      * the default homedir does start with a tilde.
394      */
395     if( opt.dry_run || opt.no_homedir_creation )
396         return;
397
398     if ( ( *defhome == '~'
399            && ( strlen(fname) >= strlen (defhome+1)
400                 && !strcmp(fname+strlen(fname)-strlen(defhome+1),
401                            defhome+1 ) ))
402          || ( *defhome != '~'
403               && !compare_filenames( fname, defhome ) )
404         ) {
405         if( mkdir( fname, S_IRUSR|S_IWUSR|S_IXUSR ) )
406             log_fatal( _("can't create directory `%s': %s\n"),
407                                         fname,  strerror(errno) );
408         else if( !opt.quiet )
409             log_info( _("directory `%s' created\n"), fname );
410         copy_options_file( fname );
411 /*      log_info(_("you have to start GnuPG again, " */
412 /*                 "so it can read the new configuration file\n") ); */
413 /*      g10_exit(1); */
414     }
415 }