Source code

Revision control

Copy as Markdown

Other Tools

/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include <stdio.h>
#include <string.h>
#if defined(WIN32)
#undef __STDC__
#include "fcntl.h"
#include "io.h"
#include <fcntl.h>
#else
#include <unistd.h>
#include <fcntl.h>
#endif
#include "secutil.h"
#include "nspr.h"
#include "prtypes.h"
#include "prtime.h"
#include "prlong.h"
#include "prinrval.h"
#include "prenv.h"
#include "pkcs11.h"
#include "pk11table.h"
#ifndef O_BINARY
#define O_BINARY 0
#endif
CK_ULONG systemFlags;
#define FLAG_NEGATE 0x80000000
#define FLAG_Verify 0x00000001
#define FLAG_VerifyFile 0x00000002
#define CKR_QUIT 0x80000000
int ArgSize(ArgType type);
const char *constLookup(const char *bp, CK_ULONG *value, ConstType *type);
int
isNum(char c)
{
return (c >= '0' && c <= '9');
}
int
isConst(const char *c)
{
CK_ULONG value;
ConstType type;
constLookup(c, &value, &type);
return type != ConstNone;
}
/*
* see if the variable is really a 'size' function. This
* function may modify var if it is a size function.
*/
char *
isSize(char *var, int *isArray)
{
char *ptr = NULL;
char *end;
int array = 0;
if (PL_strncasecmp(var, "sizeof(", /*)*/ 7) == 0) {
ptr = var + 7;
} else if (PL_strncasecmp(var, "size(", /*)*/ 5) == 0) {
ptr = var + 5;
} else if (PL_strncasecmp(var, "sizeofarray(", /*)*/ 12) == 0) {
ptr = var + 12;
array = 1;
} else if (PL_strncasecmp(var, "sizea(", /*)*/ 6) == 0) {
ptr = var + 6;
array = 1;
} else {
return NULL;
}
end = strchr(ptr, /*(*/ ')');
if (end == NULL) {
return NULL;
}
if (isArray)
*isArray = array;
*end = 0;
return ptr;
}
void
printConst(CK_ULONG value, ConstType type, int newLine)
{
int i;
for (i = 0; i < constCount; i++) {
if (consts[i].type == type && consts[i].value == value) {
printf("%s", consts[i].name);
break;
}
if (type == ConstNone && consts[i].value == value) {
printf("%s", consts[i].name);
break;
}
}
if (i == constCount) {
if ((type == ConstAvailableSizes) || (type == ConstCurrentSize)) {
printf("%lu", value);
} else {
printf("Unknown %s (%lu:0x%lx)", constTypeString[type], value, value);
}
}
if (newLine) {
printf("\n");
}
}
ConstType
getConstFromAttribute(CK_ATTRIBUTE_TYPE type)
{
int i;
for (i = 0; i < constCount; i++) {
if (consts[i].type == ConstAttribute && consts[i].value == type) {
return consts[i].attrType;
}
}
return ConstNone;
}
void
printChars(const char *name, CK_ULONG size)
{
CK_ULONG i;
for (i = 0; i < size; i++) {
if (name[i] == 0) {
break;
}
printf("%c", name[i]);
}
printf("\n");
}
#define DUMP_LEN 16
void
printDump(const unsigned char *buf, int size)
{
int i, j;
for (i = 0; i < size; i += DUMP_LEN) {
printf(" ");
for (j = 0; j < DUMP_LEN; j++) {
if (i + j < size) {
printf("%02x ", buf[i + j]);
} else {
printf(" ");
}
}
for (j = 0; j < DUMP_LEN; j++) {
if (i + j < size) {
if (buf[i + j] < ' ' || buf[i + j] >= 0x7f) {
printf(".");
} else {
printf("%c", buf[i + j]);
}
} else {
printf(" ");
}
}
printf("\n");
}
}
/*
* free an argument structure
*/
void
argFreeData(Value *arg)
{
if (arg->data && ((arg->type & ArgStatic) == 0)) {
if ((arg->type & ArgMask) == ArgAttribute) {
int i;
CK_ATTRIBUTE *template = (CK_ATTRIBUTE *)arg->data;
for (i = 0; i < arg->arraySize; i++) {
free(template[i].pValue);
}
}
if ((arg->type & ArgMask) == ArgInitializeArgs) {
CK_C_INITIALIZE_ARGS *init = (CK_C_INITIALIZE_ARGS *)arg->data;
if (init->LibraryParameters) {
free(init->LibraryParameters);
}
}
free(arg->data);
}
arg->type &= ~ArgStatic;
arg->data = NULL;
}
void
argFree(Value *arg)
{
if (arg == NULL)
return;
arg->reference--;
if (arg->reference == 0) {
if (arg->type & ArgFile) {
free(arg->filename);
}
argFreeData(arg);
free(arg);
}
}
/*
* free and argument list
*/
void
parseFree(Value **ap)
{
int i;
for (i = 0; i < MAX_ARGS; i++) {
argFree(ap[i]);
}
}
/*
* getEnd: how for to the end of this argmument list?
*/
int
getEnd(const char *bp)
{
int count = 0;
while (*bp) {
if (*bp == ' ' || *bp == '\t' || *bp == '\n')
return count;
count++;
bp++;
}
return (count);
}
/*
* strip: return the first none white space character
*/
const char *
strip(const char *bp)
{
while (*bp && (*bp == ' ' || *bp == '\t' || *bp == '\n'))
bp++;
return bp;
}
/*
* read in the next argument into dp ... don't overflow
*/
const char *
readChars(const char *bp, char *dp, int max)
{
int count = 1;
while (*bp) {
if (*bp == ' ' || *bp == '\t' || *bp == '\n') {
*dp = 0;
return bp;
}
*dp++ = *bp++;
if (++count == max)
break;
}
while (*bp && (*bp != ' ' && *bp != '\t' && *bp != '\n'))
bp++;
*dp = 0;
return (bp);
}
Value *varLookup(const char *bp, char *vname, int max, int *error);
CK_ULONG
getValue(const char *v, int *error)
{
Value *varVal = NULL;
CK_ULONG retVal = 0;
ConstType type;
char tvar[512];
*error = 0;
varVal = varLookup(v, tvar, sizeof(tvar), error);
if (varVal) {
if ((varVal->type & ArgMask) == ArgULong) {
retVal = *(CK_ULONG *)varVal->data;
} else {
fprintf(stderr, "%s: is not a ulong\n", v);
*error = 1;
}
argFree(varVal);
return retVal;
}
constLookup(v, &retVal, &type);
return retVal;
}
Value *
NewValue(ArgType type, CK_ULONG arraySize)
{
Value *value;
value = (Value *)malloc(sizeof(Value));
if (!value)
return NULL;
value->size = ArgSize(type) * arraySize;
value->type = type;
value->filename = NULL;
value->constType = ConstNone;
value->data = (void *)malloc(value->size);
if (!value->data) {
free(value);
return NULL;
}
value->reference = 1;
value->arraySize = (type == ArgChar) ? 1 : arraySize;
memset(value->data, 0, value->size);
return value;
}
#define INVALID_INDEX 0xffffffff
CK_ULONG
handleArray(char *vname, int *error)
{
char *bracket;
CK_ULONG index = INVALID_INDEX;
if ((bracket = strchr(vname, '[')) != 0) {
char *tmpv = bracket + 1;
*bracket = 0;
bracket = strchr(tmpv, ']');
if (bracket == 0) {
fprintf(stderr, "%s: missing closing brace\n", vname);
return INVALID_INDEX;
}
*bracket = 0;
index = getValue(tmpv, error);
if (*error == 1) {
return INVALID_INDEX;
} else if (index == INVALID_INDEX) {
fprintf(stderr, "%s: 0x%lx is an invalid index\n", vname, index);
*error = 1;
}
}
return index;
}
void *
makeArrayTarget(const char *vname, const Value *value, CK_ULONG index)
{
char *target;
CK_ULONG elementSize;
if (index >= (CK_ULONG)value->arraySize) {
fprintf(stderr, "%s[%lu]: index larger than array size (%d)\n",
vname, index, value->arraySize);
return NULL;
}
target = (char *)value->data;
elementSize = value->size / value->arraySize;
target += index * elementSize;
return target;
}
/*
* look up a variable from the variable chain
*/
static Variable *varHead = NULL;
Value *
varLookup(const char *bp, char *vname, int max, int *error)
{
Variable *current;
CK_ULONG index = INVALID_INDEX;
int isArray = 0;
char *ptr;
*error = 0;
if (bp != NULL) {
readChars(bp, vname, max);
}
/* don't make numbers into variables */
if (isNum(vname[0])) {
return NULL;
}
/* nor consts */
if (isConst(vname)) {
return NULL;
}
/* handle sizeof() */
if ((ptr = isSize(vname, &isArray)) != NULL) {
CK_ULONG size;
Value *targetValue = NULL;
Value *sourceValue = varLookup(NULL, ptr, 0, error);
if (!sourceValue) {
if (*error == 0) {
/* just didn't find it */
*error = 1;
fprintf(stderr, "Couldn't find variable %s to take size of\n",
ptr);
return NULL;
}
}
size = isArray ? sourceValue->arraySize : sourceValue->size;
targetValue = NewValue(ArgULong, 1);
memcpy(targetValue->data, &size, sizeof(size));
return targetValue;
}
/* modifies vname */
index = handleArray(vname, error);
if (*error == 1) {
return NULL;
}
for (current = varHead; current; current = current->next) {
if (PL_strcasecmp(current->vname, vname) == 0) {
char *target;
if (index == INVALID_INDEX) {
(current->value->reference)++;
return current->value;
}
target = makeArrayTarget(vname, current->value, index);
if (target) {
Value *element = NewValue(current->value->type, 1);
if (!element) {
fprintf(stderr, "MEMORY ERROR!\n");
*error = 1;
}
argFreeData(element);
element->data = target;
element->type |= ArgStatic;
return element;
}
*error = 1;
return NULL;
}
}
return NULL;
}
static CK_RV
list(void)
{
Variable *current;
if (varHead) {
printf(" %10s\t%16s\t%8s\tSize\tElements\n", "Name", "Type", "Const");
} else {
printf(" no variables set\n");
}
for (current = varHead; current; current = current->next) {
printf(" %10s\t%16s\t%8s\t%d\t%d\n", current->vname,
valueString[current->value->type & ArgMask],
constTypeString[current->value->constType],
current->value->size, current->value->arraySize);
}
return CKR_OK;
}
CK_RV
printFlags(const char *s, CK_ULONG flags, ConstType type)
{
CK_ULONG i;
int needComma = 0;
printf("%s", s);
for (i = 1; i; i = i << 1) {
if (flags & i) {
printf("%s", needComma ? "," : "");
printConst(i, type, 0);
needComma = 1;
}
}
if (!needComma) {
printf("Empty");
}
printf("\n");
return CKR_OK;
}
/*
* add a new variable to the chain
*/
const char *
AddVariable(const char *bp, Value **ptr)
{
char vname[512];
Variable *current;
int index = INVALID_INDEX;
int size;
int error = 0;
bp = readChars(bp, vname, sizeof(vname));
/* don't make numbers into variables */
if (isNum(vname[0])) {
return bp;
}
/* or consts */
if (isConst(vname)) {
return bp;
}
/* or NULLs */
if (vname[0] == 0) {
return bp;
}
/* or sizeof */
if (isSize(vname, NULL)) {
return bp;
}
/* arrays values should be written back to the original */
index = handleArray(vname, &error);
if (error == 1) {
return bp;
}
for (current = varHead; current; current = current->next) {
if (PL_strcasecmp(current->vname, vname) == 0) {
char *target;
/* found a complete object, return the found one */
if (index == INVALID_INDEX) {
argFree(*ptr);
*ptr = current->value;
return bp;
}
/* found an array, update the array element */
target = makeArrayTarget(vname, current->value, index);
if (target) {
memcpy(target, (*ptr)->data, (*ptr)->size);
argFreeData(*ptr);
(*ptr)->data = target;
(*ptr)->type |= ArgStatic;
}
return bp;
}
}
/* we are looking for an array and didn't find one */
if (index != INVALID_INDEX) {
return bp;
}
current = (Variable *)malloc(sizeof(Variable));
size = strlen(vname);
current->vname = (char *)malloc(size + 1);
strcpy(current->vname, vname);
current->value = *ptr;
(*ptr)->reference++;
current->next = varHead;
varHead = current;
return bp;
}
ArgType
FindTypeByName(const char *typeName)
{
int i;
for (i = 0; i < valueCount; i++) {
if (PL_strcasecmp(typeName, valueString[i]) == 0) {
return (ArgType)i;
}
if (valueString[i][0] == 'C' && valueString[i][1] == 'K' &&
valueString[i][2] == '_' &&
(PL_strcasecmp(typeName, &valueString[i][3]) == 0)) {
return (ArgType)i;
}
}
return ArgNone;
}
CK_RV
ArrayVariable(const char *bp, const char *typeName, CK_ULONG count)
{
ArgType type;
Value *value; /* new Value */
type = FindTypeByName(typeName);
if (type == ArgNone) {
fprintf(stderr, "Invalid type (%s)\n", typeName);
return CKR_FUNCTION_FAILED;
}
value = NewValue(type, count);
(void)AddVariable(bp, &value);
return CKR_OK;
}
#define MAX_TEMPLATE 25
CK_RV
ArrayTemplate(const char *bp, char *attributes)
{
char aname[512];
CK_ULONG attributeTypes[MAX_TEMPLATE];
CK_ATTRIBUTE *template;
Value *value; /* new Value */
char *ap;
int i, count = 0;
memcpy(aname, attributes, strlen(attributes) + 1);
for (ap = aname, count = 0; ap && *ap && count < MAX_TEMPLATE; count++) {
char *cur = ap;
ConstType type;
ap = strchr(ap, ',');
if (ap) {
*ap++ = 0;
}
(void)constLookup(cur, &attributeTypes[count], &type);
if ((type != ConstAttribute) && (type != ConstNone)) {
fprintf(stderr, "Unknown Attribute %s\n", cur);
return CKR_FUNCTION_FAILED;
}
}
value = NewValue(ArgAttribute, count);
template = (CK_ATTRIBUTE *)value->data;
for (i = 0; i < count; i++) {
template[i].type = attributeTypes[i];
}
(void)AddVariable(bp, &value);
return CKR_OK;
}
CK_RV
BuildTemplate(Value *vp)
{
CK_ATTRIBUTE *template = (CK_ATTRIBUTE *)vp->data;
int i;
for (i = 0; i < vp->arraySize; i++) {
if (((signed long)template[i].ulValueLen) > 0) {
if (template[i].pValue)
free(template[i].pValue);
template[i].pValue = malloc(template[i].ulValueLen);
}
}
return CKR_OK;
}
CK_RV
SetTemplate(Value *vp, CK_ULONG index, CK_ULONG value)
{
CK_ATTRIBUTE *template = (CK_ATTRIBUTE *)vp->data;
int isbool = 0;
CK_ULONG len;
ConstType attrType;
if (index >= (CK_ULONG)vp->arraySize) {
fprintf(stderr, "index (%lu) greater than array (%d)\n",
index, vp->arraySize);
return CKR_ARGUMENTS_BAD;
}
attrType = getConstFromAttribute(template[index].type);
if (attrType == ConstNone) {
fprintf(stderr, "can't set index (%lu) because ", index);
printConst(template[index].type, ConstAttribute, 0);
fprintf(stderr, " is not a CK_BBOOL or CK_ULONG\n");
return CKR_ARGUMENTS_BAD;
}
isbool = (attrType == ConstBool);
len = isbool ? sizeof(CK_BBOOL) : sizeof(CK_ULONG);
if ((template[index].ulValueLen != len) || (template[index].pValue)) {
free(template[index].pValue);
template[index].pValue = malloc(len);
template[index].ulValueLen = len;
}
if (isbool) {
*(CK_BBOOL *)template[index].pValue = (CK_BBOOL)value;
} else {
*(CK_ULONG *)template[index].pValue = (CK_ULONG)value;
}
return CKR_OK;
}
CK_RV
NewMechanism(const char *bp, CK_ULONG mechType)
{
Value *value; /* new Value */
CK_MECHANISM *mechanism;
value = NewValue(ArgMechanism, 1);
mechanism = (CK_MECHANISM *)value->data;
mechanism->mechanism = mechType;
mechanism->pParameter = NULL;
mechanism->ulParameterLen = 0;
(void)AddVariable(bp, &value);
return CKR_OK;
}
CK_RV
NewInitializeArgs(const char *bp, CK_ULONG flags, const char *param)
{
Value *value; /* new Value */
CK_C_INITIALIZE_ARGS *init;
value = NewValue(ArgInitializeArgs, 1);
init = (CK_C_INITIALIZE_ARGS *)value->data;
init->flags = flags;
if (strcmp(param, "null") != 0) {
init->LibraryParameters = (CK_CHAR_PTR *)strdup(param);
}
(void)AddVariable(bp, &value);
return CKR_OK;
}
/*
* add a new variable to the chain
*/
CK_RV
DeleteVariable(const char *bp)
{
char vname[512];
Variable **current;
bp = readChars(bp, vname, sizeof(vname));
for (current = &varHead; *current; current = &(*current)->next) {
if (PL_strcasecmp((*current)->vname, vname) == 0) {
argFree((*current)->value);
*current = (*current)->next;
break;
}
}
return CKR_OK;
}
/*
* convert an octal value to integer
*/
CK_ULONG
otoi(const char *o)
{
CK_ULONG value = 0;
while (*o) {
if ((*o >= '0') && (*o <= '7')) {
value = (value << 3) | (unsigned)(*o - '0');
} else {
break;
}
}
return value;
}
/*
* convert a hex value to integer
*/
CK_ULONG
htoi(const char *x)
{
CK_ULONG value = 0;
while (*x) {
if ((*x >= '0') && (*x <= '9')) {
value = (value << 4) | (unsigned)(*x - '0');
} else if ((*x >= 'a') && (*x <= 'f')) {
value = (value << 4) | (unsigned)(*x - 'a');
} else if ((*x >= 'A') && (*x <= 'F')) {
value = (value << 4) | (unsigned)(*x - 'A');
} else {
break;
}
}
return value;
}
/*
* look up or decode a constant value
*/
const char *
constLookup(const char *bp, CK_ULONG *value, ConstType *type)
{
char vname[512];
int i;
bp = readChars(bp, vname, sizeof(vname));
for (i = 0; i < constCount; i++) {
if ((PL_strcasecmp(consts[i].name, vname) == 0) ||
PL_strcasecmp(consts[i].name + 5, vname) == 0) {
*value = consts[i].value;
*type = consts[i].type;
return bp;
}
}
*type = ConstNone;
if (vname[0] == '0' && vname[1] == 'X') {
*value = htoi(&vname[2]);
} else if (vname[0] == '0') {
*value = otoi(&vname[1]);
} else {
*value = atoi(vname);
}
return bp;
}
int
ArgSize(ArgType type)
{
int size = 0;
type &= ArgMask;
switch (type) {
case ArgNone:
size = 0;
break;
case ArgULong:
size = sizeof(CK_ULONG);
break;
case ArgVar:
size = 1; /* get's changed later */
break;
case ArgChar:
case ArgUTF8:
size = 1;
break;
case ArgInfo:
size = sizeof(CK_INFO);
break;
case ArgSlotInfo:
size = sizeof(CK_SLOT_INFO);
break;
case ArgTokenInfo:
size = sizeof(CK_TOKEN_INFO);
break;
case ArgSessionInfo:
size = sizeof(CK_SESSION_INFO);
break;
case ArgAttribute:
size = sizeof(CK_ATTRIBUTE);
break;
case ArgMechanism:
size = sizeof(CK_MECHANISM);
break;
case ArgMechanismInfo:
size = sizeof(CK_MECHANISM_INFO);
break;
case ArgInitializeArgs:
size = sizeof(CK_C_INITIALIZE_ARGS);
break;
case ArgFunctionList:
size = sizeof(CK_FUNCTION_LIST);
break;
default:
break;
}
return (size);
}
CK_RV
restore(const char *filename, Value *ptr)
{
int fd, size;
fd = open(filename, O_RDONLY | O_BINARY);
if (fd < 0) {
perror(filename);
return CKR_FUNCTION_FAILED;
}
size = read(fd, ptr->data, ptr->size);
if (systemFlags & FLAG_VerifyFile) {
printDump(ptr->data, ptr->size);
}
if (size < 0) {
perror(filename);
return CKR_FUNCTION_FAILED;
} else if (size != ptr->size) {
fprintf(stderr, "%s: only read %d bytes, needed to read %d bytes\n",
filename, size, ptr->size);
return CKR_FUNCTION_FAILED;
}
close(fd);
return CKR_OK;
}
CK_RV
save(const char *filename, Value *ptr)
{
int fd, size;
fd = open(filename, O_WRONLY | O_BINARY | O_CREAT, 0666);
if (fd < 0) {
perror(filename);
return CKR_FUNCTION_FAILED;
}
size = write(fd, ptr->data, ptr->size);
if (size < 0) {
perror(filename);
return CKR_FUNCTION_FAILED;
} else if (size != ptr->size) {
fprintf(stderr, "%s: only wrote %d bytes, need to write %d bytes\n",
filename, size, ptr->size);
return CKR_FUNCTION_FAILED;
}
close(fd);
return CKR_OK;
}
static CK_RV
increment(Value *ptr, CK_ULONG value)
{
if ((ptr->type & ArgMask) != ArgULong) {
return CKR_ARGUMENTS_BAD;
}
*(CK_ULONG *)ptr->data += value;
return CKR_OK;
}
static CK_RV
decrement(Value *ptr, CK_ULONG value)
{
if ((ptr->type & ArgMask) != ArgULong) {
return CKR_ARGUMENTS_BAD;
}
*(CK_ULONG *)ptr->data -= value;
return CKR_OK;
}
CK_RV
printArg(Value *ptr, int arg_number)
{
ArgType type = ptr->type & ArgMask;
CK_INFO *info;
CK_SLOT_INFO *slotInfo;
CK_TOKEN_INFO *tokenInfo;
CK_SESSION_INFO *sessionInfo;
CK_ATTRIBUTE *attribute;
CK_MECHANISM *mechanism;
CK_MECHANISM_INFO *mechanismInfo;
CK_C_INITIALIZE_ARGS *initArgs;
CK_FUNCTION_LIST *functionList;
CK_RV ckrv = CKR_OK;
ConstType constType;
if (arg_number) {
printf("Arg %d: \n", arg_number);
}
if (ptr->arraySize > 1) {
Value element;
int i;
int elementSize = ptr->size / ptr->arraySize;
char *dp = (char *)ptr->data;
/* build a temporary Value to hold a single element */
element.type = type;
element.constType = ptr->constType;
element.size = elementSize;
element.filename = ptr->filename;
element.reference = 1;
element.arraySize = 1;
for (i = 0; i < ptr->arraySize; i++) {
printf(" -----[ %d ] -----\n", i);
element.data = (void *)&dp[i * elementSize];
(void)printArg(&element, 0);
}
return ckrv;
}
if (ptr->data == NULL) {
printf(" NULL ptr to a %s\n", valueString[type]);
return ckrv;
}
switch (type) {
case ArgNone:
printf(" None\n");
break;
case ArgULong:
printf(" %lu (0x%lx)\n", *((CK_ULONG *)ptr->data),
*((CK_ULONG *)ptr->data));
if (ptr->constType != ConstNone) {
printf(" ");
printConst(*(CK_ULONG *)ptr->data, ptr->constType, 1);
}
break;
case ArgVar:
printf(" %s\n", (char *)ptr->data);
break;
case ArgUTF8:
printf(" %s\n", (char *)ptr->data);
break;
case ArgChar:
printDump(ptr->data, ptr->size);
break;
case ArgInfo:
#define VERSION(x) (x).major, (x).minor
info = (CK_INFO *)ptr->data;
printf(" Cryptoki Version: %d.%02d\n",
VERSION(info->cryptokiVersion));
printf(" Manufacturer ID: ");
printChars((char *)info->manufacturerID,
sizeof(info->manufacturerID));
printFlags(" Flags: ", info->flags, ConstInfoFlags);
printf(" Library Description: ");
printChars((char *)info->libraryDescription,
sizeof(info->libraryDescription));
printf(" Library Version: %d.%02d\n",
VERSION(info->libraryVersion));
break;
case ArgSlotInfo:
slotInfo = (CK_SLOT_INFO *)ptr->data;
printf(" Slot Description: ");
printChars((char *)slotInfo->slotDescription,
sizeof(slotInfo->slotDescription));
printf(" Manufacturer ID: ");
printChars((char *)slotInfo->manufacturerID,
sizeof(slotInfo->manufacturerID));
printFlags(" Flags: ", slotInfo->flags, ConstSlotFlags);
printf(" Hardware Version: %d.%02d\n",
VERSION(slotInfo->hardwareVersion));
printf(" Firmware Version: %d.%02d\n",
VERSION(slotInfo->firmwareVersion));
break;
case ArgTokenInfo:
tokenInfo = (CK_TOKEN_INFO *)ptr->data;
printf(" Label: ");
printChars((char *)tokenInfo->label, sizeof(tokenInfo->label));
printf(" Manufacturer ID: ");
printChars((char *)tokenInfo->manufacturerID,
sizeof(tokenInfo->manufacturerID));
printf(" Model: ");
printChars((char *)tokenInfo->model, sizeof(tokenInfo->model));
printf(" Serial Number: ");
printChars((char *)tokenInfo->serialNumber,
sizeof(tokenInfo->serialNumber));
printFlags(" Flags: ", tokenInfo->flags, ConstTokenFlags);
printf(" Max Session Count: ");
printConst(tokenInfo->ulMaxSessionCount, ConstAvailableSizes, 1);
printf(" Session Count: ");
printConst(tokenInfo->ulSessionCount, ConstCurrentSize, 1);
printf(" RW Session Count: ");
printConst(tokenInfo->ulMaxRwSessionCount, ConstAvailableSizes, 1);
printf(" Max Pin Length : ");
printConst(tokenInfo->ulMaxPinLen, ConstCurrentSize, 1);
printf(" Min Pin Length : ");
printConst(tokenInfo->ulMinPinLen, ConstCurrentSize, 1);
printf(" Total Public Memory: ");
printConst(tokenInfo->ulTotalPublicMemory, ConstAvailableSizes, 1);
printf(" Free Public Memory: ");
printConst(tokenInfo->ulFreePublicMemory, ConstCurrentSize, 1);
printf(" Total Private Memory: ");
printConst(tokenInfo->ulTotalPrivateMemory, ConstAvailableSizes, 1);
printf(" Free Private Memory: ");
printConst(tokenInfo->ulFreePrivateMemory, ConstCurrentSize, 1);
printf(" Hardware Version: %d.%02d\n",
VERSION(tokenInfo->hardwareVersion));
printf(" Firmware Version: %d.%02d\n",
VERSION(tokenInfo->firmwareVersion));
printf(" UTC Time: ");
printChars((char *)tokenInfo->utcTime, sizeof(tokenInfo->utcTime));
break;
case ArgSessionInfo:
sessionInfo = (CK_SESSION_INFO *)ptr->data;
printf(" SlotID: 0x%08lx\n", sessionInfo->slotID);
printf(" State: ");
printConst(sessionInfo->state, ConstSessionState, 1);
printFlags(" Flags: ", sessionInfo->flags, ConstSessionFlags);
printf(" Device error: %lu 0x%08lx\n", sessionInfo->ulDeviceError,
sessionInfo->ulDeviceError);
break;
case ArgAttribute:
attribute = (CK_ATTRIBUTE *)ptr->data;
printf(" Attribute Type: ");
printConst(attribute->type, ConstAttribute, 1);
printf(" Attribute Data: ");
if (attribute->pValue == NULL) {
printf("NULL\n");
printf("Attribute Len: %lu\n", attribute->ulValueLen);
} else {
constType = getConstFromAttribute(attribute->type);
if (constType != ConstNone) {
CK_ULONG value = (constType == ConstBool) ? *(CK_BBOOL *)attribute->pValue
: *(CK_ULONG *)attribute->pValue;
printConst(value, constType, 1);
} else {
printf("\n");
printDump(attribute->pValue, attribute->ulValueLen);
}
}
break;
case ArgMechanism:
mechanism = (CK_MECHANISM *)ptr->data;
printf(" Mechanism Type: ");
printConst(mechanism->mechanism, ConstMechanism, 1);
printf(" Mechanism Data:\n");
printDump(mechanism->pParameter, mechanism->ulParameterLen);
break;
case ArgMechanismInfo:
mechanismInfo = (CK_MECHANISM_INFO *)ptr->data;
printf(" Minimum Key Size: %ld\n", mechanismInfo->ulMinKeySize);
printf(" Maximum Key Size: %ld\n", mechanismInfo->ulMaxKeySize);
printFlags(" Flags: ", mechanismInfo->flags, ConstMechanismFlags);
break;
case ArgInitializeArgs:
initArgs = (CK_C_INITIALIZE_ARGS *)ptr->data;
printFlags(" Flags: ", initArgs->flags, ConstInitializeFlags);
if (initArgs->LibraryParameters) {
printf("Params: %s\n", (char *)initArgs->LibraryParameters);
}
case ArgFunctionList:
functionList = (CK_FUNCTION_LIST *)ptr->data;
printf(" Version: %d.%02d\n", VERSION(functionList->version));
#ifdef notdef
#undef CK_NEED_ARG_LIST
#define CK_PKCS11_FUNCTION_INFO(func) \
printf(" %s: 0x%08lx\n", #func, (unsigned long)functionList->func);
#include "pkcs11f.h"
#undef CK_NEED_ARG_LIST
#undef CK_PKCS11_FUNCTION_INFO
#endif
default:
ckrv = CKR_ARGUMENTS_BAD;
break;
}
return ckrv;
}
/*
* Feeling ambitious? turn this whole thing into lexx yacc parser
* with full expressions.
*/
Value **
parseArgs(int index, const char *bp)
{
const Commands *cp = &commands[index];
int size = strlen(cp->fname);
int i;
CK_ULONG value;
char vname[512];
Value **argList, *possible;
ConstType constType;
/*
* skip pass the command
*/
if ((cp->fname[0] == 'C') && (cp->fname[1] == '_') && (bp[1] != '_')) {
size -= 2;
}
bp += size;
/*
* Initialize our argument list
*/
argList = (Value **)malloc(sizeof(Value *) * MAX_ARGS);
for (i = 0; i < MAX_ARGS; i++) {
argList[i] = NULL;
}
/*
* Walk the argument list parsing it...
*/
for (i = 0; i < MAX_ARGS; i++) {
ArgType type = cp->args[i] & ArgMask;
int error;
/* strip blanks */
bp = strip(bp);
/* if we hit ArgNone, we've nabbed all the arguments we need */
if (type == ArgNone) {
break;
}
/* if we run out of space in the line, we weren't given enough
* arguments... */
if (*bp == '\0') {
/* we're into optional arguments, ok to quit now */
if (cp->args[i] & ArgOpt) {
break;
}
fprintf(stderr, "%s: only %d args found,\n", cp->fname, i);
parseFree(argList);
return NULL;
}
/* collect all the rest of the command line and send
* it as a single argument */
if (cp->args[i] & ArgFull) {
int size = strlen(bp) + 1;
argList[i] = NewValue(type, size);
memcpy(argList[i]->data, bp, size);
break;
}
/*
* look up the argument in our variable list first... only
* exception is the new argument type for set...
*/
error = 0;
if ((cp->args[i] != (ArgVar | ArgNew)) &&
(possible = varLookup(bp, vname, sizeof(vname), &error))) {
/* ints are only compatible with other ints... all other types
* are interchangeable... */
if (type != ArgVar) { /* ArgVar's match anyone */
if ((type == ArgULong) ^
((possible->type & ArgMask) == ArgULong)) {
fprintf(stderr, "%s: Arg %d incompatible type with <%s>\n",
cp->fname, i + 1, vname);
argFree(possible);
parseFree(argList);
return NULL;
}
/*
* ... that is as long as they are big enough...
*/
if (ArgSize(type) > possible->size) {
fprintf(stderr,
"%s: Arg %d %s is too small (%d bytes needs to be %d bytes)\n",
cp->fname, i + 1, vname, possible->size, ArgSize(type));
argFree(possible);
parseFree(argList);
return NULL;
}
}
/* everything looks kosher here, use it */
argList[i] = possible;
bp = readChars(bp, vname, sizeof(vname));
if (cp->args[i] & ArgOut) {
possible->type |= ArgOut;
}
continue;
}
if (error == 1) {
parseFree(argList);
return NULL;
}
/* create space for our argument */
argList[i] = NewValue(type, 1);
if ((PL_strncasecmp(bp, "null", 4) == 0) && ((bp[4] == 0) ||
(bp[4] ==
' ') ||
(bp[4] ==
'\t') ||
(bp[4] == '\n'))) {
if (cp->args[i] == ArgULong) {
fprintf(stderr, "%s: Arg %d CK_ULONG can't be NULL\n",
cp->fname, i + 1);
parseFree(argList);
return NULL;
}
argFreeData(argList[i]);
argList[i]->data = NULL;
argList[i]->size = 0;
bp += 4;
if (*bp)
bp++;
continue;
}
/* if we're an output variable, we need to add it */
if (cp->args[i] & ArgOut) {
if (PL_strncasecmp(bp, "file(", 5) == 0 /* ) */) {
char filename[512];
bp = readChars(bp + 5, filename, sizeof(filename));
size = PL_strlen(filename);
if ((size > 0) && (/* ( */ filename[size - 1] == ')')) {
filename[size - 1] = 0;
}
filename[size] = 0;
argList[i]->filename = (char *)malloc(size + 1);
PL_strcpy(argList[i]->filename, filename);
argList[i]->type |= ArgOut | ArgFile;
break;
}
bp = AddVariable(bp, &argList[i]);
argList[i]->type |= ArgOut;
continue;
}
if (PL_strncasecmp(bp, "file(", 5) == 0 /* ) */) {
char filename[512];
bp = readChars(bp + 5, filename, sizeof(filename));
size = PL_strlen(filename);
if ((size > 0) && (/* ( */ filename[size - 1] == ')')) {
filename[size - 1] = 0;
}
if (restore(filename, argList[i]) != CKR_OK) {
parseFree(argList);
return NULL;
}
continue;
}
switch (type) {
case ArgULong:
bp = constLookup(bp, &value, &constType);
*(int *)argList[i]->data = value;
argList[i]->constType = constType;
break;
case ArgVar:
argFreeData(argList[i]);
size = getEnd(bp) + 1;
argList[i]->data = (void *)malloc(size);
argList[i]->size = size;
/* fall through */
case ArgInfo:
case ArgSlotInfo:
case ArgTokenInfo:
case ArgSessionInfo:
case ArgAttribute:
case ArgMechanism:
case ArgMechanismInfo:
case ArgInitializeArgs:
case ArgUTF8:
case ArgChar:
bp = readChars(bp, (char *)argList[i]->data, argList[i]->size);
case ArgNone:
default:
break;
}
}
return argList;
}
/* lookup the command in the array */
int
lookup(const char *buf)
{
int size, i;
int buflen;
buflen = PL_strlen(buf);
for (i = 0; i < commandCount; i++) {
size = PL_strlen(commands[i].fname);
if (size <= buflen) {
if (PL_strncasecmp(buf, commands[i].fname, size) == 0) {
return i;
}
}
if (size - 2 <= buflen) {
if (commands[i].fname[0] == 'C' && commands[i].fname[1] == '_' &&
(PL_strncasecmp(buf, &commands[i].fname[2], size - 2) == 0)) {
return i;
}
}
}
fprintf(stderr, "Can't find command %s\n", buf);
return -1;
}
void
putOutput(Value **ptr)
{
int i;
for (i = 0; i < MAX_ARGS; i++) {
ArgType type;
if (ptr[i] == NULL)
break;
type = ptr[i]->type;
ptr[i]->type &= ~ArgOut;
if (type == ArgNone) {
break;
}
if (type & ArgOut) {
(void)printArg(ptr[i], i + 1);
}
if (type & ArgFile) {
save(ptr[i]->filename, ptr[i]);
free(ptr[i]->filename);
ptr[i]->filename = NULL; /* paranoia */
}
}
}
CK_RV
unloadModule(Module *module)
{
char *disableUnload = NULL;
disableUnload = PR_GetEnvSecure("NSS_DISABLE_UNLOAD");
if (module->library && !disableUnload) {
PR_UnloadLibrary(module->library);
}
module->library = NULL;
module->functionList = NULL;
return CKR_OK;
}
CK_RV
loadModule(Module *module, char *library)
{
PRLibrary *newLibrary;
CK_C_GetFunctionList getFunctionList;
CK_FUNCTION_LIST *functionList;
CK_RV ckrv;
newLibrary = PR_LoadLibrary(library);
if (!newLibrary) {
fprintf(stderr, "Couldn't load library %s\n", library);
return CKR_FUNCTION_FAILED;
}
getFunctionList = (CK_C_GetFunctionList)
PR_FindSymbol(newLibrary, "C_GetFunctionList");
if (!getFunctionList) {
fprintf(stderr, "Couldn't find \"C_GetFunctionList\" in %s\n", library);
return CKR_FUNCTION_FAILED;
}
ckrv = (*getFunctionList)(&functionList);
if (ckrv != CKR_OK) {
return ckrv;
}
if (module->library) {
PR_UnloadLibrary(module->library);
}
module->library = newLibrary;
module->functionList = functionList;
return CKR_OK;
}
static void
printHelp(int index, int full)
{
int j;
printf(" %s", commands[index].fname);
for (j = 0; j < MAX_ARGS; j++) {
ArgType type = commands[index].args[j] & ArgMask;
if (type == ArgNone) {
break;
}
printf(" %s", valueString[type]);
}
printf("\n");
printf(" %s\n", commands[index].helpString);
}
/* add Topical help here ! */
static CK_RV
printTopicHelp(char *topic)
{
int size, i;
int topicLen;
topicLen = PL_strlen(topic);
for (i = 0; i < topicCount; i++) {
size = PL_strlen(topics[i].name);
if (size <= topicLen) {
if (PL_strncasecmp(topic, topics[i].name, size) == 0) {
break;
}
}
}
if (i == topicCount) {
fprintf(stderr, "Can't find topic '%s'\n", topic);
return CKR_DATA_INVALID;
}
printf(" %s", topic);
printf("\n");
printf(" %s\n", topics[i].helpString);
return CKR_OK;
}
static CK_RV
printGeneralHelp(void)
{
int i;
printf(" To get help on commands, select from the list below:");
for (i = 0; i < commandCount; i++) {
if (i % 5 == 0)
printf("\n");
printf("%s,", commands[i].fname);
}
printf("\n");
/* print help topics */
printf(" To get help on a topic, select from the list below:");
for (i = 0; i < topicCount; i++) {
if (i % 5 == 0)
printf("\n");
printf("%s,", topics[i].name);
}
printf("\n");
return CKR_OK;
}
static CK_RV
quitIf(CK_ULONG a, const char *cmp, CK_ULONG b)
{
if (strcmp(cmp, "<") == 0) {
return (a < b) ? CKR_QUIT : CKR_OK;
} else if (strcmp(cmp, ">") == 0) {
return (a > b) ? CKR_QUIT : CKR_OK;
} else if (strcmp(cmp, "<=") == 0) {
return (a <= b) ? CKR_QUIT : CKR_OK;
} else if (strcmp(cmp, ">=") == 0) {
return (a >= b) ? CKR_QUIT : CKR_OK;
} else if (strcmp(cmp, "=") == 0) {
return (a == b) ? CKR_QUIT : CKR_OK;
} else if (strcmp(cmp, "!=") == 0) {
return (a != b) ? CKR_QUIT : CKR_OK;
}
printf("Unkown integer comparator: '%s'\n", cmp);
return CKR_ARGUMENTS_BAD;
}
static CK_RV
quitIfString(const char *a, const char *cmp, const char *b)
{
if (strcmp(cmp, "=") == 0) {
return (strcmp(a, b) == 0) ? CKR_QUIT : CKR_OK;
} else if (strcmp(cmp, "!=") == 0) {
return (strcmp(a, b) != 0) ? CKR_QUIT : CKR_OK;
}
printf("Unkown string comparator: '%s'\n", cmp);
return CKR_ARGUMENTS_BAD;
}
CK_RV run(const char *);
CK_RV timeCommand(const char *);
CK_RV loop(const char *filename, const char *var,
CK_ULONG start, CK_ULONG end, CK_ULONG step);
/*
* Actually dispatch the function... Bad things happen
* if these don't match the commands array.
*/
CK_RV
do_func(int index, Value **a)
{
int value, helpIndex;
static Module module = { NULL, NULL };
CK_FUNCTION_LIST *func = module.functionList;
switch (commands[index].fType) {
case F_C_Initialize:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_Initialize((void *)a[0]->data);
case F_C_Finalize:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_Finalize((void *)a[0]->data);
case F_C_GetInfo:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_GetInfo((CK_INFO *)a[0]->data);
case F_C_GetFunctionList:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_GetFunctionList((CK_FUNCTION_LIST **)a[0]->data);
case F_C_GetSlotList:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_GetSlotList((CK_BBOOL) * (CK_ULONG *)a[0]->data,
(CK_SLOT_ID *)a[1]->data,
(CK_ULONG *)a[2]->data);
case F_C_GetSlotInfo:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_GetSlotInfo(*(CK_ULONG *)a[0]->data,
(CK_SLOT_INFO *)a[1]->data);
case F_C_GetTokenInfo:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_GetTokenInfo(*(CK_ULONG *)a[0]->data,
(CK_TOKEN_INFO *)a[1]->data);
case F_C_GetMechanismList:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
if (a[1]->data) {
a[1]->constType = ConstMechanism;
}
return func->C_GetMechanismList(*(CK_ULONG *)a[0]->data,
(CK_MECHANISM_TYPE *)a[1]->data,
(CK_ULONG *)a[2]->data);
case F_C_GetMechanismInfo:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_GetMechanismInfo(*(CK_ULONG *)a[0]->data,
*(CK_ULONG *)a[1]->data,
(CK_MECHANISM_INFO *)a[2]->data);
case F_C_InitToken:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_InitToken(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_CHAR *)a[3]->data);
case F_C_InitPIN:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_InitPIN(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data);
case F_C_SetPIN:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_SetPIN(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_CHAR *)a[3]->data,
*(CK_ULONG *)a[4]->data);
case F_C_OpenSession:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_OpenSession(*(CK_ULONG *)a[0]->data,
*(CK_ULONG *)a[1]->data,
(void *)NULL,
(CK_NOTIFY)NULL,
(CK_ULONG *)a[2]->data);
case F_C_CloseSession:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_CloseSession(*(CK_ULONG *)a[0]->data);
case F_C_CloseAllSessions:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_CloseAllSessions(*(CK_ULONG *)a[0]->data);
case F_C_GetSessionInfo:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_GetSessionInfo(*(CK_ULONG *)a[0]->data,
(CK_SESSION_INFO *)a[1]->data);
case F_C_GetOperationState:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_GetOperationState(*(CK_ULONG *)a[0]->data,
(CK_BYTE *)a[1]->data,
(CK_ULONG *)a[2]->data);
case F_C_SetOperationState:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_SetOperationState(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data,
*(CK_ULONG *)a[3]->data,
*(CK_ULONG *)a[4]->data);
case F_C_Login:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_Login(*(CK_ULONG *)a[0]->data,
*(CK_ULONG *)a[1]->data,
(CK_CHAR *)a[2]->data,
*(CK_ULONG *)a[3]->data);
case F_C_Logout:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_Logout(*(CK_ULONG *)a[0]->data);
case F_C_CreateObject:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_CreateObject(*(CK_ULONG *)a[0]->data,
(CK_ATTRIBUTE *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_ULONG *)a[3]->data);
case F_C_CopyObject:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_CopyObject(*(CK_ULONG *)a[0]->data,
*(CK_ULONG *)a[0]->data,
(CK_ATTRIBUTE *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_ULONG *)a[3]->data);
case F_C_DestroyObject:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_DestroyObject(*(CK_ULONG *)a[0]->data,
*(CK_ULONG *)a[1]->data);
case F_C_GetObjectSize:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_GetObjectSize(*(CK_ULONG *)a[0]->data,
*(CK_ULONG *)a[1]->data,
(CK_ULONG *)a[2]->data);
case F_C_GetAttributeValue:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_GetAttributeValue(*(CK_ULONG *)a[0]->data,
*(CK_ULONG *)a[1]->data,
(CK_ATTRIBUTE *)a[2]->data,
*(CK_ULONG *)a[3]->data);
case F_C_SetAttributeValue:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_SetAttributeValue(*(CK_ULONG *)a[0]->data,
*(CK_ULONG *)a[1]->data,
(CK_ATTRIBUTE *)a[2]->data,
*(CK_ULONG *)a[3]->data);
case F_C_FindObjectsInit:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_FindObjectsInit(*(CK_ULONG *)a[0]->data,
(CK_ATTRIBUTE *)a[1]->data,
*(CK_ULONG *)a[2]->data);
case F_C_FindObjects:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_FindObjects(*(CK_ULONG *)a[0]->data,
(CK_ULONG *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_ULONG *)a[3]->data);
case F_C_FindObjectsFinal:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_FindObjectsFinal(*(CK_ULONG *)a[0]->data);
case F_C_EncryptInit:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_EncryptInit(*(CK_ULONG *)a[0]->data,
(CK_MECHANISM *)a[1]->data,
*(CK_ULONG *)a[2]->data);
case F_C_Encrypt:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_Encrypt(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_CHAR *)a[3]->data,
(CK_ULONG *)a[4]->data);
case F_C_EncryptUpdate:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_EncryptUpdate(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_CHAR *)a[3]->data,
(CK_ULONG *)a[4]->data);
case F_C_EncryptFinal:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_EncryptFinal(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
(CK_ULONG *)a[2]->data);
case F_C_DecryptInit:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_DecryptInit(*(CK_ULONG *)a[0]->data,
(CK_MECHANISM *)a[1]->data,
*(CK_ULONG *)a[2]->data);
case F_C_Decrypt:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_Decrypt(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_CHAR *)a[3]->data,
(CK_ULONG *)a[4]->data);
case F_C_DecryptUpdate:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_DecryptUpdate(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_CHAR *)a[3]->data,
(CK_ULONG *)a[4]->data);
case F_C_DecryptFinal:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_DecryptFinal(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
(CK_ULONG *)a[2]->data);
case F_C_DigestInit:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_DigestInit(*(CK_ULONG *)a[0]->data,
(CK_MECHANISM *)a[1]->data);
case F_C_Digest:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_Digest(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_CHAR *)a[3]->data,
(CK_ULONG *)a[4]->data);
case F_C_DigestUpdate:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_DigestUpdate(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data);
case F_C_DigestKey:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_DigestKey(*(CK_ULONG *)a[0]->data,
*(CK_ULONG *)a[1]->data);
case F_C_DigestFinal:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_DigestFinal(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
(CK_ULONG *)a[2]->data);
case F_C_SignInit:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_SignInit(*(CK_ULONG *)a[0]->data,
(CK_MECHANISM *)a[1]->data,
*(CK_ULONG *)a[2]->data);
case F_C_Sign:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_Sign(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_CHAR *)a[3]->data,
(CK_ULONG *)a[4]->data);
case F_C_SignUpdate:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_SignUpdate(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data);
case F_C_SignFinal:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_SignFinal(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
(CK_ULONG *)a[2]->data);
case F_C_SignRecoverInit:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_SignRecoverInit(*(CK_ULONG *)a[0]->data,
(CK_MECHANISM *)a[1]->data,
*(CK_ULONG *)a[2]->data);
case F_C_SignRecover:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_SignRecover(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_CHAR *)a[3]->data,
(CK_ULONG *)a[4]->data);
case F_C_VerifyInit:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_VerifyInit(*(CK_ULONG *)a[0]->data,
(CK_MECHANISM *)a[1]->data,
*(CK_ULONG *)a[2]->data);
case F_C_Verify:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_Verify(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_CHAR *)a[3]->data,
*(CK_ULONG *)a[4]->data);
case F_C_VerifyUpdate:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_VerifyUpdate(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data);
case F_C_VerifyFinal:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_VerifyFinal(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data);
case F_C_VerifyRecoverInit:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_VerifyRecoverInit(*(CK_ULONG *)a[0]->data,
(CK_MECHANISM *)a[1]->data,
*(CK_ULONG *)a[2]->data);
case F_C_VerifyRecover:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_VerifyRecover(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_CHAR *)a[3]->data,
(CK_ULONG *)a[4]->data);
case F_C_DigestEncryptUpdate:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_DigestEncryptUpdate(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_CHAR *)a[3]->data,
(CK_ULONG *)a[4]->data);
case F_C_DecryptDigestUpdate:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_DecryptDigestUpdate(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_CHAR *)a[3]->data,
(CK_ULONG *)a[4]->data);
case F_C_SignEncryptUpdate:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_SignEncryptUpdate(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_CHAR *)a[3]->data,
(CK_ULONG *)a[4]->data);
case F_C_DecryptVerifyUpdate:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_DecryptVerifyUpdate(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_CHAR *)a[3]->data,
(CK_ULONG *)a[4]->data);
case F_C_GenerateKey:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_GenerateKey(*(CK_ULONG *)a[0]->data,
(CK_MECHANISM *)a[1]->data,
(CK_ATTRIBUTE *)a[2]->data,
*(CK_ULONG *)a[3]->data,
(CK_ULONG *)a[4]->data);
case F_C_GenerateKeyPair:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_GenerateKeyPair(*(CK_ULONG *)a[0]->data,
(CK_MECHANISM *)a[1]->data,
(CK_ATTRIBUTE *)a[2]->data,
*(CK_ULONG *)a[3]->data,
(CK_ATTRIBUTE *)a[4]->data,
*(CK_ULONG *)a[5]->data,
(CK_ULONG *)a[6]->data,
(CK_ULONG *)a[7]->data);
case F_C_WrapKey:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_WrapKey(*(CK_ULONG *)a[0]->data,
(CK_MECHANISM *)a[1]->data,
*(CK_ULONG *)a[2]->data,
*(CK_ULONG *)a[3]->data,
(CK_CHAR *)a[5]->data,
(CK_ULONG *)a[6]->data);
case F_C_UnwrapKey:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_UnwrapKey(*(CK_ULONG *)a[0]->data,
(CK_MECHANISM *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_CHAR *)a[3]->data,
*(CK_ULONG *)a[4]->data,
(CK_ATTRIBUTE *)a[5]->data,
*(CK_ULONG *)a[6]->data,
(CK_ULONG *)a[7]->data);
case F_C_DeriveKey:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_DeriveKey(*(CK_ULONG *)a[0]->data,
(CK_MECHANISM *)a[1]->data,
*(CK_ULONG *)a[2]->data,
(CK_ATTRIBUTE *)a[3]->data,
*(CK_ULONG *)a[4]->data,
(CK_ULONG *)a[5]->data);
case F_C_SeedRandom:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_SeedRandom(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data);
case F_C_GenerateRandom:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_GenerateRandom(*(CK_ULONG *)a[0]->data,
(CK_CHAR *)a[1]->data,
*(CK_ULONG *)a[2]->data);
case F_C_GetFunctionStatus:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_GetFunctionStatus(*(CK_ULONG *)a[0]->data);
case F_C_CancelFunction:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_CancelFunction(*(CK_ULONG *)a[0]->data);
case F_C_WaitForSlotEvent:
if (!func)
return CKR_CRYPTOKI_NOT_INITIALIZED;
return func->C_WaitForSlotEvent(*(CK_ULONG *)a[0]->data,
(CK_ULONG *)a[1]->data,
(void *)a[2]->data);
/* set a variable */
case F_SetVar:
case F_SetStringVar:
(void)DeleteVariable(a[0]->data);
(void)AddVariable(a[0]->data, &a[1]);
return CKR_OK;
/* print a value */
case F_Print:
return printArg(a[0], 0);
case F_SaveVar:
return save(a[0]->data, a[1]);
case F_RestoreVar:
return restore(a[0]->data, a[1]);
case F_Delete:
return DeleteVariable(a[0]->data);
case F_Increment:
return increment(a[0], *(CK_ULONG *)a[1]->data);
case F_Decrement:
return decrement(a[0], *(CK_ULONG *)a[1]->data);
case F_List:
return list();
case F_Run:
return run(a[0]->data);
case F_Time:
return timeCommand(a[0]->data);
case F_Load:
return loadModule(&module, a[0]->data);
case F_Unload:
return unloadModule(&module);
case F_NewArray:
(void)DeleteVariable(a[0]->data);
return ArrayVariable(a[0]->data, a[1]->data, *(CK_ULONG *)a[2]->data);
case F_NewTemplate:
(void)DeleteVariable(a[0]->data);
return ArrayTemplate(a[0]->data, a[1]->data);
case F_BuildTemplate:
return BuildTemplate(a[0]);
case F_SetTemplate:
return SetTemplate(a[0],
*(CK_ULONG *)a[1]->data,
*(CK_ULONG *)a[2]->data);
case F_NewMechanism:
(void)DeleteVariable(a[0]->data);
return NewMechanism(a[0]->data, *(CK_ULONG *)a[1]->data);
case F_NewInitializeArgs:
(void)DeleteVariable(a[0]->data);
return NewInitializeArgs(a[0]->data, *(CK_ULONG *)a[1]->data, a[2]->data);
case F_System:
value = *(int *)a[0]->data;
if (value & 0x80000000) {
systemFlags &= ~value;
} else {
systemFlags |= value;
}
return CKR_OK;
case F_Loop:
return loop(a[0]->data, a[1]->data, *(CK_ULONG *)a[2]->data,
*(CK_ULONG *)a[3]->data, *(CK_ULONG *)a[4]->data);
case F_Help:
if (a[0]) {
helpIndex = lookup(a[0]->data);
if (helpIndex < 0) {
return printTopicHelp(a[0]->data);
}
printHelp(helpIndex, 1);
return CKR_OK;
}
return printGeneralHelp();
case F_QuitIfString:
return quitIfString(a[0]->data, a[1]->data, a[2]->data);
case F_QuitIf:
return quitIf(*(CK_ULONG *)a[0]->data, a[1]->data, *(CK_ULONG *)a[2]->data);
case F_Quit:
return CKR_QUIT;
default:
fprintf(stderr,
"Function %s not yet supported\n", commands[index].fname);
return CKR_OK;
}
/* Not Reached */
return CKR_OK;
}
CK_RV
processCommand(const char *buf)
{
CK_RV error = CKR_OK;
int index;
const char *bp;
Value **arglist;
bp = strip(buf);
/* allow comments and blank lines in scripts */
if ((*bp == '#') || (*bp == 0) || (*bp == '\n')) {
return CKR_OK;
}
index = lookup(bp);
if (index < 0) {
return CKR_OK;
}
arglist = parseArgs(index, bp);
if (arglist == NULL) {
return CKR_OK;
}
error = do_func(index, arglist);
if (error == CKR_OK) {
putOutput(arglist);
} else if (error != CKR_QUIT) {
printf(">> Error : ");
printConst(error, ConstResult, 1);
}
parseFree(arglist);
return error;
}
CK_RV
timeCommand(const char *command)
{
CK_RV ckrv;
PRIntervalTime startTime = PR_IntervalNow();
PRIntervalTime endTime;
PRIntervalTime elapsedTime;
ckrv = processCommand(command);
endTime = PR_IntervalNow();
elapsedTime = endTime - startTime;
printf("Time -- %d msec \n",
PR_IntervalToMilliseconds(elapsedTime));
return ckrv;
}
CK_RV
process(FILE *inFile, int user)
{
char buf[2048];
CK_RV error;
CK_RV ckrv = CKR_OK;
if (user) {
printf("pkcs11> ");
fflush(stdout);
}
while (fgets(buf, 2048, inFile) != NULL) {
if (!user)
printf("* %s", buf);
error = processCommand(buf);
if (error == CKR_QUIT) {
break;
} else if (error != CKR_OK) {
ckrv = error;
}
if (user) {
printf("pkcs11> ");
fflush(stdout);
}
}
return ckrv;
}
CK_RV
run(const char *filename)
{
FILE *infile;
CK_RV ckrv;
infile = fopen(filename, "r");
if (infile == NULL) {
perror(filename);
return CKR_FUNCTION_FAILED;
}
ckrv = process(infile, 0);
fclose(infile);
return ckrv;
}
CK_RV
loop(const char *filename, const char *var,
CK_ULONG start, CK_ULONG end, CK_ULONG step)
{
CK_ULONG i = 0;
Value *value = 0;
CK_RV ckrv;
for (i = start; i < end; i += step) {
value = NewValue(ArgULong, 1);
*(CK_ULONG *)value->data = i;
DeleteVariable(var);
AddVariable(var, &value);
ckrv = run(filename);
argFree(value);
if (ckrv == CKR_QUIT) {
break;
}
}
return ckrv;
}
int
main(int argc, char **argv)
{
/* I suppose that some day we could parse some arguments */
(void)process(stdin, 1);
return 0;
}