X-Git-Url: https://the.earth.li/gitweb/?a=blobdiff_plain;f=main.c;fp=main.c;h=920db3ed8101ce6741fa8a8078eca09bda252dd8;hb=40576ee3a501070a9afd7392a53eab4aa9979e70;hp=0000000000000000000000000000000000000000;hpb=ec3e5e54e0f8b49eb8e3a1c2970d9ea819b392fc;p=usb-relay-firmware.git

diff --git a/main.c b/main.c
new file mode 100644
index 0000000..920db3e
--- /dev/null
+++ b/main.c
@@ -0,0 +1,166 @@
+/*
+ * Basic firmware for www.dcttech.com 1 port USB Relay
+ *
+ * Copyright 2016 Jonathan McDowell <noodles@earth.li>
+ *
+ * This program 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 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+#include <avr/eeprom.h>
+#include <avr/interrupt.h>
+#include <avr/io.h>
+#include <avr/wdt.h>
+#include <util/delay.h>
+
+#include <avr/pgmspace.h>
+#include "usbdrv.h"
+#include "libs-device/osccal.h"
+
+#define RELAY_BIT 8 /* Bit 3 on port B */
+#define CMD_ALL_ON 0xfe
+#define CMD_ALL_OFF 0xfc
+#define CMD_ON 0xff
+#define CMD_OFF 0xfd
+#define CMD_SET_SERIAL 0xfa
+
+uchar serno_read = 0;
+uchar serno[6];
+
+PROGMEM const char usbHidReportDescriptor[22] = {
+	0x06, 0x00, 0xff,		/* USAGE PAGE (Generic Desktop) */
+	0x09, 0x01,			/* USAGE (Vendor Usage 1) */
+	0xa1, 0x01,			/* COLLECTION (Application) */
+	0x15, 0x00,			/*   LOGICAL_MINIMUM (0) */
+	0x26, 0xff, 0x00,		/*   LOGICAL_MAXIMUM (255) */
+	0x75, 0x08,			/*   REPORT_SIZE (8) */
+	0x95, 0x08,			/*   REPORT_COUNT (8) */
+	0x09, 0x00,			/*   USAGE (Undefined) */
+	0xb2, 0x02, 0x01,		/*   FEATURE (Data, Var, Abs, Buf) */
+	0xc0				/* END_COLLECTION */
+};
+
+void fetch_serno(void)
+{
+	if (!serno_read) {
+		eeprom_read_block(serno, 0, 6);
+		if (serno[0] == 0xff) {
+			/* If the EEPROM is blank, return a default serial # */
+			serno[0] = 'U';
+			serno[1] = 'N';
+			serno[2] = 'S';
+			serno[3] = 'E';
+			serno[4] = 'T';
+			serno[5] = 0;
+		}
+		serno_read = 1;
+	}
+}
+
+void update_serno(uchar *buf, uchar len)
+{
+	uchar i;
+
+	/*
+	 * I have no idea why this gets stored 3 times, but the original
+	 * firmware does it.
+	 */
+	eeprom_write_block(buf, (void *) 0x00, len);
+	eeprom_write_block(buf, (void *) 0x40, len);
+	eeprom_write_block(buf, (void *) 0x80, len);
+
+	for (i = 0; i < 6; i++) {
+		serno[i] = buf[i];
+	}
+}
+
+usbMsgLen_t usbFunctionSetup(uchar data[8])
+{
+	usbRequest_t *rq = (void *) data;
+
+	if ((rq->bmRequestType & USBRQ_TYPE_MASK) == USBRQ_TYPE_CLASS) {
+		if ((rq->bRequest == USBRQ_HID_GET_REPORT) ||
+				(rq->bRequest == USBRQ_HID_SET_REPORT)) {
+			return 0xFF;
+		}
+	}
+
+	return 0;
+}
+
+uchar usbFunctionRead(uchar *data, uchar len)
+{
+	uchar i;
+
+	if (len != 0) {
+		fetch_serno();
+		for (i = 0; i < 6; i++) {
+			data[i] = serno[i];
+		}
+		data[6] = data[7] = 0;
+		if (PORTB & RELAY_BIT) {
+			data[7] = 1;
+		}
+		return len;
+	}
+
+	return 0;
+}
+
+uchar usbFunctionWrite(uchar *data, uchar len)
+{
+	if (data[0] == CMD_ALL_ON) {
+		PORTB |= RELAY_BIT;
+	} else if (data[0] == CMD_ALL_OFF) {
+		PORTB &= ~(RELAY_BIT);
+	} else if (data[0] == CMD_ON) {
+		if (data[1] == 1) {
+			PORTB |= RELAY_BIT;
+		}
+	} else if (data[0] == CMD_OFF) {
+		if (data[1] == 1) {
+			PORTB &= ~(RELAY_BIT);
+		}
+	} else if (data[0] == CMD_SET_SERIAL) {
+		update_serno(&data[1], 6);
+	}
+
+	return len;
+}
+
+int __attribute__((noreturn)) main(void)
+{
+	unsigned char i;
+
+	wdt_enable(WDTO_1S);
+
+	usbInit();
+	usbDeviceDisconnect();
+
+	i = 0;
+	while (--i) {
+		wdt_reset();
+		_delay_ms(1);
+	}
+
+	usbDeviceConnect();
+
+	/* Set the relay bit to output mode */
+	DDRB |= RELAY_BIT;
+
+	sei(); /* We're ready to go; enable interrupts */
+
+	while (1) {
+		wdt_reset();
+		usbPoll();
+	}
+}