| /* Copyright (c) 2014-2015 The Linux Foundation. All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions are |
| * met: |
| * * Redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer. |
| * * Redistributions in binary form must reproduce the above |
| * copyright notice, this list of conditions and the following |
| * disclaimer in the documentation and/or other materials provided |
| * with the distribution. |
| * * Neither the name of The Linux Foundation nor the names of its |
| * contributors may be used to endorse or promote products derived |
| * from this software without specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESS OR IMPLIED |
| * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF |
| * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT |
| * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS |
| * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
| * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
| * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR |
| * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, |
| * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE |
| * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN |
| * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| #include <debug.h> |
| #include <reg.h> |
| #include <ufs_hw.h> |
| #include <utp.h> |
| #include <upiu.h> |
| #include <uic.h> |
| #include <ucs.h> |
| #include <dme.h> |
| #include <rpmb.h> |
| #include <rand.h> |
| #include <string.h> |
| #include <stdlib.h> |
| #include <endian.h> |
| #include <target.h> |
| #include "ufs_rpmb.h" |
| |
| void rpmb_run_test() |
| { |
| bool result; |
| uint16_t sector_address = 1, rpmb_num_blocks = 4; |
| void *dev; |
| dev = target_mmc_device(); |
| result = rpmb_test((struct ufs_dev *)dev, sector_address, rpmb_num_blocks); |
| if (!result) |
| dprintf(INFO, "RPMB test failed"); |
| } |
| |
| int verify_rpmb_frame(struct rpmb_frame *request_frame, struct rpmb_frame *result_frame, int type) |
| { |
| int i = 0; |
| if(result_frame->result[0] == MAXED_WR_COUNTER) |
| { |
| dprintf(INFO, "Max write counter value reached\n"); |
| return MAXED_WR_COUNTER; |
| } |
| if(result_frame->result[1] != OPERATION_OK) |
| { |
| dprintf(INFO, "RPMB operation error: 0x%x\n", result_frame->result[1]); |
| return result_frame->result[1]; |
| } |
| for(i = 0; i < 16; i++) |
| { |
| if(request_frame->nonce[i] != result_frame->nonce[i]) |
| return NONCE_MISMATCH; |
| } |
| return OPERATION_OK; |
| } |
| |
| void dump_rpmb_data(struct rpmb_frame *result_frame) |
| { |
| int data_counter; |
| for (data_counter = 0; data_counter < 256; data_counter++) |
| { |
| printf("0x%x ", result_frame->data[data_counter]); |
| if ((data_counter + 1) % 16 == 0) |
| printf("\n"); |
| } |
| printf("\n"); |
| } |
| |
| bool swp_test() |
| { |
| int ret = 0; |
| qsee_stor_secure_wp_info_t cb = {0}; |
| // Write protect 4 partitions in Lun 0. |
| cb.lun_number = 0; |
| cb.num_entries = 4; |
| cb.wp_entries[0].wp_enable = 0x1; |
| cb.wp_entries[0].wp_type_mask = 0x1; |
| cb.wp_entries[0].addr = 8; |
| cb.wp_entries[0].num_blocks = 8192; |
| |
| cb.wp_entries[1].wp_enable = 0x1; |
| cb.wp_entries[1].wp_type_mask = 0x1; |
| cb.wp_entries[1].addr = 74120; |
| cb.wp_entries[1].num_blocks = 32; |
| |
| cb.wp_entries[2].wp_enable = 0x1; |
| cb.wp_entries[2].wp_type_mask = 0x1; |
| cb.wp_entries[2].addr = 6; |
| cb.wp_entries[2].num_blocks = 2; |
| |
| cb.wp_entries[3].wp_enable = 0x1; |
| cb.wp_entries[3].wp_type_mask = 0x1; |
| cb.wp_entries[3].addr = 73736; |
| cb.wp_entries[3].num_blocks = 256; |
| |
| ret = swp_write(cb); |
| if (ret) |
| { |
| dprintf(CRITICAL, "SWP Write Test Failed\n"); |
| return false; |
| } |
| else |
| dprintf(CRITICAL, "SWP Write Test Passed\n"); |
| return true; |
| } |
| |
| bool rpmb_test(struct ufs_dev *dev, uint16_t address, uint16_t rpmb_num_blocks) |
| { |
| struct rpmb_frame data_frame, result_frame[rpmb_num_blocks]; |
| int i = 0, ret; |
| uint32_t response_len = 0; |
| // check if address + sectors requested for read do not exceed total size of rpmb |
| if ((address + rpmb_num_blocks) > dev->rpmb_num_blocks) |
| { |
| dprintf(CRITICAL, "Invalid request to rpmb\n"); |
| return false; |
| } |
| memset(&data_frame, 0, sizeof(data_frame)); |
| memset(&result_frame, 0, sizeof(result_frame)); |
| data_frame.address[0] = (address & 0xff00) >> 8; |
| data_frame.address[1] = address & 0x00ff; |
| data_frame.blockcount[0] = (rpmb_num_blocks & 0xff00) >> 8; |
| data_frame.blockcount[1] = rpmb_num_blocks & 0x00ff; |
| data_frame.requestresponse[1] = AUTH_READ; |
| for(i = 0 ; i < 16; i++) |
| { |
| data_frame.nonce[i] = (rand() % 256) + 1; |
| } |
| #ifdef DEBUG_UFS_RPMB |
| dprintf(INFO, "Dumping RPMB Request frame\n"); |
| dprintf(INFO, "--------------------------\n"); |
| dump((void *) &data_frame, sizeof(struct rpmb_frame)); |
| #endif |
| ret = ucs_do_scsi_rpmb_read(dev, (uint32_t *) &data_frame, rpmb_num_blocks, |
| (uint32_t *) &result_frame, &response_len); |
| if (ret) |
| { |
| dprintf(CRITICAL, "RPMB Read error\n"); |
| return false; |
| } |
| for (i = 0; i < rpmb_num_blocks; i++) |
| { |
| ret = verify_rpmb_frame(&data_frame, &result_frame[i], AUTH_READ); |
| if(ret) |
| { |
| dprintf(CRITICAL, "Error in verifying RPMB frame\n"); |
| dump((void *) &result_frame[i], sizeof(struct rpmb_frame)); |
| } |
| } |
| #ifdef DEBUG_UFS_RPMB |
| dprintf(INFO, "Dumping RPMB Response frames\n"); |
| dprintf(INFO, "----------------------------\n"); |
| dump((void *) &result_frame, sizeof(struct rpmb_frame)*rpmb_num_blocks); |
| #endif |
| dprintf(INFO, "Data dump for RPMB read request\n"); |
| printf("-------------------------------\n"); |
| |
| for (i = 0; i < rpmb_num_blocks; i++) |
| dump_rpmb_data((void *) &result_frame[i]); |
| |
| printf("-------------------------------\n"); |
| |
| return true; |
| } |