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