You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

74 lines
2.6 KiB

/*
* Copyright 2002-2019 Intel Corporation.
*
* This software is provided to you as Sample Source Code as defined in the accompanying
* End User License Agreement for the Intel(R) Software Development Products ("Agreement")
* section 1.L.
*
* This software and the related documents are provided as is, with no express or implied
* warranties, other than those that are expressly stated in the License.
*/
/*! @file
* An example of SMC application.
*/
#include "smc_util.h"
#include "../Utils/sys_memory.h"
#include <stdio.h>
/*!
* Exit with the specified error message
*/
static void Abort(string msg)
{
fprintf (stderr, "%s\n", msg.c_str());
exit(1);
}
/*!
* The main procedure of the application.
*/
int main(int argc, char *argv[])
{
fprintf (stderr, "SMC in the image of the application\n");
// buffer to move foo routines into and execute
static char staticBuffer1[PI_FUNC::MAX_SIZE];
static char staticBuffer2[PI_FUNC::MAX_SIZE];
static char staticBuffer3[PI_FUNC::MAX_SIZE];
// Set read-write-execute protection for the buffer
size_t pageSize = GetPageSize();
char * firstPage1 = (char *)(((size_t)staticBuffer1) & ~(pageSize - 1));
char * endPage1 = (char *)(((size_t)staticBuffer1 + sizeof(staticBuffer1) + pageSize - 1) & ~(pageSize - 1));
char *firstPage2 = (char *)(((size_t)staticBuffer2) & ~(pageSize - 1));
char * endPage2 = (char *)(((size_t)staticBuffer2 + sizeof(staticBuffer2) + pageSize - 1) & ~(pageSize - 1));
char *firstPage3 = (char *)(((size_t)staticBuffer3) & ~(pageSize - 1));
char * endPage3 = (char *)(((size_t)staticBuffer3 + sizeof(staticBuffer3) + pageSize - 1) & ~(pageSize - 1));
if (!MemProtect(firstPage1, endPage1 - firstPage1, MEM_READ_WRITE_EXEC)) {Abort("MemProtect failed");}
if (!MemProtect(firstPage2, endPage2 - firstPage2, MEM_READ_WRITE_EXEC)) {Abort("MemProtect failed");}
if (!MemProtect(firstPage3, endPage3 - firstPage3, MEM_READ_WRITE_EXEC)) {Abort("MemProtect failed");}
int printed = 0;
FOO_FUNC fooFunc;
FILE *fp = fopen ("smcapp3_a.out", "w");
fprintf (fp, "%p\n", staticBuffer1);
fclose(fp);
fooFunc.Copy(staticBuffer1).Execute().AssertStatus();
fp = fopen ("smcapp3_b.out", "w");
fprintf (fp, "%p\n", staticBuffer2);
fclose(fp);
fooFunc.Copy(staticBuffer2).Execute().AssertStatus();
fooFunc.Copy(staticBuffer3).Execute().AssertStatus();
return 0;
}
/* ===================================================================== */
/* eof */
/* ===================================================================== */