device_romstage: Add a way to move to the next device

When trying to loop through all the devices in romstage, there was
no function to just go from one to the next.

This allows an easy way to go all the way down the chain of devices.

Change-Id: Id205b24610d75de060b0d48fa283a2ab92d1df0a
Signed-off-by: Martin Roth <gaumless@gmail.com>
Reviewed-on: http://review.coreboot.org/5732
Tested-by: build bot (Jenkins)
Reviewed-by: Marc Jones <marc.jones@se-eng.com>
Reviewed-by: Paul Menzel <paulepanter@users.sourceforge.net>
diff --git a/src/device/device_romstage.c b/src/device/device_romstage.c
index 591cf88..6dcf6d2 100644
--- a/src/device/device_romstage.c
+++ b/src/device/device_romstage.c
@@ -55,6 +55,30 @@
 }
 
 /**
+ * Given a device pointer, find the next PCI device.
+ *
+ * @param previous_dev A pointer to a PCI device structure.
+ * @return Pointer to the next device structure (if found), 0 otherwise.
+ */
+ROMSTAGE_CONST struct device *dev_find_next_pci_device(
+		ROMSTAGE_CONST struct device *previous_dev)
+{
+	ROMSTAGE_CONST struct device *dev, *result;
+
+	if (previous_dev == NULL)
+		previous_dev = all_devices;
+
+	result = 0;
+	for (dev = previous_dev->next; dev; dev = dev->next) {
+		if (dev->path.type == DEVICE_PATH_PCI) {
+			result = dev;
+			break;
+		}
+	}
+	return result;
+}
+
+/**
  * Given an SMBus bus and a device number, find the device structure.
  *
  * @param bus The bus number.
diff --git a/src/include/device/device.h b/src/include/device/device.h
index a4ef456..dcd93f6 100644
--- a/src/include/device/device.h
+++ b/src/include/device/device.h
@@ -236,6 +236,8 @@
 
 ROMSTAGE_CONST struct device * dev_find_slot (unsigned int bus,
 						unsigned int devfn);
+ROMSTAGE_CONST struct device *dev_find_next_pci_device(
+						ROMSTAGE_CONST struct device *previous_dev);
 ROMSTAGE_CONST struct device * dev_find_slot_on_smbus (unsigned int bus,
 							unsigned int addr);