]> sigrok.org Git - libsigrok.git/blobdiff - hardware/openbench-logic-sniffer/ols.c
Add sr_ prefix to session_{add,remove}.
[libsigrok.git] / hardware / openbench-logic-sniffer / ols.c
index df4edd2ddea9c3fd2e07f90b896f30e02a65872e..3c17e7e1201c112c80405ee3dbce29bd54b0de3d 100644 (file)
 #include <sys/stat.h>
 #include <fcntl.h>
 #include <unistd.h>
+#ifdef _WIN32
+#include <windows.h>
+#else
 #include <termios.h>
+#endif
 #include <string.h>
 #include <sys/time.h>
 #include <inttypes.h>
+#ifdef _WIN32
+/* TODO */
+#else
+#include <arpa/inet.h>
+#endif
 #include <glib.h>
-#include "sigrok.h"
-
-#define NUM_PROBES                             32
-#define NUM_TRIGGER_STAGES             4
-#define TRIGGER_TYPES                  "01"
-#define SERIAL_SPEED                   B115200
-/* TODO: SERIAL_ bits, parity, stop bit */
-#define CLOCK_RATE                             100000000
-
-
-/* command opcodes */
-#define CMD_RESET                                      0x00
-#define CMD_ID                                         0x02
-#define CMD_SET_FLAGS                          0x82
-#define CMD_SET_DIVIDER                        0x80
-#define CMD_RUN                                        0x01
-#define CMD_CAPTURE_SIZE                       0x81
-#define CMD_SET_TRIGGER_MASK_0         0xc0
-#define CMD_SET_TRIGGER_MASK_1         0xc4
-#define CMD_SET_TRIGGER_MASK_2         0xc8
-#define CMD_SET_TRIGGER_MASK_3         0xcc
-#define CMD_SET_TRIGGER_VALUE_0        0xc1
-#define CMD_SET_TRIGGER_VALUE_1        0xc5
-#define CMD_SET_TRIGGER_VALUE_2        0xc9
-#define CMD_SET_TRIGGER_VALUE_3        0xcd
-#define CMD_SET_TRIGGER_CONFIG_0       0xc2
-#define CMD_SET_TRIGGER_CONFIG_1       0xc6
-#define CMD_SET_TRIGGER_CONFIG_2       0xca
-#define CMD_SET_TRIGGER_CONFIG_3       0xce
-
-/* bitmasks for CMD_FLAGS */
-#define FLAG_DEMUX                             0x01
-#define FLAG_FILTER                    0x02
-#define FLAG_CHANNELGROUP_1    0x04
-#define FLAG_CHANNELGROUP_2    0x08
-#define FLAG_CHANNELGROUP_3    0x10
-#define FLAG_CHANNELGROUP_4    0x20
-#define FLAG_CLOCK_EXTERNAL    0x40
-#define FLAG_CLOCK_INVERTED    0x80
-#define FLAG_RLE                               0x0100
-
+#include <sigrok.h>
+#include <sigrok-internal.h>
+
+#ifdef _WIN32
+#define O_NONBLOCK FIONBIO
+#endif
+
+#define NUM_PROBES             32
+#define NUM_TRIGGER_STAGES     4
+#define TRIGGER_TYPES          "01"
+#define SERIAL_SPEED           B115200
+#define CLOCK_RATE             MHZ(100)
+#define MIN_NUM_SAMPLES        4
+
+/* Command opcodes */
+#define CMD_RESET                  0x00
+#define CMD_ID                     0x02
+#define CMD_SET_FLAGS              0x82
+#define CMD_SET_DIVIDER            0x80
+#define CMD_RUN                    0x01
+#define CMD_CAPTURE_SIZE           0x81
+#define CMD_SET_TRIGGER_MASK_0     0xc0
+#define CMD_SET_TRIGGER_MASK_1     0xc4
+#define CMD_SET_TRIGGER_MASK_2     0xc8
+#define CMD_SET_TRIGGER_MASK_3     0xcc
+#define CMD_SET_TRIGGER_VALUE_0    0xc1
+#define CMD_SET_TRIGGER_VALUE_1    0xc5
+#define CMD_SET_TRIGGER_VALUE_2    0xc9
+#define CMD_SET_TRIGGER_VALUE_3    0xcd
+#define CMD_SET_TRIGGER_CONFIG_0   0xc2
+#define CMD_SET_TRIGGER_CONFIG_1   0xc6
+#define CMD_SET_TRIGGER_CONFIG_2   0xca
+#define CMD_SET_TRIGGER_CONFIG_3   0xce
+
+/* Bitmasks for CMD_FLAGS */
+#define FLAG_DEMUX                 0x01
+#define FLAG_FILTER                0x02
+#define FLAG_CHANNELGROUP_1        0x04
+#define FLAG_CHANNELGROUP_2        0x08
+#define FLAG_CHANNELGROUP_3        0x10
+#define FLAG_CHANNELGROUP_4        0x20
+#define FLAG_CLOCK_EXTERNAL        0x40
+#define FLAG_CLOCK_INVERTED        0x80
+#define FLAG_RLE                   0x0100
 
 static int capabilities[] = {
-       HWCAP_LOGIC_ANALYZER,
-       HWCAP_SAMPLERATE,
-       HWCAP_CAPTURE_RATIO,
-       HWCAP_LIMIT_SAMPLES,
-       0
+       SR_HWCAP_LOGIC_ANALYZER,
+       SR_HWCAP_SAMPLERATE,
+       SR_HWCAP_CAPTURE_RATIO,
+       SR_HWCAP_LIMIT_SAMPLES,
+       0,
 };
 
-static struct samplerates samplerates = {
-       1,
+static struct sr_samplerates samplerates = {
+       10,
        MHZ(200),
        1,
-       0
+       0,
 };
 
-/* list of struct serial_device_instance  */
+/* List of struct sr_serial_device_instance */
 static GSList *device_instances = NULL;
 
-/* current state of the flag register */
-uint32_t flag_reg = 0;
+/* Current state of the flag register */
+static uint32_t flag_reg = 0;
 
 static uint64_t cur_samplerate = 0;
 static uint64_t limit_samples = 0;
-/* pre/post trigger capture ratio, in percentage. 0 means no pre-trigger data. */
-int capture_ratio = 0;
-static uint32_t probe_mask = 0, trigger_mask[4] = {0}, trigger_value[4] = {0};
-
-
-
-int send_shortcommand(int fd, uint8_t command)
+/*
+ * Pre/post trigger capture ratio, in percentage.
+ * 0 means no pre-trigger data.
+ */
+static int capture_ratio = 0;
+static int trigger_at = -1;
+static uint32_t probe_mask = 0xffffffff;
+static uint32_t trigger_mask[4] = { 0, 0, 0, 0 };
+static uint32_t trigger_value[4] = { 0, 0, 0, 0 };
+static int num_stages = 0;
+
+static int send_shortcommand(int fd, uint8_t command)
 {
        char buf[1];
 
-       g_message("ols: sending cmd 0x%.2x", command);
+       g_debug("ols: sending cmd 0x%.2x", command);
        buf[0] = command;
-       if(write(fd, buf, 1) != 1)
-               return SIGROK_ERR;
+       if (serial_write(fd, buf, 1) != 1)
+               return SR_ERR;
 
-       return SIGROK_OK;
+       return SR_OK;
 }
 
-
-int send_longcommand(int fd, uint8_t command, uint32_t data)
+static int send_longcommand(int fd, uint8_t command, uint32_t data)
 {
        char buf[5];
 
-       g_message("ols: sending cmd 0x%.2x data 0x%.8x", command, data);
+       g_debug("ols: sending cmd 0x%.2x data 0x%.8x", command, data);
        buf[0] = command;
-       buf[1] = data & 0xff;
-       buf[2] = data & 0xff00 >> 8;
-       buf[3] = data & 0xff0000 >> 16;
-       buf[4] = data & 0xff000000 >> 24;
-       if(write(fd, buf, 5) != 5)
-               return SIGROK_ERR;
-
-       return SIGROK_OK;
+       buf[1] = (data & 0xff000000) >> 24;
+       buf[2] = (data & 0xff0000) >> 16;
+       buf[3] = (data & 0xff00) >> 8;
+       buf[4] = data & 0xff;
+       if (serial_write(fd, buf, 5) != 5)
+               return SR_ERR;
+
+       return SR_OK;
 }
 
 static int configure_probes(GSList *probes)
 {
-       struct probe *probe;
+       struct sr_probe *probe;
        GSList *l;
-       int probe_bit, changrp_mask, stage, i;
+       int probe_bit, stage, i;
        char *tc;
 
        probe_mask = 0;
-       for(i = 0; i < NUM_TRIGGER_STAGES; i++)
-       {
+       for (i = 0; i < NUM_TRIGGER_STAGES; i++) {
                trigger_mask[i] = 0;
                trigger_value[i] = 0;
        }
 
-       for(l = probes; l; l = l->next)
-       {
-               probe = (struct probe *) l->data;
+       num_stages = 0;
+       for (l = probes; l; l = l->next) {
+               probe = (struct sr_probe *)l->data;
+               if (!probe->enabled)
+                       continue;
+
+               /*
+                * Set up the probe mask for later configuration into the
+                * flag register.
+                */
                probe_bit = 1 << (probe->index - 1);
                probe_mask |= probe_bit;
-               if(probe->trigger)
-               {
-                       stage = 0;
-                       for(tc = probe->trigger; *tc; tc++)
-                       {
-                               trigger_mask[stage] |= probe_bit;
-                               if(*tc == '1')
-                                       trigger_value[stage] |= probe_bit;
-                               stage++;
-                               if(stage > 3)
-                               {
-                                       /* only supporting parallel mode, with up to 4 stages */
-                                       return SIGROK_ERR;
-                               }
-                       }
+
+               if (!probe->trigger)
+                       continue;
+
+               /* Configure trigger mask and value. */
+               stage = 0;
+               for (tc = probe->trigger; tc && *tc; tc++) {
+                       trigger_mask[stage] |= probe_bit;
+                       if (*tc == '1')
+                               trigger_value[stage] |= probe_bit;
+                       stage++;
+                       if (stage > 3)
+                               /*
+                                * TODO: Only supporting parallel mode, with
+                                * up to 4 stages.
+                                */
+                               return SR_ERR;
                }
+               if (stage > num_stages)
+                       num_stages = stage;
        }
 
-       /* enable/disable channel groups in the flag register according
-        * to the probe mask we just made. The register stores them backwards,
-        * hence shift right from 1000.
-        */
-       changrp_mask = 0;
-       for(i = 0; i < 4; i++)
-       {
-               if(probe_mask & (0xff << i))
-                       changrp_mask |= 8 >> i;
-       }
-       /* but the flag register wants them here */
-       flag_reg |= changrp_mask << 2;
+       return SR_OK;
+}
+
+static uint32_t reverse16(uint32_t in)
+{
+       uint32_t out;
 
-       return SIGROK_OK;
+       out = (in & 0xff) << 8;
+       out |= (in & 0xff00) >> 8;
+       out |= (in & 0xff0000) << 8;
+       out |= (in & 0xff000000) >> 8;
+
+       return out;
 }
 
+static uint32_t reverse32(uint32_t in)
+{
+       uint32_t out;
+
+       out = (in & 0xff) << 24;
+       out |= (in & 0xff00) << 8;
+       out |= (in & 0xff0000) >> 8;
+       out |= (in & 0xff000000) >> 24;
+
+       return out;
+}
 
 static int hw_init(char *deviceinfo)
 {
-       struct sigrok_device_instance *sdi;
+       struct sr_device_instance *sdi;
        GSList *ports, *l;
        GPollFD *fds;
-       struct termios term, *prev_termios;
-       int devcnt, final_devcnt, num_ports, fd, i;
-       char buf[8], **device_names;
+       int devcnt, final_devcnt, num_ports, fd, ret, i;
+       char buf[8], **device_names, **serial_params;
 
-       if(deviceinfo)
-               ports = g_slist_append(NULL, g_strdup(deviceinfo));
+       if (deviceinfo)
+               ports = g_slist_append(NULL, strdup(deviceinfo));
        else
-               /* no specific device given, so scan all serial ports */
+               /* No specific device given, so scan all serial ports. */
                ports = list_serial_ports();
 
        num_ports = g_slist_length(ports);
-       fds = g_malloc0(num_ports * sizeof(GPollFD));
-       device_names = g_malloc(num_ports * (sizeof(char *)));
-       prev_termios = g_malloc(num_ports * sizeof(struct termios));
+       fds = calloc(1, num_ports * sizeof(GPollFD));
+       device_names = malloc(num_ports * sizeof(char *));
+       serial_params = malloc(num_ports * sizeof(char *));
        devcnt = 0;
-       for(l = ports; l; l = l->next) {
-               /* The discovery procedure is like this: first send the Reset command (0x00) 5 times,
-                * since the device could be anywhere in a 5-byte command. Then send the ID command
-                * (0x02). If the device responds with 4 bytes ("OLS1" or "SLA1"), we have a match.
-                * Since it may take the device a while to respond at 115Kb/s, we do all the sending
-                * first, then wait for all of them to respond with g_poll().
+       for (l = ports; l; l = l->next) {
+               /* The discovery procedure is like this: first send the Reset
+                * command (0x00) 5 times, since the device could be anywhere
+                * in a 5-byte command. Then send the ID command (0x02).
+                * If the device responds with 4 bytes ("OLS1" or "SLA1"), we
+                * have a match.
+                *
+                * Since it may take the device a while to respond at 115Kb/s,
+                * we do all the sending first, then wait for all of them to
+                * respond with g_poll().
                 */
-               fd = open(l->data, O_RDWR | O_NONBLOCK);
-               if(fd != -1) {
-                       tcgetattr(fd, &prev_termios[devcnt]);
-                       tcgetattr(fd, &term);
-                       cfsetispeed(&term, SERIAL_SPEED);
-                       tcsetattr(fd, TCSADRAIN, &term);
-                       memset(buf, CMD_RESET, 5);
-                       buf[5] = CMD_ID;
-                       if(write(fd, buf, 6) == 6) {
-                               fds[devcnt].fd = fd;
-                               fds[devcnt].events = G_IO_IN;
-                               device_names[devcnt] = l->data;
-                               devcnt++;
-                               g_message("probed device %s", (char *) l->data);
+               g_message("ols: probing %s...", (char *)l->data);
+               fd = serial_open(l->data, O_RDWR | O_NONBLOCK);
+               if (fd != -1) {
+                       serial_params[devcnt] = serial_backup_params(fd);
+                       serial_set_params(fd, 115200, 8, 0, 1, 2);
+                       ret = SR_OK;
+                       for (i = 0; i < 5; i++) {
+                               if ((ret = send_shortcommand(fd,
+                                       CMD_RESET)) != SR_OK) {
+                                       /* Serial port is not writable. */
+                                       break;
+                               }
                        }
-                       else {
-                               /* restore port settings. of course, we've crapped all over the port. */
-                               tcsetattr(fd, TCSADRAIN, &prev_termios[devcnt]);
-                               g_free(l->data);
+                       if (ret != SR_OK) {
+                               serial_restore_params(fd,
+                                       serial_params[devcnt]);
+                               serial_close(fd);
+                               continue;
                        }
+                       send_shortcommand(fd, CMD_ID);
+                       fds[devcnt].fd = fd;
+                       fds[devcnt].events = G_IO_IN;
+                       device_names[devcnt] = strdup(l->data);
+                       devcnt++;
                }
+               free(l->data);
        }
 
-       /* 2ms should do it, that's enough time for 28 bytes to go over the bus */
-       usleep(2000);
+       /* 2ms isn't enough for reliable transfer with pl2303, let's try 10 */
+       usleep(10000);
 
        final_devcnt = 0;
        g_poll(fds, devcnt, 1);
-       for(i = 0; i < devcnt; i++) {
-               if(fds[i].revents == G_IO_IN) {
-                       if(read(fds[i].fd, buf, 4) == 4) {
-                               if(!strncmp(buf, "1SLO", 4) || !strncmp(buf, "1ALS", 4)) {
-                                       if(!strncmp(buf, "1SLO", 4))
-                                               sdi = sigrok_device_instance_new(final_devcnt, ST_INACTIVE,
-                                                               "Openbench", "Logic Sniffer", "v1.0");
+       for (i = 0; i < devcnt; i++) {
+               if (fds[i].revents == G_IO_IN) {
+                       if (serial_read(fds[i].fd, buf, 4) == 4) {
+                               if (!strncmp(buf, "1SLO", 4)
+                                   || !strncmp(buf, "1ALS", 4)) {
+                                       if (!strncmp(buf, "1SLO", 4))
+                                               sdi = sr_device_instance_new
+                                                   (final_devcnt, SR_ST_INACTIVE,
+                                                    "Openbench",
+                                                    "Logic Sniffer", "v1.0");
                                        else
-                                               sdi = sigrok_device_instance_new(final_devcnt, ST_INACTIVE,
-                                                               "Sump", "Logic Analyzer", "v1.0");
-                                       sdi->serial = serial_device_instance_new(device_names[i], -1);
-                                       device_instances = g_slist_append(device_instances, sdi);
+                                               sdi = sr_device_instance_new
+                                                   (final_devcnt, SR_ST_INACTIVE,
+                                                    "Openbench", "Logic Sniffer",
+                                                    "v1.0");
+                                       sdi->serial = sr_serial_device_instance_new
+                                           (device_names[i], -1);
+                                       device_instances =
+                                           g_slist_append(device_instances, sdi);
                                        final_devcnt++;
+                                       serial_close(fds[i].fd);
                                        fds[i].fd = 0;
                                }
                        }
+                       free(device_names[i]);
                }
 
-               if(fds[i].fd != 0)
-                       tcsetattr(fds[i].fd, TCSADRAIN, &prev_termios[i]);
-               close(fds[i].fd);
+               if (fds[i].fd != 0) {
+                       serial_restore_params(fds[i].fd, serial_params[i]);
+                       serial_close(fds[i].fd);
+               }
+               free(serial_params[i]);
        }
 
-       g_free(fds);
-       g_free(device_names);
-       g_free(prev_termios);
+       free(fds);
+       free(device_names);
+       free(serial_params);
        g_slist_free(ports);
 
+       cur_samplerate = KHZ(200);
+
        return final_devcnt;
 }
 
-
 static int hw_opendev(int device_index)
 {
-       struct sigrok_device_instance *sdi;
+       struct sr_device_instance *sdi;
 
-       if(!(sdi = get_sigrok_device_instance(device_instances, device_index)))
-               return SIGROK_ERR;
+       if (!(sdi = sr_get_device_instance(device_instances, device_index)))
+               return SR_ERR;
 
-       sdi->serial->fd = open(sdi->serial->port, O_RDWR);
-       if(sdi->serial->fd == -1)
-               return SIGROK_ERR;
+       sdi->serial->fd = serial_open(sdi->serial->port, O_RDWR);
+       if (sdi->serial->fd == -1)
+               return SR_ERR;
 
-       sdi->status = ST_ACTIVE;
+       sdi->status = SR_ST_ACTIVE;
 
-       return SIGROK_OK;
+       return SR_OK;
 }
 
-
 static void hw_closedev(int device_index)
 {
-       struct sigrok_device_instance *sdi;
+       struct sr_device_instance *sdi;
 
-       if(!(sdi = get_sigrok_device_instance(device_instances, device_index)))
+       if (!(sdi = sr_get_device_instance(device_instances, device_index)))
                return;
 
-       if(sdi->serial->fd != -1) {
-               close(sdi->serial->fd);
+       if (sdi->serial->fd != -1) {
+               serial_close(sdi->serial->fd);
                sdi->serial->fd = -1;
-               sdi->status = ST_INACTIVE;
+               sdi->status = SR_ST_INACTIVE;
        }
-
 }
 
-
 static void hw_cleanup(void)
 {
        GSList *l;
-       struct sigrok_device_instance *sdi;
+       struct sr_device_instance *sdi;
 
-       /* properly close all devices */
-       for(l = device_instances; l; l = l->next) {
+       /* Properly close all devices. */
+       for (l = device_instances; l; l = l->next) {
                sdi = l->data;
-               if(sdi->serial->fd != -1)
-                       close(sdi->serial->fd);
-               sigrok_device_instance_free(sdi);
+               if (sdi->serial->fd != -1)
+                       serial_close(sdi->serial->fd);
+               sr_device_instance_free(sdi);
        }
        g_slist_free(device_instances);
        device_instances = NULL;
-
 }
 
-
 static void *hw_get_device_info(int device_index, int device_info_id)
 {
-       struct sigrok_device_instance *sdi;
+       struct sr_device_instance *sdi;
        void *info;
 
-       if( !(sdi = get_sigrok_device_instance(device_instances, device_index)) )
+       if (!(sdi = sr_get_device_instance(device_instances, device_index)))
                return NULL;
 
        info = NULL;
-       switch(device_info_id)
-       {
-       case DI_INSTANCE:
+       switch (device_info_id) {
+       case SR_DI_INSTANCE:
                info = sdi;
                break;
-       case DI_NUM_PROBES:
+       case SR_DI_NUM_PROBES:
                info = GINT_TO_POINTER(NUM_PROBES);
                break;
-       case DI_SAMPLERATES:
+       case SR_DI_SAMPLERATES:
                info = &samplerates;
                break;
-       case DI_TRIGGER_TYPES:
-               info = (char *) TRIGGER_TYPES;
+       case SR_DI_TRIGGER_TYPES:
+               info = (char *)TRIGGER_TYPES;
                break;
-       case DI_CUR_SAMPLERATE:
+       case SR_DI_CUR_SAMPLERATE:
                info = &cur_samplerate;
                break;
        }
@@ -352,279 +402,401 @@ static void *hw_get_device_info(int device_index, int device_info_id)
        return info;
 }
 
-
 static int hw_get_status(int device_index)
 {
-       struct sigrok_device_instance *sdi;
+       struct sr_device_instance *sdi;
 
-       if(!(sdi = get_sigrok_device_instance(device_instances, device_index)))
-               return ST_NOT_FOUND;
+       if (!(sdi = sr_get_device_instance(device_instances, device_index)))
+               return SR_ST_NOT_FOUND;
 
        return sdi->status;
 }
 
-
 static int *hw_get_capabilities(void)
 {
-
        return capabilities;
 }
 
-
-static int set_configuration_samplerate(struct sigrok_device_instance *sdi, uint64_t samplerate)
+static int set_configuration_samplerate(struct sr_device_instance *sdi,
+                                       uint64_t samplerate)
 {
        uint32_t divider;
 
-       if(samplerate < samplerates.low || samplerate > samplerates.high)
-               return SIGROK_ERR_SAMPLERATE;
+       if (samplerate < samplerates.low || samplerate > samplerates.high)
+               return SR_ERR_SAMPLERATE;
 
-       if(samplerate  > CLOCK_RATE) {
+       if (samplerate > CLOCK_RATE) {
                flag_reg |= FLAG_DEMUX;
-               divider = (uint8_t) (CLOCK_RATE * 2 / samplerate) - 1;
+               divider = (CLOCK_RATE * 2 / samplerate) - 1;
+       } else {
+               flag_reg &= ~FLAG_DEMUX;
+               divider = (CLOCK_RATE / samplerate) - 1;
        }
-       else {
-               flag_reg &= FLAG_DEMUX;
-               divider = (uint8_t) (CLOCK_RATE / samplerate) - 1;
-       }
-       divider <<= 8;
 
-       g_message("setting samplerate to %"PRIu64" Hz (divider %u, demux %s)", samplerate, divider,
-                       flag_reg & FLAG_DEMUX ? "on" : "off");
-       if(send_longcommand(sdi->serial->fd, CMD_SET_DIVIDER, divider) != SIGROK_OK)
-               return SIGROK_ERR;
+       g_message("ols: setting samplerate to %" PRIu64 " Hz (divider %u, demux %s)",
+                       samplerate, divider, flag_reg & FLAG_DEMUX ? "on" : "off");
+
+       if (send_longcommand(sdi->serial->fd, CMD_SET_DIVIDER, reverse32(divider)) != SR_OK)
+               return SR_ERR;
        cur_samplerate = samplerate;
 
-       return SIGROK_OK;
+       return SR_OK;
 }
 
-
 static int hw_set_configuration(int device_index, int capability, void *value)
 {
-       struct sigrok_device_instance *sdi;
+       struct sr_device_instance *sdi;
        int ret;
        uint64_t *tmp_u64;
 
-       if(!(sdi = get_sigrok_device_instance(device_instances, device_index)))
-               return SIGROK_ERR;
+       if (!(sdi = sr_get_device_instance(device_instances, device_index)))
+               return SR_ERR;
 
-       if(sdi->status != ST_ACTIVE)
-               return SIGROK_ERR;
+       if (sdi->status != SR_ST_ACTIVE)
+               return SR_ERR;
 
-       if(capability == HWCAP_SAMPLERATE) {
+       switch (capability) {
+       case SR_HWCAP_SAMPLERATE:
                tmp_u64 = value;
                ret = set_configuration_samplerate(sdi, *tmp_u64);
-       }
-       else if(capability == HWCAP_PROBECONFIG)
-               ret = configure_probes( (GSList *) value);
-       else if(capability == HWCAP_LIMIT_SAMPLES) {
-               limit_samples = strtoull(value, NULL, 10);
-               ret = SIGROK_OK;
-       }
-       else if(capability == HWCAP_CAPTURE_RATIO) {
-               capture_ratio = strtol(value, NULL, 10);
-               if(capture_ratio < 0 || capture_ratio > 100) {
+               break;
+       case SR_HWCAP_PROBECONFIG:
+               ret = configure_probes((GSList *) value);
+               break;
+       case SR_HWCAP_LIMIT_SAMPLES:
+               tmp_u64 = value;
+               if (*tmp_u64 < MIN_NUM_SAMPLES)
+                       return SR_ERR;
+               limit_samples = *tmp_u64;
+               g_message("ols: sample limit %" PRIu64, limit_samples);
+               ret = SR_OK;
+               break;
+       case SR_HWCAP_CAPTURE_RATIO:
+               tmp_u64 = value;
+               capture_ratio = *tmp_u64;
+               if (capture_ratio < 0 || capture_ratio > 100) {
                        capture_ratio = 0;
-                       ret = SIGROK_ERR;
-               }
-               else
-                       ret = SIGROK_OK;
+                       ret = SR_ERR;
+               } else
+                       ret = SR_OK;
+               break;
+       default:
+               ret = SR_ERR;
        }
-       else
-               ret = SIGROK_ERR;
 
        return ret;
 }
 
-
 static int receive_data(int fd, int revents, void *user_data)
 {
-       static int num_transfers = 0;
+       static unsigned int num_transfers = 0;
        static int num_bytes = 0;
-       static char last_sample[4] = {0xff};
-       static unsigned char sample[4];
-       int count, buflen, i;
-       struct datafeed_packet packet;
+       static char last_sample[4] = { 0xff, 0xff, 0xff, 0xff };
+       static unsigned char sample[4] = { 0, 0, 0, 0 };
+       static unsigned char tmp_sample[4];
+       static unsigned char *raw_sample_buf = NULL;
+       int count, buflen, num_channels, offset, i, j;
+       struct sr_datafeed_packet packet;
        unsigned char byte, *buffer;
 
-       if(num_transfers++ == 0) {
-               /* first time round, means the device started sending data, and will not
-                * stop until done. if it stops sending for longer than it takes to send
-                * a byte, that means it's finished. we'll double that to 30ms to be sure...
+       if (num_transfers++ == 0) {
+               /*
+                * First time round, means the device started sending data,
+                * and will not stop until done. If it stops sending for
+                * longer than it takes to send a byte, that means it's
+                * finished. We'll double that to 30ms to be sure...
                 */
-               source_remove(fd);
-               source_add(fd, G_IO_IN, 30, receive_data, user_data);
+               sr_source_remove(fd);
+               sr_source_add(fd, G_IO_IN, 30, receive_data, user_data);
+               raw_sample_buf = malloc(limit_samples * 4);
+               /* fill with 1010... for debugging */
+               memset(raw_sample_buf, 0x82, limit_samples * 4);
+       }
+
+       num_channels = 0;
+       for (i = 0x20; i > 0x02; i /= 2) {
+               if ((flag_reg & i) == 0)
+                       num_channels++;
        }
 
-       /* TODO: / 4 depends on # of channels used */
-       if(revents == G_IO_IN && num_transfers / 4 <= limit_samples){
-               if(read(fd, &byte, 1) != 1)
+       if (revents == G_IO_IN
+           && num_transfers / num_channels <= limit_samples) {
+               if (serial_read(fd, &byte, 1) != 1)
                        return FALSE;
 
                sample[num_bytes++] = byte;
-               if(num_bytes == 4) {
-                       g_message("got sample 0x%.2x%.2x%.2x%.2x", sample[3], sample[2], sample[1], sample[0]);
-                       /* got a full sample */
-                       if(flag_reg & FLAG_RLE) {
-                               /* in RLE mode -1 should never come in as a sample, because
-                                * bit 31 is the "count" flag */
-                               /* TODO: endianness may be wrong here, could be sample[3] */
-                               if(sample[0] & 0x80 && !(last_sample[0] & 0x80)) {
-                                       count = (int) (*sample) & 0x7fffffff;
+               g_debug("ols: received byte 0x%.2x", byte);
+               if (num_bytes == num_channels) {
+                       g_debug("ols: received sample 0x%.*x", num_bytes * 2, (int) *sample);
+                       /* Got a full sample. */
+                       if (flag_reg & FLAG_RLE) {
+                               /*
+                                * In RLE mode -1 should never come in as a
+                                * sample, because bit 31 is the "count" flag.
+                                * TODO: Endianness may be wrong here, could be
+                                * sample[3].
+                                */
+                               if (sample[0] & 0x80
+                                   && !(last_sample[0] & 0x80)) {
+                                       count = (int)(*sample) & 0x7fffffff;
                                        buffer = g_malloc(count);
                                        buflen = 0;
-                                       for(i = 0; i < count; i++)
-                                       {
-                                               memcpy(buffer + buflen , last_sample, 4);
+                                       for (i = 0; i < count; i++) {
+                                               memcpy(buffer + buflen, last_sample, 4);
                                                buflen += 4;
                                        }
-                               }
-                               else {
-                                       /* just a single sample, next sample will probably be a count
-                                        * referring to this -- but this one is still a part of the stream
+                               } else {
+                                       /*
+                                        * Just a single sample, next sample
+                                        * will probably be a count referring
+                                        * to this -- but this one is still a
+                                        * part of the stream.
                                         */
                                        buffer = sample;
                                        buflen = 4;
                                }
-                       }
-                       else {
-                               /* no compression */
+                       } else {
+                               /* No compression. */
                                buffer = sample;
                                buflen = 4;
                        }
 
-                       /* send it all to the session bus */
-                       packet.type = DF_LOGIC32;
-                       packet.length = buflen;
-                       packet.payload = buffer;
-                       session_bus(user_data, &packet);
-                       if(buffer == sample)
-                               memcpy(last_sample, buffer, 4);
+                       if (num_channels < 4) {
+                               /*
+                                * Some channel groups may have been turned
+                                * off, to speed up transfer between the
+                                * hardware and the PC. Expand that here before
+                                * submitting it over the session bus --
+                                * whatever is listening on the bus will be
+                                * expecting a full 32-bit sample, based on
+                                * the number of probes.
+                                */
+                               j = 0;
+                               memset(tmp_sample, 0, 4);
+                               for (i = 0; i < 4; i++) {
+                                       if (((flag_reg >> 2) & (1 << i)) == 0) {
+                                               /*
+                                                * This channel group was
+                                                * enabled, copy from received
+                                                * sample.
+                                                */
+                                               tmp_sample[i] = sample[j++];
+                                       }
+                               }
+                               memcpy(sample, tmp_sample, 4);
+                               g_debug("ols: full sample 0x%.8x", (int) *sample);
+                       }
+
+                       /* the OLS sends its sample buffer backwards.
+                        * store it in reverse order here, so we can dump
+                        * this on the session bus later.
+                        */
+                       offset = (limit_samples - num_transfers / num_channels) * 4;
+                       memcpy(raw_sample_buf + offset, sample, 4);
+
+                       if (buffer == sample)
+                               memcpy(last_sample, buffer, num_channels);
                        else
                                g_free(buffer);
 
+                       memset(sample, 0, 4);
                        num_bytes = 0;
                }
-       }
-       else {
-               /* this is the main loop telling us a timeout was reached -- we're done */
-               tcflush(fd, TCIOFLUSH);
-               close(fd);
-               packet.type = DF_END;
+       } else {
+               /*
+                * This is the main loop telling us a timeout was reached, or
+                * we've acquired all the samples we asked for -- we're done.
+                * Send the (properly-ordered) buffer to the frontend.
+                */
+               if (trigger_at != -1) {
+                       /* a trigger was set up, so we need to tell the frontend
+                        * about it.
+                        */
+                       if (trigger_at > 0) {
+                               /* there are pre-trigger samples, send those first */
+                               packet.type = SR_DF_LOGIC;
+                               packet.length = trigger_at * 4;
+                               packet.unitsize = 4;
+                               packet.payload = raw_sample_buf;
+                               sr_session_bus(user_data, &packet);
+                       }
+
+                       packet.type = SR_DF_TRIGGER;
+                       packet.length = 0;
+                       sr_session_bus(user_data, &packet);
+
+                       packet.type = SR_DF_LOGIC;
+                       packet.length = (limit_samples * 4) - (trigger_at * 4);
+                       packet.unitsize = 4;
+                       packet.payload = raw_sample_buf + trigger_at * 4;
+                       sr_session_bus(user_data, &packet);
+               } else {
+                       packet.type = SR_DF_LOGIC;
+                       packet.length = limit_samples * 4;
+                       packet.unitsize = 4;
+                       packet.payload = raw_sample_buf;
+                       sr_session_bus(user_data, &packet);
+               }
+               free(raw_sample_buf);
+
+               serial_flush(fd);
+               serial_close(fd);
+               packet.type = SR_DF_END;
                packet.length = 0;
-               session_bus(user_data, &packet);
+               sr_session_bus(user_data, &packet);
        }
 
        return TRUE;
 }
 
-
 static int hw_start_acquisition(int device_index, gpointer session_device_id)
 {
-       struct datafeed_packet *packet;
-       struct datafeed_header *header;
-       struct sigrok_device_instance *sdi;
+       int i;
+       struct sr_datafeed_packet *packet;
+       struct sr_datafeed_header *header;
+       struct sr_device_instance *sdi;
+       uint32_t trigger_config[4];
        uint32_t data;
-
-       if(!(sdi = get_sigrok_device_instance(device_instances, device_index)))
-               return SIGROK_ERR;
-
-       if(sdi->status != ST_ACTIVE)
-               return SIGROK_ERR;
-
-       /* reset again */
-       if(send_longcommand(sdi->serial->fd, CMD_RESET, 0) != SIGROK_OK)
-               return SIGROK_ERR;
-
-       /* send flag register */
-       data = flag_reg << 24;
-       if(send_longcommand(sdi->serial->fd, CMD_SET_FLAGS, data) != SIGROK_OK)
-               return SIGROK_ERR;
-
-       /* send sample limit and pre/post-trigger capture ratio */
-       data = limit_samples / 4 << 16;
-       if(capture_ratio)
-               data |= (limit_samples - (limit_samples / 100 * capture_ratio)) / 4;
-       data = 0x00190019;
-       if(send_longcommand(sdi->serial->fd, CMD_CAPTURE_SIZE, data) != SIGROK_OK)
-               return SIGROK_ERR;
-
-       /* trigger masks */
-       if(send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_MASK_0, trigger_mask[0]) != SIGROK_OK)
-               return SIGROK_ERR;
-//     if(send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_MASK_1, trigger_mask[1]) != SIGROK_OK)
-//             return SIGROK_ERR;
-//     if(send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_MASK_2, trigger_mask[2]) != SIGROK_OK)
-//             return SIGROK_ERR;
-//     if(send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_MASK_3, trigger_mask[3]) != SIGROK_OK)
-//             return SIGROK_ERR;
-
-       if(send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_VALUE_0, trigger_value[0]) != SIGROK_OK)
-               return SIGROK_ERR;
-//     if(send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_VALUE_1, trigger_value[1]) != SIGROK_OK)
-//             return SIGROK_ERR;
-//     if(send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_VALUE_2, trigger_value[2]) != SIGROK_OK)
-//             return SIGROK_ERR;
-//     if(send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_VALUE_3, trigger_value[3]) != SIGROK_OK)
-//             return SIGROK_ERR;
-
-       /* trigger configuration */
-       /* TODO: the start flag should only be on the last used stage I think... */
-       if(send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_CONFIG_0, 0x00000008) != SIGROK_OK)
-               return SIGROK_ERR;
-//     if(send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_CONFIG_1, 0x00000000) != SIGROK_OK)
-//             return SIGROK_ERR;
-//     if(send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_CONFIG_2, 0x00000000) != SIGROK_OK)
-//             return SIGROK_ERR;
-//     if(send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_CONFIG_3, 0x00000000) != SIGROK_OK)
-//             return SIGROK_ERR;
+       uint16_t readcount, delaycount;
+       uint8_t changrp_mask;
+
+       if (!(sdi = sr_get_device_instance(device_instances, device_index)))
+               return SR_ERR;
+
+       if (sdi->status != SR_ST_ACTIVE)
+               return SR_ERR;
+
+       readcount = limit_samples / 4;
+
+       memset(trigger_config, 0, 16);
+       trigger_config[num_stages-1] |= 0x08;
+       if (trigger_mask[0]) {
+               delaycount = readcount * (1 - capture_ratio / 100.0);
+               trigger_at = (readcount - delaycount) * 4 - num_stages;
+
+               if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_MASK_0,
+                       reverse32(trigger_mask[0])) != SR_OK)
+                       return SR_ERR;
+               if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_VALUE_0,
+                       reverse32(trigger_value[0])) != SR_OK)
+                       return SR_ERR;
+               if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_CONFIG_0,
+                       trigger_config[0]) != SR_OK)
+                       return SR_ERR;
+
+               if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_MASK_1,
+                       reverse32(trigger_mask[1])) != SR_OK)
+                       return SR_ERR;
+               if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_VALUE_1,
+                       reverse32(trigger_value[1])) != SR_OK)
+                       return SR_ERR;
+               if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_CONFIG_1,
+                       trigger_config[1]) != SR_OK)
+                       return SR_ERR;
+
+               if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_MASK_2,
+                       reverse32(trigger_mask[2])) != SR_OK)
+                       return SR_ERR;
+               if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_VALUE_2,
+                       reverse32(trigger_value[2])) != SR_OK)
+                       return SR_ERR;
+               if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_CONFIG_2,
+                       trigger_config[2]) != SR_OK)
+                       return SR_ERR;
+
+               if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_MASK_3,
+                       reverse32(trigger_mask[3])) != SR_OK)
+                       return SR_ERR;
+               if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_VALUE_3,
+                       reverse32(trigger_value[3])) != SR_OK)
+                       return SR_ERR;
+               if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_CONFIG_3,
+                       trigger_config[3]) != SR_OK)
+                       return SR_ERR;
+       } else {
+               if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_MASK_0,
+                    trigger_mask[0]) != SR_OK)
+                       return SR_ERR;
+               if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_VALUE_0,
+                    trigger_value[0]) != SR_OK)
+                       return SR_ERR;
+               if (send_longcommand(sdi->serial->fd, CMD_SET_TRIGGER_CONFIG_0,
+                    0x00000008) != SR_OK)
+                       return SR_ERR;
+               delaycount = readcount;
+       }
 
        set_configuration_samplerate(sdi, cur_samplerate);
 
-       /* start acquisition on the device */
-       if(send_shortcommand(sdi->serial->fd, CMD_RUN) != SIGROK_OK)
-               return SIGROK_ERR;
+       /* Send sample limit and pre/post-trigger capture ratio. */
+       data = ((readcount - 1) & 0xffff) << 16;
+       data |= (delaycount - 1) & 0xffff;
+       if (send_longcommand(sdi->serial->fd, CMD_CAPTURE_SIZE, reverse16(data)) != SR_OK)
+               return SR_ERR;
 
-       source_add(sdi->serial->fd, G_IO_IN, -1, receive_data, session_device_id);
+       /*
+        * Enable/disable channel groups in the flag register according to the
+        * probe mask.
+        */
+       changrp_mask = 0;
+       for (i = 0; i < 4; i++) {
+               if (probe_mask & (0xff << (i * 8)))
+                       changrp_mask |= (1 << i);
+       }
 
-       /* send header packet to the session bus */
-       packet = g_malloc(sizeof(struct datafeed_packet));
-       header = g_malloc(sizeof(struct datafeed_header));
-       if(!packet || !header)
-               return SIGROK_ERR;
-       packet->type = DF_HEADER;
-       packet->length = sizeof(struct datafeed_header);
-       packet->payload = (unsigned char *) header;
+       /* The flag register wants them here, and 1 means "disable channel". */
+       flag_reg |= ~(changrp_mask << 2) & 0x3c;
+       flag_reg |= FLAG_FILTER;
+       data = flag_reg << 24;
+       if (send_longcommand(sdi->serial->fd, CMD_SET_FLAGS, data) != SR_OK)
+               return SR_ERR;
+
+       /* Start acquisition on the device. */
+       if (send_shortcommand(sdi->serial->fd, CMD_RUN) != SR_OK)
+               return SR_ERR;
+
+       sr_source_add(sdi->serial->fd, G_IO_IN, -1, receive_data,
+                     session_device_id);
+
+       /* Send header packet to the session bus. */
+       packet = g_malloc(sizeof(struct sr_datafeed_packet));
+       header = g_malloc(sizeof(struct sr_datafeed_header));
+       if (!packet || !header)
+               return SR_ERR;
+       packet->type = SR_DF_HEADER;
+       packet->length = sizeof(struct sr_datafeed_header);
+       packet->payload = (unsigned char *)header;
        header->feed_version = 1;
        gettimeofday(&header->starttime, NULL);
        header->samplerate = cur_samplerate;
-       header->protocol_id = PROTO_RAW;
-       header->num_probes = NUM_PROBES;
-       session_bus(session_device_id, packet);
+       header->protocol_id = SR_PROTO_RAW;
+       header->num_logic_probes = NUM_PROBES;
+       header->num_analog_probes = 0;
+       sr_session_bus(session_device_id, packet);
        g_free(header);
        g_free(packet);
 
-       return SIGROK_OK;
+       return SR_OK;
 }
 
-
 static void hw_stop_acquisition(int device_index, gpointer session_device_id)
 {
-       struct datafeed_packet packet;
+       struct sr_datafeed_packet packet;
 
-       packet.type = DF_END;
-       packet.length = 0;
-       session_bus(session_device_id, &packet);
+       /* Avoid compiler warnings. */
+       device_index = device_index;
 
+       packet.type = SR_DF_END;
+       packet.length = 0;
+       sr_session_bus(session_device_id, &packet);
 }
 
-
-
-struct device_plugin ols_plugin_info = {
-       "sump",
+struct sr_device_plugin ols_plugin_info = {
+       "ols",
+       "Openbench Logic Sniffer",
        1,
        hw_init,
        hw_cleanup,
-
        hw_opendev,
        hw_closedev,
        hw_get_device_info,
@@ -632,6 +804,5 @@ struct device_plugin ols_plugin_info = {
        hw_get_capabilities,
        hw_set_configuration,
        hw_start_acquisition,
-       hw_stop_acquisition
+       hw_stop_acquisition,
 };
-