diff options
Diffstat (limited to 'common')
-rw-r--r-- | common/extension.c | 48 | ||||
-rw-r--r-- | common/tpm_registers.c | 10 |
2 files changed, 49 insertions, 9 deletions
diff --git a/common/extension.c b/common/extension.c index d28bfad337..63ae8f0495 100644 --- a/common/extension.c +++ b/common/extension.c @@ -10,10 +10,10 @@ #define CPRINTF(format, args...) cprintf(CC_EXTENSION, format, ## args) -uint32_t extension_route_command(uint16_t command_code, - void *buffer, - size_t in_size, - size_t *out_size) +static uint32_t extension_route_command(uint16_t command_code, + void *buffer, + size_t in_size, + size_t *out_size) { struct extension_command *cmd_p; struct extension_command *end_p; @@ -34,3 +34,43 @@ uint32_t extension_route_command(uint16_t command_code, *out_size = 0; return VENDOR_RC_NO_SUCH_COMMAND; } + +uint32_t usb_extension_route_command(uint16_t command_code, + void *buffer, + size_t in_size, + size_t *out_size) +{ + int is_allowed = 0; + + switch (command_code) { +#ifdef CR50_DEV + case VENDOR_CC_IMMEDIATE_RESET: +#endif /* defined(CR50_DEV) */ + case EXTENSION_POST_RESET: /* Always need to be able to reset. */ + is_allowed = 1; + break; + + default: + break; + } + + if (is_allowed) + return extension_route_command(command_code, buffer, in_size, + out_size); + + /* Otherwise, we don't allow this command. */ + CPRINTF("%s: ignoring vendor cmd %d\n", __func__, command_code); + *out_size = 0; + return VENDOR_RC_NO_SUCH_COMMAND; +} + +uint32_t tpm_extension_route_command(uint16_t command_code, + void *buffer, + size_t in_size, + size_t *out_size) +{ + /* + * TODO(aaboagye): Determine what commands (if any) should be filtered. + */ + return extension_route_command(command_code, buffer, in_size, out_size); +} diff --git a/common/tpm_registers.c b/common/tpm_registers.c index a70fd0e028..c2728522ae 100644 --- a/common/tpm_registers.c +++ b/common/tpm_registers.c @@ -626,11 +626,11 @@ static void call_extension_command(struct tpm_cmd_header *tpmh, *total_size -= sizeof(struct tpm_cmd_header); subcommand_code = be16toh(tpmh->subcommand_code); - rc = extension_route_command(subcommand_code, - tpmh + 1, - command_size - - sizeof(struct tpm_cmd_header), - total_size); + rc = tpm_extension_route_command(subcommand_code, + tpmh + 1, + command_size - + sizeof(struct tpm_cmd_header), + total_size); /* Add the header size back. */ *total_size += sizeof(struct tpm_cmd_header); tpmh->size = htobe32(*total_size); |