Kernel - Slight reworks to timer code
[tpg/acess2.git] / Kernel / modules.c
index 39c9966..98ca1a1 100644 (file)
  * Acess2
  * - Module Loader
  */
+#define DEBUG  0
 #include <acess.h>
 #include <modules.h>
 
 #define        USE_EDI 0
-#define        USE_UDI 1
+#define        USE_UDI 0
 
 // === PROTOTYPES ===
- int   Modules_LoadBuiltins();
- int   Module_LoadMem(void *Buffer, Uint Length, char *ArgString);
- int   Module_LoadFile(char *Path, char *ArgString);
+ int   Module_int_Initialise(tModule *Module, const char *ArgString);
+void   Modules_int_GetBuiltinArray(void);
+void   Modules_LoadBuiltins(void);
+void   Modules_SetBuiltinParams(const char *Name, char *ArgString);
+ int   Modules_InitialiseBuiltin(const char *Name);
+// int Module_RegisterLoader(tModuleLoader *Loader);
+// int Module_LoadMem(void *Buffer, Uint Length, char *ArgString);
+// int Module_LoadFile(char *Path, char *ArgString);
  int   Module_int_ResolveDeps(tModule *Info);
- int   Module_IsLoaded(char *Name);
+ int   Module_IsLoaded(const char *Name);
+// int Module_EnsureLoaded(const char *Name);
+
+// === EXPORTS ===
+EXPORT(Module_RegisterLoader);
 
 // === IMPORTS ===
 #if USE_UDI
 extern int     UDI_LoadDriver(void *Base);
 #endif
-extern void    StartupPrint(char *Str);
-extern tModule gKernelModules[];
-extern void    gKernelModulesEnd;
+extern void    StartupPrint(const char *Str);
+extern tModule gKernelModules;
+extern tModule gKernelModulesEnd;
 
 // === GLOBALS ===
  int   giNumBuiltinModules = 0;
- int   giModuleSpinlock = 0;
+tShortSpinlock glModuleSpinlock;
 tModule        *gLoadedModules = NULL;
 tModuleLoader  *gModule_Loaders = NULL;
+tModule        *gLoadingModules = NULL;
+tModule        **gapBuiltinModules = NULL;
+char   **gasBuiltinModuleArgs;
 
 // === CODE ===
-int Modules_LoadBuiltins()
+/**
+ * \brief Initialises a module
+ * \param Module       Pointer to the module header
+ * \param ArgString    Comma separated list of module arguments
+ * \return Zero on success, eModuleErrors or -1 on error
+ * \retval -1  Returned if a dependency fails, or a circular dependency
+ *              exists.
+ * \retval 0   Returned on success
+ * \retval >0  Error code form the module's initialisation function
+ */
+int Module_int_Initialise(tModule *Module, const char *ArgString)
 {
-        int    i, j, k;
+        int    i, j;
         int    ret;
-        int    numToInit = 0;
-       Uint8   *baIsLoaded;
-       char    **deps;
+       const char      **deps;
+       char    **args;
+       tModule *mod;
        
-       // Count modules
-       giNumBuiltinModules = (Uint)&gKernelModulesEnd - (Uint)&gKernelModules;
-       giNumBuiltinModules /= sizeof(tModule);
+       ENTER("pModule", Module);
+       LOG("Module->Magic = 0x%x", Module->Magic);
+       if(Module->Magic != MODULE_MAGIC) {
+               Log_Warning(
+                       "Module",
+                       "Module %p is no a valid Acess2 module (0x%08x != 0x%08x)",
+                       Module, Module->Magic, MODULE_MAGIC
+                       );
+               LEAVE('i', MODULE_ERR_BADMODULE);
+               return MODULE_ERR_BADMODULE;
+       }
+       LOG("Module->Name = %p \"%s\"", Module->Name, Module->Name);
        
-       // Allocate loaded array
-       baIsLoaded = calloc( giNumBuiltinModules, sizeof(*baIsLoaded) );
+       if(Module->Arch != MODULE_ARCH_ID) {
+               Log_Warning(
+                       "Module",
+                       "Module %p (%s) is for another architecture (%i)",
+                       Module, Module->Name, Module->Arch
+                       );
+       }
        
-       // Pass 1 - Check for dependencies
-       for( i = 0; i < giNumBuiltinModules; i++ )
+       deps = Module->Dependencies;
+       
+       // Check if the module has been loaded
+       for( mod = gLoadedModules; mod; mod = mod->Next )
        {
-               deps = gKernelModules[i].Dependencies;
-               if(deps)
-               {
-                       for( j = 0; deps[j]; j++ )
-                       {
-                               for( k = 0; k < giNumBuiltinModules; k++ ) {
-                                       if(strcmp(deps[j], gKernelModules[k].Name) == 0)
-                                               break;
-                               }
-                               if(k == giNumBuiltinModules) {
-                                       Warning(
-                                               "[MOD  ] Dependency '%s' for module '%s' was not compiled in",
-                                               deps[j], gKernelModules[i].Name
-                                               );
-                                       
-                                       baIsLoaded[i] = -1;     // Don't Load
-                                       break;
-                               }
-                       }
-               }
-               numToInit ++;
+               if(mod == Module)       LEAVE_RET('i', 0);
        }
        
-       // Pass 2 - Intialise modules in order
-       while(numToInit)
+       // Add to the "loading" (prevents circular deps)
+       Module->Next = gLoadingModules;
+       gLoadingModules = Module;
+       
+       // Scan dependency list
+       for( j = 0; deps && deps[j]; j++ )
        {
-               for( i = 0; i < giNumBuiltinModules; i++ )
+               // Check if the module is already loaded
+               for( mod = gLoadedModules; mod; mod = mod->Next )
                {
-                       if( baIsLoaded[i] )     continue;       // Ignore already loaded modules
+                       if(strcmp(deps[j], mod->Name) == 0)
+                               break;
+               }
+               if( mod )       continue;       // Dependency is loaded, check the rest
                
-                       deps = gKernelModules[i].Dependencies;
-                       
-                       if( deps )
-                       {
-                               for( j = 0; deps[j]; j++ )
-                               {
-                                       for( k = 0; k < giNumBuiltinModules; k++ ) {
-                                               if(strcmp(deps[j], gKernelModules[k].Name) == 0)
-                                                       break;
-                                       }
-                                       // `k` is assumed to be less than `giNumBuiltinModules`
-                                       // We checked this in pass 1
-                                       
-                                       // If a dependency failed, skip and mark as failed
-                                       if( baIsLoaded[k] == -1 ) {
-                                               baIsLoaded[i] = -1;
-                                               numToInit --;
-                                               break;
-                                       }
-                                       // If a dependency is not intialised, skip this module
-                                       // and come back later
-                                       if( !baIsLoaded[k] )    break;
-                               }
-                               // Check for breakouts
-                               if( deps[j] )   continue;
-                       }
-                       
-                       // All Dependencies OK? Initialise
-                       StartupPrint(gKernelModules[i].Name);
-                       Log("[MOD  ] Initialising %p '%s' v%i.%i...",
-                               &gKernelModules[i],
-                               gKernelModules[i].Name,
-                               gKernelModules[i].Version>>8, gKernelModules[i].Version & 0xFF
+               // Ok, check if it's loading
+               for( mod = gLoadingModules->Next; mod; mod = mod->Next )
+               {
+                       if(strcmp(deps[j], mod->Name) == 0)
+                               break;
+               }
+               if( mod ) {
+                       Log_Warning("Module", "Circular dependency detected (%s and %s)",
+                               mod->Name, Module->Name);
+                       LEAVE_RET('i', -1);
+               }
+               
+               // So, if it's not loaded, we better load it then
+               for( i = 0; i < giNumBuiltinModules; i ++ )
+               {
+                       if( strcmp(deps[j], gapBuiltinModules[i]->Name) == 0 )
+                               break;
+               }
+               if( i == giNumBuiltinModules ) {
+                       Log_Warning("Module", "Dependency '%s' for module '%s' failed",
+                               deps[j], Module->Name);
+                       return -1;
+               }
+               
+               // Dependency is not loaded, so load it
+               ret = Module_int_Initialise(
+                       gapBuiltinModules[i],
+                       gasBuiltinModuleArgs ? gasBuiltinModuleArgs[i] : NULL
+                       );
+               if( ret )
+               {
+                       // The only "ok" error is NOTNEEDED
+                       if(ret != MODULE_ERR_NOTNEEDED)
+                               LEAVE_RET('i', -1);
+               }
+       }
+       
+       // All Dependencies OK? Initialise
+       StartupPrint(Module->Name);
+       Log_Log("Module", "Starting %p '%s' v%i.%i",
+               Module, Module->Name,
+               Module->Version >> 8, Module->Version & 0xFF
+               );
+       
+       if( ArgString )
+               args = str_split( ArgString, ',' );
+       else
+               args = NULL;
+       
+       ret = Module->Init(args);
+       
+       if(args)        free(args);
+       
+       // Remove from loading list
+       gLoadingModules = gLoadingModules->Next;
+       
+       if( ret != MODULE_ERR_OK ) {
+               switch(ret)
+               {
+               case MODULE_ERR_MISC:
+                       Log_Warning("Module", "Unable to load, reason: Miscelanious");
+                       break;
+               case MODULE_ERR_NOTNEEDED:
+                       Log_Debug("Module", "Unable to load, reason: Module not needed");
+                       break;
+               case MODULE_ERR_MALLOC:
+                       Log_Warning("Module", "Unable to load, reason: Error in malloc/realloc/calloc, probably not good");
+                       break;
+               default:
+                       Log_Warning("Module", "Unable to load reason - Unknown code %i", ret);
+                       break;
+               }
+               LEAVE_RET('i', ret);
+               return ret;
+       }
+       LOG("ret = %i", ret);
+       
+       // Add to loaded list
+       SHORTLOCK( &glModuleSpinlock );
+       Module->Next = gLoadedModules;
+       gLoadedModules = Module;
+       SHORTREL( &glModuleSpinlock );
+       
+       LEAVE_RET('i', 0);
+}
+
+/**
+ * \brief Scans the builtin modules and creates an array of them
+ */
+void Modules_int_GetBuiltinArray(void)
+{
+        int    i;
+       tModule *module;
+       
+       // Count
+       module = &gKernelModules;
+       i = 0;
+       while( (tVAddr)module < (tVAddr)&gKernelModulesEnd )
+       {
+               if(module->Magic == MODULE_MAGIC) {
+                       i ++;
+                       module ++;
+               }
+               else
+                       module = (void*)( (tVAddr)module + 4 );
+       }
+       
+       // Create
+       giNumBuiltinModules = i;
+       gasBuiltinModuleArgs = calloc( giNumBuiltinModules, sizeof(char*) );
+       gapBuiltinModules = malloc( giNumBuiltinModules * sizeof(tModule*) );
+       
+       
+       // Fill
+       module = &gKernelModules;
+       i = 0;
+       while( (tVAddr)module < (tVAddr)&gKernelModulesEnd )
+       {
+               if(module->Magic == MODULE_MAGIC) {
+                       gapBuiltinModules[i] = module;
+                       i ++;
+                       module ++;
+               }
+               else
+                       module = (void*)( (tVAddr)module + 4 );
+       }
+}
+
+/**
+ * \brief Initialises builtin modules
+ */
+void Modules_LoadBuiltins()
+{
+        int    i;
+       
+       if( !gapBuiltinModules )
+               Modules_int_GetBuiltinArray();
+       
+       for( i = 0; i < giNumBuiltinModules; i++ )
+       {
+               Module_int_Initialise(
+                       gapBuiltinModules[i],
+                       (gasBuiltinModuleArgs ? gasBuiltinModuleArgs[i] : NULL)
+                       );
+       }
+       
+       if( gasBuiltinModuleArgs != NULL )
+               free(gasBuiltinModuleArgs);
+}
+
+/**
+ * \brief Initialise a builtin module given it's name
+ * 
+ * E.g. Used by VTerm to load an alternate video driver at runtime
+ */
+int Modules_InitialiseBuiltin(const char *Name)
+{
+        int    i;
+       
+       // Check if it's loaded
+       if( Module_IsLoaded(Name) )
+               return 0;
+       
+       if( !gapBuiltinModules )
+               Modules_int_GetBuiltinArray();
+       
+       for( i = 0; i < giNumBuiltinModules; i++ )
+       {
+               if( strcmp(gapBuiltinModules[i]->Name, Name) == 0 ) {
+                       return Module_int_Initialise(gapBuiltinModules[i],
+                               (gasBuiltinModuleArgs ? gasBuiltinModuleArgs[i] : NULL)
                                );
-                       
-                       ret = gKernelModules[i].Init(NULL);
-                       if( ret != MODULE_ERR_OK ) {
-                               Log("[MOD  ] Loading Failed, all modules that depend on this will also fail");
-                               switch(ret)
-                               {
-                               case MODULE_ERR_MISC:
-                                       Log("[MOD  ] Reason: Miscelanious");
-                                       break;
-                               case MODULE_ERR_NOTNEEDED:
-                                       Log("[MOD  ] Reason: Module not needed (probably hardware not found)");
-                                       break;
-                               case MODULE_ERR_MALLOC:
-                                       Log("[MOD  ] Reason: Error in malloc/realloc/calloc, probably not good");
-                                       break;
-                               default:
-                                       Log("[MOD  ] Reason - Unknown code %i", ret);
-                                       break;
-                               }
-                               baIsLoaded[i] = -1;
-                       }
-                       // Mark as loaded
-                       else
-                               baIsLoaded[i] = 1;
-                       numToInit --;
                }
        }
+       return -1;
+}
+
+/**
+ * \brief Sets the parameters for a builtin module
+ */
+void Modules_SetBuiltinParams(const char *Name, char *ArgString)
+{
+        int    i;
        
-       free(baIsLoaded);
+       if( gasBuiltinModuleArgs == NULL )
+       {
+               Modules_int_GetBuiltinArray();
+       }
        
-       return 0;
+       // I hate expensive scans
+       for( i = 0; i < giNumBuiltinModules; i++ )
+       {
+               if(strcmp( gapBuiltinModules[i]->Name, Name ) == 0) {
+                       gasBuiltinModuleArgs[i] = ArgString;
+                       return ;
+               }
+       }
+       
+       Log_Warning("Modules", "Unknown builtin kernel module '%s'", Name);
 }
 
 /**
@@ -163,7 +324,7 @@ int Module_RegisterLoader(tModuleLoader *Loader)
  * \fn int Module_LoadMem(void *Buffer, Uint Length, char *ArgString)
  * \brief Load a module from a memory location
  */
-int Module_LoadMem(void *Buffer, Uint Length, char *ArgString)
+int Module_LoadMem(void *Buffer, Uint Length, const char *ArgString)
 {
        char    path[VFS_MEMPATH_SIZE];
        
@@ -173,10 +334,10 @@ int Module_LoadMem(void *Buffer, Uint Length, char *ArgString)
 }
 
 /**
- * \fn int Module_LoadFile(char *Path, char *ArgString)
+ * \fn int Module_LoadFile(const char *Path, const char *ArgString)
  * \brief Load a module from a file
  */
-int Module_LoadFile(char *Path, char *ArgString)
+int Module_LoadFile(const char *Path, const char *ArgString)
 {
        void    *base;
        tModule *info;
@@ -186,7 +347,7 @@ int Module_LoadFile(char *Path, char *ArgString)
        
        // Error check
        if(base == NULL) {
-               Warning("Module_LoadFile: Unable to load '%s'", Path);
+               Log_Warning("Module", "Module_LoadFile - Unable to load '%s'", Path);
                return 0;
        }
        
@@ -212,53 +373,20 @@ int Module_LoadFile(char *Path, char *ArgString)
                // Unknown module type?, return error
                Binary_Unload(base);
                #if USE_EDI
-               Warning("Module_LoadFile: Module has neither a Module Info struct, nor an EDI entrypoint");
+               Log_Warning("Module", "Module '%s' has neither a Module Info struct, nor an EDI entrypoint", Path);
                #else
-               Warning("Module_LoadFile: Module does not have a Module Info struct");
+               Log_Warning("Module", "Module '%s' does not have a Module Info struct", Path);
                #endif
                return 0;
        }
        
-       // Check magic number
-       if(info->Magic != MODULE_MAGIC)
+       // Initialise (and register)
+       if( Module_int_Initialise( info, ArgString ) )
        {
-               Warning("Module_LoadFile: Module's magic value is invalid (0x%x != 0x%x)", info->Magic, MODULE_MAGIC);
-               return 0;
-       }
-       
-       // Check Architecture
-       if(info->Arch != MODULE_ARCH_ID)
-       {
-               Warning("Module_LoadFile: Module is for a different architecture");
-               return 0;
-       }
-       
-       // Resolve Dependencies
-       if( !Module_int_ResolveDeps(info) ) {
                Binary_Unload(base);
                return 0;
        }
        
-       Log("Initialising %p '%s' v%i.%i...",
-                               info,
-                               info->Name,
-                               info->Version>>8, info->Version & 0xFF
-                               );
-       
-       // Call Initialiser
-       //if( info->Init( ArgString ) != 0 )
-       if( info->Init( NULL ) == 0 )
-       {
-               Binary_Unload(base);
-               return 0;
-       }
-       
-       // Add to list
-       LOCK( &giModuleSpinlock );
-       info->Next = gLoadedModules;
-       gLoadedModules = info;
-       RELEASE( &giModuleSpinlock );
-       
        return 1;
 }
 
@@ -270,14 +398,14 @@ int Module_LoadFile(char *Path, char *ArgString)
  */
 int Module_int_ResolveDeps(tModule *Info)
 {
-       char    **names = Info->Dependencies;
+       const char      **names = Info->Dependencies;
        
        // Walk dependencies array
        for( ; *names; names++ )
        {
                // Check if the module is loaded
                if( !Module_IsLoaded(*names) ) {
-                       Warning("Module `%s' requires `%s', which is not loaded\n", Info->Name, *names);
+                       Log_Warning("Module", "Module `%s' requires `%s', which is not loaded\n", Info->Name, *names);
                        return 0;
                }
        }
@@ -285,11 +413,11 @@ int Module_int_ResolveDeps(tModule *Info)
 }
 
 /**
- * \fn int Module_IsLoaded(char *Name)
+ * \fn int Module_IsLoaded(const char *Name)
  * \brief Checks if a module is loaded
  * \param Name Name of module to find
  */
-int Module_IsLoaded(char *Name)
+int Module_IsLoaded(const char *Name)
 {
        tModule *mod = gLoadedModules;
        
@@ -303,3 +431,19 @@ int Module_IsLoaded(char *Name)
        // not found - return false
        return 0;
 }
+
+/**
+ * \brief Load a module if needed
+ */
+int Module_EnsureLoaded(const char *Name)
+{
+       if( Module_IsLoaded(Name) )
+               return 0;
+
+       if( Modules_InitialiseBuiltin(Name) == 0 )
+               return 0;
+
+       // TODO: Load from a file?
+
+       return -1;
+}

UCC git Repository :: git.ucc.asn.au