test/partition.c

   1 /*
   2     Copyright © 1995-2014, The AROS Development Team. All rights reserved.
   3     $Id$
   4 */
   5
   6 #define DEBUG 0
   7
   8 #include <aros/debug.h>
   9 #include <proto/exec.h>
  10 #include <proto/partition.h>
  11 #include <libraries/partition.h>
  12 #include <utility/tagitem.h>
  13
  14 #include <stdio.h>
  15 #include <stdlib.h>
  16
  17 struct PartitionBase *PartitionBase;
  18
  19 LONG PrintPartitionTable(struct PartitionHandle *root, ULONG i);
  20
  21 static UQUAD getStartBlock(struct PartitionHandle *ph)
  22 {
  23     ULONG ret = 0;
  24
  25     while (ph)
  26     {
  27         UQUAD start;
  28
  29         GetPartitionAttrsTags(ph, PT_STARTBLOCK, &start, TAG_DONE);
  30         ret += start;
  31         ph = ph->root;
  32     }
  33
  34     return ret;
  35 }
  36
  37 void PrintDE(struct DosEnvec *de, ULONG i) {
  38 ULONG a;
  39
  40     for (a=i;a;a--)
  41         printf("  ");
  42     printf("SizeBlock      = %ld\n", de->de_SizeBlock<<2);
  43     for (a=i;a;a--)
  44         printf("  ");
  45     printf("Surfaces       = %ld\n", de->de_Surfaces);
  46     for (a=i;a;a--)
  47         printf("  ");
  48     printf("BlocksPerTrack = %ld\n", de->de_BlocksPerTrack);
  49     for (a=i;a;a--)
  50         printf("  ");
  51     printf("LowCyl         = %ld\n", de->de_LowCyl);
  52     for (a=i;a;a--)
  53         printf("  ");
  54     printf("HighCyl        = %ld\n", de->de_HighCyl);
  55 }
  56
  57 void PrintPInfo(struct PartitionHandle *ph, ULONG i)
  58 {
  59     struct DosEnvec de;
  60     UBYTE name[32];
  61     struct PartitionType type;
  62     ULONG a;
  63     UQUAD start, end, abs;
  64
  65     GetPartitionAttrsTags(ph,
  66                           PT_DOSENVEC  , &de,
  67                           PT_NAME      , name,
  68                           PT_TYPE      , &type,
  69                           PT_STARTBLOCK, &start,
  70                           PT_ENDBLOCK  , &end,
  71                           TAG_DONE);
  72
  73     for (a=i;a;a--)
  74         printf("  ");
  75     printf("name: %s\n", name);
  76     for (a=i+1;a;a--)
  77         printf("  ");
  78
  79     printf("type:");
  80     for (a = 0; a < type.id_len; a++)
  81         printf(" %02X", type.id[a]);
  82     printf("\n");
  83
  84     PrintDE(&de, i+1);
  85
  86     for (a = i + 1; a; a--)
  87         printf("  ");
  88     printf("StartBlock     = %llu\n", (unsigned long long)start);
  89     for (a = i + 1; a; a--)
  90         printf("  ");
  91     printf("EndBlock       = %llu\n", (unsigned long long)end);
  92
  93     abs = getStartBlock(ph->root);
  94     for (a = i + 1; a; a--)
  95         printf("  ");
  96     printf("Abs StartBlock = %llu\n", (unsigned long long)(start + abs));
  97     for (a = i + 1; a; a--)
  98         printf("  ");
  99     printf("Abs EndBlock   = %llu\n", (unsigned long long)(end + abs));
 100
 101     PrintPartitionTable(ph, i + 1);
 102 }
 103
 104 void PrintPartitions(struct PartitionHandle *root, ULONG i) {
 105 struct PartitionHandle *ph;
 106
 107     ph = (struct PartitionHandle *)root->table->list.lh_Head;
 108     while (ph->ln.ln_Succ)
 109     {
 110         D(printf("PartitionHandle 0x%p\n", ph));
 111         PrintPInfo(ph, i);
 112         ph = (struct PartitionHandle *)ph->ln.ln_Succ;
 113     }
 114 }
 115
 116 LONG PrintPartitionTable(struct PartitionHandle *root, ULONG i)
 117 {
 118     struct DosEnvec de;
 119     ULONG type = 0;
 120     ULONG reserved = 0;
 121     ULONG a;
 122
 123     a = OpenPartitionTable(root);
 124     if (a)
 125         return a;
 126
 127     GetPartitionTableAttrsTags(root,
 128                                PTT_TYPE    , &type,
 129                                PTT_RESERVED, &reserved,
 130                                TAG_DONE);
 131     GetPartitionAttrsTags(root, PT_DOSENVEC, &de, TAG_DONE);
 132
 133     for (a=i;a;a--)
 134         printf("  ");
 135     printf("Partition table type is ");
 136     switch (type)
 137     {
 138     case PHPTT_RDB:
 139         printf("Rigid Disk Block\n");
 140         break;
 141
 142     case PHPTT_MBR:
 143         printf("MBR -> PC\n");
 144         break;
 145
 146     case PHPTT_EBR:
 147         printf("EBR -> PC\n");
 148         break;
 149
 150     default:
 151         printf("unknown\n");
 152         break;
 153     }
 154
 155     for (a=i;a;a--)
 156         printf("  ");
 157     printf("reserved blocks: %d\n", (int)reserved);
 158
 159     PrintDE(&de,i);
 160     for (a=i;a;a--)
 161         printf("  ");
 162     printf("partitions:\n");
 163     PrintPartitions(root,i+1);
 164     ClosePartitionTable(root);
 165
 166     return 0;
 167 }
 168
 169 int main(int argc, char **argv)
 170 {
 171     struct PartitionHandle *root;
 172     char *device = "fdsk.device";
 173     ULONG unit = 1;
 174
 175     if (argc > 2)
 176     {
 177         device = argv[1];
 178         unit = atoi(argv[2]);
 179     }
 180
 181     PartitionBase = (struct PartitionBase *)OpenLibrary("partition.library", 1);
 182     if (PartitionBase)
 183     {
 184         root = OpenRootPartition(device, unit);
 185         if (root)
 186         {
 187             printf("got root handle of %s unit %d\n", device, (int)unit);
 188
 189             if (PrintPartitionTable(root, 0))
 190                 printf("Couldn't read partition table\n");
 191
 192             CloseRootPartition(root);
 193         }
 194         else
 195             printf("No root handle for %s unit %d\n", device, (int)unit);
 196         CloseLibrary((struct Library *)PartitionBase);
 197     }
 198     else
 199         printf("No partition.library\n");
 200
 201     return 0;
 202 }
 203