/*
 * Copyright (C) 2009-2012 Free Software Foundation, Inc.
 *
 * Author: Simon Josefsson
 *
 * 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 GnuTLS; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <gnutls/gnutls.h>
#include <gnutls/x509.h>
#include <gnutls/pkcs12.h>

#include "utils.h"

#include <error.h>
#include <stdio.h>
#include <stdlib.h>

static char client_pem[] =
  "-----BEGIN CERTIFICATE-----\n"
  "MIICHjCCAYmgAwIBAgIERiYdNzALBgkqhkiG9w0BAQUwGTEXMBUGA1UEAxMOR251\n"
  "VExTIHRlc3QgQ0EwHhcNMDcwNDE4MTMyOTI3WhcNMDgwNDE3MTMyOTI3WjAdMRsw\n"
  "GQYDVQQDExJHbnVUTFMgdGVzdCBjbGllbnQwgZwwCwYJKoZIhvcNAQEBA4GMADCB\n"
  "iAKBgLtmQ/Xyxde2jMzF3/WIO7HJS2oOoa0gUEAIgKFPXKPQ+GzP5jz37AR2ExeL\n"
  "ZIkiW8DdU3w77XwEu4C5KL6Om8aOoKUSy/VXHqLnu7czSZ/ju0quak1o/8kR4jKN\n"
  "zj2AC41179gAgY8oBAOgIo1hBAf6tjd9IQdJ0glhaZiQo1ipAgMBAAGjdjB0MAwG\n"
  "A1UdEwEB/wQCMAAwEwYDVR0lBAwwCgYIKwYBBQUHAwIwDwYDVR0PAQH/BAUDAweg\n"
  "ADAdBgNVHQ4EFgQUTLkKm/odNON+3svSBxX+odrLaJEwHwYDVR0jBBgwFoAU6Twc\n"
  "+62SbuYGpFYsouHAUyfI8pUwCwYJKoZIhvcNAQEFA4GBALujmBJVZnvaTXr9cFRJ\n"
  "jpfc/3X7sLUsMvumcDE01ls/cG5mIatmiyEU9qI3jbgUf82z23ON/acwJf875D3/\n"
  "U7jyOsBJ44SEQITbin2yUeJMIm1tievvdNXBDfW95AM507ShzP12sfiJkJfjjdhy\n"
  "dc8Siq5JojruiMizAf0pA7in\n" "-----END CERTIFICATE-----\n";
const gnutls_datum_t client_dat = { (void*)client_pem, sizeof (client_pem) };

static char ca_pem[] =
  "-----BEGIN CERTIFICATE-----\n"
  "MIIB5zCCAVKgAwIBAgIERiYdJzALBgkqhkiG9w0BAQUwGTEXMBUGA1UEAxMOR251\n"
  "VExTIHRlc3QgQ0EwHhcNMDcwNDE4MTMyOTExWhcNMDgwNDE3MTMyOTExWjAZMRcw\n"
  "FQYDVQQDEw5HbnVUTFMgdGVzdCBDQTCBnDALBgkqhkiG9w0BAQEDgYwAMIGIAoGA\n"
  "vuyYeh1vfmslnuggeEKgZAVmQ5ltSdUY7H25WGSygKMUYZ0KT74v8C780qtcNt9T\n"
  "7EPH/N6RvB4BprdssgcQLsthR3XKA84jbjjxNCcaGs33lvOz8A1nf8p3hD+cKfRi\n"
  "kfYSW2JazLrtCC4yRCas/SPOUxu78of+3HiTfFm/oXUCAwEAAaNDMEEwDwYDVR0T\n"
  "AQH/BAUwAwEB/zAPBgNVHQ8BAf8EBQMDBwQAMB0GA1UdDgQWBBTpPBz7rZJu5gak\n"
  "Viyi4cBTJ8jylTALBgkqhkiG9w0BAQUDgYEAiaIRqGfp1jPpNeVhABK60SU0KIAy\n"
  "njuu7kHq5peUgYn8Jd9zNzExBOEp1VOipGsf6G66oQAhDFp2o8zkz7ZH71zR4HEW\n"
  "KoX6n5Emn6DvcEH/9pAhnGxNHJAoS7czTKv/JDZJhkqHxyrE1fuLsg5Qv25DTw7+\n"
  "PfqUpIhz5Bbm7J4=\n" "-----END CERTIFICATE-----\n";
const gnutls_datum_t ca_dat = { (void*)ca_pem, sizeof (ca_pem) };

static void
tls_log_func (int level, const char *str)
{
  fprintf (stderr, "|<%d>| %s", level, str);
}

void
doit (void)
{
  gnutls_pkcs12_t pkcs12;
  gnutls_x509_crt_t client;
  gnutls_x509_crt_t ca;
  gnutls_pkcs12_bag_t bag;
  unsigned char key_id_buf[20];
  gnutls_datum_t key_id;
  int ret, indx;
  char outbuf[10240];
  size_t size;
  int i;

  ret = gnutls_global_init ();
  if (ret < 0)
    error (EXIT_FAILURE, 0, "gnutls_global_init %d", ret);

  gnutls_global_set_log_function (tls_log_func);
  if (debug)
    gnutls_global_set_log_level (4711);

  /* Read certs. */
  ret = gnutls_x509_crt_init (&client);
  if (ret < 0)
    error (EXIT_FAILURE, 0, "crt_init: %d", ret);

  ret = gnutls_x509_crt_import (client, &client_dat, GNUTLS_X509_FMT_PEM);
  if (ret < 0)
    error (EXIT_FAILURE, 0, "crt_import: %d", ret);

  ret = gnutls_x509_crt_init (&ca);
  if (ret < 0)
    error (EXIT_FAILURE, 0, "ca_init: %d", ret);

  ret = gnutls_x509_crt_import (ca, &ca_dat, GNUTLS_X509_FMT_PEM);
  if (ret < 0)
    error (EXIT_FAILURE, 0, "ca_import: %d", ret);

  /* Create PKCS#12 structure. */
  ret = gnutls_pkcs12_init (&pkcs12);
  if (ret < 0)
    error (EXIT_FAILURE, 0, "pkcs12_init: %d", ret);

  /* Generate and add PKCS#12 cert bags. */
  for (i = 0; i < 2; i++)
    {
      ret = gnutls_pkcs12_bag_init (&bag);
      if (ret < 0)
        error (EXIT_FAILURE, 0, "bag_init: %d", ret);

      ret = gnutls_pkcs12_bag_set_crt (bag, i == 0 ? client : ca);
      if (ret < 0)
        error (EXIT_FAILURE, 0, "set_crt: %d", ret);

      indx = ret;

      ret = gnutls_pkcs12_bag_set_friendly_name (bag, indx,
                                                 i == 0 ? "client" : "ca");
      if (ret < 0)
        error (EXIT_FAILURE, 0, "set_friendly_name: %d", ret);

      size = sizeof (key_id_buf);
      ret = gnutls_x509_crt_get_key_id (i == 0 ? client : ca, 0,
                                        key_id_buf, &size);
      if (ret < 0)
        error (EXIT_FAILURE, 0, "get_key_id: %d", ret);

      key_id.data = key_id_buf;
      key_id.size = size;

      ret = gnutls_pkcs12_bag_set_key_id (bag, indx, &key_id);
      if (ret < 0)
        error (EXIT_FAILURE, 0, "bag_set_key_id: %d", ret);

      ret = gnutls_pkcs12_bag_encrypt (bag, "pass",
                                       i == 0 ? GNUTLS_PKCS8_USE_PKCS12_3DES
                                       : GNUTLS_PKCS_USE_PKCS12_RC2_40);
      if (ret < 0)
        error (EXIT_FAILURE, 0, "bag_encrypt: %d: %s", ret,
               i == 0 ? "3DES" : "RC2-40");

      ret = gnutls_pkcs12_set_bag (pkcs12, bag);
      if (ret < 0)
        error (EXIT_FAILURE, 0, "set_bag: %d", ret);

      gnutls_pkcs12_bag_deinit (bag);
    }

  /* MAC the structure, export and print. */
  ret = gnutls_pkcs12_generate_mac (pkcs12, "pass");
  if (ret < 0)
    error (EXIT_FAILURE, 0, "generate_mac: %d", ret);

  size = sizeof (outbuf);
  ret = gnutls_pkcs12_export (pkcs12, GNUTLS_X509_FMT_PEM, outbuf, &size);
  if (ret < 0)
    error (EXIT_FAILURE, 0, "pkcs12_export: %d", ret);

  if (debug)
    fwrite (outbuf, size, 1, stdout);

  /* Cleanup. */
  gnutls_pkcs12_deinit (pkcs12);
  gnutls_x509_crt_deinit (client);
  gnutls_x509_crt_deinit (ca);
  gnutls_global_deinit ();
}