Ejemplo n.º 1
0
int main(int argc, char** argv)
{
    int 	rc = 0;
    int start;
    int update;
    int complete;
    int verbose;

    unsigned char buffer1[] = "1234567890123456789012345678901234567890123456789012345678901234";	
    unsigned char buffer2[] = "12345678901234567890123456789012";
    unsigned char expect[] = {0xbf, 0x63, 0xee, 0xe7, 0x1c, 0x21, 0x1f, 0x83,
			      0xc9, 0x63, 0xf1, 0x41, 0xd2, 0xff, 0xd4, 0x0a,
			      0x01, 0x9f, 0xb0, 0x90};
    TPM_DIGEST	actual;
    int		not_equal;
    uint32_t 	maxNumBytes = 0;	/* return from TPM_SHA1Start */

    /* get caller's command line arguments */
    if (rc == 0) {
	rc = getArgs(&start,
		     &update,
		     &complete,
		     &verbose,
		     argc, argv);
    }
    if ((rc == 0) && start) {
	rc = TPM_SHA1Start(&maxNumBytes);	/* ignore, buffer is small */
	if (rc != 0) {
	    printf("sha1parts: Error in TPM_SHA1Start\n");
	}
    }    
    if ((rc == 0) && update) {
	rc = TPM_SHA1Update(buffer1, 64);
 	if (rc != 0) {
	    printf("sha1parts: Error in TPM_SHA1Update\n");
	}
    }    
    if ((rc == 0) && complete) {
	rc = TPM_SHA1Complete(buffer2, 32, actual);
	if (rc != 0) {
	    printf("sha1parts: Error in TPM_SHA1Complete\n");
	}
    }    
    if ((rc == 0) && complete) {
	not_equal = memcmp(expect, actual, TPM_DIGEST_SIZE);
	if (not_equal) {
	    printf("sha1parts: Error in digest\n");
	    rc = -1;
	}
    }
    if (rc != 0) {
	printf("sha1parts: Error\n");
    }
    return rc;
}
Ejemplo n.º 2
0
int main(int argc, char * argv[]) {
	char * data = NULL;
	char * filename = NULL;
	int i = 1;
	int offset = 0;
	uint32_t len;
	uint32_t ret = 0;
	unsigned char hashbuffer[TPM_HASH_SIZE];
	unsigned char pcrValue[TPM_HASH_SIZE];
	int index = -1;
	TPM_BOOL verbose = FALSE;
	uint32_t maxNumBytes = 0;	/* return from TPM_SHA1Start */

	TPM_setlog(0);
	
	while (i < argc) {
		if (!strcmp("-ic",argv[i])) {
			i++;
			if (i < argc) {
				data = argv[i];
			} else {
				printf("Missing mandatory parameter for -ic.\n");
				usage();
			}
		} else
		if (!strcmp("-if",argv[i])) {
			i++;
			if (i < argc) {
				filename = argv[i];
			} else {
				printf("Missing mandatory parameter for -if.\n");
				usage();
			}
		} else
		if (!strcmp("-ix",argv[i])) {
			i++;
			if (i < argc) {
				index = atoi(argv[i]);
			} else {
				printf("Missing mandatory parameter for -ix.\n");
				usage();
			}
		} else
		if (!strcmp("-v",argv[i])) {
			TPM_setlog(1);
			verbose = TRUE;
		} else
		    if (!strcmp("-h",argv[i])) {
			usage();
		} else {
			printf("\n%s is not a valid option\n", argv[i]);
			usage();
		}
		i++;
	}

	if (NULL == data && filename == NULL) {
		printf("Input parameters wrong or missing!\n");
		usage();
	}
	/* Common processing, for data or file */
	ret = TPM_SHA1Start(&maxNumBytes);
	if (0 != ret) {
	    printf("Error from TPM_SHA1Start(): %d (0x%x)\n",
		   ret,
		   ret);
	    exit(-1);
	}
	if (maxNumBytes < 64) {
	    printf("The size parameter returned from SHA1Start() is bad.\n");
	    exit(-1);
	}

	/* If the connection is to a hardware device, the driver limits the buffer size to
	   2048, even if the TPM can support a larger size.

	   The -20 is presumably for the TPM command packet tag, paramSize, ordinal,
	   numBytes, although this only adds up to 14 bytes.
	*/
#if defined TPM_USE_CHARDEV || defined XCRYPTO_USE_CCA
	maxNumBytes = MIN(maxNumBytes, (2 * 1024) - 20);
	maxNumBytes &= ~63;
#else
	/* Even if the connection is to a socket, it may be then through a proxy to a
	   hardware device.  Setting this env variable flag again limits the buffer to
	   2048. */
	if (getenv("TPM_HW_DRIVER") != NULL) {
	    maxNumBytes = MIN(maxNumBytes, (2 * 1024) - 20);
	    maxNumBytes &= ~63;
	}
#endif

	
	if (NULL != data) {
		len = strlen(data);
		printf("SHA1 hash for '%s': ",data);
		offset = 0;
		while (len > 64) {
			unsigned int chunksize;
			chunksize = MIN(maxNumBytes, len & ~63);

			if (verbose)
				printf("Chunksize to hash: %d\n",chunksize);

			ret = TPM_SHA1Update(&data[offset], chunksize);
			if (ret != 0) {
				printf("Error %s from SHA1Update.\n",
				       TPM_GetErrMsg(ret));
				exit(ret);
			}
			offset += chunksize;
			len -= chunksize;
		}

		if (index >= 0) {
			ret = TPM_SHA1CompleteExtend(&data[offset],len,
			                             index,
			                             hashbuffer,
			                             pcrValue);
		} else {
			ret = TPM_SHA1Complete(&data[offset],len,hashbuffer);
		}
		if (0 == ret) {
			printf("Hash: ");
			i = 0;
			while (i < (int)sizeof(hashbuffer)) {
				printf("%02x",hashbuffer[i]);
				i++;
			}
			printf("\n");
		
			if (index >= 0) {
				printf("New value of PCR: ");
				i = 0;
				while (i < (int)sizeof(pcrValue)) {
					printf("%02x",pcrValue[i]);
					i++;
				}
				printf("\n");
			}
		} else {
			printf("Error '%s' from SHA1Complete/SHA1CompleteExtend.\n",
			       TPM_GetErrMsg(ret));
		}
	}
	
	if (NULL != filename) {
		FILE * f = fopen(filename,"rb");
		if (NULL != f) {
			int n = 0;
			uint32_t chunksize;
			char *buf;
			uint32_t total = 0;
			printf("SHA1 hash for file '%s': \n",filename);

			buf = malloc(maxNumBytes);
			if (!buf) {
				printf("Could not allocated buffer.\n");
				exit(-1);
			}

			while ((n = fread(buf,1,maxNumBytes,f)) >= 64) {

				chunksize = MIN(maxNumBytes, 
				                (uint32_t)(n & ~63));
				
				if (verbose)
					printf("Chunksize to hash: %d\n",
					       chunksize);
				ret = TPM_SHA1Update(buf,chunksize);
				if (ret != 0) {
					printf("Error %s from SHA1Update.\n",
					       TPM_GetErrMsg(ret));
					exit(-1);
				}

				total += chunksize;
				if ((chunksize - n)) {
					offset = chunksize;
					n &= 63;
					break;
				}
			}
			total += n;
			
			if (index >= 0) {
				ret = TPM_SHA1CompleteExtend(&buf[offset],n,
				                             index,
				                             hashbuffer,
				                             pcrValue);
			} else {
				ret = TPM_SHA1Complete(&buf[offset],n,hashbuffer);
			}

			fclose(f);
			free(buf);

			if (0 == ret) {
				printf("Hash: ");
				i = 0;
				while (i < (int)sizeof(hashbuffer)) {
					printf("%02x",hashbuffer[i]);
					i++;
				}
				printf("\n");
			
				if (index >= 0) {
					printf("New value of PCR: ");
					i = 0;
					while (i < (int)sizeof(pcrValue)) {
						printf("%02x",pcrValue[i]);
						i++;
					}
					printf("\n");
				}
			} else {
				printf("Error '%s' from SHA1Complete/SHA1CompleteExtend.\n",
				       TPM_GetErrMsg(ret));
			}
		} else {
			printf("Could not find file '%s'.\n",filename);
		}
	}
	return 0;
}