/*
   *	drivers/pci/setup-bus.c
   *
   * Extruded from code written by
   *      Dave Rusling (david.rusling@reo.mts.dec.com)
   *      David Mosberger (davidm@cs.arizona.edu)
   *	David Miller (davem@redhat.com)
   *
   * Support routines for initializing a PCI subsystem.
   */
  
  /*
   * Nov 2000, Ivan Kokshaysky <ink@jurassic.park.msu.ru>
   *	     PCI-PCI bridges cleanup, sorted resource allocation.
   * Feb 2002, Ivan Kokshaysky <ink@jurassic.park.msu.ru>
   *	     Converted to allocation in 3 passes, which gives
   *	     tighter packing. Prefetchable range support.
   */
  
  #include <linux/init.h>
  #include <linux/kernel.h>
  #include <linux/module.h>
  #include <linux/pci.h>
  #include <linux/errno.h>
  #include <linux/ioport.h>
  #include <linux/cache.h>
  #include <linux/slab.h>

  #include "pci.h"

  struct resource_list_x {
  	struct resource_list_x *next;
  	struct resource *res;
  	struct pci_dev *dev;
  	resource_size_t start;
  	resource_size_t end;

  	resource_size_t add_size;

  	resource_size_t min_align;

  	unsigned long flags;
  };

  #define free_list(type, head) do {                      \
  	struct type *list, *tmp;			\
  	for (list = (head)->next; list;) {		\
  		tmp = list;				\
  		list = list->next;			\
  		kfree(tmp);				\
  	}						\
  	(head)->next = NULL;				\
  } while (0)

  int pci_realloc_enable = 0;
  #define pci_realloc_enabled() pci_realloc_enable
  void pci_realloc(void)
  {
  	pci_realloc_enable = 1;
  }

  /**
   * add_to_list() - add a new resource tracker to the list
   * @head:	Head of the list
   * @dev:	device corresponding to which the resource
   *		belongs
   * @res:	The resource to be tracked
   * @add_size:	additional size to be optionally added
   *              to the resource
   */
  static void add_to_list(struct resource_list_x *head,
  		 struct pci_dev *dev, struct resource *res,

  		 resource_size_t add_size, resource_size_t min_align)

  {
  	struct resource_list_x *list = head;
  	struct resource_list_x *ln = list->next;
  	struct resource_list_x *tmp;
  
  	tmp = kmalloc(sizeof(*tmp), GFP_KERNEL);
  	if (!tmp) {

  		pr_warning("add_to_list: kmalloc() failed!
  ");

  		return;
  	}
  
  	tmp->next = ln;
  	tmp->res = res;
  	tmp->dev = dev;
  	tmp->start = res->start;
  	tmp->end = res->end;
  	tmp->flags = res->flags;

  	tmp->add_size = add_size;

  	tmp->min_align = min_align;

  	list->next = tmp;
  }

  static void add_to_failed_list(struct resource_list_x *head,
  				struct pci_dev *dev, struct resource *res)
  {

  	add_to_list(head, dev, res,
  			0 /* dont care */,
  			0 /* dont care */);

  }

  static void __dev_sort_resources(struct pci_dev *dev,
  				 struct resource_list *head)

  {

  	u16 class = dev->class >> 8;

  	/* Don't touch classless devices or host bridges or ioapics.  */
  	if (class == PCI_CLASS_NOT_DEFINED || class == PCI_CLASS_BRIDGE_HOST)
  		return;

  	/* Don't touch ioapic devices already enabled by firmware */
  	if (class == PCI_CLASS_SYSTEM_PIC) {
  		u16 command;
  		pci_read_config_word(dev, PCI_COMMAND, &command);
  		if (command & (PCI_COMMAND_IO | PCI_COMMAND_MEMORY))
  			return;
  	}

  	pdev_sort_resources(dev, head);
  }

  static inline void reset_resource(struct resource *res)
  {
  	res->start = 0;
  	res->end = 0;
  	res->flags = 0;
  }

  /**

   * reassign_resources_sorted() - satisfy any additional resource requests

   *

   * @realloc_head : head of the list tracking requests requiring additional

   *             resources
   * @head     : head of the list tracking requests with allocated
   *             resources
   *

   * Walk through each element of the realloc_head and try to procure

   * additional resources for the element, provided the element
   * is in the head list.
   */

  static void reassign_resources_sorted(struct resource_list_x *realloc_head,

  		struct resource_list *head)

  {
  	struct resource *res;

  	struct resource_list_x *list, *tmp, *prev;
  	struct resource_list *hlist;
  	resource_size_t add_size;

  	int idx;

  	prev = realloc_head;
  	for (list = realloc_head->next; list;) {

  		res = list->res;

  		/* skip resource that has been reset */
  		if (!res->flags)
  			goto out;
  
  		/* skip this resource if not found in head list */
  		for (hlist = head->next; hlist && hlist->res != res;
  				hlist = hlist->next);
  		if (!hlist) { /* just skip */
  			prev = list;
  			list = list->next;
  			continue;
  		}

  		idx = res - &list->dev->resource[0];

  		add_size=list->add_size;

  		if (!resource_size(res)) {

  			res->start = list->start;

  			res->end = res->start + add_size - 1;
  			if(pci_assign_resource(list->dev, idx))

  				reset_resource(res);

  		} else {
  			resource_size_t align = list->min_align;
  			res->flags |= list->flags & (IORESOURCE_STARTALIGN|IORESOURCE_SIZEALIGN);
  			if (pci_reassign_resource(list->dev, idx, add_size, align))
  				dev_printk(KERN_DEBUG, &list->dev->dev, "failed to add optional resources res=%pR
  ",
  							res);

  		}
  out:
  		tmp = list;
  		prev->next = list = list->next;
  		kfree(tmp);
  	}
  }
  
  /**
   * assign_requested_resources_sorted() - satisfy resource requests
   *
   * @head : head of the list tracking requests for resources
   * @failed_list : head of the list tracking requests that could
   *		not be allocated
   *
   * Satisfy resource requests of each element in the list. Add
   * requests that could not satisfied to the failed_list.
   */
  static void assign_requested_resources_sorted(struct resource_list *head,
  				 struct resource_list_x *fail_head)
  {
  	struct resource *res;
  	struct resource_list *list;
  	int idx;

  	for (list = head->next; list; list = list->next) {
  		res = list->res;
  		idx = res - &list->dev->resource[0];
  		if (resource_size(res) && pci_assign_resource(list->dev, idx)) {

  			if (fail_head && !pci_is_root_bus(list->dev->bus)) {
  				/*
  				 * if the failed res is for ROM BAR, and it will
  				 * be enabled later, don't add it to the list
  				 */
  				if (!((idx == PCI_ROM_RESOURCE) &&
  				      (!(res->flags & IORESOURCE_ROM_ENABLE))))
  					add_to_failed_list(fail_head, list->dev, res);
  			}

  			reset_resource(res);

  		}

  	}
  }

  static void __assign_resources_sorted(struct resource_list *head,

  				 struct resource_list_x *realloc_head,

  				 struct resource_list_x *fail_head)
  {
  	/* Satisfy the must-have resource requests */
  	assign_requested_resources_sorted(head, fail_head);

  	/* Try to satisfy any additional optional resource

  		requests */

  	if (realloc_head)
  		reassign_resources_sorted(realloc_head, head);

  	free_list(resource_list, head);
  }

  static void pdev_assign_resources_sorted(struct pci_dev *dev,
  				 struct resource_list_x *fail_head)
  {
  	struct resource_list head;
  
  	head.next = NULL;
  	__dev_sort_resources(dev, &head);

  	__assign_resources_sorted(&head, NULL, fail_head);

  
  }
  
  static void pbus_assign_resources_sorted(const struct pci_bus *bus,

  					 struct resource_list_x *realloc_head,

  					 struct resource_list_x *fail_head)
  {
  	struct pci_dev *dev;
  	struct resource_list head;
  
  	head.next = NULL;
  	list_for_each_entry(dev, &bus->devices, bus_list)
  		__dev_sort_resources(dev, &head);

  	__assign_resources_sorted(&head, realloc_head, fail_head);

  }

  void pci_setup_cardbus(struct pci_bus *bus)

  {
  	struct pci_dev *bridge = bus->self;

  	struct resource *res;

  	struct pci_bus_region region;

  	dev_info(&bridge->dev, "CardBus bridge to [bus %02x-%02x]
  ",
  		 bus->secondary, bus->subordinate);

  	res = bus->resource[0];
  	pcibios_resource_to_bus(bridge, &region, res);
  	if (res->flags & IORESOURCE_IO) {

  		/*
  		 * The IO resource is allocated a range twice as large as it
  		 * would normally need.  This allows us to set both IO regs.
  		 */

  		dev_info(&bridge->dev, "  bridge window %pR
  ", res);

  		pci_write_config_dword(bridge, PCI_CB_IO_BASE_0,
  					region.start);
  		pci_write_config_dword(bridge, PCI_CB_IO_LIMIT_0,
  					region.end);
  	}

  	res = bus->resource[1];
  	pcibios_resource_to_bus(bridge, &region, res);
  	if (res->flags & IORESOURCE_IO) {
  		dev_info(&bridge->dev, "  bridge window %pR
  ", res);

  		pci_write_config_dword(bridge, PCI_CB_IO_BASE_1,
  					region.start);
  		pci_write_config_dword(bridge, PCI_CB_IO_LIMIT_1,
  					region.end);
  	}

  	res = bus->resource[2];
  	pcibios_resource_to_bus(bridge, &region, res);
  	if (res->flags & IORESOURCE_MEM) {
  		dev_info(&bridge->dev, "  bridge window %pR
  ", res);

  		pci_write_config_dword(bridge, PCI_CB_MEMORY_BASE_0,
  					region.start);
  		pci_write_config_dword(bridge, PCI_CB_MEMORY_LIMIT_0,
  					region.end);
  	}

  	res = bus->resource[3];
  	pcibios_resource_to_bus(bridge, &region, res);
  	if (res->flags & IORESOURCE_MEM) {
  		dev_info(&bridge->dev, "  bridge window %pR
  ", res);

  		pci_write_config_dword(bridge, PCI_CB_MEMORY_BASE_1,
  					region.start);
  		pci_write_config_dword(bridge, PCI_CB_MEMORY_LIMIT_1,
  					region.end);
  	}
  }

  EXPORT_SYMBOL(pci_setup_cardbus);

  
  /* Initialize bridges with base/limit values we have collected.
     PCI-to-PCI Bridge Architecture Specification rev. 1.1 (1998)
     requires that if there is no I/O ports or memory behind the
     bridge, corresponding range must be turned off by writing base
     value greater than limit to the bridge's base/limit registers.
  
     Note: care must be taken when updating I/O base/limit registers
     of bridges which support 32-bit I/O. This update requires two
     config space writes, so it's quite possible that an I/O window of
     the bridge will have some undesirable address (e.g. 0) after the
     first write. Ditto 64-bit prefetchable MMIO.  */

  static void pci_setup_bridge_io(struct pci_bus *bus)

  {
  	struct pci_dev *bridge = bus->self;

  	struct resource *res;

  	struct pci_bus_region region;

  	u32 l, io_upper16;

  
  	/* Set up the top and bottom of the PCI I/O segment for this bus. */

  	res = bus->resource[0];
  	pcibios_resource_to_bus(bridge, &region, res);
  	if (res->flags & IORESOURCE_IO) {

  		pci_read_config_dword(bridge, PCI_IO_BASE, &l);
  		l &= 0xffff0000;
  		l |= (region.start >> 8) & 0x00f0;
  		l |= region.end & 0xf000;
  		/* Set up upper 16 bits of I/O base/limit. */
  		io_upper16 = (region.end & 0xffff0000) | (region.start >> 16);

  		dev_info(&bridge->dev, "  bridge window %pR
  ", res);

  	} else {

  		/* Clear upper 16 bits of I/O base/limit. */
  		io_upper16 = 0;
  		l = 0x00f0;

  	}
  	/* Temporarily disable the I/O range before updating PCI_IO_BASE. */
  	pci_write_config_dword(bridge, PCI_IO_BASE_UPPER16, 0x0000ffff);
  	/* Update lower 16 bits of I/O base/limit. */
  	pci_write_config_dword(bridge, PCI_IO_BASE, l);
  	/* Update upper 16 bits of I/O base/limit. */
  	pci_write_config_dword(bridge, PCI_IO_BASE_UPPER16, io_upper16);

  }
  
  static void pci_setup_bridge_mmio(struct pci_bus *bus)
  {
  	struct pci_dev *bridge = bus->self;
  	struct resource *res;
  	struct pci_bus_region region;
  	u32 l;

  	/* Set up the top and bottom of the PCI Memory segment for this bus. */

  	res = bus->resource[1];
  	pcibios_resource_to_bus(bridge, &region, res);
  	if (res->flags & IORESOURCE_MEM) {

  		l = (region.start >> 16) & 0xfff0;
  		l |= region.end & 0xfff00000;

  		dev_info(&bridge->dev, "  bridge window %pR
  ", res);

  	} else {

  		l = 0x0000fff0;

  	}
  	pci_write_config_dword(bridge, PCI_MEMORY_BASE, l);

  }
  
  static void pci_setup_bridge_mmio_pref(struct pci_bus *bus)
  {
  	struct pci_dev *bridge = bus->self;
  	struct resource *res;
  	struct pci_bus_region region;
  	u32 l, bu, lu;

  
  	/* Clear out the upper 32 bits of PREF limit.
  	   If PCI_PREF_BASE_UPPER32 was non-zero, this temporarily
  	   disables PREF range, which is ok. */
  	pci_write_config_dword(bridge, PCI_PREF_LIMIT_UPPER32, 0);
  
  	/* Set up PREF base/limit. */

  	bu = lu = 0;

  	res = bus->resource[2];
  	pcibios_resource_to_bus(bridge, &region, res);
  	if (res->flags & IORESOURCE_PREFETCH) {

  		l = (region.start >> 16) & 0xfff0;
  		l |= region.end & 0xfff00000;

  		if (res->flags & IORESOURCE_MEM_64) {

  			bu = upper_32_bits(region.start);
  			lu = upper_32_bits(region.end);

  		}

  		dev_info(&bridge->dev, "  bridge window %pR
  ", res);

  	} else {

  		l = 0x0000fff0;

  	}
  	pci_write_config_dword(bridge, PCI_PREF_MEMORY_BASE, l);

  	/* Set the upper 32 bits of PREF base & limit. */
  	pci_write_config_dword(bridge, PCI_PREF_BASE_UPPER32, bu);
  	pci_write_config_dword(bridge, PCI_PREF_LIMIT_UPPER32, lu);

  }
  
  static void __pci_setup_bridge(struct pci_bus *bus, unsigned long type)
  {
  	struct pci_dev *bridge = bus->self;

  	dev_info(&bridge->dev, "PCI bridge to [bus %02x-%02x]
  ",
  		 bus->secondary, bus->subordinate);
  
  	if (type & IORESOURCE_IO)
  		pci_setup_bridge_io(bus);
  
  	if (type & IORESOURCE_MEM)
  		pci_setup_bridge_mmio(bus);
  
  	if (type & IORESOURCE_PREFETCH)
  		pci_setup_bridge_mmio_pref(bus);

  
  	pci_write_config_word(bridge, PCI_BRIDGE_CONTROL, bus->bridge_ctl);
  }

  void pci_setup_bridge(struct pci_bus *bus)

  {
  	unsigned long type = IORESOURCE_IO | IORESOURCE_MEM |
  				  IORESOURCE_PREFETCH;
  
  	__pci_setup_bridge(bus, type);
  }

  /* Check whether the bridge supports optional I/O and
     prefetchable memory ranges. If not, the respective
     base/limit registers must be read-only and read as 0. */

  static void pci_bridge_check_ranges(struct pci_bus *bus)

  {
  	u16 io;
  	u32 pmem;
  	struct pci_dev *bridge = bus->self;
  	struct resource *b_res;
  
  	b_res = &bridge->resource[PCI_BRIDGE_RESOURCES];
  	b_res[1].flags |= IORESOURCE_MEM;
  
  	pci_read_config_word(bridge, PCI_IO_BASE, &io);
  	if (!io) {
  		pci_write_config_word(bridge, PCI_IO_BASE, 0xf0f0);
  		pci_read_config_word(bridge, PCI_IO_BASE, &io);
   		pci_write_config_word(bridge, PCI_IO_BASE, 0x0);
   	}
   	if (io)
  		b_res[0].flags |= IORESOURCE_IO;
  	/*  DECchip 21050 pass 2 errata: the bridge may miss an address
  	    disconnect boundary by one PCI data phase.
  	    Workaround: do not use prefetching on this device. */
  	if (bridge->vendor == PCI_VENDOR_ID_DEC && bridge->device == 0x0001)
  		return;
  	pci_read_config_dword(bridge, PCI_PREF_MEMORY_BASE, &pmem);
  	if (!pmem) {
  		pci_write_config_dword(bridge, PCI_PREF_MEMORY_BASE,
  					       0xfff0fff0);
  		pci_read_config_dword(bridge, PCI_PREF_MEMORY_BASE, &pmem);
  		pci_write_config_dword(bridge, PCI_PREF_MEMORY_BASE, 0x0);
  	}

  	if (pmem) {

  		b_res[2].flags |= IORESOURCE_MEM | IORESOURCE_PREFETCH;

  		if ((pmem & PCI_PREF_RANGE_TYPE_MASK) ==
  		    PCI_PREF_RANGE_TYPE_64) {

  			b_res[2].flags |= IORESOURCE_MEM_64;

  			b_res[2].flags |= PCI_PREF_RANGE_TYPE_64;
  		}

  	}
  
  	/* double check if bridge does support 64 bit pref */
  	if (b_res[2].flags & IORESOURCE_MEM_64) {
  		u32 mem_base_hi, tmp;
  		pci_read_config_dword(bridge, PCI_PREF_BASE_UPPER32,
  					 &mem_base_hi);
  		pci_write_config_dword(bridge, PCI_PREF_BASE_UPPER32,
  					       0xffffffff);
  		pci_read_config_dword(bridge, PCI_PREF_BASE_UPPER32, &tmp);
  		if (!tmp)
  			b_res[2].flags &= ~IORESOURCE_MEM_64;
  		pci_write_config_dword(bridge, PCI_PREF_BASE_UPPER32,
  				       mem_base_hi);
  	}

  }
  
  /* Helper function for sizing routines: find first available
     bus resource of a given type. Note: we intentionally skip
     the bus resources which have already been assigned (that is,
     have non-NULL parent resource). */

  static struct resource *find_free_bus_resource(struct pci_bus *bus, unsigned long type)

  {
  	int i;
  	struct resource *r;
  	unsigned long type_mask = IORESOURCE_IO | IORESOURCE_MEM |
  				  IORESOURCE_PREFETCH;

  	pci_bus_for_each_resource(bus, r, i) {

  		if (r == &ioport_resource || r == &iomem_resource)
  			continue;

  		if (r && (r->flags & type_mask) == type && !r->parent)
  			return r;

  	}
  	return NULL;
  }

  static resource_size_t calculate_iosize(resource_size_t size,
  		resource_size_t min_size,
  		resource_size_t size1,
  		resource_size_t old_size,
  		resource_size_t align)
  {
  	if (size < min_size)
  		size = min_size;
  	if (old_size == 1 )
  		old_size = 0;
  	/* To be fixed in 2.5: we should have sort of HAVE_ISA
  	   flag in the struct pci_bus. */
  #if defined(CONFIG_ISA) || defined(CONFIG_EISA)
  	size = (size & 0xff) + ((size & ~0xffUL) << 2);
  #endif
  	size = ALIGN(size + size1, align);
  	if (size < old_size)
  		size = old_size;
  	return size;
  }
  
  static resource_size_t calculate_memsize(resource_size_t size,
  		resource_size_t min_size,
  		resource_size_t size1,
  		resource_size_t old_size,
  		resource_size_t align)
  {
  	if (size < min_size)
  		size = min_size;
  	if (old_size == 1 )
  		old_size = 0;
  	if (size < old_size)
  		size = old_size;
  	size = ALIGN(size + size1, align);
  	return size;
  }

  static resource_size_t get_res_add_size(struct resource_list_x *realloc_head,

  					struct resource *res)
  {
  	struct resource_list_x *list;

  	/* check if it is in realloc_head list */
  	for (list = realloc_head->next; list && list->res != res;

  			list = list->next);
  	if (list)
  		return list->add_size;
  
  	return 0;
  }

  /**
   * pbus_size_io() - size the io window of a given bus
   *
   * @bus : the bus
   * @min_size : the minimum io window that must to be allocated
   * @add_size : additional optional io window

   * @realloc_head : track the additional io window on this list

   *
   * Sizing the IO windows of the PCI-PCI bridge is trivial,
   * since these windows have 4K granularity and the IO ranges
   * of non-bridge PCI devices are limited to 256 bytes.
   * We must be careful with the ISA aliasing though.
   */
  static void pbus_size_io(struct pci_bus *bus, resource_size_t min_size,

  		resource_size_t add_size, struct resource_list_x *realloc_head)

  {
  	struct pci_dev *dev;
  	struct resource *b_res = find_free_bus_resource(bus, IORESOURCE_IO);

  	unsigned long size = 0, size0 = 0, size1 = 0;

  	resource_size_t children_add_size = 0;

  
  	if (!b_res)
   		return;
  
  	list_for_each_entry(dev, &bus->devices, bus_list) {
  		int i;
  
  		for (i = 0; i < PCI_NUM_RESOURCES; i++) {
  			struct resource *r = &dev->resource[i];
  			unsigned long r_size;
  
  			if (r->parent || !(r->flags & IORESOURCE_IO))
  				continue;

  			r_size = resource_size(r);

  
  			if (r_size < 0x400)
  				/* Might be re-aligned for ISA */
  				size += r_size;
  			else
  				size1 += r_size;

  			if (realloc_head)
  				children_add_size += get_res_add_size(realloc_head, r);

  		}
  	}

  	size0 = calculate_iosize(size, min_size, size1,
  			resource_size(b_res), 4096);

  	if (children_add_size > add_size)
  		add_size = children_add_size;

  	size1 = (!realloc_head || (realloc_head && !add_size)) ? size0 :

  		calculate_iosize(size, min_size+add_size, size1,

  			resource_size(b_res), 4096);

  	if (!size0 && !size1) {

  		if (b_res->start || b_res->end)
  			dev_info(&bus->self->dev, "disabling bridge window "
  				 "%pR to [bus %02x-%02x] (unused)
  ", b_res,
  				 bus->secondary, bus->subordinate);

  		b_res->flags = 0;
  		return;
  	}
  	/* Alignment of the IO window is always 4K */
  	b_res->start = 4096;

  	b_res->end = b_res->start + size0 - 1;

  	b_res->flags |= IORESOURCE_STARTALIGN;

  	if (size1 > size0 && realloc_head)
  		add_to_list(realloc_head, bus->self, b_res, size1-size0, 4096);

  }

  /**
   * pbus_size_mem() - size the memory window of a given bus
   *
   * @bus : the bus
   * @min_size : the minimum memory window that must to be allocated
   * @add_size : additional optional memory window

   * @realloc_head : track the additional memory window on this list

   *
   * Calculate the size of the bus and minimal alignment which
   * guarantees that all child resources fit in this size.
   */

  static int pbus_size_mem(struct pci_bus *bus, unsigned long mask,

  			 unsigned long type, resource_size_t min_size,
  			resource_size_t add_size,

  			struct resource_list_x *realloc_head)

  {
  	struct pci_dev *dev;

  	resource_size_t min_align, align, size, size0, size1;

  	resource_size_t aligns[12];	/* Alignments from 1Mb to 2Gb */

  	int order, max_order;
  	struct resource *b_res = find_free_bus_resource(bus, type);

  	unsigned int mem64_mask = 0;

  	resource_size_t children_add_size = 0;

  
  	if (!b_res)
  		return 0;
  
  	memset(aligns, 0, sizeof(aligns));
  	max_order = 0;
  	size = 0;

  	mem64_mask = b_res->flags & IORESOURCE_MEM_64;
  	b_res->flags &= ~IORESOURCE_MEM_64;

  	list_for_each_entry(dev, &bus->devices, bus_list) {
  		int i;

  		for (i = 0; i < PCI_NUM_RESOURCES; i++) {
  			struct resource *r = &dev->resource[i];

  			resource_size_t r_size;

  
  			if (r->parent || (r->flags & mask) != type)
  				continue;

  			r_size = resource_size(r);

  #ifdef CONFIG_PCI_IOV
  			/* put SRIOV requested res to the optional list */

  			if (realloc_head && i >= PCI_IOV_RESOURCES &&

  					i <= PCI_IOV_RESOURCE_END) {
  				r->end = r->start - 1;

  				add_to_list(realloc_head, dev, r, r_size, 0/* dont' care */);

  				children_add_size += r_size;
  				continue;
  			}
  #endif

  			/* For bridges size != alignment */

  			align = pci_resource_alignment(dev, r);

  			order = __ffs(align) - 20;
  			if (order > 11) {

  				dev_warn(&dev->dev, "disabling BAR %d: %pR "
  					 "(bad alignment %#llx)
  ", i, r,
  					 (unsigned long long) align);

  				r->flags = 0;
  				continue;
  			}
  			size += r_size;
  			if (order < 0)
  				order = 0;
  			/* Exclude ranges with size > align from
  			   calculation of the alignment. */
  			if (r_size == align)
  				aligns[order] += align;
  			if (order > max_order)
  				max_order = order;

  			mem64_mask &= r->flags & IORESOURCE_MEM_64;

  			if (realloc_head)
  				children_add_size += get_res_add_size(realloc_head, r);

  		}
  	}

  	align = 0;
  	min_align = 0;
  	for (order = 0; order <= max_order; order++) {

  		resource_size_t align1 = 1;
  
  		align1 <<= (order + 20);

  		if (!align)
  			min_align = align1;

  		else if (ALIGN(align + min_align, min_align) < align1)

  			min_align = align1 >> 1;
  		align += aligns[order];
  	}

  	size0 = calculate_memsize(size, min_size, 0, resource_size(b_res), min_align);

  	if (children_add_size > add_size)
  		add_size = children_add_size;

  	size1 = (!realloc_head || (realloc_head && !add_size)) ? size0 :

  		calculate_memsize(size, min_size+add_size, 0,

  				resource_size(b_res), min_align);

  	if (!size0 && !size1) {

  		if (b_res->start || b_res->end)
  			dev_info(&bus->self->dev, "disabling bridge window "
  				 "%pR to [bus %02x-%02x] (unused)
  ", b_res,
  				 bus->secondary, bus->subordinate);

  		b_res->flags = 0;
  		return 1;
  	}
  	b_res->start = min_align;

  	b_res->end = size0 + min_align - 1;
  	b_res->flags |= IORESOURCE_STARTALIGN | mem64_mask;

  	if (size1 > size0 && realloc_head)
  		add_to_list(realloc_head, bus->self, b_res, size1-size0, min_align);

  	return 1;
  }

  unsigned long pci_cardbus_resource_alignment(struct resource *res)
  {
  	if (res->flags & IORESOURCE_IO)
  		return pci_cardbus_io_size;
  	if (res->flags & IORESOURCE_MEM)
  		return pci_cardbus_mem_size;
  	return 0;
  }
  
  static void pci_bus_size_cardbus(struct pci_bus *bus,

  			struct resource_list_x *realloc_head)

  {
  	struct pci_dev *bridge = bus->self;
  	struct resource *b_res = &bridge->resource[PCI_BRIDGE_RESOURCES];
  	u16 ctrl;
  
  	/*
  	 * Reserve some resources for CardBus.  We reserve
  	 * a fixed amount of bus space for CardBus bridges.
  	 */

  	b_res[0].start = 0;

  	b_res[0].flags |= IORESOURCE_IO | IORESOURCE_SIZEALIGN;

  	if (realloc_head)
  		add_to_list(realloc_head, bridge, b_res, pci_cardbus_io_size, 0 /* dont care */);

  	b_res[1].start = 0;

  	b_res[1].flags |= IORESOURCE_IO | IORESOURCE_SIZEALIGN;

  	if (realloc_head)
  		add_to_list(realloc_head, bridge, b_res+1, pci_cardbus_io_size, 0 /* dont care */);

  
  	/*
  	 * Check whether prefetchable memory is supported
  	 * by this bridge.
  	 */
  	pci_read_config_word(bridge, PCI_CB_BRIDGE_CONTROL, &ctrl);
  	if (!(ctrl & PCI_CB_BRIDGE_CTL_PREFETCH_MEM0)) {
  		ctrl |= PCI_CB_BRIDGE_CTL_PREFETCH_MEM0;
  		pci_write_config_word(bridge, PCI_CB_BRIDGE_CONTROL, ctrl);
  		pci_read_config_word(bridge, PCI_CB_BRIDGE_CONTROL, &ctrl);
  	}
  
  	/*
  	 * If we have prefetchable memory support, allocate
  	 * two regions.  Otherwise, allocate one region of
  	 * twice the size.
  	 */
  	if (ctrl & PCI_CB_BRIDGE_CTL_PREFETCH_MEM0) {

  		b_res[2].start = 0;

  		b_res[2].flags |= IORESOURCE_MEM | IORESOURCE_PREFETCH | IORESOURCE_SIZEALIGN;

  		if (realloc_head)
  			add_to_list(realloc_head, bridge, b_res+2, pci_cardbus_mem_size, 0 /* dont care */);

  		b_res[3].start = 0;

  		b_res[3].flags |= IORESOURCE_MEM | IORESOURCE_SIZEALIGN;

  		if (realloc_head)
  			add_to_list(realloc_head, bridge, b_res+3, pci_cardbus_mem_size, 0 /* dont care */);

  	} else {

  		b_res[3].start = 0;

  		b_res[3].flags |= IORESOURCE_MEM | IORESOURCE_SIZEALIGN;

  		if (realloc_head)
  			add_to_list(realloc_head, bridge, b_res+3, pci_cardbus_mem_size * 2, 0 /* dont care */);

  	}

  
  	/* set the size of the resource to zero, so that the resource does not
  	 * get assigned during required-resource allocation cycle but gets assigned
  	 * during the optional-resource allocation cycle.
   	 */
  	b_res[0].start = b_res[1].start = b_res[2].start = b_res[3].start = 1;
  	b_res[0].end = b_res[1].end = b_res[2].end = b_res[3].end = 0;

  }

  void __ref __pci_bus_size_bridges(struct pci_bus *bus,

  			struct resource_list_x *realloc_head)

  {
  	struct pci_dev *dev;
  	unsigned long mask, prefmask;

  	resource_size_t additional_mem_size = 0, additional_io_size = 0;

  
  	list_for_each_entry(dev, &bus->devices, bus_list) {
  		struct pci_bus *b = dev->subordinate;
  		if (!b)
  			continue;
  
  		switch (dev->class >> 8) {
  		case PCI_CLASS_BRIDGE_CARDBUS:

  			pci_bus_size_cardbus(b, realloc_head);

  			break;
  
  		case PCI_CLASS_BRIDGE_PCI:
  		default:

  			__pci_bus_size_bridges(b, realloc_head);

  			break;
  		}
  	}
  
  	/* The root bus? */
  	if (!bus->self)
  		return;
  
  	switch (bus->self->class >> 8) {
  	case PCI_CLASS_BRIDGE_CARDBUS:
  		/* don't size cardbuses yet. */
  		break;
  
  	case PCI_CLASS_BRIDGE_PCI:
  		pci_bridge_check_ranges(bus);

  		if (bus->self->is_hotplug_bridge) {

  			additional_io_size  = pci_hotplug_io_size;
  			additional_mem_size = pci_hotplug_mem_size;

  		}

  		/*
  		 * Follow thru
  		 */

  	default:

  		pbus_size_io(bus, 0, additional_io_size, realloc_head);

  		/* If the bridge supports prefetchable range, size it
  		   separately. If it doesn't, or its prefetchable window
  		   has already been allocated by arch code, try
  		   non-prefetchable range for both types of PCI memory
  		   resources. */
  		mask = IORESOURCE_MEM;
  		prefmask = IORESOURCE_MEM | IORESOURCE_PREFETCH;

  		if (pbus_size_mem(bus, prefmask, prefmask, 0, additional_mem_size, realloc_head))

  			mask = prefmask; /* Success, size non-prefetch only. */

  		else

  			additional_mem_size += additional_mem_size;

  		pbus_size_mem(bus, mask, IORESOURCE_MEM, 0, additional_mem_size, realloc_head);

  		break;
  	}
  }

  
  void __ref pci_bus_size_bridges(struct pci_bus *bus)
  {
  	__pci_bus_size_bridges(bus, NULL);
  }

  EXPORT_SYMBOL(pci_bus_size_bridges);

  static void __ref __pci_bus_assign_resources(const struct pci_bus *bus,

  					 struct resource_list_x *realloc_head,

  					 struct resource_list_x *fail_head)

  {
  	struct pci_bus *b;
  	struct pci_dev *dev;

  	pbus_assign_resources_sorted(bus, realloc_head, fail_head);

  	list_for_each_entry(dev, &bus->devices, bus_list) {
  		b = dev->subordinate;
  		if (!b)
  			continue;

  		__pci_bus_assign_resources(b, realloc_head, fail_head);

  
  		switch (dev->class >> 8) {
  		case PCI_CLASS_BRIDGE_PCI:

  			if (!pci_is_enabled(dev))
  				pci_setup_bridge(b);

  			break;
  
  		case PCI_CLASS_BRIDGE_CARDBUS:
  			pci_setup_cardbus(b);
  			break;
  
  		default:

  			dev_info(&dev->dev, "not setting up bridge for bus "
  				 "%04x:%02x
  ", pci_domain_nr(b), b->number);

  			break;
  		}
  	}
  }

  
  void __ref pci_bus_assign_resources(const struct pci_bus *bus)
  {

  	__pci_bus_assign_resources(bus, NULL, NULL);

  }

  EXPORT_SYMBOL(pci_bus_assign_resources);

  static void __ref __pci_bridge_assign_resources(const struct pci_dev *bridge,
  					 struct resource_list_x *fail_head)
  {
  	struct pci_bus *b;
  
  	pdev_assign_resources_sorted((struct pci_dev *)bridge, fail_head);
  
  	b = bridge->subordinate;
  	if (!b)
  		return;

  	__pci_bus_assign_resources(b, NULL, fail_head);

  
  	switch (bridge->class >> 8) {
  	case PCI_CLASS_BRIDGE_PCI:
  		pci_setup_bridge(b);
  		break;
  
  	case PCI_CLASS_BRIDGE_CARDBUS:
  		pci_setup_cardbus(b);
  		break;
  
  	default:
  		dev_info(&bridge->dev, "not setting up bridge for bus "
  			 "%04x:%02x
  ", pci_domain_nr(b), b->number);
  		break;
  	}
  }

  static void pci_bridge_release_resources(struct pci_bus *bus,
  					  unsigned long type)
  {
  	int idx;
  	bool changed = false;
  	struct pci_dev *dev;
  	struct resource *r;
  	unsigned long type_mask = IORESOURCE_IO | IORESOURCE_MEM |
  				  IORESOURCE_PREFETCH;
  
  	dev = bus->self;
  	for (idx = PCI_BRIDGE_RESOURCES; idx <= PCI_BRIDGE_RESOURCE_END;
  	     idx++) {
  		r = &dev->resource[idx];
  		if ((r->flags & type_mask) != type)
  			continue;
  		if (!r->parent)
  			continue;
  		/*
  		 * if there are children under that, we should release them
  		 *  all
  		 */
  		release_child_resources(r);
  		if (!release_resource(r)) {
  			dev_printk(KERN_DEBUG, &dev->dev,
  				 "resource %d %pR released
  ", idx, r);
  			/* keep the old size */
  			r->end = resource_size(r) - 1;
  			r->start = 0;
  			r->flags = 0;
  			changed = true;
  		}
  	}
  
  	if (changed) {
  		/* avoiding touch the one without PREF */
  		if (type & IORESOURCE_PREFETCH)
  			type = IORESOURCE_PREFETCH;
  		__pci_setup_bridge(bus, type);
  	}
  }
  
  enum release_type {
  	leaf_only,
  	whole_subtree,
  };
  /*
   * try to release pci bridge resources that is from leaf bridge,
   * so we can allocate big new one later
   */
  static void __ref pci_bus_release_bridge_resources(struct pci_bus *bus,
  						   unsigned long type,
  						   enum release_type rel_type)
  {
  	struct pci_dev *dev;
  	bool is_leaf_bridge = true;
  
  	list_for_each_entry(dev, &bus->devices, bus_list) {
  		struct pci_bus *b = dev->subordinate;
  		if (!b)
  			continue;
  
  		is_leaf_bridge = false;
  
  		if ((dev->class >> 8) != PCI_CLASS_BRIDGE_PCI)
  			continue;
  
  		if (rel_type == whole_subtree)
  			pci_bus_release_bridge_resources(b, type,
  						 whole_subtree);
  	}
  
  	if (pci_is_root_bus(bus))
  		return;
  
  	if ((bus->self->class >> 8) != PCI_CLASS_BRIDGE_PCI)
  		return;
  
  	if ((rel_type == whole_subtree) || is_leaf_bridge)
  		pci_bridge_release_resources(bus, type);
  }

  static void pci_bus_dump_res(struct pci_bus *bus)
  {

  	struct resource *res;
  	int i;

  	pci_bus_for_each_resource(bus, res, i) {

  		if (!res || !res->end || !res->flags)

                          continue;

  		dev_printk(KERN_DEBUG, &bus->dev, "resource %d %pR
  ", i, res);

          }
  }
  
  static void pci_bus_dump_resources(struct pci_bus *bus)
  {
  	struct pci_bus *b;
  	struct pci_dev *dev;
  
  
  	pci_bus_dump_res(bus);
  
  	list_for_each_entry(dev, &bus->devices, bus_list) {
  		b = dev->subordinate;
  		if (!b)
  			continue;
  
  		pci_bus_dump_resources(b);
  	}
  }

  static int __init pci_bus_get_depth(struct pci_bus *bus)
  {
  	int depth = 0;
  	struct pci_dev *dev;
  
  	list_for_each_entry(dev, &bus->devices, bus_list) {
  		int ret;
  		struct pci_bus *b = dev->subordinate;
  		if (!b)
  			continue;
  
  		ret = pci_bus_get_depth(b);
  		if (ret + 1 > depth)
  			depth = ret + 1;
  	}
  
  	return depth;
  }
  static int __init pci_get_max_depth(void)
  {
  	int depth = 0;
  	struct pci_bus *bus;
  
  	list_for_each_entry(bus, &pci_root_buses, node) {
  		int ret;
  
  		ret = pci_bus_get_depth(bus);
  		if (ret > depth)
  			depth = ret;
  	}
  
  	return depth;
  }

  /*
   * first try will not touch pci bridge res
   * second  and later try will clear small leaf bridge res
   * will stop till to the max  deepth if can not find good one
   */

  void __init
  pci_assign_unassigned_resources(void)
  {
  	struct pci_bus *bus;

  	struct resource_list_x realloc_list; /* list of resources that

  					want additional resources */

  	int tried_times = 0;
  	enum release_type rel_type = leaf_only;
  	struct resource_list_x head, *list;
  	unsigned long type_mask = IORESOURCE_IO | IORESOURCE_MEM |
  				  IORESOURCE_PREFETCH;
  	unsigned long failed_type;
  	int max_depth = pci_get_max_depth();
  	int pci_try_num;
  
  
  	head.next = NULL;

  	realloc_list.next = NULL;

  
  	pci_try_num = max_depth + 1;
  	printk(KERN_DEBUG "PCI: max bus depth: %d pci_try_num: %d
  ",
  		 max_depth, pci_try_num);
  
  again:

  	/* Depth first, calculate sizes and alignments of all
  	   subordinate buses. */

  	list_for_each_entry(bus, &pci_root_buses, node)

  		__pci_bus_size_bridges(bus, &realloc_list);

  	/* Depth last, allocate resources and update the hardware. */

  	list_for_each_entry(bus, &pci_root_buses, node)

  		__pci_bus_assign_resources(bus, &realloc_list, &head);
  	BUG_ON(realloc_list.next);

  	tried_times++;
  
  	/* any device complain? */
  	if (!head.next)
  		goto enable_and_dump;

  
  	/* don't realloc if asked to do so */
  	if (!pci_realloc_enabled()) {
  		free_list(resource_list_x, &head);
  		goto enable_and_dump;
  	}

  	failed_type = 0;
  	for (list = head.next; list;) {
  		failed_type |= list->flags;
  		list = list->next;
  	}
  	/*
  	 * io port are tight, don't try extra
  	 * or if reach the limit, don't want to try more
  	 */
  	failed_type &= type_mask;
  	if ((failed_type == IORESOURCE_IO) || (tried_times >= pci_try_num)) {
  		free_list(resource_list_x, &head);
  		goto enable_and_dump;
  	}
  
  	printk(KERN_DEBUG "PCI: No. %d try to assign unassigned res
  ",
  			 tried_times + 1);
  
  	/* third times and later will not check if it is leaf */
  	if ((tried_times + 1) > 2)
  		rel_type = whole_subtree;
  
  	/*
  	 * Try to release leaf bridge's resources that doesn't fit resource of
  	 * child device under that bridge
  	 */
  	for (list = head.next; list;) {
  		bus = list->dev->bus;
  		pci_bus_release_bridge_resources(bus, list->flags & type_mask,
  						  rel_type);
  		list = list->next;
  	}
  	/* restore size and flags */
  	for (list = head.next; list;) {
  		struct resource *res = list->res;
  
  		res->start = list->start;
  		res->end = list->end;
  		res->flags = list->flags;
  		if (list->dev->subordinate)
  			res->flags = 0;
  
  		list = list->next;
  	}
  	free_list(resource_list_x, &head);
  
  	goto again;
  
  enable_and_dump:
  	/* Depth last, update the hardware. */
  	list_for_each_entry(bus, &pci_root_buses, node)
  		pci_enable_bridges(bus);

  
  	/* dump the resource on buses */

  	list_for_each_entry(bus, &pci_root_buses, node)

  		pci_bus_dump_resources(bus);

  }

  
  void pci_assign_unassigned_bridge_resources(struct pci_dev *bridge)
  {
  	struct pci_bus *parent = bridge->subordinate;

  	int tried_times = 0;
  	struct resource_list_x head, *list;

  	int retval;

  	unsigned long type_mask = IORESOURCE_IO | IORESOURCE_MEM |
  				  IORESOURCE_PREFETCH;
  
  	head.next = NULL;

  again:

  	pci_bus_size_bridges(parent);

  	__pci_bridge_assign_resources(bridge, &head);

  
  	tried_times++;
  
  	if (!head.next)

  		goto enable_all;

  
  	if (tried_times >= 2) {
  		/* still fail, don't need to try more */

  		free_list(resource_list_x, &head);

  		goto enable_all;

  	}
  
  	printk(KERN_DEBUG "PCI: No. %d try to assign unassigned res
  ",
  			 tried_times + 1);
  
  	/*
  	 * Try to release leaf bridge's resources that doesn't fit resource of
  	 * child device under that bridge
  	 */
  	for (list = head.next; list;) {
  		struct pci_bus *bus = list->dev->bus;
  		unsigned long flags = list->flags;
  
  		pci_bus_release_bridge_resources(bus, flags & type_mask,
  						 whole_subtree);
  		list = list->next;
  	}
  	/* restore size and flags */
  	for (list = head.next; list;) {
  		struct resource *res = list->res;
  
  		res->start = list->start;
  		res->end = list->end;
  		res->flags = list->flags;
  		if (list->dev->subordinate)
  			res->flags = 0;
  
  		list = list->next;
  	}

  	free_list(resource_list_x, &head);

  
  	goto again;

  
  enable_all:
  	retval = pci_reenable_device(bridge);
  	pci_set_master(bridge);
  	pci_enable_bridges(parent);

  }
  EXPORT_SYMBOL_GPL(pci_assign_unassigned_bridge_resources);