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