ReactOS 0.4.15-dev-7842-g558ab78
setlocal.c File Reference
#include "precomp.h"
#include <direct.h>
Include dependency graph for setlocal.c:

Go to the source code of this file.

Classes

struct  _SETLOCAL
 

Typedefs

typedef struct _SETLOCAL SETLOCAL
 
typedef struct _SETLOCALPSETLOCAL
 

Functions

LPTSTR DuplicateEnvironment (VOID)
 
INT cmd_setlocal (LPTSTR param)
 
INT cmd_endlocal (LPTSTR param)
 

Typedef Documentation

◆ PSETLOCAL

◆ SETLOCAL

Function Documentation

◆ cmd_endlocal()

INT cmd_endlocal ( LPTSTR  param)

Definition at line 110 of file setlocal.c.

111{
112 LPTSTR Environ, Name, Value;
113 PSETLOCAL Saved;
114 TCHAR drvEnvVar[] = _T("=?:");
115 TCHAR szCurrent[MAX_PATH];
116
117 if (!_tcscmp(param, _T("/?")))
118 {
119 // FIXME
120 ConOutPuts(_T("ENDLOCAL help not implemented yet!\n"));
121 return 0;
122 }
123
124 /* Pop a SETLOCAL struct off of this batch context's stack */
125 if (!bc || !(Saved = bc->setlocal))
126 return 0;
127 bc->setlocal = Saved->Prev;
128
131
132 /* First, clear out the environment. Since making any changes to the
133 * environment invalidates pointers obtained from GetEnvironmentStrings(),
134 * we must make a copy of it and get the variable names from that. */
135 Environ = DuplicateEnvironment();
136 if (Environ)
137 {
138 for (Name = Environ; *Name; Name += _tcslen(Name) + 1)
139 {
140 if (!(Value = _tcschr(Name + 1, _T('='))))
141 continue;
142 *Value++ = _T('\0');
144 Name = Value;
145 }
146 cmd_free(Environ);
147 }
148
149 /* Now, restore variables from the copy saved by cmd_setlocal() */
150 for (Name = Saved->Environment; *Name; Name += _tcslen(Name) + 1)
151 {
152 if (!(Value = _tcschr(Name + 1, _T('='))))
153 continue;
154 *Value++ = _T('\0');
156 Name = Value;
157 }
158
159 /* Restore the current drive and its current directory from the environment */
160 drvEnvVar[1] = _T('A') + Saved->CurDrive - 1;
161 if (!GetEnvironmentVariable(drvEnvVar, szCurrent, ARRAYSIZE(szCurrent)))
162 {
163 _stprintf(szCurrent, _T("%C:\\"), _T('A') + Saved->CurDrive - 1);
164 }
165 _tchdir(szCurrent); // SetRootPath(NULL, szCurrent);
166
167 cmd_free(Saved->Environment);
168 cmd_free(Saved);
169 return 0;
170}
struct NameRec_ * Name
Definition: cdprocs.h:460
PBATCH_CONTEXT bc
Definition: batch.c:67
BOOL bDelayedExpansion
Definition: cmd.c:162
#define ConOutPuts(szStr)
Definition: console.h:29
#define cmd_free(ptr)
Definition: cmddbg.h:31
#define NULL
Definition: types.h:112
#define ARRAYSIZE(array)
Definition: filtermapper.c:47
#define MAX_PATH
Definition: compat.h:34
GLfloat param
Definition: glext.h:5796
#define _tcscmp
Definition: tchar.h:1424
#define _tchdir
Definition: tchar.h:672
#define _tcschr
Definition: tchar.h:1406
#define _stprintf
Definition: utility.h:124
BOOL bEnableExtensions
Definition: more.c:53
LPTSTR DuplicateEnvironment(VOID)
Definition: setlocal.c:25
struct _SETLOCAL * setlocal
Definition: batch.h:42
struct _SETLOCAL * Prev
Definition: setlocal.c:16
INT CurDrive
Definition: setlocal.c:18
LPTSTR Environment
Definition: setlocal.c:17
BOOL DelayedExpansion
Definition: setlocal.c:20
BOOL EnableExtensions
Definition: setlocal.c:19
#define _T(x)
Definition: vfdio.h:22
_Must_inspect_result_ _In_ WDFKEY _In_ PCUNICODE_STRING _Out_opt_ PUSHORT _Inout_opt_ PUNICODE_STRING Value
Definition: wdfregistry.h:413
#define GetEnvironmentVariable
Definition: winbase.h:3749
#define SetEnvironmentVariable
Definition: winbase.h:3843
char TCHAR
Definition: xmlstorage.h:189
CHAR * LPTSTR
Definition: xmlstorage.h:192
#define _tcslen
Definition: xmlstorage.h:198

Referenced by ExitBatch().

◆ cmd_setlocal()

INT cmd_setlocal ( LPTSTR  param)

Definition at line 42 of file setlocal.c.

43{
44 PSETLOCAL Saved;
45 LPTSTR* arg;
46 INT argc, i;
47
48 if (!_tcscmp(param, _T("/?")))
49 {
50 // FIXME
51 ConOutPuts(_T("SETLOCAL help not implemented yet!\n"));
52 return 0;
53 }
54
55 /* SETLOCAL only works inside a batch context */
56 if (!bc)
57 return 0;
58
59 Saved = cmd_alloc(sizeof(SETLOCAL));
60 if (!Saved)
61 {
62 WARN("Cannot allocate memory for Saved!\n");
64 return 1;
65 }
66
68 if (!Saved->Environment)
69 {
71 cmd_free(Saved);
72 return 1;
73 }
74 /*
75 * Save the current drive; the duplicated environment
76 * contains the corresponding current directory.
77 */
78 Saved->CurDrive = _getdrive();
79
82
83 Saved->Prev = bc->setlocal;
84 bc->setlocal = Saved;
85
86 nErrorLevel = 0;
87
89 for (i = 0; i < argc; i++)
90 {
91 if (!_tcsicmp(arg[i], _T("ENABLEEXTENSIONS")))
93 else if (!_tcsicmp(arg[i], _T("DISABLEEXTENSIONS")))
95 else if (!_tcsicmp(arg[i], _T("ENABLEDELAYEDEXPANSION")))
97 else if (!_tcsicmp(arg[i], _T("DISABLEDELAYEDEXPANSION")))
99 else
100 {
102 break;
103 }
104 }
105 freep(arg);
106
107 return nErrorLevel;
108}
static int argc
Definition: ServiceArgs.c:12
INT nErrorLevel
Definition: cmd.c:158
LPTSTR * splitspace(LPTSTR, LPINT)
Definition: misc.c:381
VOID error_out_of_memory(VOID)
Definition: error.c:138
VOID error_invalid_parameter_format(PCTSTR s)
Definition: error.c:145
#define WARN(fmt,...)
Definition: debug.h:112
static VOID freep(LPSTR *p)
Definition: cmdcons.c:98
#define cmd_alloc(size)
Definition: cmddbg.h:29
_Check_return_ _CRTIMP int __cdecl _getdrive(void)
Definition: getdrive.c:20
#define TRUE
Definition: types.h:120
#define FALSE
Definition: types.h:117
GLsizei GLenum const GLvoid GLsizei GLenum GLbyte GLbyte GLbyte GLdouble GLdouble GLdouble GLfloat GLfloat GLfloat GLint GLint GLint GLshort GLshort GLshort GLubyte GLubyte GLubyte GLuint GLuint GLuint GLushort GLushort GLushort GLbyte GLbyte GLbyte GLbyte GLdouble GLdouble GLdouble GLdouble GLfloat GLfloat GLfloat GLfloat GLint GLint GLint GLint GLshort GLshort GLshort GLshort GLubyte GLubyte GLubyte GLubyte GLuint GLuint GLuint GLuint GLushort GLushort GLushort GLushort GLboolean const GLdouble const GLfloat const GLint const GLshort const GLbyte const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLdouble const GLfloat const GLfloat const GLint const GLint const GLshort const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort GLenum GLenum GLenum GLfloat GLenum GLint GLenum GLenum GLenum GLfloat GLenum GLenum GLint GLenum GLfloat GLenum GLint GLint GLushort GLenum GLenum GLfloat GLenum GLenum GLint GLfloat const GLubyte GLenum GLenum GLenum const GLfloat GLenum GLenum const GLint GLenum GLint GLint GLsizei GLsizei GLint GLenum GLenum const GLvoid GLenum GLenum const GLfloat GLenum GLenum const GLint GLenum GLenum const GLdouble GLenum GLenum const GLfloat GLenum GLenum const GLint GLsizei GLuint GLfloat GLuint GLbitfield GLfloat GLint GLuint GLboolean GLenum GLfloat GLenum GLbitfield GLenum GLfloat GLfloat GLint GLint const GLfloat GLenum GLfloat GLfloat GLint GLint GLfloat GLfloat GLint GLint const GLfloat GLint GLfloat GLfloat GLint GLfloat GLfloat GLint GLfloat GLfloat const GLdouble const GLfloat const GLdouble const GLfloat GLint i
Definition: glfuncs.h:248
int32_t INT
Definition: typedefs.h:58
void * arg
Definition: msvc.h:10
#define _tcsicmp
Definition: xmlstorage.h:205

◆ DuplicateEnvironment()

LPTSTR DuplicateEnvironment ( VOID  )

Definition at line 25 of file setlocal.c.

26{
27 LPTSTR Environ = GetEnvironmentStrings();
28 LPTSTR End, EnvironCopy;
29
30 if (!Environ) return NULL;
31
32 for (End = Environ; *End; End += _tcslen(End) + 1) ;
33 EnvironCopy = cmd_alloc((End + 1 - Environ) * sizeof(TCHAR));
34
35 if (EnvironCopy)
36 memcpy(EnvironCopy, Environ, (End + 1 - Environ) * sizeof(TCHAR));
37
39 return EnvironCopy;
40}
#define memcpy(s1, s2, n)
Definition: mkisofs.h:878
#define FreeEnvironmentStrings
Definition: winbase.h:3731
LPSTR WINAPI GetEnvironmentStrings(void)

Referenced by _tmain(), cmd_endlocal(), and cmd_setlocal().