]> sigrok.org Git - libsigrok.git/blobdiff - src/hardware/saleae-logic-pro/protocol.c
ols: Use symbolic name for "success" error code.
[libsigrok.git] / src / hardware / saleae-logic-pro / protocol.c
index ed0654616968f0df66f3ec38212d76ce450b8d69..351032af39ff043f1b0ed764b9f8915fb3b39531 100644 (file)
 #define COMMAND_START_CAPTURE  0x01
 #define COMMAND_STOP_CAPTURE   0x02
 #define COMMAND_READ_EEPROM    0x07
+#define COMMAND_INIT_BITSTREAM  0x7e
+#define COMMAND_SEND_BITSTREAM  0x7f
 #define COMMAND_WRITE_REG      0x80
 #define COMMAND_READ_REG       0x81
+#define COMMAND_READ_TEMP      0x86
 #define COMMAND_WRITE_I2C      0x87
 #define COMMAND_READ_I2C       0x88
 #define COMMAND_WAKE_I2C       0x89
 #define COMMAND_READ_FW_VER    0x8b
 
+#define REG_ADC_IDX            0x03
+#define REG_ADC_VAL_LSB                0x04
+#define REG_ADC_VAL_MSB                0x05
 #define REG_LED_RED            0x0f
 #define REG_LED_GREEN          0x10
 #define REG_LED_BLUE           0x11
+#define REG_STATUS             0x40
 
 static void iterate_lfsr(const struct sr_dev_inst *sdi)
 {
@@ -50,11 +57,11 @@ static void iterate_lfsr(const struct sr_dev_inst *sdi)
                          (lfsr >> 31)          \
                          ) << 31);
        }
-       sr_dbg("Iterate 0x%08x -> 0x%08x", devc->lfsr, lfsr);
+       sr_spew("Iterate 0x%08x -> 0x%08x", devc->lfsr, lfsr);
        devc->lfsr = lfsr;
 }
 
-static void encrypt(const struct sr_dev_inst *sdi, const uint8_t *in, uint8_t *out, uint8_t len)
+static void encrypt(const struct sr_dev_inst *sdi, const uint8_t *in, uint8_t *out, uint16_t len)
 {
        struct dev_context *devc = sdi->priv;
        uint32_t lfsr = devc->lfsr;
@@ -63,7 +70,7 @@ static void encrypt(const struct sr_dev_inst *sdi, const uint8_t *in, uint8_t *o
 
        for (i = 0; i < len; i++) {
                value = in[i];
-               mask = lfsr >> (i%4*8);
+               mask = lfsr >> (i % 4 * 8);
                if (i == 0)
                        value = (value & 0x28) | ((value ^ mask) & ~0x28);
                else
@@ -73,28 +80,27 @@ static void encrypt(const struct sr_dev_inst *sdi, const uint8_t *in, uint8_t *o
        iterate_lfsr(sdi);
 }
 
-static void decrypt(const struct sr_dev_inst *sdi, uint8_t *data, uint8_t len)
+static void decrypt(const struct sr_dev_inst *sdi, uint8_t *data, uint16_t len)
 {
        struct dev_context *devc = sdi->priv;
        uint32_t lfsr = devc->lfsr;
        int i;
 
-       for (i = 0; i < len; i++) {
-               data[i] ^= (lfsr >> (i%4*8));
-       }
+       for (i = 0; i < len; i++)
+               data[i] ^= (lfsr >> (i % 4 * 8));
        iterate_lfsr(sdi);
 }
 
 static int transact(const struct sr_dev_inst *sdi,
-                   const uint8_t *req, uint8_t req_len,
-                   uint8_t *rsp, uint8_t rsp_len)
+                   const uint8_t *req, uint16_t req_len,
+                   uint8_t *rsp, uint16_t rsp_len)
 {
        struct sr_usb_dev_inst *usb = sdi->conn;
        uint8_t *req_enc;
        uint8_t rsp_dummy[1] = {};
        int ret, xfer;
 
-       if (req_len < 2 || req_len > 64 || rsp_len > 128 ||
+       if (req_len < 2 || req_len > 1024 || rsp_len > 128 ||
            !req || (rsp_len > 0 && !rsp))
                return SR_ERR_ARG;
 
@@ -113,7 +119,7 @@ static int transact(const struct sr_dev_inst *sdi,
                return SR_ERR;
        }
 
-       if (req[0] == 0x20) { // reseed
+       if (req[0] == 0x20) { /* Reseed. */
                return SR_OK;
        } else if (rsp_len == 0) {
                rsp = rsp_dummy;
@@ -164,7 +170,7 @@ static int write_regs(const struct sr_dev_inst *sdi, uint8_t (*regs)[2], uint8_t
                req[4 + 2 * i] = regs[i][1];
        }
 
-       return transact(sdi, req, 3 + 2*cnt, NULL, 0);
+       return transact(sdi, req, 3 + (2 * cnt), NULL, 0);
 }
 
 static int write_reg(const struct sr_dev_inst *sdi,
@@ -175,6 +181,74 @@ static int write_reg(const struct sr_dev_inst *sdi,
        return write_regs(sdi, &regs, 1);
 }
 
+static int read_regs(const struct sr_dev_inst *sdi,
+                    const uint8_t *regs, uint8_t *values,
+                    uint8_t cnt)
+{
+       uint8_t req[33];
+
+       if (cnt < 1 || cnt > 30)
+               return SR_ERR_ARG;
+
+       req[0] = 0x00;
+       req[1] = COMMAND_READ_REG;
+       req[2] = cnt;
+       if (cnt)
+               memcpy(&req[3], regs, cnt);
+
+       return transact(sdi, req, 3 + cnt, values, cnt);
+}
+
+static int read_reg(const struct sr_dev_inst *sdi,
+                   uint8_t address, uint8_t *value)
+{
+       return read_regs(sdi, &address, value, 1);
+}
+
+static int write_adc(const struct sr_dev_inst *sdi,
+                    uint8_t address, uint16_t value)
+{
+       uint8_t regs[][2] = {
+               {REG_ADC_IDX, address},
+               {REG_ADC_VAL_LSB, value},
+               {REG_ADC_VAL_MSB, value >> 8},
+       };
+
+       return write_regs(sdi, ARRAY_AND_SIZE(regs));
+}
+
+static int read_eeprom(const struct sr_dev_inst *sdi,
+                      uint16_t address, uint8_t *data, uint16_t len)
+{
+       uint8_t req[8] = {
+               0x00, COMMAND_READ_EEPROM,
+               0x33, 0x81, /* Unknown values */
+               address, address >> 8,
+               len, len >> 8
+       };
+
+       return transact(sdi, req, sizeof(req), data, len);
+}
+
+static int read_eeprom_serial(const struct sr_dev_inst *sdi,
+                             uint8_t data[8])
+{
+       return read_eeprom(sdi, 0x08, data, 0x8);
+}
+
+static int read_eeprom_magic(const struct sr_dev_inst *sdi,
+                            uint8_t data[16])
+{
+       return read_eeprom(sdi, 0x10, data, 0x10);
+}
+
+static int read_temperature(const struct sr_dev_inst *sdi, int8_t *temp)
+{
+       uint8_t req[2] = {0x00, COMMAND_READ_TEMP};
+
+       return transact(sdi, req, sizeof(req), (uint8_t*)temp, 1);
+}
+
 static int get_firmware_version(const struct sr_dev_inst *sdi)
 {
        uint8_t req[2] = {0x00, COMMAND_READ_FW_VER};
@@ -193,7 +267,7 @@ static int get_firmware_version(const struct sr_dev_inst *sdi)
 static int read_i2c(const struct sr_dev_inst *sdi, uint8_t *data, uint8_t len)
 {
        uint8_t req[5];
-       uint8_t rsp[1+128];
+       uint8_t rsp[1 + 128];
        int ret;
 
        if (len < 1 || len > 128 || !data)
@@ -201,20 +275,19 @@ static int read_i2c(const struct sr_dev_inst *sdi, uint8_t *data, uint8_t len)
 
        req[0] = 0x00;
        req[1] = COMMAND_READ_I2C;
-       req[2] = 0xc0; // fixed address
+       req[2] = 0xc0; /* Fixed address */
        req[3] = len;
-       req[4] = 0; // len msb?
+       req[4] = 0; /* Len MSB? */
 
        ret = transact(sdi, req, sizeof(req), rsp, 1 + len);
-       if (ret != SR_OK) {
+       if (ret != SR_OK)
                return ret;
-       }
        if (rsp[0] != 0x02) {
                sr_dbg("Failed to do I2C read (0x%02x).", rsp[0]);
                return SR_ERR;
        }
 
-       memcpy(data, rsp+1, len);
+       memcpy(data, rsp + 1, len);
        return SR_OK;
 }
 
@@ -229,15 +302,14 @@ static int write_i2c(const struct sr_dev_inst *sdi, const uint8_t *data, uint8_t
 
        req[0] = 0x00;
        req[1] = COMMAND_WRITE_I2C;
-       req[2] = 0xc0; // fixed address
+       req[2] = 0xc0; /* Fixed address */
        req[3] = len;
-       req[4] = 0; // len msb?
+       req[4] = 0; /* Len MSB? */
        memcpy(req + 5, data, len);
 
        ret = transact(sdi, req, 5 + len, rsp, sizeof(rsp));
-       if (ret != SR_OK) {
+       if (ret != SR_OK)
                return ret;
-       }
        if (rsp[0] != 0x02) {
                sr_dbg("Failed to do I2C write (0x%02x).", rsp[0]);
                return SR_ERR;
@@ -250,13 +322,12 @@ static int wake_i2c(const struct sr_dev_inst *sdi)
 {
        uint8_t req[] = {0x00, COMMAND_WAKE_I2C};
        uint8_t rsp[1] = {};
-       uint8_t i2c_rsp[1+1+2] = {};
+       uint8_t i2c_rsp[1 + 1 + 2] = {};
        int ret;
 
        ret = transact(sdi, req, sizeof(req), rsp, sizeof(rsp));
-       if (ret != SR_OK) {
+       if (ret != SR_OK)
                return ret;
-       }
        if (rsp[0] != 0x00) {
                sr_dbg("Failed to do I2C wake trigger (0x%02x).", rsp[0]);
                return SR_ERR;
@@ -277,53 +348,47 @@ static int wake_i2c(const struct sr_dev_inst *sdi)
 static int crypto_random(const struct sr_dev_inst *sdi, uint8_t *data)
 {
        uint8_t i2c_req[8] = {0x03, 0x07, 0x1b, 0x00, 0x00, 0x00, 0x24, 0xcd};
-       uint8_t i2c_rsp[1+32+2] = {};
+       uint8_t i2c_rsp[1 + 32 + 2] = {};
        int ret;
 
        ret = write_i2c(sdi, i2c_req, sizeof(i2c_req));
-       if (ret != SR_OK) {
+       if (ret != SR_OK)
                return ret;
-       }
 
-       g_usleep(100000); // TODO: poll instead
+       g_usleep(100000); /* TODO: Poll instead. */
 
        ret = read_i2c(sdi, i2c_rsp, sizeof(i2c_rsp));
-       if (ret != SR_OK) {
+       if (ret != SR_OK)
                return ret;
-       }
 
-       if (data) {
-               memcpy(data, i2c_rsp+1, 32);
-       }
+       if (data)
+               memcpy(data, i2c_rsp + 1, 32);
 
        return SR_OK;
 }
 
 static int crypto_nonce(const struct sr_dev_inst *sdi, uint8_t *data)
 {
-       uint8_t i2c_req[6+20+2] = {0x03, 0x1b, 0x16, 0x00, 0x00, 0x00};
-       uint8_t i2c_rsp[1+32+2] = {};
+       uint8_t i2c_req[6 + 20 + 2] = {0x03, 0x1b, 0x16, 0x00, 0x00, 0x00};
+       uint8_t i2c_rsp[1 + 32 + 2] = {};
        int ret;
 
-       // CRC
+       /* CRC */
        i2c_req[26] = 0x7d;
        i2c_req[27] = 0xe0;
 
        ret = write_i2c(sdi, i2c_req, sizeof(i2c_req));
-       if (ret != SR_OK) {
+       if (ret != SR_OK)
                return ret;
-       }
 
-       g_usleep(100000); // TODO: poll instead
+       g_usleep(100000); /* TODO: Poll instead. */
 
        ret = read_i2c(sdi, i2c_rsp, sizeof(i2c_rsp));
-       if (ret != SR_OK) {
+       if (ret != SR_OK)
                return ret;
-       }
 
-       if (data) {
-               memcpy(data, i2c_rsp+1, 32);
-       }
+       if (data)
+               memcpy(data, i2c_rsp + 1, 32);
 
        return SR_OK;
 }
@@ -331,23 +396,21 @@ static int crypto_nonce(const struct sr_dev_inst *sdi, uint8_t *data)
 static int crypto_sign(const struct sr_dev_inst *sdi, uint8_t *data, uint8_t *crc)
 {
        uint8_t i2c_req[8] = {0x03, 0x07, 0x41, 0x80, 0x00, 0x00, 0x28, 0x05};
-       uint8_t i2c_rsp[1+64+2] = {};
+       uint8_t i2c_rsp[1 + 64 + 2] = {};
        int ret;
 
        ret = write_i2c(sdi, i2c_req, sizeof(i2c_req));
-       if (ret != SR_OK) {
+       if (ret != SR_OK)
                return ret;
-       }
 
-       g_usleep(100000); // TODO: poll instead
+       g_usleep(100000); /* TODO: Poll instead. */
 
        ret = read_i2c(sdi, i2c_rsp, sizeof(i2c_rsp));
-       if (ret != SR_OK) {
+       if (ret != SR_OK)
                return ret;
-       }
 
-       memcpy(data, i2c_rsp+1, 64);
-       memcpy(crc, i2c_rsp+1+64, 2);
+       memcpy(data, i2c_rsp + 1, 64);
+       memcpy(crc, i2c_rsp + 1 + 64, 2);
 
        return SR_OK;
 }
@@ -389,7 +452,7 @@ static int authenticate(const struct sr_dev_inst *sdi)
 
        lfsr = 0;
        for (i = 0; i < 28; i++)
-               lfsr ^= nonce[i] << (8*(i%4));
+               lfsr ^= nonce[i] << (8 * (i % 4));
        lfsr ^= sig_crc[0] | sig_crc[1] << 8;
 
        sr_dbg("Authenticate 0x%08x -> 0x%08x", devc->lfsr, lfsr);
@@ -398,6 +461,94 @@ static int authenticate(const struct sr_dev_inst *sdi)
        return SR_OK;
 }
 
+static int upload_bitstream_part(const struct sr_dev_inst *sdi,
+                                const uint8_t *data, uint16_t len)
+{
+       uint8_t req[4 + 1020];
+       uint8_t rsp[1];
+       int ret;
+
+       if (len < 1 || len > 1020 || !data)
+               return SR_ERR_ARG;
+
+       req[0] = 0x00;
+       req[1] = COMMAND_SEND_BITSTREAM;
+       req[2] = len;
+       req[3] = len >> 8;
+       memcpy(req + 4, data, len);
+
+       ret = transact(sdi, req, 4 + len, rsp, sizeof(rsp));
+       if (ret != SR_OK)
+               return ret;
+       if (rsp[0] != 0x00) {
+               sr_dbg("Failed to do bitstream upload (0x%02x).", rsp[0]);
+               return SR_ERR;
+       }
+
+       return SR_OK;
+}
+
+static int upload_bitstream(const struct sr_dev_inst *sdi,
+                           const char *name)
+{
+       struct drv_context *drvc = sdi->driver->context;
+       unsigned char *bitstream = NULL;
+       uint8_t req[2];
+       uint8_t rsp[1];
+       uint8_t reg_val;
+       int ret = SR_ERR;
+       size_t bs_size, bs_offset = 0, bs_part_size;
+
+       bitstream = sr_resource_load(drvc->sr_ctx, SR_RESOURCE_FIRMWARE,
+                                    name, &bs_size, 512 * 1024);
+       if (!bitstream)
+               goto out;
+
+       sr_info("Uploading bitstream '%s'.", name);
+
+       req[0] = 0x00;
+       req[1] = COMMAND_INIT_BITSTREAM;
+
+       ret = transact(sdi, req, sizeof(req), rsp, sizeof(rsp));
+       if (ret != SR_OK)
+               return ret;
+       if (rsp[0] != 0x00) {
+               sr_err("Failed to start bitstream upload (0x%02x).", rsp[0]);
+               ret = SR_ERR;
+               goto out;
+       }
+
+       while (bs_offset < bs_size) {
+               bs_part_size = MIN(bs_size - bs_offset, 1020);
+               sr_spew("Uploading %zd bytes.", bs_part_size);
+               ret = upload_bitstream_part(sdi, bitstream + bs_offset, bs_part_size);
+               if (ret != SR_OK)
+                       goto out;
+               bs_offset += bs_part_size;
+       }
+
+       sr_info("Bitstream upload done.");
+
+       /* Check a scratch register? */
+       ret = write_reg(sdi, 0x7f, 0xaa);
+       if (ret != SR_OK)
+               goto out;
+       ret = read_reg(sdi, 0x7f, &reg_val);
+       if (ret != SR_OK)
+               goto out;
+       if (reg_val != 0xaa) {
+               sr_err("Failed FPGA register read-back (0x%02x != 0xaa).", rsp[0]);
+               ret = SR_ERR;
+               goto out;
+       }
+
+ out:
+       g_free(bitstream);
+
+       return ret;
+}
+
+#if 0
 static int set_led(const struct sr_dev_inst *sdi, uint8_t r, uint8_t g, uint8_t b)
 {
        uint8_t regs[][2] = {
@@ -408,8 +559,9 @@ static int set_led(const struct sr_dev_inst *sdi, uint8_t r, uint8_t g, uint8_t
 
        authenticate(sdi);
 
-       return write_regs(sdi, regs, G_N_ELEMENTS(regs));
+       return write_regs(sdi, ARRAY_AND_SIZE(regs));
 }
+#endif
 
 static int configure_channels(const struct sr_dev_inst *sdi)
 {
@@ -431,17 +583,139 @@ static int configure_channels(const struct sr_dev_inst *sdi)
 
        }
        sr_dbg("%d channels enabled (0x%04x)",
-              devc->dig_channel_cnt,
-              devc->dig_channel_mask);
+              devc->dig_channel_cnt, devc->dig_channel_mask);
 
        return SR_OK;
 }
 
 SR_PRIV int saleae_logic_pro_init(const struct sr_dev_inst *sdi)
 {
-       reseed(sdi);
-       get_firmware_version(sdi);
-       /* setting the LED doesn't work yet */
+       uint8_t reg_val;
+       uint8_t dummy[8];
+       uint8_t serial[8];
+       uint8_t magic[16];
+       int8_t temperature;
+       int ret, i;
+
+       ret = reseed(sdi);
+       if (ret != SR_OK)
+               return ret;
+
+       ret = get_firmware_version(sdi);
+       if (ret != SR_OK)
+               return ret;
+
+       sr_dbg("read serial");
+       ret = read_eeprom_serial(sdi, serial);
+       if (ret != SR_OK)
+               return ret;
+
+       /* Check if we need to upload the bitstream. */
+       ret = read_reg(sdi, 0x7f, &reg_val);
+       if (ret != SR_OK)
+               return ret;
+       if (reg_val == 0xaa) {
+               sr_info("Skipping bitstream upload.");
+       } else {
+               ret = upload_bitstream(sdi, "saleae-logicpro16-fpga.bitstream");
+               if (ret != SR_OK)
+                       return ret;
+       }
+
+       /* Reset the ADC? */
+       sr_dbg("reset ADC");
+       ret = write_reg(sdi, 0x00, 0x00);
+       if (ret != SR_OK)
+               return ret;
+       ret = write_reg(sdi, 0x00, 0x80);
+       if (ret != SR_OK)
+               return ret;
+
+       sr_dbg("init ADC");
+       ret = write_adc(sdi, 0x11, 0x0444);
+       if (ret != SR_OK)
+               return ret;
+       ret = write_adc(sdi, 0x12, 0x0777);
+       if (ret != SR_OK)
+               return ret;
+       ret = write_adc(sdi, 0x25, 0x0000);
+       if (ret != SR_OK)
+               return ret;
+       ret = write_adc(sdi, 0x45, 0x0000);
+       if (ret != SR_OK)
+               return ret;
+       ret = write_adc(sdi, 0x2a, 0x1111);
+       if (ret != SR_OK)
+               return ret;
+       ret = write_adc(sdi, 0x2b, 0x1111);
+       if (ret != SR_OK)
+               return ret;
+       ret = write_adc(sdi, 0x46, 0x0004);
+       if (ret != SR_OK)
+               return ret;
+       ret = write_adc(sdi, 0x50, 0x0000);
+       if (ret != SR_OK)
+               return ret;
+       ret = write_adc(sdi, 0x55, 0x0020);
+       if (ret != SR_OK)
+               return ret;
+       ret = write_adc(sdi, 0x56, 0x0000);
+       if (ret != SR_OK)
+               return ret;
+
+       ret = write_reg(sdi, 0x15, 0x00);
+       if (ret != SR_OK)
+               return ret;
+
+       ret = write_adc(sdi, 0x0f, 0x0100);
+       if (ret != SR_OK)
+               return ret;
+
+       /* Resets? */
+       sr_dbg("resets");
+       ret = write_reg(sdi, 0x00, 0x02); /* bit 1 */
+       if (ret != SR_OK)
+               return ret;
+       ret = write_reg(sdi, 0x00, 0x00);
+       if (ret != SR_OK)
+               return ret;
+       ret = write_reg(sdi, 0x00, 0x04); /* bit 2 */
+       if (ret != SR_OK)
+               return ret;
+       ret = write_reg(sdi, 0x00, 0x00);
+       if (ret != SR_OK)
+               return ret;
+       ret = write_reg(sdi, 0x00, 0x08); /* bit 3 */
+       if (ret != SR_OK)
+               return ret;
+       ret = write_reg(sdi, 0x00, 0x00);
+       if (ret != SR_OK)
+               return ret;
+
+       sr_dbg("read dummy");
+       for (i = 0; i < 8; i++) {
+               ret = read_reg(sdi, 0x41 + i, &dummy[i]);
+               if (ret != SR_OK)
+                       return ret;
+       }
+
+       /* Read and write back magic EEPROM value. */
+       sr_dbg("read/write magic");
+       ret = read_eeprom_magic(sdi, magic);
+       if (ret != SR_OK)
+               return ret;
+       for (i = 0; i < 16; i++) {
+               ret = write_reg(sdi, 0x17, magic[i]);
+               if (ret != SR_OK)
+                       return ret;
+       }
+
+       ret = read_temperature(sdi, &temperature);
+       if (ret != SR_OK)
+               return ret;
+       sr_dbg("temperature = %d", temperature);
+
+       /* Setting the LED doesn't work yet. */
        /* set_led(sdi, 0x00, 0x00, 0xff); */
 
        return SR_OK;
@@ -457,30 +731,30 @@ SR_PRIV int saleae_logic_pro_prepare(const struct sr_dev_inst *sdi)
        };
        uint8_t regs_config[][2] = {
                {0x00, 0x00},
-               {0x08, 0x00}, /* analog channel mask (LSB) */
-               {0x09, 0x00}, /* analog channel mask (MSB) */
-               {0x06, 0x01}, /* digital channel mask (LSB) */
-               {0x07, 0x00}, /* digital channel mask (MSB) */
-               {0x0a, 0x00}, /* analog sample rate? */
-               {0x0b, 0x64}, /* digital sample rate? */
+               {0x08, 0x00}, /* Analog channel mask (LSB) */
+               {0x09, 0x00}, /* Analog channel mask (MSB) */
+               {0x06, 0x01}, /* Digital channel mask (LSB) */
+               {0x07, 0x00}, /* Digital channel mask (MSB) */
+               {0x0a, 0x00}, /* Analog sample rate? */
+               {0x0b, 0x64}, /* Digital sample rate? */
                {0x0c, 0x00},
-               {0x0d, 0x00}, /* analog mux rate? */
-               {0x0e, 0x01}, /* digital mux rate? */
+               {0x0d, 0x00}, /* Analog mux rate? */
+               {0x0e, 0x01}, /* Digital mux rate? */
                {0x12, 0x04},
                {0x13, 0x00},
-               {0x14, 0xff}, /* pre-divider? */
+               {0x14, 0xff}, /* Pre-divider? */
        };
        uint8_t start_req[] = {0x00, 0x01};
        uint8_t start_rsp[2] = {};
 
        configure_channels(sdi);
 
-       /* digital channel mask and muxing */
+       /* Digital channel mask and muxing */
        regs_config[3][1] = devc->dig_channel_mask;
        regs_config[4][1] = devc->dig_channel_mask >> 8;
        regs_config[9][1] = devc->dig_channel_cnt;
 
-       /* samplerate */
+       /* Samplerate */
        switch (devc->dig_samplerate) {
        case SR_MHZ(1):
                regs_config[6][1] = 0x64;
@@ -509,8 +783,8 @@ SR_PRIV int saleae_logic_pro_prepare(const struct sr_dev_inst *sdi)
        authenticate(sdi);
 
        write_reg(sdi, 0x15, 0x03);
-       write_regs(sdi, regs_unknown, G_N_ELEMENTS(regs_unknown));
-       write_regs(sdi, regs_config, G_N_ELEMENTS(regs_config));
+       write_regs(sdi, ARRAY_AND_SIZE(regs_unknown));
+       write_regs(sdi, ARRAY_AND_SIZE(regs_config));
 
        transact(sdi, start_req, sizeof(start_req), start_rsp, sizeof(start_rsp));
 
@@ -533,10 +807,20 @@ SR_PRIV int saleae_logic_pro_stop(const struct sr_dev_inst *sdi)
 {
        uint8_t stop_req[] = {0x00, 0x02};
        uint8_t stop_rsp[2] = {};
+       uint8_t status;
+       int ret;
 
        write_reg(sdi, 0x00, 0x00);
        transact(sdi, stop_req, sizeof(stop_req), stop_rsp, sizeof(stop_rsp));
 
+       ret = read_reg(sdi, 0x40, &status);
+       if (ret != SR_OK)
+               return ret;
+       if (status != 0x20) {
+               sr_err("Capture error (status reg = 0x%02x).", status);
+               return SR_ERR;
+       }
+
        return SR_OK;
 }
 
@@ -571,9 +855,9 @@ static void saleae_logic_pro_convert_data(const struct sr_dev_inst *sdi,
        unsigned int sample_index, batch_index;
        uint16_t *dst_batch;
 
-       /* copy partial batch to the beginning */
-       memcpy(dst, dst+devc->conv_size, CONV_BATCH_SIZE);
-       /* reset converted size */
+       /* Copy partial batch to the beginning. */
+       memcpy(dst, dst + devc->conv_size, CONV_BATCH_SIZE);
+       /* Reset converted size. */
        devc->conv_size = 0;
 
        batch_index = devc->batch_index;
@@ -581,17 +865,17 @@ static void saleae_logic_pro_convert_data(const struct sr_dev_inst *sdi,
                samples = *src++;
                dst_batch = (uint16_t*)dst;
 
-               /* first index of the batch */
+               /* First index of the batch. */
                if (batch_index == 0)
                        memset(dst, 0, CONV_BATCH_SIZE);
 
-               /* convert one channel */
+               /* Convert one channel. */
                channel_mask = devc->dig_channel_masks[batch_index];
                for (sample_index = 0; sample_index <= 31; sample_index++)
-                       if ((samples >> (31-sample_index)) & 1)
+                       if ((samples >> (31 - sample_index)) & 1)
                                dst_batch[sample_index] |= channel_mask;
 
-               /* last index of the batch */
+               /* Last index of the batch. */
                if (++batch_index == devc->dig_channel_cnt) {
                        devc->conv_size += CONV_BATCH_SIZE;
                        batch_index = 0;
@@ -615,11 +899,11 @@ SR_PRIV void LIBUSB_CALL saleae_logic_pro_receive_data(struct libusb_transfer *t
        case LIBUSB_TRANSFER_TIMED_OUT: /* We may have received some data though. */
                break;
        default:
-               // FIXME
+               /* FIXME */
                return;
        }
 
-       saleae_logic_pro_convert_data(sdi, (uint32_t*)transfer->buffer, 16*1024/4);
+       saleae_logic_pro_convert_data(sdi, (uint32_t*)transfer->buffer, 16 * 1024 / 4);
        saleae_logic_pro_send_data(sdi, devc->conv_buffer, devc->conv_size, 2);
 
        if ((ret = libusb_submit_transfer(transfer)) != LIBUSB_SUCCESS)