** code courtesy of Christophe Fergeau **

	* src/itdb_itunesdb.c (mk_mhbd): write extended header needed
          for new iPod Nanos (3G Video) and iPod Classics.

	  src/itdb_device.c
	  src/itdb_device.h: Code to parse SysInfoExtended and SysInfo
	  for the FireWireGUID. You must either copy the iPod
	  description XML file to Device/SysInfoExtended or add a line
	  'FirewireGuid: 000A27....' to Device/SysInfo. You can get
	  your FirewireGuid by using the tests/test-fw-id test program.

	* src/sha1.c
	  src/sha1.h
	  src/itdb_sha1.c
	  src/itdb_sha1.h
	  New files for obscure hash generation code.

	* src/Makefile.am: added new files.

        * tests/test-checksum.c
	  tests/test-fw-id.c
	  tests/Makefile.am: test programs to retrieve the
	  FirewireGuid and calculate/write the obscure hash.



git-svn-id: https://gtkpod.svn.sf.net/svnroot/gtkpod/libgpod/trunk@1698 f01d2545-417e-4e96-918e-98f8d0dbbcb6

3 files changed, 253 insertions, 2 deletions

diff --git a/tests/Makefile.am b/tests/Makefile.am
index 4d3e89f..17c9cb7 100644
--- a/tests/Makefile.am
+++ b/tests/Makefile.am
@@ -1,5 +1,5 @@
 if HAVE_GDKPIXBUF
-TESTTHUMBS=test-thumbnails test-write-thumbnails test-photos get-timezone
+TESTTHUMBS=test-thumbnails test-write-thumbnails test-photos get-timezone 
 
 test_thumbnails_SOURCES = test-covers.c
 test_thumbnails_CFLAGS = $(AM_CFLAGS)
@@ -33,9 +33,14 @@ test_ls_LDADD =
 test_init_ipod_SOURCES = test-init-ipod.c
 test_init_ipod_LDADD = 
 
+test_checksum_SOURCES = test-checksum.c
+
+test_firewire_id_SOURCES = test-fw-id.c
+
 get_timezone_SOURCES = get-timezone.c
 
-noinst_PROGRAMS=test-itdb test-ls $(TESTTHUMBS) $(TESTTAGLIB) $(TESTMISC)
+noinst_PROGRAMS=test-itdb test-ls test-checksum test-firewire-id \
+	        $(TESTTHUMBS) $(TESTTAGLIB) $(TESTMISC)
 
 INCLUDES=$(LIBGPOD_CFLAGS) -I$(top_srcdir)/src -DPACKAGE_LOCALE_DIR=\""$(prefix)/$(DATADIRNAME)/locale"\"
 LIBS=$(LIBGPOD_LIBS) $(top_builddir)/src/libgpod.la
diff --git a/tests/test-checksum.c b/tests/test-checksum.c
new file mode 100644
index 0000000..3f9ef85
--- /dev/null
+++ b/tests/test-checksum.c
@@ -0,0 +1,181 @@
+/*
+|  Copyright (C) 2007 Christophe Fergeau <teuf@gnome.org>
+|
+|  The code in this file is heavily based on the proof-of-concept code
+|  written by wtbw
+|
+| 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.
+|   3. The name of the author may not be used to endorse or promote
+| products derived from this software without specific prior written
+| permission.
+|
+| 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.
+|
+|
+|  iTunes and iPod are trademarks of Apple
+|
+|  This product is not supported/written/published by Apple!
+*/
+#ifdef HAVE_CONFIG_H
+#  include <config.h>
+#endif
+
+#include <glib.h>
+
+#include <errno.h>
+#include <fcntl.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <sys/mman.h>
+#include <sys/stat.h>
+
+#include "itdb.h"
+#include "itdb_sha1.h"
+
+static unsigned char *
+calculate_db_checksum (const char *itdb_path, guint64 fwid)
+{
+    int fd;
+    struct stat stat_buf;
+    int result;
+    unsigned char *itdb_data;
+    unsigned char *checksum;
+
+    fd = open (itdb_path, O_RDONLY);
+    if (fd < 0) {
+	g_warning ("Couldn't open %s", itdb_path);
+	return NULL;
+    }
+
+    result = fstat (fd, &stat_buf);
+    if (result != 0) {
+	g_warning ("Couldn't stat %s", itdb_path);
+	close (fd);
+	return NULL;
+    }
+
+    if (stat_buf.st_size < 0x80) {
+	g_warning ("%s is too small", itdb_path);
+	close (fd);
+	return NULL;
+    }
+
+    itdb_data = mmap (NULL, stat_buf.st_size, 
+		      PROT_READ | PROT_WRITE, 
+		      MAP_PRIVATE, fd, 0);
+    if (itdb_data == MAP_FAILED) {
+	g_warning ("Couldn't mmap %s", itdb_path);
+	close (fd);
+	return NULL;
+    }
+
+    /* Those fields must be zero'ed out for the sha1 calculation */
+    memset(itdb_data+0x18, 0, 8);
+    memset(itdb_data+0x32, 0, 20);
+    memset(itdb_data+0x58, 0, 20);
+
+    checksum = itdb_compute_hash (fwid, itdb_data, stat_buf.st_size);
+
+    munmap (itdb_data, stat_buf.st_size);
+    close (fd);
+
+    return checksum;
+}
+
+static gboolean write_checksum_to_file (const char *path, 
+					const unsigned char *checksum,
+					size_t size)
+{
+    FILE *f;
+    int result;
+    size_t count;
+
+    f = fopen (path, "rb+");
+    if (f == NULL) {
+	return FALSE;
+    }
+    
+    result = fseek (f, 0x58, SEEK_SET);
+    if (result != 0) {
+	fclose (f);
+	return FALSE;
+    }
+    
+    count = fwrite (checksum, size, 1, f);
+    fclose (f);
+   
+    return (count == 1);
+}
+
+int
+main (int argc, char *argv[])
+{
+    char *itdb_path;
+    guint64 firewire_id;
+    unsigned char *checksum;
+    int i;
+
+    if (argc < 3) {
+	g_print ("Usage: %s <mountpoint> <firewire_id> [write]\n"
+		 "       firewire_id in hexadecimal, e.g. 00A745....\n"
+		 "       If any third argument is given, the checksum is written back to the iTunesDB.\n",
+		 g_basename(argv[0]));
+	exit (0);
+    }
+    
+    errno = 0;
+    firewire_id = strtoull (argv[2], NULL, 16);
+    if (errno != 0) {
+	g_warning ("Couldn't parse %s as a firewire ID\n", argv[2]);
+	exit (0);	
+    }
+
+    itdb_path = itdb_get_itunesdb_path (argv[1]);
+    checksum = calculate_db_checksum (itdb_path, firewire_id);
+
+    if (argc >= 4)
+    {
+	gboolean success;
+
+	success = write_checksum_to_file (itdb_path, checksum, 20);
+	if (!success)
+	{
+	    g_warning ("Couldn't write checksum back to file '%s'\n",
+		       argv[1]);
+	}
+    }
+
+    g_free (itdb_path);
+    
+    if (checksum == NULL) {
+	g_warning ("Couldn't compute checksum");
+	exit (0);
+    }
+
+    for (i = 0; i < strlen ((char *)checksum); i++) {
+	g_print ("%02x ", checksum[i]);
+    }
+    g_print ("\n");
+    g_free (checksum);
+
+    return 0;
+}
diff --git a/tests/test-fw-id.c b/tests/test-fw-id.c
new file mode 100644
index 0000000..9ec40e9
--- /dev/null
+++ b/tests/test-fw-id.c
@@ -0,0 +1,65 @@
+/*
+|   Copyright (C) 2007 Christophe Fergeau  <teuf@gnome.org>
+|
+|   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 2 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, write to the Free Software
+|   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+|
+|  iTunes and iPod are trademarks of Apple
+|
+|  This product is not supported/written/published by Apple!
+|
+*/
+
+#ifdef HAVE_CONFIG_H
+#  include <config.h>
+#endif
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <time.h>
+#include <libintl.h>
+
+#include "itdb.h"
+#include "itdb_device.h"
+
+int
+main (int argc, char *argv[])
+{
+    Itdb_Device *device;
+    char *fwid;
+
+    if (argc < 2) {
+        g_print ("Usage: %s <mountpoint>\n", g_basename (argv[0]));
+        return 1;
+
+    }
+
+    device = itdb_device_new ();
+    if (device == NULL) {
+        return 1;
+    }
+
+    itdb_device_set_mountpoint (device, argv[1]);
+
+    fwid = itdb_device_get_sysinfo (device, "FirewireGuid");
+    if (fwid == NULL) {
+        g_print ("Couldn't find firewire ID\n");
+        return 1;
+    } else {
+	g_print ("FireWire ID: %s\n", fwid);
+    }
+
+    return 0;
+}