Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-08-01 19:43:33 +00:00
Code Issues Projects Releases Packages Wiki Activity

[CMD] Code formatting; don't hardcode the string buffer sizes in function calls.

Browse source
This commit is contained in:
Hermès Bélusca-Maïto 2019-12-29 16:53:36 +01:00
parent 4bc5d44f06
commit 0bede0062a
No known key found for this signature in database
GPG key ID: 3B2539C65E7B93D0
1 changed files with 20 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

41
base/shell/cmd/prompt.c
View file

@ -52,7 +52,7 @@
static TCHAR DefaultPrompt[] = _T("$P$G");
/*
* Initialize prompt support
* Initialize prompt support.
*/
VOID InitPrompt(VOID)
{
@ -62,12 +62,12 @@ VOID InitPrompt(VOID)
* Set the PROMPT environment variable if it doesn't exist already.
* You can change the PROMPT environment variable before cmd starts.
*/
if (GetEnvironmentVariable(_T("PROMPT"), Buffer, ARRAYSIZE(Buffer)) == 0)
if (GetEnvironmentVariable(_T("PROMPT"), Buffer, _countof(Buffer)) == 0)
SetEnvironmentVariable(_T("PROMPT"), DefaultPrompt);
}
/*
* Print an information line on top of the screen
* Print an information line on top of the screen.
*/
VOID PrintInfoLine(VOID)
{
@ -104,14 +104,15 @@ VOID PrintInfoLine(VOID)
}
/*
* Print the command-line prompt
* Print the command-line prompt.
*/
VOID PrintPrompt(VOID)
{
TCHAR szPrompt[256];
LPTSTR pr;
TCHAR szPrompt[256];
TCHAR szPath[MAX_PATH];
if (GetEnvironmentVariable(_T("PROMPT"), szPrompt, 256))
if (GetEnvironmentVariable(_T("PROMPT"), szPrompt, _countof(szPrompt)))
pr = szPrompt;
else
pr = DefaultPrompt;
@ -171,20 +172,18 @@ VOID PrintPrompt(VOID)
break;
case _T('N'):
{
TCHAR szPath[MAX_PATH];
GetCurrentDirectory(MAX_PATH, szPath);
ConOutChar(szPath[0]);
}
{
GetCurrentDirectory(_countof(szPath), szPath);
ConOutChar(szPath[0]);
break;
}
case _T('P'):
{
TCHAR szPath[MAX_PATH];
GetCurrentDirectory(MAX_PATH, szPath);
ConOutPrintf(_T("%s"), szPath);
}
{
GetCurrentDirectory(_countof(szPath), szPath);
ConOutPrintf(_T("%s"), szPath);
break;
}
case _T('Q'):
ConOutChar(_T('='));
@ -212,12 +211,12 @@ VOID PrintPrompt(VOID)
#ifdef FEATURE_DIRECTORY_STACK
case '+':
{
INT i;
for (i = 0; i < GetDirectoryStackDepth (); i++)
ConOutChar(_T('+'));
}
{
INT i;
for (i = 0; i < GetDirectoryStackDepth(); i++)
ConOutChar(_T('+'));
break;
}
#endif
}
}