rbutil/ipodpatcher/ipodio-posix.c

   1 /***************************************************************************
   2  *             __________               __   ___.
   3  *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
   4  *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
   5  *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
   6  *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
   7  *                     \/            \/     \/    \/            \/
   8  * $Id$
   9  *
  10  * Copyright (C) 2006-2007 Dave Chapman
  11  *
  12  * This program is free software; you can redistribute it and/or
  13  * modify it under the terms of the GNU General Public License
  14  * as published by the Free Software Foundation; either version 2
  15  * of the License, or (at your option) any later version.
  16  *
  17  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
  18  * KIND, either express or implied.
  19  *
  20  ****************************************************************************/
  21
  22 #include <stdio.h>
  23 #include <unistd.h>
  24 #include <fcntl.h>
  25 #include <string.h>
  26 #include <stdlib.h>
  27 #include <sys/types.h>
  28 #include <sys/stat.h>
  29 #include <sys/ioctl.h>
  30 #include <errno.h>
  31
  32 #include "ipodio.h"
  33
  34 #if defined(linux) || defined (__linux)
  35 #include <sys/mount.h>
  36 #include <linux/hdreg.h>
  37 #define IPOD_SECTORSIZE_IOCTL BLKSSZGET
  38
  39 static void get_geometry(struct ipod_t* ipod)
  40 {
  41     struct hd_geometry geometry;
  42
  43     if (!ioctl(ipod->dh, HDIO_GETGEO, &geometry)) {
  44         /* never use geometry.cylinders - it is truncated */
  45         ipod->num_heads = geometry.heads;
  46         ipod->sectors_per_track = geometry.sectors;
  47     } else {
  48         ipod->num_heads = 0;
  49         ipod->sectors_per_track = 0;
  50     }
  51 }
  52
  53 #elif defined(__FreeBSD__) || defined(__NetBSD__) || defined(__OpenBSD__) \
  54       || defined(__bsdi__) || defined(__DragonFly__)
  55 #include <sys/disk.h>
  56 #define IPOD_SECTORSIZE_IOCTL DIOCGSECTORSIZE
  57
  58 /* TODO: Implement this function for BSD */
  59 static void get_geometry(struct ipod_t* ipod)
  60 {
  61     /* Are these universal for all ipods? */
  62     ipod->num_heads = 255;
  63     ipod->sectors_per_track = 63;
  64 }
  65
  66 #elif defined(__APPLE__) && defined(__MACH__)
  67 #include <sys/disk.h>
  68 #define IPOD_SECTORSIZE_IOCTL DKIOCGETBLOCKSIZE
  69
  70 /* TODO: Implement this function for Mac OS X */
  71 static void get_geometry(struct ipod_t* ipod)
  72 {
  73     /* Are these universal for all ipods? */
  74     ipod->num_heads = 255;
  75     ipod->sectors_per_track = 63;
  76 }
  77
  78 #else
  79     #error No sector-size detection implemented for this platform
  80 #endif
  81
  82 #if defined(__APPLE__) && defined(__MACH__)
  83 static int ipod_unmount(struct ipod_t* ipod)
  84 {
  85     char cmd[4096];
  86     int res;
  87
  88     sprintf(cmd, "/usr/sbin/diskutil unmount \"%ss2\"",ipod->diskname);
  89     fprintf(stderr,"[INFO] ");
  90     res = system(cmd);
  91
  92     if (res==0) {
  93         return 0;
  94     } else {
  95         perror("Unmount failed");
  96         return -1;
  97     }
  98 }
  99 #endif
 100
 101 void print_error(char* msg)
 102 {
 103     perror(msg);
 104 }
 105
 106 int ipod_open(struct ipod_t* ipod, int silent)
 107 {
 108     ipod->dh=open(ipod->diskname,O_RDONLY);
 109     if (ipod->dh < 0) {
 110         if (!silent) perror(ipod->diskname);
 111         if(errno == EACCES) return -2;
 112         else return -1;
 113     }
 114
 115     /* Read information about the disk */
 116
 117     if(ioctl(ipod->dh,IPOD_SECTORSIZE_IOCTL,&ipod->sector_size) < 0) {
 118         ipod->sector_size=512;
 119         if (!silent) {
 120             fprintf(stderr,"[ERR] ioctl() call to get sector size failed, defaulting to %d\n"
 121                    ,ipod->sector_size);
 122         }
 123     }
 124
 125     get_geometry(ipod);
 126
 127     return 0;
 128 }
 129
 130
 131 int ipod_reopen_rw(struct ipod_t* ipod)
 132 {
 133 #if defined(__APPLE__) && defined(__MACH__)
 134     if (ipod_unmount(ipod) < 0)
 135         return -1;
 136 #endif
 137
 138     close(ipod->dh);
 139     ipod->dh=open(ipod->diskname,O_RDWR);
 140     if (ipod->dh < 0) {
 141         perror(ipod->diskname);
 142         return -1;
 143     }
 144     return 0;
 145 }
 146
 147 int ipod_close(struct ipod_t* ipod)
 148 {
 149     close(ipod->dh);
 150     return 0;
 151 }
 152
 153 int ipod_alloc_buffer(unsigned char** sectorbuf, int bufsize)
 154 {
 155     *sectorbuf=malloc(bufsize);
 156     if (*sectorbuf == NULL) {
 157         return -1;
 158     }
 159     return 0;
 160 }
 161
 162 int ipod_seek(struct ipod_t* ipod, unsigned long pos)
 163 {
 164     off_t res;
 165
 166     res = lseek(ipod->dh, pos, SEEK_SET);
 167
 168     if (res == -1) {
 169        return -1;
 170     }
 171     return 0;
 172 }
 173
 174 ssize_t ipod_read(struct ipod_t* ipod, unsigned char* buf, int nbytes)
 175 {
 176     return read(ipod->dh, buf, nbytes);
 177 }
 178
 179 ssize_t ipod_write(struct ipod_t* ipod, unsigned char* buf, int nbytes)
 180 {
 181     return write(ipod->dh, buf, nbytes);
 182 }