2 * This file is part of the libsigrok project.
4 * Copyright (C) 2013 Bert Vermeulen <bert@biot.com>
6 * This program is free software: you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation, either version 3 of the License, or
9 * (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with this program. If not, see <http://www.gnu.org/licenses/>.
20 #include <sys/types.h>
26 #include "libsigrok.h"
27 #include "libsigrok-internal.h"
29 #define LOG_PREFIX "virtual-session"
31 /* size of payloads sent across the session bus */
33 #define CHUNKSIZE (512 * 1024)
40 struct zip_file *capfile;
49 static GSList *dev_insts = NULL;
50 static const int hwcaps[] = {
52 SR_CONF_CAPTURE_UNITSIZE,
56 static int receive_data(int fd, int revents, void *cb_data)
58 struct sr_dev_inst *sdi;
59 struct session_vdev *vdev;
60 struct sr_datafeed_packet packet;
61 struct sr_datafeed_logic logic;
72 for (l = dev_insts; l; l = l->next) {
76 /* Already done with this instance. */
80 /* No capture file opened yet, or finished with the last
82 if (vdev->cur_chunk == 0) {
83 /* capturefile is always the unchunked base name. */
84 if (zip_stat(vdev->archive, vdev->capturefile, 0, &zs) != -1) {
85 /* No chunks, just a single capture file. */
87 if (!(vdev->capfile = zip_fopen(vdev->archive,
88 vdev->capturefile, 0)))
90 sr_dbg("Opened %s.", vdev->capturefile);
92 /* Try as first chunk filename. */
93 snprintf(capturefile, 15, "%s-1", vdev->capturefile);
94 if (zip_stat(vdev->archive, capturefile, 0, &zs) != -1) {
96 if (!(vdev->capfile = zip_fopen(vdev->archive,
99 sr_dbg("Opened %s.", capturefile);
101 sr_err("No capture file '%s' in " "session file '%s'.",
102 vdev->capturefile, vdev->sessionfile);
107 /* Capture data is chunked, advance to the next chunk. */
109 snprintf(capturefile, 15, "%s-%d", vdev->capturefile,
111 if (zip_stat(vdev->archive, capturefile, 0, &zs) != -1) {
112 if (!(vdev->capfile = zip_fopen(vdev->archive,
115 sr_dbg("Opened %s.", capturefile);
117 /* We got all the chunks, finish up. */
118 vdev->finished = TRUE;
124 if (!(buf = g_try_malloc(CHUNKSIZE))) {
125 sr_err("%s: buf malloc failed", __func__);
129 ret = zip_fread(vdev->capfile, buf,
130 CHUNKSIZE / vdev->unitsize * vdev->unitsize);
132 if (ret % vdev->unitsize != 0)
133 sr_warn("Read size %d not a multiple of the"
134 " unit size %d.", ret, vdev->unitsize);
136 packet.type = SR_DF_LOGIC;
137 packet.payload = &logic;
139 logic.unitsize = vdev->unitsize;
141 vdev->bytes_read += ret;
142 sr_session_send(cb_data, &packet);
144 /* done with this capture file */
145 zip_fclose(vdev->capfile);
146 vdev->capfile = NULL;
147 if (vdev->cur_chunk == 0) {
148 /* It was the only file. */
149 vdev->finished = TRUE;
151 /* There might be more chunks, so don't fall through
152 * to the SR_DF_END here. */
161 packet.type = SR_DF_END;
162 sr_session_send(cb_data, &packet);
163 sr_session_source_remove(-1);
169 /* driver callbacks */
171 static int init(struct sr_context *sr_ctx)
178 static int dev_clear(void)
182 for (l = dev_insts; l; l = l->next)
183 sr_dev_inst_free(l->data);
184 g_slist_free(dev_insts);
190 static int dev_open(struct sr_dev_inst *sdi)
192 struct session_vdev *vdev;
194 vdev = g_try_malloc0(sizeof(struct session_vdev));
196 dev_insts = g_slist_append(dev_insts, sdi);
201 static int dev_close(struct sr_dev_inst *sdi)
203 const struct session_vdev *const vdev = sdi->priv;
204 g_free(vdev->sessionfile);
205 g_free(vdev->capturefile);
213 static int config_get(int id, GVariant **data, const struct sr_dev_inst *sdi,
214 const struct sr_probe_group *probe_group)
216 struct session_vdev *vdev;
221 case SR_CONF_SAMPLERATE:
224 *data = g_variant_new_uint64(vdev->samplerate);
235 static int config_set(int id, GVariant *data, const struct sr_dev_inst *sdi,
236 const struct sr_probe_group *probe_group)
238 struct session_vdev *vdev;
245 case SR_CONF_SAMPLERATE:
246 vdev->samplerate = g_variant_get_uint64(data);
247 sr_info("Setting samplerate to %" PRIu64 ".", vdev->samplerate);
249 case SR_CONF_SESSIONFILE:
250 g_free(vdev->sessionfile);
251 vdev->sessionfile = g_strdup(g_variant_get_string(data, NULL));
252 sr_info("Setting sessionfile to '%s'.", vdev->sessionfile);
254 case SR_CONF_CAPTUREFILE:
255 g_free(vdev->capturefile);
256 vdev->capturefile = g_strdup(g_variant_get_string(data, NULL));
257 sr_info("Setting capturefile to '%s'.", vdev->capturefile);
259 case SR_CONF_CAPTURE_UNITSIZE:
260 vdev->unitsize = g_variant_get_uint64(data);
262 case SR_CONF_NUM_LOGIC_PROBES:
263 vdev->num_probes = g_variant_get_uint64(data);
272 static int config_list(int key, GVariant **data, const struct sr_dev_inst *sdi,
273 const struct sr_probe_group *probe_group)
279 case SR_CONF_DEVICE_OPTIONS:
280 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_INT32,
281 hwcaps, ARRAY_SIZE(hwcaps), sizeof(int32_t));
290 static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data)
292 struct session_vdev *vdev;
297 vdev->bytes_read = 0;
299 vdev->finished = FALSE;
301 sr_info("Opening archive %s file %s", vdev->sessionfile,
304 if (!(vdev->archive = zip_open(vdev->sessionfile, 0, &ret))) {
305 sr_err("Failed to open session file '%s': "
306 "zip error %d.", vdev->sessionfile, ret);
310 /* Send header packet to the session bus. */
311 std_session_send_df_header(cb_data, LOG_PREFIX);
313 /* freewheeling source */
314 sr_session_source_add(-1, 0, 0, receive_data, cb_data);
320 SR_PRIV struct sr_dev_driver session_driver = {
321 .name = "virtual-session",
322 .longname = "Session-emulating driver",
325 .cleanup = dev_clear,
328 .dev_clear = dev_clear,
329 .config_get = config_get,
330 .config_set = config_set,
331 .config_list = config_list,
332 .dev_open = dev_open,
333 .dev_close = dev_close,
334 .dev_acquisition_start = dev_acquisition_start,
335 .dev_acquisition_stop = NULL,