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