#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <pci/pci.h>
#include "libldetect.h"
#include "libldetect-private.h"
#include "common.h"

static int pci_find_modules(struct pciusb_entries entries, int no_subid) {
  return pciusb_find_modules(entries, "pcitable", no_subid);
}

extern struct pciusb_entries pci_probe(int probe_type) {
  FILE *f;
  char buf[512];
  unsigned short devbusfn;
  unsigned int id;
  struct pciusb_entry t[100];
  struct pciusb_entries r;

  if (!(f = fopen("/proc/bus/pci/devices", "r"))) exit(1);

  for (r.nb = 0; fgets(buf, sizeof(buf) - 1, f) && r.nb < psizeof(t); r.nb++) {
    struct pciusb_entry e;
    pciusb_initialize(&e);

    sscanf(buf, "%hx %x", &devbusfn, &id);
    e.vendor = id >> 16;
    e.device = id & 0xffff;
    e.pci_bus = devbusfn >> 8;
    e.pci_device = (devbusfn & 0xff) >> 3;
    e.pci_function = (devbusfn & 0xff) & 0x07;
    {
      char file[25];
      FILE *f;
      snprintf(file, sizeof(file), "/proc/bus/pci/%02x/%02x.%d", e.pci_bus, e.pci_device, e.pci_function);
      if (probe_type == 1 && (f = fopen(file, "r"))) {
        if (fseek(f, 10, SEEK_SET) == 0) 
	  fread(&e.class_, 2, 1, f);
        if (fseek(f, 0x2c, SEEK_SET) == 0) 
	  fread(&e.subvendor, 2, 1, f), fread(&e.subdevice, 2, 1, f);

	if ((e.subvendor == 0 && e.subdevice == 0) ||
	    (e.subvendor == e.vendor && e.subdevice == e.device)) {
	  e.subvendor = 0xffff;
	  e.subdevice = 0xffff;
	}

        if (fseek(f, 9, SEEK_SET) == 0) {
	  unsigned char class_prog = 0;
	  fread(&class_prog, 1, 1, f);
	  if (e.class_ == PCI_CLASS_SERIAL_USB) {
	    /* taken from kudzu's pci.c */
	    e.module = strdup(class_prog == 0 ? "usb-uhci" : "usb-ohci");
	  }
	}

	fclose(f);
      }
    }
    t[r.nb] = e;
  }
  fclose(f);
  r.entries = memdup(t, sizeof(struct pciusb_entry) * r.nb);

  if (pci_find_modules(r, probe_type)) return r;

  /* ok, let's try again with subids */
  if (probe_type == 0) return pci_probe(1);

  /* should not happen */
  exit(1);
}