workbench/tools/HDToolBox/harddisks.c

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