diff mbox series

[12/14] selftests/sgx: Add page permission and exception test

Message ID a6e69ea22a2694d252302af283ee3e3f023d3577.1631731214.git.reinette.chatre@intel.com
State Accepted
Commit abc5cec4735080d12d644c2d39f96cf98c0a367c
Headers show
Series selftests/sgx: Oversubscription, page permission, thread entry | expand

Commit Message

Reinette Chatre Sept. 15, 2021, 8:31 p.m. UTC
The Enclave Page Cache Map (EPCM) is a secure structure used by the
processor to track the contents of the enclave page cache. The EPCM
contains permissions with which enclave pages can be accessed. SGX
support allows EPCM and PTE page permissions to differ - as long as
the PTE permissions do not exceed the EPCM permissions.

Add a test to ensure that (1) PTE permissions can be changed as long as
they do not exceed EPCM permissions, and (2) even if EPCM permissions
allow a page to be written to, if the PTE permissions do not then a #PF
should be generated when attempting to write to a (from PTE perspective)
read-only page.

This introduces the first test of SGX exception handling. In this test
the issue that caused the exception (PTE page permissions) can be fixed
from outside the enclave and after doing so it is possible to re-enter
enclave at original entrypoint with ERESUME.

Signed-off-by: Reinette Chatre <reinette.chatre@intel.com>
---
 tools/testing/selftests/sgx/defines.h   |  14 +++
 tools/testing/selftests/sgx/main.c      | 134 ++++++++++++++++++++++++
 tools/testing/selftests/sgx/test_encl.c |  21 ++++
 3 files changed, 169 insertions(+)

Comments

Jarkko Sakkinen Sept. 16, 2021, 3:21 p.m. UTC | #1
On Wed, 2021-09-15 at 13:31 -0700, Reinette Chatre wrote:
> The Enclave Page Cache Map (EPCM) is a secure structure used by the

> processor to track the contents of the enclave page cache. The EPCM

> contains permissions with which enclave pages can be accessed. SGX

> support allows EPCM and PTE page permissions to differ - as long as

> the PTE permissions do not exceed the EPCM permissions.

> 

> Add a test to ensure that (1) PTE permissions can be changed as long as

> they do not exceed EPCM permissions, and (2) even if EPCM permissions

> allow a page to be written to, if the PTE permissions do not then a #PF

> should be generated when attempting to write to a (from PTE perspective)

> read-only page.

> 

> This introduces the first test of SGX exception handling. In this test

> the issue that caused the exception (PTE page permissions) can be fixed

> from outside the enclave and after doing so it is possible to re-enter

> enclave at original entrypoint with ERESUME.

> 

> Signed-off-by: Reinette Chatre <reinette.chatre@intel.com>

> ---

>  tools/testing/selftests/sgx/defines.h   |  14 +++

>  tools/testing/selftests/sgx/main.c      | 134 ++++++++++++++++++++++++

>  tools/testing/selftests/sgx/test_encl.c |  21 ++++

>  3 files changed, 169 insertions(+)

> 

> diff --git a/tools/testing/selftests/sgx/defines.h b/tools/testing/selftests/sgx/defines.h

> index 9ea0c7882dfb..0bbda6f0c7d3 100644

> --- a/tools/testing/selftests/sgx/defines.h

> +++ b/tools/testing/selftests/sgx/defines.h

> @@ -21,6 +21,8 @@

>  enum encl_op_type {

>  	ENCL_OP_PUT_TO_BUFFER,

>  	ENCL_OP_GET_FROM_BUFFER,

> +	ENCL_OP_PUT_TO_ADDRESS,

> +	ENCL_OP_GET_FROM_ADDRESS,

>  	ENCL_OP_MAX,

>  };

>  

> @@ -38,4 +40,16 @@ struct encl_op_get_from_buf {

>  	uint64_t value;

>  };

>  

> +struct encl_op_put_to_addr {

> +	struct encl_op_header header;

> +	uint64_t value;

> +	uint64_t addr;

> +};

> +

> +struct encl_op_get_from_addr {

> +	struct encl_op_header header;

> +	uint64_t value;

> +	uint64_t addr;

> +};

> +

>  #endif /* DEFINES_H */

> diff --git a/tools/testing/selftests/sgx/main.c b/tools/testing/selftests/sgx/main.c

> index 3eb9b89ece5f..308cf09ab02a 100644

> --- a/tools/testing/selftests/sgx/main.c

> +++ b/tools/testing/selftests/sgx/main.c

> @@ -21,6 +21,7 @@

>  #include "main.h"

>  

>  static const uint64_t MAGIC = 0x1122334455667788ULL;

> +static const uint64_t MAGIC2 = 0x8877665544332211ULL;

>  vdso_sgx_enter_enclave_t vdso_sgx_enter_enclave;

>  

>  struct vdso_symtab {

> @@ -107,6 +108,25 @@ static Elf64_Sym *vdso_symtab_get(struct vdso_symtab *symtab, const char *name)

>  	return NULL;

>  }

>  

> +/*

> + * Return the offset in the enclave where the data segment can be found.

> + * The first RW segment loaded is the TCS, skip that to get info on the

> + * data segment.

> + */

> +static off_t encl_get_data_offset(struct encl *encl)

> +{

> +	int i;

> +

> +	for (i = 0; i < encl->nr_segments; i++) {

> +		struct encl_segment *seg = &encl->segment_tbl[i];

> +

> +		if (i != 0 && seg->prot == (PROT_READ | PROT_WRITE))

> +			return seg->offset;


So, why not

	for (i = 1; i < encl->nr_segments; i++)

?

> +	}

> +

> +	return -1;

> +}

> +

>  FIXTURE(enclave) {

>  	struct encl encl;

>  	struct sgx_enclave_run run;

> @@ -373,4 +393,118 @@ TEST_F(enclave, clobbered_vdso_and_user_function)

>  	EXPECT_EQ(self->run.user_data, 0);

>  }

>  

> +/*

> + * Second page of .data segment is used to test changing PTE permissions.

> + * This spans the local encl_buffer within the test enclave.

> + *

> + * 1) Start with a sanity check: a value is written to the target page within

> + *    the enclave and read back to ensure target page can be written to.

> + * 2) Change PTE permissions (RW -> RO) of target page within enclave.

> + * 3) Repeat (1) - this time expecting a regular #PF communicated via the

> + *    vDSO.

> + * 4) Change PTE permissions of target page within enclave back to be RW.

> + * 5) Repeat (1) by resuming enclave, now expected to be possible to write to

> + *    and read from target page within enclave.

> + */

> +TEST_F(enclave, pte_permissions)

> +{

> +	struct encl_op_get_from_addr get_addr_op;

> +	struct encl_op_put_to_addr put_addr_op;

> +	unsigned long data_start;

> +	int ret;

> +

> +	ASSERT_TRUE(setup_test_encl(ENCL_HEAP_SIZE_DEFAULT, &self->encl, _metadata));

> +

> +	memset(&self->run, 0, sizeof(self->run));

> +	self->run.tcs = self->encl.encl_base;

> +

> +	data_start = self->encl.encl_base +

> +		     encl_get_data_offset(&self->encl) +

> +		     PAGE_SIZE;

> +

> +	/*

> +	 * Sanity check to ensure it is possible to write to page that will

> +	 * have its permissions manipulated.

> +	 */

> +

> +	/* Write MAGIC to page */

> +	put_addr_op.value = MAGIC;

> +	put_addr_op.addr = data_start;

> +	put_addr_op.header.type = ENCL_OP_PUT_TO_ADDRESS;

> +

> +	EXPECT_EQ(ENCL_CALL(&put_addr_op, &self->run, true), 0);

> +

> +	EXPECT_EEXIT(&self->run);

> +	EXPECT_EQ(self->run.exception_vector, 0);

> +	EXPECT_EQ(self->run.exception_error_code, 0);

> +	EXPECT_EQ(self->run.exception_addr, 0);

> +

> +	/*

> +	 * Read memory that was just written to, confirming that it is the

> +	 * value previously written (MAGIC).

> +	 */

> +	get_addr_op.value = 0;

> +	get_addr_op.addr = data_start;

> +	get_addr_op.header.type = ENCL_OP_GET_FROM_ADDRESS;

> +

> +	EXPECT_EQ(ENCL_CALL(&get_addr_op, &self->run, true), 0);

> +

> +	EXPECT_EQ(get_addr_op.value, MAGIC);

> +	EXPECT_EEXIT(&self->run);

> +	EXPECT_EQ(self->run.exception_vector, 0);

> +	EXPECT_EQ(self->run.exception_error_code, 0);

> +	EXPECT_EQ(self->run.exception_addr, 0);

> +

> +	/* Change PTE permissions of target page within the enclave */

> +	ret = mprotect((void *)data_start, PAGE_SIZE, PROT_READ);

> +	if (ret)

> +		perror("mprotect");

> +

> +	/*

> +	 * PTE permissions of target page changed to read-only, EPCM

> +	 * permissions unchanged (EPCM permissions are RW), attempt to

> +	 * write to the page, expecting a regular #PF.

> +	 */

> +

> +	put_addr_op.value = MAGIC2;

> +

> +	EXPECT_EQ(ENCL_CALL(&put_addr_op, &self->run, true), 0);

> +

> +	EXPECT_EQ(self->run.exception_vector, 14);

> +	EXPECT_EQ(self->run.exception_error_code, 0x7);

> +	EXPECT_EQ(self->run.exception_addr, data_start);

> +

> +	self->run.exception_vector = 0;

> +	self->run.exception_error_code = 0;

> +	self->run.exception_addr = 0;

> +

> +	/*

> +	 * Change PTE permissions back to enable enclave to write to the

> +	 * target page and resume enclave - do not expect any exceptions this

> +	 * time.

> +	 */

> +	ret = mprotect((void *)data_start, PAGE_SIZE, PROT_READ | PROT_WRITE);

> +	if (ret)

> +		perror("mprotect");

> +

> +	EXPECT_EQ(vdso_sgx_enter_enclave((unsigned long)&put_addr_op, 0,

> +					 0, ERESUME, 0, 0, &self->run),

> +		 0);

> +

> +	EXPECT_EEXIT(&self->run);

> +	EXPECT_EQ(self->run.exception_vector, 0);

> +	EXPECT_EQ(self->run.exception_error_code, 0);

> +	EXPECT_EQ(self->run.exception_addr, 0);

> +

> +	get_addr_op.value = 0;

> +

> +	EXPECT_EQ(ENCL_CALL(&get_addr_op, &self->run, true), 0);

> +

> +	EXPECT_EQ(get_addr_op.value, MAGIC2);

> +	EXPECT_EEXIT(&self->run);

> +	EXPECT_EQ(self->run.exception_vector, 0);

> +	EXPECT_EQ(self->run.exception_error_code, 0);

> +	EXPECT_EQ(self->run.exception_addr, 0);

> +}

> +

>  TEST_HARNESS_MAIN

> diff --git a/tools/testing/selftests/sgx/test_encl.c b/tools/testing/selftests/sgx/test_encl.c

> index 4e8da738173f..5d86e3e6456a 100644

> --- a/tools/testing/selftests/sgx/test_encl.c

> +++ b/tools/testing/selftests/sgx/test_encl.c

> @@ -4,6 +4,11 @@

>  #include <stddef.h>

>  #include "defines.h"

>  

> +/*

> + * Data buffer spanning two pages that will be placed first in .data

> + * segment. Even if not used internally the second page is needed by

> + * external test manipulating page permissions.

> + */

>  static uint8_t encl_buffer[8192] = { 1 };

>  

>  static void *memcpy(void *dest, const void *src, size_t n)

> @@ -30,11 +35,27 @@ static void do_encl_op_get_from_buf(void *op)

>  	memcpy(&op2->value, &encl_buffer[0], 8);

>  }

>  

> +static void do_encl_op_put_to_addr(void *_op)

> +{

> +	struct encl_op_put_to_addr *op = _op;

> +

> +	memcpy((void *)op->addr, &op->value, 8);

> +}

> +

> +static void do_encl_op_get_from_addr(void *_op)

> +{

> +	struct encl_op_get_from_addr *op = _op;

> +

> +	memcpy(&op->value, (void *)op->addr, 8);

> +}

> +

>  void encl_body(void *rdi,  void *rsi)

>  {

>  	const void (*encl_op_array[ENCL_OP_MAX])(void *) = {

>  		do_encl_op_put_to_buf,

>  		do_encl_op_get_from_buf,

> +		do_encl_op_put_to_addr,

> +		do_encl_op_get_from_addr,

>  	};

>  

>  	struct encl_op_header *op = (struct encl_op_header *)rdi;


/Jarkko
Dave Hansen Sept. 16, 2021, 3:30 p.m. UTC | #2
On 9/15/21 1:31 PM, Reinette Chatre wrote:
> Add a test to ensure that (1) PTE permissions can be changed as long as

> they do not exceed EPCM permissions, and (2) even if EPCM permissions

> allow a page to be written to, if the PTE permissions do not then a #PF

> should be generated when attempting to write to a (from PTE perspective)

> read-only page.


It took me a minute to figure out what this was trying to say.  Maybe
breaking it down into these three steps would help:

Add a test that:
 (1) Creates an SGX enclave page with writable EPCM permission
 (2) Changes the PTE permission on the page to read-only.  This should
     be permitted because the permission does not exceed the EPCM
     permission.
 (3) Attempts a write to the page and generate a page fault (#PF)
     because of the read-only PTE.
Reinette Chatre Sept. 16, 2021, 3:37 p.m. UTC | #3
Hi Jarkko,

On 9/16/2021 8:21 AM, Jarkko Sakkinen wrote:
> On Wed, 2021-09-15 at 13:31 -0700, Reinette Chatre wrote:


...

>> +/*

>> + * Return the offset in the enclave where the data segment can be found.

>> + * The first RW segment loaded is the TCS, skip that to get info on the

>> + * data segment.

>> + */

>> +static off_t encl_get_data_offset(struct encl *encl)

>> +{

>> +	int i;

>> +

>> +	for (i = 0; i < encl->nr_segments; i++) {

>> +		struct encl_segment *seg = &encl->segment_tbl[i];

>> +

>> +		if (i != 0 && seg->prot == (PROT_READ | PROT_WRITE))

>> +			return seg->offset;

> 

> So, why not

> 

> 	for (i = 1; i < encl->nr_segments; i++)

> 

> ?


Thank you for catching this. Will do.

Reinette
Reinette Chatre Sept. 16, 2021, 3:50 p.m. UTC | #4
Hi Dave,

On 9/16/2021 8:30 AM, Dave Hansen wrote:
> On 9/15/21 1:31 PM, Reinette Chatre wrote:

>> Add a test to ensure that (1) PTE permissions can be changed as long as

>> they do not exceed EPCM permissions, and (2) even if EPCM permissions

>> allow a page to be written to, if the PTE permissions do not then a #PF

>> should be generated when attempting to write to a (from PTE perspective)

>> read-only page.

> 

> It took me a minute to figure out what this was trying to say.


The goal was to describe what features/functionalities are being tested. 
You accurately point out that it is not clear how the test implemented 
in the patch matches with these test goals.

>  Maybe

> breaking it down into these three steps would help:

> 

> Add a test that:

>   (1) Creates an SGX enclave page with writable EPCM permission

>   (2) Changes the PTE permission on the page to read-only.  This should

>       be permitted because the permission does not exceed the EPCM

>       permission.

>   (3) Attempts a write to the page and generate a page fault (#PF)

>       because of the read-only PTE.


Thank you for the suggestion. What I understand from your feedback is 
that I should mix the description of the actual test with what 
features/functionalities are being tested. You do so in your suggestion 
for (2) and to do the same for (3) I now plan to expand it to:

    (3) Attempts a write to the page. This should generate a page fault
        (#PF) because of the read-only PTE even though the EPCM
        permissions allow the page to be written to.

Reinette
diff mbox series

Patch

diff --git a/tools/testing/selftests/sgx/defines.h b/tools/testing/selftests/sgx/defines.h
index 9ea0c7882dfb..0bbda6f0c7d3 100644
--- a/tools/testing/selftests/sgx/defines.h
+++ b/tools/testing/selftests/sgx/defines.h
@@ -21,6 +21,8 @@ 
 enum encl_op_type {
 	ENCL_OP_PUT_TO_BUFFER,
 	ENCL_OP_GET_FROM_BUFFER,
+	ENCL_OP_PUT_TO_ADDRESS,
+	ENCL_OP_GET_FROM_ADDRESS,
 	ENCL_OP_MAX,
 };
 
@@ -38,4 +40,16 @@  struct encl_op_get_from_buf {
 	uint64_t value;
 };
 
+struct encl_op_put_to_addr {
+	struct encl_op_header header;
+	uint64_t value;
+	uint64_t addr;
+};
+
+struct encl_op_get_from_addr {
+	struct encl_op_header header;
+	uint64_t value;
+	uint64_t addr;
+};
+
 #endif /* DEFINES_H */
diff --git a/tools/testing/selftests/sgx/main.c b/tools/testing/selftests/sgx/main.c
index 3eb9b89ece5f..308cf09ab02a 100644
--- a/tools/testing/selftests/sgx/main.c
+++ b/tools/testing/selftests/sgx/main.c
@@ -21,6 +21,7 @@ 
 #include "main.h"
 
 static const uint64_t MAGIC = 0x1122334455667788ULL;
+static const uint64_t MAGIC2 = 0x8877665544332211ULL;
 vdso_sgx_enter_enclave_t vdso_sgx_enter_enclave;
 
 struct vdso_symtab {
@@ -107,6 +108,25 @@  static Elf64_Sym *vdso_symtab_get(struct vdso_symtab *symtab, const char *name)
 	return NULL;
 }
 
+/*
+ * Return the offset in the enclave where the data segment can be found.
+ * The first RW segment loaded is the TCS, skip that to get info on the
+ * data segment.
+ */
+static off_t encl_get_data_offset(struct encl *encl)
+{
+	int i;
+
+	for (i = 0; i < encl->nr_segments; i++) {
+		struct encl_segment *seg = &encl->segment_tbl[i];
+
+		if (i != 0 && seg->prot == (PROT_READ | PROT_WRITE))
+			return seg->offset;
+	}
+
+	return -1;
+}
+
 FIXTURE(enclave) {
 	struct encl encl;
 	struct sgx_enclave_run run;
@@ -373,4 +393,118 @@  TEST_F(enclave, clobbered_vdso_and_user_function)
 	EXPECT_EQ(self->run.user_data, 0);
 }
 
+/*
+ * Second page of .data segment is used to test changing PTE permissions.
+ * This spans the local encl_buffer within the test enclave.
+ *
+ * 1) Start with a sanity check: a value is written to the target page within
+ *    the enclave and read back to ensure target page can be written to.
+ * 2) Change PTE permissions (RW -> RO) of target page within enclave.
+ * 3) Repeat (1) - this time expecting a regular #PF communicated via the
+ *    vDSO.
+ * 4) Change PTE permissions of target page within enclave back to be RW.
+ * 5) Repeat (1) by resuming enclave, now expected to be possible to write to
+ *    and read from target page within enclave.
+ */
+TEST_F(enclave, pte_permissions)
+{
+	struct encl_op_get_from_addr get_addr_op;
+	struct encl_op_put_to_addr put_addr_op;
+	unsigned long data_start;
+	int ret;
+
+	ASSERT_TRUE(setup_test_encl(ENCL_HEAP_SIZE_DEFAULT, &self->encl, _metadata));
+
+	memset(&self->run, 0, sizeof(self->run));
+	self->run.tcs = self->encl.encl_base;
+
+	data_start = self->encl.encl_base +
+		     encl_get_data_offset(&self->encl) +
+		     PAGE_SIZE;
+
+	/*
+	 * Sanity check to ensure it is possible to write to page that will
+	 * have its permissions manipulated.
+	 */
+
+	/* Write MAGIC to page */
+	put_addr_op.value = MAGIC;
+	put_addr_op.addr = data_start;
+	put_addr_op.header.type = ENCL_OP_PUT_TO_ADDRESS;
+
+	EXPECT_EQ(ENCL_CALL(&put_addr_op, &self->run, true), 0);
+
+	EXPECT_EEXIT(&self->run);
+	EXPECT_EQ(self->run.exception_vector, 0);
+	EXPECT_EQ(self->run.exception_error_code, 0);
+	EXPECT_EQ(self->run.exception_addr, 0);
+
+	/*
+	 * Read memory that was just written to, confirming that it is the
+	 * value previously written (MAGIC).
+	 */
+	get_addr_op.value = 0;
+	get_addr_op.addr = data_start;
+	get_addr_op.header.type = ENCL_OP_GET_FROM_ADDRESS;
+
+	EXPECT_EQ(ENCL_CALL(&get_addr_op, &self->run, true), 0);
+
+	EXPECT_EQ(get_addr_op.value, MAGIC);
+	EXPECT_EEXIT(&self->run);
+	EXPECT_EQ(self->run.exception_vector, 0);
+	EXPECT_EQ(self->run.exception_error_code, 0);
+	EXPECT_EQ(self->run.exception_addr, 0);
+
+	/* Change PTE permissions of target page within the enclave */
+	ret = mprotect((void *)data_start, PAGE_SIZE, PROT_READ);
+	if (ret)
+		perror("mprotect");
+
+	/*
+	 * PTE permissions of target page changed to read-only, EPCM
+	 * permissions unchanged (EPCM permissions are RW), attempt to
+	 * write to the page, expecting a regular #PF.
+	 */
+
+	put_addr_op.value = MAGIC2;
+
+	EXPECT_EQ(ENCL_CALL(&put_addr_op, &self->run, true), 0);
+
+	EXPECT_EQ(self->run.exception_vector, 14);
+	EXPECT_EQ(self->run.exception_error_code, 0x7);
+	EXPECT_EQ(self->run.exception_addr, data_start);
+
+	self->run.exception_vector = 0;
+	self->run.exception_error_code = 0;
+	self->run.exception_addr = 0;
+
+	/*
+	 * Change PTE permissions back to enable enclave to write to the
+	 * target page and resume enclave - do not expect any exceptions this
+	 * time.
+	 */
+	ret = mprotect((void *)data_start, PAGE_SIZE, PROT_READ | PROT_WRITE);
+	if (ret)
+		perror("mprotect");
+
+	EXPECT_EQ(vdso_sgx_enter_enclave((unsigned long)&put_addr_op, 0,
+					 0, ERESUME, 0, 0, &self->run),
+		 0);
+
+	EXPECT_EEXIT(&self->run);
+	EXPECT_EQ(self->run.exception_vector, 0);
+	EXPECT_EQ(self->run.exception_error_code, 0);
+	EXPECT_EQ(self->run.exception_addr, 0);
+
+	get_addr_op.value = 0;
+
+	EXPECT_EQ(ENCL_CALL(&get_addr_op, &self->run, true), 0);
+
+	EXPECT_EQ(get_addr_op.value, MAGIC2);
+	EXPECT_EEXIT(&self->run);
+	EXPECT_EQ(self->run.exception_vector, 0);
+	EXPECT_EQ(self->run.exception_error_code, 0);
+	EXPECT_EQ(self->run.exception_addr, 0);
+}
+
 TEST_HARNESS_MAIN
diff --git a/tools/testing/selftests/sgx/test_encl.c b/tools/testing/selftests/sgx/test_encl.c
index 4e8da738173f..5d86e3e6456a 100644
--- a/tools/testing/selftests/sgx/test_encl.c
+++ b/tools/testing/selftests/sgx/test_encl.c
@@ -4,6 +4,11 @@ 
 #include <stddef.h>
 #include "defines.h"
 
+/*
+ * Data buffer spanning two pages that will be placed first in .data
+ * segment. Even if not used internally the second page is needed by
+ * external test manipulating page permissions.
+ */
 static uint8_t encl_buffer[8192] = { 1 };
 
 static void *memcpy(void *dest, const void *src, size_t n)
@@ -30,11 +35,27 @@  static void do_encl_op_get_from_buf(void *op)
 	memcpy(&op2->value, &encl_buffer[0], 8);
 }
 
+static void do_encl_op_put_to_addr(void *_op)
+{
+	struct encl_op_put_to_addr *op = _op;
+
+	memcpy((void *)op->addr, &op->value, 8);
+}
+
+static void do_encl_op_get_from_addr(void *_op)
+{
+	struct encl_op_get_from_addr *op = _op;
+
+	memcpy(&op->value, (void *)op->addr, 8);
+}
+
 void encl_body(void *rdi,  void *rsi)
 {
 	const void (*encl_op_array[ENCL_OP_MAX])(void *) = {
 		do_encl_op_put_to_buf,
 		do_encl_op_get_from_buf,
+		do_encl_op_put_to_addr,
+		do_encl_op_get_from_addr,
 	};
 
 	struct encl_op_header *op = (struct encl_op_header *)rdi;