[Skiboot] [PATCH v2 1/7] libflash: Return a pointer to internal structure.

[Cyril Bur]

There are some functions (notably: erase_chip and set_4b_mode) which cannot
be abstracted away by the blocklevel interface, this means that if they are
really needed (by pflash for example) then at the moment a program like
pflash would need to pass a blocklevel_device struct to libflash.

This forces libflash to trust that it was given a blocklevel that it did
init. If it didn't init it the container_of call will return junk and
libflash has no way to protect its self. This method (while very useful)
has destroyed type safety. As such, this commit reintroduces some
typesafety back into this stack.

Signed-off-by: Cyril Bur <cyril.bur at au1.ibm.com>
---
 libflash/libflash.c        | 6 +++++-
 libflash/libflash.h        | 3 ++-
 libflash/test/test-flash.c | 2 +-
 platforms/astbmc/pnor.c    | 2 +-
 platforms/rhesus/rhesus.c  | 2 +-
 5 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/libflash/libflash.c b/libflash/libflash.c
index 90ae653..831bdd4 100644
--- a/libflash/libflash.c
+++ b/libflash/libflash.c
@@ -798,7 +798,8 @@ static int flash_get_info(struct blocklevel_device *bl, const char **name,
 	return 0;
 }
 
-int flash_init(struct spi_flash_ctrl *ctrl, struct blocklevel_device **bl)
+int flash_init(struct spi_flash_ctrl *ctrl, struct blocklevel_device **bl,
+		struct flash_chip **flash_chip)
 {
 	struct flash_chip *c;
 	int rc;
@@ -842,6 +843,9 @@ bail:
 	c->bl.flags = WRITE_NEED_ERASE;
 
 	*bl = &(c->bl);
+	if (flash_chip)
+		*flash_chip = c;
+
 	return 0;
 }
 
diff --git a/libflash/libflash.h b/libflash/libflash.h
index f3973ab..fa5143a 100644
--- a/libflash/libflash.h
+++ b/libflash/libflash.h
@@ -45,7 +45,8 @@ extern bool libflash_debug;
 struct flash_chip;
 struct spi_flash_ctrl;
 
-int flash_init(struct spi_flash_ctrl *ctrl, struct blocklevel_device **bl);
+int flash_init(struct spi_flash_ctrl *ctrl, struct blocklevel_device **bl,
+		struct flash_chip **flash_chip);
 void flash_exit(struct blocklevel_device *bl);
 
 /* libflash sets the 4b mode automatically based on the flash
diff --git a/libflash/test/test-flash.c b/libflash/test/test-flash.c
index cf2b3a5..c93aef5 100644
--- a/libflash/test/test-flash.c
+++ b/libflash/test/test-flash.c
@@ -378,7 +378,7 @@ int main(void)
 	memset(sim_image, 0xff, sim_image_sz);
 	test = malloc(0x10000 * 2);
 
-	rc = flash_init(&sim_ctrl, &bl);
+	rc = flash_init(&sim_ctrl, &bl, NULL);
 	if (rc) {
 		ERR("flash_init failed with err %d\n", rc);
 		exit(1);
diff --git a/platforms/astbmc/pnor.c b/platforms/astbmc/pnor.c
index 2cd57c6..74353ea 100644
--- a/platforms/astbmc/pnor.c
+++ b/platforms/astbmc/pnor.c
@@ -46,7 +46,7 @@ int pnor_init(void)
 		goto fail;
 	}
 
-	rc = flash_init(pnor_ctrl, &bl);
+	rc = flash_init(pnor_ctrl, &bl, NULL);
 	if (rc) {
 		prerror("PLAT: Failed to open init PNOR driver\n");
 		goto fail;
diff --git a/platforms/rhesus/rhesus.c b/platforms/rhesus/rhesus.c
index b597ba0..2131ec0 100644
--- a/platforms/rhesus/rhesus.c
+++ b/platforms/rhesus/rhesus.c
@@ -129,7 +129,7 @@ static int rhesus_pnor_init(void)
 		prerror("PLAT: Failed to open PNOR flash controller\n");
 		goto fail;
 	}
-	rc = flash_init(pnor_ctrl, &bl);
+	rc = flash_init(pnor_ctrl, &bl, NULL);
 	if (rc) {
 		prerror("PLAT: Failed to open init PNOR driver\n");
 		goto fail;
-- 
2.5.0