1 files changed, 409 insertions, 0 deletions

diff --git a/utils/ipodpatcher/ipodio-posix.c b/utils/ipodpatcher/ipodio-posix.c
new file mode 100644
index 0000000000..9b386d994f
--- /dev/null
+++ b/utils/ipodpatcher/ipodio-posix.c
@@ -0,0 +1,409 @@
+/***************************************************************************
+ *             __________               __   ___.
+ *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
+ *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
+ *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
+ *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
+ *                     \/            \/     \/    \/            \/
+ * $Id$
+ *
+ * Copyright (C) 2006-2007 Dave Chapman
+ *
+ * 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 software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
+ * KIND, either express or implied.
+ *
+ ****************************************************************************/
+
+#if !defined(_WIN32) /* all non-Windows platforms are considered POSIX. */
+
+#include <stdio.h>
+#include <unistd.h>
+#include <fcntl.h>
+#include <string.h>
+#include <stdlib.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <sys/ioctl.h>
+#include <errno.h>
+
+#include "ipodio.h"
+
+#if defined(linux) || defined (__linux)
+#include <sys/mount.h>
+#include <linux/hdreg.h>
+#include <scsi/scsi_ioctl.h>
+#include <scsi/sg.h>
+
+#define IPOD_SECTORSIZE_IOCTL BLKSSZGET
+
+static void get_geometry(struct ipod_t* ipod)
+{
+    struct hd_geometry geometry;
+
+    if (!ioctl(ipod->dh, HDIO_GETGEO, &geometry)) {
+        /* never use geometry.cylinders - it is truncated */
+        ipod->num_heads = geometry.heads;
+        ipod->sectors_per_track = geometry.sectors;
+    } else {
+        ipod->num_heads = 0;
+        ipod->sectors_per_track = 0;
+    }
+}
+
+/* Linux SCSI Inquiry code based on the documentation and example code from
+   http://www.ibm.com/developerworks/linux/library/l-scsi-api/index.html
+*/
+
+int ipod_scsi_inquiry(struct ipod_t* ipod, int page_code,
+                      unsigned char* buf, int bufsize)
+{
+    unsigned char cdb[6];
+    struct sg_io_hdr hdr;
+    unsigned char sense_buffer[255];
+
+    memset(&hdr, 0, sizeof(hdr));
+
+    hdr.interface_id = 'S'; /* this is the only choice we have! */
+    hdr.flags = SG_FLAG_LUN_INHIBIT; /* this would put the LUN to 2nd byte of cdb*/
+
+    /* Set xfer data */
+    hdr.dxferp = buf;
+    hdr.dxfer_len = bufsize;
+
+    /* Set sense data */
+    hdr.sbp = sense_buffer;
+    hdr.mx_sb_len = sizeof(sense_buffer);
+
+    /* Set the cdb format */
+    cdb[0] = 0x12;
+    cdb[1] = 1;   /* Enable Vital Product Data (EVPD) */
+    cdb[2] = page_code & 0xff;
+    cdb[3] = 0;
+    cdb[4] = 0xff;
+    cdb[5] = 0; /* For control filed, just use 0 */
+
+    hdr.dxfer_direction = SG_DXFER_FROM_DEV;
+    hdr.cmdp = cdb;
+    hdr.cmd_len = 6;
+
+    int ret = ioctl(ipod->dh, SG_IO, &hdr);
+
+    if (ret < 0) {
+        return -1;
+    } else {
+        return 0;
+    }
+}
+
+#elif defined(__FreeBSD__) || defined(__NetBSD__) || defined(__OpenBSD__) \
+      || defined(__bsdi__) || defined(__DragonFly__)
+#include <sys/disk.h>
+#define IPOD_SECTORSIZE_IOCTL DIOCGSECTORSIZE
+
+/* TODO: Implement this function for BSD */
+static void get_geometry(struct ipod_t* ipod)
+{
+    /* Are these universal for all ipods? */
+    ipod->num_heads = 255;
+    ipod->sectors_per_track = 63;
+}
+
+int ipod_scsi_inquiry(struct ipod_t* ipod, int page_code,
+                      unsigned char* buf, int bufsize)
+{
+    /* TODO: Implement for BSD */
+    (void)ipod;
+    (void)page_code;
+    (void)buf;
+    (void)bufsize;
+    return -1;
+}
+
+#elif defined(__APPLE__) && defined(__MACH__)
+/* OS X IOKit includes don't like VERSION being defined! */
+#undef VERSION
+#include <sys/disk.h>
+#include <CoreFoundation/CoreFoundation.h>
+#include <IOKit/IOKitLib.h>
+#include <IOKit/scsi/SCSITaskLib.h>
+#include <IOKit/scsi/SCSICommandOperationCodes.h>
+#define IPOD_SECTORSIZE_IOCTL DKIOCGETBLOCKSIZE
+
+/* TODO: Implement this function for Mac OS X */
+static void get_geometry(struct ipod_t* ipod)
+{
+    /* Are these universal for all ipods? */
+    ipod->num_heads = 255;
+    ipod->sectors_per_track = 63;
+}
+
+int ipod_scsi_inquiry(struct ipod_t* ipod, int page_code,
+                      unsigned char* buf, int bufsize)
+{
+    /* OS X doesn't allow to simply send out a SCSI inquiry request but
+     * requires registering an interface handler first.
+     * Currently this is done on each inquiry request which is somewhat
+     * inefficient but the current ipodpatcher API doesn't really fit here.
+     * Based on the documentation in Apple's document
+     * "SCSI Architecture Model Device Interface Guide".
+     *
+     * WARNING: this code currently doesn't take the selected device into
+     *          account. It simply looks for an Ipod on the system and uses
+     *          the first match.
+     */
+    (void)ipod;
+    int result = 0;
+    /* first, create a dictionary to match the device. This is needed to get the
+     * service. */
+    CFMutableDictionaryRef match_dict;
+    match_dict = CFDictionaryCreateMutable(kCFAllocatorDefault, 0, NULL, NULL);
+    if(match_dict == NULL)
+        return -1;
+
+    /* set value to match. In case of the Ipod this is "iPodUserClientDevice". */
+    CFMutableDictionaryRef sub_dict;
+    sub_dict = CFDictionaryCreateMutable(kCFAllocatorDefault, 0, NULL, NULL);
+    if(sub_dict == NULL)
+        return -1;
+    CFDictionarySetValue(sub_dict, CFSTR(kIOPropertySCSITaskDeviceCategory),
+                         CFSTR("iPodUserClientDevice"));
+    CFDictionarySetValue(match_dict, CFSTR(kIOPropertyMatchKey), sub_dict);
+
+    /* get an iterator for searching for the service. */
+    kern_return_t kr;
+    io_iterator_t iterator = IO_OBJECT_NULL;
+    /* get matching services from IO registry. Consumes one reference to
+     * the dictionary, so no need to release that. */
+    kr = IOServiceGetMatchingServices(kIOMasterPortDefault, match_dict, &iterator);
+
+    if(!iterator | (kr != kIOReturnSuccess))
+        return -1;
+
+    /* get interface and obtain exclusive access */
+    SInt32 score;
+    HRESULT herr;
+    kern_return_t err;
+    IOCFPlugInInterface **plugin_interface = NULL;
+    SCSITaskDeviceInterface **interface = NULL;
+    io_service_t device = IO_OBJECT_NULL;
+    device = IOIteratorNext(iterator);
+
+    err = IOCreatePlugInInterfaceForService(device, kIOSCSITaskDeviceUserClientTypeID,
+                                            kIOCFPlugInInterfaceID, &plugin_interface,
+                                            &score);
+
+    if(err != noErr) {
+        return -1;
+    }
+    /* query the plugin interface for task interface */
+    herr = (*plugin_interface)->QueryInterface(plugin_interface,
+                            CFUUIDGetUUIDBytes(kIOSCSITaskDeviceInterfaceID), (LPVOID*)&interface);
+    if(herr != S_OK) {
+        IODestroyPlugInInterface(plugin_interface);
+        return -1;
+    }
+
+    err = (*interface)->ObtainExclusiveAccess(interface);
+    if(err != noErr) {
+        (*interface)->Release(interface);
+        IODestroyPlugInInterface(plugin_interface);
+        return -1;
+    }
+
+    /* do the inquiry */
+    SCSITaskInterface **task = NULL;
+
+    task = (*interface)->CreateSCSITask(interface);
+    if(task != NULL) {
+        kern_return_t err;
+        SCSITaskStatus task_status;
+        IOVirtualRange* range;
+        SCSI_Sense_Data sense_data;
+        SCSICommandDescriptorBlock cdb;
+        UInt64 transfer_count = 0;
+        memset(buf, 0, bufsize);
+        /* allocate virtual range for buffer. */
+        range = (IOVirtualRange*) malloc(sizeof(IOVirtualRange));
+        memset(&sense_data, 0, sizeof(sense_data));
+        memset(cdb, 0, sizeof(cdb));
+        /* set up range. address is buffer address, length is request size. */
+        range->address = (IOVirtualAddress)buf;
+        range->length = bufsize;
+        /* setup CDB */
+        cdb[0] = 0x12; /* inquiry */
+        cdb[1] = 1;
+        cdb[2] = page_code;
+        cdb[4] = bufsize;
+
+        /* set cdb in task */
+        err = (*task)->SetCommandDescriptorBlock(task, cdb, kSCSICDBSize_6Byte);
+        if(err != kIOReturnSuccess) {
+            result = -1;
+            goto cleanup;
+        }
+        err = (*task)->SetScatterGatherEntries(task, range, 1, bufsize,
+                kSCSIDataTransfer_FromTargetToInitiator);
+        if(err != kIOReturnSuccess) {
+            result = -1;
+            goto cleanup;
+        }
+        /* set timeout */
+        err = (*task)->SetTimeoutDuration(task, 10000);
+        if(err != kIOReturnSuccess) {
+            result = -1;
+            goto cleanup;
+        }
+
+        /* request data */
+        err = (*task)->ExecuteTaskSync(task, &sense_data, &task_status, &transfer_count);
+        if(err != kIOReturnSuccess) {
+            result = -1;
+            goto cleanup;
+        }
+        /* cleanup */
+        free(range);
+
+        /* release task interface */
+        (*task)->Release(task);
+    }
+    else {
+        result = -1;
+    }
+cleanup:
+    /* cleanup interface */
+    (*interface)->ReleaseExclusiveAccess(interface);
+    (*interface)->Release(interface);
+    IODestroyPlugInInterface(plugin_interface);
+
+    return result;
+}
+
+#else
+    #error No sector-size detection implemented for this platform
+#endif
+
+#if defined(__APPLE__) && defined(__MACH__)
+static int ipod_unmount(struct ipod_t* ipod)
+{
+    char cmd[4096];
+    int res;
+
+    sprintf(cmd, "/usr/sbin/diskutil unmount \"%ss2\"",ipod->diskname);
+    fprintf(stderr,"[INFO] ");
+    res = system(cmd);
+
+    if (res==0) {
+        return 0;
+    } else {
+        perror("Unmount failed");
+        return -1;
+    }
+}
+#endif
+
+void ipod_print_error(char* msg)
+{
+    perror(msg);
+}
+
+int ipod_open(struct ipod_t* ipod, int silent)
+{
+    ipod->dh=open(ipod->diskname,O_RDONLY);
+    if (ipod->dh < 0) {
+        if (!silent) perror(ipod->diskname);
+        if(errno == EACCES) return -2;
+        else return -1;
+    }
+
+    /* Read information about the disk */
+
+    if(ioctl(ipod->dh,IPOD_SECTORSIZE_IOCTL,&ipod->sector_size) < 0) {
+        ipod->sector_size=512;
+        if (!silent) {
+            fprintf(stderr,"[ERR] ioctl() call to get sector size failed, defaulting to %d\n"
+                   ,ipod->sector_size);
+        }
+    }
+
+    get_geometry(ipod);
+
+    return 0;
+}
+
+
+int ipod_reopen_rw(struct ipod_t* ipod)
+{
+#if defined(__APPLE__) && defined(__MACH__)
+    if (ipod_unmount(ipod) < 0)
+        return -1;
+#endif
+
+    close(ipod->dh);
+    ipod->dh=open(ipod->diskname,O_RDWR);
+    if (ipod->dh < 0) {
+        perror(ipod->diskname);
+        return -1;
+    }
+    return 0;
+}
+
+int ipod_close(struct ipod_t* ipod)
+{
+    close(ipod->dh);
+    return 0;
+}
+
+int ipod_alloc_buffer(struct ipod_t* ipod, int bufsize)
+{
+    ipod->sectorbuf = malloc(bufsize);
+    if (ipod->sectorbuf== NULL) {
+        return -1;
+    }
+    return 0;
+}
+
+int ipod_dealloc_buffer(struct ipod_t* ipod)
+{
+    if (ipod->sectorbuf == NULL) {
+        return -1;
+    }
+    free(ipod->sectorbuf);
+    ipod->sectorbuf = NULL;
+    return 0;
+}
+
+int ipod_seek(struct ipod_t* ipod, unsigned long pos)
+{
+    off_t res;
+
+    res = lseek(ipod->dh, pos, SEEK_SET);
+
+    if (res == -1) {
+       return -1;
+    }
+    return 0;
+}
+
+ssize_t ipod_read(struct ipod_t* ipod, int nbytes)
+{
+    if(ipod->sectorbuf == NULL) {
+        return -1;
+    }
+    return read(ipod->dh, ipod->sectorbuf, nbytes);
+}
+
+ssize_t ipod_write(struct ipod_t* ipod, int nbytes)
+{
+    if(ipod->sectorbuf == NULL) {
+        return -1;
+    }
+    return write(ipod->dh, ipod->sectorbuf, nbytes);
+}
+
+#endif
+