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