1397 lines · 41.5 KB · C
RawDownload
// SPDX-License-Identifier: GPL-2.0/* Copyright (C) 2012-2019 ARM Limited (or its affiliates). */ #include <crypto/internal/aead.h>#include <crypto/authenc.h>#include <crypto/scatterwalk.h>#include <linux/dmapool.h>#include <linux/dma-mapping.h> #include "cc_buffer_mgr.h"#include "cc_lli_defs.h"#include "cc_cipher.h"#include "cc_hash.h"#include "cc_aead.h" union buffer_array_entry {	struct scatterlist *sgl;	dma_addr_t buffer_dma;}; struct buffer_array {	unsigned int num_of_buffers;	union buffer_array_entry entry[MAX_NUM_OF_BUFFERS_IN_MLLI];	unsigned int offset[MAX_NUM_OF_BUFFERS_IN_MLLI];	int nents[MAX_NUM_OF_BUFFERS_IN_MLLI];	int total_data_len[MAX_NUM_OF_BUFFERS_IN_MLLI];	bool is_last[MAX_NUM_OF_BUFFERS_IN_MLLI];	u32 *mlli_nents[MAX_NUM_OF_BUFFERS_IN_MLLI];}; static inline char *cc_dma_buf_type(enum cc_req_dma_buf_type type){	switch (type) {	case CC_DMA_BUF_NULL:		return "BUF_NULL";	case CC_DMA_BUF_DLLI:		return "BUF_DLLI";	case CC_DMA_BUF_MLLI:		return "BUF_MLLI";	default:		return "BUF_INVALID";	}} /** * cc_copy_mac() - Copy MAC to temporary location * * @dev: device object * @req: aead request object * @dir: [IN] copy from/to sgl */static void cc_copy_mac(struct device *dev, struct aead_request *req,			enum cc_sg_cpy_direct dir){	struct aead_req_ctx *areq_ctx = aead_request_ctx_dma(req);	u32 skip = req->assoclen + req->cryptlen; 	cc_copy_sg_portion(dev, areq_ctx->backup_mac, req->src,			   (skip - areq_ctx->req_authsize), skip, dir);} /** * cc_get_sgl_nents() - Get scatterlist number of entries. * * @dev: Device object * @sg_list: SG list * @nbytes: [IN] Total SGL data bytes. * @lbytes: [OUT] Returns the amount of bytes at the last entry * * Return: * Number of entries in the scatterlist */static unsigned int cc_get_sgl_nents(struct device *dev,				     struct scatterlist *sg_list,				     unsigned int nbytes, u32 *lbytes){	unsigned int nents = 0; 	*lbytes = 0; 	while (nbytes && sg_list) {		nents++;		/* get the number of bytes in the last entry */		*lbytes = nbytes;		nbytes -= (sg_list->length > nbytes) ?				nbytes : sg_list->length;		sg_list = sg_next(sg_list);	} 	dev_dbg(dev, "nents %d last bytes %d\n", nents, *lbytes);	return nents;} /** * cc_copy_sg_portion() - Copy scatter list data, * from to_skip to end, to dest and vice versa * * @dev: Device object * @dest: Buffer to copy to/from * @sg: SG list * @to_skip: Number of bytes to skip before copying * @end: Offset of last byte to copy * @direct: Transfer direction (true == from SG list to buffer, false == from *          buffer to SG list) */void cc_copy_sg_portion(struct device *dev, u8 *dest, struct scatterlist *sg,			u32 to_skip, u32 end, enum cc_sg_cpy_direct direct){	u32 nents; 	nents = sg_nents_for_len(sg, end);	sg_copy_buffer(sg, nents, dest, (end - to_skip + 1), to_skip,		       (direct == CC_SG_TO_BUF));} static int cc_render_buff_to_mlli(struct device *dev, dma_addr_t buff_dma,				  u32 buff_size, u32 *curr_nents,				  u32 **mlli_entry_pp){	u32 *mlli_entry_p = *mlli_entry_pp;	u32 new_nents; 	/* Verify there is no memory overflow*/	new_nents = (*curr_nents + buff_size / CC_MAX_MLLI_ENTRY_SIZE + 1);	if (new_nents > MAX_NUM_OF_TOTAL_MLLI_ENTRIES) {		dev_err(dev, "Too many mlli entries. current %d max %d\n",			new_nents, MAX_NUM_OF_TOTAL_MLLI_ENTRIES);		return -ENOMEM;	} 	/*handle buffer longer than 64 kbytes */	while (buff_size > CC_MAX_MLLI_ENTRY_SIZE) {		cc_lli_set_addr(mlli_entry_p, buff_dma);		cc_lli_set_size(mlli_entry_p, CC_MAX_MLLI_ENTRY_SIZE);		dev_dbg(dev, "entry[%d]: single_buff=0x%08X size=%08X\n",			*curr_nents, mlli_entry_p[LLI_WORD0_OFFSET],			mlli_entry_p[LLI_WORD1_OFFSET]);		buff_dma += CC_MAX_MLLI_ENTRY_SIZE;		buff_size -= CC_MAX_MLLI_ENTRY_SIZE;		mlli_entry_p = mlli_entry_p + 2;		(*curr_nents)++;	}	/*Last entry */	cc_lli_set_addr(mlli_entry_p, buff_dma);	cc_lli_set_size(mlli_entry_p, buff_size);	dev_dbg(dev, "entry[%d]: single_buff=0x%08X size=%08X\n",		*curr_nents, mlli_entry_p[LLI_WORD0_OFFSET],		mlli_entry_p[LLI_WORD1_OFFSET]);	mlli_entry_p = mlli_entry_p + 2;	*mlli_entry_pp = mlli_entry_p;	(*curr_nents)++;	return 0;} static int cc_render_sg_to_mlli(struct device *dev, struct scatterlist *sgl,				u32 sgl_data_len, u32 sgl_offset,				u32 *curr_nents, u32 **mlli_entry_pp){	struct scatterlist *curr_sgl = sgl;	u32 *mlli_entry_p = *mlli_entry_pp;	s32 rc = 0; 	for ( ; (curr_sgl && sgl_data_len);	      curr_sgl = sg_next(curr_sgl)) {		u32 entry_data_len =			(sgl_data_len > sg_dma_len(curr_sgl) - sgl_offset) ?				sg_dma_len(curr_sgl) - sgl_offset :				sgl_data_len;		sgl_data_len -= entry_data_len;		rc = cc_render_buff_to_mlli(dev, sg_dma_address(curr_sgl) +					    sgl_offset, entry_data_len,					    curr_nents, &mlli_entry_p);		if (rc)			return rc; 		sgl_offset = 0;	}	*mlli_entry_pp = mlli_entry_p;	return 0;} static int cc_generate_mlli(struct device *dev, struct buffer_array *sg_data,			    struct mlli_params *mlli_params, gfp_t flags){	u32 *mlli_p;	u32 total_nents = 0, prev_total_nents = 0;	int rc = 0, i; 	dev_dbg(dev, "NUM of SG's = %d\n", sg_data->num_of_buffers); 	/* Allocate memory from the pointed pool */	mlli_params->mlli_virt_addr =		dma_pool_alloc(mlli_params->curr_pool, flags,			       &mlli_params->mlli_dma_addr);	if (!mlli_params->mlli_virt_addr) {		dev_err(dev, "dma_pool_alloc() failed\n");		rc = -ENOMEM;		goto build_mlli_exit;	}	/* Point to start of MLLI */	mlli_p = mlli_params->mlli_virt_addr;	/* go over all SG's and link it to one MLLI table */	for (i = 0; i < sg_data->num_of_buffers; i++) {		union buffer_array_entry *entry = &sg_data->entry[i];		u32 tot_len = sg_data->total_data_len[i];		u32 offset = sg_data->offset[i]; 		rc = cc_render_sg_to_mlli(dev, entry->sgl, tot_len, offset,					  &total_nents, &mlli_p);		if (rc)			return rc; 		/* set last bit in the current table */		if (sg_data->mlli_nents[i]) {			/*Calculate the current MLLI table length for the			 *length field in the descriptor			 */			*sg_data->mlli_nents[i] +=				(total_nents - prev_total_nents);			prev_total_nents = total_nents;		}	} 	/* Set MLLI size for the bypass operation */	mlli_params->mlli_len = (total_nents * LLI_ENTRY_BYTE_SIZE); 	dev_dbg(dev, "MLLI params: virt_addr=%p dma_addr=%pad mlli_len=0x%X\n",		mlli_params->mlli_virt_addr, &mlli_params->mlli_dma_addr,		mlli_params->mlli_len); build_mlli_exit:	return rc;} static void cc_add_sg_entry(struct device *dev, struct buffer_array *sgl_data,			    unsigned int nents, struct scatterlist *sgl,			    unsigned int data_len, unsigned int data_offset,			    bool is_last_table, u32 *mlli_nents){	unsigned int index = sgl_data->num_of_buffers; 	dev_dbg(dev, "index=%u nents=%u sgl=%p data_len=0x%08X is_last=%d\n",		index, nents, sgl, data_len, is_last_table);	sgl_data->nents[index] = nents;	sgl_data->entry[index].sgl = sgl;	sgl_data->offset[index] = data_offset;	sgl_data->total_data_len[index] = data_len;	sgl_data->is_last[index] = is_last_table;	sgl_data->mlli_nents[index] = mlli_nents;	if (sgl_data->mlli_nents[index])		*sgl_data->mlli_nents[index] = 0;	sgl_data->num_of_buffers++;} static int cc_map_sg(struct device *dev, struct scatterlist *sg,		     unsigned int nbytes, int direction, u32 *nents,		     u32 max_sg_nents, u32 *lbytes, u32 *mapped_nents){	int ret = 0; 	if (!nbytes) {		*mapped_nents = 0;		*lbytes = 0;		*nents = 0;		return 0;	} 	*nents = cc_get_sgl_nents(dev, sg, nbytes, lbytes);	if (*nents > max_sg_nents) {		*nents = 0;		dev_err(dev, "Too many fragments. current %d max %d\n",			*nents, max_sg_nents);		return -ENOMEM;	} 	ret = dma_map_sg(dev, sg, *nents, direction);	if (!ret) {		*nents = 0;		dev_err(dev, "dma_map_sg() sg buffer failed %d\n", ret);		return -ENOMEM;	} 	*mapped_nents = ret; 	return 0;} static intcc_set_aead_conf_buf(struct device *dev, struct aead_req_ctx *areq_ctx,		     u8 *config_data, struct buffer_array *sg_data,		     unsigned int assoclen){	dev_dbg(dev, " handle additional data config set to DLLI\n");	/* create sg for the current buffer */	sg_init_one(&areq_ctx->ccm_adata_sg, config_data,		    AES_BLOCK_SIZE + areq_ctx->ccm_hdr_size);	if (dma_map_sg(dev, &areq_ctx->ccm_adata_sg, 1, DMA_TO_DEVICE) != 1) {		dev_err(dev, "dma_map_sg() config buffer failed\n");		return -ENOMEM;	}	dev_dbg(dev, "Mapped curr_buff: dma_address=%pad page=%p addr=%p offset=%u length=%u\n",		&sg_dma_address(&areq_ctx->ccm_adata_sg),		sg_page(&areq_ctx->ccm_adata_sg),		sg_virt(&areq_ctx->ccm_adata_sg),		areq_ctx->ccm_adata_sg.offset, areq_ctx->ccm_adata_sg.length);	/* prepare for case of MLLI */	if (assoclen > 0) {		cc_add_sg_entry(dev, sg_data, 1, &areq_ctx->ccm_adata_sg,				(AES_BLOCK_SIZE + areq_ctx->ccm_hdr_size),				0, false, NULL);	}	return 0;} static int cc_set_hash_buf(struct device *dev, struct ahash_req_ctx *areq_ctx,			   u8 *curr_buff, u32 curr_buff_cnt,			   struct buffer_array *sg_data){	dev_dbg(dev, " handle curr buff %x set to   DLLI\n", curr_buff_cnt);	/* create sg for the current buffer */	sg_init_one(areq_ctx->buff_sg, curr_buff, curr_buff_cnt);	if (dma_map_sg(dev, areq_ctx->buff_sg, 1, DMA_TO_DEVICE) != 1) {		dev_err(dev, "dma_map_sg() src buffer failed\n");		return -ENOMEM;	}	dev_dbg(dev, "Mapped curr_buff: dma_address=%pad page=%p addr=%p offset=%u length=%u\n",		&sg_dma_address(areq_ctx->buff_sg), sg_page(areq_ctx->buff_sg),		sg_virt(areq_ctx->buff_sg), areq_ctx->buff_sg->offset,		areq_ctx->buff_sg->length);	areq_ctx->data_dma_buf_type = CC_DMA_BUF_DLLI;	areq_ctx->curr_sg = areq_ctx->buff_sg;	areq_ctx->in_nents = 0;	/* prepare for case of MLLI */	cc_add_sg_entry(dev, sg_data, 1, areq_ctx->buff_sg, curr_buff_cnt, 0,			false, NULL);	return 0;} void cc_unmap_cipher_request(struct device *dev, void *ctx,				unsigned int ivsize, struct scatterlist *src,				struct scatterlist *dst){	struct cipher_req_ctx *req_ctx = (struct cipher_req_ctx *)ctx; 	if (req_ctx->gen_ctx.iv_dma_addr) {		dev_dbg(dev, "Unmapped iv: iv_dma_addr=%pad iv_size=%u\n",			&req_ctx->gen_ctx.iv_dma_addr, ivsize);		dma_unmap_single(dev, req_ctx->gen_ctx.iv_dma_addr,				 ivsize, DMA_BIDIRECTIONAL);	}	/* Release pool */	if (req_ctx->dma_buf_type == CC_DMA_BUF_MLLI &&	    req_ctx->mlli_params.mlli_virt_addr) {		dma_pool_free(req_ctx->mlli_params.curr_pool,			      req_ctx->mlli_params.mlli_virt_addr,			      req_ctx->mlli_params.mlli_dma_addr);	} 	if (src != dst) {		dma_unmap_sg(dev, src, req_ctx->in_nents, DMA_TO_DEVICE);		dma_unmap_sg(dev, dst, req_ctx->out_nents, DMA_FROM_DEVICE);		dev_dbg(dev, "Unmapped req->dst=%p\n", sg_virt(dst));		dev_dbg(dev, "Unmapped req->src=%p\n", sg_virt(src));	} else {		dma_unmap_sg(dev, src, req_ctx->in_nents, DMA_BIDIRECTIONAL);		dev_dbg(dev, "Unmapped req->src=%p\n", sg_virt(src));	}} int cc_map_cipher_request(struct cc_drvdata *drvdata, void *ctx,			  unsigned int ivsize, unsigned int nbytes,			  void *info, struct scatterlist *src,			  struct scatterlist *dst, gfp_t flags){	struct cipher_req_ctx *req_ctx = (struct cipher_req_ctx *)ctx;	struct mlli_params *mlli_params = &req_ctx->mlli_params;	struct device *dev = drvdata_to_dev(drvdata);	struct buffer_array sg_data;	u32 dummy = 0;	int rc = 0;	u32 mapped_nents = 0;	int src_direction = (src != dst ? DMA_TO_DEVICE : DMA_BIDIRECTIONAL); 	req_ctx->dma_buf_type = CC_DMA_BUF_DLLI;	mlli_params->curr_pool = NULL;	sg_data.num_of_buffers = 0; 	/* Map IV buffer */	if (ivsize) {		dump_byte_array("iv", info, ivsize);		req_ctx->gen_ctx.iv_dma_addr =			dma_map_single(dev, info, ivsize, DMA_BIDIRECTIONAL);		if (dma_mapping_error(dev, req_ctx->gen_ctx.iv_dma_addr)) {			dev_err(dev, "Mapping iv %u B at va=%p for DMA failed\n",				ivsize, info);			return -ENOMEM;		}		dev_dbg(dev, "Mapped iv %u B at va=%p to dma=%pad\n",			ivsize, info, &req_ctx->gen_ctx.iv_dma_addr);	} else {		req_ctx->gen_ctx.iv_dma_addr = 0;	} 	/* Map the src SGL */	rc = cc_map_sg(dev, src, nbytes, src_direction, &req_ctx->in_nents,		       LLI_MAX_NUM_OF_DATA_ENTRIES, &dummy, &mapped_nents);	if (rc)		goto cipher_exit;	if (mapped_nents > 1)		req_ctx->dma_buf_type = CC_DMA_BUF_MLLI; 	if (src == dst) {		/* Handle inplace operation */		if (req_ctx->dma_buf_type == CC_DMA_BUF_MLLI) {			req_ctx->out_nents = 0;			cc_add_sg_entry(dev, &sg_data, req_ctx->in_nents, src,					nbytes, 0, true,					&req_ctx->in_mlli_nents);		}	} else {		/* Map the dst sg */		rc = cc_map_sg(dev, dst, nbytes, DMA_FROM_DEVICE,			       &req_ctx->out_nents, LLI_MAX_NUM_OF_DATA_ENTRIES,			       &dummy, &mapped_nents);		if (rc)			goto cipher_exit;		if (mapped_nents > 1)			req_ctx->dma_buf_type = CC_DMA_BUF_MLLI; 		if (req_ctx->dma_buf_type == CC_DMA_BUF_MLLI) {			cc_add_sg_entry(dev, &sg_data, req_ctx->in_nents, src,					nbytes, 0, true,					&req_ctx->in_mlli_nents);			cc_add_sg_entry(dev, &sg_data, req_ctx->out_nents, dst,					nbytes, 0, true,					&req_ctx->out_mlli_nents);		}	} 	if (req_ctx->dma_buf_type == CC_DMA_BUF_MLLI) {		mlli_params->curr_pool = drvdata->mlli_buffs_pool;		rc = cc_generate_mlli(dev, &sg_data, mlli_params, flags);		if (rc)			goto cipher_exit;	} 	dev_dbg(dev, "areq_ctx->dma_buf_type = %s\n",		cc_dma_buf_type(req_ctx->dma_buf_type)); 	return 0; cipher_exit:	cc_unmap_cipher_request(dev, req_ctx, ivsize, src, dst);	return rc;} void cc_unmap_aead_request(struct device *dev, struct aead_request *req){	struct aead_req_ctx *areq_ctx = aead_request_ctx_dma(req);	unsigned int hw_iv_size = areq_ctx->hw_iv_size;	struct cc_drvdata *drvdata = dev_get_drvdata(dev);	int src_direction = (req->src != req->dst ? DMA_TO_DEVICE : DMA_BIDIRECTIONAL); 	if (areq_ctx->mac_buf_dma_addr) {		dma_unmap_single(dev, areq_ctx->mac_buf_dma_addr,				 MAX_MAC_SIZE, DMA_BIDIRECTIONAL);	} 	if (areq_ctx->cipher_mode == DRV_CIPHER_GCTR) {		if (areq_ctx->hkey_dma_addr) {			dma_unmap_single(dev, areq_ctx->hkey_dma_addr,					 AES_BLOCK_SIZE, DMA_BIDIRECTIONAL);		} 		if (areq_ctx->gcm_block_len_dma_addr) {			dma_unmap_single(dev, areq_ctx->gcm_block_len_dma_addr,					 AES_BLOCK_SIZE, DMA_TO_DEVICE);		} 		if (areq_ctx->gcm_iv_inc1_dma_addr) {			dma_unmap_single(dev, areq_ctx->gcm_iv_inc1_dma_addr,					 AES_BLOCK_SIZE, DMA_TO_DEVICE);		} 		if (areq_ctx->gcm_iv_inc2_dma_addr) {			dma_unmap_single(dev, areq_ctx->gcm_iv_inc2_dma_addr,					 AES_BLOCK_SIZE, DMA_TO_DEVICE);		}	} 	if (areq_ctx->ccm_hdr_size != ccm_header_size_null) {		if (areq_ctx->ccm_iv0_dma_addr) {			dma_unmap_single(dev, areq_ctx->ccm_iv0_dma_addr,					 AES_BLOCK_SIZE, DMA_TO_DEVICE);		} 		dma_unmap_sg(dev, &areq_ctx->ccm_adata_sg, 1, DMA_TO_DEVICE);	}	if (areq_ctx->gen_ctx.iv_dma_addr) {		dma_unmap_single(dev, areq_ctx->gen_ctx.iv_dma_addr,

Lines 1–500 of 1397.