int main(int argc, char **argv) {
KSI_CTX *ksi = NULL;
int res = KSI_UNKNOWN_ERROR;
FILE *in = NULL;
FILE *out = NULL;
KSI_DataHasher *hsr = NULL;
KSI_DataHash *hsh = NULL;
KSI_Signature *sign = NULL;
unsigned char *raw = NULL;
unsigned raw_len;
unsigned char buf[1024];
unsigned buf_len;
char *signerIdentity = NULL;
FILE *logFile = NULL;
/* Handle command line parameters */
/* Handle command line parameters */
if (argc != 7) {
fprintf(stderr, "Usage:\n"
" %s <in-data-file> <out-sign-file> <aggregator-uri> <user> <pass> <pub-file url | -> \n", argv[0]);
res = KSI_INVALID_ARGUMENT;
goto cleanup;
}
/* Input file */
in = fopen(argv[1], "rb");
if (in == NULL) {
fprintf(stderr, "Unable to open input file '%s'\n", argv[1]);
res = KSI_IO_ERROR;
goto cleanup;
}
/* Create new KSI context for this thread. */
res = KSI_CTX_new(&ksi);
if (res != KSI_OK) {
fprintf(stderr, "Unable to create context.\n");
goto cleanup;
}
logFile = fopen("ksi_sign.log", "w");
if (logFile == NULL) {
fprintf(stderr, "Unable to open log file.\n");
}
KSI_CTX_setLoggerCallback(ksi, KSI_LOG_StreamLogger, logFile);
KSI_CTX_setLogLevel(ksi, KSI_LOG_DEBUG);
KSI_LOG_info(ksi, "Using KSI version: '%s'", KSI_getVersion());
res = KSI_CTX_setAggregator(ksi, argv[3], argv[4], argv[5]);
if (res != KSI_OK) goto cleanup;
/* Check publications file url. */
if (strncmp("-", argv[6], 1)) {
res = KSI_CTX_setPublicationUrl(ksi, argv[6]);
if (res != KSI_OK) {
fprintf(stderr, "Unable to set publications file url.\n");
goto cleanup;
}
}
/* Create a data hasher using default algorithm. */
res = KSI_DataHasher_open(ksi, KSI_getHashAlgorithmByName("default"), &hsr);
if (res != KSI_OK) {
fprintf(stderr, "Unable to create hasher.\n");
goto cleanup;
}
/* Read the input file and calculate the hash of its contents. */
while (!feof(in)) {
buf_len = (unsigned)fread(buf, 1, sizeof(buf), in);
/* Add next block to the calculation. */
res = KSI_DataHasher_add(hsr, buf, buf_len);
if (res != KSI_OK) {
fprintf(stderr, "Unable to add data to hasher.\n");
goto cleanup;
}
}
/* Close the data hasher and retreive the data hash. */
res = KSI_DataHasher_close(hsr, &hsh);
if (res != KSI_OK) {
fprintf(stderr, "Unable to create hash.\n");
goto cleanup;
}
/* Sign the data hash. */
res = KSI_createSignature(ksi, hsh, &sign);
if (res != KSI_OK) {
fprintf(stderr, "Unable to sign %d.\n", res);
goto cleanup;
}
res = KSI_Signature_verify(sign, ksi);
if (res != KSI_OK) {
fprintf(stderr, "Failed to verify signature.\n");
goto cleanup;
}
/* Output the signer id */
res = KSI_Signature_getSignerIdentity(sign, &signerIdentity);
if (res == KSI_OK) {
printf("Signer id: %s\n", signerIdentity);
} else {
fprintf(stderr, "Unable to extract signer identity.\n");
}
/* Serialize the signature. */
res = KSI_Signature_serialize(sign, &raw, &raw_len);
if (res != KSI_OK) {
fprintf(stderr, "Unable to serialize signature.");
goto cleanup;
}
/* Output file */
out = fopen(argv[2], "wb");
if (out == NULL) {
fprintf(stderr, "Unable to open input file '%s'\n", argv[2]);
res = KSI_IO_ERROR;
goto cleanup;
}
/* Write the signature file. */
if (!fwrite(raw, 1, raw_len, out)) {
fprintf(stderr, "Unable to write output file.\n");
res = KSI_IO_ERROR;
goto cleanup;
}
/* Only print message when signature output is not stdout. */
if (out != NULL) {
printf("Signature saved.\n");
}
res = KSI_OK;
cleanup:
if (logFile != NULL) fclose(logFile);
if (res != KSI_OK && ksi != NULL) {
KSI_ERR_statusDump(ksi, stderr);
}
if (in != NULL) fclose(in);
if (out != NULL) fclose(out);
KSI_free(signerIdentity);
KSI_Signature_free(sign);
KSI_DataHash_free(hsh);
KSI_DataHasher_free(hsr);
KSI_free(raw);
KSI_CTX_free(ksi);
return res;
}
int main(int argc, char **argv) {
KSI_CTX *ksi = NULL;
int res = KSI_UNKNOWN_ERROR;
KSI_DataHash *hsh = NULL;
KSI_RequestHandle *handle[REQUESTS];
KSI_NetworkClient *http = NULL;
FILE *logFile = NULL;
size_t i;
KSI_DataHasher *hsr = NULL;
const KSI_CertConstraint pubFileCertConstr[] = {
{ KSI_CERT_EMAIL, "*****@*****.**"},
{ NULL, NULL }
};
struct {
size_t ok;
size_t nok;
} stat;
stat.ok = 0;
stat.nok = 0;
/* Create new KSI context for this thread. */
res = KSI_CTX_new(&ksi);
if (res != KSI_OK) {
fprintf(stderr, "Unable to create context.\n");
goto cleanup;
}
res = KSI_CTX_setDefaultPubFileCertConstraints(ksi, pubFileCertConstr);
if (res != KSI_OK) {
fprintf(stderr, "Unable to configure publications file cert constraints.\n");
goto cleanup;
}
/* Configure the logger. */
res = OpenLogging(ksi, "multi_curl.log", &logFile);
if (res != KSI_OK) goto cleanup;
KSI_LOG_info(ksi, "Using KSI version: '%s'", KSI_getVersion());
/* Check publications file url. */
res = KSI_CTX_setPublicationUrl(ksi, "http://verify.guardtime.com/ksi-publications.bin");
if (res != KSI_OK) {
fprintf(stderr, "Unable to set publications file url.\n");
goto cleanup;
}
res = KSI_HttpClient_new(ksi, &http);
if (res != KSI_OK) {
fprintf(stderr, "Unable to create http client.\n");
goto cleanup;
}
res = KSI_HttpClient_setAggregator(http, "http://ksigw.test.guardtime.com:3332", "anon", "anon");
if (res != KSI_OK) {
fprintf(stderr, "Unable to set aggregator url.\n");
goto cleanup;
}
KSI_HttpClient_setReadTimeoutSeconds(http, 10);
KSI_HttpClient_setConnectTimeoutSeconds(http, 10);
for (i = 0; i < REQUESTS; i++) {
char buf[100];
size_t len;
KSI_AggregationReq *req = NULL;
len = KSI_snprintf(buf, sizeof(buf), "Hello %d", i);
res = KSI_DataHash_create(ksi, buf, len, KSI_getHashAlgorithmByName("default"), &hsh);
if (res != KSI_OK) {
fprintf(stderr, "Unable to create hash.");
goto cleanup;
}
res = KSI_AggregationReq_new(ksi, &req);
if (res != KSI_OK) {
fprintf(stderr, "Unable to create request.");
goto cleanup;
}
res = KSI_AggregationReq_setRequestHash(req, hsh);
if (res != KSI_OK) {
fprintf(stderr, "Unable to set request hash.");
goto cleanup;
}
res = KSI_NetworkClient_sendSignRequest(http, req, &handle[i]);
if (res != KSI_OK) {
fprintf(stderr, "Unable to send aggregation request.");
goto cleanup;
}
KSI_AggregationReq_free(req);
}
res = KSI_NetworkClient_performAll(http, handle, REQUESTS);
if (res != KSI_OK) {
fprintf(stderr, "Unable to perform requests.");
goto cleanup;
}
for (i = 0; i < REQUESTS; i++) {
KSI_AggregationResp *resp = NULL;
res = KSI_RequestHandle_getAggregationResponse(handle[i], &resp);
if (res != KSI_OK) {
const KSI_RequestHandleStatus *st = NULL;
res = KSI_RequestHandle_getResponseStatus(handle[i], &st);
if (res == KSI_OK) {
printf("Status code = %ld: %s\n", st->code, st->errm);
}
KSI_ERR_statusDump(ksi, stdout);
stat.nok++;
} else {
stat.ok ++;
}
KSI_AggregationResp_free(resp);
}
printf("Requests:\n"
" Successful: %llu\n"
" Failed: %llu\n"
" TOTAL: %llu\n", (unsigned long long)stat.ok, (unsigned long long)stat.nok, (unsigned long long)(stat.ok + stat.nok));
res = KSI_OK;
cleanup:
if (logFile != NULL) fclose(logFile);
if (res != KSI_OK && ksi != NULL) {
KSI_ERR_statusDump(ksi, stderr);
}
KSI_DataHash_free(hsh);
KSI_DataHasher_free(hsr);
KSI_CTX_free(ksi);
return res;
}
