[Debug] Add kcov options support to kcov tester

Change-Id: I9cbd7caa9a1ab1ae7b7fc5303236a5e4a545aab1
diff --git a/hyp_kcov_test/hyp_kcov.c b/hyp_kcov_test/hyp_kcov.c
index 53acaf7..d87b894 100644
--- a/hyp_kcov_test/hyp_kcov.c
+++ b/hyp_kcov_test/hyp_kcov.c
@@ -20,6 +20,8 @@
 
 #define KCOV_TRACE_PC  0
 #define KCOV_TRACE_CMP 1
+#define KCOV_ENABLE_HYP (0x1ull << 48)
+#define KCOV_ENABLE_HYP_ONLY (0x1ull << 49)
 
 struct kcov_meta_buffer {
 	unsigned long num_entries;
@@ -80,13 +82,10 @@ int main(int argc, char **argv)
 		perror("mmap"), exit(1);
 	printf("Mmaped buffer\n");
 
-	if (ioctl(kcov_fd, KCOV_ENABLE, KCOV_TRACE_PC))
-		perror("ioctl"), exit(1);
-	printf("KCOV enabled\n");
-
-	if (ioctl(kcov_fd, KCOV_ENABLE, KCOV_TRACE_PC))
-		perror("ioctl"), exit(1);
-	printf("KCOV enabled2\n");
+	if (ioctl(kcov_fd, KCOV_ENABLE, KCOV_TRACE_PC | KCOV_ENABLE_HYP | KCOV_ENABLE_HYP_ONLY))
+	/* if (ioctl(kcov_fd, KCOV_ENABLE, KCOV_TRACE_PC)) */
+			perror("ioctl"), exit(1);
+	/* printf("KCOV enabled\n"); */
 
 	/* kvm_fd = open("/dev/kvm", O_RDWR); */