diff options
Diffstat (limited to 'ina260_full.c')
| -rwxr-xr-x | ina260_full.c | 542 |
1 files changed, 542 insertions, 0 deletions
diff --git a/ina260_full.c b/ina260_full.c new file mode 100755 index 0000000..283757c --- /dev/null +++ b/ina260_full.c @@ -0,0 +1,542 @@ +#include "linux/module.h" +#include "linux/uaccess.h" +#include "linux/i2c.h" +#include "linux/kobject.h" +#include "linux/slab.h" +#include "linux/kernel.h" +#include <linux/sysfs.h> + +// ina260 chip registers +#define INA260_REG_CONFIGURATION 0x00 +#define INA260_REG_CURRENT 0x01 +#define INA260_REG_VOLTAGE 0x02 +#define INA260_REG_POWER 0x03 +#define INA260_REG_MASKENABLE 0x06 +#define INA260_REG_ALERTLIMIT 0x07 +#define INA260_REG_MANUFACTURER 0xFE +#define INA260_REG_DIE 0xFF + +#define INA260_IS_ATTR(_name) (strcmp(attr->attr.name, #_name) == 0) + +// ina260 average modes list +static int ina260_avgs[8]={ + 1,4,16,64,128,256,512,1024 +}; + +// ina260 operating modes list +static char ina260_modes[8][42]={ + "Power-Down (or Shutdown)","Shunt Current, Triggered", + "Bus Voltage, Triggered","Shunt Current and Bus Voltage, Triggered", + "Power-Down (or Shutdown)","Shunt Current, Continuous","Bus Voltage, Continuous", + "Shunt Current and Bus Voltage, Continuous" +}; + +// ina260 conversion times list +static char ina260_ishcts_vbusct[8][9]={ + "140 µs", "204 µs","332 µs", "588 µs","1.1 ms", "2.116 ms","4.156 ms","8.244 ms" +}; + +// Driver root +static struct kobject* ina260_kobj; + +// Data attached to i2c clients +struct client_data { + struct kobject kobj; + struct i2c_client *client; + unsigned char reg; // Slave selected register +}; + +static const struct i2c_device_id ina260_ids[] = { + { "ina260", 0 }, + { } +}; +MODULE_DEVICE_TABLE(i2c,ina260_ids); + +/** + * @brief Read from ina260 registers + * + * @param cdata client data to use to communicate + * @param reg register to read + * @param value register content output + * @return int 0 on success, 1 on communication errors + */ +static int ina260_read_register(struct client_data* cdata, unsigned char reg, int *value){ + unsigned char bytes[2]; + if(cdata->reg == reg){ + if(i2c_master_recv(cdata->client,bytes,2)<0) + return 1; + } else { + if(i2c_master_send(cdata->client,®,1)<0) + return 1; + cdata->reg = reg; + if(i2c_master_recv(cdata->client,bytes,2)<0) + return 1; + } + *value=(bytes[0]<<8) | bytes[1]; + return 0; +} + +/** + * @brief Write to ina260 registers + * + * @param cdata client data to use to communicate + * @param reg register to write to + * @param value value to write in @a reg + * @return int 0 on success, 1 on communication errors + */ +static int ina260_write_register(struct client_data* cdata, unsigned char reg, int value){ + unsigned char data[3]; + data[0]=reg; + data[1]=(value>>8) & 0xFF; // MSB + data[2]=value & 0xFF; // LSB + if(i2c_master_send(cdata->client,data,3)<0) + return 1; + return 0; +} + +static ssize_t attr_show(struct kobject *_kobj, + struct kobj_attribute *attr, + char *buf) +{ + int rvalue; + struct client_data *cdata=container_of(_kobj,struct client_data,kobj); + unsigned char reg=INA260_REG_POWER; + + if(!INA260_IS_ATTR(power)){ + if (INA260_IS_ATTR(current)){ + reg=INA260_REG_CURRENT; + } else if(INA260_IS_ATTR(bus_voltage)) { + reg=INA260_REG_VOLTAGE; + } else if(INA260_IS_ATTR(mask_enable)) { + reg=INA260_REG_MASKENABLE; + } else if(INA260_IS_ATTR(alert_limit)) { + reg=INA260_REG_ALERTLIMIT; + } else if(INA260_IS_ATTR(configuration)) { + reg=INA260_REG_CONFIGURATION; + } else if(INA260_IS_ATTR(manufacturer_id)) { + reg=INA260_REG_MANUFACTURER; + } else { + reg=INA260_REG_DIE; + } + } + + if(ina260_read_register(cdata,reg,&rvalue)) + return -1; + + return sprintf(buf, "%x\n", rvalue); +} + + +static ssize_t attr_store(struct kobject *_kobj, + struct kobj_attribute *attr, + const char *buf, size_t count) +{ + int uvalue; + struct client_data *cdata=container_of(_kobj,struct client_data,kobj); + unsigned char reg=INA260_REG_CONFIGURATION; + + if(kstrtoint(buf, 0,&uvalue)) + return -EINVAL; + + if (INA260_IS_ATTR(mask_enable)){ + reg=INA260_REG_MASKENABLE; + } else if (INA260_IS_ATTR(alert_limit)){ + reg=INA260_REG_ALERTLIMIT; + } + + if(ina260_write_register(cdata, reg, uvalue)) + return -1; + + return count; +} + +static ssize_t attr_metric_show(struct kobject *_kobj, + struct kobj_attribute *attr, + char *buf) +{ + int rvalue; + int op1, op2; + struct client_data *cdata=container_of(_kobj,struct client_data,kobj); + + if(INA260_IS_ATTR(power)){ + if(ina260_read_register(cdata, INA260_REG_POWER,&rvalue)) + return -1; + op1=10*rvalue; + return sprintf(buf, "%d.%d\n", op1/1000,(op1) % 1000); + } else if(INA260_IS_ATTR(voltage)){ + if(ina260_read_register(cdata, INA260_REG_VOLTAGE,&rvalue)) + return -1; + op1=rvalue*25/100 + rvalue; + op2=rvalue*25%100; + return sprintf(buf, "%d.%03d%d\n",(op1)/1000,(op1)%1000,op2); + } else if(INA260_IS_ATTR(current)){ + if(ina260_read_register(cdata, INA260_REG_CURRENT,&rvalue)) + return -1; + op1=rvalue*25/100 + rvalue; + op2=rvalue*25%100; + return sprintf(buf, "%d.%03d%d\n",(op1)/1000,(op1)%1000,op2); + } + + return -1; +} + +static ssize_t attr_field_show(struct kobject *_kobj, + struct kobj_attribute *attr, + char *buf) +{ + int rvalue; + struct client_data *cdata=container_of(_kobj,struct client_data,kobj); + if(INA260_IS_ATTR(avg)){ + if(ina260_read_register(cdata, INA260_REG_CONFIGURATION,&rvalue)) + return -1; + return sprintf(buf, "%d\n", ina260_avgs[(rvalue>>9)&0x7]); + } else if (INA260_IS_ATTR(mode)){ + if(ina260_read_register(cdata, INA260_REG_CONFIGURATION,&rvalue)) + return -1; + return sprintf(buf, "%s\n", ina260_modes[rvalue&0x7]); + } else if (INA260_IS_ATTR(ishct)){ + if(ina260_read_register(cdata, INA260_REG_CONFIGURATION,&rvalue)) + return -1; + return sprintf(buf, "%s\n", ina260_ishcts_vbusct[(rvalue>>3)&0x7]); + } else if (INA260_IS_ATTR(vbusct)){ + if(ina260_read_register(cdata, INA260_REG_CONFIGURATION,&rvalue)) + return -1; + return sprintf(buf, "%s\n", ina260_ishcts_vbusct[(rvalue>>6)&0x7]); + } else if (INA260_IS_ATTR(aff)){ + if(ina260_read_register(cdata, INA260_REG_MASKENABLE,&rvalue)) + return -1; + return sprintf(buf, "%d\n", (rvalue>>4)&0x1); + } else if (INA260_IS_ATTR(cvrf)){ + if(ina260_read_register(cdata, INA260_REG_MASKENABLE,&rvalue)) + return -1; + return sprintf(buf, "%d\n", (rvalue>>3)&0x1); + } else if (INA260_IS_ATTR(ocl)){ + if(ina260_read_register(cdata, INA260_REG_MASKENABLE,&rvalue)) + return -1; + return sprintf(buf, "%d\n", (rvalue>>15)&0x1); + }else if (INA260_IS_ATTR(ucl)){ + if(ina260_read_register(cdata, INA260_REG_MASKENABLE,&rvalue)) + return -1; + return sprintf(buf, "%d\n", (rvalue>>14)&0x1); + }else if (INA260_IS_ATTR(bol)){ + if(ina260_read_register(cdata, INA260_REG_MASKENABLE,&rvalue)) + return -1; + return sprintf(buf, "%d\n", (rvalue>>13)&0x1); + }else if (INA260_IS_ATTR(bul)){ + if(ina260_read_register(cdata, INA260_REG_MASKENABLE,&rvalue)) + return -1; + return sprintf(buf, "%d\n", (rvalue>>12)&0x1); + }else if (INA260_IS_ATTR(pol)){ + if(ina260_read_register(cdata, INA260_REG_MASKENABLE,&rvalue)) + return -1; + return sprintf(buf, "%d\n", (rvalue>>11)&0x1); + }else if (INA260_IS_ATTR(cnvr)){ + if(ina260_read_register(cdata, INA260_REG_MASKENABLE,&rvalue)) + return -1; + return sprintf(buf, "%d\n", (rvalue>>10)&0x1); + }else if (INA260_IS_ATTR(ovf)){ + if(ina260_read_register(cdata, INA260_REG_MASKENABLE,&rvalue)) + return -1; + return sprintf(buf, "%d\n", (rvalue>>2)&0x1); + }else if (INA260_IS_ATTR(apol)){ + if(ina260_read_register(cdata, INA260_REG_MASKENABLE,&rvalue)) + return -1; + return sprintf(buf, "%d\n", (rvalue>>1)&0x1); + }else if (INA260_IS_ATTR(len)){ + if(ina260_read_register(cdata, INA260_REG_MASKENABLE,&rvalue)) + return -1; + return sprintf(buf, "%d\n", rvalue&0x1); + }else if (INA260_IS_ATTR(did)){ + if(ina260_read_register(cdata, INA260_REG_DIE,&rvalue)) + return -1; + return sprintf(buf, "0x%x\n", (rvalue>>4)); + }else if (INA260_IS_ATTR(rid)){ + if(ina260_read_register(cdata, INA260_REG_DIE,&rvalue)) + return -1; + return sprintf(buf, "0x%x\n", rvalue&0x7); + }else if (INA260_IS_ATTR(reset)){ + return sprintf(buf, "0\n"); + } + return 0; +} + +static int ina260_write_field3(struct client_data *cdata, unsigned char reg, unsigned char n, int uvalue){ + int rvalue; + if(!(uvalue>=0 && uvalue <8)) + return -EINVAL; + // Fetch register value: + if(ina260_read_register(cdata,reg,&rvalue)) + return 1; + // Write bits: + rvalue &= ~(0x7 << n); // clear bits + rvalue |= (uvalue << n); + // Write register value: + if(ina260_write_register(cdata,reg,rvalue)) + return 1; + return 0; +} + +static int ina260_write_field1(struct client_data *cdata, unsigned char reg, unsigned char n, int uvalue){ + int rvalue; + if(!(uvalue==0 || uvalue == 1)) + return -EINVAL; + // Fetch register value: + if(ina260_read_register(cdata,reg,&rvalue)) + return 1; + // Set bit: + rvalue &= ~(1<< n); // clear bit + rvalue |= (uvalue << n); + // Write register value + if(ina260_write_register(cdata,reg,rvalue)) + return 1; + return 0; +} + +static ssize_t attr_field_store(struct kobject *_kobj, + struct kobj_attribute *attr, + const char __user *buf, size_t count) +{ + int uvalue, ret=0; + struct client_data *cdata=container_of(_kobj,struct client_data,kobj); + // Extract user supplied value: + if(kstrtoint(buf, 0,&uvalue)) + return -EINVAL; + // Store: + if(INA260_IS_ATTR(reset) && uvalue!=0){ + ret=ina260_write_register(cdata, INA260_REG_CONFIGURATION, 0xFFFF); // Much quicker this way + } else if(INA260_IS_ATTR(avg)){ + ret=ina260_write_field3(cdata,INA260_REG_CONFIGURATION,9,uvalue); + } else if(INA260_IS_ATTR(mode)){ + ret=ina260_write_field3(cdata,INA260_REG_CONFIGURATION,0,uvalue); + } else if(INA260_IS_ATTR(ishct)){ + ret=ina260_write_field3(cdata,INA260_REG_CONFIGURATION,3,uvalue); + } else if(INA260_IS_ATTR(vbusct)){ + ret=ina260_write_field3(cdata,INA260_REG_CONFIGURATION,6,uvalue); + } else if(INA260_IS_ATTR(ocl)){ + ret=ina260_write_field1(cdata,INA260_REG_MASKENABLE,15,uvalue); + } else if(INA260_IS_ATTR(ucl)){ + ret=ina260_write_field1(cdata,INA260_REG_MASKENABLE,14,uvalue); + } else if(INA260_IS_ATTR(bol)){ + ret=ina260_write_field1(cdata,INA260_REG_MASKENABLE,13,uvalue); + } else if(INA260_IS_ATTR(bul)){ + ret=ina260_write_field1(cdata,INA260_REG_MASKENABLE,12,uvalue); + } else if(INA260_IS_ATTR(pol)){ + ret=ina260_write_field1(cdata,INA260_REG_MASKENABLE,11,uvalue); + } else if(INA260_IS_ATTR(cnvr)){ + ret=ina260_write_field1(cdata,INA260_REG_MASKENABLE,10,uvalue); + } else if(INA260_IS_ATTR(apol)){ + ret=ina260_write_field1(cdata,INA260_REG_MASKENABLE,1,uvalue); + } else if(INA260_IS_ATTR(len)){ + ret=ina260_write_field1(cdata,INA260_REG_MASKENABLE,0,uvalue); + } + return ret ? ret: count; +} + + +// ----- Registers ----- +static struct kobj_attribute configuration_attribute = + __ATTR(configuration, 0664, attr_show, attr_store); +static struct kobj_attribute current_attribute = { + .attr = {.name = "current", // current is a defined macro so need to do it manually + .mode = VERIFY_OCTAL_PERMISSIONS(0444) }, + .show = attr_show, + .store = attr_store, +}; +static struct kobj_attribute bus_voltage_attribute = + __ATTR(bus_voltage, 0444, attr_show, attr_store); +static struct kobj_attribute power_attribute = + __ATTR(power, 0444, attr_show, attr_store); +static struct kobj_attribute mask_enable_attribute = + __ATTR(mask_enable, 0664, attr_show, attr_store); +static struct kobj_attribute alert_limit_attribute = + __ATTR(alert_limit, 0664, attr_show, attr_store); +static struct kobj_attribute manufacturer_id_attribute = + __ATTR(manufacturer_id, 0664, attr_show, attr_store); +static struct kobj_attribute die_id_attribute= + __ATTR(die_id, 0664, attr_show, attr_store); +static struct attribute *registers_attrs[] = { + &configuration_attribute.attr, + ¤t_attribute.attr, + &bus_voltage_attribute.attr, + &power_attribute.attr, + &mask_enable_attribute.attr, + &alert_limit_attribute.attr, + &manufacturer_id_attribute.attr, + &die_id_attribute.attr, + NULL, +}; +static const struct attribute_group registers_group = { + .attrs = registers_attrs, + .name = "registers" +}; +// ----- Metrics ----- +static struct kobj_attribute metric_power_attribute = + __ATTR(power, 0444, attr_metric_show, NULL); +static struct kobj_attribute metric_voltage_attribute = + __ATTR(voltage, 0444, attr_metric_show, NULL); +static struct kobj_attribute metric_current_attribute = { + .attr = {.name = "current", // current is a defined macro so need to do it manually + .mode = VERIFY_OCTAL_PERMISSIONS(0444) }, + .show = attr_metric_show, + .store = NULL +}; +static struct attribute *metrics_attrs[] = { + &metric_power_attribute.attr, + &metric_voltage_attribute.attr, + &metric_current_attribute.attr, + NULL, +}; +static const struct attribute_group metrics_group = { + .attrs = metrics_attrs +}; +// ----- Fields ----- +static struct kobj_attribute reset_field_attribute = + __ATTR(reset, 0664, attr_field_show, attr_field_store); +static struct kobj_attribute avg_field_attribute = + __ATTR(avg, 0664, attr_field_show, attr_field_store); +static struct kobj_attribute mode_field_attribute = + __ATTR(mode, 0664, attr_field_show, attr_field_store); +static struct kobj_attribute ishct_field_attribute = + __ATTR(ishct, 0664, attr_field_show, attr_field_store); +static struct kobj_attribute vbusct_field_attribute = + __ATTR(vbusct, 0664, attr_field_show, attr_field_store); +static struct kobj_attribute aff_field_attribute = + __ATTR(aff, 0444, attr_field_show, attr_field_store); +static struct kobj_attribute cvrf_field_attribute = + __ATTR(cvrf, 0444, attr_field_show, attr_field_store); +static struct kobj_attribute ocl_field_attribute = + __ATTR(ocl, 0664, attr_field_show, attr_field_store); +static struct kobj_attribute ucl_field_attribute = + __ATTR(ucl, 0664, attr_field_show, attr_field_store); +static struct kobj_attribute bol_field_attribute = + __ATTR(bol, 0664, attr_field_show, attr_field_store); +static struct kobj_attribute bul_field_attribute = + __ATTR(bul, 0664, attr_field_show, attr_field_store); +static struct kobj_attribute pol_field_attribute = + __ATTR(pol, 0664, attr_field_show, attr_field_store); +static struct kobj_attribute cnvr_field_attribute = + __ATTR(cnvr, 0664, attr_field_show, attr_field_store); +static struct kobj_attribute ovf_field_attribute = + __ATTR(ovf, 0444, attr_field_show, attr_field_store); +static struct kobj_attribute apol_field_attribute = + __ATTR(apol, 0664, attr_field_show, attr_field_store); +static struct kobj_attribute len_field_attribute = + __ATTR(len, 0664, attr_field_show, attr_field_store); +static struct kobj_attribute did_field_attribute = + __ATTR(did, 0444, attr_field_show, attr_field_store); +static struct kobj_attribute rid_field_attribute = + __ATTR(rid, 0444, attr_field_show, attr_field_store); +static struct attribute *fields_attrs[] = { + &reset_field_attribute.attr, + &avg_field_attribute.attr, + &mode_field_attribute.attr, + &ishct_field_attribute.attr, + &vbusct_field_attribute.attr, + &aff_field_attribute.attr, + &cvrf_field_attribute.attr, + &ocl_field_attribute.attr, + &ucl_field_attribute.attr, + &bol_field_attribute.attr, + &bul_field_attribute.attr, + &pol_field_attribute.attr, + &cnvr_field_attribute.attr, + &ovf_field_attribute.attr, + &apol_field_attribute.attr, + &len_field_attribute.attr, + &did_field_attribute.attr, + &rid_field_attribute.attr, + NULL, +}; +static const struct attribute_group fields_group = { + .attrs = fields_attrs, + .name = "fields" +}; + + +static struct kobj_type ina260_ktype = { + .sysfs_ops = &kobj_sysfs_ops, +}; + +/** + * @brief Compare on ina260 register to a supplied value + * + * @param client i2c client to use for communications + * @param reg register to use for comparison + * @param value register expected value + * @return int 0 if register contains value and 1 otherwise + */ +static int ina260_probe_register(struct i2c_client *client, unsigned char reg, int value){ + unsigned char bytes[2]; + if(i2c_master_send(client,®,1)<0) + return 1; + if(i2c_master_recv(client,bytes,2)<0) + return 1; + return ((bytes[0]<<8) | bytes[1])!=value; +} + +static int ina260_probe_new(struct i2c_client *client){ + struct client_data *p; + // Attempt device discovery: + if(ina260_probe_register(client,INA260_REG_MANUFACTURER,0x5449) || + ina260_probe_register(client,INA260_REG_DIE,0x2270)){ + printk("ina260 probe fails bus=%d address=0x%02x\n",client->adapter->nr,client->addr); + return 1; + } + // Initialize client data: + printk("ina260 detected bus=%d address=0x%02x\n",client->adapter->nr,client->addr); + p=kzalloc(sizeof(struct client_data),GFP_KERNEL); + kobject_init(&p->kobj,&ina260_ktype); + // Setup sysfs root kobject: + if(kobject_add(&p->kobj,ina260_kobj,client->dev.kobj.name)){ + kobject_put(&p->kobj); + kfree(p); + return 1; + } + p->client=client; + p->reg=INA260_REG_DIE; // Maintain cache coherence + i2c_set_clientdata(client,p); + // Setup sysfs groups: + if(sysfs_create_group(&p->kobj,®isters_group)|| + sysfs_create_group(&p->kobj,&metrics_group)|| + sysfs_create_group(&p->kobj,&fields_group)){ + kobject_put(&p->kobj); + kfree(p); + return 1; + } + return 0; +} + +static int ina260_remove(struct i2c_client *client){ + struct client_data *p=i2c_get_clientdata(client); + kobject_put(&p->kobj); + kfree(p); + printk("ina260 removed bus=%d address=0x%02x\n",client->adapter->nr,client->addr); + return 0; +} + +static struct i2c_driver ina260_driver = { + .driver = { + .name = "ina260" + }, + .probe_new = ina260_probe_new, + .remove = ina260_remove, + .id_table = ina260_ids +}; + +static int __init ina260_init(void){ + ina260_kobj = kobject_create_and_add("ina260",kernel_kobj); + i2c_add_driver(&ina260_driver); + return 0; +} + +static void __exit ina260_exit(void){ + i2c_del_driver(&ina260_driver); + kobject_put(ina260_kobj); +} + +module_init(ina260_init); +module_exit(ina260_exit); +MODULE_LICENSE("GPL"); +MODULE_AUTHOR("Loïc Guegan"); +MODULE_DESCRIPTION("INA260 Texas Instruments"); +MODULE_VERSION("1.0"); |
