]> sigrok.org Git - libsigrok.git/blob - hardware/fx2lafw/protocol.c
fx2lafw: pass sdi to command functions.
[libsigrok.git] / hardware / fx2lafw / protocol.c
1 /*
2  * This file is part of the libsigrok project.
3  *
4  * Copyright (C) 2013 Bert Vermeulen <bert@biot.com>
5  * Copyright (C) 2012 Joel Holdsworth <joel@airwebreathe.org.uk>
6  *
7  * This program is free software: you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation, either version 3 of the License, or
10  * (at your option) any later version.
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
19  */
20
21 #include "protocol.h"
22
23 /* Protocol commands */
24 #define CMD_GET_FW_VERSION              0xb0
25 #define CMD_START                       0xb1
26 #define CMD_GET_REVID_VERSION           0xb2
27
28 #define CMD_START_FLAGS_WIDE_POS        5
29 #define CMD_START_FLAGS_CLK_SRC_POS     6
30
31 #define CMD_START_FLAGS_SAMPLE_8BIT     (0 << CMD_START_FLAGS_WIDE_POS)
32 #define CMD_START_FLAGS_SAMPLE_16BIT    (1 << CMD_START_FLAGS_WIDE_POS)
33
34 #define CMD_START_FLAGS_CLK_30MHZ       (0 << CMD_START_FLAGS_CLK_SRC_POS)
35 #define CMD_START_FLAGS_CLK_48MHZ       (1 << CMD_START_FLAGS_CLK_SRC_POS)
36
37 #pragma pack(push, 1)
38
39 struct version_info {
40         uint8_t major;
41         uint8_t minor;
42 };
43
44 struct cmd_start_acquisition {
45         uint8_t flags;
46         uint8_t sample_delay_h;
47         uint8_t sample_delay_l;
48 };
49
50 #pragma pack(pop)
51
52 static int command_get_fw_version(libusb_device_handle *devhdl,
53                                   struct version_info *vi)
54 {
55         int ret;
56
57         ret = libusb_control_transfer(devhdl, LIBUSB_REQUEST_TYPE_VENDOR |
58                 LIBUSB_ENDPOINT_IN, CMD_GET_FW_VERSION, 0x0000, 0x0000,
59                 (unsigned char *)vi, sizeof(struct version_info), 100);
60
61         if (ret < 0) {
62                 sr_err("Unable to get version info: %s.",
63                        libusb_error_name(ret));
64                 return SR_ERR;
65         }
66
67         return SR_OK;
68 }
69
70 static int command_get_revid_version(struct sr_dev_inst *sdi, uint8_t *revid)
71 {
72         struct sr_usb_dev_inst *usb = sdi->conn;
73         libusb_device_handle *devhdl = usb->devhdl;
74         int ret;
75
76         ret = libusb_control_transfer(devhdl, LIBUSB_REQUEST_TYPE_VENDOR |
77                 LIBUSB_ENDPOINT_IN, CMD_GET_REVID_VERSION, 0x0000, 0x0000,
78                 revid, 1, 100);
79
80         if (ret < 0) {
81                 sr_err("Unable to get REVID: %s.", libusb_error_name(ret));
82                 return SR_ERR;
83         }
84
85         return SR_OK;
86 }
87
88 SR_PRIV int fx2lafw_command_start_acquisition(const struct sr_dev_inst *sdi)
89 {
90         struct dev_context *devc = sdi->priv;
91         struct sr_usb_dev_inst *usb = sdi->conn;
92         libusb_device_handle *devhdl = usb->devhdl;
93         uint64_t samplerate = devc->cur_samplerate;
94         gboolean samplewide = devc->sample_wide;
95         struct cmd_start_acquisition cmd = { 0 };
96         int delay = 0, ret;
97
98         /* Compute the sample rate. */
99         if (samplewide && samplerate > MAX_16BIT_SAMPLE_RATE) {
100                 sr_err("Unable to sample at %" PRIu64 "Hz "
101                        "when collecting 16-bit samples.", samplerate);
102                 return SR_ERR;
103         }
104
105         if ((SR_MHZ(48) % samplerate) == 0) {
106                 cmd.flags = CMD_START_FLAGS_CLK_48MHZ;
107                 delay = SR_MHZ(48) / samplerate - 1;
108                 if (delay > MAX_SAMPLE_DELAY)
109                         delay = 0;
110         }
111
112         if (delay == 0 && (SR_MHZ(30) % samplerate) == 0) {
113                 cmd.flags = CMD_START_FLAGS_CLK_30MHZ;
114                 delay = SR_MHZ(30) / samplerate - 1;
115         }
116
117         sr_info("GPIF delay = %d, clocksource = %sMHz.", delay,
118                 (cmd.flags & CMD_START_FLAGS_CLK_48MHZ) ? "48" : "30");
119
120         if (delay <= 0 || delay > MAX_SAMPLE_DELAY) {
121                 sr_err("Unable to sample at %" PRIu64 "Hz.", samplerate);
122                 return SR_ERR;
123         }
124
125         cmd.sample_delay_h = (delay >> 8) & 0xff;
126         cmd.sample_delay_l = delay & 0xff;
127
128         /* Select the sampling width. */
129         cmd.flags |= samplewide ? CMD_START_FLAGS_SAMPLE_16BIT :
130                 CMD_START_FLAGS_SAMPLE_8BIT;
131
132         /* Send the control message. */
133         ret = libusb_control_transfer(devhdl, LIBUSB_REQUEST_TYPE_VENDOR |
134                         LIBUSB_ENDPOINT_OUT, CMD_START, 0x0000, 0x0000,
135                         (unsigned char *)&cmd, sizeof(cmd), 100);
136         if (ret < 0) {
137                 sr_err("Unable to send start command: %s.",
138                        libusb_error_name(ret));
139                 return SR_ERR;
140         }
141
142         return SR_OK;
143 }
144
145 /**
146  * Check the USB configuration to determine if this is an fx2lafw device.
147  *
148  * @return TRUE if the device's configuration profile match fx2lafw
149  *         configuration, FALSE otherwise.
150  */
151 SR_PRIV gboolean fx2lafw_check_conf_profile(libusb_device *dev)
152 {
153         struct libusb_device_descriptor des;
154         struct libusb_device_handle *hdl;
155         gboolean ret;
156         unsigned char strdesc[64];
157
158         hdl = NULL;
159         ret = FALSE;
160         while (!ret) {
161                 /* Assume the FW has not been loaded, unless proven wrong. */
162                 if (libusb_get_device_descriptor(dev, &des) != 0)
163                         break;
164
165                 if (libusb_open(dev, &hdl) != 0)
166                         break;
167
168                 if (libusb_get_string_descriptor_ascii(hdl,
169                     des.iManufacturer, strdesc, sizeof(strdesc)) < 0)
170                         break;
171                 if (strncmp((const char *)strdesc, "sigrok", 6))
172                         break;
173
174                 if (libusb_get_string_descriptor_ascii(hdl,
175                                 des.iProduct, strdesc, sizeof(strdesc)) < 0)
176                         break;
177                 if (strncmp((const char *)strdesc, "fx2lafw", 7))
178                         break;
179
180                 /* If we made it here, it must be an fx2lafw. */
181                 ret = TRUE;
182         }
183         if (hdl)
184                 libusb_close(hdl);
185
186         return ret;
187 }
188
189 SR_PRIV int fx2lafw_dev_open(struct sr_dev_inst *sdi, struct sr_dev_driver *di)
190 {
191         libusb_device **devlist;
192         struct sr_usb_dev_inst *usb;
193         struct libusb_device_descriptor des;
194         struct dev_context *devc;
195         struct drv_context *drvc;
196         struct version_info vi;
197         int ret, skip, i, device_count;
198         uint8_t revid;
199
200         drvc = di->priv;
201         devc = sdi->priv;
202         usb = sdi->conn;
203
204         if (sdi->status == SR_ST_ACTIVE)
205                 /* Device is already in use. */
206                 return SR_ERR;
207
208         skip = 0;
209         device_count = libusb_get_device_list(drvc->sr_ctx->libusb_ctx, &devlist);
210         if (device_count < 0) {
211                 sr_err("Failed to get device list: %s.",
212                        libusb_error_name(device_count));
213                 return SR_ERR;
214         }
215
216         for (i = 0; i < device_count; i++) {
217                 if ((ret = libusb_get_device_descriptor(devlist[i], &des))) {
218                         sr_err("Failed to get device descriptor: %s.",
219                                libusb_error_name(ret));
220                         continue;
221                 }
222
223                 if (des.idVendor != devc->profile->vid
224                     || des.idProduct != devc->profile->pid)
225                         continue;
226
227                 if (sdi->status == SR_ST_INITIALIZING) {
228                         if (skip != sdi->index) {
229                                 /* Skip devices of this type that aren't the one we want. */
230                                 skip += 1;
231                                 continue;
232                         }
233                 } else if (sdi->status == SR_ST_INACTIVE) {
234                         /*
235                          * This device is fully enumerated, so we need to find
236                          * this device by vendor, product, bus and address.
237                          */
238                         if (libusb_get_bus_number(devlist[i]) != usb->bus
239                                 || libusb_get_device_address(devlist[i]) != usb->address)
240                                 /* This is not the one. */
241                                 continue;
242                 }
243
244                 if (!(ret = libusb_open(devlist[i], &usb->devhdl))) {
245                         if (usb->address == 0xff)
246                                 /*
247                                  * First time we touch this device after FW
248                                  * upload, so we don't know the address yet.
249                                  */
250                                 usb->address = libusb_get_device_address(devlist[i]);
251                 } else {
252                         sr_err("Failed to open device: %s.",
253                                libusb_error_name(ret));
254                         break;
255                 }
256
257                 ret = command_get_fw_version(usb->devhdl, &vi);
258                 if (ret != SR_OK) {
259                         sr_err("Failed to get firmware version.");
260                         break;
261                 }
262
263                 ret = command_get_revid_version(sdi, &revid);
264                 if (ret != SR_OK) {
265                         sr_err("Failed to get REVID.");
266                         break;
267                 }
268
269                 /*
270                  * Changes in major version mean incompatible/API changes, so
271                  * bail out if we encounter an incompatible version.
272                  * Different minor versions are OK, they should be compatible.
273                  */
274                 if (vi.major != FX2LAFW_REQUIRED_VERSION_MAJOR) {
275                         sr_err("Expected firmware version %d.x, "
276                                "got %d.%d.", FX2LAFW_REQUIRED_VERSION_MAJOR,
277                                vi.major, vi.minor);
278                         break;
279                 }
280
281                 sdi->status = SR_ST_ACTIVE;
282                 sr_info("Opened device %d on %d.%d, "
283                         "interface %d, firmware %d.%d.",
284                         sdi->index, usb->bus, usb->address,
285                         USB_INTERFACE, vi.major, vi.minor);
286
287                 sr_info("Detected REVID=%d, it's a Cypress CY7C68013%s.",
288                         revid, (revid != 1) ? " (FX2)" : "A (FX2LP)");
289
290                 break;
291         }
292         libusb_free_device_list(devlist, 1);
293
294         if (sdi->status != SR_ST_ACTIVE)
295                 return SR_ERR;
296
297         return SR_OK;
298 }
299
300 SR_PRIV int fx2lafw_configure_channels(const struct sr_dev_inst *sdi)
301 {
302         struct dev_context *devc;
303         struct sr_channel *ch;
304         GSList *l;
305         int channel_bit, stage, i;
306         char *tc;
307
308         devc = sdi->priv;
309         for (i = 0; i < NUM_TRIGGER_STAGES; i++) {
310                 devc->trigger_mask[i] = 0;
311                 devc->trigger_value[i] = 0;
312         }
313
314         stage = -1;
315         for (l = sdi->channels; l; l = l->next) {
316                 ch = (struct sr_channel *)l->data;
317                 if (ch->enabled == FALSE)
318                         continue;
319
320                 if (ch->index > 7)
321                         devc->sample_wide = TRUE;
322
323                 channel_bit = 1 << (ch->index);
324                 if (!(ch->trigger))
325                         continue;
326
327                 stage = 0;
328                 for (tc = ch->trigger; *tc; tc++) {
329                         devc->trigger_mask[stage] |= channel_bit;
330                         if (*tc == '1')
331                                 devc->trigger_value[stage] |= channel_bit;
332                         stage++;
333                         if (stage > NUM_TRIGGER_STAGES)
334                                 return SR_ERR;
335                 }
336         }
337
338         if (stage == -1) {
339                 /*
340                  * We didn't configure any triggers, make sure acquisition
341                  * doesn't wait for any.
342                  */
343                 devc->trigger_fired = TRUE;
344         } else {
345                 devc->trigger_fired = FALSE;
346                 devc->trigger_stage = 0;
347         }
348
349         return SR_OK;
350 }
351
352 SR_PRIV struct dev_context *fx2lafw_dev_new(void)
353 {
354         struct dev_context *devc;
355
356         if (!(devc = g_try_malloc(sizeof(struct dev_context)))) {
357                 sr_err("Device context malloc failed.");
358                 return NULL;
359         }
360
361         devc->profile = NULL;
362         devc->fw_updated = 0;
363         devc->cur_samplerate = 0;
364         devc->limit_samples = 0;
365         devc->sample_wide = FALSE;
366
367         return devc;
368 }
369
370 SR_PRIV void fx2lafw_abort_acquisition(struct dev_context *devc)
371 {
372         int i;
373
374         devc->acq_aborted = TRUE;
375
376         for (i = devc->num_transfers - 1; i >= 0; i--) {
377                 if (devc->transfers[i])
378                         libusb_cancel_transfer(devc->transfers[i]);
379         }
380 }
381
382 static void finish_acquisition(struct dev_context *devc)
383 {
384         struct sr_datafeed_packet packet;
385
386         /* Terminate session. */
387         packet.type = SR_DF_END;
388         sr_session_send(devc->cb_data, &packet);
389
390         /* Remove fds from polling. */
391         usb_source_remove(devc->ctx);
392
393         devc->num_transfers = 0;
394         g_free(devc->transfers);
395 }
396
397 static void free_transfer(struct libusb_transfer *transfer)
398 {
399         struct dev_context *devc;
400         unsigned int i;
401
402         devc = transfer->user_data;
403
404         g_free(transfer->buffer);
405         transfer->buffer = NULL;
406         libusb_free_transfer(transfer);
407
408         for (i = 0; i < devc->num_transfers; i++) {
409                 if (devc->transfers[i] == transfer) {
410                         devc->transfers[i] = NULL;
411                         break;
412                 }
413         }
414
415         devc->submitted_transfers--;
416         if (devc->submitted_transfers == 0)
417                 finish_acquisition(devc);
418 }
419
420 static void resubmit_transfer(struct libusb_transfer *transfer)
421 {
422         int ret;
423
424         if ((ret = libusb_submit_transfer(transfer)) == LIBUSB_SUCCESS)
425                 return;
426
427         free_transfer(transfer);
428         /* TODO: Stop session? */
429
430         sr_err("%s: %s", __func__, libusb_error_name(ret));
431 }
432
433 SR_PRIV void fx2lafw_receive_transfer(struct libusb_transfer *transfer)
434 {
435         gboolean packet_has_error = FALSE;
436         struct sr_datafeed_packet packet;
437         struct sr_datafeed_logic logic;
438         struct dev_context *devc;
439         unsigned int trigger_offset, num_samples;
440         int cur_sample_count, sample_width, i;
441         uint8_t *cur_buf;
442         uint16_t cur_sample;
443
444         devc = transfer->user_data;
445
446         /*
447          * If acquisition has already ended, just free any queued up
448          * transfer that come in.
449          */
450         if (devc->acq_aborted) {
451                 free_transfer(transfer);
452                 return;
453         }
454
455         sr_info("receive_transfer(): status %d received %d bytes.",
456                 transfer->status, transfer->actual_length);
457
458         /* Save incoming transfer before reusing the transfer struct. */
459         cur_buf = transfer->buffer;
460         sample_width = devc->sample_wide ? 2 : 1;
461         cur_sample_count = transfer->actual_length / sample_width;
462
463         switch (transfer->status) {
464         case LIBUSB_TRANSFER_NO_DEVICE:
465                 fx2lafw_abort_acquisition(devc);
466                 free_transfer(transfer);
467                 return;
468         case LIBUSB_TRANSFER_COMPLETED:
469         case LIBUSB_TRANSFER_TIMED_OUT: /* We may have received some data though. */
470                 break;
471         default:
472                 packet_has_error = TRUE;
473                 break;
474         }
475
476         if (transfer->actual_length == 0 || packet_has_error) {
477                 devc->empty_transfer_count++;
478                 if (devc->empty_transfer_count > MAX_EMPTY_TRANSFERS) {
479                         /*
480                          * The FX2 gave up. End the acquisition, the frontend
481                          * will work out that the samplecount is short.
482                          */
483                         fx2lafw_abort_acquisition(devc);
484                         free_transfer(transfer);
485                 } else {
486                         resubmit_transfer(transfer);
487                 }
488                 return;
489         } else {
490                 devc->empty_transfer_count = 0;
491         }
492
493         trigger_offset = 0;
494         if (!devc->trigger_fired) {
495                 for (i = 0; i < cur_sample_count; i++) {
496                         cur_sample = devc->sample_wide ?
497                                 *((uint16_t *)cur_buf + i) :
498                                 *((uint8_t *)cur_buf + i);
499
500                         if ((cur_sample & devc->trigger_mask[devc->trigger_stage]) ==
501                                 devc->trigger_value[devc->trigger_stage]) {
502                                 /* Match on this trigger stage. */
503                                 devc->trigger_buffer[devc->trigger_stage] = cur_sample;
504                                 devc->trigger_stage++;
505
506                                 if (devc->trigger_stage == NUM_TRIGGER_STAGES ||
507                                         devc->trigger_mask[devc->trigger_stage] == 0) {
508                                         /* Match on all trigger stages, we're done. */
509                                         trigger_offset = i;
510
511                                         /*
512                                          * TODO: Send pre-trigger buffer to session bus.
513                                          * Tell the frontend we hit the trigger here.
514                                          */
515                                         packet.type = SR_DF_TRIGGER;
516                                         packet.payload = NULL;
517                                         sr_session_send(devc->cb_data, &packet);
518
519                                         /*
520                                          * Send the samples that triggered it,
521                                          * since we're skipping past them.
522                                          */
523                                         packet.type = SR_DF_LOGIC;
524                                         packet.payload = &logic;
525                                         num_samples = cur_sample_count - trigger_offset;
526                                         if (devc->limit_samples &&
527                                                         num_samples > devc->limit_samples - devc->sent_samples)
528                                                 num_samples = devc->limit_samples - devc->sent_samples;
529                                         logic.length = num_samples * sample_width;
530                                         logic.unitsize = sample_width;
531                                         logic.data = cur_buf + trigger_offset * sample_width;
532                                         sr_session_send(devc->cb_data, &packet);
533                                         devc->sent_samples += num_samples;
534
535                                         devc->trigger_fired = TRUE;
536                                         break;
537                                 }
538                         } else if (devc->trigger_stage > 0) {
539                                 /*
540                                  * We had a match before, but not in the next sample. However, we may
541                                  * have a match on this stage in the next bit -- trigger on 0001 will
542                                  * fail on seeing 00001, so we need to go back to stage 0 -- but at
543                                  * the next sample from the one that matched originally, which the
544                                  * counter increment at the end of the loop takes care of.
545                                  */
546                                 i -= devc->trigger_stage;
547                                 if (i < -1)
548                                         i = -1; /* Oops, went back past this buffer. */
549                                 /* Reset trigger stage. */
550                                 devc->trigger_stage = 0;
551                         }
552                 }
553         } else if (devc->sent_samples < devc->limit_samples) {
554                 /* Send the incoming transfer to the session bus. */
555                 packet.type = SR_DF_LOGIC;
556                 packet.payload = &logic;
557                 if (devc->sent_samples + cur_sample_count > devc->limit_samples)
558                         num_samples = devc->limit_samples - devc->sent_samples;
559                 else
560                         num_samples = cur_sample_count;
561                 logic.length = num_samples * sample_width;
562                 logic.unitsize = sample_width;
563                 logic.data = cur_buf;
564                 sr_session_send(devc->cb_data, &packet);
565                 devc->sent_samples += cur_sample_count;
566         }
567
568         if (devc->limit_samples && devc->sent_samples >= devc->limit_samples) {
569                 fx2lafw_abort_acquisition(devc);
570                 free_transfer(transfer);
571                 return;
572         }
573
574         resubmit_transfer(transfer);
575 }
576
577 static unsigned int to_bytes_per_ms(unsigned int samplerate)
578 {
579         return samplerate / 1000;
580 }
581
582 SR_PRIV size_t fx2lafw_get_buffer_size(struct dev_context *devc)
583 {
584         size_t s;
585
586         /*
587          * The buffer should be large enough to hold 10ms of data and
588          * a multiple of 512.
589          */
590         s = 10 * to_bytes_per_ms(devc->cur_samplerate);
591         return (s + 511) & ~511;
592 }
593
594 SR_PRIV unsigned int fx2lafw_get_number_of_transfers(struct dev_context *devc)
595 {
596         unsigned int n;
597
598         /* Total buffer size should be able to hold about 500ms of data. */
599         n = (500 * to_bytes_per_ms(devc->cur_samplerate) /
600                 fx2lafw_get_buffer_size(devc));
601
602         if (n > NUM_SIMUL_TRANSFERS)
603                 return NUM_SIMUL_TRANSFERS;
604
605         return n;
606 }
607
608 SR_PRIV unsigned int fx2lafw_get_timeout(struct dev_context *devc)
609 {
610         size_t total_size;
611         unsigned int timeout;
612
613         total_size = fx2lafw_get_buffer_size(devc) *
614                         fx2lafw_get_number_of_transfers(devc);
615         timeout = total_size / to_bytes_per_ms(devc->cur_samplerate);
616         return timeout + timeout / 4; /* Leave a headroom of 25% percent. */
617 }