ata blockdev is an universal child of a blockctrl device, be it (P/S)ATA or SCSI
partition blockdev is a child of other blockdev drivers

Signed-off-by: Pavel Herrmann <[email protected]>
---
 drivers/blockdev/Makefile    |   2 +-
 drivers/blockdev/ata.c       | 234 +++++++++++++++++++++++++++++++++++++++++++
 drivers/blockdev/ata.h       |  37 +++++++
 drivers/blockdev/partition.c | 179 +++++++++++++++++++++++++++++++++
 4 files changed, 451 insertions(+), 1 deletion(-)
 create mode 100644 drivers/blockdev/ata.c
 create mode 100644 drivers/blockdev/ata.h
 create mode 100644 drivers/blockdev/partition.c

diff --git a/drivers/blockdev/Makefile b/drivers/blockdev/Makefile
index a988924..668ffbd 100644
--- a/drivers/blockdev/Makefile
+++ b/drivers/blockdev/Makefile
@@ -21,7 +21,7 @@ include $(TOPDIR)/config.mk
 
 LIB    := $(obj)libblockdev.o
 
-COBJS-${CONFIG_DM_BLOCK} := core.o
+COBJS-${CONFIG_DM_BLOCK} := core.o ata.o partition.o
 COBJS-${CONFIG_DOS_PARTITION} += part_types/part_dos.o
 
 COBJS  := $(COBJS-y)
diff --git a/drivers/blockdev/ata.c b/drivers/blockdev/ata.c
new file mode 100644
index 0000000..698dc18
--- /dev/null
+++ b/drivers/blockdev/ata.c
@@ -0,0 +1,234 @@
+/*
+ * (C) Copyright 2012
+ * Pavel Herrmann <[email protected]>
+ *
+ * See file CREDITS for list of people who contributed to this
+ * project.
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License as
+ * published by the Free Software Foundation; either version 2 of
+ * the License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * 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., 59 Temple Place, Suite 330, Boston,
+ * MA 02111-1307 USA
+ */
+
+#include "ata.h"
+#include <dm/blockctrl.h>
+#include <dm/manager.h>
+#include <common.h>
+#include <malloc.h>
+
+static lbaint_t read(struct instance *dev, lbaint_t start, lbaint_t blkcnt,
+               void *buffer)
+{
+       if (!buffer)
+               return 0;
+
+       struct blockdev_ata_platform_data *platform = dev->info->platform_data;
+       struct blockdev_ata_private_data *priv = dev->private_data;
+       /* check for access beyond disk boundary */
+       if ((start + blkcnt) > priv->block_count)
+               blkcnt = (priv->block_count - start);
+
+       return blockctrl_read(dev->bus, platform->port_number,
+               start, blkcnt, buffer);
+}
+
+static lbaint_t write(struct instance *dev, lbaint_t start, lbaint_t blkcnt,
+               void *buffer)
+{
+       if (!buffer)
+               return 0;
+
+       struct blockdev_ata_platform_data *platform = dev->info->platform_data;
+       struct blockdev_ata_private_data *priv = dev->private_data;
+       /* check for access beyond disk boundary */
+       if ((start + blkcnt) > priv->block_count)
+               blkcnt = (priv->block_count - start);
+
+       return blockctrl_write(dev->bus, platform->port_number,
+               start, blkcnt, buffer);
+}
+
+static lbaint_t erase(struct instance *dev, lbaint_t start, lbaint_t blkcnt)
+{
+       /* TRIM is not supported on ATA, so we just pretend to do something */
+       return blkcnt;
+}
+
+static int get_option(struct instance *dev, enum blockdev_option_code op,
+               struct option *result)
+{
+       struct blockdev_ata_private_data *priv = dev->private_data;
+       struct blockdev_ata_platform_data *plat = dev->info->platform_data;
+       int port = plat->port_number;
+
+       if (!result || !priv)
+               return -EINVAL;
+
+       switch (op) {
+       case BLKD_OPT_IFTYPE:
+               result->flags = OPTION_TYPE_U;
+               result->data.data_u = BLOCKDEV_IFTYPE_ATA;
+               return 0;
+       case BLKD_OPT_TYPE:
+               result->flags = OPTION_TYPE_U;
+               result->data.data_u = BLOCKDEV_TYPE_HARDDISK;
+               return 0;
+       case BLKD_OPT_OFFSET:
+               result->flags = OPTION_TYPE_U;
+               result->data.data_u = 0;
+               return 0;
+       case BLKD_OPT_BLOCKSIZE:
+               result->flags = OPTION_TYPE_U;
+               result->data.data_u = priv->block_size;
+               return 0;
+       case BLKD_OPT_BLOCKCOUNT:
+               result->flags = OPTION_TYPE_U;
+               result->data.data_u = priv->block_count;
+               return 0;
+       case BLKD_OPT_VENDOR:
+               result->flags = OPTION_TYPE_S;
+               result->data.data_s = priv->vendor;
+               return 0;
+       case BLKD_OPT_PRODUCT:
+               result->flags = OPTION_TYPE_S;
+               result->data.data_s = priv->product;
+               return 0;
+       case BLKD_OPT_REVISION:
+               result->flags = OPTION_TYPE_S;
+               result->data.data_s = priv->revision;
+               return 0;
+       case BLKD_OPT_LBA48:
+               return blockctrl_get_port_option(dev->bus, port,
+                       BLKP_OPT_LBA48, result);
+       default:
+               break;
+               /* TODO: more options */
+
+       }
+       return -EINVAL;
+}
+
+static int set_option(struct instance *dev, enum blockdev_option_code op,
+               struct option *value)
+{
+       /* there are currently no meaningful options to set */
+       return -EINVAL;
+}
+
+static struct blockdev_ops ops = {
+       .read = read,
+       .write = write,
+       .erase = erase,
+       .get_option = get_option,
+       .set_option = set_option,
+};
+
+
+static int bind(struct instance *dev)
+{
+       /* prepare hint for the core to properly identify this as a ATA disk */
+       struct blockdev_core_hint hint = {
+               .iftype = BLOCKDEV_IFTYPE_ATA
+       };
+       core_bind(CORE_BLOCKDEV, dev, &ops, &hint);
+       return 0;
+}
+
+static int probe(struct instance *dev)
+{
+       struct option op;
+       struct blockdev_ata_private_data *priv;
+       int error;
+       struct blockdev_ata_platform_data *plat = dev->info->platform_data;
+       int port = plat->port_number;
+
+       priv = calloc(sizeof(*priv), 1);
+       if (!priv)
+               return -ENOMEM;
+
+       error = blockctrl_get_port_option(dev->bus, port, BLKP_OPT_BLOCKSIZE,
+                       &op);
+       if (error || (OPTION_TYPE(op) != OPTION_TYPE_U))
+               return -EINVAL;
+       priv->block_size = op.data.data_u;
+
+       error = blockctrl_get_port_option(dev->bus, port, BLKP_OPT_BLOCKCOUNT,
+                       &op);
+       if (error || (OPTION_TYPE(op) != OPTION_TYPE_U))
+               return -EINVAL;
+       priv->block_count = op.data.data_u;
+
+       error = blockctrl_get_port_option(dev->bus, port, BLKP_OPT_VENDOR, &op);
+       if (error || (OPTION_TYPE(op) != OPTION_TYPE_S))
+               return -EINVAL;
+       strncpy(priv->vendor, op.data.data_s, 40);
+       priv->vendor[40] = 0;
+       if (op.flags & OPTION_PTR_MALLOCED)
+               free(op.data.data_s);
+
+       error = blockctrl_get_port_option(dev->bus, port, BLKP_OPT_PRODUCT,
+                       &op);
+       if (error || (OPTION_TYPE(op) != OPTION_TYPE_S))
+               return -EINVAL;
+       strncpy(priv->product, op.data.data_s, 20);
+       priv->vendor[20] = 0;
+       if (op.flags & OPTION_PTR_MALLOCED)
+               free(op.data.data_s);
+
+       error = blockctrl_get_port_option(dev->bus, port, BLKP_OPT_REVISION,
+                       &op);
+       if (error || (OPTION_TYPE(op) != OPTION_TYPE_S))
+               return -EINVAL;
+       strncpy(priv->revision, op.data.data_s, 8);
+       priv->revision[8] = 0;
+       if (op.flags & OPTION_PTR_MALLOCED)
+               free(op.data.data_s);
+
+       dev->private_data = priv;
+
+       return 0;
+}
+
+static int reloc(struct instance *dev, struct instance *old)
+{
+       core_replace(CORE_BLOCKDEV, dev, old);
+       dev->private_data = malloc(sizeof(struct blockdev_ata_private_data));
+       if (!dev->private_data)
+               return -ENOMEM;
+
+       memcpy(dev->private_data, old->private_data,
+               sizeof(struct blockdev_ata_private_data));
+
+       return 0;
+}
+
+static int remove(struct instance *dev)
+{
+       /* remove our metadata cache */
+       free(dev->private_data);
+       return 0;
+}
+
+static int unbind(struct instance *dev)
+{
+       core_unbind(CORE_BLOCKDEV, dev);
+       return 0;
+}
+
+U_BOOT_DRIVER(blockdev_ata,
+       bind,
+       probe,
+       reloc,
+       remove,
+       unbind);
diff --git a/drivers/blockdev/ata.h b/drivers/blockdev/ata.h
new file mode 100644
index 0000000..da63b5d
--- /dev/null
+++ b/drivers/blockdev/ata.h
@@ -0,0 +1,37 @@
+/*
+ * (C) Copyright 2012
+ * Pavel Herrmann <[email protected]>
+ *
+ * See file CREDITS for list of people who contributed to this
+ * project.
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License as
+ * published by the Free Software Foundation; either version 2 of
+ * the License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * 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., 59 Temple Place, Suite 330, Boston,
+ * MA 02111-1307 USA
+ */
+
+#ifndef _BLOCKDEV_ATA_H_
+#define _BLOCKDEV_ATA_H_
+#include <dm/blockdev.h>
+
+struct blockdev_ata_private_data {
+       /* cache for get_option() */
+       lbaint_t block_size;
+       lbaint_t block_count;
+       char vendor[40+1];
+       char product[20+1];
+       char revision[8+1];
+};
+
+#endif
diff --git a/drivers/blockdev/partition.c b/drivers/blockdev/partition.c
new file mode 100644
index 0000000..c5c0f48
--- /dev/null
+++ b/drivers/blockdev/partition.c
@@ -0,0 +1,179 @@
+/*
+ * (C) Copyright 2012
+ * Pavel Herrmann <[email protected]>
+ *
+ * See file CREDITS for list of people who contributed to this
+ * project.
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License as
+ * published by the Free Software Foundation; either version 2 of
+ * the License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * 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., 59 Temple Place, Suite 330, Boston,
+ * MA 02111-1307 USA
+ */
+
+#include <dm/blockdev.h>
+#include <dm/manager.h>
+
+static lbaint_t read(struct instance *dev, lbaint_t start, lbaint_t blkcnt,
+               void *buffer)
+{
+       if (!buffer)
+               return 0;
+
+       struct blockdev_partition_platform_data *platform =
+               dev->info->platform_data;
+       /* check for access beyond partition boundary */
+       if ((start + blkcnt) > platform->block_count)
+               blkcnt = (platform->block_count - start);
+
+       start += platform->offset;
+
+       return blockdev_read(dev->bus, start, blkcnt, buffer);
+}
+
+static lbaint_t write(struct instance *dev, lbaint_t start, lbaint_t blkcnt,
+               void *buffer)
+{
+       if (!buffer)
+               return 0;
+
+       struct blockdev_partition_platform_data *platform =
+               dev->info->platform_data;
+       /* check for access beyond partition boundary */
+       if ((start + blkcnt) > platform->block_count)
+               blkcnt = (platform->block_count - start);
+
+       start += platform->offset;
+
+       return blockdev_write(dev->bus, start, blkcnt, buffer);
+}
+
+static lbaint_t erase(struct instance *dev, lbaint_t start, lbaint_t blkcnt)
+{
+       struct blockdev_partition_platform_data *platform =
+               dev->info->platform_data;
+       /* check for access beyond partition boundary */
+       if ((start + blkcnt) > platform->block_count)
+               blkcnt = (platform->block_count - start);
+
+       start += platform->offset;
+
+       return blockdev_erase(dev->bus, start, blkcnt);
+}
+
+static int get_option(struct instance *dev, enum blockdev_option_code op,
+               struct option *result)
+{
+       if (!result)
+               return -EINVAL;
+
+       struct blockdev_partition_platform_data *platform =
+               dev->info->platform_data;
+
+       switch (op) {
+       case BLKD_OPT_TYPE:
+               result->flags = OPTION_TYPE_U;
+               result->data.data_u = BLOCKDEV_TYPE_PARTITION;
+               return 0;
+       case BLKD_OPT_BLOCKCOUNT:
+               result->flags = OPTION_TYPE_U;
+               result->data.data_u = platform->block_count;
+               return 0;
+       case BLKD_OPT_OFFSET:
+               result->flags = OPTION_TYPE_U;
+               result->data.data_u = platform->offset;
+               return 0;
+       case BLKD_OPT_IFTYPE:
+       case BLKD_OPT_BLOCKSIZE:
+       case BLKD_OPT_REMOVABLE:
+       case BLKD_OPT_LBA48:
+       case BLKD_OPT_VENDOR:
+       case BLKD_OPT_PRODUCT:
+       case BLKD_OPT_REVISION:
+       case BLKD_OPT_SCSILUN:
+       case BLKD_OPT_SCSITARGET:
+               return blockdev_get_option(dev->bus, op, result);
+
+       default:
+               break;
+       }
+
+       return -EINVAL;
+}
+
+
+static int set_option(struct instance *dev, enum blockdev_option_code op,
+               struct option *value)
+{
+       /* there are currently no meaningful options to set */
+       return -EINVAL;
+}
+
+static struct blockdev_ops ops = {
+       .read = read,
+       .write = write,
+       .erase = erase,
+       .get_option = get_option,
+       .set_option = set_option
+};
+
+
+static int bind(struct instance *dev)
+{
+       /* prepare hint for the core to properly identify this as a partition */
+       struct blockdev_partition_platform_data *platform =
+               dev->info->platform_data;
+
+       struct blockdev_core_hint hint = {
+               .iftype = BLOCKDEV_IFTYPE_PARTITION,
+               .part_number = platform->part_number
+       };
+       core_bind(CORE_BLOCKDEV, dev, &ops, &hint);
+
+       return 0;
+}
+
+static int probe(struct instance *dev)
+{
+       /* there is nothing to activate */
+       return 0;
+}
+
+static int reloc(struct instance *dev, struct instance *old)
+{
+       core_replace(CORE_BLOCKDEV, dev, old);
+       /* there are no private data to relocate */
+
+       return 0;
+}
+
+static int remove(struct instance *dev)
+{
+       /* nothing to do here */
+       return 0;
+}
+
+static int unbind(struct instance *dev)
+{
+       core_unbind(CORE_BLOCKDEV, dev);
+
+       return 0;
+}
+
+
+U_BOOT_DRIVER(blockdev_partition,
+       bind,
+       probe,
+       reloc,
+       remove,
+       unbind);
-- 
1.7.12

_______________________________________________
U-Boot mailing list
[email protected]
http://lists.denx.de/mailman/listinfo/u-boot

Reply via email to