/* * ---------------------------------------------------------------------------- * "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 * ---------------------------------------------------------------------------- * * $Id: disk.c,v 1.19.2.2 1995/06/05 02:24:27 jkh Exp $ * */ #include #include #include #include #include #include #include #include #include #include #include #include "libdisk.h" #define DOSPTYP_EXTENDED 5 #define DOSPTYP_ONTRACK 84 char *chunk_n[] = { "whole", "unknown", "fat", "freebsd", "extended", "part", "unused", NULL }; struct disk * Open_Disk(char *name) { return Int_Open_Disk(name,0); } struct disk * Int_Open_Disk(char *name, u_long size) { int i,fd; struct diskslices ds; struct disklabel dl; char device[64]; struct disk *d; struct dos_partition *dp; void *p; u_long offset = 0; strcpy(device,"/dev/r"); strcat(device,name); d = (struct disk *)malloc(sizeof *d); if(!d) err(1,"malloc failed"); memset(d,0,sizeof *d); fd = open(device,O_RDONLY); if (fd < 0) { warn("open(%s) failed",device); return 0; } memset(&dl,0,sizeof dl); ioctl(fd,DIOCGDINFO,&dl); i = ioctl(fd,DIOCGSLICEINFO,&ds); if (i < 0) { warn("DIOCGSLICEINFO(%s) failed",device); close(fd); return 0; } #ifdef DEBUG for(i=0;idp_start >= size) continue; if (dp->dp_start+dp->dp_size >= size) continue; if (!dp->dp_size) continue; if (dp->dp_typ == DOSPTYP_ONTRACK) { d->flags |= DISK_ON_TRACK; offset = 63; } } free(p); 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); if (Add_Chunk(d, -offset, size, name,whole,0,0)) warn("Failed to add 'whole' chunk"); 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, 0) && j != 3) warn( "Failed to add chunk for partition %c [%lu,%lu]", j + 'a',dl.d_partitions[j].p_offset, dl.d_partitions[j].p_size); } } } close(fd); Fixup_Names(d); Bios_Limit_Chunk(d->chunks,1024*d->bios_hd*d->bios_sect); 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\n",d->bios_cyl,d->bios_hd,d->bios_sect); printf(" boot1=%p, boot2=%p, bootmgr=%p\n", d->boot1,d->boot2,d->bootmgr); Debug_Chunk(d->chunks); } void Free_Disk(struct disk *d) { if(d->chunks) Free_Chunk(d->chunks); if(d->name) free(d->name); if(d->bootmgr) free(d->bootmgr); if(d->boot1) free(d->boot1); if(d->boot2) free(d->boot2); free(d); } struct disk * Clone_Disk(struct disk *d) { struct disk *d2; d2 = (struct disk*) malloc(sizeof *d2); if(!d2) err(1,"malloc failed"); *d2 = *d; d2->name = strdup(d2->name); d2->chunks = Clone_Chunk(d2->chunks); if(d2->bootmgr) { d2->bootmgr = malloc(DOSPARTOFF); memcpy(d2->bootmgr,d->bootmgr,DOSPARTOFF); } 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); } return d2; } #if 0 void Collapse_Disk(struct disk *d) { while(Collapse_Chunk(d,d->chunks)) ; } #endif static char * device_list[] = {"wd","sd",0}; char ** Disk_Names() { int i,j,k; char disk[25]; char diskname[25]; struct stat st; struct diskslices ds; int fd; static char **disks; disks = malloc(sizeof *disks * (1 + MAX_NO_DISKS)); memset(disks,0,sizeof *disks * (1 + MAX_NO_DISKS)); k = 0; for (j = 0; device_list[j]; j++) { for (i = 0; i < 10; i++) { sprintf(diskname, "%s%d", device_list[j], i); sprintf(disk, "/dev/r%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) { close(fd); continue; } disks[k++] = strdup(diskname); if(k == MAX_NO_DISKS) return disks; } } return disks; } void Set_Boot_Mgr(struct disk *d, u_char *b) { if (d->bootmgr) free(d->bootmgr); if (!b) { d->bootmgr = 0; } else { d->bootmgr = malloc(DOSPARTOFF); if(!d->bootmgr) err(1,"malloc failed"); memcpy(d->bootmgr,b,DOSPARTOFF); } } void Set_Boot_Blocks(struct disk *d, u_char *b1, u_char *b2) { if (d->boot1) free(d->boot1); d->boot1 = malloc(512); if(!d->boot1) err(1,"malloc failed"); memcpy(d->boot1,b1,512); if (d->boot2) free(d->boot2); d->boot2 = malloc(15*512); if(!d->boot2) err(1,"malloc failed"); memcpy(d->boot2,b2,15*512); }