workbench/tools/HDToolBox/main.c

   1 /*
   2     Copyright © 1995-2012, The AROS Development Team. All rights reserved.
   3     $Id$
   4 */
   5
   6 #include <proto/alib.h>
   7 #include <proto/exec.h>
   8 #include <proto/dos.h>
   9 #include <proto/intuition.h>
  10
  11 #include <exec/memory.h>
  12 #include <exec/nodes.h>
  13
  14 #include <stdio.h>
  15 #include <stdlib.h>
  16
  17 #include "devices.h"
  18 #include "error.h"
  19 #include "gui.h"
  20 #include "hdtoolbox_support.h"
  21 #include "locale.h"
  22 #include "partitions.h"
  23 #include "partitiontables.h"
  24 #include "platform.h"
  25 #include "ptclass.h"
  26 #include "prefs.h"
  27
  28 #include "debug.h"
  29
  30 extern struct ListNode root;
  31
  32 struct IntuitionBase    *IntuitionBase = NULL;
  33 struct GfxBase          *GfxBase = NULL;
  34 struct PartitionBase    *PartitionBase = NULL;
  35
  36 LONG initEnv(char *device)
  37 {
  38     LONG retval;
  39
  40     D(bug("[HDToolBox] initEnv('%s')\n", device));
  41
  42     if (!InitListNode(&root, NULL))
  43         return ERR_MEMORY;
  44     IntuitionBase = (struct IntuitionBase *)OpenLibrary("intuition.library", 37);
  45     if (!IntuitionBase)
  46         return ERR_INTUI;
  47     GfxBase = (struct GfxBase *)OpenLibrary("graphics.library", 37);
  48     if (!GfxBase)
  49         return ERR_GFX;
  50     PartitionBase = (struct PartitionBase *)OpenLibrary("partition.library", 1);
  51     if (!PartitionBase)
  52         return ERR_PARTITION;
  53
  54     retval = initGUI();
  55     if (retval != ERR_NONE)
  56         return retval;
  57
  58     LoadPrefs("ENV:hdtoolbox.prefs");
  59
  60     return ERR_NONE;
  61 }
  62
  63 void uninitEnv()
  64 {
  65     D(bug("[HDToolBox] uninitEnv()\n"));
  66
  67     deinitGUI();
  68     freeDeviceList();
  69
  70     if (PartitionBase)
  71         CloseLibrary((struct Library *)PartitionBase);
  72     if (GfxBase)
  73         CloseLibrary((struct Library *)GfxBase);
  74     if (IntuitionBase)
  75         CloseLibrary((struct Library *)IntuitionBase);
  76 }
  77
  78 void waitMessage()
  79 {
  80     ULONG sigs = 0;
  81
  82     D(bug("[HDToolBox] waitMessage()\n"));
  83
  84     while (!QuitGUI(&sigs))
  85     {
  86         if (sigs)
  87         {
  88             sigs = Wait(sigs | SIGBREAKF_CTRL_C | SIGBREAKF_CTRL_D);
  89             if (sigs & SIGBREAKF_CTRL_C)
  90                 break;
  91             if (sigs & SIGBREAKF_CTRL_D)
  92                 break;
  93         }
  94     }
  95 }
  96
  97 int main(int argc, char **argv)
  98 {
  99     ULONG error;
 100     char *device;
 101
 102     D(bug("[HDToolBox] main()\n"));
 103
 104     InitLocale("System/Tools/HDToolBox.catalog", 2);
 105     device = argc > 1 ? argv[1] : NULL;
 106     if ((error=initEnv(device))==ERR_NONE)
 107     {
 108         waitMessage();
 109     }
 110     else
 111         printf("Error %d\n", (int)error);
 112
 113     uninitEnv();
 114     CleanupLocale();
 115     return 0;
 116 }