source: ps2launchargs/source/uLaunchELF/hdl_info/apa.c@ 1101

//--------------------------------------------------------------
//File name:   apa.c
//--------------------------------------------------------------
#include <thbase.h>
#include <sysclib.h>
#include <cdvdman.h>
#include <iomanX.h>
#include <sysmem.h>

#include "ps2_hdd.h"
#include "hdd.h"
#include "hdl.h"
#include "apa.h"

#define AUTO_DELETE_EMPTY

#define _MB * (1024 * 1024) /* really ugly :-) */

typedef struct ps2_partition_run_type
{
 unsigned long sector;
 u_long size_in_mb;
} ps2_partition_run_t;

//Remove this line, and uncomment the next line, to reactivate 'apa_check'
//static int apa_check(const apa_partition_table_t *table);

//--------------------------------------------------------------
u_long apa_partition_checksum(const ps2_partition_header_t *part)
{
 const u_long *p = (const u_long*)part;
 register u_long i;
 u_long sum = 0;
 for(i=1; i<256; ++i)
  sum += get_u32(p + i);
 return sum;
}
//------------------------------
//endfunc apa_partition_checksum
//--------------------------------------------------------------
static apa_partition_table_t* apa_ptable_alloc(void)
{
 apa_partition_table_t *table = AllocSysMemory(0, sizeof (apa_partition_table_t), NULL);
 if(table != NULL)
  memset(table, 0, sizeof (apa_partition_table_t));
 return table;
}
//------------------------------
//endfunc apa_ptable_alloc
//--------------------------------------------------------------
void apa_ptable_free(apa_partition_table_t *table)
{
        if (table != NULL){
                if (table->chunks_map != NULL)
                        FreeSysMemory(table->chunks_map);
                if (table->parts != NULL)
                        FreeSysMemory(table->parts);
                FreeSysMemory(table);
        }
}
//------------------------------
//endfunc apa_ptable_free
//--------------------------------------------------------------
static int apa_part_add (apa_partition_table_t *table, const ps2_partition_header_t *part, int existing, int linked)
{
        if (table->part_count == table->part_alloc_){ /* grow buffer */
                u_long bytes = (table->part_alloc_ + 16) * sizeof (apa_partition_t);
                apa_partition_t *tmp = AllocSysMemory(0, bytes, NULL);
                if(tmp != NULL)
                {
                        memset(tmp, 0, bytes);
                        if (table->parts != NULL) /* copy existing */
                                memcpy(tmp, table->parts, table->part_count * sizeof (apa_partition_t));
                        FreeSysMemory(table->parts);
                        table->parts = tmp;
                        table->part_alloc_ += 16;
                }
                else return -2;
        }

        memcpy(&table->parts[table->part_count].header, part, sizeof (ps2_partition_header_t));
        table->parts[table->part_count].existing = existing;
        table->parts[table->part_count].modified = !existing;
        table->parts[table->part_count].linked = linked;
        ++table->part_count;

        return 0;
}
//------------------------------
//endfunc apa_part_add
//--------------------------------------------------------------
/* //Remove this line and a similar one below to reactivate 'apa_setup_statistics'
static int apa_setup_statistics(apa_partition_table_t *table)
{
 u_long i;
 char *map;

 table->total_chunks = table->device_size_in_mb / 128;
 map = AllocSysMemory(0, table->total_chunks * sizeof (char), NULL);
 if(map != NULL)
 {
  for(i=0; i<table->total_chunks; ++i)
   map [i] = MAP_AVAIL;

  // build occupided/available space map
  table->allocated_chunks = 0;
  table->free_chunks = table->total_chunks;
  for(i=0; i<table->part_count; ++i)
  {
   const ps2_partition_header_t *part = &table->parts [i].header;
   u_long part_no = get_u32(&part->start) / ((128 _MB) / 512);
   u_long num_parts = get_u32(&part->length) / ((128 _MB) / 512);

   // "alloc" num_parts starting at part_no
   while (num_parts)
   {
    if(map[part_no] == MAP_AVAIL)
     map[part_no] = get_u32(&part->main) == 0 ? MAP_MAIN : MAP_SUB;
    else
     map[part_no] = MAP_COLL; // collision
    ++part_no;
    --num_parts;
    ++table->allocated_chunks;
    --table->free_chunks;
   }
  }

  if(table->chunks_map != NULL)
  FreeSysMemory(table->chunks_map);
  table->chunks_map = map;

  return 0;
 }
 else return -2;
}
*/ //Remove this line and a similar one below to reactivate 'apa_setup_statistics'
//------------------------------
//endfunc apa_setup_statistics
//--------------------------------------------------------------
int apa_ptable_read_ex ( hio_t *hio, apa_partition_table_t **table)
{
        u_long size_in_kb;
        int result = hio->stat(hio, &size_in_kb);
        if(result == 0){
                u_long total_sectors;
                // limit HDD size to 128GB - 1KB; that is: exclude the last 128MB chunk
                //if (size_in_kb > 128 * 1024 * 1024 - 1)
                // size_in_kb = 128 * 1024 * 1024 - 1;

                total_sectors = size_in_kb * 2; /* 1KB = 2 sectors of 512 bytes, each */

                *table = apa_ptable_alloc();
                if(*table != NULL){
                        u_long sector = 0;
                        do {
                                u_long bytes;
                                ps2_partition_header_t part;
                                result = hio->read(hio, sector, sizeof(part) / 512, &part, &bytes);
                                if(result == 0){
                                        if(bytes == sizeof(part) &&
                                          get_u32(&part.checksum) == apa_partition_checksum(&part) &&
                                          memcmp(part.magic, PS2_PARTITION_MAGIC, 4) == 0)
                                        {
                                                if(get_u32(&part.start) < total_sectors &&
                                                  get_u32(&part.start) + get_u32(&part.length) < total_sectors)
                                                {
                                                        if((get_u16(&part.flags)==0x0000) && (get_u16(&part.type) ==0x1337))
                                                                result = apa_part_add(*table, &part, 1, 1);
                                                        if(result == 0)
                                                                sector = get_u32(&part.next);
                                                } else { /* partition behind end-of-HDD */
                                                        result = 7; /* data behind end-of-HDD */
                                                        break;
                                                }
                                        } else
                                                result = 1;
                                }
                                /* TODO: check whether next partition is not loaded already --
                                * do not deadlock; that is a quick-and-dirty hack */
                                if ((*table)->part_count > 10000)
                                        result = 7;
                        } while (result == 0 && sector != 0);

                        if (result == 0){
                                (*table)->device_size_in_mb = size_in_kb / 1024;
                                //NB: uncommenting the next lines requires changes elsewhere too
                                //result = apa_setup_statistics (*table);
                                //if (result == 0)
                                        //result = apa_check (*table);
                        }

                        if (result != 0){
                                result = 20000+(*table)->part_count;
                                apa_ptable_free (*table);
                        }
                }
                else result = -2;
        }
        return result;
}
//------------------------------
//endfunc apa_ptable_read_ex
//--------------------------------------------------------------
/* //Remove this line and a similar one below to reactivate 'apa_check'
static int apa_check (const apa_partition_table_t *table) {

  u_long i, j, k;

  const u_long total_sectors = table->device_size_in_mb * 1024 * 2;

  for (i=0; i<table->part_count; ++i)
    {
      const ps2_partition_header_t *part = &table->parts [i].header;
      if (get_u32 (&part->checksum) != apa_partition_checksum (part))
 return 7; // bad checksum

      if (get_u32 (&part->start) < total_sectors &&
   get_u32 (&part->start) + get_u32 (&part->length) <= total_sectors)
 ;
      else
 {
     return 7; // data behind end-of-HDD
 }

      if ((get_u32 (&part->length) % ((128 _MB) / 512)) != 0)
 return 7; // partition size not multiple to 128MB

      if ((get_u32 (&part->start) % get_u32 (&part->length)) != 0)
 return 7; // partition start not multiple on partition size

      if (get_u32 (&part->main) == 0 &&
   get_u16 (&part->flags) == 0 &&
   get_u32 (&part->start) != 0)
 { // check sub-partitions
   u_long count = 0;
   for (j=0; j<table->part_count; ++j)
     {
       const ps2_partition_header_t *part2 = &table->parts [j].header;
       if (get_u32 (&part2->main) == get_u32 (&part->start))
  { // sub-partition of current main partition
    int found;
    if (get_u16 (&part2->flags) != PS2_PART_FLAG_SUB)
      return 7;

    found = 0;
    for (k=0; k<get_u32 (&part->nsub); ++k)
      if (get_u32 (&part->subs [k].start) == get_u32 (&part2->start))
        { // in list
   if (get_u32 (&part->subs [k].length) != get_u32 (&part2->length))
     return 7;
   found = 1;
   break;
        }
    if (!found)
      return 7; // not found in the list

    ++count;
  }
     }
   if (count != get_u32 (&part->nsub))
     return 7; // wrong number of sub-partitions
 }
    }

  // verify double-linked list
  for (i=0; i<table->part_count; ++i)
    {
      apa_partition_t *prev = table->parts + (i > 0 ? i - 1 : table->part_count - 1);
      apa_partition_t *curr = table->parts + i;
      apa_partition_t *next = table->parts + (i + 1 < table->part_count ? i + 1 : 0);
      if (get_u32 (&curr->header.prev) != get_u32 (&prev->header.start) ||
   get_u32 (&curr->header.next) != get_u32 (&next->header.start))
 return 7; // bad links
    }

  return 0;
}
*/ //Remove this line and a similar one above to reactivate 'apa_check'
//------------------------------
//endfunc apa_check
//--------------------------------------------------------------
u_long get_u32 (const void *buffer)
{
  const u_char *p = buffer;
  return ((((u_long) p[3]) << 24) |
          (((u_long) p[2]) << 16) |
          (((u_long) p[1]) <<  8) |
          (((u_long) p[0]) <<  0));
}
//------------------------------
//endfunc get_u32
//--------------------------------------------------------------
void set_u32 (void *buffer, u_long val)
{
  u_char *p = buffer;
  p [3] = (u_char) (val >> 24);
  p [2] = (u_char) (val >> 16);
  p [1] = (u_char) (val >>  8);
  p [0] = (u_char) (val >>  0);
}
//------------------------------
//endfunc set_u32
//--------------------------------------------------------------
u_short get_u16 (const void *buffer)
{
  const u_char *p = buffer;
  return ((((u_short) p[1]) << 8) |
          (((u_short) p[0]) << 0));
}
//------------------------------
//endfunc get_u16
//--------------------------------------------------------------
void set_u16 (void *buffer, u_short val)
{
  u_char *p = buffer;
  p [1] = (u_char) (val >> 8);
  p [0] = (u_char) (val >> 0);
}
//------------------------------
//endfunc set_u16
//--------------------------------------------------------------
//End of file: apa.c
//--------------------------------------------------------------