+++ /dev/null
-/*
- * Simple interactive CSP RNG exerciser.
- */
-#include <stdlib.h>
-#include <stdio.h>
-#include <time.h>
-#include <Security/cssm.h>
-#include <Security/cssmapple.h>
-#include "cspwrap.h"
-#include "common.h"
-
-#define RAND_ALG CSSM_ALGID_APPLE_YARROW
-#define BUFSIZE 32
-
-static void usage(char **argv)
-{
- printf("usage: %s [options]\n", argv[0]);
- printf("Options:\n");
- printf(" D (CSP/DL; default = bare CSP)\n");
- exit(1);
-}
-
-static void dumpBuf(uint8 *buf,
- unsigned len)
-{
- unsigned i;
-
- printf(" ");
- for(i=0; i<len; i++) {
- printf("%02X ", buf[i]);
- if((i % 8) == 7) {
- printf("\n ");
- }
- }
- printf("\n");
-}
-
-void doGenRand(
- CSSM_CSP_HANDLE cspHand,
- CSSM_CC_HANDLE ccHand, // 0 ==> gen a new one
- // else use this
- CSSM_CRYPTO_DATA_PTR seed, // optional
- unsigned len,
- CSSM_BOOL weMalloc)
-{
- CSSM_RETURN crtn;
- CSSM_DATA data = {0, NULL};
-
- /* optional existing context */
- if(ccHand == 0) {
- crtn = CSSM_CSP_CreateRandomGenContext(
- cspHand,
- RAND_ALG,
- seed,
- len,
- &ccHand);
- if(crtn) {
- printError("CSSM_CSP_CreateRandomGenContext", crtn);
- return;
- }
- }
-
- /* who mallocs the data? */
- if(weMalloc) {
- data.Data = (uint8 *)appMalloc(len, NULL);
- if(data.Data == NULL) {
- printf("***malloc failure\n");
- return;
- }
- data.Length = len;
- }
-
- /* go for it */
- crtn = CSSM_GenerateRandom(ccHand, &data);
- if(crtn) {
- printError("CSSM_GenerateRandom", crtn);
- return;
- }
-
- dumpBuf(data.Data, data.Length);
- appFree(data.Data, NULL);
- return;
-}
-
-#define SEED_SIZE 32
-
-/*
- * CryptoData callback for optional random seed.
- */
-CSSM_RETURN seedCallback(
- CSSM_DATA_PTR OutData,
- void *CallerCtx)
-{
- int i, j;
- static unsigned char seed[SEED_SIZE];
-
- OutData->Length = SEED_SIZE;
- OutData->Data = seed;
- for(i=SEED_SIZE, j=0; i>0; i--, j++) {
- seed[j] = i;
- }
- return CSSM_OK;
-}
-
-int main(int argc, char **argv)
-{
- int arg;
- char *argp;
- CSSM_CSP_HANDLE cspHand;
- CSSM_CC_HANDLE ccHand = 0;
- CSSM_RETURN crtn;
- CSSM_BOOL bareCsp = CSSM_TRUE;
- char resp = 'n';
- // initial op = get random data
- CSSM_BOOL weMalloc = CSSM_FALSE;
- unsigned char seed[SEED_SIZE];
- CSSM_CRYPTO_DATA cseed;
- int i;
- unsigned reqLen = 16;
- CSSM_BOOL explicitSeed;
-
- for(arg=1; arg<argc; arg++) {
- argp = argv[arg];
- switch(argv[arg][0]) {
- case 'D':
- bareCsp = CSSM_FALSE;
- break;
- default:
- usage(argv);
- }
- }
- cspHand = cspDlDbStartup(bareCsp, NULL);
- if(cspHand == 0) {
- exit(1);
- }
-
- /* set up for explicit seed */
- cseed.Param.Length = SEED_SIZE;
- cseed.Param.Data = seed;
- for(i=0; i<SEED_SIZE; i++) {
- seed[i] = i;
- }
- cseed.Callback = NULL;
- cseed.CallerCtx = NULL;
- explicitSeed = CSSM_TRUE;
-
- while(1) {
- printf(" g Get random data w/seed\n");
- printf(" n Get random data w/o seed\n");
- printf(" c Get random data using current context\n");
- printf(" m we malloc (current: %s)\n",
- weMalloc ? "True" : "False");
- printf(" M CSP mallocs\n");
- printf(" s seed via explicit data (current: %d)\n", (int)explicitSeed);
- printf(" S seed via callback\n");
- printf(" q quit\n");
- fpurge(stdin);
- printf("\ncommand me: ");
- nextChar:
- resp = getchar();
- if(resp == 'q') {
- break;
- }
- switch(resp) {
- case 'g':
- if(ccHand != 0) {
- crtn = CSSM_DeleteContext(ccHand);
- if(crtn) {
- printError("CSSM_DeleteContext", crtn);
- }
- ccHand = 0;
- }
- doGenRand(cspHand, ccHand, &cseed, reqLen, weMalloc);
- break;
- case 'n':
- if(ccHand != 0) {
- crtn = CSSM_DeleteContext(ccHand);
- if(crtn) {
- printError("CSSM_DeleteContext", crtn);
- }
- ccHand = 0;
- }
- doGenRand(cspHand, ccHand, NULL, reqLen, weMalloc);
- break;
- case 'c':
- doGenRand(cspHand, ccHand, NULL, reqLen, weMalloc);
- break;
- case 'm':
- weMalloc = CSSM_TRUE;
- break;
- case 'M':
- weMalloc = CSSM_FALSE;
- break;
- case 'l':
- printf("New length: ");
- scanf("%d", &reqLen);
- break;
- case 's':
- /* explicit seed - presence of callback is the determinant */
- cseed.Callback = NULL;
- explicitSeed = CSSM_TRUE;
- break;
- case 'S':
- /* seed by calllback */
- cseed.Callback = seedCallback;
- explicitSeed = CSSM_FALSE;
- break;
- case '\n':
- goto nextChar;
- default:
- printf("Huh?\n");
- }
- }
- if((crtn = CSSM_ModuleDetach(cspHand))) {
- printError("CSSM_ModuleDetach", crtn);
- exit(1);
- }
- return 0;
-}
projects / apple / security.git / blobdiff