]> sigrok.org Git - libsigrok.git/blobdiff - hardware/genericdmm/api.c
Return SR_ERR_MALLOC upon allocation errors.
[libsigrok.git] / hardware / genericdmm / api.c
index 035ffedfd55eafd998cc585ee2fb55df3007d236..5b12b2e42b48ffeaef132f6766effcfb5093b886 100644 (file)
 
 
 extern SR_PRIV struct dmmchip dmmchip_fs9922;
+extern SR_PRIV struct dmmchip dmmchip_victor70c;
 
 static struct sr_hwopt victor_70c_vidpid[] = {
        { SR_HWOPT_CONN, "1244.d237" },
        { 0, NULL }
 };
 static struct dev_profile dev_profiles[] = {
-       { "victor-70c", "Victor", "70C", &dmmchip_fs9922,
-               DMM_TRANSPORT_USBHID, victor_70c_vidpid
+       { "victor-70c", "Victor", "70C", &dmmchip_victor70c,
+               DMM_TRANSPORT_USBHID, 1000, victor_70c_vidpid
        },
-       { "mastech-va18b", "Mastech", "VA18B", NULL, DMM_TRANSPORT_SERIAL, NULL},
-       { NULL, NULL, NULL, NULL, 0, NULL }
+       { "mastech-va18b", "Mastech", "VA18B", NULL, DMM_TRANSPORT_SERIAL, 0, NULL},
+       { NULL, NULL, NULL, NULL, 0, 0, NULL }
 };
 
 static const int hwopts[] = {
@@ -64,7 +65,9 @@ static const char *probe_names[] = {
 SR_PRIV struct sr_dev_driver genericdmm_driver_info;
 static struct sr_dev_driver *gdi = &genericdmm_driver_info;
 /* TODO need a way to keep this local to the static library */
-SR_PRIV libusb_context *genericdmm_usb_context = NULL;
+static libusb_context *genericdmm_usb_context = NULL;
+static int hw_dev_acquisition_stop(const struct sr_dev_inst *sdi,
+               void *cb_data);
 
 
 static GSList *connect_usb(const char *conn)
@@ -72,6 +75,7 @@ static GSList *connect_usb(const char *conn)
        struct sr_dev_inst *sdi;
        struct drv_context *drvc;
        struct dev_context *devc;
+       struct sr_probe *probe;
        libusb_device **devlist;
        struct libusb_device_descriptor des;
        GSList *devices;
@@ -117,12 +121,12 @@ static GSList *connect_usb(const char *conn)
                return NULL;
 
        if (bus > 64) {
-               sr_err("invalid bus");
+               sr_err("Invalid bus.");
                return NULL;
        }
 
        if (addr > 127) {
-               sr_err("invalid address");
+               sr_err("Invalid address.");
                return NULL;
        }
 
@@ -131,7 +135,7 @@ static GSList *connect_usb(const char *conn)
        libusb_get_device_list(genericdmm_usb_context, &devlist);
        for (i = 0; devlist[i]; i++) {
                if ((err = libusb_get_device_descriptor(devlist[i], &des))) {
-                       sr_err("genericdmm: failed to get device descriptor: %d", err);
+                       sr_err("Failed to get device descriptor: %d.", err);
                        continue;
                }
 
@@ -147,17 +151,20 @@ static GSList *connect_usb(const char *conn)
 
                /* Found one. */
                if (!(devc = g_try_malloc0(sizeof(struct dev_context)))) {
-                       sr_err("genericdmm: devc malloc failed.");
-                       return 0;
+                       sr_err("Device context malloc failed.");
+                       return NULL;
                }
 
                devcnt = g_slist_length(drvc->instances);
-               if (!(sdi = sr_dev_inst_new(devcnt, SR_ST_ACTIVE,
+               if (!(sdi = sr_dev_inst_new(devcnt, SR_ST_INACTIVE,
                                NULL, NULL, NULL))) {
-                       sr_err("genericdmm: sr_dev_inst_new returned NULL.");
+                       sr_err("sr_dev_inst_new returned NULL.");
                        return NULL;
                }
                sdi->priv = devc;
+               if (!(probe = sr_probe_new(0, SR_PROBE_ANALOG, TRUE, "P1")))
+                       return NULL;
+               sdi->probes = g_slist_append(sdi->probes, probe);
                devc->usb = sr_usb_dev_inst_new(
                                libusb_get_bus_number(devlist[i]),
                                libusb_get_device_address(devlist[i]), NULL);
@@ -175,7 +182,7 @@ static GSList *connect_serial(const char *conn, const char *serialcomm)
        devices = NULL;
 
        /* TODO */
-       sr_dbg("not yet implemented");
+       sr_dbg("Not yet implemented.");
 
        return devices;
 }
@@ -219,6 +226,52 @@ static GSList *default_scan(GSList *options)
        return devices;
 }
 
+static int open_usb(struct sr_dev_inst *sdi)
+{
+       libusb_device **devlist;
+       struct libusb_device_descriptor des;
+       struct dev_context *devc;
+       int ret, tmp, cnt, i;
+
+       devc = sdi->priv;
+
+       if (sdi->status == SR_ST_ACTIVE)
+               /* already in use */
+               return SR_ERR;
+
+       cnt = libusb_get_device_list(genericdmm_usb_context, &devlist);
+       if (cnt < 0) {
+               sr_err("Failed to retrieve device list (%d).", cnt);
+               return SR_ERR;
+       }
+
+       ret = SR_ERR;
+       for (i = 0; i < cnt; i++) {
+               if ((tmp = libusb_get_device_descriptor(devlist[i], &des))) {
+                       sr_err("Failed to get device descriptor: %d.", tmp);
+                       continue;
+               }
+
+               if (libusb_get_bus_number(devlist[i]) != devc->usb->bus
+                       || libusb_get_device_address(devlist[i]) != devc->usb->address)
+                       /* this is not the one */
+                       continue;
+
+               if ((tmp = libusb_open(devlist[i], &devc->usb->devhdl))) {
+                       sr_err("Failed to open device: %d.", tmp);
+                       break;
+               }
+
+               sr_info("Opened device %s on %d.%d.", devc->profile->modelid,
+                       devc->usb->bus, devc->usb->address);
+               ret = SR_OK;
+               break;
+       }
+       libusb_free_device_list(devlist, 1);
+
+       return ret;
+}
+
 static int clear_instances(void)
 {
        GSList *l;
@@ -233,19 +286,18 @@ static int clear_instances(void)
        for (l = drvc->instances; l; l = l->next) {
                if (!(sdi = l->data)) {
                        /* Log error, but continue cleaning up the rest. */
-                       sr_err("genericdmm: sdi was NULL, continuing.");
+                       sr_err("sdi was NULL, continuing.");
                        continue;
                }
                if (!(devc = sdi->priv)) {
                        /* Log error, but continue cleaning up the rest. */
-                       sr_err("genericdmm: sdi->priv was NULL, continuing.");
+                       sr_err("sdi->priv was NULL, continuing.");
                        continue;
                }
 
                if (devc->profile) {
                        switch (devc->profile->transport) {
                        case DMM_TRANSPORT_USBHID:
-                               /* TODO */
                                sr_usb_dev_inst_free(devc->usb);
                                break;
                        case DMM_TRANSPORT_SERIAL:
@@ -270,12 +322,12 @@ static int hw_init(void)
        struct drv_context *drvc;
 
        if (!(drvc = g_try_malloc0(sizeof(struct drv_context)))) {
-               sr_err("genericdmm: driver context malloc failed.");
-               return SR_ERR;
+               sr_err("Driver context malloc failed.");
+               return SR_ERR_MALLOC;
        }
 
        if (libusb_init(&genericdmm_usb_context) != 0) {
-               sr_err("genericdmm: Failed to initialize USB.");
+               sr_err("Failed to initialize USB.");
                return SR_ERR;
        }
 
@@ -291,6 +343,7 @@ static GSList *hw_scan(GSList *options)
        struct dev_profile *pr, *profile;
        struct sr_dev_inst *sdi;
        struct drv_context *drvc;
+       struct dev_context *devc;
        const char *model;
 
        drvc = gdi->priv;
@@ -371,6 +424,8 @@ static GSList *hw_scan(GSList *options)
                         * or model. Do that now.
                         */
                        sdi = l->data;
+                       devc = sdi->priv;
+                       devc->profile = profile;
                        sdi->driver = gdi;
                        if (!sdi->vendor)
                                sdi->vendor = g_strdup(profile->vendor);
@@ -396,32 +451,34 @@ static GSList *hw_dev_list(void)
 static int hw_dev_open(struct sr_dev_inst *sdi)
 {
        struct dev_context *devc;
+       int ret;
 
        if (!(devc = sdi->priv)) {
-               sr_err("genericdmm: sdi->priv was NULL.");
+               sr_err("sdi->priv was NULL.");
                return SR_ERR_BUG;
        }
 
+       ret = SR_OK;
        switch (devc->profile->transport) {
        case DMM_TRANSPORT_USBHID:
-               /* TODO */
+               ret = open_usb(sdi);
                break;
        case DMM_TRANSPORT_SERIAL:
-               /* TODO: O_NONBLOCK? */
-               sr_dbg("genericdmm: Opening serial port '%s'.", devc->serial->port);
+               sr_dbg("Opening serial port '%s'.", devc->serial->port);
                devc->serial->fd = serial_open(devc->serial->port, O_RDWR | O_NONBLOCK);
                if (devc->serial->fd == -1) {
-                       sr_err("genericdmm: Couldn't open serial port '%s'.",
+                       sr_err("Couldn't open serial port '%s'.",
                               devc->serial->port);
-                       return SR_ERR;
+                       ret = SR_ERR;
                }
                //      serial_set_params(devc->serial->fd, 2400, 8, 0, 1, 2);
                break;
        default:
                sr_err("No transport set.");
+               ret = SR_ERR;
        }
 
-       return SR_OK;
+       return ret;
 }
 
 static int hw_dev_close(struct sr_dev_inst *sdi)
@@ -429,7 +486,7 @@ static int hw_dev_close(struct sr_dev_inst *sdi)
        struct dev_context *devc;
 
        if (!(devc = sdi->priv)) {
-               sr_err("genericdmm: %s: sdi->priv was NULL.", __func__);
+               sr_err("%s: sdi->priv was NULL.", __func__);
                return SR_ERR_BUG;
        }
 
@@ -484,6 +541,7 @@ static int hw_info_get(int info_id, const void **data,
        case SR_DI_CUR_SAMPLERATE:
                /* TODO get rid of this */
                *data = NULL;
+               return SR_ERR_ARG;
                break;
        default:
                /* Unknown device info ID. */
@@ -499,27 +557,28 @@ static int hw_dev_config_set(const struct sr_dev_inst *sdi, int hwcap,
        struct dev_context *devc;
 
        if (!(devc = sdi->priv)) {
-               sr_err("genericdmm: sdi->priv was NULL.");
+               sr_err("sdi->priv was NULL.");
                return SR_ERR_BUG;
        }
 
        switch (hwcap) {
        case SR_HWCAP_LIMIT_MSEC:
+               /* TODO: not yet implemented */
                if (*(const uint64_t *)value == 0) {
-                       sr_err("genericdmm: LIMIT_MSEC can't be 0.");
+                       sr_err("LIMIT_MSEC can't be 0.");
                        return SR_ERR;
                }
                devc->limit_msec = *(const uint64_t *)value;
-               sr_dbg("genericdmm: Setting LIMIT_MSEC to %" PRIu64 ".",
+               sr_dbg("Setting time limit to %" PRIu64 "ms.",
                       devc->limit_msec);
                break;
        case SR_HWCAP_LIMIT_SAMPLES:
                devc->limit_samples = *(const uint64_t *)value;
-               sr_dbg("genericdmm: Setting LIMIT_SAMPLES to %" PRIu64 ".",
+               sr_dbg("Setting sample limit to %" PRIu64 ".",
                       devc->limit_samples);
                break;
        default:
-               sr_err("genericdmm: Unknown capability: %d.", hwcap);
+               sr_err("Unknown capability: %d.", hwcap);
                return SR_ERR;
                break;
        }
@@ -532,26 +591,28 @@ static int receive_data(int fd, int revents, void *cb_data)
        struct sr_dev_inst *sdi;
        struct dev_context *devc;
 
+       (void)revents;
+
        if (!(sdi = cb_data))
-               return FALSE;
+               return TRUE;
 
        if (!(devc = sdi->priv))
-               return FALSE;
-
-       if (revents != G_IO_IN) {
-               sr_err("genericdmm: No data?");
-               return FALSE;
-       }
+               return TRUE;
 
        switch (devc->profile->transport) {
        case DMM_TRANSPORT_USBHID:
-               /* TODO */
+               if (devc->profile->chip->data)
+                       devc->profile->chip->data(sdi);
                break;
        case DMM_TRANSPORT_SERIAL:
                /* TODO */
+               fd = fd;
                break;
        }
 
+       if (devc->num_samples >= devc->limit_samples)
+               hw_dev_acquisition_stop(sdi, cb_data);
+
        return TRUE;
 }
 
@@ -564,16 +625,16 @@ static int hw_dev_acquisition_start(const struct sr_dev_inst *sdi,
        struct dev_context *devc;
 
        if (!(devc = sdi->priv)) {
-               sr_err("genericdmm: sdi->priv was NULL.");
+               sr_err("sdi->priv was NULL.");
                return SR_ERR_BUG;
        }
 
-       sr_dbg("genericdmm: Starting acquisition.");
+       sr_dbg("Starting acquisition.");
 
        devc->cb_data = cb_data;
 
        /* Send header packet to the session bus. */
-       sr_dbg("genericdmm: Sending SR_DF_HEADER.");
+       sr_dbg("Sending SR_DF_HEADER.");
        packet.type = SR_DF_HEADER;
        packet.payload = (uint8_t *)&header;
        header.feed_version = 1;
@@ -581,7 +642,7 @@ static int hw_dev_acquisition_start(const struct sr_dev_inst *sdi,
        sr_session_send(devc->cb_data, &packet);
 
        /* Send metadata about the SR_DF_ANALOG packets to come. */
-       sr_dbg("genericdmm: Sending SR_DF_META_ANALOG.");
+       sr_dbg("Sending SR_DF_META_ANALOG.");
        packet.type = SR_DF_META_ANALOG;
        packet.payload = &meta;
        meta.num_probes = 1;
@@ -590,7 +651,14 @@ static int hw_dev_acquisition_start(const struct sr_dev_inst *sdi,
        /* Hook up a proxy handler to receive data from the device. */
        switch (devc->profile->transport) {
        case DMM_TRANSPORT_USBHID:
-               /* TODO libusb FD setup */
+               /* Callously using stdin here. This works because no G_IO_* flags
+                * are set, but will certainly break when any other driver does
+                * this, and runs at the same time as genericdmm.
+                * We'll need a timeout-only source when revamping the whole
+                * driver source system.
+                */
+               sr_source_add(0, 0, devc->profile->poll_timeout,
+                               receive_data, (void *)sdi);
                break;
        case DMM_TRANSPORT_SERIAL:
                /* TODO serial FD setup */
@@ -609,13 +677,15 @@ static int hw_dev_acquisition_stop(const struct sr_dev_inst *sdi,
        /* Avoid compiler warnings. */
        (void)sdi;
 
-       sr_dbg("genericdmm: Stopping acquisition.");
+       sr_dbg("Stopping acquisition.");
 
        /* Send end packet to the session bus. */
-       sr_dbg("genericdmm: Sending SR_DF_END.");
+       sr_dbg("Sending SR_DF_END.");
        packet.type = SR_DF_END;
        sr_session_send(cb_data, &packet);
 
+       sr_source_remove(0);
+
        return SR_OK;
 }