workbench/tools/HDToolBox/harddisks.c

   1 /*
   2     Copyright © 1995-2008, The AROS Development Team. All rights reserved.
   3     $Id$
   4 */
   5
   6 #include <proto/exec.h>
   7 #include <proto/partition.h>
   8
   9 #include <devices/scsidisk.h>
  10 #include <exec/io.h>
  11 #include <exec/memory.h>
  12
  13 #include <stdio.h>
  14
  15 #define DEBUG 0
  16 #include "debug.h"
  17
  18 #include "harddisks.h"
  19 #include "hdtoolbox_support.h"
  20 #include "platform.h"
  21
  22 extern struct GUIGadgets gadgets;
  23
  24 void w2strcpy(STRPTR name, UWORD *wstr, ULONG len)
  25 {
  26     while (len)
  27     {
  28         *((UWORD *)name) = AROS_BE2WORD(*wstr);
  29         name += sizeof(UWORD);
  30         len -= 2;
  31         wstr++;
  32     }
  33     name -= 2;
  34     while ((*name==0) || (*name==' '))
  35         *name-- = 0;
  36 }
  37
  38 BOOL identify(struct IOStdReq *ioreq, STRPTR name)
  39 {
  40     struct SCSICmd scsicmd;
  41     UWORD data[256];
  42     UBYTE cmd = 0xEC; /* identify */
  43
  44     D(bug("[HDToolBox] identify('%s')\n", name));
  45
  46     scsicmd.scsi_Data = data;
  47     scsicmd.scsi_Length = 512;
  48     scsicmd.scsi_Command = &cmd;
  49     scsicmd.scsi_CmdLength = 1;
  50     ioreq->io_Command = HD_SCSICMD;
  51     ioreq->io_Data = &scsicmd;
  52     ioreq->io_Length = sizeof(struct SCSICmd);
  53
  54     if (DoIO((struct IORequest *)ioreq))
  55         return FALSE;
  56
  57     w2strcpy(name, &data[27], 40);
  58     return TRUE;
  59 }
  60
  61 void findHDs(struct HDTBDevice *parent)
  62 {
  63     struct IOStdReq *ioreq;
  64     struct MsgPort *mp;
  65     struct HDNode *node;
  66     int i;
  67
  68     D(bug("[HDToolBox] findHDs()\n"));
  69
  70     mp = CreateMsgPort();
  71     if (mp)
  72     {
  73         ioreq = (struct IOStdReq *)CreateIORequest(mp, sizeof(struct IOStdReq));
  74         if (ioreq)
  75         {
  76             int maxunits = 8;
  77             if (parent->maxunits > maxunits)
  78                 maxunits = parent->maxunits;
  79
  80             for (i=0;i<maxunits;i++)
  81             {
  82                 node = AllocMem(sizeof(struct HDNode), MEMF_PUBLIC | MEMF_CLEAR);
  83                 if (node)
  84                 {
  85                     node->root_partition.listnode.ln.ln_Name = AllocVec(100, MEMF_PUBLIC | MEMF_CLEAR);
  86                     if (node->root_partition.listnode.ln.ln_Name)
  87                     {
  88                         if (InitListNode(&node->root_partition.listnode, (struct ListNode *)parent))
  89                         {
  90                             node->root_partition.listnode.ln.ln_Type = LNT_Harddisk;
  91                             node->unit = i;
  92                             if (OpenDevice(parent->listnode.ln.ln_Name, i, (struct IORequest *)ioreq, 0) == 0)
  93                             {
  94                                 D(bug("[HDToolBox] findHDs: Opened %s:%d\n",parent->listnode.ln.ln_Name, i));
  95                                 if (!identify(ioreq, node->root_partition.listnode.ln.ln_Name))
  96                                     sprintf(node->root_partition.listnode.ln.ln_Name, "Unit %d", i);
  97                                 CloseDevice((struct IORequest *)ioreq);
  98                                 node->root_partition.ph = OpenRootPartition(parent->listnode.ln.ln_Name, node->unit);
  99                                 if (node->root_partition.ph)
 100                                 {
 101                                     D(bug("[HDToolBox] - appending ROOT partition %p to list %p\n", &node->root_partition.listnode.ln, &parent->listnode.list));
 102                                     D(bug("[HDToolBox] - first entry at %p\n", node->root_partition.listnode.list.lh_Head));
 103                                     AddTail(&parent->listnode.list, &node->root_partition.listnode.ln);
 104                                     if (findPartitionTable(&node->root_partition))
 105                                     {
 106                                         D(bug("[HDToolBox] - partition table found. searching for partitions\n"));
 107                                         findPartitions(&node->root_partition.listnode, &node->root_partition);
 108                                         node->root_partition.listnode.flags |= LNF_Listable;
 109                                     }
 110                                     else
 111                                     {
 112                                         D(bug("[HDToolBox] - partition table not found.\n"));
 113                                     }
 114                                     GetPartitionAttrsA
 115                                     (
 116                                         node->root_partition.ph,
 117                                         PT_GEOMETRY, &node->root_partition.dg,
 118                                         PT_DOSENVEC, &node->root_partition.de,
 119                                         TAG_DONE
 120                                     );
 121                                     continue;
 122                                 }
 123                             }
 124                             UninitListNode(&node->root_partition.listnode);
 125                         }
 126                         FreeVec(node->root_partition.listnode.ln.ln_Name);
 127                     }
 128                     FreeMem(node, sizeof(struct HDNode));
 129                 }
 130             }
 131             DeleteIORequest((struct IORequest *)ioreq);
 132         }
 133         DeleteMsgPort(mp);
 134     }
 135     D(bug("[HDToolBox] findHDs() successful\n"));
 136 }
 137
 138 void freeHDList(struct List *list)
 139 {
 140     struct HDNode *node;
 141     struct HDNode *next;
 142
 143     D(bug("[HDToolBox] freeHDList()\n"));
 144
 145     node = (struct HDNode *)list->lh_Head;
 146     while (node->root_partition.listnode.ln.ln_Succ)
 147     {
 148         next = (struct HDNode *)node->root_partition.listnode.ln.ln_Succ;
 149         if (node->root_partition.listnode.ln.ln_Type != LNT_Parent)
 150         {
 151             Remove(&node->root_partition.listnode.ln);
 152             if (node->root_partition.table)
 153             {
 154                 freePartitionList(&node->root_partition.listnode.list);
 155                 freePartitionTable(&node->root_partition);
 156             }
 157             CloseRootPartition(node->root_partition.ph);
 158             UninitListNode(&node->root_partition.listnode);
 159             FreeVec(node->root_partition.listnode.ln.ln_Name);
 160             FreeMem(node, sizeof(struct HDNode));
 161         }
 162         node = next;
 163     }
 164 }