aboutsummaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/Makefile.am6
-rw-r--r--src/download.c224
-rw-r--r--src/download.h111
-rw-r--r--src/gpt.c342
-rw-r--r--src/gpt.h75
-rw-r--r--src/hdlc.c355
-rw-r--r--src/hdlc.h45
-rw-r--r--src/lg-downloader.c404
-rw-r--r--src/lg-downloader.h39
-rw-r--r--src/usb.c170
-rw-r--r--src/usb.h50
11 files changed, 1821 insertions, 0 deletions
diff --git a/src/Makefile.am b/src/Makefile.am
new file mode 100644
index 0000000..7f919e2
--- /dev/null
+++ b/src/Makefile.am
@@ -0,0 +1,6 @@
+bin_PROGRAMS = lg-downloader
+
+lg_downloader_SOURCES = lg-downloader.c lg-downloader.h download.c download.h \
+ gpt.c gpt.h hdlc.c hdlc.h usb.c usb.h
+
+MAINTAINERCLEANFILES = Makefile.in
diff --git a/src/download.c b/src/download.c
new file mode 100644
index 0000000..741e2e5
--- /dev/null
+++ b/src/download.c
@@ -0,0 +1,224 @@
+/*
+ * Copyright (C) 2016 Paul Kocialkowski <contact@paulk.fr>
+ *
+ * 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 <stdlib.h>
+#include <string.h>
+#include <endian.h>
+
+#include "lg-downloader.h"
+#include "hdlc.h"
+#include "usb.h"
+#include "download.h"
+
+static int download_response_check(struct download_response *response)
+{
+ if (response == NULL)
+ return -1;
+
+ if (response->ack != DOWNLOAD_ACK)
+ return -1;
+
+ return 0;
+}
+
+static int download_response(struct context *context)
+{
+ struct download_response response;
+ int rc;
+
+ rc = hdlc_recv(context, &response, sizeof(response));
+ if (rc < 0)
+ return -1;
+
+ rc = download_response_check(&response);
+ if (rc < 0)
+ return -1;
+
+ return 0;
+}
+
+static int download_request(struct context *context, unsigned char command)
+{
+ struct download_request request;
+ int rc;
+
+ memset(&request, 0, sizeof(request));
+ request.command = command;
+
+ rc = hdlc_send(context, &request, sizeof(request));
+ if (rc < 0)
+ return -1;
+
+ return 0;
+}
+
+int download_start_flash_write(struct context *context)
+{
+ int rc;
+
+ rc = download_request(context, DOWNLOAD_START_FLASH_WRITE);
+ if (rc < 0)
+ return -1;
+
+ rc = download_response(context);
+ if (rc < 0)
+ return -1;
+
+ return 0;
+}
+
+int download_write(struct context *context, void *buffer, unsigned int address, unsigned int length)
+{
+ struct download_write_request request;
+ struct download_response response;
+ int rc;
+
+ memset(&request, 0, sizeof(request));
+ request.command = DOWNLOAD_WRITE;
+ request.address = htole32(address);
+ request.length = htole32(length);
+
+ rc = hdlc_send(context, &request, sizeof(request));
+ if (rc < 0)
+ return -1;
+
+ rc = usb_send(context, buffer, length);
+ if (rc < 0)
+ return -1;
+
+ rc = hdlc_recv(context, &response, sizeof(response));
+ if (rc < 0)
+ return -1;
+
+ if (response.ack != DOWNLOAD_WRITE)
+ return -1;
+
+ return 0;
+}
+
+int download_erase(struct context *context, unsigned int address, unsigned int length)
+{
+ struct download_erase_request request;
+ int rc;
+
+ memset(&request, 0, sizeof(request));
+ request.command = DOWNLOAD_ERASE;
+ request.address = htole32(address);
+ request.length = htole32(length);
+
+ rc = hdlc_send(context, &request, sizeof(request));
+ if (rc < 0)
+ return -1;
+
+ rc = download_response(context);
+ if (rc < 0)
+ return -1;
+
+ return 0;
+}
+
+int download_read(struct context *context, void *buffer, unsigned int length)
+{
+ struct download_read_request request;
+ int rc;
+
+ memset(&request, 0, sizeof(request));
+ request.command = DOWNLOAD_READ;
+ request.length = htole32(length);
+
+ rc = hdlc_send(context, &request, sizeof(request));
+ if (rc < 0)
+ return -1;
+
+ rc = usb_recv(context, buffer, length);
+ if (rc < 0)
+ return -1;
+
+ return 0;
+}
+
+int download_initialize_partition(struct context *context, unsigned int address, unsigned int length)
+{
+ struct download_initialize_partition_request request;
+ int rc;
+
+ memset(&request, 0, sizeof(request));
+ request.command = DOWNLOAD_INITIALIZE_PARTITION;
+ request.address = htole32(address);
+ request.length = htole32(length);
+
+ rc = hdlc_send(context, &request, sizeof(request));
+ if (rc < 0)
+ return -1;
+
+ rc = download_response(context);
+ if (rc < 0)
+ return -1;
+
+ return 0;
+}
+
+int download_ready_read(struct context *context, unsigned int address, unsigned int length)
+{
+ struct download_ready_read_request request;
+ int rc;
+
+ memset(&request, 0, sizeof(request));
+ request.command = DOWNLOAD_READY_READ;
+ request.address = htole32(address);
+ request.length = htole32(length);
+
+ rc = hdlc_send(context, &request, sizeof(request));
+ if (rc < 0)
+ return -1;
+
+ rc = download_response(context);
+ if (rc < 0)
+ return -1;
+
+ return 0;
+}
+
+int download_reset(struct context *context)
+{
+ int rc;
+
+ rc = download_request(context, DOWNLOAD_RESET);
+ if (rc < 0)
+ return -1;
+
+ rc = download_response(context);
+ if (rc < 0)
+ return -1;
+
+ return 0;
+}
+
+int download_notify_start_dl(struct context *context)
+{
+ int rc;
+
+ rc = download_request(context, DOWNLOAD_NOTIFY_START_DL);
+ if (rc < 0)
+ return -1;
+
+ rc = download_response(context);
+ if (rc < 0)
+ return -1;
+
+ return 0;
+}
diff --git a/src/download.h b/src/download.h
new file mode 100644
index 0000000..4117e7c
--- /dev/null
+++ b/src/download.h
@@ -0,0 +1,111 @@
+/*
+ * Copyright (C) 2016 Paul Kocialkowski <contact@paulk.fr>
+ *
+ * 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/>.
+ */
+
+#ifndef _DOWNLOAD_H_
+#define _DOWNLOAD_H_
+
+#include "lg-downloader.h"
+
+/*
+ * Values
+ */
+
+#define DOWNLOAD_START_FLASH_WRITE 0x01
+#define DOWNLOAD_WRITE 0x02
+#define DOWNLOAD_COMPLETE_FLASH_WRITE 0x03
+#define DOWNLOAD_ERASE 0x04
+#define DOWNLOAD_READ 0x05
+#define DOWNLOAD_INITIALIZE_PARTITION 0x06
+#define DOWNLOAD_WRITE_ROM_COPY 0x07
+#define DOWNLOAD_READY_READ 0x08
+#define DOWNLOAD_WRITE_ASYNC 0x09
+#define DOWNLOAD_GO 0x20
+#define DOWNLOAD_NO_OP 0x21
+#define DOWNLOAD_RESET 0x22
+#define DOWNLOAD_POWERDOWN 0x23
+#define DOWNLOAD_PRE_RESET 0x24
+#define DOWNLOAD_CP_USB_SWITCH 0x25
+#define DOWNLOAD_SOFTWARE_VER 0x40
+#define DOWNLOAD_NOTIFY_START_DL 0x41
+
+#define DOWNLOAD_ACK 0x50
+#define DOWNLOAD_NACK 0x51
+
+#define DOWNLOAD_BUFFER_SIZE 1024
+
+/*
+ * Structures
+ */
+
+struct download_request {
+ unsigned char command;
+} __attribute__((__packed__));
+
+struct download_write_request {
+ unsigned char command;
+ unsigned char binary_type;
+ unsigned short reserved;
+ unsigned int address;
+ unsigned int length;
+} __attribute__((__packed__));
+
+struct download_erase_request {
+ unsigned char command;
+ unsigned char reserved[3];
+ unsigned int address;
+ unsigned int length;
+} __attribute__((__packed__));
+
+struct download_read_request {
+ unsigned char command;
+ unsigned char reserved;
+ unsigned int length;
+} __attribute__((__packed__));
+
+struct download_initialize_partition_request {
+ unsigned char command;
+ unsigned char binary_type;
+ unsigned int address;
+ unsigned int length;
+} __attribute__((__packed__));
+
+struct download_ready_read_request {
+ unsigned char command;
+ unsigned char reserved;
+ unsigned int address;
+ unsigned int length;
+} __attribute__((__packed__));
+
+struct download_response {
+ unsigned char ack;
+} __attribute__((__packed__));
+
+/*
+ * Functions
+ */
+
+int download_start_flash_write(struct context *context);
+int download_write(struct context *context, void *buffer, unsigned int address, unsigned int length);
+int download_erase(struct context *context, unsigned int address, unsigned int length);
+int download_read(struct context *context, void *buffer, unsigned int length);
+int download_initialize_partition(struct context *context, unsigned int address, unsigned int length);
+int download_ready_read(struct context *context, unsigned int address, unsigned int length);
+int download_reset(struct context *context);
+int download_notify_start_dl(struct context *context);
+
+
+#endif
diff --git a/src/gpt.c b/src/gpt.c
new file mode 100644
index 0000000..40ac4a5
--- /dev/null
+++ b/src/gpt.c
@@ -0,0 +1,342 @@
+/*
+ * Copyright (C) 2016 Paul Kocialkowski <contact@paulk.fr>
+ *
+ * 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 usefu,
+ * 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 <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+#include <ctype.h>
+
+#include "lg-downloader.h"
+#include "download.h"
+#include "gpt.h"
+
+/* Lookup table for CRC32 with generator polynomial 0x04C11Db7. */
+static const unsigned int gpt_crc_table[] = {
+ 0x00000000, 0x77073096, 0xee0e612c, 0x990951ba, 0x076dc419,
+ 0x706af48f, 0xe963a535, 0x9e6495a3, 0x0edb8832, 0x79dcb8a4,
+ 0xe0d5e91e, 0x97d2d988, 0x09b64c2b, 0x7eb17cbd, 0xe7b82d07,
+ 0x90bf1d91, 0x1db71064, 0x6ab020f2, 0xf3b97148, 0x84be41de,
+ 0x1adad47d, 0x6ddde4eb, 0xf4d4b551, 0x83d385c7, 0x136c9856,
+ 0x646ba8c0, 0xfd62f97a, 0x8a65c9ec, 0x14015c4f, 0x63066cd9,
+ 0xfa0f3d63, 0x8d080df5, 0x3b6e20c8, 0x4c69105e, 0xd56041e4,
+ 0xa2677172, 0x3c03e4d1, 0x4b04d447, 0xd20d85fd, 0xa50ab56b,
+ 0x35b5a8fa, 0x42b2986c, 0xdbbbc9d6, 0xacbcf940, 0x32d86ce3,
+ 0x45df5c75, 0xdcd60dcf, 0xabd13d59, 0x26d930ac, 0x51de003a,
+ 0xc8d75180, 0xbfd06116, 0x21b4f4b5, 0x56b3c423, 0xcfba9599,
+ 0xb8bda50f, 0x2802b89e, 0x5f058808, 0xc60cd9b2, 0xb10be924,
+ 0x2f6f7c87, 0x58684c11, 0xc1611dab, 0xb6662d3d, 0x76dc4190,
+ 0x01db7106, 0x98d220bc, 0xefd5102a, 0x71b18589, 0x06b6b51f,
+ 0x9fbfe4a5, 0xe8b8d433, 0x7807c9a2, 0x0f00f934, 0x9609a88e,
+ 0xe10e9818, 0x7f6a0dbb, 0x086d3d2d, 0x91646c97, 0xe6635c01,
+ 0x6b6b51f4, 0x1c6c6162, 0x856530d8, 0xf262004e, 0x6c0695ed,
+ 0x1b01a57b, 0x8208f4c1, 0xf50fc457, 0x65b0d9c6, 0x12b7e950,
+ 0x8bbeb8ea, 0xfcb9887c, 0x62dd1ddf, 0x15da2d49, 0x8cd37cf3,
+ 0xfbd44c65, 0x4db26158, 0x3ab551ce, 0xa3bc0074, 0xd4bb30e2,
+ 0x4adfa541, 0x3dd895d7, 0xa4d1c46d, 0xd3d6f4fb, 0x4369e96a,
+ 0x346ed9fc, 0xad678846, 0xda60b8d0, 0x44042d73, 0x33031de5,
+ 0xaa0a4c5f, 0xdd0d7cc9, 0x5005713c, 0x270241aa, 0xbe0b1010,
+ 0xc90c2086, 0x5768b525, 0x206f85b3, 0xb966d409, 0xce61e49f,
+ 0x5edef90e, 0x29d9c998, 0xb0d09822, 0xc7d7a8b4, 0x59b33d17,
+ 0x2eb40d81, 0xb7bd5c3b, 0xc0ba6cad, 0xedb88320, 0x9abfb3b6,
+ 0x03b6e20c, 0x74b1d29a, 0xead54739, 0x9dd277af, 0x04db2615,
+ 0x73dc1683, 0xe3630b12, 0x94643b84, 0x0d6d6a3e, 0x7a6a5aa8,
+ 0xe40ecf0b, 0x9309ff9d, 0x0a00ae27, 0x7d079eb1, 0xf00f9344,
+ 0x8708a3d2, 0x1e01f268, 0x6906c2fe, 0xf762575d, 0x806567cb,
+ 0x196c3671, 0x6e6b06e7, 0xfed41b76, 0x89d32be0, 0x10da7a5a,
+ 0x67dd4acc, 0xf9b9df6f, 0x8ebeeff9, 0x17b7be43, 0x60b08ed5,
+ 0xd6d6a3e8, 0xa1d1937e, 0x38d8c2c4, 0x4fdff252, 0xd1bb67f1,
+ 0xa6bc5767, 0x3fb506dd, 0x48b2364b, 0xd80d2bda, 0xaf0a1b4c,
+ 0x36034af6, 0x41047a60, 0xdf60efc3, 0xa867df55, 0x316e8eef,
+ 0x4669be79, 0xcb61b38c, 0xbc66831a, 0x256fd2a0, 0x5268e236,
+ 0xcc0c7795, 0xbb0b4703, 0x220216b9, 0x5505262f, 0xc5ba3bbe,
+ 0xb2bd0b28, 0x2bb45a92, 0x5cb36a04, 0xc2d7ffa7, 0xb5d0cf31,
+ 0x2cd99e8b, 0x5bdeae1d, 0x9b64c2b0, 0xec63f226, 0x756aa39c,
+ 0x026d930a, 0x9c0906a9, 0xeb0e363f, 0x72076785, 0x05005713,
+ 0x95bf4a82, 0xe2b87a14, 0x7bb12bae, 0x0cb61b38, 0x92d28e9b,
+ 0xe5d5be0d, 0x7cdcefb7, 0x0bdbdf21, 0x86d3d2d4, 0xf1d4e242,
+ 0x68ddb3f8, 0x1fda836e, 0x81be16cd, 0xf6b9265b, 0x6fb077e1,
+ 0x18b74777, 0x88085ae6, 0xff0f6a70, 0x66063bca, 0x11010b5c,
+ 0x8f659eff, 0xf862ae69, 0x616bffd3, 0x166ccf45, 0xa00ae278,
+ 0xd70dd2ee, 0x4e048354, 0x3903b3c2, 0xa7672661, 0xd06016f7,
+ 0x4969474d, 0x3e6e77db, 0xaed16a4a, 0xd9d65adc, 0x40df0b66,
+ 0x37d83bf0, 0xa9bcae53, 0xdebb9ec5, 0x47b2cf7f, 0x30b5ffe9,
+ 0xbdbdf21c, 0xcabac28a, 0x53b39330, 0x24b4a3a6, 0xbad03605,
+ 0xcdd70693, 0x54de5729, 0x23d967bf, 0xb3667a2e, 0xc4614ab8,
+ 0x5d681b02, 0x2a6f2b94, 0xb40bbe37, 0xc30c8ea1, 0x5a05df1b,
+ 0x2d02ef8d
+};
+
+static unsigned int gpt_crc_step(unsigned int crc, unsigned char data)
+{
+ return ((crc >> 8) ^ gpt_crc_table[(crc ^ data) & 0xff]);
+}
+
+static unsigned int gpt_crc_final(unsigned int crc)
+{
+ return crc ^ GPT_CRC_SEED;
+}
+
+static void gpt_partition_name(struct gpt_partition *partition, char *name)
+{
+ unsigned int i;
+ unsigned int j;
+ char c;
+
+ if (partition == NULL || name == NULL)
+ return;
+
+ for (i = 0, j = 0; i < GPT_PARTITION_NAME_COUNT; i++) {
+ c = (char) le16toh(partition->name[i]);
+
+ if (isprint(c))
+ name[j++] = c;
+ }
+
+ name[j] = '\0';
+}
+
+int gpt_header_verify(struct gpt_header *header)
+{
+ struct gpt_header header_copy;
+ unsigned char *p;
+ unsigned int crc;
+ unsigned int i;
+
+ if (header == NULL)
+ return -1;
+
+ if (strncmp(header->signature, GPT_SIGNATURE, sizeof(header->signature)) != 0) {
+ fprintf(stderr, "Verifying GPT signature failed\n");
+ return -1;
+ }
+
+ if (le32toh(header->version) != GPT_VERSION) {
+ fprintf(stderr, "Verifying GPT version failed\n");
+ return -1;
+ }
+
+ memcpy(&header_copy, header, sizeof(struct gpt_header));
+ header_copy.crc = 0;
+
+ crc = GPT_CRC_SEED;
+ p = (unsigned char *) &header_copy;
+
+ for (i = 0; i < sizeof(struct gpt_header); i++)
+ crc = gpt_crc_step(crc, *p++);
+
+ crc = gpt_crc_final(crc);
+
+ if (le32toh(header->crc) != crc) {
+ fprintf(stderr, "Verifying GPT CRC failed\n");
+ return -1;
+ }
+
+ if (le32toh(header->partitions_size) != sizeof(struct gpt_partition))
+ return -1;
+
+ if ((GPT_BLOCK_SIZE % le32toh(header->partitions_size)) != 0)
+ return -1;
+
+ if (le32toh(header->partitions_lba) == 0 || le32toh(header->partitions_count) == 0)
+ return -1;
+
+ return 0;
+}
+
+int gpt_header_read(struct context *context, struct gpt_header *header)
+{
+ unsigned int address;
+ unsigned int length;
+ int rc;
+
+ address = GPT_HEADER_LBA;
+ length = sizeof(struct gpt_header);
+
+ rc = download_ready_read(context, address, length);
+ if (rc < 0)
+ return -1;
+
+ rc = download_read(context, (void *) header, length);
+ if (rc < 0)
+ return -1;
+
+ return 0;
+}
+
+int gpt_partition_find(struct context *context)
+{
+ struct gpt_header header;
+ struct gpt_partition *partition;
+ void *buffer = NULL;
+ char name[GPT_PARTITION_NAME_COUNT + 1];
+ unsigned int address;
+ unsigned int length;
+ unsigned int count;
+ unsigned int index;
+ unsigned int i;
+ int rc;
+
+ if (context == NULL || context->partition == NULL)
+ return -1;
+
+ if (context->verbose)
+ printf("Reading GPT header\n");
+
+ rc = gpt_header_read(context, &header);
+ if (rc < 0) {
+ fprintf(stderr, "Reading GPT header failed\n");
+ goto error;
+ }
+
+ if (context->verbose)
+ printf("Verifying GPT header\n");
+
+ rc = gpt_header_verify(&header);
+ if (rc < 0) {
+ fprintf(stderr, "Verifying GPT header failed\n");
+ goto error;
+ }
+
+ address = le32toh(header.partitions_lba);
+ length = le32toh(header.partitions_size);
+ count = le32toh(header.partitions_count);
+
+ buffer = calloc(1, GPT_BLOCK_SIZE);
+
+ for (i = 0; i < count; i++) {
+ index = i % (GPT_BLOCK_SIZE / length);
+
+ if (index == 0) {
+ rc = download_ready_read(context, address, GPT_BLOCK_SIZE);
+ if (rc < 0)
+ goto error;
+
+ rc = download_read(context, buffer, GPT_BLOCK_SIZE);
+ if (rc < 0)
+ goto error;
+
+ address++;
+ }
+
+ partition = ((struct gpt_partition *) buffer) + index;
+
+ gpt_partition_name(partition, (char *) &name);
+
+ if (name[0] != '\0') {
+ if (context->verbose)
+ printf("Found GPT partition: %s\n", name);
+
+ if (strcmp(name, context->partition) == 0) {
+ printf("Matched GPT partition: %s\n", name);
+
+ context->address = (unsigned int) le64toh(partition->first_lba);
+ context->length = (unsigned int) (le64toh(partition->last_lba) - le64toh(partition->first_lba) + 1) * GPT_BLOCK_SIZE;
+
+ rc = 0;
+ goto complete;
+ }
+
+ }
+ }
+
+ fprintf(stderr, "Matching GPT partition: %s failed\n", context->partition);
+
+error:
+ rc = -1;
+
+complete:
+ if (buffer != NULL)
+ free(buffer);
+
+ return rc;
+}
+
+int gpt_partitions_print(struct context *context)
+{
+ struct gpt_header header;
+ struct gpt_partition *partition;
+ void *buffer = NULL;
+ char name[GPT_PARTITION_NAME_COUNT + 1];
+ unsigned int address;
+ unsigned int length;
+ unsigned int count;
+ unsigned int index;
+ unsigned int i;
+ int rc;
+
+ if (context == NULL)
+ return -1;
+
+ if (context->verbose)
+ printf("Reading GPT header\n");
+
+ rc = gpt_header_read(context, &header);
+ if (rc < 0) {
+ fprintf(stderr, "Reading GPT header failed\n");
+ goto error;
+ }
+
+ if (context->verbose)
+ printf("Verifying GPT header\n");
+
+ rc = gpt_header_verify(&header);
+ if (rc < 0) {
+ fprintf(stderr, "Verifying GPT header failed\n");
+ goto error;
+ }
+
+ address = le32toh(header.partitions_lba);
+ length = le32toh(header.partitions_size);
+ count = le32toh(header.partitions_count);
+
+ buffer = calloc(1, GPT_BLOCK_SIZE);
+
+ for (i = 0; i < count; i++) {
+ index = i % (GPT_BLOCK_SIZE / length);
+
+ if (index == 0) {
+ rc = download_ready_read(context, address, GPT_BLOCK_SIZE);
+ if (rc < 0)
+ goto error;
+
+ rc = download_read(context, buffer, GPT_BLOCK_SIZE);
+ if (rc < 0)
+ goto error;
+
+ address++;
+ }
+
+ partition = ((struct gpt_partition *) buffer) + index;
+
+ gpt_partition_name(partition, (char *) &name);
+
+ if (name[0] != '\0')
+ printf("Partition name: %s, size: %d bytes\n", name, (unsigned int) (le64toh(partition->last_lba) - le64toh(partition->first_lba) + 1) * GPT_BLOCK_SIZE);
+ }
+
+ rc = 0;
+ goto complete;
+
+error:
+ rc = -1;
+
+complete:
+ if (buffer != NULL)
+ free(buffer);
+
+ return rc;
+}
diff --git a/src/gpt.h b/src/gpt.h
new file mode 100644
index 0000000..79b5b8c
--- /dev/null
+++ b/src/gpt.h
@@ -0,0 +1,75 @@
+/*
+ * Copyright (C) 2016 Paul Kocialkowski <contact@paulk.fr>
+ *
+ * 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/>.
+ */
+
+#ifndef _GPT_H_
+#define _GPT_H_
+
+/*
+ * Values
+ */
+
+#define GPT_BLOCK_SIZE 512
+
+#define GPT_HEADER_LBA 1
+#define GPT_VERSION 0x00010000
+
+#define GPT_CRC_SEED 0xffffffff
+
+#define GPT_PARTITION_NAME_COUNT 36
+
+#define GPT_SIGNATURE "EFI PART"
+
+/*
+ * Structures
+ */
+
+struct gpt_header {
+ char signature[8];
+ unsigned int version;
+ unsigned int size;
+ unsigned int crc;
+ unsigned int reserved;
+ unsigned long long current_lba;
+ unsigned long long backup_lba;
+ unsigned long long first_lba;
+ unsigned long long last_lba;
+ unsigned char guid[16];
+ unsigned long long partitions_lba;
+ unsigned int partitions_count;
+ unsigned int partitions_size;
+ unsigned int partitions_crc;
+} __attribute__((__packed__));
+
+struct gpt_partition {
+ unsigned char type_guid[16];
+ unsigned char guid[16];
+ unsigned long long first_lba;
+ unsigned long long last_lba;
+ unsigned long long flags;
+ short name[GPT_PARTITION_NAME_COUNT];
+} __attribute__((__packed__));
+
+/*
+ * Functions
+ */
+
+int gpt_header_verify(struct gpt_header *header);
+int gpt_header_read(struct context *context, struct gpt_header *header);
+int gpt_partition_find(struct context *context);
+int gpt_partitions_print(struct context *context);
+
+#endif
diff --git a/src/hdlc.c b/src/hdlc.c
new file mode 100644
index 0000000..cfc9784
--- /dev/null
+++ b/src/hdlc.c
@@ -0,0 +1,355 @@
+/*
+ * Copyright (C) 2016 Paul Kocialkowski <contact@paulk.fr>
+ *
+ * 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 <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+#include <endian.h>
+
+#include "lg-downloader.h"
+#include "hdlc.h"
+
+/* Lookup table for CCITT-16 CRC with generator polynomial 0x8408. */
+static const unsigned short hdlc_crc_table[] = {
+ 0x0000, 0x1189, 0x2312, 0x329b, 0x4624, 0x57ad, 0x6536, 0x74bf,
+ 0x8c48, 0x9dc1, 0xaf5a, 0xbed3, 0xca6c, 0xdbe5, 0xe97e, 0xf8f7,
+ 0x1081, 0x0108, 0x3393, 0x221a, 0x56a5, 0x472c, 0x75b7, 0x643e,
+ 0x9cc9, 0x8d40, 0xbfdb, 0xae52, 0xdaed, 0xcb64, 0xf9ff, 0xe876,
+ 0x2102, 0x308b, 0x0210, 0x1399, 0x6726, 0x76af, 0x4434, 0x55bd,
+ 0xad4a, 0xbcc3, 0x8e58, 0x9fd1, 0xeb6e, 0xfae7, 0xc87c, 0xd9f5,
+ 0x3183, 0x200a, 0x1291, 0x0318, 0x77a7, 0x662e, 0x54b5, 0x453c,
+ 0xbdcb, 0xac42, 0x9ed9, 0x8f50, 0xfbef, 0xea66, 0xd8fd, 0xc974,
+ 0x4204, 0x538d, 0x6116, 0x709f, 0x0420, 0x15a9, 0x2732, 0x36bb,
+ 0xce4c, 0xdfc5, 0xed5e, 0xfcd7, 0x8868, 0x99e1, 0xab7a, 0xbaf3,
+ 0x5285, 0x430c, 0x7197, 0x601e, 0x14a1, 0x0528, 0x37b3, 0x263a,
+ 0xdecd, 0xcf44, 0xfddf, 0xec56, 0x98e9, 0x8960, 0xbbfb, 0xaa72,
+ 0x6306, 0x728f, 0x4014, 0x519d, 0x2522, 0x34ab, 0x0630, 0x17b9,
+ 0xef4e, 0xfec7, 0xcc5c, 0xddd5, 0xa96a, 0xb8e3, 0x8a78, 0x9bf1,
+ 0x7387, 0x620e, 0x5095, 0x411c, 0x35a3, 0x242a, 0x16b1, 0x0738,
+ 0xffcf, 0xee46, 0xdcdd, 0xcd54, 0xb9eb, 0xa862, 0x9af9, 0x8b70,
+ 0x8408, 0x9581, 0xa71a, 0xb693, 0xc22c, 0xd3a5, 0xe13e, 0xf0b7,
+ 0x0840, 0x19c9, 0x2b52, 0x3adb, 0x4e64, 0x5fed, 0x6d76, 0x7cff,
+ 0x9489, 0x8500, 0xb79b, 0xa612, 0xd2ad, 0xc324, 0xf1bf, 0xe036,
+ 0x18c1, 0x0948, 0x3bd3, 0x2a5a, 0x5ee5, 0x4f6c, 0x7df7, 0x6c7e,
+ 0xa50a, 0xb483, 0x8618, 0x9791, 0xe32e, 0xf2a7, 0xc03c, 0xd1b5,
+ 0x2942, 0x38cb, 0x0a50, 0x1bd9, 0x6f66, 0x7eef, 0x4c74, 0x5dfd,
+ 0xb58b, 0xa402, 0x9699, 0x8710, 0xf3af, 0xe226, 0xd0bd, 0xc134,
+ 0x39c3, 0x284a, 0x1ad1, 0x0b58, 0x7fe7, 0x6e6e, 0x5cf5, 0x4d7c,
+ 0xc60c, 0xd785, 0xe51e, 0xf497, 0x8028, 0x91a1, 0xa33a, 0xb2b3,
+ 0x4a44, 0x5bcd, 0x6956, 0x78df, 0x0c60, 0x1de9, 0x2f72, 0x3efb,
+ 0xd68d, 0xc704, 0xf59f, 0xe416, 0x90a9, 0x8120, 0xb3bb, 0xa232,
+ 0x5ac5, 0x4b4c, 0x79d7, 0x685e, 0x1ce1, 0x0d68, 0x3ff3, 0x2e7a,
+ 0xe70e, 0xf687, 0xc41c, 0xd595, 0xa12a, 0xb0a3, 0x8238, 0x93b1,
+ 0x6b46, 0x7acf, 0x4854, 0x59dd, 0x2d62, 0x3ceb, 0x0e70, 0x1ff9,
+ 0xf78f, 0xe606, 0xd49d, 0xc514, 0xb1ab, 0xa022, 0x92b9, 0x8330,
+ 0x7bc7, 0x6a4e, 0x58d5, 0x495c, 0x3de3, 0x2c6a, 0x1ef1, 0x0f78
+};
+
+static unsigned short hdlc_crc_step(unsigned short crc, unsigned char data)
+{
+ return ((crc >> 8) ^ hdlc_crc_table[(crc ^ data) & 0xff]);
+}
+
+static unsigned short hdlc_crc_final(unsigned short crc)
+{
+ return crc ^ HDLC_CRC_SEED;
+}
+
+static unsigned int hdlc_escape_size(const void *data, unsigned int size)
+{
+ unsigned int length;
+ unsigned char *p;
+ unsigned int i;
+
+ if (data == NULL || size == 0)
+ return 0;
+
+ p = (unsigned char *) data;
+ length = size;
+
+ for (i = 0; i < size; i++)
+ if (p[i] == HDLC_FLAG || p[i] == HDLC_ESCAPE)
+ length++;
+
+ return length;
+}
+
+static int hdlc_escape(const void *data, unsigned int size, void *buffer)
+{
+ unsigned char *p;
+ unsigned char *q;
+ unsigned int i;
+ unsigned int j;
+
+ if (buffer == NULL)
+ return -1;
+
+ p = (unsigned char *) data;
+ q = (unsigned char *) buffer;
+
+ for (i = 0, j = 0; i < size; i++) {
+ if (p[i] == HDLC_FLAG || p[i] == HDLC_ESCAPE) {
+ q[j++] = HDLC_ESCAPE;
+ q[j++] = p[i] ^ HDLC_ESCAPE_MASK;
+ } else {
+ q[j++] = p[i];
+ }
+ }
+
+ return 0;
+}
+
+static int hdlc_unescape_size(const void *data, unsigned int size)
+{
+ if (data == NULL || size == 0)
+ return 0;
+
+ unsigned int length;
+ unsigned char *p;
+ unsigned int i;
+
+ if (data == NULL || size == 0)
+ return 0;
+
+ p = (unsigned char *) data;
+ length = size;
+
+ for (i = 0; i < size; i++)
+ if (p[i] == HDLC_ESCAPE)
+ length--;
+
+ return length;
+}
+
+static int hdlc_unescape(const void *data, unsigned int size, void *buffer)
+{
+ unsigned char *p;
+ unsigned char *q;
+ unsigned int i;
+ unsigned int j;
+
+ if (buffer == NULL)
+ return -1;
+
+ p = (unsigned char *) data;
+ q = (unsigned char *) buffer;
+
+ for (i = 0, j = 0; i < size; i++) {
+ if (p[i] == HDLC_ESCAPE && i < (size - 1))
+ q[j++] = p[++i] ^ HDLC_ESCAPE_MASK;
+ else
+ q[j++] = p[i];
+ }
+
+ return 0;
+}
+
+int hdlc_send(struct context *context, const void *data, unsigned int size)
+{
+ void *buffer = NULL;
+ unsigned int length;
+ unsigned short crc;
+ unsigned char *p;
+ unsigned short *q;
+ unsigned int i;
+ int rc;
+
+ if (data == NULL || size == 0)
+ return -1;
+
+ /* Escaped length */
+
+ length = hdlc_escape_size(data, size);
+ if (length == 0)
+ return -1;
+
+ /* CRC */
+
+ crc = HDLC_CRC_SEED;
+ p = (unsigned char *) data;
+
+ /* CRC is calculated from unescaped data */
+ for (i = 0; i < size; i++)
+ crc = hdlc_crc_step(crc, *p++);
+
+ /* Trailing zeros must be taken in account */
+ if (size < HDLC_MIN_SIZE)
+ for (i = 0; i < (HDLC_MIN_SIZE - size); i++)
+ crc = hdlc_crc_step(crc, 0);
+
+ crc = hdlc_crc_final(crc);
+
+ /* Final length */
+
+ /* Minimum size is relative to unescaped data */
+ if (size < HDLC_MIN_SIZE);
+ length = (length - size) + HDLC_MIN_SIZE;
+
+ length = HDLC_HEAD_SIZE + length + HDLC_TAIL_SIZE;
+
+ /* Copy */
+
+ buffer = calloc(1, length);
+
+ p = (unsigned char *) buffer;
+
+ *p = HDLC_FLAG;
+ p += HDLC_HEAD_SIZE;
+
+ rc = hdlc_escape(data, size, p);
+ if (rc < 0)
+ goto error;
+
+ p += (length - HDLC_HEAD_SIZE - HDLC_TAIL_SIZE);
+
+ q = (unsigned short *) p;
+
+ *q = htole16(crc);
+ q++;
+
+ p = (unsigned char *) q;
+
+ *p++ = HDLC_FLAG;
+
+ /* USB send */
+
+ rc = usb_send(context, buffer, length);
+ if (rc < 0) {
+ fprintf(stderr, "Sending HDLC data failed\n");
+ goto error;
+ }
+
+ rc = 0;
+ goto complete;
+
+error:
+ rc = -1;
+
+complete:
+ if (buffer != NULL)
+ free(buffer);
+
+ return rc;
+}
+
+int hdlc_recv(struct context *context, void *data, unsigned int size)
+{
+ void *contents = NULL;
+ void *buffer = NULL;
+ unsigned int length;
+ unsigned int count;
+ unsigned short crc;
+ unsigned char *p;
+ unsigned int i;
+ int rc;
+
+ if (data == NULL || size == 0)
+ return -1;
+
+ /* Length and buffer */
+
+ if (size < HDLC_MIN_SIZE)
+ length = HDLC_MIN_SIZE;
+ else
+ length = size;
+
+ /* Worst case: all bytes are escaped */
+ length = 2 * length + HDLC_TAIL_SIZE;
+
+ buffer = calloc(1, length);
+
+ count = 0;
+ p = (unsigned char *) buffer;
+
+ /* USB recv */
+
+ while (count < length) {
+ rc = usb_recv_available(context, p, length - count);
+ if (rc < 0) {
+ fprintf(stderr, "Reading HDLC data failed\n");
+ fprintf(stderr, "\nYou may want to try disconnecting and reconnecting the USB cable\n");
+ goto error;
+ }
+
+ for (i = 0; i < rc; i++) {
+ if (p[i] == HDLC_FLAG) {
+ count += i + 1;
+ goto check;
+ }
+ }
+
+ count += rc;
+ p += rc;
+ }
+
+ /* Check */
+check:
+ p = (unsigned char *) buffer;
+
+ /* Check for bare minimum size and flag */
+ if (count < (HDLC_TAIL_SIZE + 1) || p[count - 1] != HDLC_FLAG) {
+ fprintf(stderr, "Read invalid HDLC data\n");
+ goto error;
+ }
+
+ /* HDLC unescape */
+
+ /* Eliminate the HDLC flag */
+ count--;
+
+ length = hdlc_unescape_size(buffer, count);
+ if (length < size)
+ goto error;
+
+ contents = calloc(1, length);
+
+ rc = hdlc_unescape(buffer, count, contents);
+ if (rc < 0)
+ goto error;
+
+ /* CRC */
+
+ crc = HDLC_CRC_SEED;
+ p = (unsigned char *) contents;
+
+ for (i = 0; i < length; i++)
+ crc = hdlc_crc_step(crc, *p++);
+
+ if (crc != HDLC_CRC_FINAL) {
+ fprintf(stderr, "Invalid CRC for HDLC data\n");
+ goto error;
+ }
+
+ /* Copy */
+
+ memcpy(data, contents, size);
+
+ rc = 0;
+ goto complete;
+
+error:
+ rc = -1;
+
+complete:
+ if (contents != NULL)
+ free(contents);
+
+ if (buffer != NULL)
+ free(buffer);
+
+ return rc;
+}
diff --git a/src/hdlc.h b/src/hdlc.h
new file mode 100644
index 0000000..52f3756
--- /dev/null
+++ b/src/hdlc.h
@@ -0,0 +1,45 @@
+/*
+ * Copyright (C) 2016 Paul Kocialkowski <contact@paulk.fr>
+ *
+ * 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/>.
+ */
+
+#ifndef _HDLC_H_
+#define _HDLC_H_
+
+#include "lg-downloader.h"
+
+/*
+ * Values
+ */
+
+#define HDLC_FLAG 0x7e
+#define HDLC_ESCAPE 0x7d
+#define HDLC_ESCAPE_MASK 0x20
+
+#define HDLC_CRC_SEED 0xffff
+#define HDLC_CRC_FINAL 0xf0b8
+
+#define HDLC_MIN_SIZE 12
+#define HDLC_HEAD_SIZE 1
+#define HDLC_TAIL_SIZE 3
+
+/*
+ * Functions
+ */
+
+int hdlc_send(struct context *context, const void *data, unsigned int size);
+int hdlc_recv(struct context *context, void *data, unsigned int size);
+
+#endif
diff --git a/src/lg-downloader.c b/src/lg-downloader.c
new file mode 100644
index 0000000..5c3ecfc
--- /dev/null
+++ b/src/lg-downloader.c
@@ -0,0 +1,404 @@
+/*
+ * Copyright (C) 2016 Paul Kocialkowski <contact@paulk.fr>
+ *
+ * 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 <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+#include <fcntl.h>
+#include <sys/stat.h>
+
+#include "lg-downloader.h"
+#include "gpt.h"
+#include "download.h"
+#include "usb.h"
+
+static int usage_print(void)
+{
+ printf("Usage: lg-downloader [OPTIONS] [OPERATION]\n\n"
+ "Options:\n"
+ " -v verbose\n\n"
+ "Operations:\n"
+ " reboot reboot device\n"
+ " erase [partition] [filename] erase a partition\n"
+ " flash [partition] [filename] flash a file to partition\n"
+ " dump [partition] [filename] dump a partition to file\n");
+}
+
+static int arguments_parse(struct context *context, int argc, char *argv[])
+{
+ int i;
+
+ if (context == NULL || argc < 2)
+ return -1;
+
+ for (i = 1; i < argc; i++) {
+ if (strcmp(argv[i], "reboot") == 0) {
+ if (context->operation)
+ return -1;
+
+ context->operation = 'r';
+ } else if (strcmp(argv[i], "erase") == 0) {
+ if (context->operation || argc <= (i + 1))
+ return -1;
+
+ context->operation = 'e';
+ context->partition = argv[++i];
+ context->filename = argv[++i];
+ } else if (strcmp(argv[i], "flash") == 0) {
+ if (context->operation || argc <= (i + 2))
+ return -1;
+
+ context->operation = 'f';
+ context->partition = argv[++i];
+ context->filename = argv[++i];
+ } else if (strcmp(argv[i], "dump") == 0) {
+ if (context->operation || argc <= (i + 2))
+ return -1;
+
+ context->operation = 'd';
+ context->partition = argv[++i];
+ context->filename = argv[++i];
+ } else if (strcmp(argv[i], "partitions") == 0) {
+ if (context->operation)
+ return -1;
+
+ context->operation = 'p';
+ } else if (strcmp(argv[i], "-v") == 0) {
+ context->verbose = 1;
+ } else {
+ return -1;
+ }
+ }
+
+ if (!context->operation)
+ return -1;
+
+ return 0;
+}
+
+int erase(struct context *context)
+{
+ int rc;
+
+ rc = gpt_partition_find(context);
+ if (rc < 0) {
+ fprintf(stderr, "Finding GPT partition failed\n");
+ return -1;
+ }
+
+ if (context->verbose)
+ printf("Erasing partition with address %d blocks and length %d bytes\n", context->address, context->length);
+
+ rc = download_erase(context, context->address, context->length);
+ if (rc < 0)
+ return -1;
+
+ return 0;
+}
+
+int flash(struct context *context)
+{
+ void *buffer = NULL;
+ unsigned int address;
+ unsigned int length;
+ unsigned int count;
+ unsigned int chunk;
+ unsigned char *p;
+ unsigned int i;
+ struct stat st;
+ int fd = -1;
+ int rc;
+
+ if (context == NULL || context->filename == NULL)
+ return -1;
+
+ rc = gpt_partition_find(context);
+ if (rc < 0) {
+ fprintf(stderr, "Finding GPT partition failed\n");
+ goto error;
+ }
+
+ rc = stat(context->filename, &st);
+ if (rc < 0) {
+ fprintf(stderr, "Stating file failed\n");
+ goto error;
+ }
+
+ length = st.st_size;
+
+ if (length > context->length) {
+ fprintf(stderr, "File size (%d bytes) is too big for partition (%d bytes)\n", length, context->length);
+ goto error;
+ }
+
+ fd = open(context->filename, O_RDONLY);
+ if (fd < 0) {
+ fprintf(stderr, "Opening file failed\n");
+ goto error;
+ }
+
+ if (context->verbose)
+ printf("Writing to partition with address %d blocks and length %d bytes\n", context->address, length);
+
+ buffer = calloc(1, GPT_BLOCK_SIZE);
+
+ rc = download_start_flash_write(context);
+ if (rc < 0)
+ goto error;
+
+ address = context->address;
+ count = 0;
+
+ rc = download_initialize_partition(context, address, GPT_BLOCK_SIZE);
+ if (rc < 0)
+ goto error;
+
+ while (count < length) {
+ chunk = (length - count) < GPT_BLOCK_SIZE ? (length - count) : GPT_BLOCK_SIZE;
+
+ /* Writes are always block-aligned: fill the rest with zeros */
+ if (chunk < GPT_BLOCK_SIZE)
+ memset(buffer, 0, GPT_BLOCK_SIZE);
+
+ p = (unsigned char *) buffer;
+ i = 0;
+
+ while (i < chunk) {
+ rc = read(fd, p, chunk - i);
+ if (rc <= 0)
+ goto error;
+
+ i += rc;
+ p += rc;
+ }
+
+ rc = download_write(context, buffer, address, GPT_BLOCK_SIZE);
+ if (rc < 0)
+ goto error;
+
+ count += chunk;
+ address++;
+ }
+
+ rc = 0;
+ goto complete;
+
+error:
+ rc = -1;
+
+complete:
+ if (buffer != NULL)
+ free(buffer);
+
+ if (fd >= 0)
+ close(fd);
+
+ return rc;
+}
+
+int dump(struct context *context)
+{
+ void *buffer = NULL;
+ unsigned int address;
+ unsigned int length;
+ unsigned int count;
+ unsigned int chunk;
+ unsigned char *p;
+ unsigned int c;
+ int fd = -1;
+ int rc;
+
+ if (context == NULL || context->filename == NULL)
+ return -1;
+
+ rc = gpt_partition_find(context);
+ if (rc < 0) {
+ fprintf(stderr, "Finding GPT partition failed\n");
+ goto error;
+ }
+
+ fd = open(context->filename, O_WRONLY | O_CREAT | O_TRUNC, 644);
+ if (fd < 0) {
+ fprintf(stderr, "Opening file failed\n");
+ goto error;
+ }
+
+ if (context->verbose)
+ printf("Dumping partition with address %d blocks and length %d bytes\n", context->address, context->length);
+
+ buffer = calloc(1, GPT_BLOCK_SIZE);
+
+ address = context->address;
+ length = context->length;
+ count = 0;
+
+ rc = download_ready_read(context, address, GPT_BLOCK_SIZE);
+ if (rc < 0)
+ goto error;
+
+ while (count < length) {
+ chunk = (length - count) < GPT_BLOCK_SIZE ? (length - count) : GPT_BLOCK_SIZE;
+
+ if (chunk != GPT_BLOCK_SIZE) {
+ rc = download_ready_read(context, address, chunk);
+ if (rc < 0)
+ goto error;
+ }
+
+ rc = download_read(context, buffer, chunk);
+ if (rc < 0)
+ goto error;
+
+ c = 0;
+ p = (unsigned char *) buffer;
+
+ while (c < chunk) {
+ rc = write(fd, p, chunk - c);
+ if (rc <= 0)
+ goto error;
+
+ c += rc;
+ p += rc;
+ }
+
+ count += chunk;
+ address++;
+ }
+
+ rc = 0;
+ goto complete;
+
+error:
+ rc = -1;
+
+complete:
+ if (buffer != NULL)
+ free(buffer);
+
+ if (fd >= 0)
+ close(fd);
+
+ return rc;
+}
+
+static int partitions(struct context *context)
+{
+ int rc;
+
+ rc = gpt_partitions_print(context);
+ if (rc < 0)
+ return -1;
+
+ return 0;
+}
+
+int main(int argc, char *argv[])
+{
+ struct context context = { 0 };
+ int rc;
+
+ rc = arguments_parse(&context, argc, argv);
+ if (rc < 0) {
+ usage_print();
+ goto error;
+ }
+
+ if (context.verbose)
+ printf("Finding and opening USB device\n");
+
+ rc = usb_open(&context);
+ if (rc < 0)
+ goto error;
+
+ if (context.verbose)
+ printf("Notifying download\n");
+
+ rc = download_notify_start_dl(&context);
+ if (rc < 0)
+ goto error;
+
+ switch (context.operation) {
+ case 'r':
+ printf("Rebooting device\n");
+
+ rc = download_reset(&context);
+ if (rc < 0) {
+ fprintf(stderr, "Rebooting device failed\n");
+ goto error;
+ }
+
+ break;
+ case 'e':
+ printf("Erasing partition %s\n", context.partition);
+
+ /* Give the user a chance to hit Ctrl+C */
+ sleep(1);
+
+ rc = erase(&context);
+ if (rc < 0) {
+ fprintf(stderr, "Erasing partition failed\n");
+ goto error;
+ }
+
+ break;
+ case 'f':
+ printf("Flashing %s to partition %s\n", context.filename, context.partition);
+
+ rc = flash(&context);
+ if (rc < 0) {
+ fprintf(stderr, "Flashing partition failed\n");
+ goto error;
+ }
+
+ break;
+ case 'd':
+ printf("Dumping partition %s to %s\n", context.partition, context.filename);
+
+ rc = dump(&context);
+ if (rc < 0) {
+ fprintf(stderr, "Dumping partition failed\n");
+ goto error;
+ }
+
+ break;
+ case 'p':
+ printf("Reading partitions table\n");
+
+ rc = partitions(&context);
+ if (rc < 0) {
+ fprintf(stderr, "Reading partitions table failed\n");
+ goto error;
+ }
+
+ break;
+ default:
+ goto error;
+ }
+
+ printf("Done!\n");
+
+ rc = 0;
+ goto complete;
+
+error:
+ rc = 1;
+
+complete:
+ usb_close(&context);
+
+ return rc;
+}
diff --git a/src/lg-downloader.h b/src/lg-downloader.h
new file mode 100644
index 0000000..a645fae
--- /dev/null
+++ b/src/lg-downloader.h
@@ -0,0 +1,39 @@
+/*
+ * Copyright (C) 2016 Paul Kocialkowski <contact@paulk.fr>
+ *
+ * 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/>.
+ */
+
+#ifndef _LG_DOWNLOADER_H_
+#define _LG_DOWNLOADER_H_
+
+#include <libusb.h>
+
+/*
+ * Structures
+ */
+
+struct context {
+ struct libusb_device_handle *usb_handle;
+ int verbose;
+ char operation;
+
+ char *partition;
+ char *filename;
+
+ unsigned int address;
+ unsigned int length;
+};
+
+#endif
diff --git a/src/usb.c b/src/usb.c
new file mode 100644
index 0000000..8e87474
--- /dev/null
+++ b/src/usb.c
@@ -0,0 +1,170 @@
+/*
+ * Copyright (C) 2016 Paul Kocialkowski <contact@paulk.fr>
+ *
+ * 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 <stdlib.h>
+#include <stdio.h>
+#include <libusb.h>
+
+#include "lg-downloader.h"
+#include "usb.h"
+
+int usb_open(struct context *context)
+{
+ struct libusb_device_handle *handle;
+ int rc;
+
+ if (context == NULL)
+ return -1;
+
+ rc = libusb_init(NULL);
+ if (rc < 0) {
+ fprintf(stderr, "Initializing libusb failed\n");
+ return -1;
+ }
+
+ handle = libusb_open_device_with_vid_pid(NULL, USB_VENDOR_LG, USB_DEVICE_DOWNLOAD);
+ if (handle == NULL) {
+ fprintf(stderr, "Finding and opening USB device failed\n");
+ fprintf(stderr, "\nDevice might be missing or you may not have the rights to do this\n");
+ goto error;
+ }
+
+ libusb_detach_kernel_driver(handle, USB_INTERFACE);
+
+ rc = libusb_set_configuration(handle, USB_CONFIGURATION);
+ if (rc < 0) {
+ fprintf(stderr, "Setting USB configuration failed\n");
+ goto error;
+ }
+
+ rc = libusb_claim_interface(handle, USB_INTERFACE);
+ if (rc < 0) {
+ fprintf(stderr, "Claiming USB interface failed\n");
+ goto error;
+ }
+
+ context->usb_handle = handle;
+
+ rc = 0;
+ goto complete;
+
+error:
+ rc = -1;
+
+ if (handle != NULL)
+ libusb_close(handle);
+
+ libusb_exit(NULL);
+
+complete:
+ return rc;
+}
+
+void usb_close(struct context *context)
+{
+ if (context == NULL)
+ return;
+
+ if (context->usb_handle != NULL) {
+ libusb_release_interface(context->usb_handle, USB_INTERFACE);
+ libusb_close(context->usb_handle);
+
+ context->usb_handle = NULL;
+
+ libusb_exit(NULL);
+ }
+}
+
+int usb_send(struct context *context, const void *data, unsigned int size)
+{
+ unsigned int count;
+ unsigned int chunk;
+ unsigned char *p;
+ int transferred;
+ int rc;
+
+ if (context == NULL)
+ return -1;
+
+ count = 0;
+ p = (unsigned char *) data;
+
+ while (count < size) {
+ chunk = (size - count) < USB_SEND_CHUNK ? (size - count) : USB_SEND_CHUNK;
+
+ rc = libusb_bulk_transfer(context->usb_handle, USB_ENDPOINT_OUT, p, chunk, &transferred, USB_TIMEOUT);
+ if (rc < 0 || transferred <= 0) {
+ fprintf(stderr, "sBulk USB transfer failed\n");
+ return -1;
+ }
+
+ count += transferred;
+ p += transferred;
+ }
+
+ return 0;
+}
+
+int usb_recv(struct context *context, void *data, unsigned int size)
+{
+ unsigned int count;
+ unsigned int chunk;
+ unsigned char *p;
+ int transferred;
+ int rc;
+
+ if (context == NULL)
+ return -1;
+
+ count = 0;
+ p = (unsigned char *) data;
+
+ while (count < size) {
+ chunk = (size - count) < USB_RECV_CHUNK ? (size - count) : USB_RECV_CHUNK;
+
+ rc = libusb_bulk_transfer(context->usb_handle, USB_ENDPOINT_IN, p, chunk, &transferred, USB_TIMEOUT);
+ if (rc < 0 || transferred <= 0) {
+ fprintf(stderr, "Bulk USB transfer failed\n");
+ return -1;
+ }
+
+ count += transferred;
+ p += transferred;
+ }
+
+ return 0;
+}
+
+int usb_recv_available(struct context *context, void *data, unsigned int size)
+{
+ unsigned int chunk;
+ int transferred;
+ int rc;
+
+ if (context == NULL)
+ return -1;
+
+ chunk = size < USB_RECV_CHUNK ? size : USB_RECV_CHUNK;
+
+ rc = libusb_bulk_transfer(context->usb_handle, USB_ENDPOINT_IN, data, chunk, &transferred, USB_TIMEOUT);
+ if (rc < 0 || transferred <= 0) {
+ fprintf(stderr, "Bulk USB transfer failed\n");
+ return -1;
+ }
+
+ return transferred;
+}
diff --git a/src/usb.h b/src/usb.h
new file mode 100644
index 0000000..0666ec4
--- /dev/null
+++ b/src/usb.h
@@ -0,0 +1,50 @@
+/*
+ * Copyright (C) 2016 Paul Kocialkowski <contact@paulk.fr>
+ *
+ * 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/>.
+ */
+
+#ifndef _USB_H_
+#define _USB_H_
+
+#include "lg-downloader.h"
+
+/*
+ * Values
+ */
+
+#define USB_VENDOR_LG 0x1004
+#define USB_DEVICE_DOWNLOAD 0x6000
+
+#define USB_CONFIGURATION 1
+#define USB_INTERFACE 1
+
+#define USB_ENDPOINT_OUT 0x02
+#define USB_ENDPOINT_IN 0x83
+
+#define USB_TIMEOUT 300
+#define USB_SEND_CHUNK 128
+#define USB_RECV_CHUNK 512
+
+/*
+ * Functions
+ */
+
+int usb_open(struct context *context);
+void usb_close(struct context *context);
+int usb_send(struct context *context, const void *data, unsigned int size);
+int usb_recv(struct context *context, void *data, unsigned int size);
+int usb_recv_available(struct context *context, void *data, unsigned int size);
+
+#endif
generated by cgit v1.2.3 (git 2.25.1) at 2024-05-21 20:24:05 +0000