/* * ---------------------------------------------------------------------------- * "THE BEER-WARE LICENSE" (Revision 42): * wrote this file. As long as you retain this notice you * can do whatever you want with this stuff. If we meet some day, and you think * this stuff is worth it, you can buy me a beer in return. Poul-Henning Kamp * ---------------------------------------------------------------------------- * * $FreeBSD: src/lib/libdisk/disk.c,v 1.50.2.15 2001/12/30 09:56:12 phk Exp $ * $DragonFly: src/lib/libdisk/Attic/disk.c,v 1.3 2003/11/10 06:14:40 dillon Exp $ * */ #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include "libdisk.h" #define DOSPTYP_EXTENDED 5 #define DOSPTYP_ONTRACK 84 const char *chunk_n[] = { "whole", "unknown", "fat", "freebsd", "extended", "part", "unused", NULL }; struct disk * Open_Disk(const char *name) { return Int_Open_Disk(name, 0); } #ifndef PC98 static u_int32_t Read_Int32(u_int32_t *p) { u_int8_t *bp = (u_int8_t *)p; return bp[0] | (bp[1] << 8) | (bp[2] << 16) | (bp[3] << 24); } #endif struct disk * Int_Open_Disk(const char *name, u_long size) { int i,fd; struct diskslices ds; struct disklabel dl; char device[64], *buf; struct disk *d; u_long sector_size; #ifdef PC98 unsigned char *p; #else struct dos_partition *dp; void *p; #endif u_long offset = 0; strcpy(device, _PATH_DEV); strcat(device, name); d = (struct disk *)malloc(sizeof *d); if(!d) return NULL; memset(d, 0, sizeof *d); fd = open(device, O_RDONLY); if (fd < 0) { #ifdef DEBUG warn("open(%s) failed", device); #endif return 0; } memset(&dl, 0, sizeof dl); ioctl(fd, DIOCGDINFO, &dl); i = ioctl(fd, DIOCGSLICEINFO, &ds); if (i < 0) { #ifdef DEBUG warn("DIOCGSLICEINFO(%s) failed", device); #endif close(fd); return 0; } #ifdef DEBUG for(i = 0; i < ds.dss_nslices; i++) if(ds.dss_slices[i].ds_openmask) printf(" open(%d)=0x%2x", i, ds.dss_slices[i].ds_openmask); printf("\n"); #endif /* XXX --- ds.dss_slice[WHOLE_DISK_SLICE].ds.size of MO disk is wrong!!! */ #ifdef PC98 if (!size) size = dl.d_ncylinders * dl.d_ntracks * dl.d_nsectors; #else if (!size) size = ds.dss_slices[WHOLE_DISK_SLICE].ds_size; #endif /* determine media sector size */ if ((buf = malloc(MAX_SEC_SIZE)) == NULL) return NULL; for (sector_size = MIN_SEC_SIZE; sector_size <= MAX_SEC_SIZE; sector_size *= 2) { if (read(fd, buf, sector_size) == sector_size) { d->sector_size = sector_size; break; } } free (buf); if (sector_size > MAX_SEC_SIZE) return NULL; /* could not determine sector size */ #ifdef PC98 p = (unsigned char*)read_block(fd, 1, sector_size); #else p = read_block(fd, 0, sector_size); dp = (struct dos_partition*)(p + DOSPARTOFF); for (i = 0; i < NDOSPART; i++) { if (Read_Int32(&dp->dp_start) >= size) continue; if (Read_Int32(&dp->dp_start) + Read_Int32(&dp->dp_size) >= size) continue; if (!Read_Int32(&dp->dp_size)) continue; if (dp->dp_typ == DOSPTYP_ONTRACK) { d->flags |= DISK_ON_TRACK; offset = 63; } } free(p); #endif d->bios_sect = dl.d_nsectors; d->bios_hd = dl.d_ntracks; d->name = strdup(name); if (dl.d_ntracks && dl.d_nsectors) d->bios_cyl = size / (dl.d_ntracks * dl.d_nsectors); #ifdef PC98 if (Add_Chunk(d, -offset, size, name, whole, 0, 0, "-")) #else if (Add_Chunk(d, -offset, size, name, whole, 0, 0)) #endif #ifdef DEBUG warn("Failed to add 'whole' chunk"); #else {} #endif #ifdef __i386__ #ifdef PC98 /* XXX -- Quick Hack! * Check MS-DOS MO */ if ((*p == 0xf0 || *p == 0xf8) && (*(p+1) == 0xff) && (*(p+2) == 0xff)) { Add_Chunk(d, 0, size, name, fat, 0xa0a0, 0, name); free(p); goto pc98_mo_done; } free(p); #endif /* PC98 */ for(i=BASE_SLICE;i ds.dss_slices[i].ds_size) continue; sprintf(pname, "%s%c", sname, j + 'a'); if (Add_Chunk(d, dl.d_partitions[j].p_offset + ds.dss_slices[i].ds_offset, dl.d_partitions[j].p_size, pname,part, dl.d_partitions[j].p_fstype, #ifdef PC98 0, ds.dss_slices[i].ds_name) && j != 3) #else 0) && j != 3) #endif #ifdef DEBUG warn( "Failed to add chunk for partition %c [%lu,%lu]", j + 'a', dl.d_partitions[j].p_offset, dl.d_partitions[j].p_size); #else {} #endif } } } #endif /* __i386__ */ #ifdef __alpha__ { struct disklabel dl; char pname[20]; int j,k; strcpy(pname, _PATH_DEV); strcat(pname, name); j = open(pname, O_RDONLY); if (j < 0) { #ifdef DEBUG warn("open(%s)", pname); #endif goto nolabel; } k = ioctl(j, DIOCGDINFO, &dl); if (k < 0) { #ifdef DEBUG warn("ioctl(%s, DIOCGDINFO)", pname); #endif close(j); goto nolabel; } close(j); All_FreeBSD(d, 1); for(j = 0; j <= dl.d_npartitions; j++) { if (j == RAW_PART) continue; if (j == 3) continue; if (j == dl.d_npartitions) { j = 3; dl.d_npartitions = 0; } if (!dl.d_partitions[j].p_size) continue; if (dl.d_partitions[j].p_size + dl.d_partitions[j].p_offset > ds.dss_slices[WHOLE_DISK_SLICE].ds_size) continue; sprintf(pname, "%s%c", name, j + 'a'); if (Add_Chunk(d, dl.d_partitions[j].p_offset, dl.d_partitions[j].p_size, pname,part, dl.d_partitions[j].p_fstype, 0) && j != 3) #ifdef DEBUG warn( "Failed to add chunk for partition %c [%lu,%lu]", j + 'a', dl.d_partitions[j].p_offset, dl.d_partitions[j].p_size); #else {} #endif } nolabel:; } #endif /* __alpha__ */ #ifdef PC98 pc98_mo_done: #endif close(fd); Fixup_Names(d); return d; } void Debug_Disk(struct disk *d) { printf("Debug_Disk(%s)", d->name); printf(" flags=%lx", d->flags); #if 0 printf(" real_geom=%lu/%lu/%lu", d->real_cyl, d->real_hd, d->real_sect); #endif printf(" bios_geom=%lu/%lu/%lu = %lu\n", d->bios_cyl, d->bios_hd, d->bios_sect, d->bios_cyl * d->bios_hd * d->bios_sect); #if defined(PC98) printf(" boot1=%p, boot2=%p, bootipl=%p, bootmenu=%p\n", d->boot1, d->boot2, d->bootipl, d->bootmenu); #elif defined(__i386__) printf(" boot1=%p, boot2=%p, bootmgr=%p\n", d->boot1, d->boot2, d->bootmgr); #elif defined(__alpha__) printf(" boot1=%p, bootmgr=%p\n", d->boot1, d->bootmgr); #endif Debug_Chunk(d->chunks); } void Free_Disk(struct disk *d) { if(d->chunks) Free_Chunk(d->chunks); if(d->name) free(d->name); #ifdef PC98 if(d->bootipl) free(d->bootipl); if(d->bootmenu) free(d->bootmenu); #else if(d->bootmgr) free(d->bootmgr); #endif if(d->boot1) free(d->boot1); #if defined(__i386__) if(d->boot2) free(d->boot2); #endif free(d); } struct disk * Clone_Disk(struct disk *d) { struct disk *d2; d2 = (struct disk*) malloc(sizeof *d2); if(!d2) return NULL; *d2 = *d; d2->name = strdup(d2->name); d2->chunks = Clone_Chunk(d2->chunks); #ifdef PC98 if(d2->bootipl) { d2->bootipl = malloc(d2->bootipl_size); memcpy(d2->bootipl, d->bootipl, d2->bootipl_size); } if(d2->bootmenu) { d2->bootmenu = malloc(d2->bootmenu_size); memcpy(d2->bootmenu, d->bootmenu, d2->bootmenu_size); } #else if(d2->bootmgr) { d2->bootmgr = malloc(d2->bootmgr_size); memcpy(d2->bootmgr, d->bootmgr, d2->bootmgr_size); } #endif #if defined(__i386__) if(d2->boot1) { d2->boot1 = malloc(512); memcpy(d2->boot1, d->boot1, 512); } if(d2->boot2) { d2->boot2 = malloc(512 * 15); memcpy(d2->boot2, d->boot2, 512 * 15); } #elif defined(__alpha__) if(d2->boot1) { d2->boot1 = malloc(512 * 15); memcpy(d2->boot1, d->boot1, 512 * 15); } #endif return d2; } #if 0 void Collapse_Disk(struct disk *d) { while(Collapse_Chunk(d, d->chunks)) ; } #endif #ifdef PC98 static char * device_list[] = {"wd", "aacd", "ad", "da", "afd", "fla", "idad", "mlxd", "amrd", "twed", "ar", "fd", 0}; #else static char * device_list[] = {"aacd", "ad", "da", "afd", "fla", "idad", "mlxd", "amrd", "twed", "ar", "fd", 0}; #endif int qstrcmp(const void* a, const void* b) { char *str1 = *(char**)a; char *str2 = *(char**)b; return strcmp(str1, str2); } char ** Disk_Names() { int i,j,disk_cnt; char disk[25]; char diskname[25]; struct stat st; struct diskslices ds; int fd; static char **disks; int error; size_t listsize; char *disklist, **dp; disks = malloc(sizeof *disks * (1 + MAX_NO_DISKS)); memset(disks,0,sizeof *disks * (1 + MAX_NO_DISKS)); #if !defined(PC98) && !defined(KERN_DISKS_BROKEN) error = sysctlbyname("kern.disks", NULL, &listsize, NULL, 0); if (!error) { disklist = (char *)malloc(listsize); error = sysctlbyname("kern.disks", disklist, &listsize, NULL, 0); if (error) return NULL; disk_cnt = 0; for (dp = disks; ((*dp = strsep(&disklist, " ")) != NULL) && disk_cnt < MAX_NO_DISKS; disk_cnt++, dp++); } else { warn("kern.disks sysctl not available"); #endif disk_cnt = 0; for (j = 0; device_list[j]; j++) { if(disk_cnt >= MAX_NO_DISKS) break; for (i = 0; i < MAX_NO_DISKS; i++) { sprintf(diskname, "%s%d", device_list[j], i); sprintf(disk, _PATH_DEV"%s", diskname); if (stat(disk, &st) || !(st.st_mode & S_IFCHR)) continue; if ((fd = open(disk, O_RDWR)) == -1) continue; if (ioctl(fd, DIOCGSLICEINFO, &ds) == -1) { #ifdef DEBUG warn("DIOCGSLICEINFO %s", disk); #endif close(fd); continue; } close(fd); disks[disk_cnt++] = strdup(diskname); if(disk_cnt >= MAX_NO_DISKS) break; } } #if !defined(PC98) && !defined(KERN_DISKS_BROKEN) } #endif qsort(disks, disk_cnt, sizeof(char*), qstrcmp); return disks; } #ifdef PC98 void Set_Boot_Mgr(struct disk *d, const u_char *bootipl, const size_t bootipl_size, const u_char *bootmenu, const size_t bootmenu_size) #else void Set_Boot_Mgr(struct disk *d, const u_char *b, const size_t s) #endif { #ifdef PC98 if (bootipl_size % d->sector_size != 0) return; if (d->bootipl) free(d->bootipl); if (!bootipl) { d->bootipl = NULL; } else { d->bootipl_size = bootipl_size; d->bootipl = malloc(bootipl_size); if(!d->bootipl) return; memcpy(d->bootipl, bootipl, bootipl_size); } if (bootmenu_size % d->sector_size != 0) return; if (d->bootmenu) free(d->bootmenu); if (!bootmenu) { d->bootmenu = NULL; } else { d->bootmenu_size = bootmenu_size; d->bootmenu = malloc(bootmenu_size); if(!d->bootmenu) return; memcpy(d->bootmenu, bootmenu, bootmenu_size); } #else if (s % d->sector_size != 0) return; if (d->bootmgr) free(d->bootmgr); if (!b) { d->bootmgr = NULL; } else { d->bootmgr_size = s; d->bootmgr = malloc(s); if(!d->bootmgr) return; memcpy(d->bootmgr, b, s); } #endif } int Set_Boot_Blocks(struct disk *d, const u_char *b1, const u_char *b2) { #if defined(__i386__) if (d->boot1) free(d->boot1); d->boot1 = malloc(512); if(!d->boot1) return -1; memcpy(d->boot1, b1, 512); if (d->boot2) free(d->boot2); d->boot2 = malloc(15 * 512); if(!d->boot2) return -1; memcpy(d->boot2, b2, 15 * 512); #elif defined(__alpha__) if (d->boot1) free(d->boot1); d->boot1 = malloc(15 * 512); if(!d->boot1) return -1; memcpy(d->boot1, b1, 15 * 512); #endif return 0; } const char * slice_type_name( int type, int subtype ) { switch (type) { case 0: return "whole"; #ifndef PC98 case 1: switch (subtype) { case 1: return "fat (12-bit)"; case 2: return "XENIX /"; case 3: return "XENIX /usr"; case 4: return "fat (16-bit,<=32Mb)"; case 5: return "extended DOS"; case 6: return "fat (16-bit,>32Mb)"; case 7: return "NTFS/HPFS/QNX"; case 8: return "AIX bootable"; case 9: return "AIX data"; case 10: return "OS/2 bootmgr"; case 11: return "fat (32-bit)"; case 12: return "fat (32-bit,LBA)"; case 14: return "fat (16-bit,>32Mb,LBA)"; case 15: return "extended DOS, LBA"; case 18: return "Compaq Diagnostic"; case 84: return "OnTrack diskmgr"; case 100: return "Netware 2.x"; case 101: return "Netware 3.x"; case 115: return "SCO UnixWare"; case 128: return "Minix 1.1"; case 129: return "Minix 1.5"; case 130: return "linux_swap"; case 131: return "ext2fs"; case 166: return "OpenBSD FFS"; /* 0xA6 */ case 169: return "NetBSD FFS"; /* 0xA9 */ case 182: return "OpenBSD"; /* dedicated */ case 183: return "bsd/os"; case 184: return "bsd/os swap"; default: return "unknown"; } #endif case 2: return "fat"; case 3: switch (subtype) { #ifdef PC98 case 0xc494: return "freebsd"; #else case 165: return "freebsd"; #endif default: return "unknown"; } #ifndef PC98 case 4: return "extended"; case 5: return "part"; case 6: return "unused"; #endif default: return "unknown"; } }