1 files changed, 39 insertions, 6 deletions

diff --git a/test/qemu-pagestep.c b/test/qemu-pagestep.c
index eea8c40..d46debd 100644
--- a/test/qemu-pagestep.c
+++ b/test/qemu-pagestep.c
@@ -15,23 +15,48 @@
 #include <stdio.h>
 #include <stdlib.h>
 
+struct cpc_event *batch_buffer;
+
+void
+report(struct cpc_event *event)
+{
+	if (event->type == CPC_EVENT_TRACK_PAGE) {
+		printf("Page event: rip:%016llx prev:%08llx next:%08llx ret:%llu\n",
+			vm_get_rip(), event->page.inst_gfn_prev,
+			event->page.inst_gfn, event->page.retinst);
+	} else if (event->type == CPC_EVENT_GUEST) {
+		printf("Guest event: type:%u arg:%u\n",
+			event->guest.type, event->guest.val);
+	} else {
+		printf("Unexpected event type %i\n", event->type);
+	}
+	printf("\n");
+}
+
 void
 monitor(void)
 {
 	struct cpc_event event;
+	struct cpc_event_batch batch;
+	size_t i;
 	int ret;
 
 	ret = ioctl(kvm_dev, KVM_CPC_POLL_EVENT, &event);
 	if (ret && errno == EAGAIN) return;
 	if (ret) err(1, "KVM_CPC_POLL_EVENT");
 
-	if (event.type == CPC_EVENT_TRACK_PAGE) {
-		printf("Event: rip:%016llx prev:%08llx next:%08llx ret:%llu\n",
-			vm_get_rip(), event.page.inst_gfn_prev,
-			event.page.inst_gfn, event.page.retinst);
-		printf("\n");
+	if (event.type == CPC_EVENT_BATCH) {
+		batch.cnt = 0;
+		batch.maxcnt = CPC_EVENT_BATCH_MAX;
+		batch.buf = batch_buffer;
+
+		ret = ioctl(kvm_dev, KVM_CPC_READ_BATCH, &batch);
+		if (ret) err(1, "KVM_CPC_READ_BATCH");
+
+		for (i = 0; i < batch.cnt; i++)
+			report(&batch.buf[i]);
 	} else {
-		printf("Unexpected event type %i\n", event.type);
+		report(&event);
 	}
 
 	ret = ioctl(kvm_dev, KVM_CPC_ACK_EVENT, &event.id);
@@ -42,8 +67,12 @@ int
 main(int argc, const char **argv)
 {
 	struct cpc_track_cfg cfg;
+	uint32_t arg;
 	int ret;
 
+	batch_buffer = malloc(sizeof(struct cpc_event) * CPC_EVENT_BATCH_MAX);
+	if (!batch_buffer) err(1, "malloc");
+
 	setvbuf(stdout, NULL, _IONBF, 0);
 
 	kvm_setup_init();
@@ -53,6 +82,10 @@ main(int argc, const char **argv)
 	ret = ioctl(kvm_dev, KVM_CPC_RESET);
 	if (ret) err(1, "KVM_CPC_RESET");
 
+	arg = true;
+	ret = ioctl(kvm_dev, KVM_CPC_BATCH_EVENTS, &arg);
+	if (ret) err(1, "KVM_CPC_BATCH_EVENTS");
+
 	memset(&cfg, 0, sizeof(cfg));
 	cfg.mode = CPC_TRACK_PAGES;
 	ret = ioctl(kvm_dev, KVM_CPC_TRACK_MODE, &cfg);