Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 440bad0b authored by Wenhao Wang's avatar Wenhao Wang
Browse files

trusty:storageproxyd: Add wakelock to the UFS commands 

We add a wakelock to the sequence of UFS commands so that
the sequence will not be disrrupted when devices get suspended.

Bug: 193456223
Test: Trusty storage tests
Merged-In: Ib90f8b284017cf261d2a2aea940834a42c21de02
Change-Id: Ib90f8b284017cf261d2a2aea940834a42c21de02
parent c66e99bf

trusty/storage/proxy/Android.bp

+4 −1
Original line number Original line Diff line number Diff line
@@ -29,7 +29,10 @@ cc_binary { 
        "proxy.c",

        "proxy.c",

    ],

    ],





    shared_libs: ["liblog"],

    shared_libs: [

        "liblog",

        "libhardware_legacy",

    ],

    header_libs: ["libcutils_headers"],

    header_libs: ["libcutils_headers"],





    static_libs: [

    static_libs: [

trusty/storage/proxy/rpmb.c

+16 −0
Original line number Original line Diff line number Diff line
@@ -29,6 +29,8 @@ 
#include <linux/major.h>

#include <linux/major.h>

#include <linux/mmc/ioctl.h>

#include <linux/mmc/ioctl.h>





#include <hardware_legacy/power.h>



#include "ipc.h"

#include "ipc.h"

#include "log.h"

#include "log.h"

#include "rpmb.h"

#include "rpmb.h"
@@ -100,6 +102,8 @@ static int rpmb_fd = -1; 
static uint8_t read_buf[4096];

static uint8_t read_buf[4096];

static enum dev_type dev_type = UNKNOWN_RPMB;

static enum dev_type dev_type = UNKNOWN_RPMB;





static const char* UFS_WAKE_LOCK_NAME = "ufs_seq_wakelock";



#ifdef RPMB_DEBUG

#ifdef RPMB_DEBUG





static void print_buf(const char* prefix, const uint8_t* buf, size_t size) {

static void print_buf(const char* prefix, const uint8_t* buf, size_t size) {
@@ -194,6 +198,7 @@ static int send_mmc_rpmb_req(int mmc_fd, const struct storage_rpmb_send_req* req 




static int send_ufs_rpmb_req(int sg_fd, const struct storage_rpmb_send_req* req) {

static int send_ufs_rpmb_req(int sg_fd, const struct storage_rpmb_send_req* req) {

    int rc;

    int rc;

    int wl_rc;

    const uint8_t* write_buf = req->payload;

    const uint8_t* write_buf = req->payload;

    /*

    /*

     * Meaning of member values are stated on the definition of struct sec_proto_cdb.

     * Meaning of member values are stated on the definition of struct sec_proto_cdb.
@@ -202,6 +207,12 @@ static int send_ufs_rpmb_req(int sg_fd, const struct storage_rpmb_send_req* req) 
    struct sec_proto_cdb out_cdb = {0xB5, 0xEC, 0x00, 0x01, 0x00, 0x00, 0, 0x00, 0x00};

    struct sec_proto_cdb out_cdb = {0xB5, 0xEC, 0x00, 0x01, 0x00, 0x00, 0, 0x00, 0x00};

    unsigned char sense_buffer[32];

    unsigned char sense_buffer[32];





    wl_rc = acquire_wake_lock(PARTIAL_WAKE_LOCK, UFS_WAKE_LOCK_NAME);

    if (wl_rc < 0) {

        ALOGE("%s: failed to acquire wakelock: %d, %s\n", __func__, wl_rc, strerror(errno));

        return wl_rc;

    }



    if (req->reliable_write_size) {

    if (req->reliable_write_size) {

        /* Prepare SECURITY PROTOCOL OUT command. */

        /* Prepare SECURITY PROTOCOL OUT command. */

        out_cdb.length = __builtin_bswap32(req->reliable_write_size);

        out_cdb.length = __builtin_bswap32(req->reliable_write_size);
@@ -244,6 +255,11 @@ static int send_ufs_rpmb_req(int sg_fd, const struct storage_rpmb_send_req* req) 
    }

    }





err_op:

err_op:

    wl_rc = release_wake_lock(UFS_WAKE_LOCK_NAME);

    if (wl_rc < 0) {

        ALOGE("%s: failed to release wakelock: %d, %s\n", __func__, wl_rc, strerror(errno));

    }



    return rc;

    return rc;

}

}