// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * ts5500_flash.c -- MTD map driver for Technology Systems TS-5500 board
 *
 * Copyright (C) 2004 Sean Young <sean@mess.org>
 *
 * Note:
 * - In order for detection to work, jumper 3 must be set.
 * - Drive A and B use the resident flash disk (RFD) flash translation layer.
 * - If you have created your own jffs file system and the bios overwrites
 *   it during boot, try disabling Drive A: and B: in the boot order.
 */

#include <linux/init.h>
#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/mtd/map.h>
#include <linux/mtd/mtd.h>
#include <linux/mtd/partitions.h>
#include <linux/types.h>


#define WINDOW_ADDR 0x09400000
#define WINDOW_SIZE 0x00200000

static struct map_info ts5500_map = {
 .name = "TS-5500 Flash",
 .size = WINDOW_SIZE,
 .bankwidth = 1,
 .phys = WINDOW_ADDR
};

static const struct mtd_partition ts5500_partitions[] = {
 {
  .name = "Drive A",
  .offset = 0,
  .size = 0x0e0000
 },
 {
  .name = "BIOS",
  .offset = 0x0e0000,
  .size = 0x020000,
 },
 {
  .name = "Drive B",
  .offset = 0x100000,
  .size = 0x100000
 }
};

#define NUM_PARTITIONS ARRAY_SIZE(ts5500_partitions)

static struct mtd_info *mymtd;

static int __init init_ts5500_map(void)
{
 int rc = 0;

 ts5500_map.virt = ioremap(ts5500_map.phys, ts5500_map.size);

 if (!ts5500_map.virt) {
  printk(KERN_ERR "Failed to ioremap\n");
  rc = -EIO;
  goto err2;
 }

 simple_map_init(&ts5500_map);

 mymtd = do_map_probe("jedec_probe", &ts5500_map);
 if (!mymtd)
  mymtd = do_map_probe("map_rom", &ts5500_map);

 if (!mymtd) {
  rc = -ENXIO;
  goto err1;
 }

 mymtd->owner = THIS_MODULE;
 mtd_device_register(mymtd, ts5500_partitions, NUM_PARTITIONS);

 return 0;

err1:
 iounmap(ts5500_map.virt);
err2:
 return rc;
}

static void __exit cleanup_ts5500_map(void)
{
 if (mymtd) {
  mtd_device_unregister(mymtd);
  map_destroy(mymtd);
 }

 if (ts5500_map.virt) {
  iounmap(ts5500_map.virt);
  ts5500_map.virt = NULL;
 }
}

module_init(init_ts5500_map);
module_exit(cleanup_ts5500_map);

MODULE_LICENSE("GPL");
MODULE_AUTHOR("Sean Young <sean@mess.org>");
MODULE_DESCRIPTION("MTD map driver for Technology Systems TS-5500 board");