]> sigrok.org Git - libsigrok.git/blobdiff - src/hardware/ftdi-la/api.c
Put driver pointers into special section
[libsigrok.git] / src / hardware / ftdi-la / api.c
index a4309774c9e8c342a8c5ca83e71786b3b09a084e..34f79bf71b71d0066621a422f3cfa45b27a756b5 100644 (file)
  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
+#include <config.h>
 #include <ftdi.h>
 #include <libusb.h>
+#include <libsigrok/libsigrok.h>
+#include "libsigrok-internal.h"
 #include "protocol.h"
 
-SR_PRIV struct sr_dev_driver ftdi_la_driver_info;
-
 static const uint32_t scanopts[] = {
-       /* TODO: SR_CONF_CONN to be able to specify the USB address. */
+       SR_CONF_CONN,
 };
 
 static const uint32_t devopts[] = {
        SR_CONF_LOGIC_ANALYZER,
+       SR_CONF_CONTINUOUS,
        SR_CONF_LIMIT_SAMPLES | SR_CONF_SET,
        SR_CONF_SAMPLERATE | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
+       SR_CONF_CONN | SR_CONF_GET,
 };
 
 static const uint64_t samplerates[] = {
@@ -63,13 +66,13 @@ static const struct ftdi_chip_desc ft232r_desc = {
        .samplerate_div = 30,
        .channel_names = {
                "TXD",
-               "RI#",
-               "DCD#",
-               "DSR#",
-               "DTR#",
-               "CTS#",
-               "RTS#",
                "RXD",
+               "RTS#",
+               "CTS#",
+               "DTR#",
+               "DSR#",
+               "DCD#",
+               "RI#",
                NULL
        }
 };
@@ -79,29 +82,32 @@ static const struct ftdi_chip_desc *chip_descs[] = {
        &ft232r_desc,
 };
 
-static int init(struct sr_dev_driver *di, struct sr_context *sr_ctx)
+static void scan_device(struct sr_dev_driver *di, struct libusb_device *dev, GSList **devices)
 {
-       return std_init(sr_ctx, di, LOG_PREFIX);
-}
-
-static GSList *scan(struct sr_dev_driver *di, GSList *options)
-{
-       struct drv_context *drvc;
-       struct dev_context *devc;
-       struct sr_dev_inst *sdi;
-       GSList *devices;
-       struct ftdi_device_list *devlist = 0;
-       struct ftdi_device_list *curdev;
        struct libusb_device_descriptor usb_desc;
        const struct ftdi_chip_desc *desc;
+       struct dev_context *devc;
        char *vendor, *model, *serial_num;
-       int ret;
-
-       (void)options;
+       struct sr_dev_inst *sdi;
+       struct drv_context *drvc;
+       int rv;
 
-       devices = NULL;
        drvc = di->context;
-       drvc->instances = NULL;
+       libusb_get_device_descriptor(dev, &usb_desc);
+
+       desc = NULL;
+       for (unsigned long i = 0; i < ARRAY_SIZE(chip_descs); i++) {
+               desc = chip_descs[i];
+               if (desc->vendor == usb_desc.idVendor &&
+                       desc->product == usb_desc.idProduct)
+                       break;
+       }
+
+       if (!desc) {
+               sr_spew("Unsupported FTDI device 0x%4x:0x%4x.",
+                       usb_desc.idVendor, usb_desc.idProduct);
+               return;
+       }
 
        /* Allocate memory for our private device context. */
        devc = g_malloc0(sizeof(struct dev_context));
@@ -116,10 +122,78 @@ static GSList *scan(struct sr_dev_driver *di, GSList *options)
                goto err_free_data_buf;
        }
 
-       ret = ftdi_usb_find_all(devc->ftdic, &devlist, 0, 0);
+       devc->usbdev = dev;
+       devc->desc = desc;
+
+       vendor = g_malloc(32);
+       model = g_malloc(32);
+       serial_num = g_malloc(32);
+       rv = ftdi_usb_get_strings(devc->ftdic, dev, vendor, 32,
+                            model, 32, serial_num, 32);
+       switch (rv) {
+       case 0:
+               break;
+       case -9:
+               sr_dbg("The device lacks a serial number.");
+               g_free(serial_num);
+               serial_num = NULL;
+               break;
+       default:
+               sr_err("Failed to get the FTDI strings: %d", rv);
+               goto err_free_strings;
+       }
+       sr_dbg("Found an FTDI device: %s.", model);
+
+       /* Register the device with libsigrok. */
+       sdi = g_malloc0(sizeof(struct sr_dev_inst));
+       sdi->status = SR_ST_INACTIVE;
+       sdi->vendor = vendor;
+       sdi->model = model;
+       sdi->serial_num = serial_num;
+       sdi->driver = di;
+       sdi->priv = devc;
+
+       for (char *const *chan = &(desc->channel_names[0]); *chan; chan++)
+               sr_channel_new(sdi, chan - &(desc->channel_names[0]),
+                               SR_CHANNEL_LOGIC, TRUE, *chan);
+
+       *devices = g_slist_append(*devices, sdi);
+       drvc->instances = g_slist_append(drvc->instances, sdi);
+       return;
+
+err_free_strings:
+       g_free(vendor);
+       g_free(model);
+       g_free(serial_num);
+       ftdi_free(devc->ftdic);
+err_free_data_buf:
+       g_free(devc->data_buf);
+       g_free(devc);
+}
+
+static GSList *scan_all(struct sr_dev_driver *di, GSList *options)
+{
+       GSList *devices;
+       struct ftdi_device_list *devlist = 0;
+       struct ftdi_device_list *curdev;
+       struct ftdi_context *ftdic;
+       int ret;
+
+       (void)options;
+
+       devices = NULL;
+
+       /* Allocate memory for the FTDI context (ftdic) and initialize it. */
+       ftdic = ftdi_new();
+       if (!ftdic) {
+               sr_err("Failed to initialize libftdi.");
+               return NULL;
+       }
+
+       ret = ftdi_usb_find_all(ftdic, &devlist, 0, 0);
        if (ret < 0) {
                sr_err("Failed to list devices (%d): %s", ret,
-                      ftdi_get_error_string(devc->ftdic));
+                      ftdi_get_error_string(ftdic));
                goto err_free_ftdic;
        }
 
@@ -127,64 +201,59 @@ static GSList *scan(struct sr_dev_driver *di, GSList *options)
 
        curdev = devlist;
        while (curdev) {
-               libusb_get_device_descriptor(curdev->dev, &usb_desc);
-
-               desc = NULL;
-               for (unsigned long i = 0; i < ARRAY_SIZE(chip_descs); i++) {
-                       desc = chip_descs[i];
-                       if (desc->vendor == usb_desc.idVendor &&
-                               desc->product == usb_desc.idProduct)
-                               break;
-               }
-
-               if (!desc) {
-                       sr_spew("Unsupported FTDI device 0x%4x:0x%4x.",
-                               usb_desc.idVendor, usb_desc.idProduct);
-                       continue;
-               }
-               devc->usbdev = curdev->dev;
-               devc->desc = desc;
-
-               vendor = g_malloc(32);
-               model = g_malloc(32);
-               serial_num = g_malloc(32);
-               ftdi_usb_get_strings(devc->ftdic, curdev->dev, vendor, 32,
-                                    model, 32, serial_num, 32);
-               sr_dbg("Found an FTDI device: %s.", model);
-
-               /* Register the device with libsigrok. */
-               sdi = g_malloc0(sizeof(struct sr_dev_inst));
-               sdi->status = SR_ST_INITIALIZING;
-               sdi->vendor = vendor;
-               sdi->model = model;
-               sdi->serial_num = serial_num;
-               sdi->driver = di;
-               sdi->priv = devc;
-
-               for (char *const *chan = &(desc->channel_names[0]); *chan; chan++)
-                       sr_channel_new(sdi, &(desc->channel_names[0]) - chan,
-                                       SR_CHANNEL_LOGIC, TRUE, *chan);
-
-               devices = g_slist_append(devices, sdi);
-               drvc->instances = g_slist_append(drvc->instances, sdi);
-
+               scan_device(di, curdev->dev, &devices);
                curdev = curdev->next;
        }
 
+       ftdi_list_free(&devlist);
+       ftdi_free(ftdic);
+
        return devices;
 
 err_free_ftdic:
-       ftdi_free(devc->ftdic); /* NOT free() or g_free()! */
-err_free_data_buf:
-       g_free(devc->data_buf);
-       g_free(devc);
+       ftdi_free(ftdic); /* NOT free() or g_free()! */
 
        return NULL;
 }
 
-static GSList *dev_list(const struct sr_dev_driver *di)
+static GSList *scan(struct sr_dev_driver *di, GSList *options)
 {
-       return ((struct drv_context *)(di->context))->instances;
+       struct sr_config *src;
+       struct sr_usb_dev_inst *usb;
+       const char *conn;
+       GSList *l, *conn_devices;
+       GSList *devices;
+       struct drv_context *drvc;
+       libusb_device **devlist;
+       int i;
+
+       drvc = di->context;
+       drvc->instances = NULL;
+       conn = NULL;
+       for (l = options; l; l = l->next) {
+               src = l->data;
+               if (src->key == SR_CONF_CONN) {
+                       conn = g_variant_get_string(src->data, NULL);
+                       break;
+               }
+       }
+       if (conn) {
+               devices = NULL;
+               libusb_get_device_list(drvc->sr_ctx->libusb_ctx, &devlist);
+               for (i = 0; devlist[i]; i++) {
+                       conn_devices = sr_usb_find(drvc->sr_ctx->libusb_ctx, conn);
+                       for (l = conn_devices; l; l = l->next) {
+                               usb = l->data;
+                               if (usb->bus == libusb_get_bus_number(devlist[i])
+                                       && usb->address == libusb_get_device_address(devlist[i])) {
+                                       scan_device(di, devlist[i], &devices);
+                               }
+                       }
+               }
+               libusb_free_device_list(devlist, 1);
+               return devices;
+       } else
+               return scan_all(di, options);
 }
 
 static void clear_helper(void *priv)
@@ -267,20 +336,13 @@ static int dev_close(struct sr_dev_inst *sdi)
        return SR_OK;
 }
 
-static int cleanup(const struct sr_dev_driver *di)
-{
-       dev_clear(di);
-
-       /* TODO: Free other driver resources, if any. */
-
-       return SR_OK;
-}
-
 static int config_get(uint32_t key, GVariant **data,
        const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
 {
        int ret;
        struct dev_context *devc;
+       struct sr_usb_dev_inst *usb;
+       char str[128];
 
        (void)cg;
 
@@ -291,6 +353,13 @@ static int config_get(uint32_t key, GVariant **data,
        case SR_CONF_SAMPLERATE:
                *data = g_variant_new_uint64(devc->cur_samplerate);
                break;
+       case SR_CONF_CONN:
+               if (!sdi || !sdi->conn)
+                       return SR_ERR_ARG;
+               usb = sdi->conn;
+               snprintf(str, 128, "%d.%d", usb->bus, usb->address);
+               *data = g_variant_new_string(str);
+               break;
        default:
                return SR_ERR_NA;
        }
@@ -298,20 +367,6 @@ static int config_get(uint32_t key, GVariant **data,
        return ret;
 }
 
-static int ftdi_la_set_samplerate(struct dev_context *devc)
-{
-       int ret;
-
-       ret = ftdi_set_baudrate(devc->ftdic,
-                       devc->cur_samplerate / devc->desc->samplerate_div);
-       if (ret < 0) {
-               sr_err("Failed to set baudrate (%d): %s.", devc->cur_samplerate,
-                      ftdi_get_error_string(devc->ftdic));
-               return SR_ERR;
-       }
-       return SR_OK;
-}
-
 static int config_set(uint32_t key, GVariant *data,
        const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
 {
@@ -385,12 +440,10 @@ static int config_list(uint32_t key, GVariant **data,
        return ret;
 }
 
-static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data)
+static int dev_acquisition_start(const struct sr_dev_inst *sdi)
 {
        struct dev_context *devc;
 
-       (void)cb_data;
-
        devc = sdi->priv;
 
        if (sdi->status != SR_ST_ACTIVE)
@@ -401,13 +454,10 @@ static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data)
 
        ftdi_set_bitmode(devc->ftdic, 0, BITMODE_BITBANG);
 
-       devc->cb_data = cb_data;
-
        /* Properly reset internal variables before every new acquisition. */
        devc->samples_sent = 0;
        devc->bytes_received = 0;
 
-       /* Send header packet to the session bus. */
        std_session_send_df_header(sdi, LOG_PREFIX);
 
        /* Hook up a dummy handler to receive data from the device. */
@@ -417,34 +467,27 @@ static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data)
        return SR_OK;
 }
 
-static int dev_acquisition_stop(struct sr_dev_inst *sdi, void *cb_data)
+static int dev_acquisition_stop(struct sr_dev_inst *sdi)
 {
-       struct sr_datafeed_packet packet;
-
-       (void)cb_data;
-
        if (sdi->status != SR_ST_ACTIVE)
                return SR_ERR_DEV_CLOSED;
 
        sr_dbg("Stopping acquisition.");
        sr_session_source_remove(sdi->session, -1);
 
-       /* Send end packet to the session bus. */
-       sr_dbg("Sending SR_DF_END.");
-       packet.type = SR_DF_END;
-       sr_session_send(sdi, &packet);
+       std_session_send_df_end(sdi, LOG_PREFIX);
 
        return SR_OK;
 }
 
-SR_PRIV struct sr_dev_driver ftdi_la_driver_info = {
+static struct sr_dev_driver ftdi_la_driver_info = {
        .name = "ftdi-la",
        .longname = "FTDI LA",
        .api_version = 1,
-       .init = init,
-       .cleanup = cleanup,
+       .init = std_init,
+       .cleanup = std_cleanup,
        .scan = scan,
-       .dev_list = dev_list,
+       .dev_list = std_dev_list,
        .dev_clear = dev_clear,
        .config_get = config_get,
        .config_set = config_set,
@@ -455,3 +498,4 @@ SR_PRIV struct sr_dev_driver ftdi_la_driver_info = {
        .dev_acquisition_stop = dev_acquisition_stop,
        .context = NULL,
 };
+SR_REGISTER_DEV_DRIVER(ftdi_la_driver_info);