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         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 #include <aros/shcommands.h>
  87
  88 AROS_SH3(Setenv, 45.0,
  89 AROS_SHA(STRPTR, ,NAME,     , NULL),
  90 AROS_SHA(IPTR  , ,SAVE,   /S, NULL),
  91 AROS_SHA(STRPTR, ,STRING, /F, NULL))
  92 {
  93     AROS_SHCOMMAND_INIT
  94
  95     int     Return_Value;
  96     BOOL    Success;
  97     BPTR    lock;
  98     TEXT    progname[108];
  99
 100
 101     GetProgramName(progname, sizeof(progname));
 102     Return_Value = RETURN_OK;
 103
 104     if (SHArg(NAME) != NULL || SHArg(STRING) != NULL)
 105     {
 106         /* Make sure we get to here is either arguments are
 107          * provided on the command line.
 108          */
 109         if (SHArg(NAME) != NULL && SHArg(STRING) != NULL)
 110         {
 111             /* Add the new global variable to the list.
 112              */
 113              Success = SetVar(SHArg(NAME),
 114                               SHArg(STRING),
 115                               -1,
 116                               SHArg(SAVE) ? GVF_GLOBAL_ONLY | GVF_SAVE_VAR : GVF_GLOBAL_ONLY);
 117
 118              if (Success == FALSE)
 119              {
 120                  UBYTE buf[FAULT_MAX+128];
 121
 122                  Fault(-141, NULL, buf, sizeof(buf));
 123                  Printf(buf, SHArg(NAME));
 124                  PrintFault(IoErr(), progname);
 125
 126                  Return_Value = RETURN_ERROR;
 127              }
 128         }
 129     }
 130     else
 131     {
 132         /* Display a list of global variables.
 133          */
 134
 135         lock = Lock("ENV:", ACCESS_READ);
 136         if (lock)
 137         {
 138             struct FileInfoBlock *FIB = AllocDosObject(DOS_FIB, NULL);
 139
 140             if (FIB)
 141             {
 142                 if(Examine(lock, FIB))
 143                 {
 144                     while(ExNext(lock, FIB))
 145                     {
 146                         /* don't show dirs */
 147                         if (FIB->fib_DirEntryType < 0)
 148                         {
 149                             PutStr(FIB->fib_FileName);
 150                             PutStr("\n");
 151                         }
 152                     }
 153                 }
 154                 FreeDosObject(DOS_FIB, FIB);
 155             }
 156             UnLock(lock);
 157         }
 158         else
 159         {
 160             PrintFault(IoErr(), progname);
 161
 162             Return_Value = RETURN_FAIL;
 163         }
 164     }
 165
 166     return Return_Value;
 167
 168     AROS_SHCOMMAND_EXIT
 169 } /* main */