Merge tag 'u-boot-rockchip-20190729' of https://gitlab.denx.de/u-boot/custodians...

[platform/kernel/u-boot.git] / include / avb_verify.h

diff --git a/include/avb_verify.h b/include/avb_verify.h
index eaa60f5..18278f8 100644 (file)
--- a/include/avb_verify.h
+++ b/include/avb_verify.h
@@ -9,6 +9,7 @@
 #define _AVB_VERIFY_H
 
 #include <../lib/libavb/libavb.h>
+#include <mapmem.h>
 #include <mmc.h>
 
 #define AVB_MAX_ARGS                   1024
@@ -27,6 +28,10 @@ struct AvbOpsData {
        struct AvbOps ops;
        int mmc_dev;
        enum avb_boot_state boot_state;
+#ifdef CONFIG_OPTEE_TA_AVB
+       struct udevice *tee;
+       u32 session;
+#endif
 };
 
 struct mmc_part {
@@ -72,7 +77,7 @@ static inline size_t get_sector_buf_size(void)
 
 static inline void *get_sector_buf(void)
 {
-       return (void *)CONFIG_FASTBOOT_BUF_ADDR;
+       return map_sysmem(CONFIG_FASTBOOT_BUF_ADDR, CONFIG_FASTBOOT_BUF_SIZE);
 }
 
 static inline bool is_buf_unaligned(void *buffer)