1 /*-
2 * Copyright (c) 2014 Juniper Networks, Inc.
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 #include <sys/errno.h>
29 #include <stdint.h>
30 #include <stdlib.h>
31 #include <string.h>
32
33 #include <bsd.h>
34
35 #include "endian.h"
36 #include "image.h"
37 #include "mkimg.h"
38 #include "scheme.h"
39
40 static struct mkimg_alias bsd_aliases[] = {
41 { ALIAS_FREEBSD_NANDFS, ALIAS_INT2TYPE(FS_NANDFS) },
42 { ALIAS_FREEBSD_SWAP, ALIAS_INT2TYPE(FS_SWAP) },
43 { ALIAS_FREEBSD_UFS, ALIAS_INT2TYPE(FS_BSDFFS) },
44 { ALIAS_FREEBSD_VINUM, ALIAS_INT2TYPE(FS_VINUM) },
45 { ALIAS_FREEBSD_ZFS, ALIAS_INT2TYPE(FS_ZFS) },
46 { ALIAS_NONE, 0 }
47 };
48
49 static lba_t
bsd_metadata(u_int where,lba_t blk)50 bsd_metadata(u_int where, lba_t blk)
51 {
52
53 if (where == SCHEME_META_IMG_START)
54 blk += BSD_BOOTBLOCK_SIZE / secsz;
55 else if (where == SCHEME_META_IMG_END)
56 blk = round_cylinder(blk);
57 else
58 blk = round_block(blk);
59 return (blk);
60 }
61
62 static int
bsd_write(lba_t imgsz,void * bootcode)63 bsd_write(lba_t imgsz, void *bootcode)
64 {
65 u_char *buf, *p;
66 struct disklabel *d;
67 struct partition *dp;
68 struct part *part;
69 int bsdparts, error, n;
70 uint16_t checksum;
71
72 buf = malloc(BSD_BOOTBLOCK_SIZE);
73 if (buf == NULL)
74 return (ENOMEM);
75 if (bootcode != NULL) {
76 memcpy(buf, bootcode, BSD_BOOTBLOCK_SIZE);
77 memset(buf + secsz, 0, sizeof(struct disklabel));
78 } else
79 memset(buf, 0, BSD_BOOTBLOCK_SIZE);
80
81 bsdparts = nparts + 1; /* Account for c partition */
82 if (bsdparts < BSD_NPARTS_MIN)
83 bsdparts = BSD_NPARTS_MIN;
84
85 d = (void *)(buf + secsz);
86 le32enc(&d->d_magic, BSD_MAGIC);
87 le32enc(&d->d_secsize, secsz);
88 le32enc(&d->d_nsectors, nsecs);
89 le32enc(&d->d_ntracks, nheads);
90 le32enc(&d->d_ncylinders, ncyls);
91 le32enc(&d->d_secpercyl, nsecs * nheads);
92 le32enc(&d->d_secperunit, imgsz);
93 le16enc(&d->d_rpm, 3600);
94 le32enc(&d->d_magic2, BSD_MAGIC);
95 le16enc(&d->d_npartitions, bsdparts);
96 le32enc(&d->d_bbsize, BSD_BOOTBLOCK_SIZE);
97
98 dp = &d->d_partitions[BSD_PART_RAW];
99 le32enc(&dp->p_size, imgsz);
100 TAILQ_FOREACH(part, &partlist, link) {
101 n = part->index + ((part->index >= BSD_PART_RAW) ? 1 : 0);
102 dp = &d->d_partitions[n];
103 le32enc(&dp->p_size, part->size);
104 le32enc(&dp->p_offset, part->block);
105 le32enc(&dp->p_fsize, 0);
106 dp->p_fstype = ALIAS_TYPE2INT(part->type);
107 dp->p_frag = 0;
108 le16enc(&dp->p_cpg, 0);
109 }
110
111 dp = &d->d_partitions[bsdparts];
112 checksum = 0;
113 for (p = (void *)d; p < (u_char *)dp; p += 2)
114 checksum ^= le16dec(p);
115 le16enc(&d->d_checksum, checksum);
116
117 error = image_write(0, buf, BSD_BOOTBLOCK_SIZE / secsz);
118 free(buf);
119 return (error);
120 }
121
122 static struct mkimg_scheme bsd_scheme = {
123 .name = "bsd",
124 .description = "BSD disk label",
125 .aliases = bsd_aliases,
126 .metadata = bsd_metadata,
127 .write = bsd_write,
128 .nparts = BSD_NPARTS_MAX - 1,
129 .bootcode = BSD_BOOTBLOCK_SIZE,
130 .maxsecsz = 512
131 };
132
133 SCHEME_DEFINE(bsd_scheme);
134