workbench/tools/SysExplorer/Modules/AHCI/ahci_enum.c

   1 /*
   2     Copyright (C) 2018-2019, The AROS Development Team.
   3     $Id$
   4 */
   5
   6 #include <aros/debug.h>
   7
   8 #define __STORAGE_NOLIBBASE__
   9
  10 #include <proto/sysexp.h>
  11 #include <proto/storage.h>
  12
  13 #include <proto/alib.h>
  14 #include <proto/dos.h>
  15 #include <proto/exec.h>
  16 #include <proto/muimaster.h>
  17 #include <proto/oop.h>
  18 #include <proto/utility.h>
  19 #include <proto/intuition.h>
  20
  21 #include <zune/customclasses.h>
  22 #include <mui/NListtree_mcc.h>
  23 #include <mui/NListview_mcc.h>
  24 #include <utility/tagitem.h>
  25 #include <utility/hooks.h>
  26
  27 #include <hidd/storage.h>
  28 #include <hidd/hidd.h>
  29 #include <hidd/ahci.h>
  30
  31 #include "locale.h"
  32 #include "storage.h"
  33
  34 #include "ahci_classes.h"
  35 #include "ahci_intern.h"
  36
  37 extern int AHCIBusWindow_Initialize(struct SysexpBase *SysexpBase);
  38 extern void AHCIBusWindow_Deinitialize(struct SysexpBase *SysexpBase);
  39 extern int AHCIUnitWindow_Initialize(struct SysexpBase *SysexpBase);
  40 extern void AHCIUnitWindow_Deinitialize(struct SysexpBase *SysexpBase);
  41
  42 #if (1) // TODO : Move into libbase
  43 extern OOP_AttrBase HiddStorageUnitAB;
  44 extern OOP_AttrBase HiddAHCIUnitAB;
  45 #endif
  46
  47 struct MUI_CustomClass *DevicePage_CLASS;
  48
  49 AROS_UFH3S(BOOL, ahciunitenumFunc,
  50     AROS_UFHA(struct Hook *, h,  A0),
  51     AROS_UFHA(OOP_Object*, unit, A2),
  52     AROS_UFHA(void *, parent, A1))
  53 {
  54     AROS_USERFUNC_INIT
  55
  56     struct SysexpEnum_data *edata = (struct SysexpEnum_data *)h->h_Data;
  57     struct InsertObjectMsg msg =
  58     {
  59         .obj      = unit,
  60     };
  61     CONST_STRPTR name;
  62     struct SysexpBase *SysexpBase;
  63     struct SysexpAHCIBase *AhciBase;
  64
  65     D(bug("[ahci.sysexp] %s()\n", __func__));
  66
  67     if (!edata)
  68         return TRUE;
  69
  70     SysexpBase = edata->ed_sysexpbase;
  71
  72     D(bug("[ahci.sysexp] ahciunit @ %p\n", unit));
  73     if (!unit)
  74         return TRUE;
  75
  76     AhciBase = GetBase("AHCI.Module");
  77     msg.winClass = AhciBase->seab_AHCIUnitWindowCLASS;
  78     SysexpBase->GlobalCount++;
  79
  80     OOP_GetAttr(unit, aHidd_StorageUnit_Model, (IPTR *)&name);
  81     DoMethod(SysexpBase->sesb_Tree, MUIM_NListtree_Insert, name, &msg,
  82              parent, MUIV_NListtree_Insert_PrevNode_Tail, 0);
  83
  84     return FALSE; /* Continue enumeration */
  85
  86     AROS_USERFUNC_EXIT
  87 }
  88
  89 struct SysexpStorageBase *StorageBase;
  90 static struct SysexpEnum_data privatehookdata =
  91 {
  92     NULL,
  93     NULL
  94 };
  95
  96 static void ahciBusEnum(OOP_Object *obj, struct MUI_NListtree_TreeNode *parent)
  97 {
  98     D(bug("[ahci.sysexp] ahcibus: \n"));
  99
 100     EnumBusUnits(obj, parent, ahciunitenumFunc, &privatehookdata);
 101 }
 102
 103 void AHCIStartup(struct SysexpBase *SysexpBase)
 104 {
 105     struct SysexpAHCIBase *AhciBase;
 106     D(bug("[ahci.sysexp] %s(%p)\n", __func__, SysexpBase));
 107
 108     StorageBase = GetBase("Storage.Module");
 109     D(bug("[ahci.sysexp] %s: StorageBase @ %p\n", __func__, StorageBase));
 110     if (!StorageBase)
 111     {
 112             __showerror
 113         ( (char *)
 114             "AHCI enumerator requires `storage.sysexp'.", NULL
 115         );
 116     }
 117     AhciBase = GetBase("AHCI.Module");
 118     D(bug("[ahci.sysexp] %s: AhciBase @ %p\n", __func__, AhciBase));
 119
 120     privatehookdata.ed_sysexpbase = SysexpBase;
 121
 122     AhciBase->seab_DevicePageCLASS = GetBase("DevicePage.Class");
 123     AhciBase->seab_GenericWindowCLASS = GetBase("GenericWindow.Class");
 124
 125     AHCIBusWindow_Initialize(SysexpBase);
 126     AHCIUnitWindow_Initialize(SysexpBase);
 127
 128     RegisterStorageControllerHandler((CONST_STRPTR)CLID_Hidd_AHCI, 90, AhciBase->seab_GenericWindowCLASS, NULL, NULL);
 129     RegisterStorageBusHandler((CONST_STRPTR)CLID_Hidd_AHCIBus, 90, AhciBase->seab_AHCIBusWindowCLASS, ahciBusEnum, NULL);
 130 }
 131
 132 void AHCIShutdown(struct SysexpBase *SysexpBase)
 133 {
 134     D(bug("[ahci.sysexp] %s(%p)\n", __func__, SysexpBase));
 135
 136     AHCIUnitWindow_Deinitialize(SysexpBase);
 137     AHCIBusWindow_Deinitialize(SysexpBase);
 138 }