g10/photoid.c

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