summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/Makefile.am2
-rw-r--r--src/common.h7
-rw-r--r--src/download.c104
-rw-r--r--src/download.h24
-rw-r--r--src/idevicerestore.c61
5 files changed, 196 insertions, 2 deletions
diff --git a/src/Makefile.am b/src/Makefile.am
index 5fbcdef..93b0d98 100644
--- a/src/Makefile.am
+++ b/src/Makefile.am
@@ -18,6 +18,6 @@ AM_LDFLAGS =\
bin_PROGRAMS = idevicerestore
-idevicerestore_SOURCES = idevicerestore.c common.c tss.c img3.c ipsw.c normal.c dfu.c recovery.c restore.c asr.c libirecovery.c limera1n.c
+idevicerestore_SOURCES = idevicerestore.c common.c tss.c img3.c ipsw.c normal.c dfu.c recovery.c restore.c asr.c libirecovery.c limera1n.c download.c
idevicerestore_CFLAGS = $(AM_CFLAGS)
idevicerestore_LDFLAGS = $(AM_LDFLAGS) -lusb-1.0
diff --git a/src/common.h b/src/common.h
index 9a84188..19ce881 100644
--- a/src/common.h
+++ b/src/common.h
@@ -129,6 +129,7 @@ struct idevicerestore_device_t {
struct idevicerestore_client_t {
int flags;
plist_t tss;
+ plist_t version_data;
uint64_t ecid;
unsigned char* nonce;
int nonce_size;
@@ -183,6 +184,12 @@ int write_file(const char* filename, const void* data, size_t size);
char *generate_guid();
+#ifdef WIN32
+#define __mkdir(path, mode) mkdir(path)
+#else
+#define __mkdir(path, mode) mkdir(path, mode)
+#endif
+
extern struct idevicerestore_client_t* idevicerestore;
#ifdef __cplusplus
diff --git a/src/download.c b/src/download.c
new file mode 100644
index 0000000..211987a
--- /dev/null
+++ b/src/download.c
@@ -0,0 +1,104 @@
+/*
+ * download.c
+ * file download helper functions
+ *
+ * Copyright (c) 2012 Nikias Bassen. All Rights Reserved.
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library 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
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+ */
+#include <stdint.h>
+#include <stdlib.h>
+#include <string.h>
+#include <curl/curl.h>
+
+#include "download.h"
+
+typedef struct {
+ int length;
+ char* content;
+} curl_response;
+
+static size_t download_write_buffer_callback(char* data, size_t size, size_t nmemb, curl_response* response) {
+ size_t total = size * nmemb;
+ if (total != 0) {
+ response->content = realloc(response->content, response->length + total + 1);
+ memcpy(response->content + response->length, data, total);
+ response->content[response->length + total] = '\0';
+ response->length += total;
+ }
+ return total;
+}
+
+int download_to_buffer(const char* url, char** buf, uint32_t* length)
+{
+ curl_global_init(CURL_GLOBAL_ALL);
+ CURL* handle = curl_easy_init();
+ if (handle == NULL) {
+ error("ERROR: could not initialize CURL\n");
+ return -1;
+ }
+
+ curl_response response;
+ response.length = 0;
+ response.content = malloc(1);
+ response.content[0] = '\0';
+
+ curl_easy_setopt(handle, CURLOPT_WRITEFUNCTION, (curl_write_callback)&download_write_buffer_callback);
+ curl_easy_setopt(handle, CURLOPT_WRITEDATA, &response);
+ curl_easy_setopt(handle, CURLOPT_USERAGENT, "InetURL/1.0");
+ curl_easy_setopt(handle, CURLOPT_URL, url);
+
+ curl_easy_perform(handle);
+ curl_easy_cleanup(handle);
+
+ if (response.content) {
+ *length = response.length;
+ *buf = response.content;
+ }
+
+ curl_global_cleanup();
+
+ return 0;
+}
+
+int download_to_file(const char* url, const char* filename)
+{
+ curl_global_init(CURL_GLOBAL_ALL);
+ CURL* handle = curl_easy_init();
+ if (handle == NULL) {
+ error("ERROR: could not initialize CURL\n");
+ return -1;
+ }
+
+ FILE* f = fopen(filename, "wb");
+ if (!f) {
+ error("ERROR: cannot open '%s' for writing\n", filename);
+ return -1;
+ }
+
+ curl_easy_setopt(handle, CURLOPT_WRITEFUNCTION, (curl_write_callback)&fwrite);
+ curl_easy_setopt(handle, CURLOPT_WRITEDATA, f);
+ curl_easy_setopt(handle, CURLOPT_USERAGENT, "InetURL/1.0");
+ curl_easy_setopt(handle, CURLOPT_URL, url);
+
+ curl_easy_perform(handle);
+ curl_easy_cleanup(handle);
+
+ fclose(f);
+
+ curl_global_cleanup();
+
+ return 0;
+}
diff --git a/src/download.h b/src/download.h
new file mode 100644
index 0000000..4900b18
--- /dev/null
+++ b/src/download.h
@@ -0,0 +1,24 @@
+/*
+ * download.h
+ * file download helper functions (header file)
+ *
+ * Copyright (c) 2012 Nikias Bassen. All Rights Reserved.
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library 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
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+ */
+#include <stdint.h>
+
+int download_to_buffer(const char* url, char** buf, uint32_t* length);
+int download_to_file(const char* url, const char* filename);
diff --git a/src/idevicerestore.c b/src/idevicerestore.c
index 23e49f5..93c6766 100644
--- a/src/idevicerestore.c
+++ b/src/idevicerestore.c
@@ -34,11 +34,14 @@
#include "common.h"
#include "normal.h"
#include "restore.h"
+#include "download.h"
#include "recovery.h"
#include "idevicerestore.h"
#include "limera1n.h"
+#define VERSION_XML "cache/version.xml"
+
int use_apple_server;
static struct option longopts[] = {
@@ -70,6 +73,59 @@ void usage(int argc, char* argv[]) {
printf("\n");
}
+static int load_version_data(struct idevicerestore_client_t* client)
+{
+ if (!client) {
+ return -1;
+ }
+
+ struct stat fst;
+ int cached = 0;
+
+ if ((stat(VERSION_XML, &fst) < 0) || ((time(NULL)-86400) > fst.st_mtime)) {
+ char tmpf[256];
+ tmpf[0] = '\0';
+ if (!tmpnam(tmpf) || (tmpf[0] == '\0')) {
+ error("ERROR: Could not get temporary filename\n");
+ return -1;
+ }
+
+ if (download_to_file("http://itunes.com/version", tmpf) == 0) {
+ __mkdir("cache", 0755);
+ remove(VERSION_XML);
+ if (rename(tmpf, VERSION_XML) < 0) {
+ error("ERROR: Could not update '" VERSION_XML "'\n");
+ } else {
+ info("NOTE: Updated version data.\n");
+ }
+ }
+ } else {
+ cached = 1;
+ }
+
+ char *verbuf = NULL;
+ size_t verlen = 0;
+ read_file(VERSION_XML, (void**)&verbuf, &verlen);
+ if (!verbuf) {
+ error("ERROR: Could not load '" VERSION_XML "'.\n");
+ return -1;
+ }
+
+ client->version_data = NULL;
+ plist_from_xml(verbuf, verlen, &client->version_data);
+
+ if (!client->version_data) {
+ error("ERROR: Cannot parse plist data from '" VERSION_XML "'.\n");
+ return -1;
+ }
+
+ if (cached) {
+ info("NOTE: using cached version data\n");
+ }
+
+ return 0;
+}
+
int main(int argc, char* argv[]) {
int opt = 0;
int optindex = 0;
@@ -151,6 +207,9 @@ int main(int argc, char* argv[]) {
client->uuid = uuid;
client->ipsw = ipsw;
+ // update version data (from cache, or apple if too old)
+ load_version_data(client);
+
// check which mode the device is currently in so we know where to start
if (check_mode(client) < 0 || client->mode->index == MODE_UNKNOWN) {
error("ERROR: Unable to discover device mode. Please make sure a device is attached.\n");
@@ -428,7 +487,7 @@ int main(int argc, char* argv[]) {
if (bin) {
char zfn[512];
sprintf(zfn, "shsh/%lld-%s-%s.shsh", (long long int)client->ecid, client->device->product, client->version);
- mkdir("shsh", 0755);
+ __mkdir("shsh", 0755);
struct stat fst;
if (stat(zfn, &fst) != 0) {
gzFile zf = gzopen(zfn, "wb");