1 /* $NetBSD: common.h,v 1.8 2017/07/29 21:04:07 riastradh Exp $ */ 2 3 /*- 4 * Copyright (c) 2013 The NetBSD Foundation, Inc. 5 * All rights reserved. 6 * 7 * This code is derived from software contributed to The NetBSD Foundation 8 * by Taylor R. Campbell. 9 * 10 * Redistribution and use in source and binary forms, with or without 11 * modification, are permitted provided that the following conditions 12 * are met: 13 * 1. Redistributions of source code must retain the above copyright 14 * notice, this list of conditions and the following disclaimer. 15 * 2. Redistributions in binary form must reproduce the above copyright 16 * notice, this list of conditions and the following disclaimer in the 17 * documentation and/or other materials provided with the distribution. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 29 * POSSIBILITY OF SUCH DAMAGE. 30 */ 31 32 #ifndef VNDCOMPRESS_COMMON_H 33 #define VNDCOMPRESS_COMMON_H 34 35 #include <sys/cdefs.h> 36 #include <sys/param.h> 37 38 #include <limits.h> 39 #include <stdint.h> 40 41 #ifndef __type_fit 42 43 /* XXX Cruft from HEAD. */ 44 45 #define __type_mask(t) (/*LINTED*/sizeof(t) < sizeof(intmax_t) ? \ 46 (~((1ULL << (sizeof(t) * NBBY)) - 1)) : 0ULL) 47 48 #define __zeroll() (0LL) 49 #define __negative_p(x) ((x) < 0) 50 51 #define __type_min_s(t) ((t)((1ULL << (sizeof(t) * NBBY - 1)))) 52 #define __type_max_s(t) ((t)~((1ULL << (sizeof(t) * NBBY - 1)))) 53 #define __type_min_u(t) ((t)0ULL) 54 #define __type_max_u(t) ((t)~0ULL) 55 #define __type_is_signed(t) (/*LINTED*/__type_min_s(t) + (t)1 < (t)1) 56 #define __type_min(t) (__type_is_signed(t) ? __type_min_s(t) : __type_min_u(t)) 57 #define __type_max(t) (__type_is_signed(t) ? __type_max_s(t) : __type_max_u(t)) 58 59 60 #define __type_fit_u(t, a) (/*LINTED*/sizeof(t) < sizeof(intmax_t) ? \ 61 (((a) & __type_mask(t)) == 0) : !__negative_p(a)) 62 63 #define __type_fit_s(t, a) (/*LINTED*/__negative_p(a) ? \ 64 ((intmax_t)((a) + __zeroll()) >= (intmax_t)__type_min_s(t)) : \ 65 ((intmax_t)((a) + __zeroll()) <= (intmax_t)__type_max_s(t))) 66 67 /* 68 * return true if value 'a' fits in type 't' 69 */ 70 #define __type_fit(t, a) (__type_is_signed(t) ? \ 71 __type_fit_s(t, a) : __type_fit_u(t, a)) 72 73 #endif 74 75 /* XXX urk */ 76 #ifndef OFF_MAX 77 #define OFF_MAX __type_max(off_t) 78 #endif 79 80 /* XXX Why is this kernel-only? */ 81 #define SET(t, f) ((t) |= (f)) 82 #define CLR(t, f) ((t) &= ~(f)) 83 #define ISSET(t, f) ((t) & (f)) 84 85 /* 86 * Bounds checks for arithmetic. 87 */ 88 #define ADD_OK(T, A, B) ((A) <= __type_max(T) - (B)) 89 90 #define MUL_OK(T, A, B) ((A) <= __type_max(T)/(B)) 91 92 #define HOWMANY(X, N) (((X) + ((N) - 1))/(N)) 93 #define TOOMANY(T, X, N, M) (!ADD_OK(T, X, (N) - 1) || HOWMANY(X, N) > (M)) 94 95 /* 96 * We require: 97 * 98 * 0 < blocksize (duh) 99 * blocksize % DEV_BSIZE == 0 (for kernel use) 100 * 101 * blocksize <= UINT32_MAX (per the format) 102 * blocksize*2 <= SIZE_MAX (for a compression buffer) 103 * blocksize*2 <= ULONG_MAX (for zlib API) 104 */ 105 #define MIN_BLOCKSIZE DEV_BSIZE 106 #define DEF_BLOCKSIZE 0x10000 107 #define MAX_BLOCKSIZE \ 108 rounddown(MIN(UINT32_MAX, (MIN(SIZE_MAX, ULONG_MAX) / 2)), \ 109 MIN_BLOCKSIZE) 110 111 /* 112 * We require: 113 * 114 * n_offsets * sizeof(uint64_t) <= SIZE_MAX (array of 64-bit offsets) 115 * n_blocks = (n_offsets + 1) (extra offset to set last block end) 116 * n_blocks <= UINT32_MAX (per the format) 117 * n_offsets <= UINT32_MAX (for the sake of simplicity) 118 */ 119 #define MAX_N_BLOCKS \ 120 (MIN(UINT32_MAX, (SIZE_MAX / sizeof(uint64_t))) - 1) 121 #define MAX_N_OFFSETS (MAX_N_BLOCKS + 1) 122 123 /* 124 * The window size is at most the number of offsets, or the largest 125 * uint32_t value. The default window size is chosen so that windows 126 * fit in one 4096-byte page of memory. We could use 64k bytes, or 127 * st_blksize, to maximize I/O transfer size, but the transfers won't 128 * be aligned without a lot of extra work. 129 */ 130 #define MAX_WINDOW_SIZE MIN(UINT32_MAX, MAX_N_OFFSETS) 131 #define DEF_WINDOW_SIZE 512 132 133 struct cloop2_header { 134 char cl2h_magic[128]; 135 uint32_t cl2h_blocksize; 136 uint32_t cl2h_n_blocks; 137 } __packed; 138 139 #define CLOOP2_OFFSET_TABLE_OFFSET 136 140 141 struct options { 142 int flags; 143 #define FLAG_c 0x0001 /* Compress */ 144 #define FLAG_d 0x0002 /* Decompress */ 145 #define FLAG_p 0x0004 /* Partial */ 146 #define FLAG_r 0x0008 /* Restart */ 147 #define FLAG_b 0x0010 /* Block size */ 148 #define FLAG_R 0x0020 /* abort on Restart failure */ 149 #define FLAG_k 0x0040 /* checKpoint blocks */ 150 #define FLAG_l 0x0080 /* Length of input */ 151 #define FLAG_w 0x0100 /* Window size */ 152 uint32_t blocksize; 153 uint32_t end_block; /* end for partial transfer */ 154 uint32_t checkpoint_blocks; /* blocks before checkpoint */ 155 uint64_t length; /* length of image in bytes */ 156 uint32_t window_size; /* size of window into offset table */ 157 }; 158 159 int vndcompress(int, char **, const struct options *); 160 int vnduncompress(int, char **, const struct options *); 161 void usage(void) __dead; 162 163 static const char cloop2_magic[] = "\ 164 #!/bin/sh\n\ 165 #V2.0 Format\n\ 166 insmod cloop.o file=$0 && mount -r -t iso9660 /dev/cloop $1\n\ 167 exit $?\n\ 168 "; 169 170 #endif /* VNDCOMPRESS_COMMON_H */ 171