lib/libdisk/write_pc98_disk.c

   1 /*
   2  * ----------------------------------------------------------------------------
   3  * "THE BEER-WARE LICENSE" (Revision 42):
   4  * <phk@FreeBSD.org> wrote this file.  As long as you retain this notice you
   5  * can do whatever you want with this stuff. If we meet some day, and you think
   6  * this stuff is worth it, you can buy me a beer in return.   Poul-Henning Kamp
   7  * ----------------------------------------------------------------------------
   8  */
   9
  10 #include <sys/cdefs.h>
  11 __FBSDID("$FreeBSD$");
  12
  13 #include <stdio.h>
  14 #include <stdlib.h>
  15 #include <unistd.h>
  16 #include <fcntl.h>
  17 #include <string.h>
  18 #include <err.h>
  19 #include <sys/types.h>
  20 #include <sys/stat.h>
  21 #include <sys/ioctl.h>
  22 #include <sys/disklabel.h>
  23 #include <sys/diskpc98.h>
  24 #include <paths.h>
  25 #include "libdisk.h"
  26
  27 /*
  28  * XXX: A lot of hardcoded 512s probably should be foo->sector_size;
  29  *      I'm not sure which, so I leave it like it worked before. --schweikh
  30  */
  31 static int
  32 Write_FreeBSD(int fd, const struct disk *new, const struct chunk *c1)
  33 {
  34         struct disklabel *dl;
  35         int i;
  36         void *p;
  37         u_char buf[BBSIZE];
  38
  39         for (i = 0; i < BBSIZE / 512; i++) {
  40                 if (!(p = read_block(fd, i + c1->offset, 512)))
  41                         return (1);
  42                 memcpy(buf + 512 * i, p, 512);
  43                 free(p);
  44         }
  45         if (new->boot1)
  46                 memcpy(buf, new->boot1, 512);
  47
  48         if (new->boot2)
  49                 memcpy(buf + 512, new->boot2, BBSIZE - 512);
  50
  51         dl = (struct disklabel *)(buf + 512 * LABELSECTOR + LABELOFFSET);
  52         Fill_Disklabel(dl, new, c1);
  53
  54         for (i = 0; i < BBSIZE / 512; i++)
  55                 write_block(fd, i + c1->offset, buf + 512 * i, 512);
  56
  57         return 0;
  58 }
  59
  60
  61 int
  62 Write_Disk(const struct disk *d1)
  63 {
  64         int fd, i, j;
  65         struct chunk *c1;
  66         int ret = 0;
  67         char device[64];
  68         u_char *mbrblk;
  69         struct pc98_partition *dp, work[NDOSPART];
  70         int s[7];
  71         int PC98_EntireDisk = 0;
  72
  73         strcpy(device, _PATH_DEV);
  74         strcat(device, d1->name);
  75
  76         /* XXX - for entire FreeBSD(98) */
  77         for (c1 = d1->chunks->part; c1; c1 = c1->next) {
  78             if ((c1->type == freebsd) || (c1->offset == 0))
  79                 device[9] = 0;
  80         }
  81
  82         fd = open(device, O_RDWR);
  83         if (fd < 0) {
  84 #ifdef DEBUG
  85                 warn("open(%s) failed", device);
  86 #endif
  87                 return 1;
  88         }
  89
  90         memset(s, 0, sizeof s);
  91         if (!(mbrblk = read_block(fd, 1, d1->sector_size))) {
  92                 close (fd);
  93                 return (1);
  94         }
  95         dp = (struct pc98_partition *)(mbrblk + DOSPARTOFF);
  96         memcpy(work, dp, sizeof work);
  97         dp = work;
  98         free(mbrblk);
  99         for (c1 = d1->chunks->part; c1; c1 = c1->next) {
 100                 if (c1->type == unused)
 101                         continue;
 102                 if (!strcmp(c1->name, "X"))
 103                         continue;
 104                 j = c1->name[strlen(d1->name) + 1] - '1';
 105                 if (j < 0 || j > 7)
 106                         continue;
 107                 s[j]++;
 108                 if (c1->type == freebsd)
 109                         ret += Write_FreeBSD(fd, d1, c1);
 110
 111                 i = c1->offset;
 112                 dp[j].dp_ssect = dp[j].dp_ipl_sct = i % d1->bios_sect;
 113                 i -= dp[j].dp_ssect;
 114                 i /= d1->bios_sect;
 115                 dp[j].dp_shd = dp[j].dp_ipl_head = i % d1->bios_hd;
 116                 i -= dp[j].dp_shd;
 117                 i /= d1->bios_hd;
 118                 dp[j].dp_scyl = dp[j].dp_ipl_cyl = i;
 119 #ifdef DEBUG
 120                 printf("S:%lu = (%x/%x/%x)", c1->offset,
 121                        dp[j].dp_scyl, dp[j].dp_shd, dp[j].dp_ssect);
 122 #endif
 123
 124                 i = c1->end;
 125 #if 1
 126                 dp[j].dp_esect = dp[j].dp_ehd = 0;
 127                 dp[j].dp_ecyl = i / (d1->bios_sect * d1->bios_hd);
 128 #else
 129                 dp[j].dp_esect = i % d1->bios_sect;
 130                 i -= dp[j].dp_esect;
 131                 i /= d1->bios_sect;
 132                 dp[j].dp_ehd = i % d1->bios_hd;
 133                 i -= dp[j].dp_ehd;
 134                 i /= d1->bios_hd;
 135                 dp[j].dp_ecyl = i;
 136 #endif
 137 #ifdef DEBUG
 138                 printf("  E:%lu = (%x/%x/%x)\n", c1->end,
 139                        dp[j].dp_ecyl, dp[j].dp_ehd, dp[j].dp_esect);
 140 #endif
 141
 142                 dp[j].dp_mid = c1->subtype & 0xff;
 143                 dp[j].dp_sid = c1->subtype >> 8;
 144                 if (c1->flags & CHUNK_ACTIVE)
 145                         dp[j].dp_mid |= 0x80;
 146
 147                 strncpy(dp[j].dp_name, c1->sname, 16);
 148         }
 149         j = 0;
 150         for (i = 0; i < NDOSPART; i++) {
 151                 if (!s[i])
 152                         memset(dp + i, 0, sizeof *dp);
 153         }
 154
 155         if (d1->bootipl)
 156                 write_block(fd, 0, d1->bootipl, d1->sector_size);
 157
 158         if (!(mbrblk = read_block(fd, 1, d1->sector_size))) {
 159                 close (fd);
 160                 return (1);
 161         }
 162         memcpy(mbrblk + DOSPARTOFF, dp, sizeof *dp * NDOSPART);
 163         /* XXX - for entire FreeBSD(98) */
 164         for (c1 = d1->chunks->part; c1; c1 = c1->next)
 165                 if (((c1->type == freebsd) || (c1->type == fat))
 166                          && (c1->offset == 0))
 167                         PC98_EntireDisk = 1;
 168         if (PC98_EntireDisk == 0)
 169                 write_block(fd, 1, mbrblk, d1->sector_size);
 170
 171         if (d1->bootmenu)
 172                 for (i = 0; i * d1->sector_size < d1->bootmenu_size; i++)
 173                         write_block(fd, 2 + i,
 174                                     &d1->bootmenu[i * d1->sector_size],
 175                                     d1->sector_size);
 176
 177         close(fd);
 178         return 0;
 179 }