FryingPan/DTLib/rSplitISOData.cpp

   1 /*
   2  * FryingPan - Amiga CD/DVD Recording Software (User Interface and supporting Libraries only)
   3  * Copyright (C) 2001-2011 Tomasz Wiszkowski Tomasz.Wiszkowski at gmail.com
   4  *
   5  * This program is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public License
   7  * as published by the Free Software Foundation; either version 2.1
   8  * of the License, or (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  18  */
  19
  20 #include "rSplitISOData.h"
  21 #include <libclass/dos.h>
  22 #include <libclass/exec.h>
  23 #include <libclass/utility.h>
  24 #include <exec/lists.h>
  25 #include <Generic/HookT.h>
  26 #include <Generic/XMLDocument.h>
  27 #include <Generic/SumMD5.h>
  28
  29 #define CHECK_FIRST_CNT 64
  30
  31 IFileReader         *rSplitISOData::openRead(const char* sFile, EDtError &rc)
  32 {
  33    rSplitISOData *pSkel = 0;
  34    if (true == checkFile(sFile, rc))
  35    {
  36       pSkel = new rSplitISOData(sFile, rc);
  37    }
  38    return pSkel;
  39 }
  40
  41 rSplitISOData::rSplitISOData(const char *sName, EDtError &rc)
  42 {
  43    rc = DT_OK;
  44    sFileName = sName;
  45    memset(&item, 0, sizeof(item));
  46
  47    sector_size    = 0;
  48    sector_count   = 0;
  49    num_parts      = 0;
  50    current_file   = 0;
  51
  52    if (false == readDescriptor(rc))
  53    {
  54       sector_count = 0;
  55       sector_size  = 0;
  56    }
  57 }
  58
  59 rSplitISOData::~rSplitISOData()
  60 {
  61    for (int i=0; i<descs.Count(); i++)
  62       delete descs[i];
  63    cleanUp();
  64 }
  65
  66 bool                 rSplitISOData::checkFile(const char* sFile, EDtError &rc)
  67 {
  68    bool res = false;
  69    FileInfoBlock *b = new FileInfoBlock;
  70
  71    BPTR l = DOS->Lock((char*)sFile, ACCESS_READ);
  72
  73    if (l != 0)
  74    {
  75       if (DOS->Examine(l, b))
  76       {
  77          // our files should never grow any larger :P
  78
  79          rc = DT_InvalidFormat;
  80
  81          if (b->fib_Size < 4*1024)
  82          {
  83             XMLDocument xdoc;
  84
  85             if (true == xdoc.ReadXML(sFile))
  86             {
  87                if (xdoc.GetName() == "SplitISOInfo")
  88                {
  89                   res = true;
  90                   rc = DT_OK;
  91                }
  92             }
  93          }
  94       }
  95       else
  96          rc = DT_UnableToOpenFile;
  97
  98       DOS->UnLock(l);
  99    }
 100    else
 101       rc = DT_UnableToOpenFile;
 102
 103    delete b;
 104
 105    return res;
 106 }
 107
 108 bool                 rSplitISOData::readDescriptor(EDtError &rc)
 109 {
 110    XMLDocument    xdoc;
 111    XMLAttribute  *elem;
 112
 113    xdoc.ReadXML(sFileName.Data());
 114
 115    sector_size = 0;
 116    elem = xdoc.FindAttribute("SectorSize");
 117    if (elem != 0)
 118    {
 119       sector_size = elem->GetValue().ToLong();
 120    }
 121
 122    sector_count = 0;
 123    elem = xdoc.FindAttribute("SectorCount");
 124    if (elem != 0)
 125    {
 126       sector_count = elem->GetValue().ToLong();
 127    }
 128
 129    num_parts = 0;
 130    elem = xdoc.FindAttribute("NumParts");
 131    if (elem != 0)
 132    {
 133       num_parts = elem->GetValue().ToLong();
 134    }
 135
 136    for (int i=0; i<num_parts; i++)
 137    {
 138       for (int j=0; j<xdoc.GetChildrenCount(); j++)
 139       {
 140          XMLElement *xel = xdoc.GetChild(j);
 141
 142          elem = xel->FindAttribute("PartNumber");
 143          if ((0 != elem) && (elem->GetValue().ToLong() == i))
 144          {
 145             PartDescriptor *pdesc = new PartDescriptor;
 146
 147             elem = xel->FindAttribute("MD5Sum");
 148             if (elem != 0)
 149                pdesc->md5sum = elem->GetValue();
 150
 151             elem = xel->FindAttribute("PartName");
 152             if (elem != 0)
 153                pdesc->fileName = elem->GetValue();
 154
 155             elem = xel->FindAttribute("PartSize");
 156             if (elem != 0)
 157                pdesc->sectors = elem->GetValue().ToLong();
 158
 159             descs << pdesc;
 160
 161             if (validatePart(pdesc) == 0)
 162             {
 163                rc = DT_FileMalformed;
 164                return false;
 165             }
 166          }
 167       }
 168    }
 169
 170    rc = DT_OK;
 171    return true;
 172 }
 173
 174 bool                 rSplitISOData::validatePart(PartDescriptor *pdesc)
 175 {
 176    BPTR   fh;
 177    uint8 *mem;
 178    SumMD5 sum;
 179    uint32 sval[4];
 180    int    len;
 181    String sstr;
 182    bool   res = false;
 183
 184    fh = DOS->Open(pdesc->fileName.Data(), MODE_OLDFILE);
 185    if (0 != fh)
 186    {
 187       mem = new uint8[CHECK_FIRST_CNT * sector_size];
 188       if (0 != mem)
 189       {
 190          len = DOS->Read(fh, mem, CHECK_FIRST_CNT * sector_size);
 191          if (len > 0)
 192          {
 193             sum.Initialize();
 194             sum.Update(mem, len);
 195             sum.Finalize();
 196
 197             sum.GetSum(sval);
 198             sstr.FormatStr("%08lx%08lx%08lx%08lx", ARRAY(sval[0], sval[1], sval[2], sval[3]));
 199
 200             if (sstr == pdesc->md5sum)
 201             {
 202                res = true;
 203             }
 204          }
 205          delete [] mem;
 206       }
 207       DOS->Close(fh);
 208    }
 209
 210    return res;
 211 }
 212
 213 const char          *rSplitISOData::static_getName()
 214 {
 215    return "Split ISO Data Track";
 216 }
 217
 218 bool                 rSplitISOData::static_isAudio()
 219 {
 220    return false;
 221 }
 222
 223 bool                 rSplitISOData::static_isData()
 224 {
 225    return true;
 226 }
 227
 228 bool                 rSplitISOData::static_isSession()
 229 {
 230    return false;
 231 }
 232
 233 bool                 rSplitISOData::isAudio()
 234 {
 235    return static_isAudio();
 236 }
 237
 238 bool                 rSplitISOData::isData()
 239 {
 240    return static_isData();
 241 }
 242
 243 const char          *rSplitISOData::getName()
 244 {
 245    return static_getName();
 246 }
 247
 248 const char*          rSplitISOData::getTrackName()
 249 {
 250    return sFileName.Data();
 251 }
 252
 253 bool                 rSplitISOData::readData(const IOptItem* item, void* pBuff, int pLen)
 254 {
 255    int32 len;
 256
 257    ASSERT(pBuff != 0);
 258    ASSERT(pLen != 0);
 259    if ((pBuff == 0) || (pLen == 0))
 260       return false;
 261
 262    if (0 == current_file)
 263    {
 264       current_file = DOS->Open(descs[current_elem]->fileName.Data(), MODE_OLDFILE);
 265       current_size = descs[current_elem]->sectors;
 266    }
 267
 268    len = current_size - current_pos;
 269    len = len < pLen ? len : pLen;
 270
 271    ASSERT(len > 0);
 272
 273    ASSERT(0 != current_file);
 274    if (current_file != 0)
 275    {
 276       DOS->Read(current_file, pBuff, len * sector_size);
 277    }
 278    else
 279    {
 280       return false;
 281    }
 282
 283    pLen        -= len;
 284    current_pos += len;
 285
 286    if (current_pos >= current_size)
 287    {
 288       if (current_file)
 289          DOS->Close(current_file);
 290
 291       current_file = 0;
 292       current_pos  = 0;
 293       current_elem++;
 294    }
 295
 296    if (pLen > 0)
 297       return readData(item, &((char*)pBuff)[len * sector_size], pLen);
 298
 299    return true;
 300 }
 301
 302 bool                 rSplitISOData::setUp()
 303 {
 304    current_file   = 0;
 305    current_pos    = 0;
 306    current_size   = 0;
 307    current_elem   = 0;
 308    return true;
 309 }
 310
 311 void                 rSplitISOData::cleanUp()
 312 {
 313    if (current_file != 0)
 314    {
 315       DOS->Close(current_file);
 316    }
 317    current_file = 0;
 318 }
 319
 320 void                 rSplitISOData::dispose()
 321 {
 322    delete this;
 323 }
 324
 325 unsigned long        rSplitISOData::getBlockCount()
 326 {
 327    return sector_count;
 328 }
 329
 330 unsigned short       rSplitISOData::getBlockSize()
 331 {
 332    return sector_size;
 333 }
 334
 335 bool                 rSplitISOData::fillOptItem(IOptItem *item)
 336 {
 337    item->setDataType(Data_Mode1);
 338    item->setDataBlockCount(sector_count);
 339    item->setSectorSize(2048);
 340
 341    return true;
 342 }
 343
 344 // vim: ts=3 et