/*
 * Guillaume Cottenceau (gc@mandrakesoft.com)
 *
 * Copyright 2000 Mandrakesoft
 *
 * This software may be freely redistributed under the terms of the GNU
 * public license.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 */

/*
 * Portions from Erik Troan (ewt@redhat.com)
 *
 * Copyright 1996 Red Hat Software 
 *
 */


/*
 * This contains stuff related to probing:
 * (1) any (actually only SCSI, NET, CPQ, USB Controllers) devices (autoprobe for PCI and USB)
 * (2) IDE media
 * (3) SCSI media
 * (4) ETH devices
 */


#include <stdlib.h>
#include <unistd.h>
#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <sys/socket.h>
#include <net/if.h>
#include <sys/ioctl.h>
#include <sys/mount.h>
#include <pci/pci.h>
#include "stage1.h"

#include "log.h"
#include "frontend.h"
#include "modules.h"
#include "pci-resource/pci-ids.h"
#ifdef ENABLE_USB
#include "usb-resource/usb-ids.h"
#endif

#include "probing.h"


struct media_info {
	char * name;
	char * model;
	enum media_type type;
};


static void warning_insmod_failed(enum insmod_return r)
{
	if (IS_AUTOMATIC && r == INSMOD_FAILED_FILE_NOT_FOUND)
		return;
	if (r != INSMOD_OK) {
		if (r == INSMOD_FAILED_FILE_NOT_FOUND)
			stg1_error_message("This floppy doesn't contain the driver.");
		else
			stg1_error_message("Warning, installation of driver failed. (please include msg from <Alt-F3> for bugreports)");
	}
}

#ifndef DISABLE_NETWORK
struct net_description_elem
{
	char * intf_name;
	char * intf_description;
};
static struct net_description_elem net_descriptions[50];
static int net_descr_number = 0;
static char * intf_descr_for_discover = NULL;
static char * net_intf_too_early_name[50]; /* for modules providing more than one net intf */
static int net_intf_too_early_number = 0;
static int net_intf_too_early_ptr = 0;

void prepare_intf_descr(const char * intf_descr)
{
	intf_descr_for_discover = strdup(intf_descr);
}

void net_discovered_interface(char * intf_name)
{
	if (!intf_descr_for_discover) {
		net_intf_too_early_name[net_intf_too_early_number++] = strdup(intf_name);
		return;
	}
	if (!intf_name) {
		if (net_intf_too_early_ptr >= net_intf_too_early_number) {
			log_message("NET: was expecting another network interface (broken net module?)");
			return;
		}
		net_descriptions[net_descr_number].intf_name = net_intf_too_early_name[net_intf_too_early_ptr++];
	}
	else
		net_descriptions[net_descr_number].intf_name = strdup(intf_name);
	net_descriptions[net_descr_number].intf_description = strdup(intf_descr_for_discover);
	intf_descr_for_discover = NULL;
	net_descr_number++;
}

char * get_net_intf_description(char * intf_name)
{
	int i;
	for (i = 0; i < net_descr_number ; i++)
		if (!strcmp(net_descriptions[i].intf_name, intf_name))
			return net_descriptions[i].intf_description;
	return strdup("unknown");
}
#endif

static struct pci_module_map_full orphan_devices[50]; /* might be overkill, we only need the driver for now */
static int orphan_devices_len = 0;

int exists_orphan_device(char *driver) {
	int i;
	for (i = 0; i < orphan_devices_len ; i++) {
		if (!strcmp(orphan_devices[i].module, driver)) {
			log_message("probing: found orphan device for module %s", driver);
			return 1;
		}
	}
        log_message("probing: no orphan device for module %s", driver);
        return 0;
}

void discovered_device(enum driver_type type,
		       unsigned short vendor, unsigned short device, unsigned short subvendor, unsigned short subdevice,
		       const char * description, const char * driver)
{
	enum insmod_return failed = INSMOD_FAILED;
	log_message("PCI: device %04x %04x %04x %04x is \"%s\", driver is %s", vendor, device, subvendor, subdevice, description, driver);
#ifndef DISABLE_MEDIAS
	if (type == SCSI_ADAPTERS) {
		wait_message("Loading driver for SCSI adapter:\n \n%s", description);
		failed = my_insmod(driver, SCSI_ADAPTERS, NULL, 1);
		remove_wait_message();
		warning_insmod_failed(failed);
	}
#endif
#ifndef DISABLE_NETWORK
	if (type == NETWORK_DEVICES) {
		wait_message("Loading driver for network device:\n \n%s", description);
		prepare_intf_descr(description);
		failed = my_insmod(driver, NETWORK_DEVICES, NULL, 1);
		warning_insmod_failed(failed);
		remove_wait_message();
		if (intf_descr_for_discover) /* for modules providing more than one net intf */
			net_discovered_interface(NULL);
	}
#endif
#ifdef ENABLE_USB
	if (type == USB_CONTROLLERS)
                /* we can't allow additional modules floppy since we need usbkbd for keystrokes of usb keyboards */
		failed = my_insmod(driver, USB_CONTROLLERS, NULL, 0);
#endif

	if (failed != INSMOD_OK) {
		struct pci_module_map_full *dev = &orphan_devices[orphan_devices_len++];
		log_message("adding orphan device (%x, %x, %s)", vendor, device, driver);
		dev->vendor = vendor;
		dev->device = device;
		dev->subvendor = subvendor;
		dev->subdevice = subdevice;
		dev->name = description;
		dev->module = driver;
	}
}

#ifdef ENABLE_USB
void probe_that_type(enum driver_type type, enum media_bus bus)
#else
void probe_that_type(enum driver_type type, enum media_bus bus __attribute__ ((unused)))
#endif
{
        static int already_probed_usb_controllers = 0;
        static int already_loaded_usb_scsi = 0;

	/* ---- PCI probe ---------------------------------------------- */
	{
		FILE * f = NULL;
		unsigned int len = 0;
		unsigned int len_full = 0;
		char buf[200];
		struct pci_module_map * pcidb = NULL;
		struct pci_module_map_full * pcidb_full = NULL;

		switch (type) {
#ifndef DISABLE_PCIADAPTERS
#ifndef DISABLE_MEDIAS
			static int already_probed_scsi_adapters = 0;
		case SCSI_ADAPTERS:
			if (already_probed_scsi_adapters)
				goto end_pci_probe;
			already_probed_scsi_adapters = 1;
			pcidb = medias_pci_ids;
			len   = medias_num_ids;
			pcidb_full = medias_pci_ids_full;
			len_full   = medias_num_ids_full;
			break;
#endif
#ifndef DISABLE_NETWORK
		case NETWORK_DEVICES:
			pcidb = network_pci_ids;
			len   = network_num_ids;
			pcidb_full = network_pci_ids_full;
			len_full   = network_num_ids_full;
			break;
#endif
#endif
#ifdef ENABLE_USB
		case USB_CONTROLLERS:
			if (already_probed_usb_controllers || IS_NOAUTO)
				goto end_pci_probe;
			already_probed_usb_controllers = 1;
			pcidb = usb_pci_ids;
			len   = usb_num_ids;
			break;
#endif
		default:
			goto end_pci_probe;
		}

		if (!(f = fopen("/proc/bus/pci/devices", "rb"))) {
			log_message("PCI: could not open proc file");
			goto end_pci_probe;
		}

		while (1) {
			unsigned int i;
			unsigned short vendor, device, subvendor, subdevice, class_, class_prog, devbusfn;
			const char *name, *module;
			enum driver_type type_ = type;

			if (!fgets(buf, sizeof(buf), f)) break;
	
			sscanf(buf, "%hx %x", &devbusfn, &i);
			device = i;
			vendor = i >> 16;

			{
				int bus = devbusfn >> 8;
				int device_p = (devbusfn & 0xff) >> 3;
				int function = (devbusfn & 0xff) & 0x07;
				char file[100];
				int sf;
				sprintf(file, "/proc/bus/pci/%02x/%02x.%d", bus, device_p, function);
				if ((sf = open(file, O_RDONLY)) == -1) {
					log_message("PCI: could not open file for full probe (%s)", file);
					continue;
				}
				if (read(sf, buf, 48) == -1) {
					log_message("PCI: could not read 48 bytes from %s", file);
					close(sf);
					continue;
				}
				close(sf);
				memcpy(&class_prog, buf+9, 1);
				memcpy(&class_, buf+10, 2);
				memcpy(&subvendor, buf+44, 2);
				memcpy(&subdevice, buf+46, 2);
			}

			/* special rules below must be in sync with ldetect/pci.c */

			if (class_ == PCI_CLASS_SERIAL_USB) {
				/* taken from kudzu's pci.c */
				module = 
					class_prog == 0 ? "usb-uhci" : 
					class_prog == 0x10 ? "usb-ohci" :
					class_prog == 0x20 ? "ehci-hcd" : NULL;
				if (module) {
					name = "USB Controller";
					type_ = USB_CONTROLLERS;
					goto found_pci_device;
				}
			}
			if (class_ == PCI_CLASS_SERIAL_FIREWIRE) {
				/* taken from kudzu's pci.c */
				if (class_prog == 0x10) {
					module = strdup("ohci1394");
					name = "Firewire Controller";
					goto found_pci_device;
				}
			}

			for (i = 0; i < len_full; i++)
				if (pcidb_full[i].vendor == vendor && pcidb_full[i].device == device) {
					if (pcidb_full[i].subvendor == subvendor && pcidb_full[i].subdevice == subdevice) {
						name = pcidb_full[i].name;
						module = pcidb_full[i].module;
						goto found_pci_device;
					}
				}
			
			for (i = 0; i < len; i++)
				if (pcidb[i].vendor == vendor && pcidb[i].device == device) {
					name = pcidb[i].name;
					module = pcidb[i].module;
					goto found_pci_device;
				}

			continue;

		found_pci_device:
			discovered_device(type_, vendor, device, subvendor, subdevice, name, module);
		}
	end_pci_probe:;
		if (f)
                        fclose(f);
	}


#ifdef ENABLE_USB
	/* ---- USB probe ---------------------------------------------- */
	if ((bus == BUS_USB || bus == BUS_ANY) && !(IS_NOAUTO)) {
		static int already_mounted_usbdev = 0;

		FILE * f = NULL;
		int len = 0;
		char buf[200];
		struct usb_module_map * usbdb = NULL;

		if (!already_probed_usb_controllers) {
			already_probed_usb_controllers = 1;
			probe_that_type(USB_CONTROLLERS, BUS_ANY);
		}

		if (!already_mounted_usbdev) {
			already_mounted_usbdev = 1;
			if (mount("none", "/proc/bus/usb", "usbfs", 0, NULL) &&
			    mount("none", "/proc/bus/usb", "usbdevfs", 0, NULL)) {
				log_message("USB: couldn't mount /proc/bus/usb");
				goto end_usb_probe;
			}
			wait_message("Detecting USB devices.");
			sleep(4); /* sucking background work */
			my_insmod("usbkbd", ANY_DRIVER_TYPE, NULL, 0);
			my_insmod("keybdev", ANY_DRIVER_TYPE, NULL, 0);
			remove_wait_message();
		}

		if (!(f = fopen("/proc/bus/usb/devices", "rb"))) {
			log_message("USB: could not open proc file");
			goto end_usb_probe;
		}

		switch (type) {
		case NETWORK_DEVICES:
			usbdb = usb_usb_ids;
			len   = usb_usb_num_ids;
			break;
		default:
			goto end_usb_probe;
		}

		while (1) {
			int i, vendor, id;

			if (!fgets(buf, sizeof(buf), f)) break;

			if (sscanf(buf, "P:  Vendor=%x ProdID=%x", &vendor, &id) != 2)
				continue;

			for (i = 0; i < len; i++) {
				if (usbdb[i].vendor == vendor && usbdb[i].id == id) {
					log_message("USB: device %04x %04x is \"%s\" (%s)", vendor, id, usbdb[i].name, usbdb[i].module);
#ifndef DISABLE_NETWORK
					if (type == NETWORK_DEVICES) {
						stg1_info_message("About to load driver for usb network device:\n \n%s", usbdb[i].name);
						prepare_intf_descr(usbdb[i].name);
						warning_insmod_failed(my_insmod(usbdb[i].module, NETWORK_DEVICES, NULL, 1));
						if (intf_descr_for_discover) /* for modules providing more than one net intf */
							net_discovered_interface(NULL);
					}
#endif

				}
			}
		}
	end_usb_probe:;
		if (f)
                        fclose(f);
	}
#endif

        /* be sure to load usb-storage after SCSI adapters, so that they are in
           same order than reboot, so that naming is the same */
        if (type == SCSI_ADAPTERS && already_probed_usb_controllers && !already_loaded_usb_scsi) {
                already_loaded_usb_scsi = 1;
                /* we can't allow additional modules floppy since we need usbkbd for keystrokes of usb keyboards */
                my_insmod("usb-storage", SCSI_ADAPTERS, NULL, 0); 
                if (module_already_present("ieee1394"))
                        my_insmod("sbp2", SCSI_ADAPTERS, NULL, 0);
                wait_message("Detecting USB mass-storage devices.");
                sleep(10); /* sucking background work */
                remove_wait_message();
        }
}


static struct media_info * medias = NULL;

static void find_media(enum media_bus bus)
{
    	char b[50];
	char buf[5000];
	struct media_info tmp[50];
	int count = 0;
        int fd;

	if (medias)
		free(medias); /* that does not free the strings, by the way */

	if (bus == BUS_SCSI || bus == BUS_USB || bus == BUS_ANY) {
                log_message("looking for SCSI adapters");
                probe_that_type(SCSI_ADAPTERS, bus);
        }
	/* ----------------------------------------------- */
	if (bus != BUS_IDE && bus != BUS_ANY)
		goto find_media_after_ide;
	log_message("looking for ide media");

    	strcpy(b, "/proc/ide/hd");
    	for (b[12] = 'a'; b[12] <= 't'; b[12]++) {
		int i;
		char ide_disk[] = "disk";
		char ide_cdrom[] = "cdrom";
		char ide_tape[] = "tape";
		char ide_floppy[] = "floppy";
		
		/* first, test if file exists (will tell if attached medium exists) */
		b[13] = '\0';
		if (access(b, R_OK))
			continue;

		tmp[count].name = strdup("hda");
		tmp[count].name[2] = b[12];

		/* media type */
		strcpy(b + 13, "/media");
		fd = open(b, O_RDONLY);
		if (fd == -1) {
			log_message("failed to open %s for reading", b);
			continue;
		}

		i = read(fd, buf, sizeof(buf));
		if (i == -1) {
			log_message("failed to read %s", b);
			continue;
		}
		buf[i] = '\0';
		close(fd);

		if (ptr_begins_static_str(buf, ide_disk))
			tmp[count].type = DISK;
		else if (ptr_begins_static_str(buf, ide_cdrom))
			tmp[count].type = CDROM;
		else if (ptr_begins_static_str(buf, ide_tape))
			tmp[count].type = TAPE;
		else if (ptr_begins_static_str(buf, ide_floppy))
			tmp[count].type = FLOPPY;
		else
			tmp[count].type = UNKNOWN_MEDIA;

		/* media model */
		strcpy(b + 13, "/model");
		fd = open(b, O_RDONLY);
		if (fd == -1) {
			log_message("failed to open %s for reading", b);
			continue;
		}

		i = read(fd, buf, sizeof(buf));
		if (i <= 0) {
			log_message("failed to read %s", b);
			tmp[count].model = strdup("(none)");
		}
		else {
			buf[i-1] = '\0'; /* eat the \n */
			tmp[count].model = strdup(buf);
		}
		close(fd);

		log_message("IDE/%d: %s is a %s", tmp[count].type, tmp[count].name, tmp[count].model);
		count++;
    	}

 find_media_after_ide:
	/* ----------------------------------------------- */
	if (bus != BUS_SCSI && bus != BUS_USB && bus != BUS_ANY)
		goto find_media_after_scsi;
	log_message("looking for scsi media");

	fd = open("/proc/scsi/scsi", O_RDONLY);
	if (fd != -1) {
                FILE * f;

		enum { SCSI_TOP, SCSI_HOST, SCSI_VENDOR, SCSI_TYPE } state = SCSI_TOP;
		char * start, * chptr, * next, * end;
		char scsi_disk_count = 'a';
		char scsi_cdrom_count = '0';
		char scsi_tape_count = '0';

		char scsi_no_devices[] = "Attached devices: none";
		char scsi_some_devices[] = "Attached devices:";
		char scsi_host[] = "Host: ";
		char scsi_vendor[] = "  Vendor: ";

		int i = read(fd, &buf, sizeof(buf)-1);
		if (i < 1) {
			close(fd);
			goto end_scsi;
		}
		close(fd);
		buf[i] = '\0';

		if (ptr_begins_static_str(buf, scsi_no_devices))
			goto end_scsi;
		
		start = buf;
		while (*start) {
			char tmp_model[50];
			char tmp_name[10];

			chptr = start;
			while (*chptr != '\n') chptr++;
			*chptr = '\0';
			next = chptr + 1;
			
			switch (state) {
			case SCSI_TOP:
				if (!ptr_begins_static_str(start, scsi_some_devices))
					goto end_scsi;
				state = SCSI_HOST;
				break;

			case SCSI_HOST:
				if (!ptr_begins_static_str(start, scsi_host))
					goto end_scsi;
				state = SCSI_VENDOR;
				break;

			case SCSI_VENDOR:
				if (!ptr_begins_static_str(start, scsi_vendor))
					goto end_scsi;

				/* (1) Grab Vendor info */
				start += 10;
				end = chptr = strstr(start, "Model:");
				if (!chptr)
					goto end_scsi;

				chptr--;
				while (*chptr == ' ')
					chptr--;
				if (*chptr == ':') {
					chptr++;
					*(chptr + 1) = '\0';
					strcpy(tmp_model,"(unknown)");
				} else {
					*(chptr + 1) = '\0';
					strcpy(tmp_model, start);
				}

				/* (2) Grab Model info */
				start = end;
				start += 7;
				
				chptr = strstr(start, "Rev:");
				if (!chptr)
					goto end_scsi;
				
				chptr--;
				while (*chptr == ' ') chptr--;
				*(chptr + 1) = '\0';
				
				strcat(tmp_model, " ");
				strcat(tmp_model, start);

				tmp[count].model = strdup(tmp_model);
				
				state = SCSI_TYPE;

				break;

			case SCSI_TYPE:
				if (strncmp("  Type:", start, 7))
					goto end_scsi;
				*tmp_name = '\0';

				if (strstr(start, "Direct-Access")) {
					sprintf(tmp_name, "sd%c", scsi_disk_count++);
					tmp[count].type = DISK;
				} else if (strstr(start, "Sequential-Access")) {
					sprintf(tmp_name, "st%c", scsi_tape_count++);
					tmp[count].type = TAPE;
				} else if (strstr(start, "CD-ROM")) {
					sprintf(tmp_name, "sr%c", scsi_cdrom_count++);
					tmp[count].type = CDROM;
				}

                                if (!(f = fopen("/proc/partitions", "rb")) || !fgets(buf, sizeof(buf), f) || !fgets(buf, sizeof(buf), f)) {
                                        log_message("Couldn't open /proc/partitions");
                                } else {
                                        while (fgets(buf, sizeof(buf), f)) {
                                                char name[100];
                                                int major, minor, blocks;
                                                memset(name, 0, sizeof(name));
                                                sscanf(buf, " %d %d %d %s", &major, &minor, &blocks, name);
                                                if (streq(name, tmp_name) && tmp[count].type == DISK && ((blocks == 1048575) || (blocks == 1440)))
                                                        tmp[count].type = FLOPPY;
                                        }
                                        fclose(f);
                                }

				if (*tmp_name) {
					tmp[count].name = strdup(tmp_name);
					log_message("SCSI/%d: %s is a %s", tmp[count].type, tmp[count].name, tmp[count].model);
					count++;
				}
				
				state = SCSI_HOST;
			}
			
			start = next;
		}
		
	end_scsi:;
	}

	/* ----------------------------------------------- */
	log_message("looking for Compaq Smart Array media");
	{
		char * procfiles[] = { "/proc/driver/cpqarray/ida0", "/proc/driver/cciss/cciss0", // 2.4 style
				       "/proc/array/ida", "/proc/cciss/cciss",                 // 2.2 style
				       NULL };
		static char cpq_descr[] = "Compaq RAID logical disk";
		char ** procfile = procfiles;
		FILE * f;
		while (procfile && *procfile && (f = fopen(*procfile, "rb"))) {
			while (fgets(buf, sizeof(buf), f)) {
				if (ptr_begins_static_str(buf, "ida/") || ptr_begins_static_str(buf, "cciss/")) {
					char * end = strchr(buf, ':');
					if (!end)
						log_message("Inconsistency in %s, line:\n%s", *procfile, buf);
					else {
						*end = '\0';
						tmp[count].name = strdup(buf);
						tmp[count].type = DISK;
						tmp[count].model = cpq_descr;
						log_message("CPQ: found %s", tmp[count].name);
						count++;
					}
				}
			}
			fclose(f);
			procfile++;
		}
	}

	/* ----------------------------------------------- */
	log_message("looking for DAC960");
	{
		FILE * f;
		if ((f = fopen("/tmp/syslog", "rb"))) {
			while (fgets(buf, sizeof(buf), f)) {
				char * start;
				if ((start = strstr(buf, "/dev/rd/"))) {
					char * end = strchr(start, ':');
					if (!end)
						log_message("Inconsistency in syslog, line:\n%s", buf);
					else {
						*end = '\0';
						tmp[count].name = strdup(start+5);
						tmp[count].type = DISK;
						start = end + 2;
						end = strchr(start, ',');
						if (end) {
							*end = '\0';
							tmp[count].model = strdup(start);
						} else
							tmp[count].model = "(unknown)";
						log_message("DAC960: found %s (%s)", tmp[count].name, tmp[count].model);
						count++;
					}
				}
			}
			fclose(f);
		}
	}
 find_media_after_scsi:

	/* ----------------------------------------------- */
	tmp[count].name = NULL;
	count++;

	medias = memdup(tmp, sizeof(struct media_info) * count);
}


/* Finds by media */
void get_medias(enum media_type media, char *** names, char *** models, enum media_bus bus)
{
	struct media_info * m;
	char * tmp_names[50];
	char * tmp_models[50];
	int count;

	find_media(bus);

	m = medias;

	count = 0;
	while (m && m->name) {
		if (m->type == media) {
			tmp_names[count] = strdup(m->name);
			tmp_models[count++] = strdup(m->model);
		}
		m++;
	}
	tmp_names[count] = NULL;
	tmp_models[count++] = NULL;

	*names = memdup(tmp_names, sizeof(char *) * count);
	*models = memdup(tmp_models, sizeof(char *) * count);
}


#ifndef DISABLE_NETWORK
int net_device_available(char * device) {
	struct ifreq req;
	int s;
    
	s = socket(AF_INET, SOCK_DGRAM, 0);
	if (s < 0) {
		log_perror(device);
		return 0;
	}
	strcpy(req.ifr_name, device);
	if (ioctl(s, SIOCGIFFLAGS, &req)) {
		/* if we can't get the flags, the networking device isn't available */
		close(s);
		return 0;
	}
	close(s);
	return 1;
}


char ** get_net_devices(void)
{
	char * devices[] = {
		"eth0", "eth1", "eth2", "eth3", "eth4", "eth5", "eth6", "eth7", "eth8", "eth9",
		"tr0",
		"plip0", "plip1", "plip2",
		"fddi0",
#ifdef ENABLE_USB
		"usb0", "usb1", "usb2", "usb3",
#endif
		NULL
	};
	char ** ptr = devices;
	char * tmp[50];
	int i = 0;
	static int already_probed = 0;

	if (!already_probed) {
		already_probed = 1; /* cut off loop brought by: probe_that_type => my_insmod => get_net_devices */
		probe_that_type(NETWORK_DEVICES, BUS_ANY);
	}

	while (ptr && *ptr) {
		if (net_device_available(*ptr))
			tmp[i++] = strdup(*ptr);
		ptr++;
	}
	tmp[i++] = NULL;

	return memdup(tmp, sizeof(char *) * i);
}
#endif /* DISABLE_NETWORK */