[jmorlan] 34566: Fix argument parsing for label command. It should not split the arguments by spaces; on Windows you can do 'label c:x y z' and it will set the label to X Y Z. (Bug 3530 #2)

Author: jmorlan
Date: Thu Jul 17 11:46:09 2008
New Revision: 34566

URL: http://svn.reactos.org/svn/reactos?rev=34566&view=rev
Log:
Fix argument parsing for label command. It should not split the arguments by spaces; on
Windows you can do 'label c:x y z' and it will set the label to X Y Z. (Bug 3530
#2)

Modified:
    trunk/reactos/base/shell/cmd/label.c

Modified: trunk/reactos/base/shell/cmd/label.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/base/shell/cmd/label.c?rev…
==============================================================================
--- trunk/reactos/base/shell/cmd/label.c [iso-8859-1] (original)
+++ trunk/reactos/base/shell/cmd/label.c [iso-8859-1] Thu Jul 17 11:46:09 2008
@@ -29,11 +29,11 @@
 	TCHAR  szLabel[80];
 	TCHAR  szOldLabel[80];
 	DWORD  dwSerialNr;
-	LPTSTR *arg;
-	INT    args;
 
 	/* set empty label string */
 	szLabel[0] = _T('\0');
+
+	nErrorLevel = 0;
 
 	/* print help */
 	if (!_tcsncmp (param, _T("/?"), 2))
@@ -42,55 +42,34 @@
 		return 0;
 	}
 
-  nErrorLevel = 0;
-
 	/* get parameters */
-	arg = split (param, &args, FALSE);
-
-	if (args > 2)
+	if (param[0] != _T('\0') && param[1] == _T(':'))
 	{
-		/* too many parameters */
-		error_too_many_parameters (arg[args - 1]);
-		freep (arg);
-    nErrorLevel = 1;
-		return 1;
+		szRootPath[0] = toupper(*param);
+		param += 2;
+		while (_istspace(*param))
+			param++;
 	}
-
-	if (args == 0)
+	else
 	{
 		/* get label of current drive */
 		TCHAR szCurPath[MAX_PATH];
 		GetCurrentDirectory (MAX_PATH, szCurPath);
 		szRootPath[0] = szCurPath[0];
 	}
-	else
-	{
-		if ((_tcslen (arg[0]) >= 2) && (arg[0][1] == _T(':')))
-		{
-			szRootPath[0] = toupper (*arg[0]);
-			if (args == 2)
-				_tcsncpy (szLabel, arg[1], 12);
-		}
-		else
-		{
-			TCHAR szCurPath[MAX_PATH];
-			GetCurrentDirectory (MAX_PATH, szCurPath);
-			szRootPath[0] = szCurPath[0];
-			_tcsncpy (szLabel, arg[0], 12);
-		}
-	}
+
+	_tcsncpy (szLabel, param, 12);
 
 	/* check root path */
 	if (!IsValidPathName (szRootPath))
 	{
 		error_invalid_drive ();
-		freep (arg);
-    nErrorLevel = 1;
+		nErrorLevel = 1;
 		return 1;
 	}
 
 	GetVolumeInformation(szRootPath, szOldLabel, 80, &dwSerialNr,
-			      NULL, NULL, NULL, 0);
+	                     NULL, NULL, NULL, 0);
 
 	/* print drive info */
 	if (szOldLabel[0] != _T('\0'))
@@ -110,7 +89,6 @@
 
 	if (szLabel[0] == _T('\0'))
 	{
-		LoadString(CMD_ModuleHandle, STRING_LABEL_HELP5, szMsg, RC_STRING_MAX_SIZE);
 		ConOutResPuts(STRING_LABEL_HELP5);
 
 		ConInString(szLabel, 80);
@@ -118,8 +96,6 @@
 
 	SetVolumeLabel(szRootPath, szLabel);
 
-	freep(arg);
-
 	return 0;
 }