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