From 9cd240b08f573f59fe7f30ff51fab9ddeb6b267c Mon Sep 17 00:00:00 2001
From: Royce Mitchell III <royce3@ev1.net>
Date: Sun, 18 Sep 2005 17:52:28 +0000
Subject: [PATCH] (hopefully) last bugfix to set /a - assignment works again

svn path=/trunk/; revision=17916
---
 reactos/subsys/system/cmd/set.c | 33 +++++++++++++--------------------
 1 file changed, 13 insertions(+), 20 deletions(-)

diff --git a/reactos/subsys/system/cmd/set.c b/reactos/subsys/system/cmd/set.c
index 20cb253cae8..c55f147b109 100644
--- a/reactos/subsys/system/cmd/set.c
+++ b/reactos/subsys/system/cmd/set.c
@@ -408,36 +408,22 @@ seta_assignment ( LPCTSTR* p_, INT* result )
 	if ( identlen )
 	{
 		if ( *p == _T('=') )
-			op = *p, p++;
+			op = *p, p = skip_ws(p+1);
 		else if ( _tcschr ( _T("*/%+-&^|"), *p ) && p[1] == _T('=') )
-			op = *p, p += 2;
+			op = *p, p = skip_ws(p+2);
 		else if ( _tcschr ( _T("<>"), *p ) && *p == p[1] && p[2] == _T('=') )
-			op = *p, p += 3;
-		else
-		{
-			_tprintf ( _T("Missing operand.\n") );
-			return FALSE;
-		}
-		p = skip_ws ( p );
+			op = *p, p = skip_ws(p+3);
 	}
 
 	/* allow to chain multiple assignments, such as: a=b=1 */
 	if ( ident && op )
-	{
-		if ( !seta_assignment ( &p, &exprval ) )
-			return FALSE;
-	}
-	else
-	{
-		if ( !seta_expr ( &p, &exprval ) )
-			return FALSE;
-	}
-
-	if ( identlen )
 	{
 		INT identval;
 		LPTSTR buf;
 
+		if ( !seta_assignment ( &p, &exprval ) )
+			return FALSE;
+
 		if ( !seta_identval ( ident, &identval ) )
 			identval = 0;
 		switch ( op )
@@ -460,6 +446,13 @@ seta_assignment ( LPCTSTR* p_, INT* result )
 		SetEnvironmentVariable ( ident, buf ); // TODO FIXME - check return value
 		exprval = identval;
 	}
+	else
+	{
+		/* restore p in case we found an ident but not an op */
+		p = *p_;
+		if ( !seta_expr ( &p, &exprval ) )
+			return FALSE;
+	}
 
 	*result = exprval;
 	*p_ = p;