]> sigrok.org Git - libsigrok.git/blobdiff - session.c
fix session saving and input file loading
[libsigrok.git] / session.c
index 2d6a74f98b784bff9e07cfe1f7aa6340fee47971..930b313ea244458f10c718555d35224283669619 100644 (file)
--- a/session.c
+++ b/session.c
 #include <unistd.h>
 #include <string.h>
 #include <zip.h>
-#include "sigrok.h"
+#include <sigrok.h>
+#include <config.h>
 
-/* there can only be one session at a time */
-struct session *session;
 
+/* There can only be one session at a time. */
+struct session *session;
 
-struct session *session_load(char *filename)
+struct session *session_load(const char *filename)
 {
        struct session *session;
 
-       /* TODO: implement */
+       /* Avoid compiler warnings. */
+       filename = filename;
+
+       /* TODO: Implement. */
        session = NULL;
 
        return session;
 }
 
-
 struct session *session_new(void)
 {
-
        session = calloc(1, sizeof(struct session));
 
        return session;
 }
 
-
 void session_destroy(void)
 {
-
        g_slist_free(session->devices);
 
-       /* TODO: loop over protocols and free them */
+       /* TODO: Loop over protocol decoders and free them. */
 
        g_free(session);
-
 }
 
-
 void session_device_clear(void)
 {
-
        g_slist_free(session->devices);
        session->devices = NULL;
-
 }
 
-
 int session_device_add(struct device *device)
 {
        int ret;
 
-       ret = device->plugin->open(device->plugin_index);
-       if(ret == SIGROK_OK)
-               session->devices = g_slist_append(session->devices, device);
+       if (device->plugin && device->plugin->open) {
+               ret = device->plugin->open(device->plugin_index);
+               if (ret != SIGROK_OK)
+                       return ret;
+       }
+
+       session->devices = g_slist_append(session->devices, device);
 
-       return ret;
+       return SIGROK_OK;
 }
 
-
 void session_pa_clear(void)
 {
-
-       /* the protocols are pointers to the global set of PA plugins, so don't free them */
+       /*
+        * The protocols are pointers to the global set of PA plugins,
+        * so don't free them.
+        */
        g_slist_free(session->analyzers);
        session->analyzers = NULL;
-
 }
 
-
 void session_pa_add(struct analyzer *an)
 {
-
        session->analyzers = g_slist_append(session->analyzers, an);
-
 }
 
-
 void session_datafeed_callback_clear(void)
 {
-
        g_slist_free(session->datafeed_callbacks);
        session->datafeed_callbacks = NULL;
-
 }
 
-
 void session_datafeed_callback_add(datafeed_callback callback)
 {
-
-       session->datafeed_callbacks = g_slist_append(session->datafeed_callbacks, callback);
-
+       session->datafeed_callbacks =
+           g_slist_append(session->datafeed_callbacks, callback);
 }
 
-
 int session_start(void)
 {
        struct device *device;
@@ -123,154 +113,143 @@ int session_start(void)
        int ret;
 
        g_message("starting acquisition");
-       for(l = session->devices; l; l = l->next)
-       {
+       for (l = session->devices; l; l = l->next) {
                device = l->data;
-               if( (ret = device->plugin->start_acquisition(device->plugin_index, device)) != SIGROK_OK)
+               if ((ret = device->plugin->start_acquisition(
+                               device->plugin_index, device)) != SIGROK_OK)
                        break;
        }
 
        return ret;
 }
 
-
 void session_stop(void)
 {
        struct device *device;
        GSList *l;
 
        g_message("stopping acquisition");
-       for(l = session->devices; l; l = l->next)
-       {
+       for (l = session->devices; l; l = l->next) {
                device = l->data;
-               device->plugin->stop_acquisition(device->plugin_index, device);
+               if (device->plugin)
+                       device->plugin->stop_acquisition(device->plugin_index, device);
        }
-
 }
 
-
 void session_bus(struct device *device, struct datafeed_packet *packet)
 {
        GSList *l;
        datafeed_callback cb;
 
-       /* TODO: send packet through PA pipe, and send the output of that to
-        * the callbacks as well
+       /*
+        * TODO: Send packet through PD pipe, and send the output of that to
+        * the callbacks as well.
         */
-
-       for(l = session->datafeed_callbacks; l; l = l->next)
-       {
+       for (l = session->datafeed_callbacks; l; l = l->next) {
                cb = l->data;
                cb(device, packet);
        }
-
 }
 
-
-void make_metadata(char *filename)
-{
-       GSList *l, *p;
-       struct device *device;
-       struct probe *probe;
-       FILE *f;
-       int devcnt;
-
-       f = fopen(filename, "wb");
-
-       /* general */
-
-       /* devices */
-       devcnt = 1;
-       for(l = session->devices; l; l = l->next) {
-               device = l->data;
-               fprintf(f, "[device]\n");
-               fprintf(f, "driver = %s\n", device->plugin->name);
-               if(device->datastore)
-                       fprintf(f, "capturefile = raw-%d\n", devcnt);
-               for(p = device->probes; p; p = p->next) {
-                       probe = p->data;
-                       if(probe->enabled)
-                       {
-                               fprintf(f, "probe %d", probe->index);
-                               if(probe->name)
-                                       fprintf(f, " name \"%s\"", probe->name);
-                               if(probe->trigger)
-                                       fprintf(f, " trigger \"%s\"", probe->trigger);
-                               fprintf(f, "\n");
-                       }
-               }
-               devcnt++;
-       }
-
-       /* TODO: protocol analyzers */
-
-       fclose(f);
-
-}
-
-
 int session_save(char *filename)
 {
-       GSList *l, *d;
+       GSList *l, *p, *d;
+       FILE *meta;
        struct device *device;
+       struct probe *probe;
        struct datastore *ds;
        struct zip *zipfile;
-       struct zip_source *src;
+       struct zip_source *versrc, *metasrc, *logicsrc;
        int bufcnt, devcnt, tmpfile, ret, error;
-       char version[1], rawname[16], metafile[32], *buf;
+       char version[1], rawname[16], metafile[32], *newfn, *buf;
 
-       /* quietly delete it first, libzip wants replace ops otherwise */
-       unlink(filename);
+       newfn = g_malloc(strlen(filename) + 10);
+       strcpy(newfn, filename);
+       if (strstr(filename, ".sigrok") != filename+strlen(filename)-7)
+               strcat(newfn, ".sigrok");
 
-       if( !(zipfile = zip_open(filename, ZIP_CREATE, &error)) )
+       /* Quietly delete it first, libzip wants replace ops otherwise. */
+       unlink(newfn);
+       if (!(zipfile = zip_open(newfn, ZIP_CREATE, &error)))
                return SIGROK_ERR;
+       g_free(newfn);
 
-       /* version */
+       /* "version" */
        version[0] = '1';
-       if( !(src = zip_source_buffer(zipfile, version, 1, 0)) )
+       if (!(versrc = zip_source_buffer(zipfile, version, 1, 0)))
                return SIGROK_ERR;
-       if(zip_add(zipfile, "version", src) == -1) {
-               g_message("error saving version into zipfile: %s", zip_strerror(zipfile));
+       if (zip_add(zipfile, "version", versrc) == -1) {
+               g_message("error saving version into zipfile: %s",
+                         zip_strerror(zipfile));
                return SIGROK_ERR;
        }
 
-       /* metadata */
+       /* init "metadata" */
        strcpy(metafile, "sigrok-meta-XXXXXX");
-       if(tmpfile = g_mkstemp(metafile)) == -1)
+       if ((tmpfile = g_mkstemp(metafile)) == -1)
                return SIGROK_ERR;
        close(tmpfile);
-       make_metadata(metafile);
-       if( !(src = zip_source_file(zipfile, metafile, 0, -1)) )
-               return SIGROK_ERR;
-       if(zip_add(zipfile, "metadata", src) == -1)
-               return SIGROK_ERR;
-       unlink(metafile);
+       meta = fopen(metafile, "wb");
+       fprintf(meta, "sigrok version = %s\n", PACKAGE_VERSION);
+       /* TODO: save protocol decoders used */
 
-       /* raw */
+       /* all datastores in all devices */
        devcnt = 1;
-       for(l = session->devices; l; l = l->next) {
+       for (l = session->devices; l; l = l->next) {
                device = l->data;
+               /* metadata */
+               fprintf(meta, "[device]\n");
+               if (device->plugin)
+                       fprintf(meta, "driver = %s\n", device->plugin->name);
+
                ds = device->datastore;
-               if(ds) {
-                       buf = malloc(ds->num_units * ds->ds_unitsize + DATASTORE_CHUNKSIZE);
+               if (ds) {
+                       /* metadata */
+                       fprintf(meta, "capturefile = logic-%d\n", devcnt);
+                       for (p = device->probes; p; p = p->next) {
+                               probe = p->data;
+                               if (probe->enabled) {
+                                       fprintf(meta, "probe %d", probe->index);
+                                       if (probe->name)
+                                               fprintf(meta, " name \"%s\"", probe->name);
+                                       if (probe->trigger)
+                                               fprintf(meta, " trigger \"%s\"",
+                                                       probe->trigger);
+                                       fprintf(meta, "\n");
+                               }
+                       }
+
+                       /* dump datastore into logic-n */
+                       buf = malloc(ds->num_units * ds->ds_unitsize +
+                                  DATASTORE_CHUNKSIZE);
                        bufcnt = 0;
-                       for(d = ds->chunklist; d; d = d->next) {
-                               memcpy(buf + bufcnt, d->data, DATASTORE_CHUNKSIZE);
+                       for (d = ds->chunklist; d; d = d->next) {
+                               memcpy(buf + bufcnt, d->data,
+                                      DATASTORE_CHUNKSIZE);
                                bufcnt += DATASTORE_CHUNKSIZE;
                        }
-                       if( !(src = zip_source_buffer(zipfile, buf, ds->num_units * ds->ds_unitsize, TRUE)) )
+                       if (!(logicsrc = zip_source_buffer(zipfile, buf,
+                                      ds->num_units * ds->ds_unitsize, TRUE)))
                                return SIGROK_ERR;
-                       snprintf(rawname, 15, "raw-%d", devcnt);
-                       if(zip_add(zipfile, rawname, src) == -1)
+                       snprintf(rawname, 15, "logic-%d", devcnt);
+                       if (zip_add(zipfile, rawname, logicsrc) == -1)
                                return SIGROK_ERR;
                }
                devcnt++;
        }
+       fclose(meta);
+
+       if (!(metasrc = zip_source_file(zipfile, metafile, 0, -1)))
+               return SIGROK_ERR;
+       if (zip_add(zipfile, "metadata", metasrc) == -1)
+               return SIGROK_ERR;
 
-       if(ret = zip_close(zipfile)) == -1) {
+       if ((ret = zip_close(zipfile)) == -1) {
                g_message("error saving zipfile: %s", zip_strerror(zipfile));
                return SIGROK_ERR;
        }
 
+       unlink(metafile);
+
        return SIGROK_OK;
 }