1 /*- 2 * Copyright (c) 2016 Marcel Moolenaar 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24 * SUCH DAMAGE. 25 */ 26 27 #include <sys/cdefs.h> 28 __FBSDID("$FreeBSD$"); 29 30 #include <stdint.h> 31 #include <stdio.h> 32 #include <string.h> 33 34 #include "endian.h" 35 #include "image.h" 36 #include "mkimg.h" 37 38 static void osdep_uuidgen(mkimg_uuid_t *); 39 40 #ifdef __APPLE__ 41 #include <uuid/uuid.h> 42 43 static void 44 osdep_uuidgen(mkimg_uuid_t *uuid) 45 { 46 47 uuid_generate_time((void *)uuid); 48 } 49 #endif /* __APPLE__ */ 50 51 #ifdef __FreeBSD__ 52 #include <sys/uuid.h> 53 54 static void 55 osdep_uuidgen(mkimg_uuid_t *uuid) 56 { 57 58 uuidgen((void *)uuid, 1); 59 } 60 #endif /* __FreeBSD__ */ 61 62 #ifdef __linux__ 63 #include <stdlib.h> 64 #include <time.h> 65 66 static void 67 osdep_uuidgen(mkimg_uuid_t *uuid) 68 { 69 struct timespec tp; 70 uint64_t time = 0x01B21DD213814000LL; 71 u_int i; 72 uint16_t seq; 73 74 if (clock_gettime(CLOCK_REALTIME, &tp) == -1) 75 abort(); 76 77 time += (uint64_t)tp.tv_sec * 10000000LL; 78 time += tp.tv_nsec / 100; 79 80 uuid->time_low = (uint32_t)time; 81 uuid->time_mid = (uint16_t)(time >> 32); 82 uuid->time_hi_and_version = (uint16_t)(time >> 48) & 0xfff; 83 uuid->time_hi_and_version |= 1 << 12; 84 85 seq = random(); 86 87 uuid->clock_seq_hi_and_reserved = (uint8_t)(seq >> 8) & 0x3f; 88 uuid->clock_seq_low = (uint8_t)seq; 89 90 for (i = 0; i < 6; i++) 91 uuid->node[i] = (uint8_t)random(); 92 uuid->node[0] |= 0x01; 93 } 94 #endif /* __linux__ */ 95 96 void 97 mkimg_uuid(mkimg_uuid_t *uuid) 98 { 99 static uint8_t gen[sizeof(mkimg_uuid_t)]; 100 u_int i; 101 102 if (!unit_testing) { 103 osdep_uuidgen(uuid); 104 return; 105 } 106 107 for (i = 0; i < sizeof(gen); i++) 108 gen[i]++; 109 memcpy(uuid, gen, sizeof(*uuid)); 110 } 111 112 void 113 mkimg_uuid_enc(void *buf, const mkimg_uuid_t *uuid) 114 { 115 uint8_t *p = buf; 116 u_int i; 117 118 le32enc(p, uuid->time_low); 119 le16enc(p + 4, uuid->time_mid); 120 le16enc(p + 6, uuid->time_hi_and_version); 121 p[8] = uuid->clock_seq_hi_and_reserved; 122 p[9] = uuid->clock_seq_low; 123 for (i = 0; i < 6; i++) 124 p[10 + i] = uuid->node[i]; 125 } 126