--- /dev/null
+/* MDS per-thread basher */
+#include <time.h>
+#include <stdio.h>
+#include "testParams.h"
+#include <security_cdsa_client/mdsclient.h>
+
+/* for malloc debug */
+#define DO_PAUSE 0
+
+using namespace Security;
+
+/* most of this is cribbed from cspxutils/mdsLookup/ */
+
+/* free attribute(s) allocated by MDS */
+static void freeAttrs(
+ CSSM_DB_RECORD_ATTRIBUTE_DATA_PTR recordAttrs)
+{
+ unsigned i;
+
+ for(i=0; i<recordAttrs->NumberOfAttributes; i++) {
+ CSSM_DB_ATTRIBUTE_DATA_PTR attrData = &recordAttrs->AttributeData[i];
+ if(attrData == NULL) {
+ /* fault of caller, who allocated the CSSM_DB_ATTRIBUTE_DATA */
+ printf("***freeAttrs screwup: NULL attrData\n");
+ return;
+ }
+ unsigned j;
+ for(j=0; j<attrData->NumberOfValues; j++) {
+ CSSM_DATA_PTR data = &attrData->Value[j];
+ if(data == NULL) {
+ /* fault of MDS, who said there was a value here */
+ printf("***freeAttrs screwup: NULL data\n");
+ return;
+ }
+ /* we're using cdsa_client, what's the appFree equivalent? */
+ free(data->Data);
+ data->Data = NULL;
+ data->Length = 0;
+ }
+ free(attrData->Value);
+ attrData->Value = NULL;
+ }
+}
+
+static int doLookup(
+ MDSClient::Directory &mds,
+
+ /* Record type, a.k.a. Relation, e.g. MDS_CDSADIR_CSP_PRIMARY_RECORDTYPE */
+ CSSM_DB_RECORDTYPE recordType,
+
+ /* key, value, valForm, and valOp are the thing we search on */
+ /* Note CSSM_DB_ATTRIBUTE_NAME_FORMAT - the format of the attribute name -
+ * is always CSSM_DB_ATTRIBUTE_NAME_AS_STRING for MDS. */
+ const char *key, // e.g. "AlgType"
+ const void *valPtr,
+ unsigned valLen,
+ CSSM_DB_ATTRIBUTE_FORMAT valForm, // CSSM_DB_ATTRIBUTE_FORMAT_STRING, etc.
+ CSSM_DB_OPERATOR valOp) // normally CSSM_DB_EQUAL
+{
+ CSSM_QUERY query;
+ CSSM_DB_UNIQUE_RECORD_PTR record = NULL;
+ CSSM_HANDLE resultHand;
+ CSSM_DB_RECORD_ATTRIBUTE_DATA recordAttrs;
+ CSSM_SELECTION_PREDICATE predicate;
+ CSSM_DATA predData;
+ CSSM_DB_ATTRIBUTE_DATA outAttr;
+ CSSM_DB_ATTRIBUTE_INFO_PTR attrInfo;
+ const char *attrName = "ModuleID";
+
+ /* We want one attribute back, name and format specified by caller */
+ recordAttrs.DataRecordType = recordType;
+ recordAttrs.SemanticInformation = 0;
+ recordAttrs.NumberOfAttributes = 1;
+ recordAttrs.AttributeData = &outAttr;
+
+ memset(&outAttr, 0, sizeof(CSSM_DB_ATTRIBUTE_DATA));
+ attrInfo = &outAttr.Info;
+ attrInfo->AttributeNameFormat = CSSM_DB_ATTRIBUTE_NAME_AS_STRING;
+ attrInfo->Label.AttributeName = (char *)attrName;
+ attrInfo->AttributeFormat = CSSM_DB_ATTRIBUTE_FORMAT_STRING;
+
+ /* one predicate - the caller's key and CSSM_DB_OPERATOR */
+ predicate.DbOperator = valOp;
+ predicate.Attribute.Info.AttributeNameFormat = CSSM_DB_ATTRIBUTE_NAME_AS_STRING;
+ predicate.Attribute.Info.Label.AttributeName = (char *)key;
+ predicate.Attribute.Info.AttributeFormat = valForm;
+ predData.Data = (uint8 *)valPtr;
+ predData.Length = valLen;
+ predicate.Attribute.Value = &predData;
+ predicate.Attribute.NumberOfValues = 1;
+
+ query.RecordType = recordType;
+ query.Conjunctive = CSSM_DB_NONE;
+ query.NumSelectionPredicates = 1;
+ query.SelectionPredicate = &predicate;
+ query.QueryLimits.TimeLimit = 0; // FIXME - meaningful?
+ query.QueryLimits.SizeLimit = 1; // FIXME - meaningful?
+ query.QueryFlags = 0; // CSSM_QUERY_RETURN_DATA...FIXME - used?
+
+ /* until this API settles down bag this test */
+ #if 0
+ printf("doLookup: mds.dlGetFirst unstable' skipping!\n");
+ return -1;
+ #else
+ try {
+ /* 8A268 has this */
+ #if 0
+ resultHand = mds.dlGetFirst(query, recordAttrs, record);
+ #else
+ /* TOT */
+ resultHand = mds.dlGetFirst(query, recordAttrs, NULL, record);
+ #endif
+ }
+ catch(...) {
+ printf("doLookup: dlGetFirst threw exception!\n");
+ return -1;
+ }
+ #endif
+ if(resultHand == 0) {
+ printf("doLookup: no record found\n");
+ return -1;
+ }
+
+ /* we could examine the record here of we wanted to */
+ mds.dlFreeUniqueId(record);
+ freeAttrs(&recordAttrs);
+ mds.dlAbortQuery(resultHand);
+ return 0;
+}
+
+/* nothing here for now */
+int mdsLookupInit(TestParams *tp)
+{
+ return 0;
+}
+
+int mdsLookup(TestParams *testParams)
+{
+ for(unsigned loopNum=0; loopNum<testParams->numLoops; loopNum++) {
+ if(testParams->verbose) {
+ printf("mdsLookup loop %d\n", loopNum);
+ }
+ else if(!testParams->quiet) {
+ printChar(testParams->progressChar);
+ }
+ try {
+ MDSClient::Directory &mds = MDSClient::mds();
+ uint32 val = CSSM_ALGID_SHA1;
+ if(doLookup(mds, MDS_CDSADIR_CSP_CAPABILITY_RECORDTYPE,
+ "AlgType", &val, sizeof(uint32),
+ CSSM_DB_ATTRIBUTE_FORMAT_UINT32, CSSM_DB_EQUAL)) {
+ printf("***thread %u aborting\n", testParams->threadNum);
+ return -1;
+ }
+ }
+ catch(CssmError &err) {
+ cssmPerror("MDS init", err.error);
+ printf("mdsLookup: MDSClient::mds() threw CssmError!\n");
+ return -1;
+ }
+ catch(...) {
+ printf("mdsLookup: MDSClient::mds() threw exception!\n");
+ return -1;
+ }
+ #if DO_PAUSE
+ fpurge(stdin);
+ printf("Hit CR to continue: ");
+ getchar();
+ #endif
+ }
+ return 0;
+}
+
projects / apple / security.git / blobdiff