workbench/c/shellcommands/Setenv.c

   1 /*
   2     Copyright © 1995-2001, The AROS Development Team. All rights reserved.
   3     $Id$
   4
   5     Desc: Setenv CLI command
   6     Lang: English
   7 */
   8
   9 /*****************************************************************************
  10
  11     NAME
  12
  13         Setenv
  14
  15     SYNOPSIS
  16
  17         NAME,SAVE/S,STRING/F
  18
  19     LOCATION
  20
  21         Workbench:c
  22
  23     FUNCTION
  24
  25         Sets a global variable from the current shell. These variables can
  26         be accessed from any program executing at any time.
  27
  28         These variables are usually not saved in the ENVARC: directory, hence they
  29         can only be used by programs during the current execution of the
  30         operating system. When using SAVE argument, the variable is also saved
  31         in ENVARC:
  32
  33         If no parameters are specified, the current list of global variables
  34         are displayed.
  35
  36     INPUTS
  37
  38         NAME    - The name of the global variable to set.
  39
  40         SAVE    - Save the variable also in ENVARC:
  41
  42         STRING  - The value of the global variable NAME.
  43
  44     RESULT
  45
  46         Standard DOS error codes.
  47
  48     NOTES
  49
  50     EXAMPLE
  51
  52         Setenv EDITOR Ed
  53
  54             Any program that accesses the variable "EDITOR" will be able to
  55             find out the name of the text-editor the user would like to use,
  56             by examining the contents of the variable.
  57
  58     BUGS
  59
  60     SEE ALSO
  61
  62         Getenv, Unsetenv
  63
  64     INTERNALS
  65
  66     HISTORY
  67
  68         30-Jul-1997     laguest     Initial inclusion into the AROS tree
  69         13-Aug-1997     srittau     Minor changes
  70
  71 ******************************************************************************/
  72
  73 #include <proto/dos.h>
  74 #include <proto/exec.h>
  75
  76 #include <dos/dos.h>
  77 #include <dos/exall.h>
  78 #include <dos/rdargs.h>
  79 #include <dos/var.h>
  80 #include <exec/memory.h>
  81 #include <exec/types.h>
  82
  83 #include <utility/tagitem.h>
  84 #include <stdio.h>
  85
  86 #define SH_GLOBAL_DOSBASE 1 /* because of Printf from amiga.lib */
  87
  88 #include <aros/shcommands.h>
  89
  90 AROS_SH3(Setenv, 45.0,
  91 AROS_SHA(STRPTR, ,NAME,     , NULL),
  92 AROS_SHA(IPTR  , ,SAVE,   /S, NULL),
  93 AROS_SHA(STRPTR, ,STRING, /F, NULL))
  94 {
  95     AROS_SHCOMMAND_INIT
  96
  97     int     Return_Value;
  98     BOOL    Success;
  99     BPTR    lock;
 100     TEXT    progname[108];
 101
 102
 103     GetProgramName(progname, sizeof(progname));
 104     Return_Value = RETURN_OK;
 105
 106     if (SHArg(NAME) != NULL || SHArg(STRING) != NULL)
 107     {
 108         /* Make sure we get to here is either arguments are
 109          * provided on the command line.
 110          */
 111         if (SHArg(NAME) != NULL && SHArg(STRING) != NULL)
 112         {
 113             /* Add the new global variable to the list.
 114              */
 115              Success = SetVar(SHArg(NAME),
 116                               SHArg(STRING),
 117                               -1,
 118                               SHArg(SAVE) ? GVF_GLOBAL_ONLY | GVF_SAVE_VAR : GVF_GLOBAL_ONLY);
 119
 120              if (Success == FALSE)
 121              {
 122                  UBYTE buf[FAULT_MAX+128];
 123
 124                  Fault(-141, NULL, buf, sizeof(buf));
 125                  Printf(buf, SHArg(NAME));
 126                  PrintFault(IoErr(), progname);
 127
 128                  Return_Value = RETURN_ERROR;
 129              }
 130         }
 131     }
 132     else
 133     {
 134         /* Display a list of global variables.
 135          */
 136
 137         lock = Lock("ENV:", ACCESS_READ);
 138         if (lock)
 139         {
 140             struct FileInfoBlock *FIB = AllocDosObject(DOS_FIB, NULL);
 141
 142             if (FIB)
 143             {
 144                 if(Examine(lock, FIB))
 145                 {
 146                     while(ExNext(lock, FIB))
 147                     {
 148                         /* don't show dirs */
 149                         if (FIB->fib_DirEntryType < 0)
 150                         {
 151                             PutStr(FIB->fib_FileName);
 152                             PutStr("\n");
 153                         }
 154                     }
 155                 }
 156                 FreeDosObject(DOS_FIB, FIB);
 157             }
 158             UnLock(lock);
 159         }
 160         else
 161         {
 162             PrintFault(IoErr(), progname);
 163
 164             Return_Value = RETURN_FAIL;
 165         }
 166     }
 167
 168     return Return_Value;
 169
 170     AROS_SHCOMMAND_EXIT
 171 } /* main */