* photoid.c (generate_photo_id): Enable readline completion and tilde
[gnupg.git] / g10 / photoid.c
1 /* photoid.c - photo ID handling code
2  * Copyright (C) 2001, 2002, 2005 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
19  * USA.
20  */
21
22 #include <config.h>
23 #include <errno.h>
24 #include <stdio.h>
25 #include <string.h>
26 #ifdef _WIN32 
27 # include <windows.h>
28 # ifndef VER_PLATFORM_WIN32_WINDOWS
29 #  define VER_PLATFORM_WIN32_WINDOWS 1
30 # endif
31 #endif
32 #include "packet.h"
33 #include "status.h"
34 #include "exec.h"
35 #include "keydb.h"
36 #include "util.h"
37 #include "i18n.h"
38 #include "iobuf.h"
39 #include "memory.h"
40 #include "options.h"
41 #include "main.h"
42 #include "photoid.h"
43 #include "ttyio.h"
44
45 /* Generate a new photo id packet, or return NULL if canceled */
46 PKT_user_id *
47 generate_photo_id(PKT_public_key *pk)
48 {
49   PKT_user_id *uid;
50   int error=1,i;
51   unsigned int len;
52   char *filename=NULL;
53   byte *photo=NULL;
54   byte header[16];
55   IOBUF file;
56
57   header[0]=0x10; /* little side of photo header length */
58   header[1]=0;    /* big side of photo header length */
59   header[2]=1;    /* 1 == version of photo header */
60   header[3]=1;    /* 1 == JPEG */
61
62   for(i=4;i<16;i++) /* The reserved bytes */
63     header[i]=0;
64
65 #define EXTRA_UID_NAME_SPACE 71
66   uid=xmalloc_clear(sizeof(*uid)+71);
67
68   tty_printf(_("\nPick an image to use for your photo ID.  "
69            "The image must be a JPEG file.\n"
70            "Remember that the image is stored within your public key.  "
71            "If you use a\n"
72            "very large picture, your key will become very large as well!\n"
73            "Keeping the image close to 240x288 is a good size to use.\n"));
74
75   while(photo==NULL)
76     {
77       char *tempname;
78
79       tty_printf("\n");
80
81       xfree(filename);
82
83       tty_enable_completion(NULL);
84
85       tempname=cpr_get("photoid.jpeg.add",
86                        _("Enter JPEG filename for photo ID: "));
87
88       tty_disable_completion();
89
90       filename=make_filename(tempname,(void *)NULL);
91
92       xfree(tempname);
93
94       if(strlen(filename)==0)
95         goto scram;
96
97       file=iobuf_open(filename);
98       if (file && is_secured_file (iobuf_get_fd (file)))
99         {
100           iobuf_close (file);
101           file = NULL;
102           errno = EPERM;
103         }
104       if(!file)
105         {
106           log_error(_("unable to open JPEG file `%s': %s\n"),
107                     filename,strerror(errno));
108           continue;
109         }
110
111       len=iobuf_get_filelength(file);
112       if(len>6144)
113         {
114           tty_printf( _("This JPEG is really large (%d bytes) !\n"),len);
115           if(!cpr_get_answer_is_yes("photoid.jpeg.size",
116                             _("Are you sure you want to use it? (y/N) ")))
117           {
118             iobuf_close(file);
119             continue;
120           }
121         }
122
123       photo=xmalloc(len);
124       iobuf_read(file,photo,len);
125       iobuf_close(file);
126
127       /* Is it a JPEG? */
128       if(photo[0]!=0xFF || photo[1]!=0xD8 ||
129          photo[6]!='J' || photo[7]!='F' || photo[8]!='I' || photo[9]!='F')
130         {
131           log_error(_("`%s' is not a JPEG file\n"),filename);
132           xfree(photo);
133           photo=NULL;
134           continue;
135         }
136
137       /* Build the packet */
138       build_attribute_subpkt(uid,1,photo,len,header,16);
139       parse_attribute_subpkts(uid);
140       make_attribute_uidname(uid, EXTRA_UID_NAME_SPACE);
141
142       /* Showing the photo is not safe when noninteractive since the
143          "user" may not be able to dismiss a viewer window! */
144       if(opt.command_fd==-1)
145         {
146           show_photos(uid->attribs,uid->numattribs,pk,NULL);
147           switch(cpr_get_answer_yes_no_quit("photoid.jpeg.okay",
148                                          _("Is this photo correct (y/N/q)? ")))
149             {
150             case -1:
151               goto scram;
152             case 0:
153               free_attributes(uid);
154               xfree(photo);
155               photo=NULL;
156               continue;
157             }
158         }
159     }
160
161   error=0;
162   uid->ref=1;
163
164  scram:
165   xfree(filename);
166   xfree(photo);
167
168   if(error)
169     {
170       free_attributes(uid);
171       xfree(uid);
172       return NULL;
173     }
174
175   return uid;
176 }
177
178 /* Returns 0 for error, 1 for valid */
179 int parse_image_header(const struct user_attribute *attr,byte *type,u32 *len)
180 {
181   u16 headerlen;
182
183   if(attr->len<3)
184     return 0;
185
186   /* For historical reasons (i.e. "oops!"), the header length is
187      little endian. */
188   headerlen=(attr->data[1]<<8) | attr->data[0];
189
190   if(headerlen>attr->len)
191     return 0;
192
193   if(type && attr->len>=4)
194     {
195       if(attr->data[2]==1) /* header version 1 */
196         *type=attr->data[3];
197       else
198         *type=0;
199     }
200
201   *len=attr->len-headerlen;
202
203   if(*len==0)
204     return 0;
205
206   return 1;
207 }
208
209 /* style==0 for extension, 1 for name, 2 for MIME type.  Remember that
210    the "name" style string could be used in a user ID name field, so
211    make sure it is not too big (see parse-packet.c:parse_attribute).
212    Extensions should be 3 characters long for the best cross-platform
213    compatibility. */
214 char *image_type_to_string(byte type,int style)
215 {
216   char *string;
217
218   switch(type)
219     {
220     case 1: /* jpeg */
221       if(style==0)
222         string="jpg";
223       else if(style==1)
224         string="jpeg";
225       else
226         string="image/jpeg";
227       break;
228
229     default:
230       if(style==0)
231         string="bin";
232       else if(style==1)
233         string="unknown";
234       else
235         string="image/x-unknown";
236       break;
237     }
238
239   return string;
240 }
241
242 #if !defined(FIXED_PHOTO_VIEWER) && !defined(DISABLE_PHOTO_VIEWER)
243 static const char *get_default_photo_command(void)
244 {
245 #if defined(_WIN32)
246   OSVERSIONINFO osvi;
247
248   memset(&osvi,0,sizeof(osvi));
249   osvi.dwOSVersionInfoSize=sizeof(osvi);
250   GetVersionEx(&osvi);
251
252   if(osvi.dwPlatformId==VER_PLATFORM_WIN32_WINDOWS)
253     return "start /w %i";
254   else
255     return "cmd /c start /w %i";
256 #elif defined(__APPLE__)
257   /* OS X.  This really needs more than just __APPLE__. */
258   return "open %I";
259 #elif defined(__riscos__)
260   return "Filer_Run %I";
261 #else
262   return "xloadimage -fork -quiet -title 'KeyID 0x%k' stdin";
263 #endif
264 }
265 #endif
266
267 void show_photos(const struct user_attribute *attrs,
268                  int count,PKT_public_key *pk,PKT_secret_key *sk)
269 {
270 #ifndef DISABLE_PHOTO_VIEWER
271   int i;
272   struct expando_args args;
273   u32 len;
274   u32 kid[2]={0,0};
275
276   memset(&args,0,sizeof(args));
277   args.pk=pk;
278   args.sk=sk;
279
280   if(pk)
281     keyid_from_pk(pk,kid);
282   else if(sk)
283     keyid_from_sk(sk,kid);
284
285   for(i=0;i<count;i++)
286     if(attrs[i].type==ATTRIB_IMAGE &&
287        parse_image_header(&attrs[i],&args.imagetype,&len))
288       {
289         char *command,*name;
290         struct exec_info *spawn;
291         int offset=attrs[i].len-len;
292
293 #ifdef FIXED_PHOTO_VIEWER
294         opt.photo_viewer=FIXED_PHOTO_VIEWER;
295 #else
296         if(!opt.photo_viewer)
297           opt.photo_viewer=get_default_photo_command();
298 #endif
299
300         /* make command grow */
301         command=pct_expando(opt.photo_viewer,&args);
302         if(!command)
303           goto fail;
304
305         name=xmalloc(16+strlen(EXTSEP_S)+
306                      strlen(image_type_to_string(args.imagetype,0))+1);
307
308         /* Make the filename.  Notice we are not using the image
309            encoding type for more than cosmetics.  Most external image
310            viewers can handle a multitude of types, and even if one
311            cannot understand a particular type, we have no way to know
312            which.  The spec permits this, by the way. -dms */
313
314 #ifdef USE_ONLY_8DOT3
315         sprintf(name,"%08lX" EXTSEP_S "%s",(ulong)kid[1],
316                 image_type_to_string(args.imagetype,0));
317 #else
318         sprintf(name,"%08lX%08lX" EXTSEP_S "%s",(ulong)kid[0],(ulong)kid[1],
319                 image_type_to_string(args.imagetype,0));
320 #endif
321
322         if(exec_write(&spawn,NULL,command,name,1,1)!=0)
323           {
324             xfree(name);
325             goto fail;
326           }
327
328 #ifdef __riscos__
329         riscos_set_filetype_by_mimetype(spawn->tempfile_in,
330                                         image_type_to_string(args.imagetype,2));
331 #endif
332
333         xfree(name);
334
335         fwrite(&attrs[i].data[offset],attrs[i].len-offset,1,spawn->tochild);
336
337         if(exec_read(spawn)!=0)
338           {
339             exec_finish(spawn);
340             goto fail;
341           }
342
343         if(exec_finish(spawn)!=0)
344           goto fail;
345       }
346
347   return;
348
349  fail:
350   log_error(_("unable to display photo ID!\n"));
351 #endif
352 }