imported source from https://github.com/justgr/arnova-tools
[android-command-line.git] / rkflashtool / rkflashtool.c
diff --git a/rkflashtool/rkflashtool.c b/rkflashtool/rkflashtool.c
new file mode 100644 (file)
index 0000000..52ce5ef
--- /dev/null
@@ -0,0 +1,189 @@
+/* rkflashtool - for RK2808, RK2818 and RK2918 based tablets
+ *
+ * Copyright (C) 2011 Ivo van Poorten   (complete rewrite for libusb)
+ * Copyright (C) 2010 FUKAUMI Naoki     (reverse engineering of protocol)
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+ * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+ * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ *
+ * Build with:
+ *
+ *      gcc -o rkflashtool rkflashtool.c -lusb-1.0 -O2 -W -Wall -s
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <stdarg.h>
+#include <stdint.h>
+#include <unistd.h>
+#include <string.h>
+#include <libusb-1.0/libusb.h>
+
+#define RKFLASHTOOL_VERSION     2
+
+#define RKFT_BLOCKSIZE  0x4000                  /* must be multiple of 512 */
+#define RKFT_OFF_INCR   (RKFT_BLOCKSIZE>>9)
+
+#define RKFT_CID        4
+#define RKFT_FLAG       12
+#define RKFT_COMMAND    13
+#define RKFT_OFFSET     17
+#define RKFT_SIZE       23
+
+#define SETBE32(a, v) ((uint8_t*)a)[3] =  v      & 0xff; \
+                      ((uint8_t*)a)[2] = (v>>8 ) & 0xff; \
+                      ((uint8_t*)a)[1] = (v>>16) & 0xff; \
+                      ((uint8_t*)a)[0] = (v>>24) & 0xff
+
+static uint8_t cmd[31] = { 'U', 'S', 'B', 'C', };
+static uint8_t res[13];
+
+static uint8_t buf[RKFT_BLOCKSIZE], cid;
+static int tmp;
+
+static const char *const strings[2] = { "info", "fatal" };
+
+static void info_and_fatal(const int s, char *f, ...) {
+    va_list ap;
+    va_start(ap,f);
+    fprintf(stderr, "rkflashtool: %s: ", strings[s]);
+    vfprintf(stderr, f, ap);
+    va_end(ap);
+    if (s) exit(s);
+}
+
+#define info(...)   info_and_fatal(0, __VA_ARGS__)
+#define fatal(...)  info_and_fatal(1, __VA_ARGS__)
+
+static void usage(void) {
+    fatal("usage:\n"
+          "\trkflashtool b                   \treboot device\n"
+          "\trkflashtool r offset size >file \tread flash\n"
+          "\trkflashtool w offset size <file \twrite flash\n\n"
+          "\toffset and size are in units of 512 bytes\n");
+}
+
+static void send_cmd(libusb_device_handle *h, int e, uint8_t flag,
+                     uint32_t command, uint32_t offset, uint8_t size) {
+    cmd[RKFT_CID ] = cid++;
+    cmd[RKFT_FLAG] = flag;
+    cmd[RKFT_SIZE] = size;
+
+    SETBE32(&cmd[RKFT_COMMAND], command);
+    SETBE32(&cmd[RKFT_OFFSET ], offset );
+
+    libusb_bulk_transfer(h, e|LIBUSB_ENDPOINT_OUT, cmd, sizeof(cmd), &tmp, 0);
+}
+
+#define send_buf(h,e,s) libusb_bulk_transfer(h, e|LIBUSB_ENDPOINT_OUT, \
+                                             buf, s, &tmp, 0)
+
+#define recv_res(h,e) libusb_bulk_transfer(h, e|LIBUSB_ENDPOINT_IN, \
+                                           res, sizeof(res), &tmp, 0)
+
+#define recv_buf(h,e,s) libusb_bulk_transfer(h, e|LIBUSB_ENDPOINT_IN, \
+                                             buf, s, &tmp, 0)
+
+#define NEXT do { argc--;argv++; }while(0)
+
+int main(int argc, char **argv) {
+    libusb_context *c;
+    libusb_device_handle *h;
+    int offset = 0, size = 0;
+    char action;
+
+    NEXT; if (!argc) usage();
+
+    action = **argv; NEXT;
+
+    switch(action) {
+    case 'b':           if (argc   ) usage(); break;
+    case 'r': case 'w': if (argc!=2) usage();
+        offset = strtoul(argv[0], NULL, 0);
+        size   = strtoul(argv[1], NULL, 0);
+        break;
+    default:
+        usage();
+    }
+
+    if (libusb_init(&c)) fatal("cannot init libusb\n");
+
+    libusb_set_debug(c, 3);
+
+    if (!(h = libusb_open_device_with_vid_pid(c, 0x2207, 0x290a)))
+        if (!(h = libusb_open_device_with_vid_pid(c, 0x2207, 0x281a)))
+            fatal("cannot open device\n");
+
+    if (libusb_kernel_driver_active(h, 0) == 1) {
+        info("kernel driver active\n");
+        if (!libusb_detach_kernel_driver(h, 0))
+            info("driver detached\n");
+    }
+
+    if (libusb_claim_interface(h, 0)<0) fatal("cannot claim interface\n");
+    info("interface claimed\n");
+
+    send_cmd(h, 2, 0x80, 0x00060000, 0x00000000, 0x00);        // INIT
+    recv_res(h, 1);
+    usleep(20*1000);
+
+    switch(action) {
+    case 'b':
+        info("rebooting device...\n");
+        send_cmd(h, 2, 0x00, 0x0006ff00, 0x00000000, 0x00);
+        recv_res(h, 1);
+        break;
+    case 'r':
+        while (size>0) {
+            info("reading flash memory at offset 0x%08x\n", offset);
+
+            send_cmd(h, 2, 0x80, 0x000a1400, offset, RKFT_OFF_INCR);
+            recv_buf(h, 1, RKFT_BLOCKSIZE);
+            recv_res(h, 1);
+
+            write(1, buf, RKFT_BLOCKSIZE);
+            offset += RKFT_OFF_INCR;
+            size   -= RKFT_OFF_INCR;
+        }
+        break;
+    case 'w':
+        while (size>0) {
+            info("writing flash memory at offset 0x%08x\n", offset);
+
+            memset(buf, 0, RKFT_BLOCKSIZE);
+            read(0, buf, RKFT_BLOCKSIZE);
+
+            send_cmd(h, 2, 0x80, 0x000a1500, offset, RKFT_OFF_INCR);
+            send_buf(h, 2, RKFT_BLOCKSIZE);
+            recv_res(h, 1);
+
+            offset += RKFT_OFF_INCR;
+            size   -= RKFT_OFF_INCR;
+        }
+        break;
+    default:
+        break;
+    }
+
+    libusb_release_interface(h, 0);
+    libusb_close(h);
+    libusb_exit(c);
+    return 0;
+}