sonic-buildimage/platform/broadcom/sonic-platform-modules-juniper/qfx5210/modules/x86-64-juniper-qfx5210-64x-psu.c

/*
 * An hwmon driver for juniper qfx5210_64x Power Module
 *
 * Modified and tested on Juniper QFX5210
 * Ciju Rajan K <crajank@juniper.net>
 *
 * Copyright (C) 2014 Accton Technology Corporation.
 * Brandon Chuang <brandon_chuang@accton.com.tw> 
 *
 * Based on ad7414.c
 * Copyright 2006 Stefan Roese <sr at denx.de>, DENX Software Engineering
 *
 * 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.
 */

#include <linux/module.h>
#include <linux/jiffies.h>
#include <linux/i2c.h>
#include <linux/hwmon.h>
#include <linux/hwmon-sysfs.h>
#include <linux/err.h>
#include <linux/mutex.h>
#include <linux/sysfs.h>
#include <linux/slab.h>
#include <linux/delay.h>
#include <linux/dmi.h>
#include <linux/notifier.h>
#include <linux/reboot.h>

#define PSU_STATUS_I2C_ADDR			0x60
#define PSU_STATUS_I2C_REG_OFFSET	0x03

#define IS_POWER_GOOD(id, value)	(!!(value & BIT(2+id)))
#define IS_PRESENT(id, value)		(!(value & BIT(id)))

static ssize_t show_status(struct device *dev, struct device_attribute *da, char *buf);
static struct qfx5210_64x_psu_data *qfx5210_64x_psu_update_device(struct device *dev);
extern int juniper_i2c_cpld_read (u8 cpld_addr, u8 reg);
/*
 * This function is defined in juniper_i2c_cpld.c
 */
extern int juniper_i2c_cpld_write(unsigned short, u8, u8);

/* Addresses scanned 
 */
static const unsigned short normal_i2c[] = { I2C_CLIENT_END };

/* Each client has this additional data 
 */
struct qfx5210_64x_psu_data {
    struct device      *hwmon_dev;
    struct mutex        update_lock;
    char                valid;           /* !=0 if registers are valid */
    unsigned long       last_updated;    /* In jiffies */
    u8  index;           /* PSU index */
    u8  status;          /* Status(present/power_good) register read from CPLD */
};             

enum qfx5210_64x_psu_sysfs_attributes {
	PSU_PRESENT,
	PSU_POWER_GOOD
};

/* sysfs attributes for hwmon 
 */
static SENSOR_DEVICE_ATTR(psu_present,    S_IRUGO, show_status, NULL, PSU_PRESENT);
static SENSOR_DEVICE_ATTR(psu_power_good, S_IRUGO, show_status, NULL, PSU_POWER_GOOD);

static struct attribute *qfx5210_64x_psu_attributes[] = {
    &sensor_dev_attr_psu_present.dev_attr.attr,
    &sensor_dev_attr_psu_power_good.dev_attr.attr,
    NULL
};

static int qfx5210_cpld_soft_reset(struct notifier_block *nb,
                                   unsigned long action, 
                                   void *data)
{
    int ret = 0;

    switch (action) {
	    case SYS_POWER_OFF:
	    case SYS_HALT:
		    printk(KERN_CRIT "System halt/power_off\n");
		    break;
	    case SYS_RESTART:
		    printk(KERN_CRIT "System restart: qfx5210_cpld_soft_reset\n");
		    ret = juniper_i2c_cpld_write(0x65, 0x04, 0x01);
		    if (ret) {
			    printk(KERN_CRIT "qfx5210_cpld_soft_reset failed\n");
		    }
		    msleep(100);
		    break;
	    default:
		    /* Do Nothing */
		    break;
    }
    return NOTIFY_DONE;
}

static struct notifier_block qfx5210_nb = {
    .notifier_call = qfx5210_cpld_soft_reset,
};

static ssize_t show_status(struct device *dev, struct device_attribute *da,
             char *buf)
{
    struct sensor_device_attribute *attr = to_sensor_dev_attr(da);
    struct qfx5210_64x_psu_data *data = qfx5210_64x_psu_update_device(dev);
    u8 status = 0;

    if (!data->valid) {
        return -EIO;
    }

    if (attr->index == PSU_PRESENT) {
		status = IS_PRESENT(data->index, data->status);
    }
    else { /* PSU_POWER_GOOD */
		status = IS_POWER_GOOD(data->index, data->status);
    }

    return sprintf(buf, "%d\n", status);
}

static const struct attribute_group qfx5210_64x_psu_group = {
    .attrs = qfx5210_64x_psu_attributes,
};

/*
 * QFX5210 power off sequence
 */
static void qfx5210_cpld_power_off(void)
{
    printk(KERN_ALERT "pm_power_off: qfx5210_cpld_power_off\n");
    (void) juniper_i2c_cpld_write(0x65, 0x14, 0x00);
    msleep(100);
    (void) juniper_i2c_cpld_write(0x77, 0x00, 0x01);
    msleep(100);
    (void) juniper_i2c_cpld_write(0x76, 0x00, 0x04);
}

/*
 * Default platform pm_power_off handler 
 */
static void (*default_pm_power_off)(void);

static int qfx5210_64x_psu_probe(struct i2c_client *client,
            const struct i2c_device_id *dev_id)
{
    struct qfx5210_64x_psu_data *data;
    int status;

    if (!i2c_check_functionality(client->adapter, I2C_FUNC_SMBUS_I2C_BLOCK)) {
        status = -EIO;
        goto exit;
    }

    data = kzalloc(sizeof(struct qfx5210_64x_psu_data), GFP_KERNEL);
    if (!data) {
        status = -ENOMEM;
        goto exit;
    }

    i2c_set_clientdata(client, data);
    data->valid = 0;
    data->index = dev_id->driver_data;
    mutex_init(&data->update_lock);

    dev_info(&client->dev, "chip found\n");

    /* Register sysfs hooks */
    status = sysfs_create_group(&client->dev.kobj, &qfx5210_64x_psu_group);
    if (status) {
        goto exit_free;
    }

    data->hwmon_dev = hwmon_device_register(&client->dev);
    if (IS_ERR(data->hwmon_dev)) {
        status = PTR_ERR(data->hwmon_dev);
        goto exit_remove;
    }

    dev_info(&client->dev, "%s: psu '%s'\n",
         dev_name(data->hwmon_dev), client->name);
    
    return 0;

exit_remove:
    sysfs_remove_group(&client->dev.kobj, &qfx5210_64x_psu_group);
exit_free:
    kfree(data);
exit:
    
    return status;
}

static int qfx5210_64x_psu_remove(struct i2c_client *client)
{
    struct qfx5210_64x_psu_data *data = i2c_get_clientdata(client);

    hwmon_device_unregister(data->hwmon_dev);
    sysfs_remove_group(&client->dev.kobj, &qfx5210_64x_psu_group);
    kfree(data);

    return 0;
}

enum psu_index 
{ 
    qfx5210_64x_psu1,
    qfx5210_64x_psu2
};

static const struct i2c_device_id qfx5210_64x_psu_id[] = {
    { "qfx5210_64x_psu1", qfx5210_64x_psu1 },
    { "qfx5210_64x_psu2", qfx5210_64x_psu2 },
    {}
};
MODULE_DEVICE_TABLE(i2c, qfx5210_64x_psu_id);

static struct i2c_driver qfx5210_64x_psu_driver = {
    .class        = I2C_CLASS_HWMON,
    .driver = {
        .name     = "qfx5210_64x_psu",
    },
    .probe        = qfx5210_64x_psu_probe,
    .remove       = qfx5210_64x_psu_remove,
    .id_table     = qfx5210_64x_psu_id,
    .address_list = normal_i2c,
};

static struct qfx5210_64x_psu_data *qfx5210_64x_psu_update_device(struct device *dev)
{
    struct i2c_client *client = to_i2c_client(dev);
    struct qfx5210_64x_psu_data *data = i2c_get_clientdata(client);
    
    mutex_lock(&data->update_lock);

    if (time_after(jiffies, data->last_updated + HZ + HZ / 2)
        || !data->valid) {
        int status;

        data->valid = 0;
        dev_dbg(&client->dev, "Starting qfx5210_64x update\n");

		/* Read psu status */
        status = juniper_i2c_cpld_read(PSU_STATUS_I2C_ADDR, PSU_STATUS_I2C_REG_OFFSET);
		
		if (status < 0) {
			dev_dbg(&client->dev, "cpld reg (0x%x) err %d\n", PSU_STATUS_I2C_ADDR, status);
			goto exit;
		}
		else {
			data->status = status;
		}

		data->last_updated = jiffies;
		data->valid = 1;
	}

exit:
	mutex_unlock(&data->update_lock);

	return data;
}

static int __init qfx5210_64x_psu_init(void)
{
    /*
     * Store the default poweroff handler for later usage 
     */
    default_pm_power_off = pm_power_off;
    /*
     * Register the cpld poweroff handler
     */
    pm_power_off = qfx5210_cpld_power_off;
    /*
     * Register the cpld soft reset handler
     */
    if(register_reboot_notifier(&qfx5210_nb)) {
    	printk(KERN_ALERT "Restart handler registration failed\n");
    }

    return i2c_add_driver(&qfx5210_64x_psu_driver);
}

static void __exit qfx5210_64x_psu_exit(void)
{
    /*
     * Restore the poweroff handler
     */
    pm_power_off = default_pm_power_off;
    /*
     * Unregister the cpld soft reset handler
     */
    if (!unregister_restart_handler(&qfx5210_nb)) {
    	printk(KERN_CRIT "Failed to uregister restart handler\n");
    }

    i2c_del_driver(&qfx5210_64x_psu_driver);
}

module_init(qfx5210_64x_psu_init);
module_exit(qfx5210_64x_psu_exit);

MODULE_AUTHOR("Brandon Chuang <brandon_chuang@accton.com.tw>");
MODULE_DESCRIPTION("qfx5210_64x_psu driver");
MODULE_LICENSE("GPL");