LCOV - code coverage report
Current view: top level - tests - run-genkey.c (source / functions) Hit Total Coverage
Test: coverage.info Lines: 0 213 0.0 %
Date: 2016-11-29 15:07:43 Functions: 0 7 0.0 %

          Line data    Source code
       1             : /* run-genkey.c  - Test tool to perform key generation
       2             :  * Copyright (C) 2016 g10 Code GmbH
       3             :  *
       4             :  * This file is part of GPGME.
       5             :  *
       6             :  * GPGME is free software; you can redistribute it and/or modify it
       7             :  * under the terms of the GNU Lesser General Public License as
       8             :  * published by the Free Software Foundation; either version 2.1 of
       9             :  * the License, or (at your option) any later version.
      10             :  *
      11             :  * GPGME is distributed in the hope that it will be useful, but
      12             :  * WITHOUT ANY WARRANTY; without even the implied warranty of
      13             :  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
      14             :  * Lesser General Public License for more details.
      15             :  *
      16             :  * You should have received a copy of the GNU Lesser General Public
      17             :  * License along with this program; if not, see <https://www.gnu.org/licenses/>.
      18             :  */
      19             : 
      20             : /* We need to include config.h so that we know whether we are building
      21             :    with large file system (LFS) support. */
      22             : #ifdef HAVE_CONFIG_H
      23             : #include <config.h>
      24             : #endif
      25             : 
      26             : #include <stdlib.h>
      27             : #include <stdio.h>
      28             : #include <string.h>
      29             : #include <assert.h>
      30             : 
      31             : #include <gpgme.h>
      32             : 
      33             : #define PGM "run-genkey"
      34             : 
      35             : #include "run-support.h"
      36             : 
      37             : 
      38             : static int verbose;
      39             : 
      40             : 
      41             : /* Tokenize STRING using the set of delimiters in DELIM.  Leading
      42             :  * spaces and tabs are removed from all tokens.  The caller must free
      43             :  * the result.
      44             :  *
      45             :  * Returns: A malloced and NULL delimited array with the tokens.  On
      46             :  *          memory error NULL is returned and ERRNO is set.
      47             :  */
      48             : static char **
      49           0 : strtokenize (const char *string, const char *delim)
      50             : {
      51             :   const char *s;
      52             :   size_t fields;
      53             :   size_t bytes, n;
      54             :   char *buffer;
      55             :   char *p, *px, *pend;
      56             :   char **result;
      57             : 
      58             :   /* Count the number of fields.  */
      59           0 :   for (fields = 1, s = strpbrk (string, delim); s; s = strpbrk (s + 1, delim))
      60           0 :     fields++;
      61           0 :   fields++; /* Add one for the terminating NULL.  */
      62             : 
      63             :   /* Allocate an array for all fields, a terminating NULL, and space
      64             :      for a copy of the string.  */
      65           0 :   bytes = fields * sizeof *result;
      66           0 :   if (bytes / sizeof *result != fields)
      67             :     {
      68           0 :       gpg_err_set_errno (ENOMEM);
      69           0 :       return NULL;
      70             :     }
      71           0 :   n = strlen (string) + 1;
      72           0 :   bytes += n;
      73           0 :   if (bytes < n)
      74             :     {
      75           0 :       gpg_err_set_errno (ENOMEM);
      76           0 :       return NULL;
      77             :     }
      78           0 :   result = malloc (bytes);
      79           0 :   if (!result)
      80           0 :     return NULL;
      81           0 :   buffer = (char*)(result + fields);
      82             : 
      83             :   /* Copy and parse the string.  */
      84           0 :   strcpy (buffer, string);
      85           0 :   for (n = 0, p = buffer; (pend = strpbrk (p, delim)); p = pend + 1)
      86             :     {
      87           0 :       *pend = 0;
      88           0 :       while (*p == ' ' || *p == '\t')
      89           0 :         p++;
      90           0 :       for (px = pend - 1; px >= p && (*px == ' ' || *px == '\t'); px--)
      91           0 :         *px = 0;
      92           0 :       result[n++] = p;
      93             :     }
      94           0 :   while (*p == ' ' || *p == '\t')
      95           0 :     p++;
      96           0 :   for (px = p + strlen (p) - 1; px >= p && (*px == ' ' || *px == '\t'); px--)
      97           0 :     *px = 0;
      98           0 :   result[n++] = p;
      99           0 :   result[n] = NULL;
     100             : 
     101           0 :   assert ((char*)(result + n + 1) == buffer);
     102             : 
     103           0 :   return result;
     104             : }
     105             : 
     106             : 
     107             : static gpg_error_t
     108           0 : status_cb (void *opaque, const char *keyword, const char *value)
     109             : {
     110             :   (void)opaque;
     111           0 :   fprintf (stderr, "status_cb: %s %s\n", nonnull(keyword), nonnull(value));
     112           0 :   return 0;
     113             : }
     114             : 
     115             : 
     116             : static void
     117           0 : progress_cb (void *opaque, const char *what, int type, int current, int total)
     118             : {
     119             :   (void)opaque;
     120             :   (void)type;
     121             : 
     122           0 :   if (total)
     123           0 :     fprintf (stderr, "progress for '%s' %u%% (%d of %d)\n",
     124             :              nonnull (what),
     125           0 :              (unsigned)(((double)current / total) * 100), current, total);
     126             :   else
     127           0 :     fprintf (stderr, "progress for '%s' %d\n", nonnull(what), current);
     128           0 :   fflush (stderr);
     129           0 : }
     130             : 
     131             : 
     132             : static unsigned long
     133           0 : parse_expire_string (const char *string)
     134             : {
     135             :   unsigned long seconds;
     136             : 
     137           0 :   if (!string || !*string || !strcmp (string, "none")
     138           0 :       || !strcmp (string, "never") || !strcmp (string, "-"))
     139           0 :     seconds = 0;
     140           0 :   else if (strspn (string, "01234567890") == strlen (string))
     141           0 :     seconds = strtoul (string, NULL, 10);
     142             :   else
     143             :     {
     144           0 :       fprintf (stderr, PGM ": invalid value '%s'\n", string);
     145           0 :       exit (1);
     146             :     }
     147             : 
     148           0 :   return seconds;
     149             : }
     150             : 
     151             : 
     152             : /* Parse a usage string and return flags for gpgme_op_createkey.  */
     153             : static unsigned int
     154           0 : parse_usage_string (const char *string)
     155             : {
     156             :   gpg_error_t err;
     157           0 :   char **tokens = NULL;
     158             :   const char *s;
     159             :   int i;
     160           0 :   unsigned int flags = 0;
     161             : 
     162           0 :   tokens = strtokenize (string, " \t,");
     163           0 :   if (!tokens)
     164             :     {
     165           0 :       err = gpg_error_from_syserror ();
     166           0 :       fprintf (stderr, PGM": strtokenize failed: %s\n", gpg_strerror (err));
     167           0 :       exit (1);
     168             :     }
     169             : 
     170           0 :   for (i=0; (s = tokens[i]); i++)
     171             :     {
     172           0 :       if (!*s)
     173             :         ;
     174           0 :       else if (!strcmp (s, "default"))
     175             :         ;
     176           0 :       else if (!strcmp (s, "sign"))
     177           0 :         flags |= GPGME_CREATE_SIGN;
     178           0 :       else if (!strcmp (s, "encr"))
     179           0 :         flags |= GPGME_CREATE_ENCR;
     180           0 :       else if (!strcmp (s, "cert"))
     181           0 :         flags |= GPGME_CREATE_CERT;
     182           0 :       else if (!strcmp (s, "auth"))
     183           0 :         flags |= GPGME_CREATE_AUTH;
     184             :       else
     185             :         {
     186           0 :           free (tokens);
     187           0 :           fprintf (stderr, PGM": invalid value '%s': %s\n",
     188             :                    string, "bad usage");
     189           0 :           exit (1);
     190             :         }
     191             :     }
     192             : 
     193           0 :   free (tokens);
     194           0 :   return flags;
     195             : }
     196             : 
     197             : 
     198             : 
     199             : static int
     200           0 : show_usage (int ex)
     201             : {
     202           0 :   fputs ("usage: " PGM " [options] ARGS\n"
     203             :          "         args: USERID [ALGO [USAGE [EXPIRESECONDS]]]\n"
     204             :          "   for addkey: FPR    [ALGO [USAGE [EXPIRESECONDS]]]\n"
     205             :          "   for adduid: FPR    USERID\n"
     206             :          "   for revuid: FPR    USERID\n"
     207             :          "Options:\n"
     208             :          "  --addkey         add a subkey to the key with FPR\n"
     209             :          "  --adduid         add a user id to the key with FPR\n"
     210             :          "  --revuid         Revoke a user id from the key with FPR\n"
     211             :          "  --verbose        run in verbose mode\n"
     212             :          "  --status         print status lines from the backend\n"
     213             :          "  --progress       print progress info\n"
     214             :          "  --openpgp        use the OpenPGP protocol (default)\n"
     215             :          "  --cms            use the CMS protocol\n"
     216             :          "  --loopback       use a loopback pinentry\n"
     217             :          "  --unprotected    do not use a passphrase\n"
     218             :          "  --force          do not check for a duplicated user id\n"
     219             :          , stderr);
     220           0 :   exit (ex);
     221             : }
     222             : 
     223             : 
     224             : int
     225           0 : main (int argc, char **argv)
     226             : {
     227           0 :   int last_argc = -1;
     228             :   gpgme_error_t err;
     229             :   gpgme_ctx_t ctx;
     230           0 :   gpgme_protocol_t protocol = GPGME_PROTOCOL_OpenPGP;
     231           0 :   int print_status = 0;
     232           0 :   int print_progress = 0;
     233           0 :   int use_loopback = 0;
     234           0 :   int addkey = 0;
     235           0 :   int adduid = 0;
     236           0 :   int revuid = 0;
     237             :   const char *userid;
     238           0 :   const char *algo = NULL;
     239           0 :   const char *newuserid = NULL;
     240           0 :   unsigned int flags = 0;
     241           0 :   unsigned long expire = 0;
     242             :   gpgme_genkey_result_t result;
     243             : 
     244           0 :   if (argc)
     245           0 :     { argc--; argv++; }
     246             : 
     247           0 :   while (argc && last_argc != argc )
     248             :     {
     249           0 :       last_argc = argc;
     250           0 :       if (!strcmp (*argv, "--"))
     251             :         {
     252           0 :           argc--; argv++;
     253           0 :           break;
     254             :         }
     255           0 :       else if (!strcmp (*argv, "--help"))
     256           0 :         show_usage (0);
     257           0 :       else if (!strcmp (*argv, "--addkey"))
     258             :         {
     259           0 :           addkey = 1;
     260           0 :           adduid = 0;
     261           0 :           revuid = 0;
     262           0 :           argc--; argv++;
     263             :         }
     264           0 :       else if (!strcmp (*argv, "--adduid"))
     265             :         {
     266           0 :           addkey = 0;
     267           0 :           adduid = 1;
     268           0 :           revuid = 0;
     269           0 :           argc--; argv++;
     270             :         }
     271           0 :       else if (!strcmp (*argv, "--revuid"))
     272             :         {
     273           0 :           addkey = 0;
     274           0 :           adduid = 0;
     275           0 :           revuid = 1;
     276           0 :           argc--; argv++;
     277             :         }
     278           0 :       else if (!strcmp (*argv, "--verbose"))
     279             :         {
     280           0 :           verbose = 1;
     281           0 :           argc--; argv++;
     282             :         }
     283           0 :       else if (!strcmp (*argv, "--status"))
     284             :         {
     285           0 :           print_status = 1;
     286           0 :           argc--; argv++;
     287             :         }
     288           0 :       else if (!strcmp (*argv, "--progress"))
     289             :         {
     290           0 :           print_progress = 1;
     291           0 :           argc--; argv++;
     292             :         }
     293           0 :       else if (!strcmp (*argv, "--openpgp"))
     294             :         {
     295           0 :           protocol = GPGME_PROTOCOL_OpenPGP;
     296           0 :           argc--; argv++;
     297             :         }
     298           0 :       else if (!strcmp (*argv, "--cms"))
     299             :         {
     300           0 :           protocol = GPGME_PROTOCOL_CMS;
     301           0 :           argc--; argv++;
     302             :         }
     303           0 :       else if (!strcmp (*argv, "--loopback"))
     304             :         {
     305           0 :           use_loopback = 1;
     306           0 :           argc--; argv++;
     307             :         }
     308           0 :       else if (!strcmp (*argv, "--unprotected"))
     309             :         {
     310           0 :           flags |= GPGME_CREATE_NOPASSWD;
     311           0 :           argc--; argv++;
     312             :         }
     313           0 :       else if (!strcmp (*argv, "--force"))
     314             :         {
     315           0 :           flags |= GPGME_CREATE_FORCE;
     316           0 :           argc--; argv++;
     317             :         }
     318           0 :       else if (!strncmp (*argv, "--", 2))
     319           0 :         show_usage (1);
     320             :     }
     321             : 
     322           0 :   if (adduid || revuid)
     323             :     {
     324           0 :       if (argc != 2)
     325           0 :         show_usage (1);
     326           0 :       userid = argv[0];
     327           0 :       newuserid = argv[1];
     328             :     }
     329             :   else
     330             :     {
     331           0 :       if (!argc || argc > 4)
     332           0 :         show_usage (1);
     333           0 :       userid = argv[0];
     334           0 :       if (argc > 1)
     335           0 :         algo = argv[1];
     336           0 :       if (argc > 2)
     337           0 :         flags |= parse_usage_string (argv[2]);
     338           0 :       if (argc > 3)
     339           0 :         expire = parse_expire_string (argv[3]);
     340             :     }
     341             : 
     342           0 :   init_gpgme (protocol);
     343             : 
     344           0 :   err = gpgme_new (&ctx);
     345           0 :   fail_if_err (err);
     346           0 :   gpgme_set_protocol (ctx, protocol);
     347           0 :   gpgme_set_armor (ctx, 1);
     348           0 :   if (print_status)
     349             :     {
     350           0 :       gpgme_set_status_cb (ctx, status_cb, NULL);
     351           0 :       gpgme_set_ctx_flag (ctx, "full-status", "1");
     352             :     }
     353           0 :   if (print_progress)
     354           0 :     gpgme_set_progress_cb (ctx, progress_cb, NULL);
     355           0 :   if (use_loopback)
     356             :     {
     357           0 :       gpgme_set_pinentry_mode (ctx, GPGME_PINENTRY_MODE_LOOPBACK);
     358           0 :       gpgme_set_passphrase_cb (ctx, passphrase_cb, NULL);
     359             :     }
     360             : 
     361           0 :   if (addkey || adduid || revuid)
     362           0 :     {
     363             :       gpgme_key_t akey;
     364             : 
     365           0 :       err = gpgme_get_key (ctx, userid, &akey, 1);
     366           0 :       if (err)
     367             :         {
     368           0 :           fprintf (stderr, PGM ": error getting secret key for '%s': %s\n",
     369             :                    userid, gpg_strerror (err));
     370           0 :           exit (1);
     371             :         }
     372             : 
     373           0 :       if (addkey)
     374             :         {
     375           0 :           err = gpgme_op_createsubkey (ctx, akey, algo, 0, expire, flags);
     376           0 :           if (err)
     377             :             {
     378           0 :               fprintf (stderr, PGM ": gpgme_op_createsubkey failed: %s\n",
     379             :                        gpg_strerror (err));
     380           0 :               exit (1);
     381             :             }
     382             :         }
     383           0 :       else if (adduid)
     384             :         {
     385           0 :           err = gpgme_op_adduid (ctx, akey, newuserid, flags);
     386           0 :           if (err)
     387             :             {
     388           0 :               fprintf (stderr, PGM ": gpgme_op_adduid failed: %s\n",
     389             :                        gpg_strerror (err));
     390           0 :               exit (1);
     391             :             }
     392             :         }
     393           0 :       else if (revuid)
     394             :         {
     395           0 :           err = gpgme_op_revuid (ctx, akey, newuserid, flags);
     396           0 :           if (err)
     397             :             {
     398           0 :               fprintf (stderr, PGM ": gpgme_op_revuid failed: %s\n",
     399             :                        gpg_strerror (err));
     400           0 :               exit (1);
     401             :             }
     402             :         }
     403           0 :       gpgme_key_unref (akey);
     404             :     }
     405             :   else
     406             :     {
     407           0 :       err = gpgme_op_createkey (ctx, userid, algo, 0, expire, NULL, flags);
     408           0 :       if (err)
     409             :         {
     410           0 :           fprintf (stderr, PGM ": gpgme_op_createkey failed: %s\n",
     411             :                    gpg_strerror (err));
     412           0 :           exit (1);
     413             :         }
     414             :     }
     415             : 
     416           0 :   result = gpgme_op_genkey_result (ctx);
     417           0 :   if (!result)
     418             :     {
     419           0 :       fprintf (stderr, PGM": gpgme_op_genkey_result returned NULL\n");
     420           0 :       exit (1);
     421             :     }
     422             : 
     423           0 :   printf ("Generated key: %s (%s)\n",
     424           0 :           result->fpr ? result->fpr : "none",
     425           0 :           result->primary ? (result->sub ? "primary, sub" : "primary")
     426           0 :           /**/            : (result->sub ? "sub" : "none"));
     427             : 
     428           0 :   if (result->fpr && strlen (result->fpr) < 40)
     429           0 :     fprintf (stderr, PGM": generated key has unexpected fingerprint\n");
     430           0 :   if (!result->primary)
     431           0 :     fprintf (stderr, PGM": primary key was not generated\n");
     432           0 :   if (!result->sub)
     433           0 :     fprintf (stderr, PGM": sub key was not generated\n");
     434           0 :   if (!result->uid)
     435           0 :     fprintf (stderr, PGM": uid was not generated\n");
     436             : 
     437           0 :   gpgme_release (ctx);
     438           0 :   return 0;
     439             : }

Generated by: LCOV version 1.11