summaryrefslogtreecommitdiff
path: root/src/certtool-extras.c
blob: 14221885ec6dff3b41f95136d67f2730f5131f3b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
/*
 * Copyright (C) 2012 Lucas Fisher *lucas.fisher [at] gmail.com*
 * Copyright (C) 2012 Free Software Foundation, Inc.
 *
 * This file is part of GnuTLS.
 *
 * GnuTLS is free software: you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * GnuTLS is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see
 * <http://www.gnu.org/licenses/>.
 */

#include <config.h>

#include <gnutls/gnutls.h>
#include <gnutls/x509.h>
#include <gnutls/openpgp.h>
#include <gnutls/pkcs12.h>
#include <gnutls/pkcs11.h>
#include <gnutls/abstract.h>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include <time.h>
#include <unistd.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <error.h>
#include "certtool-common.h"
#include "certtool-cfg.h"


#define MAX_KEYS 256

/* Loads a x509 private key list
 */
gnutls_x509_privkey_t *
load_privkey_list (int mand, size_t * privkey_size, common_info_st * info)
{
  static gnutls_x509_privkey_t key[MAX_KEYS];
  char *ptr;
  int ret, i;
  gnutls_datum_t dat, file_data;
  int ptr_size;
  unsigned int flags = 0;
  const char* pass;

  *privkey_size = 0;
  fprintf (stderr, "Loading private key list...\n");

  if (info->privkey == NULL)
    {
      if (mand)
        error (EXIT_FAILURE, 0, "missing --load-privkey");
      else
        return NULL;
    }

  ret = gnutls_load_file(info->privkey, &file_data);
  if (ret < 0)
    error (EXIT_FAILURE, errno, "%s", info->privkey);

  ptr = (void*)file_data.data;
  ptr_size = file_data.size;

  for (i = 0; i < MAX_KEYS; i++)
    {
      ret = gnutls_x509_privkey_init (&key[i]);
      if (ret < 0)
        error (EXIT_FAILURE, 0, "privkey_init: %s", gnutls_strerror (ret));

      dat.data = (void*)ptr;
      dat.size = ptr_size;

      ret = gnutls_x509_privkey_import2 (key[i], &dat, info->incert_format, NULL, 0);
      if (ret == GNUTLS_E_DECRYPTION_FAILED)
        {
          pass = get_password (info, &flags, 0);
          ret = gnutls_x509_privkey_import2 (key[i], &dat, info->incert_format, pass, flags);
        }

      if (ret < 0 && *privkey_size > 0)
        break;
      if (ret < 0)
        error (EXIT_FAILURE, 0, "privkey_import: %s", gnutls_strerror (ret));

      (*privkey_size)++;

      if (info->incert_format != GNUTLS_X509_FMT_PEM)
      	break;

      ptr = strstr (ptr, "---END");
      if (ptr == NULL)
        break;
      ptr++;

      ptr_size = file_data.size;
      ptr_size -=
        (unsigned int) ((unsigned char *) ptr - (unsigned char *) buffer);

      if (ptr_size < 0)
        break;

    }
  
  gnutls_free(file_data.data);
  fprintf (stderr, "Loaded %d private keys.\n", (int) *privkey_size);

  return key;
}