uboot/test/dm/i2c.c
<<
>>
Prefs
   1// SPDX-License-Identifier: GPL-2.0+
   2/*
   3 * Copyright (C) 2013 Google, Inc
   4 *
   5 * Note: Test coverage does not include 10-bit addressing
   6 */
   7
   8#include <common.h>
   9#include <dm.h>
  10#include <fdtdec.h>
  11#include <i2c.h>
  12#include <asm/state.h>
  13#include <asm/test.h>
  14#include <dm/device-internal.h>
  15#include <dm/test.h>
  16#include <dm/uclass-internal.h>
  17#include <dm/util.h>
  18#include <test/ut.h>
  19
  20static const int busnum;
  21static const int chip = 0x2c;
  22
  23/* Test that we can find buses and chips */
  24static int dm_test_i2c_find(struct unit_test_state *uts)
  25{
  26        struct udevice *bus, *dev;
  27        const int no_chip = 0x10;
  28
  29        ut_asserteq(-ENODEV, uclass_find_device_by_seq(UCLASS_I2C, busnum,
  30                                                       false, &bus));
  31
  32        /*
  33         * The post_bind() method will bind devices to chip selects. Check
  34         * this then remove the emulation and the slave device.
  35         */
  36        ut_assertok(uclass_get_device_by_seq(UCLASS_I2C, busnum, &bus));
  37        ut_assertok(dm_i2c_probe(bus, chip, 0, &dev));
  38        ut_asserteq(-ENOENT, dm_i2c_probe(bus, no_chip, 0, &dev));
  39        ut_asserteq(-ENODEV, uclass_get_device_by_seq(UCLASS_I2C, 1, &bus));
  40
  41        return 0;
  42}
  43DM_TEST(dm_test_i2c_find, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
  44
  45static int dm_test_i2c_read_write(struct unit_test_state *uts)
  46{
  47        struct udevice *bus, *dev;
  48        uint8_t buf[5];
  49
  50        ut_assertok(uclass_get_device_by_seq(UCLASS_I2C, busnum, &bus));
  51        ut_assertok(i2c_get_chip(bus, chip, 1, &dev));
  52        ut_assertok(dm_i2c_read(dev, 0, buf, 5));
  53        ut_assertok(memcmp(buf, "\0\0\0\0\0", sizeof(buf)));
  54        ut_assertok(dm_i2c_write(dev, 2, (uint8_t *)"AB", 2));
  55        ut_assertok(dm_i2c_read(dev, 0, buf, 5));
  56        ut_assertok(memcmp(buf, "\0\0AB\0", sizeof(buf)));
  57
  58        return 0;
  59}
  60DM_TEST(dm_test_i2c_read_write, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
  61
  62static int dm_test_i2c_speed(struct unit_test_state *uts)
  63{
  64        struct udevice *bus, *dev;
  65        uint8_t buf[5];
  66
  67        ut_assertok(uclass_get_device_by_seq(UCLASS_I2C, busnum, &bus));
  68
  69        /* Use test mode so we create the required errors for invalid speeds */
  70        sandbox_i2c_set_test_mode(bus, true);
  71        ut_assertok(i2c_get_chip(bus, chip, 1, &dev));
  72        ut_assertok(dm_i2c_set_bus_speed(bus, 100000));
  73        ut_assertok(dm_i2c_read(dev, 0, buf, 5));
  74        ut_assertok(dm_i2c_set_bus_speed(bus, 400000));
  75        ut_asserteq(400000, dm_i2c_get_bus_speed(bus));
  76        ut_assertok(dm_i2c_read(dev, 0, buf, 5));
  77        ut_asserteq(-EINVAL, dm_i2c_write(dev, 0, buf, 5));
  78        sandbox_i2c_set_test_mode(bus, false);
  79
  80        return 0;
  81}
  82DM_TEST(dm_test_i2c_speed, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
  83
  84static int dm_test_i2c_offset_len(struct unit_test_state *uts)
  85{
  86        struct udevice *bus, *dev;
  87        uint8_t buf[5];
  88
  89        ut_assertok(uclass_get_device_by_seq(UCLASS_I2C, busnum, &bus));
  90        ut_assertok(i2c_get_chip(bus, chip, 1, &dev));
  91        ut_assertok(i2c_set_chip_offset_len(dev, 1));
  92        ut_assertok(dm_i2c_read(dev, 0, buf, 5));
  93
  94        /* This is not supported by the uclass */
  95        ut_asserteq(-EINVAL, i2c_set_chip_offset_len(dev, 5));
  96
  97        return 0;
  98}
  99DM_TEST(dm_test_i2c_offset_len, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
 100
 101static int dm_test_i2c_probe_empty(struct unit_test_state *uts)
 102{
 103        struct udevice *bus, *dev;
 104
 105        ut_assertok(uclass_get_device_by_seq(UCLASS_I2C, busnum, &bus));
 106
 107        /* Use test mode so that this chip address will always probe */
 108        sandbox_i2c_set_test_mode(bus, true);
 109        ut_assertok(dm_i2c_probe(bus, SANDBOX_I2C_TEST_ADDR, 0, &dev));
 110        sandbox_i2c_set_test_mode(bus, false);
 111
 112        return 0;
 113}
 114DM_TEST(dm_test_i2c_probe_empty, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
 115
 116static int dm_test_i2c_bytewise(struct unit_test_state *uts)
 117{
 118        struct udevice *bus, *dev;
 119        struct udevice *eeprom;
 120        uint8_t buf[5];
 121
 122        ut_assertok(uclass_get_device_by_seq(UCLASS_I2C, busnum, &bus));
 123        ut_assertok(i2c_get_chip(bus, chip, 1, &dev));
 124        ut_assertok(dm_i2c_read(dev, 0, buf, 5));
 125        ut_assertok(memcmp(buf, "\0\0\0\0\0", sizeof(buf)));
 126
 127        /* Tell the EEPROM to only read/write one register at a time */
 128        ut_assertok(uclass_first_device(UCLASS_I2C_EMUL, &eeprom));
 129        ut_assertnonnull(eeprom);
 130        sandbox_i2c_eeprom_set_test_mode(eeprom, SIE_TEST_MODE_SINGLE_BYTE);
 131
 132        /* Now we only get the first byte - the rest will be 0xff */
 133        ut_assertok(dm_i2c_read(dev, 0, buf, 5));
 134        ut_assertok(memcmp(buf, "\0\xff\xff\xff\xff", sizeof(buf)));
 135
 136        /* If we do a separate transaction for each byte, it works */
 137        ut_assertok(i2c_set_chip_flags(dev, DM_I2C_CHIP_RD_ADDRESS));
 138        ut_assertok(dm_i2c_read(dev, 0, buf, 5));
 139        ut_assertok(memcmp(buf, "\0\0\0\0\0", sizeof(buf)));
 140
 141        /* This will only write A */
 142        ut_assertok(i2c_set_chip_flags(dev, 0));
 143        ut_assertok(dm_i2c_write(dev, 2, (uint8_t *)"AB", 2));
 144        ut_assertok(dm_i2c_read(dev, 0, buf, 5));
 145        ut_assertok(memcmp(buf, "\0\xff\xff\xff\xff", sizeof(buf)));
 146
 147        /* Check that the B was ignored */
 148        ut_assertok(i2c_set_chip_flags(dev, DM_I2C_CHIP_RD_ADDRESS));
 149        ut_assertok(dm_i2c_read(dev, 0, buf, 5));
 150        ut_assertok(memcmp(buf, "\0\0A\0\0\0", sizeof(buf)));
 151
 152        /* Now write it again with the new flags, it should work */
 153        ut_assertok(i2c_set_chip_flags(dev, DM_I2C_CHIP_WR_ADDRESS));
 154        ut_assertok(dm_i2c_write(dev, 2, (uint8_t *)"AB", 2));
 155        ut_assertok(dm_i2c_read(dev, 0, buf, 5));
 156        ut_assertok(memcmp(buf, "\0\xff\xff\xff\xff", sizeof(buf)));
 157
 158        ut_assertok(i2c_set_chip_flags(dev, DM_I2C_CHIP_WR_ADDRESS |
 159                                                DM_I2C_CHIP_RD_ADDRESS));
 160        ut_assertok(dm_i2c_read(dev, 0, buf, 5));
 161        ut_assertok(memcmp(buf, "\0\0AB\0\0", sizeof(buf)));
 162
 163        /* Restore defaults */
 164        sandbox_i2c_eeprom_set_test_mode(eeprom, SIE_TEST_MODE_NONE);
 165        ut_assertok(i2c_set_chip_flags(dev, 0));
 166
 167        return 0;
 168}
 169DM_TEST(dm_test_i2c_bytewise, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
 170
 171static int dm_test_i2c_offset(struct unit_test_state *uts)
 172{
 173        struct udevice *eeprom;
 174        struct udevice *dev;
 175        uint8_t buf[5];
 176
 177        ut_assertok(i2c_get_chip_for_busnum(busnum, chip, 1, &dev));
 178
 179        /* Do a transfer so we can find the emulator */
 180        ut_assertok(dm_i2c_read(dev, 0, buf, 5));
 181        ut_assertok(uclass_first_device(UCLASS_I2C_EMUL, &eeprom));
 182
 183        /* Offset length 0 */
 184        sandbox_i2c_eeprom_set_offset_len(eeprom, 0);
 185        ut_assertok(i2c_set_chip_offset_len(dev, 0));
 186        ut_assertok(dm_i2c_write(dev, 10 /* ignored */, (uint8_t *)"AB", 2));
 187        ut_assertok(dm_i2c_read(dev, 0, buf, 5));
 188        ut_assertok(memcmp(buf, "AB\0\0\0\0", sizeof(buf)));
 189
 190        /* Offset length 1 */
 191        sandbox_i2c_eeprom_set_offset_len(eeprom, 1);
 192        ut_assertok(i2c_set_chip_offset_len(dev, 1));
 193        ut_assertok(dm_i2c_write(dev, 2, (uint8_t *)"AB", 2));
 194        ut_assertok(dm_i2c_read(dev, 0, buf, 5));
 195        ut_assertok(memcmp(buf, "ABAB\0", sizeof(buf)));
 196
 197        /* Offset length 2 */
 198        sandbox_i2c_eeprom_set_offset_len(eeprom, 2);
 199        ut_assertok(i2c_set_chip_offset_len(dev, 2));
 200        ut_assertok(dm_i2c_write(dev, 0x210, (uint8_t *)"AB", 2));
 201        ut_assertok(dm_i2c_read(dev, 0x210, buf, 5));
 202        ut_assertok(memcmp(buf, "AB\0\0\0", sizeof(buf)));
 203
 204        /* Offset length 3 */
 205        sandbox_i2c_eeprom_set_offset_len(eeprom, 2);
 206        ut_assertok(i2c_set_chip_offset_len(dev, 2));
 207        ut_assertok(dm_i2c_write(dev, 0x410, (uint8_t *)"AB", 2));
 208        ut_assertok(dm_i2c_read(dev, 0x410, buf, 5));
 209        ut_assertok(memcmp(buf, "AB\0\0\0", sizeof(buf)));
 210
 211        /* Offset length 4 */
 212        sandbox_i2c_eeprom_set_offset_len(eeprom, 2);
 213        ut_assertok(i2c_set_chip_offset_len(dev, 2));
 214        ut_assertok(dm_i2c_write(dev, 0x420, (uint8_t *)"AB", 2));
 215        ut_assertok(dm_i2c_read(dev, 0x420, buf, 5));
 216        ut_assertok(memcmp(buf, "AB\0\0\0", sizeof(buf)));
 217
 218        /* Restore defaults */
 219        sandbox_i2c_eeprom_set_offset_len(eeprom, 1);
 220
 221        return 0;
 222}
 223DM_TEST(dm_test_i2c_offset, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
 224