X-Git-Url: https://sigrok.org/gitweb/?a=blobdiff_plain;f=tests%2Flib.c;h=ad0b3e1f38a2c75b9faa043385801e3ce4cc322c;hb=00ebcbf48a775e56b6e73a39e69af390db547865;hp=44777bb3cd86df272f5233250888dd36498ecfd9;hpb=79bb0e97d53526ef6deb491ea9c7698ed6e90631;p=libsigrok.git

diff --git a/tests/lib.c b/tests/lib.c
index 44777bb3..ad0b3e1f 100644
--- a/tests/lib.c
+++ b/tests/lib.c
@@ -20,8 +20,11 @@
 
 #include <stdio.h>
 #include <string.h>
+#include <glib.h>
+#include <glib/gstdio.h>
 #include <check.h>
 #include "../libsigrok.h"
+#include "lib.h"
 
 /* Get a libsigrok driver by name. */
 struct sr_dev_driver *srtest_driver_get(const char *drivername)
@@ -42,6 +45,44 @@ struct sr_dev_driver *srtest_driver_get(const char *drivername)
 	return driver;
 }
 
+/* Get a libsigrok input format by ID. */
+struct sr_input_format *srtest_input_get(const char *id)
+{
+	struct sr_input_format **inputs, *input = NULL;
+	int i;
+
+	inputs = sr_input_list();
+	fail_unless(inputs != NULL, "No input modules found.");
+
+	for (i = 0; inputs[i]; i++) {
+		if (strcmp(inputs[i]->id, id))
+			continue;
+		input = inputs[i];
+	}
+	fail_unless(input != NULL, "Input module '%s' not found.", id);
+
+	return input;
+}
+
+/* Get a libsigrok output format by ID. */
+struct sr_output_format *srtest_output_get(const char *id)
+{
+	struct sr_output_format **outputs, *output = NULL;
+	int i;
+
+	outputs = sr_output_list();
+	fail_unless(outputs != NULL, "No output modules found.");
+
+	for (i = 0; outputs[i]; i++) {
+		if (strcmp(outputs[i]->id, id))
+			continue;
+		output = outputs[i];
+	}
+	fail_unless(output != NULL, "Output module '%s' not found.", id);
+
+	return output;
+}
+
 /* Initialize a libsigrok driver. */
 void srtest_driver_init(struct sr_context *sr_ctx, struct sr_dev_driver *driver)
 {
@@ -69,15 +110,53 @@ void srtest_driver_init_all(struct sr_context *sr_ctx)
 	}
 }
 
+/* Initialize a libsigrok input module. */
+void srtest_input_init(struct sr_context *sr_ctx, struct sr_input_format *input)
+{
+	int ret;
+	struct sr_input *in;
+
+	(void)sr_ctx;
+
+	in = g_try_malloc0(sizeof(struct sr_input));
+	fail_unless(in != NULL);
+
+	in->format = input;
+	in->param = NULL;
+
+	ret = in->format->init(in, "nonexisting.dat");
+	fail_unless(ret == SR_OK, "Failed to init '%s' input module: %d.",
+		    input->id, ret);
+
+	g_free(in);
+}
+
+/* Initialize all libsigrok input modules. */
+void srtest_input_init_all(struct sr_context *sr_ctx)
+{
+	struct sr_input_format **inputs;
+	int i;
+
+	inputs = sr_input_list();
+	fail_unless(inputs != NULL, "No input modules found.");
+
+	for (i = 0; inputs[i]; i++)
+		srtest_input_init(sr_ctx, inputs[i]);
+}
+
 /* Set the samplerate for the respective driver to the specified value. */
 void srtest_set_samplerate(struct sr_dev_driver *driver, uint64_t samplerate)
 {
 	int ret;
 	struct sr_dev_inst *sdi;
+	GVariant *gvar;
 
 	sdi = g_slist_nth_data(driver->priv, 0);
 
-	ret = driver->config_set(SR_CONF_SAMPLERATE, &samplerate, sdi);
+	gvar = g_variant_new_uint64(samplerate);
+	ret = driver->config_set(SR_CONF_SAMPLERATE, gvar, sdi, NULL);
+	g_variant_unref(gvar);
+
 	fail_unless(ret == SR_OK, "%s: Failed to set SR_CONF_SAMPLERATE: %d.",
 		    driver->name, ret);
 }
@@ -86,18 +165,20 @@ void srtest_set_samplerate(struct sr_dev_driver *driver, uint64_t samplerate)
 uint64_t srtest_get_samplerate(struct sr_dev_driver *driver)
 {
 	int ret;
-	uint64_t *samplerate;
+	uint64_t samplerate;
 	struct sr_dev_inst *sdi;
+	GVariant *gvar;
 
 	sdi = g_slist_nth_data(driver->priv, 0);
 
-	ret = driver->config_get(SR_CONF_SAMPLERATE,
-				 (const void **)&samplerate, sdi);
+	ret = driver->config_get(SR_CONF_SAMPLERATE, &gvar, sdi, NULL);
+	samplerate = g_variant_get_uint64(gvar);
+	g_variant_unref(gvar);
+
 	fail_unless(ret == SR_OK, "%s: Failed to get SR_CONF_SAMPLERATE: %d.",
 		    driver->name, ret);
-	fail_unless(samplerate != NULL);
 
-	return *samplerate;
+	return samplerate;
 }
 
 /* Check whether the respective driver can set/get the correct samplerate. */
@@ -114,3 +195,37 @@ void srtest_check_samplerate(struct sr_context *sr_ctx, const char *drivername,
 	fail_unless(s == samplerate, "%s: Incorrect samplerate: %" PRIu64 ".",
 		    drivername, s);
 }
+
+void srtest_buf_to_file(const char *filename, const uint8_t *buf, uint64_t len)
+{
+	FILE *f;
+	GError *error = NULL;
+	gboolean ret;
+
+	f = g_fopen(filename, "wb");
+	fail_unless(f != NULL);
+
+	ret = g_file_set_contents(filename, (const gchar *)buf, len, &error);
+	fail_unless(ret == TRUE);
+
+	fclose(f);
+}
+
+GArray *srtest_get_enabled_logic_channels(const struct sr_dev_inst *sdi)
+{
+	struct sr_channel *ch;
+	GArray *channels;
+	GSList *l;
+
+	channels = g_array_new(FALSE, FALSE, sizeof(int));
+	for (l = sdi->channels; l; l = l->next) {
+		ch = l->data;
+		if (ch->type != SR_CHANNEL_LOGIC)
+			continue;
+		if (ch->enabled != TRUE)
+			continue;
+		g_array_append_val(channels, ch->index);
+	}
+
+	return channels;
+}