1786acdabcc15f023330d7c628aca9679e757a238Randall Spangler/* Copyright (c) 2014 The Chromium OS Authors. All rights reserved. 2786acdabcc15f023330d7c628aca9679e757a238Randall Spangler * Use of this source code is governed by a BSD-style license that can be 3786acdabcc15f023330d7c628aca9679e757a238Randall Spangler * found in the LICENSE file. 4786acdabcc15f023330d7c628aca9679e757a238Randall Spangler * 5786acdabcc15f023330d7c628aca9679e757a238Randall Spangler * Common functions between firmware and kernel verified boot. 6786acdabcc15f023330d7c628aca9679e757a238Randall Spangler * (Firmware portion) 7786acdabcc15f023330d7c628aca9679e757a238Randall Spangler */ 8786acdabcc15f023330d7c628aca9679e757a238Randall Spangler 9786acdabcc15f023330d7c628aca9679e757a238Randall Spangler#include "2sysincludes.h" 10786acdabcc15f023330d7c628aca9679e757a238Randall Spangler#include "2common.h" 117141d73c149ee203a192f0cb45ca9bfa10c0ff8cRandall Spangler#include "2rsa.h" 127141d73c149ee203a192f0cb45ca9bfa10c0ff8cRandall Spangler#include "2sha.h" 13786acdabcc15f023330d7c628aca9679e757a238Randall Spangler 146d03b527fe0107c3bb71d0bbaf6236fe980b0704Randall Spanglerint vb2_safe_memcmp(const void *s1, const void *s2, size_t size) 156d03b527fe0107c3bb71d0bbaf6236fe980b0704Randall Spangler{ 166d03b527fe0107c3bb71d0bbaf6236fe980b0704Randall Spangler const unsigned char *us1 = s1; 176d03b527fe0107c3bb71d0bbaf6236fe980b0704Randall Spangler const unsigned char *us2 = s2; 186d03b527fe0107c3bb71d0bbaf6236fe980b0704Randall Spangler int result = 0; 196d03b527fe0107c3bb71d0bbaf6236fe980b0704Randall Spangler 206d03b527fe0107c3bb71d0bbaf6236fe980b0704Randall Spangler if (0 == size) 216d03b527fe0107c3bb71d0bbaf6236fe980b0704Randall Spangler return 0; 226d03b527fe0107c3bb71d0bbaf6236fe980b0704Randall Spangler 236d03b527fe0107c3bb71d0bbaf6236fe980b0704Randall Spangler /* 246d03b527fe0107c3bb71d0bbaf6236fe980b0704Randall Spangler * Code snippet without data-dependent branch due to Nate Lawson 256d03b527fe0107c3bb71d0bbaf6236fe980b0704Randall Spangler * (nate@root.org) of Root Labs. 266d03b527fe0107c3bb71d0bbaf6236fe980b0704Randall Spangler */ 276d03b527fe0107c3bb71d0bbaf6236fe980b0704Randall Spangler while (size--) 286d03b527fe0107c3bb71d0bbaf6236fe980b0704Randall Spangler result |= *us1++ ^ *us2++; 296d03b527fe0107c3bb71d0bbaf6236fe980b0704Randall Spangler 306d03b527fe0107c3bb71d0bbaf6236fe980b0704Randall Spangler return result != 0; 316d03b527fe0107c3bb71d0bbaf6236fe980b0704Randall Spangler} 326d03b527fe0107c3bb71d0bbaf6236fe980b0704Randall Spangler 33786acdabcc15f023330d7c628aca9679e757a238Randall Spanglerint vb2_align(uint8_t **ptr, uint32_t *size, uint32_t align, uint32_t want_size) 34786acdabcc15f023330d7c628aca9679e757a238Randall Spangler{ 35786acdabcc15f023330d7c628aca9679e757a238Randall Spangler uintptr_t p = (uintptr_t)*ptr; 36786acdabcc15f023330d7c628aca9679e757a238Randall Spangler uintptr_t offs = p & (align - 1); 37786acdabcc15f023330d7c628aca9679e757a238Randall Spangler 38786acdabcc15f023330d7c628aca9679e757a238Randall Spangler if (offs) { 39786acdabcc15f023330d7c628aca9679e757a238Randall Spangler offs = align - offs; 40786acdabcc15f023330d7c628aca9679e757a238Randall Spangler 41786acdabcc15f023330d7c628aca9679e757a238Randall Spangler if (*size < offs) 422145721c3c1840561030b27d2207006b0139c16cRandall Spangler return VB2_ERROR_ALIGN_BIGGER_THAN_SIZE; 43786acdabcc15f023330d7c628aca9679e757a238Randall Spangler 44786acdabcc15f023330d7c628aca9679e757a238Randall Spangler *ptr += offs; 45786acdabcc15f023330d7c628aca9679e757a238Randall Spangler *size -= offs; 46786acdabcc15f023330d7c628aca9679e757a238Randall Spangler } 47786acdabcc15f023330d7c628aca9679e757a238Randall Spangler 48786acdabcc15f023330d7c628aca9679e757a238Randall Spangler if (*size < want_size) 492145721c3c1840561030b27d2207006b0139c16cRandall Spangler return VB2_ERROR_ALIGN_SIZE; 50786acdabcc15f023330d7c628aca9679e757a238Randall Spangler 51786acdabcc15f023330d7c628aca9679e757a238Randall Spangler return VB2_SUCCESS; 52786acdabcc15f023330d7c628aca9679e757a238Randall Spangler} 53786acdabcc15f023330d7c628aca9679e757a238Randall Spangler 54786acdabcc15f023330d7c628aca9679e757a238Randall Spanglervoid vb2_workbuf_init(struct vb2_workbuf *wb, uint8_t *buf, uint32_t size) 55786acdabcc15f023330d7c628aca9679e757a238Randall Spangler{ 56786acdabcc15f023330d7c628aca9679e757a238Randall Spangler wb->buf = buf; 57786acdabcc15f023330d7c628aca9679e757a238Randall Spangler wb->size = size; 58786acdabcc15f023330d7c628aca9679e757a238Randall Spangler 59786acdabcc15f023330d7c628aca9679e757a238Randall Spangler /* Align the buffer so allocations will be aligned */ 60786acdabcc15f023330d7c628aca9679e757a238Randall Spangler if (vb2_align(&wb->buf, &wb->size, VB2_WORKBUF_ALIGN, 0)) 61786acdabcc15f023330d7c628aca9679e757a238Randall Spangler wb->size = 0; 62786acdabcc15f023330d7c628aca9679e757a238Randall Spangler} 63786acdabcc15f023330d7c628aca9679e757a238Randall Spangler 64786acdabcc15f023330d7c628aca9679e757a238Randall Spangler/** 65786acdabcc15f023330d7c628aca9679e757a238Randall Spangler * Round up a number to a multiple of VB2_WORKBUF_ALIGN 66786acdabcc15f023330d7c628aca9679e757a238Randall Spangler * 67786acdabcc15f023330d7c628aca9679e757a238Randall Spangler * @param v Number to round up 68786acdabcc15f023330d7c628aca9679e757a238Randall Spangler * @return The number, rounded up. 69786acdabcc15f023330d7c628aca9679e757a238Randall Spangler */ 70786acdabcc15f023330d7c628aca9679e757a238Randall Spanglerstatic __inline uint32_t wb_round_up(uint32_t v) 71786acdabcc15f023330d7c628aca9679e757a238Randall Spangler{ 72786acdabcc15f023330d7c628aca9679e757a238Randall Spangler return (v + VB2_WORKBUF_ALIGN - 1) & ~(VB2_WORKBUF_ALIGN - 1); 73786acdabcc15f023330d7c628aca9679e757a238Randall Spangler} 74786acdabcc15f023330d7c628aca9679e757a238Randall Spangler 75786acdabcc15f023330d7c628aca9679e757a238Randall Spanglervoid *vb2_workbuf_alloc(struct vb2_workbuf *wb, uint32_t size) 76786acdabcc15f023330d7c628aca9679e757a238Randall Spangler{ 77786acdabcc15f023330d7c628aca9679e757a238Randall Spangler uint8_t *ptr = wb->buf; 78786acdabcc15f023330d7c628aca9679e757a238Randall Spangler 79786acdabcc15f023330d7c628aca9679e757a238Randall Spangler /* Round up size to work buffer alignment */ 80786acdabcc15f023330d7c628aca9679e757a238Randall Spangler size = wb_round_up(size); 81786acdabcc15f023330d7c628aca9679e757a238Randall Spangler 82786acdabcc15f023330d7c628aca9679e757a238Randall Spangler if (size > wb->size) 83786acdabcc15f023330d7c628aca9679e757a238Randall Spangler return NULL; 84786acdabcc15f023330d7c628aca9679e757a238Randall Spangler 85786acdabcc15f023330d7c628aca9679e757a238Randall Spangler wb->buf += size; 86786acdabcc15f023330d7c628aca9679e757a238Randall Spangler wb->size -= size; 87786acdabcc15f023330d7c628aca9679e757a238Randall Spangler 88786acdabcc15f023330d7c628aca9679e757a238Randall Spangler return ptr; 89786acdabcc15f023330d7c628aca9679e757a238Randall Spangler} 90786acdabcc15f023330d7c628aca9679e757a238Randall Spangler 91786acdabcc15f023330d7c628aca9679e757a238Randall Spanglervoid *vb2_workbuf_realloc(struct vb2_workbuf *wb, 92786acdabcc15f023330d7c628aca9679e757a238Randall Spangler uint32_t oldsize, 93786acdabcc15f023330d7c628aca9679e757a238Randall Spangler uint32_t newsize) 94786acdabcc15f023330d7c628aca9679e757a238Randall Spangler{ 95786acdabcc15f023330d7c628aca9679e757a238Randall Spangler /* 96786acdabcc15f023330d7c628aca9679e757a238Randall Spangler * Just free and allocate to update the size. No need to move/copy 97786acdabcc15f023330d7c628aca9679e757a238Randall Spangler * memory, since the new pointer is guaranteed to be the same as the 98786acdabcc15f023330d7c628aca9679e757a238Randall Spangler * old one. The new allocation can fail, if the new size is too big. 99786acdabcc15f023330d7c628aca9679e757a238Randall Spangler */ 100786acdabcc15f023330d7c628aca9679e757a238Randall Spangler vb2_workbuf_free(wb, oldsize); 101786acdabcc15f023330d7c628aca9679e757a238Randall Spangler return vb2_workbuf_alloc(wb, newsize); 102786acdabcc15f023330d7c628aca9679e757a238Randall Spangler} 103786acdabcc15f023330d7c628aca9679e757a238Randall Spangler 104786acdabcc15f023330d7c628aca9679e757a238Randall Spanglervoid vb2_workbuf_free(struct vb2_workbuf *wb, uint32_t size) 105786acdabcc15f023330d7c628aca9679e757a238Randall Spangler{ 106786acdabcc15f023330d7c628aca9679e757a238Randall Spangler /* Round up size to work buffer alignment */ 107786acdabcc15f023330d7c628aca9679e757a238Randall Spangler size = wb_round_up(size); 108786acdabcc15f023330d7c628aca9679e757a238Randall Spangler 109786acdabcc15f023330d7c628aca9679e757a238Randall Spangler wb->buf -= size; 110786acdabcc15f023330d7c628aca9679e757a238Randall Spangler wb->size += size; 111786acdabcc15f023330d7c628aca9679e757a238Randall Spangler} 112786acdabcc15f023330d7c628aca9679e757a238Randall Spangler 113786acdabcc15f023330d7c628aca9679e757a238Randall Spanglerptrdiff_t vb2_offset_of(const void *base, const void *ptr) 114786acdabcc15f023330d7c628aca9679e757a238Randall Spangler{ 115786acdabcc15f023330d7c628aca9679e757a238Randall Spangler return (uintptr_t)ptr - (uintptr_t)base; 116786acdabcc15f023330d7c628aca9679e757a238Randall Spangler} 117