cyapa_gen3.c

   1 /*
   2  * Cypress APA trackpad with I2C interface
   3  *
   4  * Author: Dudley Du <dudl@cypress.com>
   5  * Further cleanup and restructuring by:
   6  *   Daniel Kurtz <djkurtz@chromium.org>
   7  *   Benson Leung <bleung@chromium.org>
   8  *
   9  * Copyright (C) 2011-2015 Cypress Semiconductor, Inc.
  10  * Copyright (C) 2011-2012 Google, Inc.
  11  *
  12  * This file is subject to the terms and conditions of the GNU General Public
  13  * License.  See the file COPYING in the main directory of this archive for
  14  * more details.
  15  */
  16
  17 #include <linux/delay.h>
  18 #include <linux/i2c.h>
  19 #include <linux/input.h>
  20 #include <linux/input/mt.h>
  21 #include <linux/module.h>
  22 #include <linux/slab.h>
  23 #include <asm/unaligned.h>
  24 #include "cyapa.h"
  25
  26
  27 #define GEN3_MAX_FINGERS 5
  28 #define GEN3_FINGER_NUM(x) (((x) >> 4) & 0x07)
  29
  30 #define BLK_HEAD_BYTES 32
  31
  32 /* Macro for register map group offset. */
  33 #define PRODUCT_ID_SIZE  16
  34 #define QUERY_DATA_SIZE  27
  35 #define REG_PROTOCOL_GEN_QUERY_OFFSET  20
  36
  37 #define REG_OFFSET_DATA_BASE     0x0000
  38 #define REG_OFFSET_COMMAND_BASE  0x0028
  39 #define REG_OFFSET_QUERY_BASE    0x002a
  40
  41 #define CYAPA_OFFSET_SOFT_RESET  REG_OFFSET_COMMAND_BASE
  42 #define OP_RECALIBRATION_MASK    0x80
  43 #define OP_REPORT_BASELINE_MASK  0x40
  44 #define REG_OFFSET_MAX_BASELINE  0x0026
  45 #define REG_OFFSET_MIN_BASELINE  0x0027
  46
  47 #define REG_OFFSET_POWER_MODE (REG_OFFSET_COMMAND_BASE + 1)
  48 #define SET_POWER_MODE_DELAY   10000  /* Unit: us */
  49 #define SET_POWER_MODE_TRIES   5
  50
  51 #define GEN3_BL_CMD_CHECKSUM_SEED 0xff
  52 #define GEN3_BL_CMD_INITIATE_BL   0x38
  53 #define GEN3_BL_CMD_WRITE_BLOCK   0x39
  54 #define GEN3_BL_CMD_VERIFY_BLOCK  0x3a
  55 #define GEN3_BL_CMD_TERMINATE_BL  0x3b
  56 #define GEN3_BL_CMD_LAUNCH_APP    0xa5
  57
  58 /*
  59  * CYAPA trackpad device states.
  60  * Used in register 0x00, bit1-0, DeviceStatus field.
  61  * Other values indicate device is in an abnormal state and must be reset.
  62  */
  63 #define CYAPA_DEV_NORMAL  0x03
  64 #define CYAPA_DEV_BUSY    0x01
  65
  66 #define CYAPA_FW_BLOCK_SIZE     64
  67 #define CYAPA_FW_READ_SIZE      16
  68 #define CYAPA_FW_HDR_START      0x0780
  69 #define CYAPA_FW_HDR_BLOCK_COUNT  2
  70 #define CYAPA_FW_HDR_BLOCK_START  (CYAPA_FW_HDR_START / CYAPA_FW_BLOCK_SIZE)
  71 #define CYAPA_FW_HDR_SIZE         (CYAPA_FW_HDR_BLOCK_COUNT * \
  72                                         CYAPA_FW_BLOCK_SIZE)
  73 #define CYAPA_FW_DATA_START     0x0800
  74 #define CYAPA_FW_DATA_BLOCK_COUNT  480
  75 #define CYAPA_FW_DATA_BLOCK_START  (CYAPA_FW_DATA_START / CYAPA_FW_BLOCK_SIZE)
  76 #define CYAPA_FW_DATA_SIZE      (CYAPA_FW_DATA_BLOCK_COUNT * \
  77                                  CYAPA_FW_BLOCK_SIZE)
  78 #define CYAPA_FW_SIZE           (CYAPA_FW_HDR_SIZE + CYAPA_FW_DATA_SIZE)
  79 #define CYAPA_CMD_LEN           16
  80
  81 #define GEN3_BL_IDLE_FW_MAJ_VER_OFFSET 0x0b
  82 #define GEN3_BL_IDLE_FW_MIN_VER_OFFSET (GEN3_BL_IDLE_FW_MAJ_VER_OFFSET + 1)
  83
  84
  85 struct cyapa_touch {
  86         /*
  87          * high bits or x/y position value
  88          * bit 7 - 4: high 4 bits of x position value
  89          * bit 3 - 0: high 4 bits of y position value
  90          */
  91         u8 xy_hi;
  92         u8 x_lo;  /* low 8 bits of x position value. */
  93         u8 y_lo;  /* low 8 bits of y position value. */
  94         u8 pressure;
  95         /* id range is 1 - 15.  It is incremented with every new touch. */
  96         u8 id;
  97 } __packed;
  98
  99 struct cyapa_reg_data {
 100         /*
 101          * bit 0 - 1: device status
 102          * bit 3 - 2: power mode
 103          * bit 6 - 4: reserved
 104          * bit 7: interrupt valid bit
 105          */
 106         u8 device_status;
 107         /*
 108          * bit 7 - 4: number of fingers currently touching pad
 109          * bit 3: valid data check bit
 110          * bit 2: middle mechanism button state if exists
 111          * bit 1: right mechanism button state if exists
 112          * bit 0: left mechanism button state if exists
 113          */
 114         u8 finger_btn;
 115         /* CYAPA reports up to 5 touches per packet. */
 116         struct cyapa_touch touches[5];
 117 } __packed;
 118
 119 struct gen3_write_block_cmd {
 120         u8 checksum_seed;  /* Always be 0xff */
 121         u8 cmd_code;       /* command code: 0x39 */
 122         u8 key[8];         /* 8-byte security key */
 123         __be16 block_num;
 124         u8 block_data[CYAPA_FW_BLOCK_SIZE];
 125         u8 block_checksum;  /* Calculated using bytes 12 - 75 */
 126         u8 cmd_checksum;    /* Calculated using bytes 0-76 */
 127 } __packed;
 128
 129 static const u8 security_key[] = {
 130                 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07 };
 131 static const u8 bl_activate[] = { 0x00, 0xff, 0x38, 0x00, 0x01, 0x02, 0x03,
 132                 0x04, 0x05, 0x06, 0x07 };
 133 static const u8 bl_deactivate[] = { 0x00, 0xff, 0x3b, 0x00, 0x01, 0x02, 0x03,
 134                 0x04, 0x05, 0x06, 0x07 };
 135 static const u8 bl_exit[] = { 0x00, 0xff, 0xa5, 0x00, 0x01, 0x02, 0x03, 0x04,
 136                 0x05, 0x06, 0x07 };
 137
 138
 139  /* for byte read/write command */
 140 #define CMD_RESET      0
 141 #define CMD_POWER_MODE 1
 142 #define CMD_DEV_STATUS 2
 143 #define CMD_REPORT_MAX_BASELINE 3
 144 #define CMD_REPORT_MIN_BASELINE 4
 145 #define SMBUS_BYTE_CMD(cmd) (((cmd) & 0x3f) << 1)
 146 #define CYAPA_SMBUS_RESET         SMBUS_BYTE_CMD(CMD_RESET)
 147 #define CYAPA_SMBUS_POWER_MODE    SMBUS_BYTE_CMD(CMD_POWER_MODE)
 148 #define CYAPA_SMBUS_DEV_STATUS    SMBUS_BYTE_CMD(CMD_DEV_STATUS)
 149 #define CYAPA_SMBUS_MAX_BASELINE  SMBUS_BYTE_CMD(CMD_REPORT_MAX_BASELINE)
 150 #define CYAPA_SMBUS_MIN_BASELINE  SMBUS_BYTE_CMD(CMD_REPORT_MIN_BASELINE)
 151
 152  /* for group registers read/write command */
 153 #define REG_GROUP_DATA 0
 154 #define REG_GROUP_CMD 2
 155 #define REG_GROUP_QUERY 3
 156 #define SMBUS_GROUP_CMD(grp) (0x80 | (((grp) & 0x07) << 3))
 157 #define CYAPA_SMBUS_GROUP_DATA   SMBUS_GROUP_CMD(REG_GROUP_DATA)
 158 #define CYAPA_SMBUS_GROUP_CMD    SMBUS_GROUP_CMD(REG_GROUP_CMD)
 159 #define CYAPA_SMBUS_GROUP_QUERY  SMBUS_GROUP_CMD(REG_GROUP_QUERY)
 160
 161  /* for register block read/write command */
 162 #define CMD_BL_STATUS 0
 163 #define CMD_BL_HEAD 1
 164 #define CMD_BL_CMD 2
 165 #define CMD_BL_DATA 3
 166 #define CMD_BL_ALL 4
 167 #define CMD_BLK_PRODUCT_ID 5
 168 #define CMD_BLK_HEAD 6
 169 #define SMBUS_BLOCK_CMD(cmd) (0xc0 | (((cmd) & 0x1f) << 1))
 170
 171 /* register block read/write command in bootloader mode */
 172 #define CYAPA_SMBUS_BL_STATUS  SMBUS_BLOCK_CMD(CMD_BL_STATUS)
 173 #define CYAPA_SMBUS_BL_HEAD    SMBUS_BLOCK_CMD(CMD_BL_HEAD)
 174 #define CYAPA_SMBUS_BL_CMD     SMBUS_BLOCK_CMD(CMD_BL_CMD)
 175 #define CYAPA_SMBUS_BL_DATA    SMBUS_BLOCK_CMD(CMD_BL_DATA)
 176 #define CYAPA_SMBUS_BL_ALL     SMBUS_BLOCK_CMD(CMD_BL_ALL)
 177
 178 /* register block read/write command in operational mode */
 179 #define CYAPA_SMBUS_BLK_PRODUCT_ID SMBUS_BLOCK_CMD(CMD_BLK_PRODUCT_ID)
 180 #define CYAPA_SMBUS_BLK_HEAD SMBUS_BLOCK_CMD(CMD_BLK_HEAD)
 181
 182  /* for byte read/write command */
 183 #define CMD_RESET 0
 184 #define CMD_POWER_MODE 1
 185 #define CMD_DEV_STATUS 2
 186 #define CMD_REPORT_MAX_BASELINE 3
 187 #define CMD_REPORT_MIN_BASELINE 4
 188 #define SMBUS_BYTE_CMD(cmd) (((cmd) & 0x3f) << 1)
 189 #define CYAPA_SMBUS_RESET         SMBUS_BYTE_CMD(CMD_RESET)
 190 #define CYAPA_SMBUS_POWER_MODE    SMBUS_BYTE_CMD(CMD_POWER_MODE)
 191 #define CYAPA_SMBUS_DEV_STATUS    SMBUS_BYTE_CMD(CMD_DEV_STATUS)
 192 #define CYAPA_SMBUS_MAX_BASELINE  SMBUS_BYTE_CMD(CMD_REPORT_MAX_BASELINE)
 193 #define CYAPA_SMBUS_MIN_BASELINE  SMBUS_BYTE_CMD(CMD_REPORT_MIN_BASELINE)
 194
 195  /* for group registers read/write command */
 196 #define REG_GROUP_DATA  0
 197 #define REG_GROUP_CMD   2
 198 #define REG_GROUP_QUERY 3
 199 #define SMBUS_GROUP_CMD(grp) (0x80 | (((grp) & 0x07) << 3))
 200 #define CYAPA_SMBUS_GROUP_DATA  SMBUS_GROUP_CMD(REG_GROUP_DATA)
 201 #define CYAPA_SMBUS_GROUP_CMD   SMBUS_GROUP_CMD(REG_GROUP_CMD)
 202 #define CYAPA_SMBUS_GROUP_QUERY SMBUS_GROUP_CMD(REG_GROUP_QUERY)
 203
 204  /* for register block read/write command */
 205 #define CMD_BL_STATUS           0
 206 #define CMD_BL_HEAD             1
 207 #define CMD_BL_CMD              2
 208 #define CMD_BL_DATA             3
 209 #define CMD_BL_ALL              4
 210 #define CMD_BLK_PRODUCT_ID      5
 211 #define CMD_BLK_HEAD            6
 212 #define SMBUS_BLOCK_CMD(cmd) (0xc0 | (((cmd) & 0x1f) << 1))
 213
 214 /* register block read/write command in bootloader mode */
 215 #define CYAPA_SMBUS_BL_STATUS SMBUS_BLOCK_CMD(CMD_BL_STATUS)
 216 #define CYAPA_SMBUS_BL_HEAD   SMBUS_BLOCK_CMD(CMD_BL_HEAD)
 217 #define CYAPA_SMBUS_BL_CMD    SMBUS_BLOCK_CMD(CMD_BL_CMD)
 218 #define CYAPA_SMBUS_BL_DATA   SMBUS_BLOCK_CMD(CMD_BL_DATA)
 219 #define CYAPA_SMBUS_BL_ALL    SMBUS_BLOCK_CMD(CMD_BL_ALL)
 220
 221 /* register block read/write command in operational mode */
 222 #define CYAPA_SMBUS_BLK_PRODUCT_ID SMBUS_BLOCK_CMD(CMD_BLK_PRODUCT_ID)
 223 #define CYAPA_SMBUS_BLK_HEAD       SMBUS_BLOCK_CMD(CMD_BLK_HEAD)
 224
 225 struct cyapa_cmd_len {
 226         u8 cmd;
 227         u8 len;
 228 };
 229
 230 /* maps generic CYAPA_CMD_* code to the I2C equivalent */
 231 static const struct cyapa_cmd_len cyapa_i2c_cmds[] = {
 232         { CYAPA_OFFSET_SOFT_RESET, 1 },         /* CYAPA_CMD_SOFT_RESET */
 233         { REG_OFFSET_COMMAND_BASE + 1, 1 },     /* CYAPA_CMD_POWER_MODE */
 234         { REG_OFFSET_DATA_BASE, 1 },            /* CYAPA_CMD_DEV_STATUS */
 235         { REG_OFFSET_DATA_BASE, sizeof(struct cyapa_reg_data) },
 236                                                 /* CYAPA_CMD_GROUP_DATA */
 237         { REG_OFFSET_COMMAND_BASE, 0 },         /* CYAPA_CMD_GROUP_CMD */
 238         { REG_OFFSET_QUERY_BASE, QUERY_DATA_SIZE }, /* CYAPA_CMD_GROUP_QUERY */
 239         { BL_HEAD_OFFSET, 3 },                  /* CYAPA_CMD_BL_STATUS */
 240         { BL_HEAD_OFFSET, 16 },                 /* CYAPA_CMD_BL_HEAD */
 241         { BL_HEAD_OFFSET, 16 },                 /* CYAPA_CMD_BL_CMD */
 242         { BL_DATA_OFFSET, 16 },                 /* CYAPA_CMD_BL_DATA */
 243         { BL_HEAD_OFFSET, 32 },                 /* CYAPA_CMD_BL_ALL */
 244         { REG_OFFSET_QUERY_BASE, PRODUCT_ID_SIZE },
 245                                                 /* CYAPA_CMD_BLK_PRODUCT_ID */
 246         { REG_OFFSET_DATA_BASE, 32 },           /* CYAPA_CMD_BLK_HEAD */
 247         { REG_OFFSET_MAX_BASELINE, 1 },         /* CYAPA_CMD_MAX_BASELINE */
 248         { REG_OFFSET_MIN_BASELINE, 1 },         /* CYAPA_CMD_MIN_BASELINE */
 249 };
 250
 251 static const struct cyapa_cmd_len cyapa_smbus_cmds[] = {
 252         { CYAPA_SMBUS_RESET, 1 },               /* CYAPA_CMD_SOFT_RESET */
 253         { CYAPA_SMBUS_POWER_MODE, 1 },          /* CYAPA_CMD_POWER_MODE */
 254         { CYAPA_SMBUS_DEV_STATUS, 1 },          /* CYAPA_CMD_DEV_STATUS */
 255         { CYAPA_SMBUS_GROUP_DATA, sizeof(struct cyapa_reg_data) },
 256                                                 /* CYAPA_CMD_GROUP_DATA */
 257         { CYAPA_SMBUS_GROUP_CMD, 2 },           /* CYAPA_CMD_GROUP_CMD */
 258         { CYAPA_SMBUS_GROUP_QUERY, QUERY_DATA_SIZE },
 259                                                 /* CYAPA_CMD_GROUP_QUERY */
 260         { CYAPA_SMBUS_BL_STATUS, 3 },           /* CYAPA_CMD_BL_STATUS */
 261         { CYAPA_SMBUS_BL_HEAD, 16 },            /* CYAPA_CMD_BL_HEAD */
 262         { CYAPA_SMBUS_BL_CMD, 16 },             /* CYAPA_CMD_BL_CMD */
 263         { CYAPA_SMBUS_BL_DATA, 16 },            /* CYAPA_CMD_BL_DATA */
 264         { CYAPA_SMBUS_BL_ALL, 32 },             /* CYAPA_CMD_BL_ALL */
 265         { CYAPA_SMBUS_BLK_PRODUCT_ID, PRODUCT_ID_SIZE },
 266                                                 /* CYAPA_CMD_BLK_PRODUCT_ID */
 267         { CYAPA_SMBUS_BLK_HEAD, 16 },           /* CYAPA_CMD_BLK_HEAD */
 268         { CYAPA_SMBUS_MAX_BASELINE, 1 },        /* CYAPA_CMD_MAX_BASELINE */
 269         { CYAPA_SMBUS_MIN_BASELINE, 1 },        /* CYAPA_CMD_MIN_BASELINE */
 270 };
 271
 272 static int cyapa_gen3_try_poll_handler(struct cyapa *cyapa);
 273
 274 /*
 275  * cyapa_smbus_read_block - perform smbus block read command
 276  * @cyapa  - private data structure of the driver
 277  * @cmd    - the properly encoded smbus command
 278  * @len    - expected length of smbus command result
 279  * @values - buffer to store smbus command result
 280  *
 281  * Returns negative errno, else the number of bytes written.
 282  *
 283  * Note:
 284  * In trackpad device, the memory block allocated for I2C register map
 285  * is 256 bytes, so the max read block for I2C bus is 256 bytes.
 286  */
 287 ssize_t cyapa_smbus_read_block(struct cyapa *cyapa, u8 cmd, size_t len,
 288                                       u8 *values)
 289 {
 290         ssize_t ret;
 291         u8 index;
 292         u8 smbus_cmd;
 293         u8 *buf;
 294         struct i2c_client *client = cyapa->client;
 295
 296         if (!(SMBUS_BYTE_BLOCK_CMD_MASK & cmd))
 297                 return -EINVAL;
 298
 299         if (SMBUS_GROUP_BLOCK_CMD_MASK & cmd) {
 300                 /* read specific block registers command. */
 301                 smbus_cmd = SMBUS_ENCODE_RW(cmd, SMBUS_READ);
 302                 ret = i2c_smbus_read_block_data(client, smbus_cmd, values);
 303                 goto out;
 304         }
 305
 306         ret = 0;
 307         for (index = 0; index * I2C_SMBUS_BLOCK_MAX < len; index++) {
 308                 smbus_cmd = SMBUS_ENCODE_IDX(cmd, index);
 309                 smbus_cmd = SMBUS_ENCODE_RW(smbus_cmd, SMBUS_READ);
 310                 buf = values + I2C_SMBUS_BLOCK_MAX * index;
 311                 ret = i2c_smbus_read_block_data(client, smbus_cmd, buf);
 312                 if (ret < 0)
 313                         goto out;
 314         }
 315
 316 out:
 317         return ret > 0 ? len : ret;
 318 }
 319
 320 static s32 cyapa_read_byte(struct cyapa *cyapa, u8 cmd_idx)
 321 {
 322         u8 cmd;
 323
 324         if (cyapa->smbus) {
 325                 cmd = cyapa_smbus_cmds[cmd_idx].cmd;
 326                 cmd = SMBUS_ENCODE_RW(cmd, SMBUS_READ);
 327         } else {
 328                 cmd = cyapa_i2c_cmds[cmd_idx].cmd;
 329         }
 330         return i2c_smbus_read_byte_data(cyapa->client, cmd);
 331 }
 332
 333 static s32 cyapa_write_byte(struct cyapa *cyapa, u8 cmd_idx, u8 value)
 334 {
 335         u8 cmd;
 336
 337         if (cyapa->smbus) {
 338                 cmd = cyapa_smbus_cmds[cmd_idx].cmd;
 339                 cmd = SMBUS_ENCODE_RW(cmd, SMBUS_WRITE);
 340         } else {
 341                 cmd = cyapa_i2c_cmds[cmd_idx].cmd;
 342         }
 343         return i2c_smbus_write_byte_data(cyapa->client, cmd, value);
 344 }
 345
 346 ssize_t cyapa_i2c_reg_read_block(struct cyapa *cyapa, u8 reg, size_t len,
 347                                         u8 *values)
 348 {
 349         return i2c_smbus_read_i2c_block_data(cyapa->client, reg, len, values);
 350 }
 351
 352 static ssize_t cyapa_i2c_reg_write_block(struct cyapa *cyapa, u8 reg,
 353                                          size_t len, const u8 *values)
 354 {
 355         return i2c_smbus_write_i2c_block_data(cyapa->client, reg, len, values);
 356 }
 357
 358 ssize_t cyapa_read_block(struct cyapa *cyapa, u8 cmd_idx, u8 *values)
 359 {
 360         u8 cmd;
 361         size_t len;
 362
 363         if (cyapa->smbus) {
 364                 cmd = cyapa_smbus_cmds[cmd_idx].cmd;
 365                 len = cyapa_smbus_cmds[cmd_idx].len;
 366                 return cyapa_smbus_read_block(cyapa, cmd, len, values);
 367         }
 368         cmd = cyapa_i2c_cmds[cmd_idx].cmd;
 369         len = cyapa_i2c_cmds[cmd_idx].len;
 370         return cyapa_i2c_reg_read_block(cyapa, cmd, len, values);
 371 }
 372
 373 /*
 374  * Determine the Gen3 trackpad device's current operating state.
 375  *
 376  */
 377 static int cyapa_gen3_state_parse(struct cyapa *cyapa, u8 *reg_data, int len)
 378 {
 379         cyapa->state = CYAPA_STATE_NO_DEVICE;
 380
 381         /* Parse based on Gen3 characteristic registers and bits */
 382         if (reg_data[REG_BL_FILE] == BL_FILE &&
 383                 reg_data[REG_BL_ERROR] == BL_ERROR_NO_ERR_IDLE &&
 384                 (reg_data[REG_BL_STATUS] ==
 385                         (BL_STATUS_RUNNING | BL_STATUS_CSUM_VALID) ||
 386                         reg_data[REG_BL_STATUS] == BL_STATUS_RUNNING)) {
 387                 /*
 388                  * Normal state after power on or reset,
 389                  * REG_BL_STATUS == 0x11, firmware image checksum is valid.
 390                  * REG_BL_STATUS == 0x10, firmware image checksum is invalid.
 391                  */
 392                 cyapa->gen = CYAPA_GEN3;
 393                 cyapa->state = CYAPA_STATE_BL_IDLE;
 394         } else if (reg_data[REG_BL_FILE] == BL_FILE &&
 395                 (reg_data[REG_BL_STATUS] & BL_STATUS_RUNNING) ==
 396                         BL_STATUS_RUNNING) {
 397                 cyapa->gen = CYAPA_GEN3;
 398                 if (reg_data[REG_BL_STATUS] & BL_STATUS_BUSY) {
 399                         cyapa->state = CYAPA_STATE_BL_BUSY;
 400                 } else {
 401                         if ((reg_data[REG_BL_ERROR] & BL_ERROR_BOOTLOADING) ==
 402                                         BL_ERROR_BOOTLOADING)
 403                                 cyapa->state = CYAPA_STATE_BL_ACTIVE;
 404                         else
 405                                 cyapa->state = CYAPA_STATE_BL_IDLE;
 406                 }
 407         } else if ((reg_data[REG_OP_STATUS] & OP_STATUS_SRC) &&
 408                         (reg_data[REG_OP_DATA1] & OP_DATA_VALID)) {
 409                 /*
 410                  * Normal state when running in operational mode,
 411                  * may also not in full power state or
 412                  * busying in command process.
 413                  */
 414                 if (GEN3_FINGER_NUM(reg_data[REG_OP_DATA1]) <=
 415                                 GEN3_MAX_FINGERS) {
 416                         /* Finger number data is valid. */
 417                         cyapa->gen = CYAPA_GEN3;
 418                         cyapa->state = CYAPA_STATE_OP;
 419                 }
 420         } else if (reg_data[REG_OP_STATUS] == 0x0C &&
 421                         reg_data[REG_OP_DATA1] == 0x08) {
 422                 /* Op state when first two registers overwritten with 0x00 */
 423                 cyapa->gen = CYAPA_GEN3;
 424                 cyapa->state = CYAPA_STATE_OP;
 425         } else if (reg_data[REG_BL_STATUS] &
 426                         (BL_STATUS_RUNNING | BL_STATUS_BUSY)) {
 427                 cyapa->gen = CYAPA_GEN3;
 428                 cyapa->state = CYAPA_STATE_BL_BUSY;
 429         }
 430
 431         if (cyapa->gen == CYAPA_GEN3 && (cyapa->state == CYAPA_STATE_OP ||
 432                 cyapa->state == CYAPA_STATE_BL_IDLE ||
 433                 cyapa->state == CYAPA_STATE_BL_ACTIVE ||
 434                 cyapa->state == CYAPA_STATE_BL_BUSY))
 435                 return 0;
 436
 437         return -EAGAIN;
 438 }
 439
 440 /*
 441  * Enter bootloader by soft resetting the device.
 442  *
 443  * If device is already in the bootloader, the function just returns.
 444  * Otherwise, reset the device; after reset, device enters bootloader idle
 445  * state immediately.
 446  *
 447  * Returns:
 448  *   0        on success
 449  *   -EAGAIN  device was reset, but is not now in bootloader idle state
 450  *   < 0      if the device never responds within the timeout
 451  */
 452 static int cyapa_gen3_bl_enter(struct cyapa *cyapa)
 453 {
 454         int error;
 455         int waiting_time;
 456
 457         error = cyapa_poll_state(cyapa, 500);
 458         if (error)
 459                 return error;
 460         if (cyapa->state == CYAPA_STATE_BL_IDLE) {
 461                 /* Already in BL_IDLE. Skipping reset. */
 462                 return 0;
 463         }
 464
 465         if (cyapa->state != CYAPA_STATE_OP)
 466                 return -EAGAIN;
 467
 468         cyapa->operational = false;
 469         cyapa->state = CYAPA_STATE_NO_DEVICE;
 470         error = cyapa_write_byte(cyapa, CYAPA_CMD_SOFT_RESET, 0x01);
 471         if (error)
 472                 return -EIO;
 473
 474         usleep_range(25000, 50000);
 475         waiting_time = 2000;  /* For some shipset, max waiting time is 1~2s. */
 476         do {
 477                 error = cyapa_poll_state(cyapa, 500);
 478                 if (error) {
 479                         if (error == -ETIMEDOUT) {
 480                                 waiting_time -= 500;
 481                                 continue;
 482                         }
 483                         return error;
 484                 }
 485
 486                 if ((cyapa->state == CYAPA_STATE_BL_IDLE) &&
 487                         !(cyapa->status[REG_BL_STATUS] & BL_STATUS_WATCHDOG))
 488                         break;
 489
 490                 msleep(100);
 491                 waiting_time -= 100;
 492         } while (waiting_time > 0);
 493
 494         if ((cyapa->state != CYAPA_STATE_BL_IDLE) ||
 495                 (cyapa->status[REG_BL_STATUS] & BL_STATUS_WATCHDOG))
 496                 return -EAGAIN;
 497
 498         return 0;
 499 }
 500
 501 static int cyapa_gen3_bl_activate(struct cyapa *cyapa)
 502 {
 503         int error;
 504
 505         error = cyapa_i2c_reg_write_block(cyapa, 0, sizeof(bl_activate),
 506                                         bl_activate);
 507         if (error)
 508                 return error;
 509
 510         /* Wait for bootloader to activate; takes between 2 and 12 seconds */
 511         msleep(2000);
 512         error = cyapa_poll_state(cyapa, 11000);
 513         if (error)
 514                 return error;
 515         if (cyapa->state != CYAPA_STATE_BL_ACTIVE)
 516                 return -EAGAIN;
 517
 518         return 0;
 519 }
 520
 521 static int cyapa_gen3_bl_deactivate(struct cyapa *cyapa)
 522 {
 523         int error;
 524
 525         error = cyapa_i2c_reg_write_block(cyapa, 0, sizeof(bl_deactivate),
 526                                         bl_deactivate);
 527         if (error)
 528                 return error;
 529
 530         /* Wait for bootloader to switch to idle state; should take < 100ms */
 531         msleep(100);
 532         error = cyapa_poll_state(cyapa, 500);
 533         if (error)
 534                 return error;
 535         if (cyapa->state != CYAPA_STATE_BL_IDLE)
 536                 return -EAGAIN;
 537         return 0;
 538 }
 539
 540 /*
 541  * Exit bootloader
 542  *
 543  * Send bl_exit command, then wait 50 - 100 ms to let device transition to
 544  * operational mode.  If this is the first time the device's firmware is
 545  * running, it can take up to 2 seconds to calibrate its sensors.  So, poll
 546  * the device's new state for up to 2 seconds.
 547  *
 548  * Returns:
 549  *   -EIO    failure while reading from device
 550  *   -EAGAIN device is stuck in bootloader, b/c it has invalid firmware
 551  *   0       device is supported and in operational mode
 552  */
 553 static int cyapa_gen3_bl_exit(struct cyapa *cyapa)
 554 {
 555         int error;
 556
 557         error = cyapa_i2c_reg_write_block(cyapa, 0, sizeof(bl_exit), bl_exit);
 558         if (error)
 559                 return error;
 560
 561         /*
 562          * Wait for bootloader to exit, and operation mode to start.
 563          * Normally, this takes at least 50 ms.
 564          */
 565         usleep_range(50000, 100000);
 566         /*
 567          * In addition, when a device boots for the first time after being
 568          * updated to new firmware, it must first calibrate its sensors, which
 569          * can take up to an additional 2 seconds. If the device power is
 570          * running low, this may take even longer.
 571          */
 572         error = cyapa_poll_state(cyapa, 4000);
 573         if (error < 0)
 574                 return error;
 575         if (cyapa->state != CYAPA_STATE_OP)
 576                 return -EAGAIN;
 577
 578         return 0;
 579 }
 580
 581 static u16 cyapa_gen3_csum(const u8 *buf, size_t count)
 582 {
 583         int i;
 584         u16 csum = 0;
 585
 586         for (i = 0; i < count; i++)
 587                 csum += buf[i];
 588
 589         return csum;
 590 }
 591
 592 /*
 593  * Verify the integrity of a CYAPA firmware image file.
 594  *
 595  * The firmware image file is 30848 bytes, composed of 482 64-byte blocks.
 596  *
 597  * The first 2 blocks are the firmware header.
 598  * The next 480 blocks are the firmware image.
 599  *
 600  * The first two bytes of the header hold the header checksum, computed by
 601  * summing the other 126 bytes of the header.
 602  * The last two bytes of the header hold the firmware image checksum, computed
 603  * by summing the 30720 bytes of the image modulo 0xffff.
 604  *
 605  * Both checksums are stored little-endian.
 606  */
 607 static int cyapa_gen3_check_fw(struct cyapa *cyapa, const struct firmware *fw)
 608 {
 609         struct device *dev = &cyapa->client->dev;
 610         u16 csum;
 611         u16 csum_expected;
 612
 613         /* Firmware must match exact 30848 bytes = 482 64-byte blocks. */
 614         if (fw->size != CYAPA_FW_SIZE) {
 615                 dev_err(dev, "invalid firmware size = %zu, expected %u.\n",
 616                         fw->size, CYAPA_FW_SIZE);
 617                 return -EINVAL;
 618         }
 619
 620         /* Verify header block */
 621         csum_expected = (fw->data[0] << 8) | fw->data[1];
 622         csum = cyapa_gen3_csum(&fw->data[2], CYAPA_FW_HDR_SIZE - 2);
 623         if (csum != csum_expected) {
 624                 dev_err(dev, "%s %04x, expected: %04x\n",
 625                         "invalid firmware header checksum = ",
 626                         csum, csum_expected);
 627                 return -EINVAL;
 628         }
 629
 630         /* Verify firmware image */
 631         csum_expected = (fw->data[CYAPA_FW_HDR_SIZE - 2] << 8) |
 632                          fw->data[CYAPA_FW_HDR_SIZE - 1];
 633         csum = cyapa_gen3_csum(&fw->data[CYAPA_FW_HDR_SIZE],
 634                         CYAPA_FW_DATA_SIZE);
 635         if (csum != csum_expected) {
 636                 dev_err(dev, "%s %04x, expected: %04x\n",
 637                         "invalid firmware header checksum = ",
 638                         csum, csum_expected);
 639                 return -EINVAL;
 640         }
 641         return 0;
 642 }
 643
 644 /*
 645  * Write a |len| byte long buffer |buf| to the device, by chopping it up into a
 646  * sequence of smaller |CYAPA_CMD_LEN|-length write commands.
 647  *
 648  * The data bytes for a write command are prepended with the 1-byte offset
 649  * of the data relative to the start of |buf|.
 650  */
 651 static int cyapa_gen3_write_buffer(struct cyapa *cyapa,
 652                 const u8 *buf, size_t len)
 653 {
 654         int error;
 655         size_t i;
 656         unsigned char cmd[CYAPA_CMD_LEN + 1];
 657         size_t cmd_len;
 658
 659         for (i = 0; i < len; i += CYAPA_CMD_LEN) {
 660                 const u8 *payload = &buf[i];
 661
 662                 cmd_len = (len - i >= CYAPA_CMD_LEN) ? CYAPA_CMD_LEN : len - i;
 663                 cmd[0] = i;
 664                 memcpy(&cmd[1], payload, cmd_len);
 665
 666                 error = cyapa_i2c_reg_write_block(cyapa, 0, cmd_len + 1, cmd);
 667                 if (error)
 668                         return error;
 669         }
 670         return 0;
 671 }
 672
 673 /*
 674  * A firmware block write command writes 64 bytes of data to a single flash
 675  * page in the device.  The 78-byte block write command has the format:
 676  *   <0xff> <CMD> <Key> <Start> <Data> <Data-Checksum> <CMD Checksum>
 677  *
 678  *  <0xff>  - every command starts with 0xff
 679  *  <CMD>   - the write command value is 0x39
 680  *  <Key>   - write commands include an 8-byte key: { 00 01 02 03 04 05 06 07 }
 681  *  <Block> - Memory Block number (address / 64) (16-bit, big-endian)
 682  *  <Data>  - 64 bytes of firmware image data
 683  *  <Data Checksum> - sum of 64 <Data> bytes, modulo 0xff
 684  *  <CMD Checksum> - sum of 77 bytes, from 0xff to <Data Checksum>
 685  *
 686  * Each write command is split into 5 i2c write transactions of up to 16 bytes.
 687  * Each transaction starts with an i2c register offset: (00, 10, 20, 30, 40).
 688  */
 689 static int cyapa_gen3_write_fw_block(struct cyapa *cyapa,
 690                 u16 block, const u8 *data)
 691 {
 692         int ret;
 693         struct gen3_write_block_cmd write_block_cmd;
 694         u8 status[BL_STATUS_SIZE];
 695         int tries;
 696         u8 bl_status, bl_error;
 697
 698         /* Set write command and security key bytes. */
 699         write_block_cmd.checksum_seed = GEN3_BL_CMD_CHECKSUM_SEED;
 700         write_block_cmd.cmd_code = GEN3_BL_CMD_WRITE_BLOCK;
 701         memcpy(write_block_cmd.key, security_key, sizeof(security_key));
 702         put_unaligned_be16(block, &write_block_cmd.block_num);
 703         memcpy(write_block_cmd.block_data, data, CYAPA_FW_BLOCK_SIZE);
 704         write_block_cmd.block_checksum = cyapa_gen3_csum(
 705                         write_block_cmd.block_data, CYAPA_FW_BLOCK_SIZE);
 706         write_block_cmd.cmd_checksum = cyapa_gen3_csum((u8 *)&write_block_cmd,
 707                         sizeof(write_block_cmd) - 1);
 708
 709         ret = cyapa_gen3_write_buffer(cyapa, (u8 *)&write_block_cmd,
 710                         sizeof(write_block_cmd));
 711         if (ret)
 712                 return ret;
 713
 714         /* Wait for write to finish */
 715         tries = 11;  /* Programming for one block can take about 100ms. */
 716         do {
 717                 usleep_range(10000, 20000);
 718
 719                 /* Check block write command result status. */
 720                 ret = cyapa_i2c_reg_read_block(cyapa, BL_HEAD_OFFSET,
 721                                                BL_STATUS_SIZE, status);
 722                 if (ret != BL_STATUS_SIZE)
 723                         return (ret < 0) ? ret : -EIO;
 724         } while ((status[REG_BL_STATUS] & BL_STATUS_BUSY) && --tries);
 725
 726         /* Ignore WATCHDOG bit and reserved bits. */
 727         bl_status = status[REG_BL_STATUS] & ~BL_STATUS_REV_MASK;
 728         bl_error = status[REG_BL_ERROR] & ~BL_ERROR_RESERVED;
 729
 730         if (bl_status & BL_STATUS_BUSY)
 731                 ret = -ETIMEDOUT;
 732         else if (bl_status != BL_STATUS_RUNNING ||
 733                 bl_error != BL_ERROR_BOOTLOADING)
 734                 ret = -EIO;
 735         else
 736                 ret = 0;
 737
 738         return ret;
 739 }
 740
 741 static int cyapa_gen3_write_blocks(struct cyapa *cyapa,
 742                 size_t start_block, size_t block_count,
 743                 const u8 *image_data)
 744 {
 745         int error;
 746         int i;
 747
 748         for (i = 0; i < block_count; i++) {
 749                 size_t block = start_block + i;
 750                 size_t addr = i * CYAPA_FW_BLOCK_SIZE;
 751                 const u8 *data = &image_data[addr];
 752
 753                 error = cyapa_gen3_write_fw_block(cyapa, block, data);
 754                 if (error)
 755                         return error;
 756         }
 757         return 0;
 758 }
 759
 760 static int cyapa_gen3_do_fw_update(struct cyapa *cyapa,
 761                 const struct firmware *fw)
 762 {
 763         struct device *dev = &cyapa->client->dev;
 764         int error;
 765
 766         /* First write data, starting at byte 128 of fw->data */
 767         error = cyapa_gen3_write_blocks(cyapa,
 768                 CYAPA_FW_DATA_BLOCK_START, CYAPA_FW_DATA_BLOCK_COUNT,
 769                 &fw->data[CYAPA_FW_HDR_BLOCK_COUNT * CYAPA_FW_BLOCK_SIZE]);
 770         if (error) {
 771                 dev_err(dev, "FW update aborted, write image: %d\n", error);
 772                 return error;
 773         }
 774
 775         /* Then write checksum */
 776         error = cyapa_gen3_write_blocks(cyapa,
 777                 CYAPA_FW_HDR_BLOCK_START, CYAPA_FW_HDR_BLOCK_COUNT,
 778                 &fw->data[0]);
 779         if (error) {
 780                 dev_err(dev, "FW update aborted, write checksum: %d\n", error);
 781                 return error;
 782         }
 783
 784         return 0;
 785 }
 786
 787 static ssize_t cyapa_gen3_do_calibrate(struct device *dev,
 788                                      struct device_attribute *attr,
 789                                      const char *buf, size_t count)
 790 {
 791         struct cyapa *cyapa = dev_get_drvdata(dev);
 792         int tries;
 793         int ret;
 794
 795         ret = cyapa_read_byte(cyapa, CYAPA_CMD_DEV_STATUS);
 796         if (ret < 0) {
 797                 dev_err(dev, "Error reading dev status: %d\n", ret);
 798                 goto out;
 799         }
 800         if ((ret & CYAPA_DEV_NORMAL) != CYAPA_DEV_NORMAL) {
 801                 dev_warn(dev, "Trackpad device is busy, device state: 0x%02x\n",
 802                          ret);
 803                 ret = -EAGAIN;
 804                 goto out;
 805         }
 806
 807         ret = cyapa_write_byte(cyapa, CYAPA_CMD_SOFT_RESET,
 808                                OP_RECALIBRATION_MASK);
 809         if (ret < 0) {
 810                 dev_err(dev, "Failed to send calibrate command: %d\n",
 811                         ret);
 812                 goto out;
 813         }
 814
 815         tries = 20;  /* max recalibration timeout 2s. */
 816         do {
 817                 /*
 818                  * For this recalibration, the max time will not exceed 2s.
 819                  * The average time is approximately 500 - 700 ms, and we
 820                  * will check the status every 100 - 200ms.
 821                  */
 822                 usleep_range(100000, 200000);
 823
 824                 ret = cyapa_read_byte(cyapa, CYAPA_CMD_DEV_STATUS);
 825                 if (ret < 0) {
 826                         dev_err(dev, "Error reading dev status: %d\n",
 827                                 ret);
 828                         goto out;
 829                 }
 830                 if ((ret & CYAPA_DEV_NORMAL) == CYAPA_DEV_NORMAL)
 831                         break;
 832         } while (--tries);
 833
 834         if (tries == 0) {
 835                 dev_err(dev, "Failed to calibrate. Timeout.\n");
 836                 ret = -ETIMEDOUT;
 837                 goto out;
 838         }
 839         dev_dbg(dev, "Calibration successful.\n");
 840
 841 out:
 842         return ret < 0 ? ret : count;
 843 }
 844
 845 static ssize_t cyapa_gen3_show_baseline(struct device *dev,
 846                                    struct device_attribute *attr, char *buf)
 847 {
 848         struct cyapa *cyapa = dev_get_drvdata(dev);
 849         int max_baseline, min_baseline;
 850         int tries;
 851         int ret;
 852
 853         ret = cyapa_read_byte(cyapa, CYAPA_CMD_DEV_STATUS);
 854         if (ret < 0) {
 855                 dev_err(dev, "Error reading dev status. err = %d\n", ret);
 856                 goto out;
 857         }
 858         if ((ret & CYAPA_DEV_NORMAL) != CYAPA_DEV_NORMAL) {
 859                 dev_warn(dev, "Trackpad device is busy. device state = 0x%x\n",
 860                          ret);
 861                 ret = -EAGAIN;
 862                 goto out;
 863         }
 864
 865         ret = cyapa_write_byte(cyapa, CYAPA_CMD_SOFT_RESET,
 866                                OP_REPORT_BASELINE_MASK);
 867         if (ret < 0) {
 868                 dev_err(dev, "Failed to send report baseline command. %d\n",
 869                         ret);
 870                 goto out;
 871         }
 872
 873         tries = 3;  /* Try for 30 to 60 ms */
 874         do {
 875                 usleep_range(10000, 20000);
 876
 877                 ret = cyapa_read_byte(cyapa, CYAPA_CMD_DEV_STATUS);
 878                 if (ret < 0) {
 879                         dev_err(dev, "Error reading dev status. err = %d\n",
 880                                 ret);
 881                         goto out;
 882                 }
 883                 if ((ret & CYAPA_DEV_NORMAL) == CYAPA_DEV_NORMAL)
 884                         break;
 885         } while (--tries);
 886
 887         if (tries == 0) {
 888                 dev_err(dev, "Device timed out going to Normal state.\n");
 889                 ret = -ETIMEDOUT;
 890                 goto out;
 891         }
 892
 893         ret = cyapa_read_byte(cyapa, CYAPA_CMD_MAX_BASELINE);
 894         if (ret < 0) {
 895                 dev_err(dev, "Failed to read max baseline. err = %d\n", ret);
 896                 goto out;
 897         }
 898         max_baseline = ret;
 899
 900         ret = cyapa_read_byte(cyapa, CYAPA_CMD_MIN_BASELINE);
 901         if (ret < 0) {
 902                 dev_err(dev, "Failed to read min baseline. err = %d\n", ret);
 903                 goto out;
 904         }
 905         min_baseline = ret;
 906
 907         dev_dbg(dev, "Baseline report successful. Max: %d Min: %d\n",
 908                 max_baseline, min_baseline);
 909         ret = scnprintf(buf, PAGE_SIZE, "%d %d\n", max_baseline, min_baseline);
 910
 911 out:
 912         return ret;
 913 }
 914
 915 /*
 916  * cyapa_get_wait_time_for_pwr_cmd
 917  *
 918  * Compute the amount of time we need to wait after updating the touchpad
 919  * power mode. The touchpad needs to consume the incoming power mode set
 920  * command at the current clock rate.
 921  */
 922
 923 static u16 cyapa_get_wait_time_for_pwr_cmd(u8 pwr_mode)
 924 {
 925         switch (pwr_mode) {
 926         case PWR_MODE_FULL_ACTIVE: return 20;
 927         case PWR_MODE_BTN_ONLY: return 20;
 928         case PWR_MODE_OFF: return 20;
 929         default: return cyapa_pwr_cmd_to_sleep_time(pwr_mode) + 50;
 930         }
 931 }
 932
 933 /*
 934  * Set device power mode
 935  *
 936  * Write to the field to configure power state. Power states include :
 937  *   Full : Max scans and report rate.
 938  *   Idle : Report rate set by user specified time.
 939  *   ButtonOnly : No scans for fingers. When the button is triggered,
 940  *     a slave interrupt is asserted to notify host to wake up.
 941  *   Off : Only awake for i2c commands from host. No function for button
 942  *     or touch sensors.
 943  *
 944  * The power_mode command should conform to the following :
 945  *   Full : 0x3f
 946  *   Idle : Configurable from 20 to 1000ms. See note below for
 947  *     cyapa_sleep_time_to_pwr_cmd and cyapa_pwr_cmd_to_sleep_time
 948  *   ButtonOnly : 0x01
 949  *   Off : 0x00
 950  *
 951  * Device power mode can only be set when device is in operational mode.
 952  */
 953 static int cyapa_gen3_set_power_mode(struct cyapa *cyapa, u8 power_mode,
 954                 u16 always_unused, enum cyapa_pm_stage pm_stage)
 955 {
 956         struct input_dev *input = cyapa->input;
 957         u8 power;
 958         int tries;
 959         int sleep_time;
 960         int interval;
 961         int ret;
 962
 963         if (cyapa->state != CYAPA_STATE_OP)
 964                 return 0;
 965
 966         tries = SET_POWER_MODE_TRIES;
 967         while (tries--) {
 968                 ret = cyapa_read_byte(cyapa, CYAPA_CMD_POWER_MODE);
 969                 if (ret >= 0)
 970                         break;
 971                 usleep_range(SET_POWER_MODE_DELAY, 2 * SET_POWER_MODE_DELAY);
 972         }
 973         if (ret < 0)
 974                 return ret;
 975
 976         /*
 977          * Return early if the power mode to set is the same as the current
 978          * one.
 979          */
 980         if ((ret & PWR_MODE_MASK) == power_mode)
 981                 return 0;
 982
 983         sleep_time = (int)cyapa_get_wait_time_for_pwr_cmd(ret & PWR_MODE_MASK);
 984         power = ret;
 985         power &= ~PWR_MODE_MASK;
 986         power |= power_mode & PWR_MODE_MASK;
 987         tries = SET_POWER_MODE_TRIES;
 988         while (tries--) {
 989                 ret = cyapa_write_byte(cyapa, CYAPA_CMD_POWER_MODE, power);
 990                 if (!ret)
 991                         break;
 992                 usleep_range(SET_POWER_MODE_DELAY, 2 * SET_POWER_MODE_DELAY);
 993         }
 994
 995         /*
 996          * Wait for the newly set power command to go in at the previous
 997          * clock speed (scanrate) used by the touchpad firmware. Not
 998          * doing so before issuing the next command may result in errors
 999          * depending on the command's content.
1000          */
1001         if (cyapa->operational && input && input->users &&
1002             (pm_stage == CYAPA_PM_RUNTIME_SUSPEND ||
1003              pm_stage == CYAPA_PM_RUNTIME_RESUME)) {
1004                 /* Try to polling in 120Hz, read may fail, just ignore it. */
1005                 interval = 1000 / 120;
1006                 while (sleep_time > 0) {
1007                         if (sleep_time > interval)
1008                                 msleep(interval);
1009                         else
1010                                 msleep(sleep_time);
1011                         sleep_time -= interval;
1012                         cyapa_gen3_try_poll_handler(cyapa);
1013                 }
1014         } else {
1015                 msleep(sleep_time);
1016         }
1017
1018         return ret;
1019 }
1020
1021 static int cyapa_gen3_set_proximity(struct cyapa *cyapa, bool enable)
1022 {
1023         return -EOPNOTSUPP;
1024 }
1025
1026 static int cyapa_gen3_get_query_data(struct cyapa *cyapa)
1027 {
1028         u8 query_data[QUERY_DATA_SIZE];
1029         int ret;
1030
1031         if (cyapa->state != CYAPA_STATE_OP)
1032                 return -EBUSY;
1033
1034         ret = cyapa_read_block(cyapa, CYAPA_CMD_GROUP_QUERY, query_data);
1035         if (ret != QUERY_DATA_SIZE)
1036                 return (ret < 0) ? ret : -EIO;
1037
1038         memcpy(&cyapa->product_id[0], &query_data[0], 5);
1039         cyapa->product_id[5] = '-';
1040         memcpy(&cyapa->product_id[6], &query_data[5], 6);
1041         cyapa->product_id[12] = '-';
1042         memcpy(&cyapa->product_id[13], &query_data[11], 2);
1043         cyapa->product_id[15] = '\0';
1044
1045         cyapa->fw_maj_ver = query_data[15];
1046         cyapa->fw_min_ver = query_data[16];
1047
1048         cyapa->btn_capability = query_data[19] & CAPABILITY_BTN_MASK;
1049
1050         cyapa->gen = query_data[20] & 0x0f;
1051
1052         cyapa->max_abs_x = ((query_data[21] & 0xf0) << 4) | query_data[22];
1053         cyapa->max_abs_y = ((query_data[21] & 0x0f) << 8) | query_data[23];
1054
1055         cyapa->physical_size_x =
1056                 ((query_data[24] & 0xf0) << 4) | query_data[25];
1057         cyapa->physical_size_y =
1058                 ((query_data[24] & 0x0f) << 8) | query_data[26];
1059
1060         cyapa->max_z = 255;
1061
1062         return 0;
1063 }
1064
1065 static int cyapa_gen3_bl_query_data(struct cyapa *cyapa)
1066 {
1067         u8 bl_data[CYAPA_CMD_LEN];
1068         int ret;
1069
1070         ret = cyapa_i2c_reg_read_block(cyapa, 0, CYAPA_CMD_LEN, bl_data);
1071         if (ret != CYAPA_CMD_LEN)
1072                 return (ret < 0) ? ret : -EIO;
1073
1074         /*
1075          * This value will be updated again when entered application mode.
1076          * If TP failed to enter application mode, this fw version values
1077          * can be used as a reference.
1078          * This firmware version valid when fw image checksum is valid.
1079          */
1080         if (bl_data[REG_BL_STATUS] ==
1081                         (BL_STATUS_RUNNING | BL_STATUS_CSUM_VALID)) {
1082                 cyapa->fw_maj_ver = bl_data[GEN3_BL_IDLE_FW_MAJ_VER_OFFSET];
1083                 cyapa->fw_min_ver = bl_data[GEN3_BL_IDLE_FW_MIN_VER_OFFSET];
1084         }
1085
1086         return 0;
1087 }
1088
1089 /*
1090  * Check if device is operational.
1091  *
1092  * An operational device is responding, has exited bootloader, and has
1093  * firmware supported by this driver.
1094  *
1095  * Returns:
1096  *   -EBUSY  no device or in bootloader
1097  *   -EIO    failure while reading from device
1098  *   -EAGAIN device is still in bootloader
1099  *           if ->state = CYAPA_STATE_BL_IDLE, device has invalid firmware
1100  *   -EINVAL device is in operational mode, but not supported by this driver
1101  *   0       device is supported
1102  */
1103 static int cyapa_gen3_do_operational_check(struct cyapa *cyapa)
1104 {
1105         struct device *dev = &cyapa->client->dev;
1106         int error;
1107
1108         switch (cyapa->state) {
1109         case CYAPA_STATE_BL_ACTIVE:
1110                 error = cyapa_gen3_bl_deactivate(cyapa);
1111                 if (error) {
1112                         dev_err(dev, "failed to bl_deactivate: %d\n", error);
1113                         return error;
1114                 }
1115
1116         /* Fallthrough state */
1117         case CYAPA_STATE_BL_IDLE:
1118                 /* Try to get firmware version in bootloader mode. */
1119                 cyapa_gen3_bl_query_data(cyapa);
1120
1121                 error = cyapa_gen3_bl_exit(cyapa);
1122                 if (error) {
1123                         dev_err(dev, "failed to bl_exit: %d\n", error);
1124                         return error;
1125                 }
1126
1127         /* Fallthrough state */
1128         case CYAPA_STATE_OP:
1129                 /*
1130                  * Reading query data before going back to the full mode
1131                  * may cause problems, so we set the power mode first here.
1132                  */
1133                 error = cyapa_gen3_set_power_mode(cyapa,
1134                                 PWR_MODE_FULL_ACTIVE, 0, CYAPA_PM_ACTIVE);
1135                 if (error)
1136                         dev_err(dev, "%s: set full power mode failed: %d\n",
1137                                 __func__, error);
1138                 error = cyapa_gen3_get_query_data(cyapa);
1139                 if (error < 0)
1140                         return error;
1141
1142                 /* Only support firmware protocol gen3 */
1143                 if (cyapa->gen != CYAPA_GEN3) {
1144                         dev_err(dev, "unsupported protocol version (%d)",
1145                                 cyapa->gen);
1146                         return -EINVAL;
1147                 }
1148
1149                 /* Only support product ID starting with CYTRA */
1150                 if (memcmp(cyapa->product_id, product_id,
1151                                 strlen(product_id)) != 0) {
1152                         dev_err(dev, "unsupported product ID (%s)\n",
1153                                 cyapa->product_id);
1154                         return -EINVAL;
1155                 }
1156
1157                 return 0;
1158
1159         default:
1160                 return -EIO;
1161         }
1162         return 0;
1163 }
1164
1165 /*
1166  * Return false, do not continue process
1167  * Return true, continue process.
1168  */
1169 static bool cyapa_gen3_irq_cmd_handler(struct cyapa *cyapa)
1170 {
1171         /* Not gen3 irq command response, skip for continue. */
1172         if (cyapa->gen != CYAPA_GEN3)
1173                 return true;
1174
1175         if (cyapa->operational)
1176                 return true;
1177
1178         /*
1179          * Driver in detecting or other interface function processing,
1180          * so, stop cyapa_gen3_irq_handler to continue process to
1181          * avoid unwanted to error detecting and processing.
1182          *
1183          * And also, avoid the periodically asserted interrupts to be processed
1184          * as touch inputs when gen3 failed to launch into application mode,
1185          * which will cause gen3 stays in bootloader mode.
1186          */
1187         return false;
1188 }
1189
1190 static int cyapa_gen3_event_process(struct cyapa *cyapa,
1191                                     struct cyapa_reg_data *data)
1192 {
1193         struct input_dev *input = cyapa->input;
1194         int num_fingers;
1195         int i;
1196
1197         num_fingers = (data->finger_btn >> 4) & 0x0f;
1198         for (i = 0; i < num_fingers; i++) {
1199                 const struct cyapa_touch *touch = &data->touches[i];
1200                 /* Note: touch->id range is 1 to 15; slots are 0 to 14. */
1201                 int slot = touch->id - 1;
1202
1203                 input_mt_slot(input, slot);
1204                 input_mt_report_slot_state(input, MT_TOOL_FINGER, true);
1205                 input_report_abs(input, ABS_MT_POSITION_X,
1206                                  ((touch->xy_hi & 0xf0) << 4) | touch->x_lo);
1207                 input_report_abs(input, ABS_MT_POSITION_Y,
1208                                  ((touch->xy_hi & 0x0f) << 8) | touch->y_lo);
1209                 input_report_abs(input, ABS_MT_PRESSURE, touch->pressure);
1210         }
1211
1212         input_mt_sync_frame(input);
1213
1214         if (cyapa->btn_capability & CAPABILITY_LEFT_BTN_MASK)
1215                 input_report_key(input, BTN_LEFT,
1216                                  !!(data->finger_btn & OP_DATA_LEFT_BTN));
1217         if (cyapa->btn_capability & CAPABILITY_MIDDLE_BTN_MASK)
1218                 input_report_key(input, BTN_MIDDLE,
1219                                  !!(data->finger_btn & OP_DATA_MIDDLE_BTN));
1220         if (cyapa->btn_capability & CAPABILITY_RIGHT_BTN_MASK)
1221                 input_report_key(input, BTN_RIGHT,
1222                                  !!(data->finger_btn & OP_DATA_RIGHT_BTN));
1223         input_sync(input);
1224
1225         return 0;
1226 }
1227
1228 static int cyapa_gen3_irq_handler(struct cyapa *cyapa)
1229 {
1230         struct device *dev = &cyapa->client->dev;
1231         struct cyapa_reg_data data;
1232         int ret;
1233
1234         ret = cyapa_read_block(cyapa, CYAPA_CMD_GROUP_DATA, (u8 *)&data);
1235         if (ret != sizeof(data)) {
1236                 dev_err(dev, "failed to read report data, (%d)\n", ret);
1237                 return -EINVAL;
1238         }
1239
1240         if ((data.device_status & OP_STATUS_SRC) != OP_STATUS_SRC ||
1241             (data.device_status & OP_STATUS_DEV) != CYAPA_DEV_NORMAL ||
1242             (data.finger_btn & OP_DATA_VALID) != OP_DATA_VALID) {
1243                 dev_err(dev, "invalid device state bytes: %02x %02x\n",
1244                         data.device_status, data.finger_btn);
1245                 return -EINVAL;
1246         }
1247
1248         return cyapa_gen3_event_process(cyapa, &data);
1249 }
1250
1251 /*
1252  * This function will be called in the cyapa_gen3_set_power_mode function,
1253  * and it's known that it may failed in some situation after the set power
1254  * mode command was sent. So this function is aimed to avoid the knwon
1255  * and unwanted output I2C and data parse error messages.
1256  */
1257 static int cyapa_gen3_try_poll_handler(struct cyapa *cyapa)
1258 {
1259         struct cyapa_reg_data data;
1260         int ret;
1261
1262         ret = cyapa_read_block(cyapa, CYAPA_CMD_GROUP_DATA, (u8 *)&data);
1263         if (ret != sizeof(data))
1264                 return -EINVAL;
1265
1266         if ((data.device_status & OP_STATUS_SRC) != OP_STATUS_SRC ||
1267             (data.device_status & OP_STATUS_DEV) != CYAPA_DEV_NORMAL ||
1268             (data.finger_btn & OP_DATA_VALID) != OP_DATA_VALID)
1269                 return -EINVAL;
1270
1271         return cyapa_gen3_event_process(cyapa, &data);
1272
1273 }
1274
1275 static int cyapa_gen3_initialize(struct cyapa *cyapa) { return 0; }
1276 static int cyapa_gen3_bl_initiate(struct cyapa *cyapa,
1277                 const struct firmware *fw) { return 0; }
1278 static int cyapa_gen3_empty_output_data(struct cyapa *cyapa,
1279                 u8 *buf, int *len, cb_sort func) { return 0; }
1280
1281 const struct cyapa_dev_ops cyapa_gen3_ops = {
1282         .check_fw = cyapa_gen3_check_fw,
1283         .bl_enter = cyapa_gen3_bl_enter,
1284         .bl_activate = cyapa_gen3_bl_activate,
1285         .update_fw = cyapa_gen3_do_fw_update,
1286         .bl_deactivate = cyapa_gen3_bl_deactivate,
1287         .bl_initiate = cyapa_gen3_bl_initiate,
1288
1289         .show_baseline = cyapa_gen3_show_baseline,
1290         .calibrate_store = cyapa_gen3_do_calibrate,
1291
1292         .initialize = cyapa_gen3_initialize,
1293
1294         .state_parse = cyapa_gen3_state_parse,
1295         .operational_check = cyapa_gen3_do_operational_check,
1296
1297         .irq_handler = cyapa_gen3_irq_handler,
1298         .irq_cmd_handler = cyapa_gen3_irq_cmd_handler,
1299         .sort_empty_output_data = cyapa_gen3_empty_output_data,
1300         .set_power_mode = cyapa_gen3_set_power_mode,
1301
1302         .set_proximity = cyapa_gen3_set_proximity,
1303 };