From 2e31eedce5bc7d2fc37f8cd6be1a7177dd7f6da7 Mon Sep 17 00:00:00 2001 From: Paul Kocialkowski Date: Wed, 27 Jan 2016 20:25:41 +0100 Subject: lg-downloader: Tool for flashing LG devices' internal storage from download mode Signed-off-by: Paul Kocialkowski --- src/Makefile.am | 6 + src/download.c | 224 +++++++++++++++++++++++++++++ src/download.h | 111 +++++++++++++++ src/gpt.c | 342 ++++++++++++++++++++++++++++++++++++++++++++ src/gpt.h | 75 ++++++++++ src/hdlc.c | 355 +++++++++++++++++++++++++++++++++++++++++++++ src/hdlc.h | 45 ++++++ src/lg-downloader.c | 404 ++++++++++++++++++++++++++++++++++++++++++++++++++++ src/lg-downloader.h | 39 +++++ src/usb.c | 170 ++++++++++++++++++++++ src/usb.h | 50 +++++++ 11 files changed, 1821 insertions(+) create mode 100644 src/Makefile.am create mode 100644 src/download.c create mode 100644 src/download.h create mode 100644 src/gpt.c create mode 100644 src/gpt.h create mode 100644 src/hdlc.c create mode 100644 src/hdlc.h create mode 100644 src/lg-downloader.c create mode 100644 src/lg-downloader.h create mode 100644 src/usb.c create mode 100644 src/usb.h (limited to 'src') 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 + * + * 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 . + */ + +#include +#include +#include + +#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 + * + * 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 . + */ + +#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 + * + * 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 . + */ + +#include +#include +#include +#include + +#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 + * + * 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 . + */ + +#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 + * + * 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 . + */ + +#include +#include +#include +#include + +#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 + * + * 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 . + */ + +#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 + * + * 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 . + */ + +#include +#include +#include +#include +#include + +#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 + * + * 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 . + */ + +#ifndef _LG_DOWNLOADER_H_ +#define _LG_DOWNLOADER_H_ + +#include + +/* + * 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 + * + * 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 . + */ + +#include +#include +#include + +#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 + * + * 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 . + */ + +#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 -- cgit v1.2.3