174fe6c29SRuslan Bukin /*
274fe6c29SRuslan Bukin  * Copyright (c) 2016-2018, Intel Corporation
374fe6c29SRuslan Bukin  *
474fe6c29SRuslan Bukin  * Redistribution and use in source and binary forms, with or without
574fe6c29SRuslan Bukin  * modification, are permitted provided that the following conditions are met:
674fe6c29SRuslan Bukin  *
774fe6c29SRuslan Bukin  *  * Redistributions of source code must retain the above copyright notice,
874fe6c29SRuslan Bukin  *    this list of conditions and the following disclaimer.
974fe6c29SRuslan Bukin  *  * Redistributions in binary form must reproduce the above copyright notice,
1074fe6c29SRuslan Bukin  *    this list of conditions and the following disclaimer in the documentation
1174fe6c29SRuslan Bukin  *    and/or other materials provided with the distribution.
1274fe6c29SRuslan Bukin  *  * Neither the name of Intel Corporation nor the names of its contributors
1374fe6c29SRuslan Bukin  *    may be used to endorse or promote products derived from this software
1474fe6c29SRuslan Bukin  *    without specific prior written permission.
1574fe6c29SRuslan Bukin  *
1674fe6c29SRuslan Bukin  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
1774fe6c29SRuslan Bukin  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
1874fe6c29SRuslan Bukin  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
1974fe6c29SRuslan Bukin  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
2074fe6c29SRuslan Bukin  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
2174fe6c29SRuslan Bukin  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
2274fe6c29SRuslan Bukin  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
2374fe6c29SRuslan Bukin  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
2474fe6c29SRuslan Bukin  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
2574fe6c29SRuslan Bukin  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
2674fe6c29SRuslan Bukin  * POSSIBILITY OF SUCH DAMAGE.
2774fe6c29SRuslan Bukin  */
2874fe6c29SRuslan Bukin 
2974fe6c29SRuslan Bukin #include "pt_block_cache.h"
3074fe6c29SRuslan Bukin 
3174fe6c29SRuslan Bukin #include <stdlib.h>
3274fe6c29SRuslan Bukin #include <string.h>
3374fe6c29SRuslan Bukin 
3474fe6c29SRuslan Bukin 
3574fe6c29SRuslan Bukin struct pt_block_cache *pt_bcache_alloc(uint64_t nentries)
3674fe6c29SRuslan Bukin {
3774fe6c29SRuslan Bukin 	struct pt_block_cache *bcache;
3874fe6c29SRuslan Bukin 	uint64_t size;
3974fe6c29SRuslan Bukin 
4074fe6c29SRuslan Bukin 	if (!nentries || (UINT32_MAX < nentries))
4174fe6c29SRuslan Bukin 		return NULL;
4274fe6c29SRuslan Bukin 
4374fe6c29SRuslan Bukin 	size = sizeof(*bcache) + (nentries * sizeof(struct pt_bcache_entry));
4474fe6c29SRuslan Bukin 	if (SIZE_MAX < size)
4574fe6c29SRuslan Bukin 		return NULL;
4674fe6c29SRuslan Bukin 
4774fe6c29SRuslan Bukin 	bcache = malloc((size_t) size);
4874fe6c29SRuslan Bukin 	if (!bcache)
4974fe6c29SRuslan Bukin 		return NULL;
5074fe6c29SRuslan Bukin 
5174fe6c29SRuslan Bukin 	memset(bcache, 0, (size_t) size);
5274fe6c29SRuslan Bukin 	bcache->nentries = (uint32_t) nentries;
5374fe6c29SRuslan Bukin 
5474fe6c29SRuslan Bukin 	return bcache;
5574fe6c29SRuslan Bukin }
5674fe6c29SRuslan Bukin 
5774fe6c29SRuslan Bukin void pt_bcache_free(struct pt_block_cache *bcache)
5874fe6c29SRuslan Bukin {
5974fe6c29SRuslan Bukin 	free(bcache);
6074fe6c29SRuslan Bukin }
6174fe6c29SRuslan Bukin 
6274fe6c29SRuslan Bukin int pt_bcache_add(struct pt_block_cache *bcache, uint64_t index,
6374fe6c29SRuslan Bukin 		  struct pt_bcache_entry bce)
6474fe6c29SRuslan Bukin {
6574fe6c29SRuslan Bukin 	if (!bcache)
6674fe6c29SRuslan Bukin 		return -pte_internal;
6774fe6c29SRuslan Bukin 
6874fe6c29SRuslan Bukin 	if (bcache->nentries <= index)
6974fe6c29SRuslan Bukin 		return -pte_internal;
7074fe6c29SRuslan Bukin 
7174fe6c29SRuslan Bukin 	/* We rely on guaranteed atomic operations as specified in section 8.1.1
7274fe6c29SRuslan Bukin 	 * in Volume 3A of the Intel(R) Software Developer's Manual at
7374fe6c29SRuslan Bukin 	 * http://www.intel.com/sdm.
7474fe6c29SRuslan Bukin 	 */
7574fe6c29SRuslan Bukin 	bcache->entry[(uint32_t) index] = bce;
7674fe6c29SRuslan Bukin 
7774fe6c29SRuslan Bukin 	return 0;
7874fe6c29SRuslan Bukin }
7974fe6c29SRuslan Bukin 
8074fe6c29SRuslan Bukin int pt_bcache_lookup(struct pt_bcache_entry *bce,
8174fe6c29SRuslan Bukin 		     const struct pt_block_cache *bcache, uint64_t index)
8274fe6c29SRuslan Bukin {
8374fe6c29SRuslan Bukin 	if (!bce || !bcache)
8474fe6c29SRuslan Bukin 		return -pte_internal;
8574fe6c29SRuslan Bukin 
8674fe6c29SRuslan Bukin 	if (bcache->nentries <= index)
8774fe6c29SRuslan Bukin 		return -pte_internal;
8874fe6c29SRuslan Bukin 
8974fe6c29SRuslan Bukin 	/* We rely on guaranteed atomic operations as specified in section 8.1.1
9074fe6c29SRuslan Bukin 	 * in Volume 3A of the Intel(R) Software Developer's Manual at
9174fe6c29SRuslan Bukin 	 * http://www.intel.com/sdm.
9274fe6c29SRuslan Bukin 	 */
9374fe6c29SRuslan Bukin 	*bce = bcache->entry[(uint32_t) index];
9474fe6c29SRuslan Bukin 
9574fe6c29SRuslan Bukin 	return 0;
9674fe6c29SRuslan Bukin }
97