idevicebtlogger: add --format option for 'pcap' and 'packetlogger'

1 files changed, 93 insertions, 13 deletions

diff --git a/tools/idevicebtlogger.c b/tools/idevicebtlogger.c
index b73d958..c780143 100644
--- a/tools/idevicebtlogger.c
+++ b/tools/idevicebtlogger.c
@@ -32,6 +32,8 @@
 #include <stdlib.h>
 #include <unistd.h>
 #include <getopt.h>
+#include <assert.h>
+#include <fcntl.h>
 
 #ifdef WIN32
 #include <windows.h>
@@ -57,7 +59,14 @@ static idevice_t device = NULL;
 static bt_packet_logger_client_t bt_packet_logger = NULL;
 static int use_network = 0;
 static char* out_filename = NULL;
-static pcap_dumper_t * dump;
+static char* log_format_string = NULL;
+static pcap_dumper_t * pcap_dumper = NULL;
+static int packetlogger_fd = -1;
+
+static enum {
+	LOG_FORMAT_PACKETLOGGER,
+	LOG_FORMAT_PCAP
+} log_format = LOG_FORMAT_PACKETLOGGER;
 
 typedef enum {
 	HCI_COMMAND = 0x00,
@@ -67,9 +76,17 @@ typedef enum {
 } PacketLoggerPacketType;
 
 /**
+ * Callback from the packet logger service to handle packets and log to PacketLoggger format
+ */
+static void bt_packet_logger_callback_packetlogger(uint8_t * data, uint16_t len, void *user_data)
+{
+	write(packetlogger_fd, data, len);
+}
+
+/**
  * Callback from the packet logger service to handle packets and log to pcap
  */
-static void bt_packet_logger_callback(uint8_t * data, uint16_t len, void *user_data)
+static void bt_packet_logger_callback_pcap(uint8_t * data, uint16_t len, void *user_data)
 {
 	bt_packet_logger_header_t * header = (bt_packet_logger_header_t *)data;
 	uint16_t offset = sizeof(bt_packet_logger_header_t);
@@ -126,8 +143,8 @@ static void bt_packet_logger_callback(uint8_t * data, uint16_t len, void *user_d
 		// having to memcpy things around.
 		offset -= sizeof(uint32_t);
 		*(uint32_t*)&data[offset] = direction;
-		pcap_dump((unsigned char*)dump, &pcap_header, &data[offset]);
-		pcap_dump_flush(dump);
+		pcap_dump((unsigned char*)pcap_dumper, &pcap_header, &data[offset]);
+		pcap_dump_flush(pcap_dumper);
 	}
 }
 
@@ -173,7 +190,19 @@ static int start_logging(void)
 	bt_packet_logger_client_start_service(device, &bt_packet_logger, TOOL_NAME);
 
 	/* start capturing bt_packet_logger */
-	bt_packet_logger_error_t serr = bt_packet_logger_start_capture(bt_packet_logger, bt_packet_logger_callback, NULL);
+	void (*callback)(uint8_t * data, uint16_t len, void *user_data);
+	switch (log_format){
+		case LOG_FORMAT_PCAP:
+			callback = bt_packet_logger_callback_pcap;
+			break;
+		case LOG_FORMAT_PACKETLOGGER:
+			callback = bt_packet_logger_callback_packetlogger;
+			break;
+		default:
+			assert(0);
+			return 0;
+	}
+	bt_packet_logger_error_t serr = bt_packet_logger_start_capture(bt_packet_logger, callback, NULL);
 	if (serr != BT_PACKET_LOGGER_E_SUCCESS) {
 		fprintf(stderr, "ERROR: Unable to start capturing bt_packet_logger.\n");
 		bt_packet_logger_client_free(bt_packet_logger);
@@ -243,12 +272,13 @@ static void print_usage(int argc, char **argv, int is_error)
 		"Capture HCI packets from a connected device.\n" \
 		"\n" \
 		"OPTIONS:\n" \
-		"  -u, --udid UDID  target specific device by UDID\n" \
-		"  -n, --network    connect to network device\n" \
-		"  -x, --exit       exit when device disconnects\n" \
-		"  -h, --help       prints usage information\n" \
-		"  -d, --debug      enable communication debugging\n" \
-		"  -v, --version    prints version information\n" \
+		"  -u, --udid UDID     target specific device by UDID\n" \
+		"  -n, --network       connect to network device\n" \
+		"  -f, --format FORMAT logging format: packetlogger (default) or pcap\n" \
+		"  -x, --exit          exit when device disconnects\n" \
+		"  -h, --help          prints usage information\n" \
+		"  -d, --debug         enable communication debugging\n" \
+		"  -v, --version       prints version information\n" \
 		"\n" \
 		"Homepage:    <" PACKAGE_URL ">\n"
 		"Bug Reports: <" PACKAGE_BUGREPORT ">\n"
@@ -265,6 +295,7 @@ int main(int argc, char *argv[])
 		{ "debug", no_argument, NULL, 'd' },
 		{ "help", no_argument, NULL, 'h' },
 		{ "udid", required_argument, NULL, 'u' },
+		{ "format", required_argument, NULL, 'f' },
 		{ "network", no_argument, NULL, 'n' },
 		{ "exit", no_argument, NULL, 'x' },
 		{ "version", no_argument, NULL, 'v' },
@@ -278,7 +309,7 @@ int main(int argc, char *argv[])
 	signal(SIGPIPE, SIG_IGN);
 #endif
 
-	while ((c = getopt_long(argc, argv, "dhu:nxv", longopts, NULL)) != -1) {
+	while ((c = getopt_long(argc, argv, "dhu:f:nxv", longopts, NULL)) != -1) {
 		switch (c) {
 		case 'd':
 			idevice_set_debug_level(1);
@@ -292,6 +323,15 @@ int main(int argc, char *argv[])
 			free(udid);
 			udid = strdup(optarg);
 			break;
+		case 'f':
+			if (!*optarg) {
+				fprintf(stderr, "ERROR: FORMAT must not be empty!\n");
+				print_usage(argc, argv, 1);
+				return 2;
+			}
+			free(log_format_string);
+			log_format_string = strdup(optarg);
+			break;
 		case 'n':
 			use_network = 1;
 			break;
@@ -319,10 +359,23 @@ int main(int argc, char *argv[])
 		return 2;
 	}
 
+	if (log_format_string != NULL){
+		if (strcmp("packetlogger", log_format_string) == 0){
+			log_format = LOG_FORMAT_PACKETLOGGER;
+		} else if (strcmp("pcap", log_format_string) == 0){
+			log_format = LOG_FORMAT_PCAP;
+		} else {
+			printf("Unknown logging format: '%s'\n", log_format_string);
+			print_usage(argc, argv, 1);
+			return 2;
+		}
+	}
+
 	int num = 0;
 	idevice_info_t *devices = NULL;
 	idevice_get_device_list_extended(&devices, &num);
 	idevice_device_list_extended_free(devices);
+	int oflags;
 	if (num == 0) {
 		if (!udid) {
 			fprintf(stderr, "No device found. Plug in a device or pass UDID with -u to wait for device to be available.\n");
@@ -332,7 +385,27 @@ int main(int argc, char *argv[])
 		}
 	}
 
-	dump = pcap_dump_open(pcap_open_dead(DLT_BLUETOOTH_HCI_H4_WITH_PHDR, BT_MAX_PACKET_SIZE), out_filename);
+	switch (log_format){
+		case LOG_FORMAT_PCAP:
+			printf("Output Format: PCAP\n");
+			pcap_dumper = pcap_dump_open(pcap_open_dead(DLT_BLUETOOTH_HCI_H4_WITH_PHDR, BT_MAX_PACKET_SIZE), out_filename);
+			break;
+		case LOG_FORMAT_PACKETLOGGER:
+			printf("Output Format: PacketLogger\n");
+		    oflags = O_WRONLY | O_CREAT | O_TRUNC;
+#ifdef WIN32
+		    default_oflags |= O_BINARY;
+#endif
+		    packetlogger_fd = open(out_filename, oflags);
+		    if (packetlogger_fd < 0){
+		        fprintf(stderr, "Failed to open file %s, errno = %d\n", out_filename, errno);
+		        return -2;
+		    }		
+	    	break;
+		default:
+			assert(0);
+			return -2;
+	}
 	idevice_event_subscribe(device_event_cb, NULL);
 
 	while (!quit_flag) {
@@ -342,6 +415,13 @@ int main(int argc, char *argv[])
 	idevice_event_unsubscribe();
 	stop_logging();
 
+	if (pcap_dumper) {
+		pcap_dump_close(pcap_dumper);
+	}
+	if (packetlogger_fd >= 0){
+		close(packetlogger_fd);
+	}
+
 	free(udid);
 
 	return 0;