X-Git-Url: https://git.zerfleddert.de/cgi-bin/gitweb.cgi/hmcfgusb/blobdiff_plain/469ea397d6f5c4a808fb4d08d8768680d2074007..86fad1045193837717cb46380a2678ab1e6d96d6:/flash-ota.c

diff --git a/flash-ota.c b/flash-ota.c
index b9b96cc..2625209 100644
--- a/flash-ota.c
+++ b/flash-ota.c
@@ -1,6 +1,6 @@
 /* flasher for HomeMatic-devices supporting OTA updates
  *
- * Copyright (c) 2014 Michael Gernoth <michael@gernoth.net>
+ * Copyright (c) 2014-16 Michael Gernoth <michael@gernoth.net>
  *
  * Permission is hereby granted, free of charge, to any person obtaining a copy
  * of this software and associated documentation files (the "Software"), to
@@ -41,6 +41,7 @@
 #include "version.h"
 #include "hmcfgusb.h"
 #include "culfw.h"
+#include "hmuartlgw.h"
 #include "util.h"
 
 #define MAX_RETRIES		5
@@ -51,32 +52,35 @@ extern char *optarg;
 
 uint32_t hmid = 0;
 uint32_t my_hmid = 0;
+uint8_t key[16] = {0};
+int32_t kNo = -1;
 
 /* Maximum payloadlen supported by IO */
 uint32_t max_payloadlen = NORMAL_MAX_PAYLOAD;
 
-enum device_type {
-	DEVICE_TYPE_HMCFGUSB,
-	DEVICE_TYPE_CULFW,
-};
-
-struct ota_dev {
-	int type;
-	struct hmcfgusb_dev *hmcfgusb;
-	struct culfw_dev *culfw;
-};
-
 enum message_type {
 	MESSAGE_TYPE_E = 1,
 	MESSAGE_TYPE_R = 2,
 };
 
+enum hmuartlgw_state {
+	HMUARTLGW_STATE_GET_HMID,
+	HMUARTLGW_STATE_GET_FIRMWARE,
+	HMUARTLGW_STATE_GET_CREDITS,
+	HMUARTLGW_STATE_DONE,
+	HMUARTLGW_STATE_WAIT_APP,
+	HMUARTLGW_STATE_ACK_APP,
+};
+
 struct recv_data {
 	uint8_t message[64];
 	enum message_type message_type;
 	uint16_t status;
 	int speed;
 	uint16_t version;
+	uint8_t credits;
+	enum hmuartlgw_state uartlgw_state;
+	uint8_t uartlgw_version[3];
 };
 
 static int parse_hmcfgusb(uint8_t *buf, int buf_len, void *data)
@@ -108,6 +112,7 @@ static int parse_hmcfgusb(uint8_t *buf, int buf_len, void *data)
 			break;
 		case 'H':
 			rdata->version = (buf[11] << 8) | buf[12];
+			rdata->credits = buf[36];
 			my_hmid = (buf[0x1b] << 16) | (buf[0x1c] << 8) | buf[0x1d];
 			break;
 		default:
@@ -173,8 +178,25 @@ static int parse_culfw(uint8_t *buf, int buf_len, void *data)
 				*e = '\0';
 				v = atoi(s);
 				rdata->version |= v;
+
+				s = e + 1;
+				e = strchr(s, ' ');
+				if (!e) {
+					break;
+				}
+				*e = '\0';
+				if (!strcmp(s, "a-culfw")) {
+					rdata->version = 0xffff;
+				}
 			}
 			break;
+		case 'E':
+			{
+				if (!strncmp((char*)buf, "ERR:CCA", 7)) {
+					fprintf(stderr, "CCA didn't complete, too much traffic\n");
+				}
+				break;
+			}
 		default:
 			fprintf(stderr, "Unknown response from CUL: %s", buf);
 			return 0;
@@ -184,7 +206,65 @@ static int parse_culfw(uint8_t *buf, int buf_len, void *data)
 	return 1;
 }
 
-int send_hm_message(struct ota_dev *dev, struct recv_data *rdata, uint8_t *msg)
+static int parse_hmuartlgw(enum hmuartlgw_dst dst, uint8_t *buf, int buf_len, void *data)
+{
+	struct recv_data *rdata = data;
+
+	if (dst == HMUARTLGW_OS) {
+		switch (rdata->uartlgw_state) {
+			case HMUARTLGW_STATE_GET_FIRMWARE:
+				if (buf[0] == HMUARTLGW_OS_ACK) {
+					rdata->uartlgw_version[0] = buf[5];
+					rdata->uartlgw_version[1] = buf[6];
+					rdata->uartlgw_version[2] = buf[7];
+					rdata->uartlgw_state = HMUARTLGW_STATE_DONE;
+				}
+				break;
+			case HMUARTLGW_STATE_GET_CREDITS:
+				if (buf[0] == HMUARTLGW_OS_ACK) {
+					rdata->credits = buf[2] / 2;
+					rdata->uartlgw_state = HMUARTLGW_STATE_DONE;
+				}
+				break;
+			default:
+				break;
+		}
+		return 0;
+	}
+
+	switch(buf[0]) {
+		case HMUARTLGW_APP_ACK:
+			if (rdata->uartlgw_state == HMUARTLGW_STATE_GET_HMID) {
+				my_hmid = (buf[4] << 16) | (buf[5] << 8) | buf[6];
+			}
+
+			rdata->status = buf[1];
+			rdata->message_type = MESSAGE_TYPE_R;
+			rdata->uartlgw_state = HMUARTLGW_STATE_ACK_APP;
+#if 0
+			hexdump(buf, buf_len, "ACK Status: ");
+#endif
+
+			break;
+		case HMUARTLGW_APP_RECV:
+			if ((!hmid) ||
+			    ((buf[7] == ((hmid >> 16) & 0xff)) &&
+			    (buf[8] == ((hmid >> 8) & 0xff)) &&
+			    (buf[9] == (hmid & 0xff)))) {
+				memset(rdata->message, 0, sizeof(rdata->message));
+				memcpy(rdata->message + 1, buf + 4, buf_len - 4);
+				rdata->message[LEN] = buf_len - 4;
+				rdata->message_type = MESSAGE_TYPE_E;
+			}
+			break;
+		default:
+			break;
+	}
+
+	return 1;
+}
+
+int send_hm_message(struct hm_dev *dev, struct recv_data *rdata, uint8_t *msg)
 {
 	static uint32_t id = 1;
 	struct timeval tv;
@@ -218,14 +298,16 @@ int send_hm_message(struct ota_dev *dev, struct recv_data *rdata, uint8_t *msg)
 
 			while (1) {
 				if (rdata->message_type == MESSAGE_TYPE_R) {
-					if (((rdata->status & 0xff) == 0x01) ||
-					    ((rdata->status & 0xff) == 0x02)) {
+					if (((rdata->status & 0xdf) == 0x01) ||
+					    ((rdata->status & 0xdf) == 0x02)) {
 						break;
 					} else {
 						if ((rdata->status & 0xff00) == 0x0400) {
 							fprintf(stderr, "\nOut of credits!\n");
 						} else if ((rdata->status & 0xff) == 0x08) {
 							fprintf(stderr, "\nMissing ACK!\n");
+						} else if ((rdata->status & 0xff) == 0x30) {
+							fprintf(stderr, "\nUnknown AES-key requested!\n");
 						} else {
 							fprintf(stderr, "\nInvalid status: %04x\n", rdata->status);
 						}
@@ -264,7 +346,7 @@ int send_hm_message(struct ota_dev *dev, struct recv_data *rdata, uint8_t *msg)
 				}
 
 				if (msg[CTL] & 0x20) {
-					int cnt = 3;
+					int cnt = 5;
 					int pfd;
 					do {
 						errno = 0;
@@ -276,7 +358,48 @@ int send_hm_message(struct ota_dev *dev, struct recv_data *rdata, uint8_t *msg)
 							}
 						}
 						if (rdata->message_type == MESSAGE_TYPE_E) {
-							break;
+							if (rdata->message[TYPE] == 0x02) {
+								if (rdata->message[PAYLOAD] == 0x04) {
+									int32_t req_kNo;
+									uint8_t challenge[6];
+									uint8_t respbuf[16];
+									uint8_t *resp;
+
+									req_kNo = rdata->message[rdata->message[LEN]] / 2;
+									memcpy(challenge, &(rdata->message[PAYLOAD+1]), 6);
+
+									if (req_kNo != kNo) {
+										fprintf(stderr, "AES request for unknown key %d!\n", req_kNo);
+									} else {
+										resp = hm_sign(key, challenge, msg, NULL, respbuf);
+										if (resp) {
+											uint8_t rbuf[64];
+
+											memset(rbuf, 0, sizeof(rbuf));
+											rbuf[MSGID] = rdata->message[MSGID];
+											rbuf[CTL] = rdata->message[CTL];
+											rbuf[TYPE] = 0x03;
+											SET_SRC(rbuf, DST(rdata->message));
+											SET_DST(rbuf, SRC(rdata->message));
+											memcpy(&(rbuf[PAYLOAD]), resp, 16);
+											SET_LEN_FROM_PAYLOADLEN(rbuf, 16);
+
+											usleep(110000); /* Determined by a fair dice roll */
+											return send_hm_message(dev, rdata, rbuf);
+										}
+									}
+								} else if (rdata->message[PAYLOAD] >= 0x80 && rdata->message[PAYLOAD] <= 0x8f) {
+									fprintf(stderr, "NACK\n");
+								} else {	/* ACK or ACKinfo */
+									break;
+								}
+							} else {
+								fprintf(stderr, "Unexpected message received: ");
+								for (i = 0; i < rdata->message[LEN]; i++) {
+									fprintf(stderr, "%02x", rdata->message[i+1]);
+								}
+								fprintf(stderr, "\n");
+							}
 						}
 					} while(cnt--);
 
@@ -287,13 +410,52 @@ int send_hm_message(struct ota_dev *dev, struct recv_data *rdata, uint8_t *msg)
 				}
 			}
 			break;
+		case DEVICE_TYPE_HMUARTLGW:
+			memset(out, 0, sizeof(out));
+
+			out[0] = HMUARTLGW_APP_SEND;
+			out[1] = 0x00;
+			out[2] = 0x00;
+			out[3] = (msg[CTL] & 0x10) ? 0x01 : 0x00; /* Burst?! */
+			memcpy(&out[4], &msg[1], msg[0]);
+
+			memset(rdata, 0, sizeof(struct recv_data));
+			hmuartlgw_send(dev->hmuartlgw, out, msg[0] + 4, HMUARTLGW_APP);
+
+			while (1) {
+				if (rdata->message_type == MESSAGE_TYPE_R) {
+					if ((rdata->status == 0x02) ||
+					    (rdata->status == 0x03) ||
+					    (rdata->status == 0x0c)) {
+						break;
+					} else {
+						if (rdata->status == 0x0d) {
+							fprintf(stderr, "\nAES handshake failed!\n");
+						} else if (rdata->status == 0x04 || rdata->status == 0x06) {
+							fprintf(stderr, "\nMissing ACK!\n");
+						} else {
+							fprintf(stderr, "\nInvalid status: %04x\n", rdata->status);
+						}
+						return 0;
+					}
+				}
+				errno = 0;
+				pfd = hmuartlgw_poll(dev->hmuartlgw, 1000);
+				if ((pfd < 0) && errno) {
+					if (errno != ETIMEDOUT) {
+						perror("\n\nhmcfgusb_poll");
+						exit(EXIT_FAILURE);
+					}
+				}
+			}
+			break;
 	}
 
 	id++;
 	return 1;
 }
 
-static int switch_speed(struct ota_dev *dev, struct recv_data *rdata, uint8_t speed)
+static int switch_speed(struct hm_dev *dev, struct recv_data *rdata, uint8_t speed)
 {
 	uint8_t out[0x40];
 	int pfd;
@@ -328,6 +490,17 @@ static int switch_speed(struct ota_dev *dev, struct recv_data *rdata, uint8_t sp
 				return culfw_send(dev->culfw, "Ar\r\n", 4);
 			}
 			break;
+		case DEVICE_TYPE_HMUARTLGW:
+			if (speed == 100) {
+				out[0] = HMUARTLGW_OS_UPDATE_MODE;
+				out[1] = 0xe9;
+				out[2] = 0xca;
+				hmuartlgw_send(dev->hmuartlgw, out, 3, HMUARTLGW_OS);
+			} else {
+				out[0] = HMUARTLGW_OS_NORMAL_MODE;
+				hmuartlgw_send(dev->hmuartlgw, out, 1, HMUARTLGW_OS);
+			}
+			break;
 	}
 
 	return 1;
@@ -338,12 +511,18 @@ void flash_ota_syntax(char *prog)
 	fprintf(stderr, "Syntax: %s parameters options\n\n", prog);
 	fprintf(stderr, "Mandatory parameters:\n");
 	fprintf(stderr, "\t-f firmware.eq3\tfirmware file to flash\n");
-	fprintf(stderr, "\t-s SERIAL\tserial of device to flash\n");
-	fprintf(stderr, "\nPossible options:\n");
+	fprintf(stderr, "\t-s SERIAL\tserial of device to flash (optional when using -D)\n");
+	fprintf(stderr, "\nOptional parameters:\n");
 	fprintf(stderr, "\t-c device\tenable CUL-mode with CUL at path \"device\"\n");
 	fprintf(stderr, "\t-b bps\t\tuse CUL with speed \"bps\" (default: %u)\n", DEFAULT_CUL_BPS);
 	fprintf(stderr, "\t-l\t\tlower payloadlen (required for devices with little RAM, e.g. CUL v2 and CUL v4)\n");
+	fprintf(stderr, "\t-S serial\tuse HM-CFG-USB with given serial\n");
+	fprintf(stderr, "\t-U device\tuse HM-MOD-UART on given device\n");
 	fprintf(stderr, "\t-h\t\tthis help\n");
+	fprintf(stderr, "\nOptional parameters for automatically sending device to bootloader\n");
+	fprintf(stderr, "\t-C\t\tHMID of central (3 hex-bytes, no prefix, e.g. ABCDEF)\n");
+	fprintf(stderr, "\t-D\t\tHMID of device (3 hex-bytes, no prefix, e.g. 123456)\n");
+	fprintf(stderr, "\t-K\t\tKNO:KEY AES key-number and key (hex) separated by colon (Fhem hmKey attribute)\n");
 }
 
 int main(int argc, char **argv)
@@ -353,14 +532,17 @@ int main(int argc, char **argv)
 	char *fw_file = NULL;
 	char *serial = NULL;
 	char *culfw_dev = NULL;
+	char *endptr = NULL;
 	unsigned int bps = DEFAULT_CUL_BPS;
-	struct ota_dev dev;
+	struct hm_dev dev;
 	struct recv_data rdata;
 	uint8_t out[0x40];
 	uint8_t *pos;
 	uint8_t msgid = 0x1;
 	uint16_t len;
 	struct firmware *fw;
+	char *hmcfgusb_serial = NULL;
+	char *uart = NULL;
 	int block;
 	int pfd;
 	int debug = 0;
@@ -372,7 +554,7 @@ int main(int argc, char **argv)
 
 	printf("HomeMatic OTA flasher version " VERSION "\n\n");
 
-	while((opt = getopt(argc, argv, "b:c:f:hls:")) != -1) {
+	while((opt = getopt(argc, argv, "b:c:f:hls:C:D:K:S:U:")) != -1) {
 		switch (opt) {
 			case 'b':
 				bps = atoi(optarg);
@@ -390,6 +572,48 @@ int main(int argc, char **argv)
 			case 's':
 				serial = optarg;
 				break;
+			case 'C':
+				my_hmid = strtoul(optarg, &endptr, 16);
+				if (*endptr != '\0') {
+					fprintf(stderr, "Invalid central HMID!\n\n");
+					flash_ota_syntax(argv[0]);
+					exit(EXIT_FAILURE);
+				}
+				break;
+			case 'D':
+				hmid = strtoul(optarg, &endptr, 16);
+				if (*endptr != '\0') {
+					fprintf(stderr, "Invalid device HMID!\n\n");
+					flash_ota_syntax(argv[0]);
+					exit(EXIT_FAILURE);
+				}
+				break;
+			case 'K':
+				kNo = strtoul(optarg, &endptr, 10);
+				if (*endptr != ':') {
+					fprintf(stderr, "Invalid key number!\n\n");
+					flash_ota_syntax(argv[0]);
+					exit(EXIT_FAILURE);
+				}
+				endptr++;
+				for (cnt = 0; cnt < 16; cnt++) {
+					if (*endptr == '\0' || *(endptr+1) == '\0' ||
+					    !validate_nibble(*endptr) ||
+					    !validate_nibble(*(endptr+1))) {
+						fprintf(stderr, "Invalid key!\n\n");
+						flash_ota_syntax(argv[0]);
+						exit(EXIT_FAILURE);
+					}
+					key[cnt] = ascii_to_nibble(*endptr) << 4 | ascii_to_nibble(*(endptr+1));
+					endptr += 2;
+				}
+				break;
+			case 'S':
+				hmcfgusb_serial = optarg;
+				break;
+			case 'U':
+				uart = optarg;
+				break;
 			case 'h':
 			case ':':
 			case '?':
@@ -401,7 +625,7 @@ int main(int argc, char **argv)
 		}
 	}
 
-	if (!fw_file || !serial) {
+	if (!fw_file || (!serial && !hmid)) {
 		flash_ota_syntax(argv[0]);
 		exit(EXIT_FAILURE);
 	}
@@ -411,7 +635,7 @@ int main(int argc, char **argv)
 		exit(EXIT_FAILURE);
 
 	memset(&rdata, 0, sizeof(rdata));
-	memset(&dev, 0, sizeof(struct ota_dev));
+	memset(&dev, 0, sizeof(struct hm_dev));
 
 	if (culfw_dev) {
 		printf("Opening culfw-device at path %s with speed %u\n", culfw_dev, bps);
@@ -441,54 +665,119 @@ int main(int argc, char **argv)
 				break;
 		}
 
-		printf("culfw-device firmware version: %u.%02u\n", 
-			(rdata.version >> 8) & 0xff,
-			rdata.version & 0xff);
+		printf("culfw-device firmware version: ");
+		if (rdata.version != 0xffff) {
+			printf("%u.%02u\n",
+				(rdata.version >> 8) & 0xff,
+				rdata.version & 0xff);
+		} else {
+			printf("a-culfw\n");
+		}
 
 		if (rdata.version < 0x013a) {
 			fprintf(stderr, "\nThis version does _not_ support firmware upgrade mode, you need at least 1.58!\n");
 			exit(EXIT_FAILURE);
 		}
-	} else {
-		hmcfgusb_set_debug(debug);
+	} else if (uart) {
+		uint32_t new_hmid = my_hmid;
 
-		dev.hmcfgusb = hmcfgusb_init(parse_hmcfgusb, &rdata);
-		if (!dev.hmcfgusb) {
-			fprintf(stderr, "Can't initialize HM-CFG-USB\n");
+		hmuartlgw_set_debug(debug);
+
+		dev.hmuartlgw = hmuart_init(uart, parse_hmuartlgw, &rdata);
+		if (!dev.hmuartlgw) {
+			fprintf(stderr, "Can't initialize HM-MOD-UART\n");
 			exit(EXIT_FAILURE);
 		}
-		dev.type = DEVICE_TYPE_HMCFGUSB;
+		dev.type = DEVICE_TYPE_HMUARTLGW;
 
-		printf("\nRebooting HM-CFG-USB to avoid running out of credits\n\n");
+		out[0] = HMUARTLGW_APP_GET_HMID;
+		do {
+			rdata.uartlgw_state = HMUARTLGW_STATE_GET_HMID;
+			hmuartlgw_send(dev.hmuartlgw, out, 1, HMUARTLGW_APP);
+			do { hmuartlgw_poll(dev.hmuartlgw, 500); } while (rdata.uartlgw_state != HMUARTLGW_STATE_ACK_APP);
+		} while (rdata.status == 0x08);
 
-		if (!dev.hmcfgusb->bootloader) {
-			printf("HM-CFG-USB not in bootloader mode, entering bootloader.\n");
-			printf("Waiting for device to reappear...\n");
+		out[0] = HMUARTLGW_OS_GET_FIRMWARE;
+		do {
+			rdata.uartlgw_state = HMUARTLGW_STATE_GET_FIRMWARE;
+			hmuartlgw_send(dev.hmuartlgw, out, 1, HMUARTLGW_OS);
+			do { hmuartlgw_poll(dev.hmuartlgw, 500); } while (rdata.uartlgw_state != HMUARTLGW_STATE_DONE);
+		} while (rdata.status == 0x08);
 
+		out[0] = HMUARTLGW_OS_GET_CREDITS;
+		do {
+			rdata.uartlgw_state = HMUARTLGW_STATE_GET_CREDITS;
+			hmuartlgw_send(dev.hmuartlgw, out, 1, HMUARTLGW_OS);
+			do { hmuartlgw_poll(dev.hmuartlgw, 500); } while (rdata.uartlgw_state != HMUARTLGW_STATE_DONE);
+		} while (rdata.status == 0x08);
+
+		printf("HM-MOD-UART firmware version: %u.%u.%u, used credits: %u%%\n",
+			rdata.uartlgw_version[0],
+			rdata.uartlgw_version[1],
+			rdata.uartlgw_version[2],
+			rdata.credits);
+
+		if (rdata.credits >= 40) {
+			printf("\nRebooting HM-MOD-UART to avoid running out of credits\n");
+
+			hmuartlgw_enter_bootloader(dev.hmuartlgw);
+			hmuartlgw_enter_app(dev.hmuartlgw);
+		}
+
+		printf("\nHM-MOD-UART opened\n\n");
+
+		if (new_hmid && (my_hmid != new_hmid)) {
+			printf("Changing hmid from %06x to %06x\n", my_hmid, new_hmid);
+
+			out[0] = HMUARTLGW_APP_SET_HMID;
+			out[1] = (new_hmid >> 16) & 0xff;
+			out[2] = (new_hmid >> 8) & 0xff;
+			out[3] = new_hmid & 0xff;
 			do {
-				if (dev.hmcfgusb) {
-					if (!dev.hmcfgusb->bootloader)
-						hmcfgusb_enter_bootloader(dev.hmcfgusb);
-					hmcfgusb_close(dev.hmcfgusb);
-				}
-				sleep(1);
-			} while (((dev.hmcfgusb = hmcfgusb_init(parse_hmcfgusb, &rdata)) == NULL) || (!dev.hmcfgusb->bootloader));
+				rdata.uartlgw_state = HMUARTLGW_STATE_WAIT_APP;
+				hmuartlgw_send(dev.hmuartlgw, out, 4, HMUARTLGW_APP);
+				do { hmuartlgw_poll(dev.hmuartlgw, 500); } while (rdata.uartlgw_state != HMUARTLGW_STATE_ACK_APP);
+			} while (rdata.status == 0x08);
+
+			my_hmid = new_hmid;
 		}
 
-		if (dev.hmcfgusb->bootloader) {
-			printf("HM-CFG-USB in bootloader mode, rebooting\n");
+		if (kNo > 0) {
+			printf("Setting AES-key\n");
+
+			memset(out, 0, sizeof(out));
+			out[0] = HMUARTLGW_APP_SET_CURRENT_KEY;
+			memcpy(&(out[1]), key, 16);
+			out[17] = kNo;
 
 			do {
-				if (dev.hmcfgusb) {
-					if (dev.hmcfgusb->bootloader)
-						hmcfgusb_leave_bootloader(dev.hmcfgusb);
-					hmcfgusb_close(dev.hmcfgusb);
-				}
-				sleep(1);
-			} while (((dev.hmcfgusb = hmcfgusb_init(parse_hmcfgusb, &rdata)) == NULL) || (dev.hmcfgusb->bootloader));
+				rdata.uartlgw_state = HMUARTLGW_STATE_WAIT_APP;
+				hmuartlgw_send(dev.hmuartlgw, out, 18, HMUARTLGW_APP);
+				do { hmuartlgw_poll(dev.hmuartlgw, 500); } while (rdata.uartlgw_state != HMUARTLGW_STATE_ACK_APP);
+			} while (rdata.status == 0x08);
+
+			memset(out, 0, sizeof(out));
+			out[0] = HMUARTLGW_APP_SET_OLD_KEY;
+			memcpy(&(out[1]), key, 16);
+			out[17] = kNo;
+
+			do {
+				rdata.uartlgw_state = HMUARTLGW_STATE_WAIT_APP;
+				hmuartlgw_send(dev.hmuartlgw, out, 18, HMUARTLGW_APP);
+				do { hmuartlgw_poll(dev.hmuartlgw, 500); } while (rdata.uartlgw_state != HMUARTLGW_STATE_ACK_APP);
+			} while (rdata.status == 0x08);
 		}
+	} else {
+		uint32_t new_hmid = my_hmid;
 
-		printf("\n\nHM-CFG-USB opened\n\n");
+		hmcfgusb_set_debug(debug);
+
+		dev.hmcfgusb = hmcfgusb_init(parse_hmcfgusb, &rdata, hmcfgusb_serial);
+		if (!dev.hmcfgusb) {
+			fprintf(stderr, "Can't initialize HM-CFG-USB\n");
+			exit(EXIT_FAILURE);
+		}
+		dev.type = DEVICE_TYPE_HMCFGUSB;
 
 		memset(out, 0, sizeof(out));
 		out[0] = 'K';
@@ -512,7 +801,80 @@ int main(int argc, char **argv)
 			exit(EXIT_FAILURE);
 		}
 
-		printf("HM-CFG-USB firmware version: %u\n", rdata.version);
+		printf("HM-CFG-USB firmware version: %u, used credits: %u%%\n", rdata.version, rdata.credits);
+
+		if (rdata.credits >= 40) {
+			printf("\nRebooting HM-CFG-USB to avoid running out of credits\n\n");
+
+			if (!dev.hmcfgusb->bootloader) {
+				printf("HM-CFG-USB not in bootloader mode, entering bootloader.\n");
+				printf("Waiting for device to reappear...\n");
+
+				do {
+					if (dev.hmcfgusb) {
+						if (!dev.hmcfgusb->bootloader)
+							hmcfgusb_enter_bootloader(dev.hmcfgusb);
+						hmcfgusb_close(dev.hmcfgusb);
+					}
+					sleep(1);
+				} while (((dev.hmcfgusb = hmcfgusb_init(parse_hmcfgusb, &rdata, hmcfgusb_serial)) == NULL) || (!dev.hmcfgusb->bootloader));
+			}
+
+			if (dev.hmcfgusb->bootloader) {
+				printf("HM-CFG-USB in bootloader mode, rebooting\n");
+
+				do {
+					if (dev.hmcfgusb) {
+						if (dev.hmcfgusb->bootloader)
+							hmcfgusb_leave_bootloader(dev.hmcfgusb);
+						hmcfgusb_close(dev.hmcfgusb);
+					}
+					sleep(1);
+				} while (((dev.hmcfgusb = hmcfgusb_init(parse_hmcfgusb, &rdata, hmcfgusb_serial)) == NULL) || (dev.hmcfgusb->bootloader));
+			}
+		}
+
+		printf("\n\nHM-CFG-USB opened\n\n");
+
+		if (new_hmid && (my_hmid != new_hmid)) {
+			printf("Changing hmid from %06x to %06x\n", my_hmid, new_hmid);
+
+			memset(out, 0, sizeof(out));
+			out[0] = 'A';
+			out[1] = (new_hmid >> 16) & 0xff;
+			out[2] = (new_hmid >> 8) & 0xff;
+			out[3] = new_hmid & 0xff;
+
+			hmcfgusb_send(dev.hmcfgusb, out, sizeof(out), 1);
+
+			my_hmid = new_hmid;
+		}
+
+		if (kNo > 0) {
+			printf("Setting AES-key\n");
+
+			memset(out, 0, sizeof(out));
+			out[0] = 'Y';
+			out[1] = 0x01;
+			out[2] = kNo;
+			out[3] = sizeof(key);
+			memcpy(&(out[4]), key, sizeof(key));
+			hmcfgusb_send(dev.hmcfgusb, out, sizeof(out), 1);
+
+			memset(out, 0, sizeof(out));
+			out[0] = 'Y';
+			out[1] = 0x02;
+			out[2] = 0x00;
+			out[3] = 0x00;
+			hmcfgusb_send(dev.hmcfgusb, out, sizeof(out), 1);
+
+			memset(out, 0, sizeof(out));
+			out[0] = 'Y';
+			out[1] = 0x03;
+			out[2] = 0x00;
+			out[3] = 0x00;
+			hmcfgusb_send(dev.hmcfgusb, out, sizeof(out), 1);
+		}
 	}
 
 	if (!switch_speed(&dev, &rdata, 10)) {
@@ -520,7 +882,64 @@ int main(int argc, char **argv)
 		exit(EXIT_FAILURE);
 	}
 
-	printf("Waiting for device with serial %s\n", serial);
+	if (hmid && my_hmid) {
+		switch (dev.type) {
+			case DEVICE_TYPE_HMCFGUSB:
+				printf("Adding HMID\n");
+
+				memset(out, 0, sizeof(out));
+				out[0] = '+';
+				out[1] = (hmid >> 16) & 0xff;
+				out[2] = (hmid >> 8) & 0xff;
+				out[3] = hmid & 0xff;
+
+				hmcfgusb_send(dev.hmcfgusb, out, sizeof(out), 1);
+				break;
+			case DEVICE_TYPE_HMUARTLGW:
+				printf("Adding HMID\n");
+
+				memset(out, 0, sizeof(out));
+				out[0] = HMUARTLGW_APP_ADD_PEER;
+				out[1] = (hmid >> 16) & 0xff;
+				out[2] = (hmid >> 8) & 0xff;
+				out[3] = hmid & 0xff;
+				out[4] = (kNo > 0) ? kNo : 0x00; /* KeyIndex */
+				out[5] = 0x00; /* WakeUp? */
+				out[6] = 0x00; /* WakeUp? */
+
+				do {
+					rdata.uartlgw_state = HMUARTLGW_STATE_WAIT_APP;
+					hmuartlgw_send(dev.hmuartlgw, out, 7, HMUARTLGW_APP);
+					do { hmuartlgw_poll(dev.hmuartlgw, 500); } while (rdata.uartlgw_state != HMUARTLGW_STATE_ACK_APP);
+				} while (rdata.status == 0x08);
+
+				break;
+		}
+		printf("Sending device with hmid %06x to bootloader\n", hmid);
+		out[CTL] = 0x30;
+		out[TYPE] = 0x11;
+		SET_SRC(out, my_hmid);
+		SET_DST(out, hmid);
+		out[PAYLOAD] = 0xCA;
+		SET_LEN_FROM_PAYLOADLEN(out, 1);
+
+		cnt = 3;
+		do {
+			out[MSGID] = msgid++;
+			if (send_hm_message(&dev, &rdata, out)) {
+				break;
+			}
+		} while (cnt--);
+		if (cnt == -1) {
+			printf("Failed to send device to bootloader, please enter bootloader manually.\n");
+		}
+	}
+
+	if (serial) {
+		printf("Waiting for device with serial %s\n", serial);
+	} else {
+		printf("Waiting for device with HMID %06x\n", hmid);
+	}
 
 	while (1) {
 		errno = 0;
@@ -529,9 +948,14 @@ int main(int argc, char **argv)
 				pfd = culfw_poll(dev.culfw, 1000);
 				break;
 			case DEVICE_TYPE_HMCFGUSB:
-			default:
 				pfd = hmcfgusb_poll(dev.hmcfgusb, 1000);
 				break;
+			case DEVICE_TYPE_HMUARTLGW:
+				pfd = hmuartlgw_poll(dev.hmuartlgw, 1000);
+				break;
+			default:
+				pfd = -1;
+				break;
 		}
 
 		if ((pfd < 0) && errno) {
@@ -547,25 +971,49 @@ int main(int argc, char **argv)
 		    (rdata.message[TYPE] == 0x10) && /* Messagte type: Information */
 		    (DST(rdata.message) == 0x000000) && /* Broadcast */
 		    (rdata.message[PAYLOAD] == 0x00)) { /* FUP? */
-			if (!strncmp((char*)&(rdata.message[0x0b]), serial, 10)) {
+			if (serial && !strncmp((char*)&(rdata.message[0x0b]), serial, 10)) {
 				hmid = SRC(rdata.message);
 				break;
+			} else if (!serial && SRC(rdata.message) == hmid) {
+				serial = (char*)&(rdata.message[0x0b]);
+				break;
 			}
 		}
 	}
 
-	printf("Device with serial %s (hmid: %06x) entered firmware-update-mode\n", serial, hmid);
+	printf("Device with serial %s (HMID: %06x) entered firmware-update-mode\n", serial, hmid);
 
-	if (dev.type == DEVICE_TYPE_HMCFGUSB) {
-		printf("Adding HMID\n");
+	switch (dev.type) {
+		case DEVICE_TYPE_HMCFGUSB:
+			printf("Adding HMID\n");
 
-		memset(out, 0, sizeof(out));
-		out[0] = '+';
-		out[1] = (hmid >> 16) & 0xff;
-		out[2] = (hmid >> 8) & 0xff;
-		out[3] = hmid & 0xff;
+			memset(out, 0, sizeof(out));
+			out[0] = '+';
+			out[1] = (hmid >> 16) & 0xff;
+			out[2] = (hmid >> 8) & 0xff;
+			out[3] = hmid & 0xff;
 
-		hmcfgusb_send(dev.hmcfgusb, out, sizeof(out), 1);
+			hmcfgusb_send(dev.hmcfgusb, out, sizeof(out), 1);
+			break;
+		case DEVICE_TYPE_HMUARTLGW:
+			printf("Adding HMID\n");
+
+			memset(out, 0, sizeof(out));
+			out[0] = HMUARTLGW_APP_ADD_PEER;
+			out[1] = (hmid >> 16) & 0xff;
+			out[2] = (hmid >> 8) & 0xff;
+			out[3] = hmid & 0xff;
+			out[4] = 0x00; /* KeyIndex */
+			out[5] = 0x00; /* WakeUp? */
+			out[6] = 0x00; /* WakeUp? */
+
+			do {
+				rdata.uartlgw_state = HMUARTLGW_STATE_WAIT_APP;
+				hmuartlgw_send(dev.hmuartlgw, out, 7, HMUARTLGW_APP);
+				do { hmuartlgw_poll(dev.hmuartlgw, 500); } while (rdata.uartlgw_state != HMUARTLGW_STATE_ACK_APP);
+			} while (rdata.status == 0x08);
+
+			break;
 	}
 
 	switchcnt = 3;
@@ -717,8 +1165,11 @@ int main(int argc, char **argv)
 	}
 
 	printf("Waiting for device to reboot\n");
+	rdata.message_type = MESSAGE_TYPE_R;
 
 	cnt = 10;
+	if (dev.type == DEVICE_TYPE_HMUARTLGW)
+		cnt = 200; /* FIXME */
 	do {
 		errno = 0;
 		switch(dev.type) {
@@ -726,9 +1177,14 @@ int main(int argc, char **argv)
 				pfd = culfw_poll(dev.culfw, 1000);
 				break;
 			case DEVICE_TYPE_HMCFGUSB:
-			default:
 				pfd = hmcfgusb_poll(dev.hmcfgusb, 1000);
 				break;
+			case DEVICE_TYPE_HMUARTLGW:
+				pfd = hmuartlgw_poll(dev.hmuartlgw, 1000);
+				break;
+			default:
+				pfd = -1;
+				break;
 		}
 		if ((pfd < 0) && errno) {
 			if (errno != ETIMEDOUT) {