p7sign.c
上传用户:lyxiangda
上传日期:2007-01-12
资源大小:3042k
文件大小:8k
源码类别:

CA认证

开发平台:

WINDOWS

  1. /*
  2.  * The contents of this file are subject to the Mozilla Public
  3.  * License Version 1.1 (the "License"); you may not use this file
  4.  * except in compliance with the License. You may obtain a copy of
  5.  * the License at http://www.mozilla.org/MPL/
  6.  * 
  7.  * Software distributed under the License is distributed on an "AS
  8.  * IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
  9.  * implied. See the License for the specific language governing
  10.  * rights and limitations under the License.
  11.  * 
  12.  * The Original Code is the Netscape security libraries.
  13.  * 
  14.  * The Initial Developer of the Original Code is Netscape
  15.  * Communications Corporation.  Portions created by Netscape are 
  16.  * Copyright (C) 1994-2000 Netscape Communications Corporation.  All
  17.  * Rights Reserved.
  18.  * 
  19.  * Contributor(s):
  20.  * 
  21.  * Alternatively, the contents of this file may be used under the
  22.  * terms of the GNU General Public License Version 2 or later (the
  23.  * "GPL"), in which case the provisions of the GPL are applicable 
  24.  * instead of those above.  If you wish to allow use of your 
  25.  * version of this file only under the terms of the GPL and not to
  26.  * allow others to use your version of this file under the MPL,
  27.  * indicate your decision by deleting the provisions above and
  28.  * replace them with the notice and other provisions required by
  29.  * the GPL.  If you do not delete the provisions above, a recipient
  30.  * may use your version of this file under either the MPL or the
  31.  * GPL.
  32.  */
  33. /*
  34.  * p7sign -- A command to create a *detached* pkcs7 signature (over a given
  35.  * input file).
  36.  *
  37.  * $Id: p7sign.c,v 1.1 2000/03/31 20:12:30 relyea%netscape.com Exp $
  38.  */
  39. #include "nspr.h"
  40. #include "plgetopt.h"
  41. #include "secutil.h"
  42. #include "secpkcs7.h"
  43. #include "cert.h"
  44. #include "certdb.h"
  45. #include "cdbhdl.h"
  46. #if defined(XP_UNIX)
  47. #include <unistd.h>
  48. #endif
  49. #include <stdio.h>
  50. #include <string.h>
  51. #if (defined(XP_WIN) && !defined(WIN32)) || (defined(__sun) && !defined(SVR4))
  52. extern int fread(char *, size_t, size_t, FILE*);
  53. extern int fwrite(char *, size_t, size_t, FILE*);
  54. extern int fprintf(FILE *, char *, ...);
  55. #endif
  56. extern void SEC_Init(void); /* XXX */
  57. static void
  58. Usage(char *progName)
  59. {
  60.     fprintf(stderr,
  61.     "Usage:  %s -k keyname [-d keydir] [-i input] [-o output]n",
  62.     progName);
  63.     fprintf(stderr, "%-20s Nickname of key to use for signaturen",
  64.     "-k keyname");
  65.     fprintf(stderr, "%-20s Key database directory (default is ~/.netscape)n",
  66.     "-d keydir");
  67.     fprintf(stderr, "%-20s Define an input file to use (default is stdin)n",
  68.     "-i input");
  69.     fprintf(stderr, "%-20s Define an output file to use (default is stdout)n",
  70.     "-o output");
  71.     fprintf(stderr, "%-20s Encapsulate content in signature messagen",
  72.     "-e");
  73.     exit(-1);
  74. }
  75. static SECKEYKeyDBHandle *
  76. OpenKeyDB(char *progName)
  77. {
  78.     SECKEYKeyDBHandle *keyHandle;
  79.     keyHandle = SECU_OpenKeyDB(PR_FALSE);
  80.     if (keyHandle == NULL) {
  81.         SECU_PrintError(progName, "could not open key database");
  82. return NULL;
  83.     }
  84.     return(keyHandle);
  85. }
  86. static CERTCertDBHandle certHandleStatic; /* avoid having to allocate */
  87. static CERTCertDBHandle *
  88. OpenCertDB(char *progName)
  89. {
  90.     CERTCertDBHandle *certHandle;
  91.     SECStatus rv;
  92.     certHandle = &certHandleStatic;
  93.     rv = CERT_OpenCertDB(certHandle, PR_FALSE, SECU_CertDBNameCallback, NULL);
  94.     if (rv != SECSuccess) {
  95.         SECU_PrintError(progName, "could not open cert database");
  96. return NULL;
  97.     } else {
  98. CERT_SetDefaultCertDB(certHandle);
  99.     }
  100.     return certHandle;
  101. }
  102. static void
  103. SignOut(void *arg, const char *buf, unsigned long len)
  104. {
  105.    FILE *out;
  106.    out = arg; 
  107.    fwrite (buf, len, 1, out);
  108. }
  109. static int
  110. CreateDigest(SECItem *data, char *digestdata, unsigned int *len, unsigned int maxlen)
  111. {
  112.     SECHashObject *hashObj;
  113.     void *hashcx;
  114.     /* XXX probably want to extend interface to allow other hash algorithms */
  115.     hashObj = &SECHashObjects[HASH_AlgSHA1];
  116.     hashcx = (* hashObj->create)();
  117.     if (hashcx == NULL)
  118. return -1;
  119.     (* hashObj->begin)(hashcx);
  120.     (* hashObj->update)(hashcx, data->data, data->len);
  121.     (* hashObj->end)(hashcx, (unsigned char *)digestdata, len, maxlen);
  122.     (* hashObj->destroy)(hashcx, PR_TRUE);
  123.     return 0;
  124. }
  125. static int
  126. SignFile(FILE *outFile, PRFileDesc *inFile, CERTCertificate *cert, 
  127.          PRBool encapsulated)
  128. {
  129.     int nb;
  130.     char digestdata[32];
  131.     unsigned int len;
  132.     SECItem digest, data2sign;
  133.     SEC_PKCS7ContentInfo *cinfo;
  134.     SECStatus rv;
  135.     if (outFile == NULL || inFile == NULL || cert == NULL)
  136. return -1;
  137.     /* suck the file in */
  138. if (SECU_ReadDERFromFile(&data2sign, inFile, PR_FALSE) != SECSuccess)
  139. return -1;
  140.     if (!encapsulated) {
  141. /* unfortunately, we must create the digest ourselves */
  142. /* SEC_PKCS7CreateSignedData should have a flag to not include */
  143. /* the content for non-encapsulated content at encode time, but */
  144. /* should always compute the hash itself */
  145. if (CreateDigest(&data2sign, digestdata, &len, 32) < 0)
  146.     return -1;
  147. digest.data = (unsigned char *)digestdata;
  148. digest.len = len;
  149.     }
  150.     /* XXX Need a better way to handle that usage stuff! */
  151.     cinfo = SEC_PKCS7CreateSignedData (cert, certUsageEmailSigner, NULL,
  152.        SEC_OID_SHA1,
  153.        encapsulated ? NULL : &digest,
  154.        NULL, NULL);
  155.     if (cinfo == NULL)
  156. return -1;
  157.     if (encapsulated) {
  158. SEC_PKCS7SetContent(cinfo, (char *)data2sign.data, data2sign.len);
  159.     }
  160.     rv = SEC_PKCS7IncludeCertChain (cinfo, NULL);
  161.     if (rv != SECSuccess) {
  162. SEC_PKCS7DestroyContentInfo (cinfo);
  163. return -1;
  164.     }
  165.     rv = SEC_PKCS7Encode (cinfo, SignOut, outFile, NULL,
  166.   SECU_GetPassword, NULL);
  167.     SEC_PKCS7DestroyContentInfo (cinfo);
  168.     if (rv != SECSuccess)
  169. return -1;
  170.     return 0;
  171. }
  172. int
  173. main(int argc, char **argv)
  174. {
  175.     char *progName;
  176.     FILE *outFile;
  177.     PRFileDesc *inFile;
  178.     char *keyName;
  179.     SECKEYKeyDBHandle *keyHandle;
  180.     CERTCertDBHandle *certHandle;
  181.     CERTCertificate *cert;
  182.     PRBool encapsulated = PR_FALSE;
  183.     PLOptState *optstate;
  184.     PLOptStatus status;
  185.     progName = strrchr(argv[0], '/');
  186.     progName = progName ? progName+1 : argv[0];
  187.     inFile = NULL;
  188.     outFile = NULL;
  189.     keyName = NULL;
  190.     /*
  191.      * Parse command line arguments
  192.      */
  193.     optstate = PL_CreateOptState(argc, argv, "ed:k:i:o:");
  194.     while ((status = PL_GetNextOpt(optstate)) == PL_OPT_OK) {
  195. switch (optstate->option) {
  196.   case '?':
  197.     Usage(progName);
  198.     break;
  199.   case 'e':
  200.     /* create a message with the signed content encapsulated */
  201.     encapsulated = PR_TRUE;
  202.     break;
  203.   case 'd':
  204.     SECU_ConfigDirectory(optstate->value);
  205.     break;
  206.   case 'i':
  207.     inFile = PR_Open(optstate->value, PR_RDONLY, 0);
  208.     if (!inFile) {
  209. fprintf(stderr, "%s: unable to open "%s" for readingn",
  210. progName, optstate->value);
  211. return -1;
  212.     }
  213.     break;
  214.   case 'k':
  215.     keyName = strdup(optstate->value);
  216.     break;
  217.   case 'o':
  218.     outFile = fopen(optstate->value, "w");
  219.     if (!outFile) {
  220. fprintf(stderr, "%s: unable to open "%s" for writingn",
  221. progName, optstate->value);
  222. return -1;
  223.     }
  224.     break;
  225. }
  226.     }
  227.     if (!keyName) Usage(progName);
  228.     if (!inFile) inFile = PR_STDIN;
  229.     if (!outFile) outFile = stdout;
  230.     /* Call the initialization routines */
  231.     PR_Init(PR_SYSTEM_THREAD, PR_PRIORITY_NORMAL, 1);
  232.     SECU_PKCS11Init(PR_FALSE);
  233.     SEC_Init();
  234.     /* open key database */
  235.     keyHandle = OpenKeyDB(progName);
  236.     if (keyHandle == NULL) {
  237. return -1;
  238.     }
  239. #if 0
  240.     /* check if key actually exists */
  241.     if (! SECU_CheckKeyNameExists(keyHandle, keyName)) {
  242. SECU_PrintError(progName, "the key "%s" does not exist", keyName);
  243. return -1;
  244.     }
  245. #endif
  246.     SECKEY_SetDefaultKeyDB(keyHandle);
  247.     /* open cert database */
  248.     certHandle = OpenCertDB(progName);
  249.     if (certHandle == NULL) {
  250. return -1;
  251.     }
  252.     /* find cert */
  253.     cert = CERT_FindCertByNickname(certHandle, keyName);
  254.     if (cert == NULL) {
  255. SECU_PrintError(progName,
  256.         "the corresponding cert for key "%s" does not exist",
  257. keyName);
  258. return -1;
  259.     }
  260.     CERT_SetDefaultCertDB(certHandle);
  261.     if (SignFile(outFile, inFile, cert, encapsulated)) {
  262. SECU_PrintError(progName, "problem signing data");
  263. return -1;
  264.     }
  265.     return 0;
  266. }