gnu/abc-shell/environ.c

   1 #include "sh.h"
   2 #include <exec/memory.h>
   3 #include <dos/dos.h>
   4 #include <dos/var.h>
   5 #include <utility/hooks.h>
   6 #include <proto/exec.h>
   7 #include <proto/dos.h>
   8 #include <proto/utility.h>
   9
  10 char **environ = NULL;
  11
  12
  13 #if defined(__amigaos4__) || defined(__AROS__)
  14
  15 #define MAX_ENV_SIZE 1024  /* maximum number of environ entries */
  16
  17 #ifdef AUTOINIT
  18 #ifdef __GNUC__
  19 void ___makeenviron() __attribute__((constructor));
  20 void ___freeenviron() __attribute__((destructor));
  21 #endif
  22 #ifdef __VBCC__
  23 #define ___makeenviron() _INIT_9_makeenviron()
  24 #define ___freeenviron() _EXIT_9_makeenviron()
  25 #endif
  26 #endif
  27
  28
  29 uint32
  30 copy_env(struct Hook *hook, APTR userdata, struct ScanVarsMsg *message)
  31 {
  32         static uint32 env_size = 1;  /* environ is null terminated */
  33
  34         if(strlen(message->sv_GDir) <= 4)
  35         {
  36                 if ( env_size == MAX_ENV_SIZE )
  37                 {
  38                         return 0;
  39                 }
  40
  41                 char **env = (char **)hook->h_Data;
  42                 uint32 size = strlen(message->sv_Name) + 1 + message->sv_VarLen + 1 + 1;
  43                 char *buffer=(char *)malloc(size);
  44                 if ( buffer == NULL )
  45                 {
  46                         return 0;
  47                 }
  48
  49                 ++env_size;
  50
  51                 snprintf(buffer,size-1,"%s=%.*s", message->sv_Name, (int)message->sv_VarLen, message->sv_Var);
  52
  53                 *env  = buffer;
  54                 env++;
  55                 hook->h_Data = env;
  56         }
  57
  58         return 0;
  59 }
  60
  61 void
  62 ___makeenviron()
  63 {
  64         char varbuf[8];
  65         uint32 flags=0;
  66         size_t environ_size=MAX_ENV_SIZE * sizeof(char*);
  67
  68         if(GetVar("ABCSH_IMPORT_LOCAL",varbuf,sizeof(varbuf),GVF_LOCAL_ONLY) > 0)
  69         {
  70             flags = GVF_LOCAL_ONLY;
  71         }
  72         else
  73         {
  74             flags = GVF_GLOBAL_ONLY;
  75         }
  76
  77         environ = (char **)malloc(environ_size);
  78         if (!environ)
  79         {
  80                 return;
  81         }
  82
  83         memset(environ, 0, environ_size);
  84
  85         struct Hook hook;
  86         memset(&hook, 0, sizeof(struct Hook));
  87         hook.h_Entry = copy_env;
  88         hook.h_Data = environ;
  89
  90         ScanVars(&hook, flags, 0);
  91 }
  92
  93 void
  94 ___freeenviron()
  95 {
  96         if ( environ != NULL )
  97         {
  98                 char **i;
  99                 for ( i = environ; *i != NULL; i++ )
 100                 {
 101                         free(*i);
 102                         *i = NULL;
 103                 }
 104
 105                 free(environ);
 106                 environ = NULL;
 107         }
 108 }
 109 #endif