/*
 * ----------------------------------------------------------------------------
 * "THE BEER-WARE LICENSE" (Revision 42):
 * <phk@login.dknet.dk> 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.3 1995/04/29 04:00:55 phk Exp $
 *
 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <string.h>
#include <err.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/ioctl.h>
#include <sys/disklabel.h>
#include <sys/diskslice.h>
#include "libdisk.h"

#define DOSPTYP_EXTENDED        5
#define DOSPTYP_ONTRACK         84      

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;
	char device[64];
	struct disk *d;

	strcpy(device,"/dev/r");
	strcat(device,name);

	fd = open(device,O_RDONLY);
	if (fd < 0) {
		warn("open(%s) failed",device);
		return 0;
	}
	i = ioctl(fd,DIOCGSLICEINFO,&ds);
	if (i < 0) {
		warn("DIOCSLICEINFO(%s) failed",device);
		close(fd);
		return 0;
	}

	d = (struct disk *)malloc(sizeof *d);
	if(!d) err(1,"malloc failed");

	memset(d,0,sizeof *d);

	d->name = strdup(name);

	if (!size)
		size = ds.dss_slices[WHOLE_DISK_SLICE].ds_size;

	if (Add_Chunk(d, 0, size, name,whole,0,0))
		warn("Failed to add 'whole' chunk");
	if (ds.dss_slices[COMPATIBILITY_SLICE].ds_offset)
		if (Add_Chunk(d, 0, 1, "-",reserved,0,0))
			warn("Failed to add MBR chunk");
	
	for(i=BASE_SLICE;i<ds.dss_nslices;i++) {
		char sname[20];
		chunk_e ce;
		u_long flags=0;
		int subtype=0;
		if (! ds.dss_slices[i].ds_size)
			continue;
		sprintf(sname,"%ss%d",name,i-1);
		switch (ds.dss_slices[i].ds_type) {
			case 0xa5:
				ce = freebsd;
				break;
			case 0x1:
			case 0x6:
				ce = fat;
				break;
			case DOSPTYP_EXTENDED:
				ce = extended;
				break;
			default:
				ce = foo;
				subtype = -ds.dss_slices[i].ds_type;
				break;
		}	
		flags |= CHUNK_ALIGN;
		if (Add_Chunk(d,ds.dss_slices[i].ds_offset,
			ds.dss_slices[i].ds_size, sname,ce,subtype,flags))
			warn("failed to add chunk for slice %d",i - 1);
		if (ce == extended)
			if (Add_Chunk(d,ds.dss_slices[i].ds_offset,
				1, "-",reserved, subtype, flags))
				warn("failed to add MBR chunk for slice %d",i - 1);
		if (ds.dss_slices[i].ds_type == 0xa5) {
			struct disklabel *dl;
			int j;

			dl = read_disklabel(fd,
				ds.dss_slices[i].ds_offset + LABELSECTOR);
			if(dl) {
				char pname[20];
				for(j=0; j < dl->d_npartitions; j++) {
					sprintf(pname,"%s%c",sname,j+'a');
					if (j == 2 || j == 3)
						continue;
					if (!dl->d_partitions[j].p_size)
						continue;
					if (Add_Chunk(d,
						dl->d_partitions[j].p_offset,
						dl->d_partitions[j].p_size,
						pname,part,0,0))
						warn("Failed to add chunk for partition %c",j + 'a');
				}
				sprintf(pname,"%sd",sname);
				if (!dl->d_partitions[3].p_size)
					continue;
				Add_Chunk(d,
					dl->d_partitions[3].p_offset,
					dl->d_partitions[3].p_size,
					pname,part,0,0);
			}
			free(dl);
		}
	}
	close(fd);
	return d;
}

void
Debug_Disk(struct disk *d)
{
	printf("Debug_Disk(%s)",d->name);
	printf("  flags=%lx",d->flags);
	printf("  real_geom=%lu/%lu/%lu",d->real_cyl,d->real_hd,d->real_sect);
	printf("  bios_geom=%lu/%lu/%lu\n",d->bios_cyl,d->bios_hd,d->bios_sect);
	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*7);
		memcpy(d2->boot2,d->boot2,512*7);
	}
	return d2;
}

void
Collapse_Disk(struct disk *d)
{

	while(Collapse_Chunk(d,d->chunks))
		;
}

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);	
	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(7*512);
	if(!d->boot2) err(1,"malloc failed");
	memcpy(d->boot2,b2,7*512);
}