[thirdparty/u-boot.git] / include / avb_verify.h


/*
 * (C) Copyright 2018, Linaro Limited
 *
 * SPDX-License-Identifier:	GPL-2.0+
 */

#ifndef	_AVB_VERIFY_H
#define _AVB_VERIFY_H

#include <../lib/libavb/libavb.h>
#include <mapmem.h>
#include <mmc.h>

#define AVB_MAX_ARGS			1024
#define VERITY_TABLE_OPT_RESTART	"restart_on_corruption"
#define VERITY_TABLE_OPT_LOGGING	"ignore_corruption"
#define ALLOWED_BUF_ALIGN		8

enum avb_boot_state {
	AVB_GREEN,
	AVB_YELLOW,
	AVB_ORANGE,
	AVB_RED,
};

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 {
	int dev_num;
	struct mmc *mmc;
	struct blk_desc *mmc_blk;
	disk_partition_t info;
};

enum mmc_io_type {
	IO_READ,
	IO_WRITE
};

AvbOps *avb_ops_alloc(int boot_device);
void avb_ops_free(AvbOps *ops);

char *avb_set_state(AvbOps *ops, enum avb_boot_state boot_state);
char *avb_set_enforce_verity(const char *cmdline);
char *avb_set_ignore_corruption(const char *cmdline);

char *append_cmd_line(char *cmdline_orig, char *cmdline_new);

/**
 * ============================================================================
 * I/O helper inline functions
 * ============================================================================
 */
static inline uint64_t calc_offset(struct mmc_part *part, int64_t offset)
{
	u64 part_size = part->info.size * part->info.blksz;

	if (offset < 0)
		return part_size + offset;

	return offset;
}

static inline size_t get_sector_buf_size(void)
{
	return (size_t)CONFIG_FASTBOOT_BUF_SIZE;
}

static inline void *get_sector_buf(void)
{
	return map_sysmem(CONFIG_FASTBOOT_BUF_ADDR, CONFIG_FASTBOOT_BUF_SIZE);
}

static inline bool is_buf_unaligned(void *buffer)
{
	return (bool)((uintptr_t)buffer % ALLOWED_BUF_ALIGN);
}

static inline int get_boot_device(AvbOps *ops)
{
	struct AvbOpsData *data;

	if (ops) {
		data = ops->user_data;
		if (data)
			return data->mmc_dev;
	}

	return -1;
}

#endif /* _AVB_VERIFY_H */
Commit	Line	Data
3af30e44 IO	1
	2	/*
	3	* (C) Copyright 2018, Linaro Limited
	4	*
	5	* SPDX-License-Identifier: GPL-2.0+
	6	*/
	7
	8	#ifndef _AVB_VERIFY_H
	9	#define _AVB_VERIFY_H
	10
	11	#include <../lib/libavb/libavb.h>
bbddbef9	12	#include <mapmem.h>
3af30e44 IO	13	#include <mmc.h>
3af30e44 IO	14
5d4fd877 IO	15	#define AVB_MAX_ARGS 1024
	16	#define VERITY_TABLE_OPT_RESTART "restart_on_corruption"
	17	#define VERITY_TABLE_OPT_LOGGING "ignore_corruption"
	18	#define ALLOWED_BUF_ALIGN 8
	19
	20	enum avb_boot_state {
	21	AVB_GREEN,
	22	AVB_YELLOW,
	23	AVB_ORANGE,
	24	AVB_RED,
	25	};
3af30e44 IO	26
	27	struct AvbOpsData {
	28	struct AvbOps ops;
	29	int mmc_dev;
5d4fd877	30	enum avb_boot_state boot_state;
6663e074 JW	31	#ifdef CONFIG_OPTEE_TA_AVB
	32	struct udevice *tee;
	33	u32 session;
	34	#endif
3af30e44 IO	35	};
	36
	37	struct mmc_part {
	38	int dev_num;
	39	struct mmc *mmc;
	40	struct blk_desc *mmc_blk;
	41	disk_partition_t info;
	42	};
	43
	44	enum mmc_io_type {
	45	IO_READ,
	46	IO_WRITE
	47	};
	48
	49	AvbOps *avb_ops_alloc(int boot_device);
	50	void avb_ops_free(AvbOps *ops);
	51
5d4fd877 IO	52	char avb_set_state(AvbOps ops, enum avb_boot_state boot_state);
	53	char avb_set_enforce_verity(const char cmdline);
	54	char avb_set_ignore_corruption(const char cmdline);
	55
	56	char append_cmd_line(char cmdline_orig, char *cmdline_new);
	57
3af30e44 IO	58	/**
	59	* ============================================================================
	60	* I/O helper inline functions
	61	* ============================================================================
	62	*/
	63	static inline uint64_t calc_offset(struct mmc_part *part, int64_t offset)
	64	{
	65	u64 part_size = part->info.size * part->info.blksz;
	66
	67	if (offset < 0)
	68	return part_size + offset;
	69
	70	return offset;
	71	}
	72
	73	static inline size_t get_sector_buf_size(void)
	74	{
	75	return (size_t)CONFIG_FASTBOOT_BUF_SIZE;
	76	}
	77
	78	static inline void *get_sector_buf(void)
	79	{
bbddbef9	80	return map_sysmem(CONFIG_FASTBOOT_BUF_ADDR, CONFIG_FASTBOOT_BUF_SIZE);
3af30e44 IO	81	}
	82
	83	static inline bool is_buf_unaligned(void *buffer)
	84	{
	85	return (bool)((uintptr_t)buffer % ALLOWED_BUF_ALIGN);
	86	}
	87
	88	static inline int get_boot_device(AvbOps *ops)
	89	{
	90	struct AvbOpsData *data;
	91
	92	if (ops) {
	93	data = ops->user_data;
	94	if (data)
	95	return data->mmc_dev;
	96	}
	97
	98	return -1;
	99	}
	100
	101	#endif /* _AVB_VERIFY_H */