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