Link.cpp

/*
 * This file is part of OpenCorsairLink.
 * Copyright (C) 2014  Sean Nelson <audiohacked@gmail.com>

 * OpenCorsairLink is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 2 of the License, or
 * any later version.

 * OpenCorsairLink is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.

 * You should have received a copy of the GNU General Public License
 * along with OpenCorsairLink.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "Proto.h"
#include "Link.h"
#include "Fan.h"


CorsairLink::CorsairLink() {
	handle = NULL;
	CommandId = 0x81;
	max_ms_read_wait = 5000;
}

int CorsairLink::Initialize()
{
	if(handle == NULL)
	{
		if (hid_init())
			return 0;

		// Set up the command buffer.
		//memset(buf,0x00,sizeof(buf));
		//buf[0] = 0x01;
		//buf[1] = 0x81;

		// Open the device using the VID, PID,
		// and optionally the Serial number.
		// open Corsair H80i, H100i, or H110i cooler
		handle = hid_open(0x1b1c, 0x0c04, NULL);
		if (!handle)
		{
			fprintf(stderr, "Error: Unable to open Corsair H80i, H100i or H110i CPU Cooler\n");
			return 0;
		}
		hid_set_nonblocking(handle, 1);

		deviceId = this->GetDeviceId();
		if ((deviceId != 0x3b) && (deviceId != 0x3c) && (deviceId != 0x41))
		{
			fprintf(stderr, "Device ID: %2x mismatch. Not Corsair H80i, H100i or H110i CPU Cooler\n", deviceId );
			this->Close();
			return 0;
		}
	} else {
		fprintf(stderr, "Cannot initialize twice\n" );
		return 0;
	}
	return 1;
}

int CorsairLink::GetDeviceId(void)
{
	memset(buf,0,sizeof(buf));

	// Read Device ID: 0x3b = H80i. 0x3c = H100i. 0x41 = H110i
	buf[0] = 0x03; // Length
	buf[1] = this->CommandId++; // Command ID
	buf[2] = ReadOneByte; // Command Opcode
	buf[3] = DeviceID; // Command data...
	buf[4] = 0x00;

	int res = hid_write(handle, buf, 17);
	if (res < 0) {
		fprintf(stderr, "Error: Unable to write() %s\n", (char*)hid_error(handle) );
	}

	hid_read_wrapper(handle, buf);
	if (res < 0) {
		fprintf(stderr, "Error: Unable to read() %s\n", (char*)hid_error(handle) );
	}

	return buf[2];
}

int CorsairLink::GetFirmwareVersion()
{
	memset(buf,0,sizeof(buf));

	// Read Device ID: 0x3b = H80i. 0x3c = H100i. 0x41 = H110i
	buf[0] = 0x03; // Length
	buf[1] = this->CommandId++; // Command ID
	buf[2] = ReadTwoBytes; // Command Opcode
	buf[3] = FirmwareID; // Command data...
	buf[4] = 0x00;

	int res = hid_write(handle, buf, 17);
	if (res < 0) {
		fprintf(stderr, "Error: Unable to write() %s\n", (char*)hid_error(handle) );
	}

	hid_read_wrapper(handle, buf);
	if (res < 0) {
		fprintf(stderr, "Error: Unable to read() %s\n", (char*)hid_error(handle) );
	}
	int firmware = buf[3]<<8;
	firmware += buf[2];
	return firmware;
}

int CorsairLink::GetProductName(char *ostring)
{
	memset(buf,0,sizeof(buf));

	// Read Device ID: 0x3b = H80i. 0x3c = H100i. 0x41 = H110i
	buf[0] = 0x04; // Length
	buf[1] = this->CommandId++; // Command ID
	buf[2] = ReadThreeBytes; // Command Opcode
	buf[3] = ProductName; // Command data...
	buf[4] = 0x08;

	int res = hid_write(handle, buf, 8);
	if (res < 0) {
		fprintf(stderr, "Error: Unable to write() %s\n", (char*)hid_error(handle) );
	}

	hid_read_wrapper(handle, buf);
	if (res < 0) {
		fprintf(stderr, "Error: Unable to read() %s\n", (char*)hid_error(handle) );
	}
	memcpy(ostring, buf + 3, 8);
	return 0;
}

int CorsairLink::GetDeviceStatus()
{
	memset(buf,0,sizeof(buf));

	buf[0] = 0x03; // Length
	buf[1] = this->CommandId++; // Command ID
	buf[2] = ReadOneByte; // Command Opcode
	buf[3] = Status; // Command data...
	buf[4] = 0x00;

	int res = hid_write(handle, buf, 17);
	if (res < 0) {
		fprintf(stderr, "Error: Unable to write() %s\n", (char*)hid_error(handle) );
	}

	hid_read_wrapper(handle, buf);
	if (res < 0) {
		fprintf(stderr, "Error: Unable to read() %s\n", (char*)hid_error(handle) );
	}

	return buf[2];
}

char* CorsairLink::_GetManufacturer()
{
	char *str;
	char wstr[MAX_STR];
	wstr[0] = 0x0000;

	int res = hid_get_manufacturer_string(handle, (wchar_t*)wstr, MAX_STR);
	if (res < 0)
		fprintf(stderr, "Unable to read manufacturer string\n");

	str = wstr;
	return str;
}

char* CorsairLink::_GetProduct()
{
	char* str;
	char wstr[MAX_STR];
	wstr[0] = 0x0000;
	
	int res = hid_get_product_string(handle, (wchar_t*)wstr, MAX_STR);
	if (res < 0)
		fprintf(stderr, "Unable to read product string\n");

	str = wstr;
	return str;
}

int CorsairLink::hid_read_wrapper (hid_device *handle, unsigned char *buf)
{
	// Read requested state. hid_read() has been set to be
	// non-blocking by the call to hid_set_nonblocking() above.
	// This loop demonstrates the non-blocking nature of hid_read().
	int res = 0;
	int sleepTotal = 0;
	while (res == 0 && sleepTotal < this->max_ms_read_wait)
	{
		res = hid_read(handle, buf, sizeof(buf));
		if (res < 0)
			fprintf(stderr, "Unable to read()\n");
		
		this->sleep(100);
		sleepTotal += 100;
	}
	if(sleepTotal == this->max_ms_read_wait)
	{
		res = 0;
	}

#if DEBUG
	int i = 0;
	for (i = 0; i < sizeof(buf); i++)
	{
		fprintf(stdout, "Debug-hid_read_wrapper: %02X\n", buf[i]);
	}
#endif
	return 1;
}

int CorsairLink::hid_wrapper (hid_device *handle, unsigned char *buf, size_t buf_size) {
	int res = hid_write(handle, buf, buf_size);
	if (res < 0) {
		fprintf(stderr, "Error: Unable to write() %s\n", (char*)hid_error(handle) );
		//return -1;
	}
	res = this->hid_read_wrapper(handle, buf);
	if (res < 0) {
		fprintf(stderr, "Error: Unable to read() %s\n", (char*)hid_error(handle) );
		//return -1;
	}
	return res;	
}

void CorsairLink::sleep(int ms)
{
	#ifdef WIN32
	Sleep(ms);
	#else
	usleep(ms*1000);
	#endif
}

void CorsairLink::Close()
{
	if(handle != NULL)
	{
		hid_close(handle);
		hid_exit();
		handle = NULL;
	}
}

CorsairLink::~CorsairLink()
{
	this->Close();
//	if(fans != NULL) {
//		free(fans);
//	}
}