rbutil/sansapatcher/sansaio-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 #if defined(linux) || defined (__linux)
  33 #include <sys/mount.h>
  34 #define SANSA_SECTORSIZE_IOCTL BLKSSZGET
  35 #elif defined(__FreeBSD__) || defined(__NetBSD__) || defined(__OpenBSD__) \
  36       || defined(__bsdi__) || defined(__DragonFly__)
  37 #include <sys/disk.h>
  38 #define SANSA_SECTORSIZE_IOCTL DIOCGSECTORSIZE
  39 #elif defined(__APPLE__) && defined(__MACH__)
  40 #include <sys/disk.h>
  41 #define SANSA_SECTORSIZE_IOCTL DKIOCGETBLOCKSIZE
  42 #else
  43     #error No sector-size detection implemented for this platform
  44 #endif
  45
  46 #include "sansaio.h"
  47
  48 #if defined(__APPLE__) && defined(__MACH__)
  49 static int sansa_unmount(struct sansa_t* sansa)
  50 {
  51     char cmd[4096];
  52     int res;
  53
  54     sprintf(cmd, "/usr/sbin/diskutil unmount \"%ss1\"",sansa->diskname);
  55     fprintf(stderr,"[INFO] ");
  56     res = system(cmd);
  57
  58     if (res==0) {
  59         return 0;
  60     } else {
  61         perror("Unmount failed");
  62         return -1;
  63     }
  64 }
  65 #endif
  66
  67
  68 void sansa_print_error(char* msg)
  69 {
  70     perror(msg);
  71 }
  72
  73 int sansa_open(struct sansa_t* sansa, int silent)
  74 {
  75     sansa->dh=open(sansa->diskname,O_RDONLY);
  76     if (sansa->dh < 0) {
  77         if (!silent) perror(sansa->diskname);
  78         if(errno == EACCES) return -2;
  79         else return -1;
  80     }
  81
  82     if(ioctl(sansa->dh,SANSA_SECTORSIZE_IOCTL,&sansa->sector_size) < 0) {
  83         sansa->sector_size=512;
  84         if (!silent) {
  85             fprintf(stderr,"[ERR] ioctl() call to get sector size failed, defaulting to %d\n"
  86                    ,sansa->sector_size);
  87         }
  88     }
  89     return 0;
  90 }
  91
  92
  93 int sansa_reopen_rw(struct sansa_t* sansa)
  94 {
  95 #if defined(__APPLE__) && defined(__MACH__)
  96     if (sansa_unmount(sansa) < 0)
  97         return -1;
  98 #endif
  99
 100     close(sansa->dh);
 101     sansa->dh=open(sansa->diskname,O_RDWR);
 102     if (sansa->dh < 0) {
 103         perror(sansa->diskname);
 104         return -1;
 105     }
 106     return 0;
 107 }
 108
 109 int sansa_close(struct sansa_t* sansa)
 110 {
 111     close(sansa->dh);
 112     return 0;
 113 }
 114
 115 int sansa_alloc_buffer(unsigned char** sectorbuf, int bufsize)
 116 {
 117     *sectorbuf=malloc(bufsize);
 118     if (*sectorbuf == NULL) {
 119         return -1;
 120     }
 121     return 0;
 122 }
 123
 124 int sansa_seek(struct sansa_t* sansa, loff_t pos)
 125 {
 126     off_t res;
 127
 128     res = lseek64(sansa->dh, pos, SEEK_SET);
 129
 130     if (res == -1) {
 131        return -1;
 132     }
 133     return 0;
 134 }
 135
 136 int sansa_read(struct sansa_t* sansa, unsigned char* buf, int nbytes)
 137 {
 138     return read(sansa->dh, buf, nbytes);
 139 }
 140
 141 int sansa_write(struct sansa_t* sansa, unsigned char* buf, int nbytes)
 142 {
 143     return write(sansa->dh, buf, nbytes);
 144 }