workbench/c/LoadResource/main.c

   1 /*
   2     Copyright © 2004-2013, The AROS Development Team. All rights reserved.
   3     This file is part of the LoadResource program, which is distributed under
   4     the terms of version 2 of the GNU General Public License.
   5
   6     FIXME:
   7     + Implement LOCK and UNLOCK options. Argument template should then be
   8       changed to "NAME/M,LOCK/S,UNLOCK/S"; if the user doesn't provide NAME,
   9       then all currently locked resources shall be listed.
  10     + Implement support for loading devices, fonts and catalogs.
  11 */
  12
  13 #include <exec/io.h>
  14 #include <proto/exec.h>
  15 #include <proto/dos.h>
  16
  17 #include <string.h>
  18
  19 #include "locale.h"
  20
  21 #if defined(__AROSPLATFORM_SMP__)
  22 #include <aros/types/spinlock_s.h>
  23 #include <proto/execlock.h>
  24 #include <resources/execlock.h>
  25 #endif
  26
  27 enum
  28 {
  29     ARG_NAME,
  30     ARG_COUNT
  31 };
  32
  33 #define TEMPLATE     "NAME/M/A"
  34 #define ERROR_HEADER "LoadResource"
  35
  36 BOOL process(CONST_STRPTR name);
  37 BOOL dev_process(CONST_STRPTR name);
  38
  39 int main(void)
  40 {
  41     int            rc              = RETURN_OK;
  42     struct RDArgs *rdargs          = NULL;
  43     IPTR           args[ARG_COUNT] = { 0 };
  44
  45     if ((rdargs = ReadArgs(TEMPLATE, args, NULL)) != NULL)
  46     {
  47         if (args[ARG_NAME] != 0)
  48         {
  49             CONST_STRPTR  *names = (CONST_STRPTR *) args[ARG_NAME],
  50                            name  = NULL;
  51
  52             while ((name = *names++) != NULL)
  53             {
  54                 BOOL res;
  55                 int len = strlen(name);
  56
  57                 if ((len > 7) && (!strcmp(&name[len-7], ".device")))
  58                     res = dev_process(name);
  59                 else
  60                     res = process(name);
  61
  62                 if (!res)
  63                 {
  64                     rc = RETURN_WARN;
  65                 }
  66             }
  67         }
  68         else
  69         {
  70             // FIXME: List currently locked resources.
  71         }
  72
  73         FreeArgs(rdargs);
  74     }
  75     else
  76     {
  77         PrintFault(IoErr(), ERROR_HEADER);
  78         rc = RETURN_FAIL;
  79     }
  80
  81     return rc;
  82 }
  83
  84 BOOL process(CONST_STRPTR name)
  85 {
  86     struct Library *lb = OpenLibrary(name, 0L);
  87
  88     if (lb != NULL)
  89     {
  90         CloseLibrary(lb);
  91         return TRUE;
  92     }
  93     else
  94     {
  95         PutStr(ERROR_HEADER": ");
  96         Printf(_(MSG_ERROR_OPEN_LIBRARY), name);
  97         PutStr("\n");
  98         return FALSE;
  99     }
 100 }
 101
 102 BOOL dev_process(CONST_STRPTR name)
 103 {
 104     struct IORequest req;
 105     BOOL retval = FALSE;
 106     memset(&req, 0, sizeof(req));
 107     req.io_Message.mn_Length = sizeof(req);
 108 #if defined(__AROSPLATFORM_SMP__)
 109     void *ExecLockBase = OpenResource("execlock.resource");
 110 #endif
 111
 112     if (!OpenDevice(name, 0, &req, 0))
 113     {
 114         CloseDevice(&req);
 115         retval = TRUE;
 116     }
 117     else
 118     {
 119 #if defined(__AROSPLATFORM_SMP__)
 120         if (ExecLockBase)
 121             ObtainSystemLock(&SysBase->DeviceList, SPINLOCK_MODE_READ, LOCKF_FORBID);
 122         else
 123             Forbid();
 124 #else
 125         Forbid();
 126 #endif
 127         /* There can different errors, but if the device was loaded, it's OK */
 128         if (FindName(&SysBase->DeviceList, name))
 129             retval = TRUE;
 130
 131 #if defined(__AROSPLATFORM_SMP__)
 132         if (ExecLockBase)
 133             ReleaseSystemLock(&SysBase->DeviceList, LOCKF_FORBID);
 134         else
 135             Permit();
 136 #else
 137         Permit();
 138 #endif
 139         if (retval == FALSE)
 140         {
 141             PutStr(ERROR_HEADER": ");
 142             Printf(_(MSG_ERROR_OPEN_DEVICE), name);
 143             PutStr("\n");
 144         }
 145     }
 146
 147     return retval;
 148 }