X-Git-Url: https://sigrok.org/gitweb/?a=blobdiff_plain;f=input%2Finput_binary.c;h=4286a6c3459a1bcd5eaf7e1be1b863f8159be4dd;hb=719c5a934c7705466a449854b876b9962eb4cb5e;hp=a6c6005aae78c87828a96596b411a7dfba897552;hpb=34e4813f2e8b75981ed92d625c5fd55146a35e66;p=libsigrok.git

diff --git a/input/input_binary.c b/input/input_binary.c
index a6c6005a..4286a6c3 100644
--- a/input/input_binary.c
+++ b/input/input_binary.c
@@ -17,67 +17,89 @@
  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
+#include <stdlib.h>
 #include <sys/types.h>
 #include <sys/stat.h>
 #include <fcntl.h>
 #include <unistd.h>
 #include <sys/time.h>
-
 #include <sigrok.h>
 
-#define CHUNKSIZE 4096
+#define CHUNKSIZE          4096
+#define DEFAULT_NUM_PROBES    8
 
 
-static int format_match(char *filename)
+static int format_match(const char *filename)
 {
+	/* suppress compiler warning */
+	(void)filename;
 
-	filename = NULL;
-
+	/* this module will handle anything you throw at it */
 	return TRUE;
 }
 
+static int init(struct sr_input *in)
+{
+	int num_probes;
+
+	if (in->param && in->param[0]) {
+		num_probes = strtoul(in->param, NULL, 10);
+		if (num_probes < 1)
+			return SR_ERR;
+	} else
+		num_probes = DEFAULT_NUM_PROBES;
+
+	/* create a virtual device */
+	in->vdevice = sr_device_new(NULL, 0, num_probes);
+
+	return SR_OK;
+}
 
-/* TODO: number of probes hardcoded to 8 */
-static int in_loadfile(char *filename)
+static int loadfile(struct sr_input *in, const char *filename)
 {
-	struct datafeed_header header;
-	struct datafeed_packet packet;
-	char buffer[CHUNKSIZE];
-	int fd, size;
+	struct sr_datafeed_header header;
+	struct sr_datafeed_packet packet;
+	struct sr_datafeed_logic logic;
+	unsigned char buffer[CHUNKSIZE];
+	int fd, size, num_probes;
+
+	if ((fd = open(filename, O_RDONLY)) == -1)
+		return SR_ERR;
 
-	if( (fd = open(filename, O_RDONLY)) == -1)
-		return SIGROK_ERR;
+	num_probes = g_slist_length(in->vdevice->probes);
 
+	/* send header */
 	header.feed_version = 1;
-	header.num_probes = 8;
-	header.protocol_id = PROTO_RAW;
+	header.num_logic_probes = num_probes;
+	header.num_analog_probes = 0;
 	header.samplerate = 0;
 	gettimeofday(&header.starttime, NULL);
-	packet.type = DF_HEADER;
-	packet.length = sizeof(struct datafeed_header);
+	packet.type = SR_DF_HEADER;
 	packet.payload = &header;
-	session_bus(NULL, &packet);
-
-	packet.type = DF_LOGIC8;
-	packet.payload = buffer;
-	while( (size = read(fd, buffer, CHUNKSIZE)) > 0) {
-		packet.length = size;
-		session_bus(NULL, &packet);
+	sr_session_bus(in->vdevice, &packet);
+
+	/* chop up the input file into chunks and feed it into the session bus */
+	packet.type = SR_DF_LOGIC;
+	packet.payload = &logic;
+	logic.unitsize = (num_probes + 7) / 8;
+	logic.data = buffer;
+	while ((size = read(fd, buffer, CHUNKSIZE)) > 0) {
+		logic.length = size;
+		sr_session_bus(in->vdevice, &packet);
 	}
 	close(fd);
 
-	packet.type = DF_END;
-	session_bus(NULL, &packet);
+	/* end of stream */
+	packet.type = SR_DF_END;
+	sr_session_bus(in->vdevice, &packet);
 
-
-	return SIGROK_OK;
+	return SR_OK;
 }
 
-
-struct input_format input_binary = {
-		"binary",
-		"Raw binary",
-		format_match,
-		in_loadfile
+struct sr_input_format input_binary = {
+	.id = "binary",
+	.description = "Raw binary",
+	.format_match = format_match,
+	.init = init,
+	.loadfile = loadfile,
 };
-