Add support for opening a specific device

Allow filtration and matching of specified Device ID, Product ID, Serial
Number

Fix an issue in listKeyboards (hidapi) with an out of bounds search when
using serial number

Fix possible null reference problem in listKeyboards (hidapi) that
caused rare segfaults when traversing the device enumeration in
increments of two

Fix handling of output of listKeyboards (hidapi) where it was
incrementing the dev list pointer, then accessing the node to look for
device serial number (potential security risk)

Fix handling of serial number output of listKeyboards (hidapi) to handle
wchar_t instead of outputting the memory address

Fix issue in listKeyboards (libusb) failure to finish cleaning up USB
contexts, leading to a segfault if calling a separate function after
listing keyboards.

Fix issue in close (libusb) segfaulting if m_hidHandle was null, so
added a check.

Modify listKeyboards to provide a vector of DeviceInfo objects that can
be used by calling applications instead of outputting to stdout directly.

Implement a struct to hold information regarding device information and
ability for a library caller to query this information to make decisions
about the currently targeted device.

Signed-off-by: Kevin Pearson <pearson.kevin.m@gmail.com>

src/classes/Keyboard.cpp

@@ -5,6 +5,7 @@
 #include <vector>
 
 #if defined(hidapi)
+	#include <locale>
 	#include "hidapi/hidapi.h"
 #elif defined(libusb)
 	#include "libusb-1.0/libusb.h"
@@ -20,38 +21,57 @@ LedKeyboard::~LedKeyboard() {
 }
 
 
-bool LedKeyboard::listKeyboards() {
+vector<LedKeyboard::DeviceInfo> LedKeyboard::listKeyboards() {
+	vector<LedKeyboard::DeviceInfo> deviceList;
+
 	#if defined(hidapi)
-		if (hid_init() < 0) return false;
+		if (hid_init() < 0) return deviceList;
 		
 		struct hid_device_info *devs, *dev;
 		devs = hid_enumerate(0x0, 0x0);
 		dev = devs;
 		while (dev) {
-			for (int i=0; i<(int)SupportedKeyboards.size(); i++) {
+			for (size_t i=0; i<SupportedKeyboards.size(); i++) {
 				if (dev->vendor_id == SupportedKeyboards[i][0]) {
 					if (dev->product_id == SupportedKeyboards[i][1]) {
-						cout<<"0x"<<std::hex<<dev->vendor_id \
+						DeviceInfo deviceInfo;
-							<<" 0x"<<std::hex<<dev->product_id \
+							deviceInfo.vendorID=dev->vendor_id;
-							<<" "<<dev->serial_number \
+							deviceInfo.productID=dev->product_id;
-							<<" "<<dev->path<<" ";
+
+						if (dev->serial_number != NULL) {
+							char buf[256];
+							wcstombs(buf,dev->serial_number,256);
+							deviceInfo.serialNumber = string(buf);
+						}
+
+						if (dev->manufacturer_string != NULL)
+						{
+							char buf[256];
+							wcstombs(buf,dev->manufacturer_string,256);
+							deviceInfo.manufacturer = string(buf);
+						}
+
+						if (dev->product_string != NULL)
+						{
+							char buf[256];
+							wcstombs(buf,dev->product_string,256);
+							deviceInfo.product = string(buf);
+						}
+
+						deviceList.push_back(deviceInfo);
 						dev = dev->next;
-						cout<<dev->serial_number<<" "<<dev->path<<endl;
-						i++;
 						break;
 					}
 				}
 			}
-			dev = dev->next;
+			if (dev != NULL) dev = dev->next;
 		}
 		hid_free_enumeration(devs);
-		
 		hid_exit();
-		return true;
 		
 	#elif defined(libusb)
 		libusb_context *ctx = NULL;
-		if(libusb_init(&m_ctx) < 0) return false;
+		if(libusb_init(&m_ctx) < 0) return deviceList;
 		
 		libusb_device **devs;
 		ssize_t cnt = libusb_get_device_list(ctx, &devs);
@@ -62,20 +82,37 @@ bool LedKeyboard::listKeyboards() {
 			for (int i=0; i<(int)SupportedKeyboards.size(); i++) {
 				if (desc.idVendor == SupportedKeyboards[i][0]) {
 					if (desc.idProduct == SupportedKeyboards[i][1]) {
-						cout<<"0x"<<std::hex<<desc.idVendor \
+					  unsigned char buf[256];
-							<<" 0x"<<std::hex<<desc.idProduct<<endl;
+						DeviceInfo deviceInfo;
+							deviceInfo.vendorID=desc.idVendor;
+							deviceInfo.productID=desc.idProduct;
+
+						if (libusb_open(device, &m_hidHandle) != 0)	continue;
+
+						if (libusb_get_string_descriptor_ascii(m_hidHandle, desc.iSerialNumber, buf, 256) >= 1) deviceInfo.serialNumber = string((char*)buf);
+						if (libusb_get_string_descriptor_ascii(m_hidHandle, desc.iManufacturer, buf, 256) >= 1) deviceInfo.manufacturer = string((char*)buf);
+						if (libusb_get_string_descriptor_ascii(m_hidHandle, desc.iProduct, buf, 256) >= 1) deviceInfo.product = string((char*)buf);
+
+						deviceList.push_back(deviceInfo);
+						libusb_close(m_hidHandle);
+						m_hidHandle = NULL;
 						break;
 					}
 				}
 			}
 		}
 		libusb_free_device_list(devs, 1);
+
+		if (m_hidHandle != NULL) {
+			libusb_close(m_hidHandle);
+			m_hidHandle = NULL;
+		}
 		
 		libusb_exit(m_ctx);
-		return true;
+		m_ctx = NULL;
 	#endif
 	
-	return false;
+	return deviceList;
 }
 
 
@@ -86,114 +123,204 @@ bool LedKeyboard::isOpen() {
 bool LedKeyboard::open() {
 	if (m_isOpen) return true;
 	
+	return open(0x0,0x0,"");
+}
+
+bool LedKeyboard::open(uint16_t vendorID, uint16_t productID, string serial) {
+	if (m_isOpen && ! close()) return false;
+	currentDevice.model = KeyboardModel::unknown;
+
 	#if defined(hidapi)
-		if (hid_init() < 0) return false;
+	if (hid_init() < 0) return false;
-		
+
-		if (m_keyboardModel == KeyboardModel::unknown) {
+	struct hid_device_info *devs, *dev;
-			struct hid_device_info *devs, *dev;
+	devs = hid_enumerate(vendorID, productID);
-			devs = hid_enumerate(0x0, 0x0);
+	dev = devs;
-			dev = devs;
+	wstring wideSerial;
-			m_keyboardModel = KeyboardModel::unknown;
+
-			while (dev) {
+	if (!serial.empty()) {
-				for (int i=0; i<(int)SupportedKeyboards.size(); i++) {
+		wchar_t tempSerial[256];
-					if (dev->vendor_id == SupportedKeyboards[i][0]) {
+		if (mbstowcs(tempSerial, serial.c_str(), 256) < 1) return false;
-						if (dev->product_id == SupportedKeyboards[i][1]) {
+		wideSerial = wstring(tempSerial);
-							m_vendorID = dev->vendor_id;
+	}
-							m_productID = dev->product_id;
+
-							m_keyboardModel = (KeyboardModel)SupportedKeyboards[i][2];
+	while (dev) {
-							break;
+		for (int i=0; i<(int)SupportedKeyboards.size(); i++) {
-						}
+			if (dev->vendor_id == SupportedKeyboards[i][0]) {
+				if (dev->product_id == SupportedKeyboards[i][1]) {
+					if (!serial.empty() && dev->serial_number != NULL && wideSerial.compare(dev->serial_number) != 0) break; //Serial didn't match
+
+					if (dev->serial_number != NULL) {
+						char buf[256];
+						wcstombs(buf,dev->serial_number,256);
+						currentDevice.serialNumber=string(buf);
 					}
+
+					if (dev->manufacturer_string != NULL)
+					{
+						char buf[256];
+						wcstombs(buf,dev->manufacturer_string,256);
+						currentDevice.manufacturer = string(buf);
+					}
+
+					if (dev->product_string != NULL)
+					{
+						char buf[256];
+						wcstombs(buf,dev->product_string,256);
+						currentDevice.product = string(buf);
+					}
+
+					currentDevice.vendorID = dev->vendor_id;
+					currentDevice.productID = dev->product_id;
+					currentDevice.model = (KeyboardModel)SupportedKeyboards[i][2];
+					break;
 				}
-				if (m_keyboardModel != KeyboardModel::unknown) break;
-				dev = dev->next;
-			}
-			hid_free_enumeration(devs);
-			
-			if (! dev) {
-				cout<<"Keyboard not found"<<endl;
-				m_keyboardModel = KeyboardModel::unknown;
-				hid_exit();
-				return false;
 			}
 		}
-		
+		if (currentDevice.model != KeyboardModel::unknown) break;
-		m_hidHandle = hid_open(m_vendorID, m_productID, NULL);
+		dev = dev->next;
-		
+	}
-		if(m_hidHandle == 0) {
+
-			hid_exit();
+	hid_free_enumeration(devs);
-			return false;
+
-		}
+	if (! dev) {
-		
+		currentDevice.model = KeyboardModel::unknown;
-		m_isOpen = true;
+		hid_exit();
-		return true;
+		return false;
-		
+	}
+
+	if (wideSerial.empty()) m_hidHandle = hid_open(currentDevice.vendorID, currentDevice.productID, NULL);
+	else m_hidHandle = hid_open(currentDevice.vendorID, currentDevice.productID, wideSerial.c_str());
+
+	if(m_hidHandle == 0) {
+		hid_exit();
+		return false;
+	}
+
+	m_isOpen = true;
+	return true;
+
 	#elif defined(libusb)
-		if(libusb_init(&m_ctx) < 0) return false;
+	if (libusb_init(&m_ctx) < 0) return false;
 		
-		if (m_keyboardModel == KeyboardModel::unknown) {
+	libusb_device **devs;
-			libusb_device **devs;
+	libusb_device *dev = NULL;
-			ssize_t cnt = libusb_get_device_list(m_ctx, &devs);
+	ssize_t cnt = libusb_get_device_list(m_ctx, &devs);
-			if(cnt >= 0) {
+	if(cnt >= 0) {
-				for(ssize_t i = 0; i < cnt; i++) {
+		for(ssize_t i = 0; i < cnt; i++) {
-					libusb_device *device = devs[i];
+			libusb_device *device = devs[i];
-					libusb_device_descriptor desc;
+			libusb_device_descriptor desc;
-					libusb_get_device_descriptor(device, &desc);
+			libusb_get_device_descriptor(device, &desc);
+
+			if (vendorID != 0x0 && desc.idVendor != vendorID) continue;
+			else if (productID != 0x0 && desc.idProduct != productID) continue;
+			else if (! serial.empty()) {
+				if (desc.iSerialNumber <= 0) continue; //Device does not populate serial number
+
+				unsigned char buf[256];
+				if (libusb_open(device, &m_hidHandle) != 0){
+					m_hidHandle = NULL;
+					continue;
+				}
+
+				if (libusb_get_string_descriptor_ascii(m_hidHandle, desc.iSerialNumber, buf, 256) >= 1 && serial.compare((char*)buf) == 0) {
+					//Make sure entry is a supported keyboard and get model
 					for (int i=0; i<(int)SupportedKeyboards.size(); i++) {
 						if (desc.idVendor == SupportedKeyboards[i][0]) {
 							if (desc.idProduct == SupportedKeyboards[i][1]) {
-								m_vendorID = desc.idVendor;
+								if (libusb_get_string_descriptor_ascii(m_hidHandle, desc.iManufacturer, buf, 256) >= 1) currentDevice.manufacturer = string((char*)buf);
-								m_productID = desc.idProduct;
+								if (libusb_get_string_descriptor_ascii(m_hidHandle, desc.iProduct, buf, 256) >= 1) currentDevice.product = string((char*)buf);
-								m_keyboardModel = (KeyboardModel)SupportedKeyboards[i][2];
+								currentDevice.serialNumber = serial;
+								currentDevice.vendorID = desc.idVendor;
+								currentDevice.productID = desc.idProduct;
+								currentDevice.model = (KeyboardModel)SupportedKeyboards[i][2];
+
+								dev = device;
+								libusb_close(m_hidHandle);
+								m_hidHandle = NULL;
 								break;
 							}
 						}
 					}
-					if (m_keyboardModel != KeyboardModel::unknown) break;
 				}
-				libusb_free_device_list(devs, 1);
+				else {
+					libusb_close(m_hidHandle);
+					m_hidHandle = NULL;
+					continue; //Serial number set but doesn't match
+				}
 			}
+
+			//For the case where serial is not specified, find first supported device
+			for (int i=0; i<(int)SupportedKeyboards.size(); i++) {
+				if (desc.idVendor == SupportedKeyboards[i][0]) {
+					if (desc.idProduct == SupportedKeyboards[i][1]) {
+						unsigned char buf[256];
+						if (libusb_open(device, &m_hidHandle) != 0){
+							m_hidHandle = NULL;
+							continue;
+						}
+						currentDevice.vendorID = desc.idVendor;
+						currentDevice.productID = desc.idProduct;
+						currentDevice.model = (KeyboardModel)SupportedKeyboards[i][2];
+						if (libusb_get_string_descriptor_ascii(m_hidHandle, desc.iManufacturer, buf, 256) >= 1) currentDevice.manufacturer = string((char*)buf);
+						if (libusb_get_string_descriptor_ascii(m_hidHandle, desc.iProduct, buf, 256) >= 1) currentDevice.product = string((char*)buf);
+						if (libusb_get_string_descriptor_ascii(m_hidHandle, desc.iSerialNumber, buf, 256) >= 1) currentDevice.serialNumber = string((char*)buf);
+
+						libusb_close(m_hidHandle);
+						m_hidHandle=NULL;
+						break;
+					}
+				}
+			}
+			if (currentDevice.model != KeyboardModel::unknown) break;
 		}
+		libusb_free_device_list(devs, 1);
+	}
+
+	if (currentDevice.model == KeyboardModel::unknown) {
+		libusb_exit(m_ctx);
+		m_ctx = NULL;
+		return false;
+	}
 		
-		if (m_keyboardModel == KeyboardModel::unknown) {
+	if (dev == NULL) m_hidHandle = libusb_open_device_with_vid_pid(m_ctx, currentDevice.vendorID, currentDevice.productID);
-			cout<<"Keyboard not found"<<endl;
+	else libusb_open(dev, &m_hidHandle);
+
+	if(m_hidHandle == NULL) {
+		libusb_exit(m_ctx);
+		m_ctx = NULL;
+		return false;
+	}
+		
+	if(libusb_kernel_driver_active(m_hidHandle, 1) == 1) {
+		if(libusb_detach_kernel_driver(m_hidHandle, 1) != 0) {
 			libusb_exit(m_ctx);
 			m_ctx = NULL;
 			return false;
 		}
+		m_isKernellDetached = true;
+	}
 		
-		m_hidHandle = libusb_open_device_with_vid_pid(m_ctx, m_vendorID, m_productID);
+	if(libusb_claim_interface(m_hidHandle, 1) < 0) {
-		if(m_hidHandle == 0) {
+		if(m_isKernellDetached==true) {
-			libusb_exit(m_ctx);
+			libusb_attach_kernel_driver(m_hidHandle, 1);
-			m_ctx = NULL;
+			m_isKernellDetached = false;
-			return false;
 		}
+		libusb_exit(m_ctx);
+		m_ctx = NULL;
+		return false;
+	}
 		
-		if(libusb_kernel_driver_active(m_hidHandle, 1) == 1) {
+	m_isOpen = true;
-			if(libusb_detach_kernel_driver(m_hidHandle, 1) != 0) {
+	return true;
-				libusb_exit(m_ctx);
-				m_ctx = NULL;
-				return false;
-			}
-			m_isKernellDetached = true;
-		}
-		
-		if(libusb_claim_interface(m_hidHandle, 1) < 0) {
-			if(m_isKernellDetached==true) {
-				libusb_attach_kernel_driver(m_hidHandle, 1);
-				m_isKernellDetached = false;
-			}
-			libusb_exit(m_ctx);
-			m_ctx = NULL;
-			return false;
-		}
-		
-		m_isOpen = true;
-		return true;
-		
 	#endif
-	
+
-	return false;
+	return false; //In case neither is defined
+}
+
+LedKeyboard::DeviceInfo LedKeyboard::getCurrentDevice()
+{
+	return currentDevice;
 }
 
 bool LedKeyboard::close() {
@@ -202,11 +329,12 @@ bool LedKeyboard::close() {
 	
 	#if defined(hidapi)
 		hid_close(m_hidHandle);
-		m_hidHandle = 0;
+		m_hidHandle = NULL;
 		hid_exit();
 		return true;
 		
 	#elif defined(libusb)
+		if (m_hidHandle == NULL) return true;
 		if(libusb_release_interface(m_hidHandle, 1) != 0) return false;
 		if(m_isKernellDetached==true) {
 			libusb_attach_kernel_driver(m_hidHandle, 1);
@@ -224,13 +352,12 @@ bool LedKeyboard::close() {
 
 
 LedKeyboard::KeyboardModel LedKeyboard::getKeyboardModel() {
-	return m_keyboardModel;
+	return currentDevice.model;
 }
 
-
 bool LedKeyboard::commit() {
 	byte_buffer_t data;
-	switch (m_keyboardModel) {
+	switch (currentDevice.model) {
 		case KeyboardModel::g213:
 			break; // Keyboard is non-transactional
 		case KeyboardModel::g410:
@@ -268,7 +395,7 @@ bool LedKeyboard::setKeys(KeyValueArray keyValues) {
 	for (uint8_t i = 0; i < keyValues.size(); i++) {
 		switch(static_cast<LedKeyboard::KeyAddressGroup>(static_cast<uint16_t>(keyValues[i].key) >> 8 )) {
 			case LedKeyboard::KeyAddressGroup::logo:
-				switch (m_keyboardModel) {
+				switch (currentDevice.model) {
 					case LedKeyboard::KeyboardModel::g610:
 					case LedKeyboard::KeyboardModel::g810:
 						if (SortedKeys[0].size() <= 1 && keyValues[i].key == LedKeyboard::Key::logo)
@@ -285,7 +412,7 @@ bool LedKeyboard::setKeys(KeyValueArray keyValues) {
 				if (SortedKeys[1].size() <= 5) SortedKeys[1].push_back(keyValues[i]);
 				break;
 			case LedKeyboard::KeyAddressGroup::multimedia:
-				switch (m_keyboardModel) {
+				switch (currentDevice.model) {
 					case LedKeyboard::KeyboardModel::g610:
 					case LedKeyboard::KeyboardModel::g810:
 						if (SortedKeys[2].size() <= 5) SortedKeys[2].push_back(keyValues[i]);
@@ -295,7 +422,7 @@ bool LedKeyboard::setKeys(KeyValueArray keyValues) {
 				}
 				break;
 			case LedKeyboard::KeyAddressGroup::gkeys:
-				switch (m_keyboardModel) {
+				switch (currentDevice.model) {
 					case LedKeyboard::KeyboardModel::g910:
 						if (SortedKeys[3].size() <= 9) SortedKeys[3].push_back(keyValues[i]);
 						break;
@@ -304,7 +431,7 @@ bool LedKeyboard::setKeys(KeyValueArray keyValues) {
 				}
 				break;
 			case LedKeyboard::KeyAddressGroup::keys:
-				switch (m_keyboardModel) {
+				switch (currentDevice.model) {
 					case LedKeyboard::KeyboardModel::g610:
 					case LedKeyboard::KeyboardModel::g810:
 					case LedKeyboard::KeyboardModel::g910:
@@ -433,7 +560,7 @@ bool LedKeyboard::setGroupKeys(KeyGroup keyGroup, LedKeyboard::Color color) {
 bool LedKeyboard::setAllKeys(LedKeyboard::Color color) {
 	KeyValueArray keyValues;
 
-	switch (m_keyboardModel) {
+	switch (currentDevice.model) {
 		case KeyboardModel::g213:
 			for (uint8_t rIndex=0x01; rIndex <= 0x05; rIndex++) if (! setRegion(rIndex, color)) return false;
 			return true;
@@ -461,7 +588,7 @@ bool LedKeyboard::setAllKeys(LedKeyboard::Color color) {
 
 bool LedKeyboard::setMRKey(uint8_t value) {
 	LedKeyboard::byte_buffer_t data;
-	switch (m_keyboardModel) {
+	switch (currentDevice.model) {
 		case KeyboardModel::g910:
 			switch (value) {
 				case 0x00:
@@ -481,7 +608,7 @@ bool LedKeyboard::setMRKey(uint8_t value) {
 
 bool LedKeyboard::setMNKey(uint8_t value) {
 	LedKeyboard::byte_buffer_t data;
-	switch (m_keyboardModel) {
+	switch (currentDevice.model) {
 		case KeyboardModel::g910:
 			switch (value) {
 				case 0x00:
@@ -507,7 +634,7 @@ bool LedKeyboard::setMNKey(uint8_t value) {
 
 bool LedKeyboard::setGKeysMode(uint8_t value) {
 	LedKeyboard::byte_buffer_t data;
-	switch (m_keyboardModel) {
+	switch (currentDevice.model) {
 		case KeyboardModel::g910:
 			switch (value) {
 				case 0x00:
@@ -527,7 +654,7 @@ bool LedKeyboard::setGKeysMode(uint8_t value) {
 
 bool LedKeyboard::setRegion(uint8_t region, LedKeyboard::Color color) {
 	LedKeyboard::byte_buffer_t data;
-	switch (m_keyboardModel) {
+	switch (currentDevice.model) {
 		case KeyboardModel::g213:
 			data = { 0x11, 0xff, 0x0c, 0x3a, region, 0x01, color.red, color.green, color.blue };
 			data.resize(20,0x00);
@@ -542,7 +669,7 @@ bool LedKeyboard::setRegion(uint8_t region, LedKeyboard::Color color) {
 
 bool LedKeyboard::setStartupMode(StartupMode startupMode) {
 	byte_buffer_t data;
-	switch (m_keyboardModel) {
+	switch (currentDevice.model) {
 		case KeyboardModel::g213:
 		case KeyboardModel::g410:
 		case KeyboardModel::g610:
@@ -564,7 +691,7 @@ bool LedKeyboard::setStartupMode(StartupMode startupMode) {
 bool LedKeyboard::setNativeEffect(NativeEffect effect, NativeEffectPart part, uint8_t speed, Color color) {
 	uint8_t protocolByte = 0;
 	
-	switch (m_keyboardModel) {
+	switch (currentDevice.model) {
 		case KeyboardModel::g213:
 			protocolByte = 0x0c;
 			if (part == NativeEffectPart::logo) return false; //Does not have logo component
@@ -686,7 +813,7 @@ bool LedKeyboard::sendDataInternal(byte_buffer_t &data) {
 }
 
 LedKeyboard::byte_buffer_t LedKeyboard::getKeyGroupAddress(LedKeyboard::KeyAddressGroup keyAddressGroup) {
-	switch (m_keyboardModel) {
+	switch (currentDevice.model) {
 		case KeyboardModel::g213:
 		  return {}; // Device doesn't support per-key setting
 		case KeyboardModel::g410:

src/classes/Keyboard.h

@@ -109,7 +109,15 @@ class LedKeyboard {
 			ctrl_right, shift_right, alt_right, win_right
 			
 		};
-		
+
+		typedef struct {
+			uint16_t vendorID = 0x0;
+			uint16_t productID = 0x0;
+			std::string manufacturer = "";
+			std::string product = "";
+			std::string serialNumber = "";
+			KeyboardModel model;
+		} DeviceInfo;
 		
 		struct Color {
 			uint8_t red;
@@ -127,10 +135,12 @@ class LedKeyboard {
 		~LedKeyboard();
 		
 		
-		bool listKeyboards();
+		std::vector<DeviceInfo> listKeyboards();
 		
 		bool isOpen();
 		bool open();
+		bool open(uint16_t vendorID, uint16_t productID, std::string serial);
+		DeviceInfo getCurrentDevice();
 		bool close();
 		
 		KeyboardModel getKeyboardModel();
@@ -152,7 +162,6 @@ class LedKeyboard {
 		bool setNativeEffect(NativeEffect effect, NativeEffectPart part, uint8_t speed, Color color);
 		
 		
-		
 	private:
 		
 		typedef std::vector<unsigned char> byte_buffer_t;
@@ -191,9 +200,7 @@ class LedKeyboard {
 		};
 		
 		bool m_isOpen = false;
-		uint16_t m_vendorID = 0;
+		DeviceInfo currentDevice;
-		uint16_t m_productID = 0;
-		KeyboardModel m_keyboardModel = KeyboardModel::unknown;
 		
 		#if defined(hidapi)
 			hid_device *m_hidHandle;