/* i2c-core.c - a device driver for the iic-bus interface		     */
/* ------------------------------------------------------------------------- */
/*   Copyright (C) 1995-99 Simon G. Vogl

    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., 675 Mass Ave, Cambridge, MA 02139, USA.		     */
/* ------------------------------------------------------------------------- */

/* With some changes from Kyösti Mälkki <kmalkki@cc.hut.fi>.
   All SMBus-related things are written by Frodo Looijaard <frodol@dds.nl>
   SMBus 2.0 support by Mark Studebaker <mdsxyz123@yahoo.com>                */

#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/errno.h>
#include <linux/slab.h>
#include <linux/i2c.h>
#include <linux/init.h>
#include <linux/idr.h>
#include <linux/seq_file.h>
#include <asm/uaccess.h>


static LIST_HEAD(adapters);
static LIST_HEAD(drivers);
static DECLARE_MUTEX(core_lists);
static DEFINE_IDR(i2c_adapter_idr);

static int i2c_device_match(struct device *dev, struct device_driver *drv)
  {
  return 1;
  }

static int i2c_bus_suspend(struct device * dev, pm_message_t state)
  {
  int rc = 0;

  if (dev->driver && dev->driver->suspend)
    rc = dev->driver->suspend(dev, state, 0);

  return rc;
  }

static int i2c_bus_resume(struct device * dev)
  {
  int rc = 0;

  if (dev->driver && dev->driver->resume)
    rc = dev->driver->resume(dev, 0);

  return rc;
  }

struct bus_type i2c_bus_type = {
  .name =   "i2c",
  .match =  i2c_device_match,
  .suspend =      i2c_bus_suspend,
  .resume =       i2c_bus_resume,
  };

static int i2c_device_probe(struct device *dev)
  {
  return -ENODEV;
  }

static int i2c_device_remove(struct device *dev)
  {
  return 0;
  }

void i2c_adapter_dev_release(struct device *dev)
  {
  struct i2c_adapter *adap = dev_to_i2c_adapter(dev);

  complete(&adap->dev_released);
  }

struct device_driver i2c_adapter_driver = {
  .name = "i2c_adapter",
  .bus = &i2c_bus_type,
  .probe = i2c_device_probe,
  .remove = i2c_device_remove,
  };

static void i2c_adapter_class_dev_release(struct class_device *dev)
  {
  struct i2c_adapter *adap = class_dev_to_i2c_adapter(dev);

  complete(&adap->class_dev_released);
  }

struct class i2c_adapter_class = {
  .name =   "i2c-adapter",
  .release =  &i2c_adapter_class_dev_release,
  };

static ssize_t show_adapter_name(struct device *dev, struct device_attribute *attr, char *buf)
  {
  struct i2c_adapter *adap = dev_to_i2c_adapter(dev);

  return sprintf(buf, "%s\n", adap->name);
  }

static DEVICE_ATTR(name, S_IRUGO, show_adapter_name, NULL);


static void i2c_client_release(struct device *dev)
  {
  struct i2c_client *client = to_i2c_client(dev);

  complete(&client->released);
  }

static ssize_t show_client_name(struct device *dev, struct device_attribute *attr, char *buf)
  {
  struct i2c_client *client = to_i2c_client(dev);

  return sprintf(buf, "%s\n", client->name);
  }

/*
 * We can't use the DEVICE_ATTR() macro here as we want the same filename for a
 * different type of a device.  So beware if the DEVICE_ATTR() macro ever
 * changes, this definition will also have to change.
 */
static struct device_attribute dev_attr_client_name = {
  .attr = { .name = "name", .mode = S_IRUGO, .owner = THIS_MODULE },
  .show = &show_client_name,
  };


/* ---------------------------------------------------
 * registering functions
 * ---------------------------------------------------
 */

/* -----
 * i2c_add_adapter is called from within the algorithm layer,
 * when a new hw adapter registers. A new device is register to be
 * available for clients.
 */
int i2c_add_adapter(struct i2c_adapter *adap)
  {
  int id, res = 0;
  struct list_head   *item;
  struct i2c_driver  *driver;

  down(&core_lists);

  if (idr_pre_get(&i2c_adapter_idr, GFP_KERNEL) == 0)
    {
    res = -ENOMEM;
    goto out_unlock;
    }

  res = idr_get_new(&i2c_adapter_idr, adap, &id);

  if (res < 0)
    {
    if (res == -EAGAIN)
      res = -ENOMEM;

    goto out_unlock;
    }

  adap->nr =  id & MAX_ID_MASK;
  init_MUTEX(&adap->bus_lock);
  init_MUTEX(&adap->clist_lock);
  list_add_tail(&adap->list, &adapters);
  INIT_LIST_HEAD(&adap->clients);

  /* Add the adapter to the driver core.
   * If the parent pointer is not set up,
   * we add this adapter to the host bus.
   */
  if (adap->dev.parent == NULL)
    adap->dev.parent = &platform_bus;

  sprintf(adap->dev.bus_id, "i2c-%d", adap->nr);
  adap->dev.driver = &i2c_adapter_driver;
  adap->dev.release = &i2c_adapter_dev_release;
  device_register(&adap->dev);
  device_create_file(&adap->dev, &dev_attr_name);

    /* Add this adapter to the i2c_adapter class */
  memset(&adap->class_dev, 0x00, sizeof(struct class_device));
  adap->class_dev.dev = &adap->dev;
  adap->class_dev.class = &i2c_adapter_class;
  strlcpy(adap->class_dev.class_id, adap->dev.bus_id, BUS_ID_SIZE);
  class_device_register(&adap->class_dev);

  dev_dbg(&adap->dev, "adapter [%s] registered\n", adap->name);

    /* inform drivers of new adapters */
  list_for_each(item, &drivers) {
    driver = list_entry(item, struct i2c_driver, list);

    if (driver->flags & I2C_DF_NOTIFY)
        /* We ignore the return code; if it fails, too bad */
      driver->attach_adapter(adap);
    }

out_unlock:
  up(&core_lists);
  return res;
  }

int i2c_del_adapter(struct i2c_adapter *adap)
  {
  struct list_head  *item, *_n;
  struct i2c_adapter *adap_from_list;
  struct i2c_driver *driver;
  struct i2c_client *client;
  int res = 0;

  down(&core_lists);

    /* First make sure that this adapter was ever added */
  list_for_each_entry(adap_from_list, &adapters, list) {
    if (adap_from_list == adap)
      break;
    }

  if (adap_from_list != adap)
    {
    pr_debug("i2c-core: attempting to delete unregistered "
             "adapter [%s]\n", adap->name);
    res = -EINVAL;
    goto out_unlock;
    }

  list_for_each(item, &drivers) {
    driver = list_entry(item, struct i2c_driver, list);

    if (driver->detach_adapter)
      if ((res = driver->detach_adapter(adap)))
        {
        dev_err(&adap->dev, "detach_adapter failed "
                "for driver [%s]\n", driver->name);
        goto out_unlock;
        }
    }

  /* detach any active clients. This must be done first, because
   * it can fail; in which case we give up. */
  list_for_each_safe(item, _n, &adap->clients) {
    client = list_entry(item, struct i2c_client, list);

    /* detaching devices is unconditional of the set notify
     * flag, as _all_ clients that reside on the adapter
     * must be deleted, as this would cause invalid states.
     */
    if ((res = client->driver->detach_client(client)))
      {
      dev_err(&adap->dev, "detach_client failed for client "
              "[%s] at address 0x%02x\n", client->name,
              client->addr);
      goto out_unlock;
      }
    }

    /* clean up the sysfs representation */
  init_completion(&adap->dev_released);
  init_completion(&adap->class_dev_released);
  class_device_unregister(&adap->class_dev);
  device_remove_file(&adap->dev, &dev_attr_name);
  device_unregister(&adap->dev);
  list_del(&adap->list);

    /* wait for sysfs to drop all references */
  wait_for_completion(&adap->dev_released);
  wait_for_completion(&adap->class_dev_released);

    /* free dynamically allocated bus id */
  idr_remove(&i2c_adapter_idr, adap->nr);

  dev_dbg(&adap->dev, "adapter [%s] unregistered\n", adap->name);

out_unlock:
  up(&core_lists);
  return res;
  }

/* -----
 * What follows is the "upwards" interface: commands for talking to clients,
 * which implement the functions to access the physical information of the
 * chips.
 */

int i2c_add_driver(struct i2c_driver *driver)
  {
  struct list_head   *item;
  struct i2c_adapter *adapter;
  int res = 0;

  down(&core_lists);

    /* add the driver to the list of i2c drivers in the driver core */
  driver->driver.name = driver->name;
  driver->driver.bus = &i2c_bus_type;
  driver->driver.probe = i2c_device_probe;
  driver->driver.remove = i2c_device_remove;

  res = driver_register(&driver->driver);

  if (res)
    goto out_unlock;

  list_add_tail(&driver->list, &drivers);
  pr_debug("i2c-core: driver [%s] registered\n", driver->name);

    /* now look for instances of driver on our adapters */
  if (driver->flags & I2C_DF_NOTIFY)
    {
    list_for_each(item, &adapters) {
      adapter = list_entry(item, struct i2c_adapter, list);
      driver->attach_adapter(adapter);
      }
    }

out_unlock:
  up(&core_lists);
  return res;
  }

int i2c_del_driver(struct i2c_driver *driver)
  {
  struct list_head   *item1, *item2, *_n;
  struct i2c_client  *client;
  struct i2c_adapter *adap;

  int res = 0;

  down(&core_lists);

  /* Have a look at each adapter, if clients of this driver are still
   * attached. If so, detach them to be able to kill the driver
   * afterwards.
   *
   * Removing clients does not depend on the notify flag, else
   * invalid operation might (will!) result, when using stale client
   * pointers.
   */
  list_for_each(item1, &adapters) {
    adap = list_entry(item1, struct i2c_adapter, list);

    if (driver->detach_adapter)
      {
      if ((res = driver->detach_adapter(adap)))
        {
        dev_err(&adap->dev, "detach_adapter failed "
                "for driver [%s]\n", driver->name);
        goto out_unlock;
        }
      }
    else
      {
      list_for_each_safe(item2, _n, &adap->clients) {
        client = list_entry(item2, struct i2c_client, list);

        if (client->driver != driver)
          continue;

        dev_dbg(&adap->dev, "detaching client [%s] "
                "at 0x%02x\n", client->name,
                client->addr);

        if ((res = driver->detach_client(client)))
          {
          dev_err(&adap->dev, "detach_client "
                  "failed for client [%s] at "
                  "0x%02x\n", client->name,
                  client->addr);
          goto out_unlock;
          }
        }
      }
    }

  driver_unregister(&driver->driver);
  list_del(&driver->list);
  pr_debug("i2c-core: driver [%s] unregistered\n", driver->name);

out_unlock:
  up(&core_lists);
  return 0;
  }

static int __i2c_check_addr(struct i2c_adapter *adapter, unsigned int addr)
  {
  struct list_head   *item;
  struct i2c_client  *client;

  list_for_each(item, &adapter->clients) {
    client = list_entry(item, struct i2c_client, list);

    if (client->addr == addr)
      return -EBUSY;
    }
  return 0;
  }

int i2c_check_addr(struct i2c_adapter *adapter, int addr)
  {
  int rval;

  down(&adapter->clist_lock);
  rval = __i2c_check_addr(adapter, addr);
  up(&adapter->clist_lock);

  return rval;
  }

int i2c_attach_client(struct i2c_client *client)
  {
  struct i2c_adapter *adapter = client->adapter;

  down(&adapter->clist_lock);

  if (__i2c_check_addr(client->adapter, client->addr))
    {
    up(&adapter->clist_lock);
    return -EBUSY;
    }

  list_add_tail(&client->list, &adapter->clients);
  up(&adapter->clist_lock);

  if (adapter->client_register)
    {
    if (adapter->client_register(client))
      {
      dev_dbg(&adapter->dev, "client_register "
              "failed for client [%s] at 0x%02x\n",
              client->name, client->addr);
      }
    }

  if (client->flags & I2C_CLIENT_ALLOW_USE)
    client->usage_count = 0;

  client->dev.parent = &client->adapter->dev;
  client->dev.driver = &client->driver->driver;
  client->dev.bus = &i2c_bus_type;
  client->dev.release = &i2c_client_release;

  snprintf(&client->dev.bus_id[0], sizeof(client->dev.bus_id),
           "%d-%04x", i2c_adapter_id(adapter), client->addr);
  dev_dbg(&adapter->dev, "client [%s] registered with bus id %s\n",
          client->name, client->dev.bus_id);
  device_register(&client->dev);
  device_create_file(&client->dev, &dev_attr_client_name);

  return 0;
  }

int i2c_detach_client(struct i2c_client *client)
  {
  struct i2c_adapter *adapter = client->adapter;
  int res = 0;

  if ((client->flags & I2C_CLIENT_ALLOW_USE)
      && (client->usage_count > 0))
    {
    dev_warn(&client->dev, "Client [%s] still busy, "
             "can't detach\n", client->name);
    return -EBUSY;
    }

  if (adapter->client_unregister)
    {
    res = adapter->client_unregister(client);

    if (res)
      {
      dev_err(&client->dev,
              "client_unregister [%s] failed, "
              "client not detached\n", client->name);
      goto out;
      }
    }

  down(&adapter->clist_lock);
  list_del(&client->list);
  init_completion(&client->released);
  device_remove_file(&client->dev, &dev_attr_client_name);
  device_unregister(&client->dev);
  up(&adapter->clist_lock);
  wait_for_completion(&client->released);

out:
  return res;
  }

static int i2c_inc_use_client(struct i2c_client *client)
  {

  if (!try_module_get(client->driver->owner))
    return -ENODEV;

  if (!try_module_get(client->adapter->owner))
    {
    module_put(client->driver->owner);
    return -ENODEV;
    }

  return 0;
  }

static void i2c_dec_use_client(struct i2c_client *client)
  {
  module_put(client->driver->owner);
  module_put(client->adapter->owner);
  }

int i2c_use_client(struct i2c_client *client)
  {
  int ret;

  ret = i2c_inc_use_client(client);

  if (ret)
    return ret;

  if (client->flags & I2C_CLIENT_ALLOW_USE)
    {
    if (client->flags & I2C_CLIENT_ALLOW_MULTIPLE_USE)
      client->usage_count++;
    else if (client->usage_count > 0)
      goto busy;
    else
      client->usage_count++;
    }

  return 0;
busy:
  i2c_dec_use_client(client);
  return -EBUSY;
  }

int i2c_release_client(struct i2c_client *client)
  {
  if (client->flags & I2C_CLIENT_ALLOW_USE)
    {
    if (client->usage_count > 0)
      client->usage_count--;
    else
      {
      pr_debug("i2c-core: %s used one too many times\n",
               __FUNCTION__);
      return -EPERM;
      }
    }

  i2c_dec_use_client(client);

  return 0;
  }

void i2c_clients_command(struct i2c_adapter *adap, unsigned int cmd, void *arg)
  {
  struct list_head  *item;
  struct i2c_client *client;

  down(&adap->clist_lock);
  list_for_each(item, &adap->clients) {
    client = list_entry(item, struct i2c_client, list);

    if (!try_module_get(client->driver->owner))
      continue;

    if (NULL != client->driver->command)
      {
      up(&adap->clist_lock);
      client->driver->command(client, cmd, arg);
      down(&adap->clist_lock);
      }

    module_put(client->driver->owner);
    }
  up(&adap->clist_lock);
  }

static int __init i2c_init(void)
  {
  int retval;

  retval = bus_register(&i2c_bus_type);

  if (retval)
    return retval;

  retval = driver_register(&i2c_adapter_driver);

  if (retval)
    return retval;

  return class_register(&i2c_adapter_class);
  }

static void __exit i2c_exit(void)
  {
  class_unregister(&i2c_adapter_class);
  driver_unregister(&i2c_adapter_driver);
  bus_unregister(&i2c_bus_type);
  }

subsys_initcall(i2c_init);
module_exit(i2c_exit);

/* ----------------------------------------------------
 * the functional interface to the i2c busses.
 * ----------------------------------------------------
 */

int i2c_transfer(struct i2c_adapter * adap, struct i2c_msg *msgs, int num)
  {
  int ret;

  if (adap->algo->master_xfer)
    {
#ifdef DEBUG

    for (ret = 0; ret < num; ret++)
      {
      dev_dbg(&adap->dev, "master_xfer[%d] %c, addr=0x%02x, "
              "len=%d\n", ret, msgs[ret].flags & I2C_M_RD ?
              'R' : 'W', msgs[ret].addr, msgs[ret].len);
      }

#endif

    down(&adap->bus_lock);
    ret = adap->algo->master_xfer(adap, msgs, num);
    up(&adap->bus_lock);

    return ret;
    }
  else
    {
    dev_dbg(&adap->dev, "I2C level transfers not supported\n");
    return -ENOSYS;
    }
  }

int i2c_master_send(struct i2c_client *client, const char *buf, int count)
  {
  int ret;
  struct i2c_adapter *adap = client->adapter;
  struct i2c_msg msg;

  msg.addr = client->addr;
  msg.flags = client->flags & I2C_M_TEN;
  msg.len = count;
  msg.buf = (char *)buf;

  ret = i2c_transfer(adap, &msg, 1);

  /* If everything went ok (i.e. 1 msg transmitted), return #bytes
     transmitted, else error code. */
  return (ret == 1) ? count : ret;
  }

int i2c_master_recv(struct i2c_client *client, char *buf, int count)
  {
  struct i2c_adapter *adap = client->adapter;
  struct i2c_msg msg;
  int ret;

  msg.addr = client->addr;
  msg.flags = client->flags & I2C_M_TEN;
  msg.flags |= I2C_M_RD;
  msg.len = count;
  msg.buf = buf;

  ret = i2c_transfer(adap, &msg, 1);

  /* If everything went ok (i.e. 1 msg transmitted), return #bytes
     transmitted, else error code. */
  return (ret == 1) ? count : ret;
  }

int i2c_control(struct i2c_client *client,
                unsigned int cmd, unsigned long arg)
  {
  int ret = 0;
  struct i2c_adapter *adap = client->adapter;

  dev_dbg(&client->adapter->dev, "i2c ioctl, cmd: 0x%x, arg: %#lx\n", cmd, arg);

  switch (cmd)
    {
    case I2C_RETRIES:
      adap->retries = arg;
      break;

    case I2C_TIMEOUT:
      adap->timeout = arg;
      break;

    default:

      if (adap->algo->algo_control != NULL)
        ret = adap->algo->algo_control(adap, cmd, arg);
    }

  return ret;
  }

/* ----------------------------------------------------
 * the i2c address scanning function
 * Will not work for 10-bit addresses!
 * ----------------------------------------------------
 */
static int i2c_probe_address(struct i2c_adapter *adapter, int addr, int kind,
                             int (*found_proc)(struct i2c_adapter *, int, int))
  {
  int err;

    /* Make sure the address is valid */
  if (addr < 0x03 || addr > 0x77)
    {
    dev_warn(&adapter->dev, "Invalid probe address 0x%02x\n",
             addr);
    return -EINVAL;
    }

    /* Skip if already in use */
  if (i2c_check_addr(adapter, addr))
    return 0;

    /* Make sure there is something at this address, unless forced */
  if (kind < 0)
    {
    if (i2c_smbus_xfer(adapter, addr, 0, 0, 0,
                       I2C_SMBUS_QUICK, NULL) < 0)
      return 0;

      /* prevent 24RF08 corruption */
    if ((addr & ~0x0f) == 0x50)
      i2c_smbus_xfer(adapter, addr, 0, 0, 0,
                     I2C_SMBUS_QUICK, NULL);
    }

    /* Finally call the custom detection function */
  err = found_proc(adapter, addr, kind);

  /* -ENODEV can be returned if there is a chip at the given address
     but it isn't supported by this chip driver. We catch it here as
     this isn't an error. */
  return (err == -ENODEV) ? 0 : err;
  }

int i2c_probe(struct i2c_adapter *adapter,
              struct i2c_client_address_data *address_data,
              int (*found_proc)(struct i2c_adapter *, int, int))
  {
  int i, err;
  int adap_id = i2c_adapter_id(adapter);

    /* Forget it if we can't probe using SMBUS_QUICK */
  if (!i2c_check_functionality(adapter, I2C_FUNC_SMBUS_QUICK))
    return -1;

  /* Force entries are done first, and are not affected by ignore
     entries */
  if (address_data->forces)
    {
    unsigned short **forces = address_data->forces;
    int kind;

    for (kind = 0; forces[kind]; kind++)
      {
      for (i = 0; forces[kind][i] != I2C_CLIENT_END;
           i += 2)
        {
        if (forces[kind][i] == adap_id
            || forces[kind][i] == ANY_I2C_BUS)
          {
          dev_dbg(&adapter->dev, "found force "
                  "parameter for adapter %d, "
                  "addr 0x%02x, kind %d\n",
                  adap_id, forces[kind][i + 1],
                  kind);
          err = i2c_probe_address(adapter,
                                  forces[kind][i + 1],
                                  kind, found_proc);

          if (err)
            return err;
          }
        }
      }
    }

  /* Probe entries are done second, and are not affected by ignore
     entries either */
  for (i = 0; address_data->probe[i] != I2C_CLIENT_END; i += 2)
    {
    if (address_data->probe[i] == adap_id
        || address_data->probe[i] == ANY_I2C_BUS)
      {
      dev_dbg(&adapter->dev, "found probe parameter for "
              "adapter %d, addr 0x%02x\n", adap_id,
              address_data->probe[i + 1]);
      err = i2c_probe_address(adapter,
                              address_data->probe[i + 1],
                              -1, found_proc);

      if (err)
        return err;
      }
    }

    /* Normal entries are done last, unless shadowed by an ignore entry */
  for (i = 0; address_data->normal_i2c[i] != I2C_CLIENT_END; i += 1)
    {
    int j, ignore;

    ignore = 0;

    for (j = 0; address_data->ignore[j] != I2C_CLIENT_END;
         j += 2)
      {
      if ((address_data->ignore[j] == adap_id ||
           address_data->ignore[j] == ANY_I2C_BUS)
          && address_data->ignore[j + 1]
          == address_data->normal_i2c[i])
        {
        dev_dbg(&adapter->dev, "found ignore "
                "parameter for adapter %d, "
                "addr 0x%02x\n", adap_id,
                address_data->ignore[j + 1]);
        }

      ignore = 1;
      break;
      }

    if (ignore)
      continue;

    dev_dbg(&adapter->dev, "found normal entry for adapter %d, "
            "addr 0x%02x\n", adap_id,
            address_data->normal_i2c[i]);
    err = i2c_probe_address(adapter, address_data->normal_i2c[i],
                            -1, found_proc);

    if (err)
      return err;
    }

  return 0;
  }

struct i2c_adapter* i2c_get_adapter(int id)
  {
  struct i2c_adapter *adapter;

  down(&core_lists);
  adapter = (struct i2c_adapter *)idr_find(&i2c_adapter_idr, id);

  if (adapter && !try_module_get(adapter->owner))
    adapter = NULL;

  up(&core_lists);
  return adapter;
  }

void i2c_put_adapter(struct i2c_adapter *adap)
  {
  module_put(adap->owner);
  }

/* The SMBus parts */

#define POLY    (0x1070U << 3)
static u8
crc8(u16 data)
  {
  int i;

  for (i = 0; i < 8; i++)
    {
    if (data & 0x8000)
      data = data ^ POLY;

    data = data << 1;
    }

  return (u8)(data >> 8);
  }

/* CRC over count bytes in the first array plus the bytes in the rest
   array if it is non-null. rest[0] is the (length of rest) - 1
   and is included. */
static u8 i2c_smbus_partial_pec(u8 crc, int count, u8 *first, u8 *rest)
  {
  int i;

  for (i = 0; i < count; i++)
    crc = crc8((crc ^ first[i]) << 8);

  if (rest != NULL)
    for (i = 0; i <= rest[0]; i++)
      crc = crc8((crc ^ rest[i]) << 8);

  return crc;
  }

static u8 i2c_smbus_pec(int count, u8 *first, u8 *rest)
  {
  return i2c_smbus_partial_pec(0, count, first, rest);
  }

/* Returns new "size" (transaction type)
   Note that we convert byte to byte_data and byte_data to word_data
   rather than invent new xxx_PEC transactions. */
static int i2c_smbus_add_pec(u16 addr, u8 command, int size,
                             union i2c_smbus_data *data)
  {
  u8 buf[3];

  buf[0] = addr << 1;
  buf[1] = command;

  switch (size)
    {
    case I2C_SMBUS_BYTE:
      data->byte = i2c_smbus_pec(2, buf, NULL);
      size = I2C_SMBUS_BYTE_DATA;
      break;

    case I2C_SMBUS_BYTE_DATA:
      buf[2] = data->byte;
      data->word = buf[2] ||
                   (i2c_smbus_pec(3, buf, NULL) << 8);
      size = I2C_SMBUS_WORD_DATA;
      break;

    case I2C_SMBUS_WORD_DATA:
        /* unsupported */
      break;

    case I2C_SMBUS_BLOCK_DATA:
      data->block[data->block[0] + 1] =
        i2c_smbus_pec(2, buf, data->block);
      size = I2C_SMBUS_BLOCK_DATA_PEC;
      break;
    }

  return size;
  }

static int i2c_smbus_check_pec(u16 addr, u8 command, int size, u8 partial,
                               union i2c_smbus_data *data)
  {
  u8 buf[3], rpec, cpec;

  buf[1] = command;

  switch (size)
    {
    case I2C_SMBUS_BYTE_DATA:
      buf[0] = (addr << 1) | 1;
      cpec = i2c_smbus_pec(2, buf, NULL);
      rpec = data->byte;
      break;

    case I2C_SMBUS_WORD_DATA:
      buf[0] = (addr << 1) | 1;
      buf[2] = data->word & 0xff;
      cpec = i2c_smbus_pec(3, buf, NULL);
      rpec = data->word >> 8;
      break;

    case I2C_SMBUS_WORD_DATA_PEC:
        /* unsupported */
      cpec = rpec = 0;
      break;

    case I2C_SMBUS_PROC_CALL_PEC:
        /* unsupported */
      cpec = rpec = 0;
      break;

    case I2C_SMBUS_BLOCK_DATA_PEC:
      buf[0] = (addr << 1);
      buf[2] = (addr << 1) | 1;
      cpec = i2c_smbus_pec(3, buf, data->block);
      rpec = data->block[data->block[0] + 1];
      break;

    case I2C_SMBUS_BLOCK_PROC_CALL_PEC:
      buf[0] = (addr << 1) | 1;
      rpec = i2c_smbus_partial_pec(partial, 1,
                                   buf, data->block);
      cpec = data->block[data->block[0] + 1];
      break;

    default:
      cpec = rpec = 0;
      break;
    }

  if (rpec != cpec)
    {
    pr_debug("i2c-core: Bad PEC 0x%02x vs. 0x%02x\n",
             rpec, cpec);
    return -1;
    }

  return 0;
  }

s32 i2c_smbus_write_quick(struct i2c_client *client, u8 value)
  {
  return i2c_smbus_xfer(client->adapter, client->addr, client->flags,
                        value, 0, I2C_SMBUS_QUICK, NULL);
  }

s32 i2c_smbus_read_byte(struct i2c_client *client)
  {
  union i2c_smbus_data data;

  if (i2c_smbus_xfer(client->adapter, client->addr, client->flags,
                     I2C_SMBUS_READ, 0, I2C_SMBUS_BYTE, &data))
    return -1;
  else
    return 0x0FF & data.byte;
  }

s32 i2c_smbus_write_byte(struct i2c_client *client, u8 value)
  {
  union i2c_smbus_data data;  /* only for PEC */

  return i2c_smbus_xfer(client->adapter, client->addr, client->flags,
                        I2C_SMBUS_WRITE, value, I2C_SMBUS_BYTE, &data);
  }

s32 i2c_smbus_read_byte_data(struct i2c_client *client, u8 command)
  {
  union i2c_smbus_data data;

  if (i2c_smbus_xfer(client->adapter, client->addr, client->flags,
                     I2C_SMBUS_READ, command, I2C_SMBUS_BYTE_DATA, &data))
    return -1;
  else
    return 0x0FF & data.byte;
  }

s32 i2c_smbus_write_byte_data(struct i2c_client *client, u8 command, u8 value)
  {
  union i2c_smbus_data data;

  data.byte = value;
  return i2c_smbus_xfer(client->adapter, client->addr, client->flags,
                        I2C_SMBUS_WRITE, command,
                        I2C_SMBUS_BYTE_DATA, &data);
  }

s32 i2c_smbus_read_word_data(struct i2c_client *client, u8 command)
  {
  union i2c_smbus_data data;

  if (i2c_smbus_xfer(client->adapter, client->addr, client->flags,
                     I2C_SMBUS_READ, command, I2C_SMBUS_WORD_DATA, &data))
    return -1;
  else
    return 0x0FFFF & data.word;
  }

s32 i2c_smbus_write_word_data(struct i2c_client *client, u8 command, u16 value)
  {
  union i2c_smbus_data data;

  data.word = value;
  return i2c_smbus_xfer(client->adapter, client->addr, client->flags,
                        I2C_SMBUS_WRITE, command,
                        I2C_SMBUS_WORD_DATA, &data);
  }

s32 i2c_smbus_write_block_data(struct i2c_client *client, u8 command,
                               u8 length, u8 *values)
  {
  union i2c_smbus_data data;
  int i;

  if (length > I2C_SMBUS_BLOCK_MAX)
    length = I2C_SMBUS_BLOCK_MAX;

  for (i = 1; i <= length; i++)
    data.block[i] = values[i - 1];

  data.block[0] = length;
  return i2c_smbus_xfer(client->adapter, client->addr, client->flags,
                        I2C_SMBUS_WRITE, command,
                        I2C_SMBUS_BLOCK_DATA, &data);
  }

/* Returns the number of read bytes */
s32 i2c_smbus_read_i2c_block_data(struct i2c_client *client, u8 command, u8 *values)
  {
  union i2c_smbus_data data;
  int i;

  if (i2c_smbus_xfer(client->adapter, client->addr, client->flags,
                     I2C_SMBUS_READ, command,
                     I2C_SMBUS_I2C_BLOCK_DATA, &data))
    return -1;
  else
    {
    for (i = 1; i <= data.block[0]; i++)
      values[i - 1] = data.block[i];

    return data.block[0];
    }
  }

/* Simulate a SMBus command using the i2c protocol
   No checking of parameters is done!  */
static s32 i2c_smbus_xfer_emulated(struct i2c_adapter * adapter, u16 addr,
                                   unsigned short flags,
                                   char read_write, u8 command, int size,
                                   union i2c_smbus_data * data)
  {
  /* So we need to generate a series of msgs. In the case of writing, we
     need to use only one message; when reading, we need two. We initialize
     most things with sane defaults, to keep the code below somewhat
     simpler. */
  unsigned char msgbuf0[34];
  unsigned char msgbuf1[34];
  int num = read_write == I2C_SMBUS_READ ? 2 : 1;
  struct i2c_msg msg[2] = { { addr, flags, 1, msgbuf0 },
                                { addr, flags | I2C_M_RD, 0, msgbuf1 } };
  int i;

  msgbuf0[0] = command;

  switch (size)
    {
    case I2C_SMBUS_QUICK:
      msg[0].len = 0;
        /* Special case: The read/write field is used as data */
      msg[0].flags = flags | (read_write == I2C_SMBUS_READ) ? I2C_M_RD : 0;
      num = 1;
      break;

    case I2C_SMBUS_BYTE:

      if (read_write == I2C_SMBUS_READ)
        {
          /* Special case: only a read! */
        msg[0].flags = I2C_M_RD | flags;
        num = 1;
        }

      break;

    case I2C_SMBUS_BYTE_DATA:

      if (read_write == I2C_SMBUS_READ)
        msg[1].len = 1;
      else
        {
        msg[0].len = 2;
        msgbuf0[1] = data->byte;
        }

      break;

    case I2C_SMBUS_WORD_DATA:

      if (read_write == I2C_SMBUS_READ)
        msg[1].len = 2;
      else
        {
        msg[0].len = 3;
        msgbuf0[1] = data->word & 0xff;
        msgbuf0[2] = (data->word >> 8) & 0xff;
        }

      break;

    case I2C_SMBUS_PROC_CALL:
      num = 2; /* Special case */
      read_write = I2C_SMBUS_READ;
      msg[0].len = 3;
      msg[1].len = 2;
      msgbuf0[1] = data->word & 0xff;
      msgbuf0[2] = (data->word >> 8) & 0xff;
      break;

    case I2C_SMBUS_BLOCK_DATA:
    case I2C_SMBUS_BLOCK_DATA_PEC:

      if (read_write == I2C_SMBUS_READ)
        {
        dev_err(&adapter->dev, "Block read not supported "
                "under I2C emulation!\n");
        return -1;
        }
      else
        {
        msg[0].len = data->block[0] + 2;

        if (msg[0].len > I2C_SMBUS_BLOCK_MAX + 2)
          {
          dev_err(&adapter->dev, "smbus_access called with "
                  "invalid block write size (%d)\n",
                  data->block[0]);
          return -1;
          }

        if (size == I2C_SMBUS_BLOCK_DATA_PEC)
          (msg[0].len)++;

        for (i = 1; i <= msg[0].len; i++)
          msgbuf0[i] = data->block[i - 1];
        }

      break;

    case I2C_SMBUS_BLOCK_PROC_CALL:
    case I2C_SMBUS_BLOCK_PROC_CALL_PEC:
      dev_dbg(&adapter->dev, "Block process call not supported "
              "under I2C emulation!\n");
      return -1;

    case I2C_SMBUS_I2C_BLOCK_DATA:

      if (read_write == I2C_SMBUS_READ)
        msg[1].len = I2C_SMBUS_I2C_BLOCK_MAX;
      else
        {
        msg[0].len = data->block[0] + 1;

        if (msg[0].len > I2C_SMBUS_I2C_BLOCK_MAX + 1)
          {
          dev_err(&adapter->dev, "i2c_smbus_xfer_emulated called with "
                  "invalid block write size (%d)\n",
                  data->block[0]);
          return -1;
          }

        for (i = 1; i <= data->block[0]; i++)
          msgbuf0[i] = data->block[i];
        }

      break;

    default:
      dev_err(&adapter->dev, "smbus_access called with invalid size (%d)\n",
              size);
      return -1;
    }

  if (i2c_transfer(adapter, msg, num) < 0)
    return -1;

  if (read_write == I2C_SMBUS_READ)
    switch (size)
      {
      case I2C_SMBUS_BYTE:
        data->byte = msgbuf0[0];
        break;

      case I2C_SMBUS_BYTE_DATA:
        data->byte = msgbuf1[0];
        break;

      case I2C_SMBUS_WORD_DATA:
      case I2C_SMBUS_PROC_CALL:
        data->word = msgbuf1[0] | (msgbuf1[1] << 8);
        break;

      case I2C_SMBUS_I2C_BLOCK_DATA:
          /* fixed at 32 for now */
        data->block[0] = I2C_SMBUS_I2C_BLOCK_MAX;

        for (i = 0; i < I2C_SMBUS_I2C_BLOCK_MAX; i++)
          data->block[i + 1] = msgbuf1[i];

        break;
      }

  return 0;
  }

s32 i2c_smbus_xfer(struct i2c_adapter * adapter, u16 addr, unsigned short flags,
                   char read_write, u8 command, int size,
                   union i2c_smbus_data * data)
  {
  s32 res;
  int swpec = 0;
  u8 partial = 0;

  flags &= I2C_M_TEN | I2C_CLIENT_PEC;

  if ((flags & I2C_CLIENT_PEC) &&
      !(i2c_check_functionality(adapter, I2C_FUNC_SMBUS_HWPEC_CALC)))
    {
    swpec = 1;

    if (read_write == I2C_SMBUS_READ &&
        size == I2C_SMBUS_BLOCK_DATA)
      size = I2C_SMBUS_BLOCK_DATA_PEC;
    else if (size == I2C_SMBUS_PROC_CALL)
      size = I2C_SMBUS_PROC_CALL_PEC;
    else if (size == I2C_SMBUS_BLOCK_PROC_CALL)
      {
      i2c_smbus_add_pec(addr, command,
                        I2C_SMBUS_BLOCK_DATA, data);
      partial = data->block[data->block[0] + 1];
      size = I2C_SMBUS_BLOCK_PROC_CALL_PEC;
      }
    else if (read_write == I2C_SMBUS_WRITE &&
             size != I2C_SMBUS_QUICK &&
             size != I2C_SMBUS_I2C_BLOCK_DATA)
      size = i2c_smbus_add_pec(addr, command, size, data);
    }

  if (adapter->algo->smbus_xfer)
    {
    down(&adapter->bus_lock);
    res = adapter->algo->smbus_xfer(adapter, addr, flags, read_write,
                                    command, size, data);
    up(&adapter->bus_lock);
    }
  else
    res = i2c_smbus_xfer_emulated(adapter, addr, flags, read_write,
                                  command, size, data);

  if (res >= 0 && swpec &&
      size != I2C_SMBUS_QUICK && size != I2C_SMBUS_I2C_BLOCK_DATA &&
      (read_write == I2C_SMBUS_READ || size == I2C_SMBUS_PROC_CALL_PEC ||
       size == I2C_SMBUS_BLOCK_PROC_CALL_PEC))
    if (i2c_smbus_check_pec(addr, command, size, partial, data))
      return -1;

  return res;
  }

/* Next four are needed by i2c-isa */
EXPORT_SYMBOL_GPL(i2c_adapter_dev_release);
EXPORT_SYMBOL_GPL(i2c_adapter_driver);
EXPORT_SYMBOL_GPL(i2c_adapter_class);
EXPORT_SYMBOL_GPL(i2c_bus_type);

EXPORT_SYMBOL(i2c_add_adapter);
EXPORT_SYMBOL(i2c_del_adapter);
EXPORT_SYMBOL(i2c_add_driver);
EXPORT_SYMBOL(i2c_del_driver);
EXPORT_SYMBOL(i2c_attach_client);
EXPORT_SYMBOL(i2c_detach_client);
EXPORT_SYMBOL(i2c_use_client);
EXPORT_SYMBOL(i2c_release_client);
EXPORT_SYMBOL(i2c_clients_command);
EXPORT_SYMBOL(i2c_check_addr);

EXPORT_SYMBOL(i2c_master_send);
EXPORT_SYMBOL(i2c_master_recv);
EXPORT_SYMBOL(i2c_control);
EXPORT_SYMBOL(i2c_transfer);
EXPORT_SYMBOL(i2c_get_adapter);
EXPORT_SYMBOL(i2c_put_adapter);
EXPORT_SYMBOL(i2c_probe);

EXPORT_SYMBOL(i2c_smbus_xfer);
EXPORT_SYMBOL(i2c_smbus_write_quick);
EXPORT_SYMBOL(i2c_smbus_read_byte);
EXPORT_SYMBOL(i2c_smbus_write_byte);
EXPORT_SYMBOL(i2c_smbus_read_byte_data);
EXPORT_SYMBOL(i2c_smbus_write_byte_data);
EXPORT_SYMBOL(i2c_smbus_read_word_data);
EXPORT_SYMBOL(i2c_smbus_write_word_data);
EXPORT_SYMBOL(i2c_smbus_write_block_data);
EXPORT_SYMBOL(i2c_smbus_read_i2c_block_data);

MODULE_AUTHOR("Simon G. Vogl <simon@tk.uni-linz.ac.at>");
MODULE_DESCRIPTION("I2C-Bus main module");
MODULE_LICENSE("GPL");
