size_t s = _tcslen(pszText);
if (m_pszText)
- delete m_pszText;
+ delete[] m_pszText;
- m_pszText = new TCHAR [s+(b?3:1)]; // if we have spaces in unique part, we need 2 addtional chars for "
+ m_pszText = new (std::nothrow) TCHAR [s+(b?3:1)]; // if we have spaces in unique part, we need 2 addtional chars for "
if (!m_pszText)
return FALSE;
~CCompletionMatch()
{
if (m_pszText)
- delete m_pszText;
+ delete[] m_pszText;
}
private:
DeleteList();
if (m_pszContext)
- delete m_pszContext;
+ delete[] m_pszContext;
if (m_pszBegin)
- delete m_pszBegin;
+ delete[] m_pszBegin;
if (m_pszCurrentKey)
- delete m_pszCurrentKey;
+ delete[] m_pszCurrentKey;
}
void CCompletionList::Invalidate()
{
if (m_pszCurrentKey)
{
- delete m_pszCurrentKey;
+ delete[] m_pszCurrentKey;
m_pszCurrentKey = NULL;
}
}
rblnNew = TRUE;
if (m_pszContext)
{
- delete m_pszContext;
+ delete[] m_pszContext;
m_pszContext = NULL;
}
if (m_pszBegin)
{
- delete m_pszBegin;
+ delete[] m_pszBegin;
m_pszBegin = NULL;
}
if (m_pszCurrentKey)
{
- delete m_pszCurrentKey;
+ delete[] m_pszCurrentKey;
m_pszCurrentKey = NULL;
}
size_t s = _tcslen(pszContext);
- m_pszContext = new TCHAR[s+1];
+ m_pszContext = new (std::nothrow) TCHAR[s+1];
if (!m_pszContext)
return FALSE;
_tcscpy(m_pszContext,pszContext);
s = _tcslen(pszBegin);
- m_pszBegin = new TCHAR[s+1];
+ m_pszBegin = new (std::nothrow) TCHAR[s+1];
if (!m_pszBegin)
+ {
+ delete[] m_pszContext;
+ m_pszContext = NULL;
return FALSE;
+ }
_tcscpy(m_pszBegin,pszBegin);
s = _tcslen(pszCurrentKey);
- m_pszCurrentKey = new TCHAR[s+1];
+ m_pszCurrentKey = new (std::nothrow) TCHAR[s+1];
if (!m_pszCurrentKey)
+ {
+ delete[] m_pszContext;
+ delete[] m_pszBegin;
+ m_pszContext = NULL;
+ m_pszBegin = NULL;
return FALSE;
+ }
_tcscpy(m_pszCurrentKey,pszCurrentKey);
return TRUE;
{
if (_tcsnicmp(pszText,m_pszBegin,_tcslen(m_pszBegin)) != 0)
return TRUE;
- CCompletionMatch *pNode = new CCompletionMatch;
+ CCompletionMatch *pNode = new (std::nothrow) CCompletionMatch;
if (!pNode)
return FALSE;
if (!pNode->Init(pszText))
if (nError != ERROR_SUCCESS)
return FALSE;
- pszSubkeyName = new TCHAR[nKeyNameSize+dwMaxSubkeyNameLength+1];
+ pszSubkeyName = new (std::nothrow) TCHAR[nKeyNameSize+dwMaxSubkeyNameLength+1];
if (!pszSubkeyName)
goto Abort;
goto Abort;
}
- pszValueName = new TCHAR[nKeyNameSize+dwMaxValueNameSize+1];
+ pszValueName = new (std::nothrow) TCHAR[nKeyNameSize+dwMaxValueNameSize+1];
if (!pszValueName)
goto Abort;
}
if (pszValueName)
- delete pszValueName;
+ delete[] pszValueName;
if (pszSubkeyName)
- delete pszSubkeyName;
+ delete[] pszSubkeyName;
return TRUE;
Abort:
if (pszValueName)
- delete pszValueName;
+ delete[] pszValueName;
if (pszSubkeyName)
- delete pszSubkeyName;
+ delete[] pszSubkeyName;
return FALSE;
}
CConsole::~CConsole()
{
if (m_pchBuffer)
- delete m_pchBuffer;
+ delete[] m_pchBuffer;
if (m_pchBuffer1)
- delete m_pchBuffer1;
+ delete[] m_pchBuffer1;
if (m_pchBuffer2)
- delete m_pchBuffer2;
+ delete[] m_pchBuffer2;
if (m_blnOldInputModeSaved)
SetConsoleMode(m_hStdIn,m_dwOldInputMode);
if (m_pchBuffer2) delete m_pchBuffer2;
m_pchBuffer2 = NULL;
- m_pchBuffer = new TCHAR [dwBufferSize];
+ m_pchBuffer = new (std::nothrow) TCHAR [dwBufferSize];
if (!m_pchBuffer) goto Abort;
m_pchBuffer[dwBufferSize-1] = 0;
- m_pchBuffer1 = new TCHAR [dwBufferSize];
+ m_pchBuffer1 = new (std::nothrow) TCHAR [dwBufferSize];
if (!m_pchBuffer1) goto Abort;
m_pchBuffer1[dwBufferSize-1] = 0;
- m_pchBuffer2 = new TCHAR [dwBufferSize];
+ m_pchBuffer2 = new (std::nothrow) TCHAR [dwBufferSize];
if (!m_pchBuffer2) goto Abort;
m_pchBuffer2[dwBufferSize-1] = 0;
if (m_hStdOut != INVALID_HANDLE_VALUE) VERIFY(CloseHandle(m_hStdOut));
m_hStdOut = INVALID_HANDLE_VALUE;
- if (m_pchBuffer) delete m_pchBuffer;
+ if (m_pchBuffer) delete[] m_pchBuffer;
m_pchBuffer = NULL;
- if (m_pchBuffer1) delete m_pchBuffer1;
+ if (m_pchBuffer1) delete[] m_pchBuffer1;
m_pchBuffer1 = NULL;
- if (m_pchBuffer2) delete m_pchBuffer2;
+ if (m_pchBuffer2) delete[] m_pchBuffer2;
m_pchBuffer2 = NULL;
m_dwBufferSize = 0;
CPrompt::CPrompt(CRegistryTree& rTree, HRESULT& rhr):m_rTree(rTree)
{
- m_pszPrompt = new TCHAR[_tcslen(DEFAULT_PROMPT)+1];
+ m_pszPrompt = new (std::nothrow) TCHAR[_tcslen(DEFAULT_PROMPT)+1];
if (!m_pszPrompt)
{
rhr = E_OUTOFMEMORY;
return E_UNEXPECTED;
}
- m_pszPrompt = new TCHAR[_tcslen(pszPrompt)+1];
+ m_pszPrompt = new (std::nothrow) TCHAR[_tcslen(pszPrompt)+1];
if (!m_pszPrompt)
return E_OUTOFMEMORY;
CPrompt::~CPrompt()
{
if (m_pszPrompt)
- delete m_pszPrompt;
+ delete[] m_pszPrompt;
}
void CPrompt::ShowPrompt(CConsole &rConsole)
CArgumentParser Parser;
- pSettings = new CSettings();
+ pSettings = new (std::nothrow) CSettings();
if (!pSettings)
{
_ftprintf(stderr,_T("Cannot initialize settings. Out of memory.\n"));
goto Abort;
}
- pPrompt = new CPrompt(Tree,hr);
+ pPrompt = new (std::nothrow) CPrompt(Tree,hr);
if (!pPrompt)
{
_ftprintf(stderr,_T("Cannot initialize prompt. Out of memory.\n"));
{ // copy machine name
size_t size = _tcslen(pszMachineName);
- m_pszMachineName = new TCHAR [size+2];
+ m_pszMachineName = new (std::nothrow) TCHAR [size+2];
if (!m_pszMachineName)
return E_OUTOFMEMORY;
_tcscpy(m_pszMachineName,pszMachineName);
if (pszPath)
size += _tcslen(pszPath);
- m_pszKeyName = new TCHAR [size+2];
+ m_pszKeyName = new (std::nothrow) TCHAR [size+2];
if (!m_pszKeyName)
return E_OUTOFMEMORY;
_stprintf(m_pszKeyName,_T("%s%s\\"),pszPath?pszPath:_T(""),pszKeyName);
{
const TCHAR *pszKeyName = GetKeyName();
size_t size = _tcslen(pszKeyName) + _tcslen(pszSubkeyName) + 1;
- TCHAR *pszSubkeyFullName = new TCHAR [size];
+ TCHAR *pszSubkeyFullName = new (std::nothrow) TCHAR [size];
if (!pszSubkeyFullName)
{
nError = RegCloseKey(hKey);
_tcscpy(pszSubkeyFullName,pszKeyName);
_tcscat(pszSubkeyFullName,pszSubkeyName);
HRESULT hr = rKey.Init(hKey,GetKeyName(),pszSubkeyName,samDesired);
- delete pszSubkeyName;
+ delete[] pszSubkeyName;
if (FAILED(hr))
{
nError = RegCloseKey(hKey);
CRegistryTree::~CRegistryTree()
{
if (m_pszMachineName)
- delete m_pszMachineName;
+ delete[] m_pszMachineName;
CNode *pNode;
while(m_pCurrentKey->m_pUp)
const TCHAR *pszSeps = _T("\\");
// Make buffer and copy relative path into it.
- TCHAR *pszBuffer = new TCHAR[_tcslen(pszRelativePath)+1];
+ TCHAR *pszBuffer = new (std::nothrow) TCHAR[_tcslen(pszRelativePath)+1];
if (!pszBuffer)
{
SetError(ERROR_OUTOFMEMORY);
return TRUE;
Abort:
- delete pszBuffer;
+ delete[] pszBuffer;
return FALSE;
}
{
// Delete previous machine name buffer if allocated.
if (m_pszMachineName)
- delete m_pszMachineName;
+ delete[] m_pszMachineName;
m_pszMachineName = NULL;
m_Root.m_Key.InitRoot();
ASSERT(*pszMachineName); // No machine name.
- TCHAR *pszNewMachineName = new TCHAR[_tcslen(pszMachineName)+3]; // two leading backslashes + terminating null
+ TCHAR *pszNewMachineName = new (std::nothrow) TCHAR[_tcslen(pszMachineName)+3]; // two leading backslashes + terminating null
- if (!pszMachineName)
+ if (!pszNewMachineName)
{
SetError(ERROR_OUTOFMEMORY);
return FALSE;
// Delete previous machine name buffer if allocated.
if (m_pszMachineName)
- delete m_pszMachineName;
+ delete[] m_pszMachineName;
m_pszMachineName = pszNewMachineName;
return FALSE;
}
- TCHAR *pszSubkeyName = new TCHAR [dwMaxSubkeyNameLength];
+ TCHAR *pszSubkeyName = new (std::nothrow) TCHAR [dwMaxSubkeyNameLength];
rKey.InitSubkeyEnumeration(pszSubkeyName, dwMaxSubkeyNameLength);
BOOL blnKeyDeleted = FALSE;
while ((nError = rKey.GetNextSubkeyName()) == ERROR_SUCCESS)
BOOL CRegistryTree::InternalChangeCurrentKey(const TCHAR *pszSubkeyName, REGSAM DesiredAccess)
{
size_t size = _tcslen(pszSubkeyName);
- TCHAR *pszSubkeyNameBuffer = new TCHAR[size+3];
+ TCHAR *pszSubkeyNameBuffer = new (std::nothrow) TCHAR[size+3];
if (!pszSubkeyNameBuffer)
{
SetError(_T("Cannot open key : %s%s\nError %d (%s)\n"),
if (_tcscmp(pszSubkeyName,_T(".")) == 0)
{
- delete pszSubkeyNameBuffer;
+ delete[] pszSubkeyNameBuffer;
return TRUE;
}
// We are on root
ASSERT(m_pCurrentKey->m_pUp == NULL);
SetError(_T("Cannot open key. The root is not child.\n"));
- delete pszSubkeyNameBuffer;
+ delete[] pszSubkeyNameBuffer;
return FALSE;
}
if (!m_pCurrentKey->m_pUp)
{
SetInternalError();
- delete pszSubkeyNameBuffer;
+ delete[] pszSubkeyNameBuffer;
return FALSE;
}
CNode *pNode = m_pCurrentKey;
m_pCurrentKey = m_pCurrentKey->m_pUp;
delete pNode;
- delete pszSubkeyNameBuffer;
+ delete[] pszSubkeyNameBuffer;
return TRUE;
}
{
SetError(_T("Cannot open key : %s%s\nError %d (%s)\n"),
GetCurrentPath(),pszSubkeyName,ERROR_OUTOFMEMORY,GetErrorDescription(ERROR_OUTOFMEMORY));
- delete pszSubkeyNameBuffer;
+ delete[] pszSubkeyNameBuffer;
return FALSE;
}
if (!InternalGetSubkey(pszSubkeyName,DesiredAccess,pNewKey->m_Key))
{
delete pNewKey;
- delete pszSubkeyNameBuffer;
+ delete[] pszSubkeyNameBuffer;
return FALSE;
}
pNewKey->m_pUp = m_pCurrentKey;
m_pCurrentKey = pNewKey;
- delete pszSubkeyNameBuffer;
+ delete[] pszSubkeyNameBuffer;
return TRUE;
}
if (nError != ERROR_SUCCESS)
goto SkipCaseUpdate;
- pszSubkeyNameCaseUpdated = new TCHAR [dwMaxSubkeyNameLength];
+ pszSubkeyNameCaseUpdated = new (std::nothrow) TCHAR [dwMaxSubkeyNameLength];
m_pCurrentKey->m_Key.InitSubkeyEnumeration(pszSubkeyNameCaseUpdated, dwMaxSubkeyNameLength);
while ((nError = m_pCurrentKey->m_Key.GetNextSubkeyName()) == ERROR_SUCCESS)
if (_tcsicmp(pszSubkeyNameCaseUpdated, pszSubkeyName) == 0)
if (nError != ERROR_SUCCESS)
{
- delete pszSubkeyNameCaseUpdated;
+ delete[] pszSubkeyNameCaseUpdated;
pszSubkeyNameCaseUpdated = NULL;
}
goto Abort;
}
- delete pszSubkeyNameCaseUpdated;
+ delete[] pszSubkeyNameCaseUpdated;
}
else
{
return TRUE;
Abort:
if (pszSubkeyNameCaseUpdated)
- delete pszSubkeyNameCaseUpdated;
+ delete[] pszSubkeyNameCaseUpdated;
if (hNewKey)
{
return FALSE;
}
- TCHAR *pszShortKeyName = new TCHAR [size+1];
+ TCHAR *pszShortKeyName = new (std::nothrow) TCHAR [size+1];
if (!pszShortKeyName)
{
SetError(ERROR_OUTOFMEMORY);
// change to parent key
if (!Tree.InternalChangeCurrentKey(_T(".."),READ_CONTROL))
{
+ delete[] pszShortKeyName;
ASSERT(FALSE);
SetInternalError();
return FALSE;
// change back to target key
if (!Tree.InternalGetSubkey(pszShortKeyName,DesiredAccess,rKey))
{
+ delete[] pszShortKeyName;
SetError(Tree.GetLastErrorDescription());
return FALSE;
}
+ delete[] pszShortKeyName;
return TRUE;
}
{
if (m_pszPrompt)
{
- delete m_pszPrompt;
+ delete[] m_pszPrompt;
m_pszPrompt = NULL;
}
nError = RegQueryValueEx(hKey,PROMPT_VALUE_NAME,NULL,&dwType,NULL,&dwSize);
if (nError == ERROR_SUCCESS && dwType == REG_SZ)
{
- m_pszPrompt = (TCHAR *) new BYTE[dwSize];
+ m_pszPrompt = (TCHAR *) new (std::nothrow) BYTE[dwSize];
if (!m_pszPrompt)
{
hr = E_OUTOFMEMORY;
rConsole.Write(pchSID);
rConsole.Write(_T("\n"));
}
- delete pchSID;
+ delete[] pchSID;
DWORD dwNameBufferLength, dwDomainNameBufferLength;
dwNameBufferLength = 1024;
dwDomainNameBufferLength = 1024;
return 0;
}
- TCHAR *pszValueName = new TCHAR[dwMaxValueNameLength];
+ TCHAR *pszValueName = new (std::nothrow) TCHAR[dwMaxValueNameLength];
if (!pszValueName)
{
rConsole.Write("Out of memory.");
if (nError != ERROR_SUCCESS)
throw nError;
- TCHAR *pszSubkeyNameBuffer = new TCHAR[dwMaxSubkeyNameLength];
+ TCHAR *pszSubkeyNameBuffer = new (std::nothrow) TCHAR[dwMaxSubkeyNameLength];
if (!pszSubkeyNameBuffer)
throw ERROR_OUTOFMEMORY;
}
}
- delete pszSubkeyNameBuffer;
+ delete[] pszSubkeyNameBuffer;
if (nError != ERROR_NO_MORE_ITEMS)
throw nError;
if (nError != ERROR_SUCCESS)
throw nError;
- TCHAR *pchValueNameBuffer = new TCHAR[dwMaxValueNameBufferSize];
+ TCHAR *pchValueNameBuffer = new (std::nothrow) TCHAR[dwMaxValueNameBufferSize];
if (!pchValueNameBuffer)
throw ERROR_OUTOFMEMORY;
}
}
- delete pchValueNameBuffer;
+ delete[] pchValueNameBuffer;
if (nError != ERROR_NO_MORE_ITEMS)
throw nError;
goto Error;
}
- pSecurityDescriptor = (PISECURITY_DESCRIPTOR) new unsigned char [dwSecurityDescriptorLength];
+ pSecurityDescriptor = (PISECURITY_DESCRIPTOR) new (std::nothrow) unsigned char [dwSecurityDescriptorLength];
if (!pSecurityDescriptor)
{
_tcsncpy(pszError_msg,_T("\nOut of memory.\n"),ERROR_MSG_BUFFER_SIZE-1);
BOOL blnRet = GetTextualSid(pSID,NULL,&dwSIDStringSize);
ASSERT(!blnRet);
ASSERT(GetLastError() == ERROR_INSUFFICIENT_BUFFER);
- TCHAR *pszSID = new TCHAR[dwSIDStringSize];
+ TCHAR *pszSID = new (std::nothrow) TCHAR[dwSIDStringSize];
if (!pszSID)
{
rConsole.Write(pszSID);
rConsole.Write(_T("\n"));
}
- delete pszSID;
+ delete[] pszSID;
TCHAR *pszName, *pszDomainName;
DWORD dwNameBufferLength, dwDomainNameBufferLength;
dwNameBufferLength = 1024;
dwDomainNameBufferLength = 1024;
- pszName = new TCHAR [dwNameBufferLength];
+ pszName = new (std::nothrow) TCHAR [dwNameBufferLength];
if (!pszName)
{
_tcsncpy(pszError_msg,_T("\nOut of memory.\n"),ERROR_MSG_BUFFER_SIZE-1);
goto Error;
}
- pszDomainName = new TCHAR [dwDomainNameBufferLength];
+ pszDomainName = new (std::nothrow) TCHAR [dwDomainNameBufferLength];
if (!pszDomainName)
{
_tcsncpy(pszError_msg,_T("\nOut of memory.\n"),ERROR_MSG_BUFFER_SIZE-1);
+ delete[] pszName;
goto Error;
}
{
rConsole.Write(_T("\t\tKEY_QUERY_VALUE\n"));
}
+
+ delete[] pszName;
+ delete[] pszDomainName;
} // for
AbortDumpSACL:
SkipCommand:
if (pDataBuffer)
- delete pDataBuffer;
+ delete[] pDataBuffer;
return 0;
CommandNAonRoot:
SkipValueCommand:
if (pDataBuffer)
- delete pDataBuffer;
+ delete[] pDataBuffer;
return 0;
ValueCommandNAonRoot:
rConsole.Write(VALUE_CMD COMMAND_NA_ON_ROOT);
void CShellCommandsLinkedList::AddCommand(CShellCommand *pCommand)
{
// Create new node
- SNode *pNewNode = new SNode;
+ SNode *pNewNode = new (std::nothrow) SNode;
if (pNewNode == NULL)
return;
CTextHistory::~CTextHistory()
{
- if (m_pHistoryBuffer) delete m_pHistoryBuffer;
+ if (m_pHistoryBuffer) delete[] m_pHistoryBuffer;
}
BOOL CTextHistory::Init(DWORD dwMaxHistoryLineSize, DWORD dwMaxHistoryLines)
ASSERT(FALSE);
return FALSE;
}
- if (m_pHistoryBuffer) delete m_pHistoryBuffer;
+ if (m_pHistoryBuffer) delete[] m_pHistoryBuffer;
m_dwFirstHistoryIndex = 0;
m_dwLastHistoryIndex = 0;
m_dwHisoryFull = 0;
m_dwMaxHistoryLines = dwMaxHistoryLines;
m_dwMaxHistoryLineSize = dwMaxHistoryLineSize;
- m_pHistoryBuffer = new TCHAR [m_dwMaxHistoryLines*dwMaxHistoryLineSize];
+ m_pHistoryBuffer = new (std::nothrow) TCHAR [m_dwMaxHistoryLines*dwMaxHistoryLineSize];
if (!m_pHistoryBuffer) return FALSE;
return TRUE;
}
#include <winuser.h>
#include <winreg.h>
-#include <assert.h>
+#include <cassert>
#define ASSERT assert
#ifdef _DEBUG
#define VERIFY ASSERT
#endif
#include <conio.h>
-#include <limits.h>
+#include <climits>
#include <tchar.h>
-#include <stdio.h>
-#include <stdlib.h>
+#include <cstdio>
+#include <cstdlib>
#include <ctype.h>
+#include <new>
// INHERITED_ACE is from windows 2000
#ifndef INHERITED_ACE