Counter Strike : Global Offensive Source Code
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

1488 lines
46 KiB

//====== Copyright © 1996-2004, Valve Corporation, All rights reserved. =======
//
// Purpose:
//
//=============================================================================
#ifndef UTLSTRING_H
#define UTLSTRING_H
#ifdef _WIN32
#pragma once
#endif
#include "tier1/utlmemory.h"
#include "tier1/strtools.h"
#include "limits.h"
#ifndef SIZE_MAX
#define SIZE_MAX ((size_t) -1)
#endif
//-----------------------------------------------------------------------------
// Base class, containing simple memory management
//-----------------------------------------------------------------------------
class CUtlBinaryBlock
{
public:
CUtlBinaryBlock( int growSize = 0, int initSize = 0 );
~CUtlBinaryBlock()
{
#ifdef _DEBUG
m_nActualLength = 0x7BADF00D;
#else
m_nActualLength = 0;
#endif
}
// NOTE: nInitialLength indicates how much of the buffer starts full
CUtlBinaryBlock( void* pMemory, int nSizeInBytes, int nInitialLength );
CUtlBinaryBlock( const void* pMemory, int nSizeInBytes );
CUtlBinaryBlock( const CUtlBinaryBlock& src );
CUtlBinaryBlock &operator=( const CUtlBinaryBlock &src );
#if VALVE_CPP11
CUtlBinaryBlock( CUtlBinaryBlock&& src );
CUtlBinaryBlock &operator=( CUtlBinaryBlock&& src );
#endif
void Get( void *pValue, int nMaxLen ) const;
void Set( const void *pValue, int nLen );
const void *Get( ) const;
void *Get( );
unsigned char& operator[]( int i );
const unsigned char& operator[]( int i ) const;
int Length() const;
void SetLength( int nLength ); // Undefined memory will result
bool IsEmpty() const;
void Clear();
void Purge();
bool IsReadOnly() const;
// Test for equality
bool operator==( const CUtlBinaryBlock &src ) const;
private:
CUtlMemory<unsigned char> m_Memory;
int m_nActualLength;
};
//-----------------------------------------------------------------------------
// class inlines
//-----------------------------------------------------------------------------
#if VALVE_CPP11
inline CUtlBinaryBlock::CUtlBinaryBlock( CUtlBinaryBlock&& src )
: m_Memory( Move(src.m_Memory) )
, m_nActualLength( src.m_nActualLength )
{
src.m_nActualLength = 0;
}
inline CUtlBinaryBlock& CUtlBinaryBlock::operator= ( CUtlBinaryBlock&& src )
{
int length = src.m_nActualLength;
src.m_nActualLength = 0;
m_Memory = Move( src.m_Memory );
m_nActualLength = length;
return *this;
}
#endif
inline const void *CUtlBinaryBlock::Get( ) const
{
return m_Memory.Base();
}
inline void *CUtlBinaryBlock::Get( )
{
return m_Memory.Base();
}
inline int CUtlBinaryBlock::Length() const
{
return m_nActualLength;
}
inline unsigned char& CUtlBinaryBlock::operator[]( int i )
{
return m_Memory[i];
}
inline const unsigned char& CUtlBinaryBlock::operator[]( int i ) const
{
return m_Memory[i];
}
inline bool CUtlBinaryBlock::IsReadOnly() const
{
return m_Memory.IsReadOnly();
}
inline bool CUtlBinaryBlock::IsEmpty() const
{
return Length() == 0;
}
inline void CUtlBinaryBlock::Clear()
{
SetLength( 0 );
}
inline void CUtlBinaryBlock::Purge()
{
SetLength( 0 );
m_Memory.Purge();
}
//-----------------------------------------------------------------------------
// Simple string class.
// NOTE: This is *not* optimal! Use in tools, but not runtime code
//-----------------------------------------------------------------------------
class CUtlString
{
public:
CUtlString(); // = default;
CUtlString( const char *pString ); // initialize from c-style string
// Attaches the string to external memory. Useful for avoiding a copy
CUtlString( void* pMemory, int nSizeInBytes, int nInitialLength );
CUtlString( const void* pMemory, int nSizeInBytes );
// Copy/move constructor/assignment
// Moves are extremely efficient as the underlying memory is not copied, just the pointers.
CUtlString( const CUtlString& string ); // = default;
CUtlString &operator=( const CUtlString &src ); // = default;
#if VALVE_CPP11
CUtlString( CUtlString&& moveFrom ); // = default;
CUtlString &operator=( CUtlString&& moveFrom ); // = default;
#endif
// Also can assign from a regular C-style string
CUtlString &operator=( const char *src );
char *Access() { return Get(); }
const char *Get( ) const;
char *Get();
char* GetForModify();
void Clear() { Set( NULL ); }
// CUtlString can be used anywhere a c-style string would be required
// via this implicit conversion
operator const char*() const;
// for compatibility switching items from UtlSymbol
const char *String() const { return Get(); }
// Returns strlen
int Length() const;
bool IsEmpty() const;
// GS - Added for chromehtml
bool IsValid()
{
return (Length() != 0);
}
// Sets the length (used to serialize into the buffer )
// Note: If nLen != 0, then this adds an extra byte for a null-terminator.
void Set( const char *pValue );
void SetLength( int nLen );
void Purge();
void Swap(CUtlString &src);
// Case Change
void ToUpper();
void ToLower( );
void Append( const char *pchAddition );
void Append(const char *pchAddition, int nMaxChars);
void Append(char chAddition) {
char temp[2] = { chAddition, 0 };
Append(temp);
}
// Strips the trailing slash
void StripTrailingSlash();
char operator[] ( int idx ) const;
char& operator[] ( int idx );
// Test for equality
bool operator==( const CUtlString &src ) const;
bool operator==( const char *src ) const;
bool operator!=( const CUtlString &src ) const { return !operator==( src ); }
bool operator!=( const char *src ) const { return !operator==( src ); }
// If these are not defined, CUtlString as rhs will auto-convert
// to const char* and do logical operations on the raw pointers. Ugh.
inline friend bool operator==( const char *lhs, const CUtlString &rhs ) { return rhs.operator==( lhs ); }
inline friend bool operator!=( const char *lhs, const CUtlString &rhs ) { return rhs.operator!=( lhs ); }
CUtlString &operator+=( const CUtlString &rhs );
CUtlString &operator+=( const char *rhs );
CUtlString &operator+=( char c );
CUtlString &operator+=( int rhs );
CUtlString &operator+=( double rhs );
CUtlString operator+( const char *pOther )const;
CUtlString operator+( int rhs )const;
bool MatchesPattern( const CUtlString &Pattern, int nFlags = 0 ); // case SENSITIVE, use * for wildcard in pattern string
#if ! defined(SWIG)
// Don't let SWIG see the PRINTF_FORMAT_STRING attribute or it will complain.
int Format( PRINTF_FORMAT_STRING const char *pFormat, ... ) FMTFUNCTION( 2, 3 );
int FormatV( PRINTF_FORMAT_STRING const char *pFormat, va_list marker );
#else
int Format( const char *pFormat, ... );
int FormatV( const char *pFormat, va_list marker );
#endif
void SetDirect( const char *pValue, int nChars );
// Defining AltArgumentType_t hints that associative container classes should
// also implement Find/Insert/Remove functions that take const char* params.
typedef const char *AltArgumentType_t;
// Take a piece out of the string.
// If you only specify nStart, it'll go from nStart to the end.
// You can use negative numbers and it'll wrap around to the start.
CUtlString Slice( int32 nStart=0, int32 nEnd=INT_MAX );
// Grab a substring starting from the left or the right side.
CUtlString Left( int32 nChars );
CUtlString Right( int32 nChars );
CUtlString Remove(char const *pTextToRemove, bool bCaseSensitive) const;
// Replace all instances of one character with another.
CUtlString Replace( char cFrom, char cTo );
/// Replace one string with the other (single pass). Passing a NULL to pchTo is same as calling Remove
CUtlString Replace( char const *pchFrom, const char *pchTo, bool bCaseSensitive = false ) const;
/// helper func for caseless replace
CUtlString ReplaceCaseless( char const *pchFrom, const char *pchTo ) const { return Replace( pchFrom, pchTo, false ); }
void RemoveDotSlashes(char separator = CORRECT_PATH_SEPARATOR);
// Trim whitespace
void TrimLeft( const char *szTargets = "\t\r\n " );
void TrimRight( const char *szTargets = "\t\r\n " );
void Trim( const char *szTargets = "\t\r\n " );
// Calls right through to V_MakeAbsolutePath.
CUtlString AbsPath( const char *pStartingDir=NULL ) const;
CUtlString AbsPath(const char *pStartingDir, bool bLowercaseName) const
{
CUtlString result = AbsPath(pStartingDir);
if (bLowercaseName)
{
result.ToLower();
}
return result;
}
// Gets the filename (everything except the path.. c:\a\b\c\somefile.txt -> somefile.txt).
CUtlString UnqualifiedFilename() const;
// Strips off one directory. Uses V_StripLastDir but strips the last slash also!
CUtlString DirName();
// Get a string with the extension removed (with V_StripExtension).
CUtlString StripExtension() const;
// Get a string with the filename removed (uses V_UnqualifiedFileName and also strips the last slash)
CUtlString StripFilename() const;
// Get a string with the base filename (with V_FileBase).
CUtlString GetBaseFilename() const;
// Get a string with the file extension (with V_FileBase).
CUtlString GetExtension() const;
// Works like V_ComposeFileName.
static CUtlString PathJoin( const char *pStr1, const char *pStr2 );
// These can be used for utlvector sorts.
static int __cdecl SortCaseInsensitive( const CUtlString *pString1, const CUtlString *pString2 );
static int __cdecl SortCaseSensitive( const CUtlString *pString1, const CUtlString *pString2 );
// From Src2
void FixSlashes( char cSeparator = CORRECT_PATH_SEPARATOR )
{
for ( int nLength = Length() - 1; nLength >= 0; nLength-- )
{
char *pname = (char*)&m_Storage[ nLength ];
if ( *pname == INCORRECT_PATH_SEPARATOR || *pname == CORRECT_PATH_SEPARATOR )
{
*pname = cSeparator;
}
}
}
bool IsEqual_CaseSensitive( const char *src ) const
{
if ( !src )
{
return ( Length() == 0 );
}
return ( V_strcmp( Get(), src ) == 0 );
}
bool IsEqual_CaseInsensitive(const char *src) const
{
if (!src)
{
return (Length() == 0);
}
return (V_stricmp(Get(), src) == 0);
}
private:
CUtlBinaryBlock m_Storage;
};
//-----------------------------------------------------------------------------
// Inline methods
//-----------------------------------------------------------------------------
#if VALVE_CPP11
inline CUtlString::CUtlString( CUtlString&& moveFrom )
: m_Storage( Move( moveFrom.m_Storage ) )
{
}
inline CUtlString& CUtlString::operator=( CUtlString&& moveFrom )
{
m_Storage = Move( moveFrom.m_Storage );
return *this;
}
#endif
inline bool CUtlString::IsEmpty() const
{
return Length() == 0;
}
inline int __cdecl CUtlString::SortCaseInsensitive( const CUtlString *pString1, const CUtlString *pString2 )
{
return V_stricmp( pString1->String(), pString2->String() );
}
inline int __cdecl CUtlString::SortCaseSensitive( const CUtlString *pString1, const CUtlString *pString2 )
{
return V_strcmp( pString1->String(), pString2->String() );
}
inline char CUtlString::operator [] ( int index ) const
{
return Get()[index];
}
inline char& CUtlString::operator[] ( int index )
{
return Access()[index];
}
inline CUtlString::operator const char *( ) const
{
return Get();
}
//-----------------------------------------------------------------------------
// Purpose: Implementation of low-level string functionality for character types.
//-----------------------------------------------------------------------------
template < typename T >
class StringFuncs
{
public:
static T *Duplicate( const T *pValue );
static void Copy( T *out_pOut, const T *pIn, int iLength );
static int Compare( const T *pLhs, const T *pRhs );
static int Length( const T *pValue );
static const T *FindChar( const T *pStr, const T cSearch );
static const T *EmptyString();
};
template < >
class StringFuncs<char>
{
public:
static char *Duplicate( const char *pValue ) { return strdup( pValue ); }
static void Copy( char *out_pOut, const char *pIn, int iLength ) { strncpy( out_pOut, pIn, iLength ); }
static int Compare( const char *pLhs, const char *pRhs ) { return strcmp( pLhs, pRhs ); }
static int Length( const char *pValue ) { return (int)strlen( pValue ); }
static const char *FindChar( const char *pStr, const char cSearch ) { return strchr( pStr, cSearch ); }
static const char *EmptyString() { return ""; }
};
template < >
class StringFuncs<wchar_t>
{
public:
static wchar_t *Duplicate( const wchar_t *pValue ) { return wcsdup( pValue ); }
static void Copy( wchar_t *out_pOut, const wchar_t *pIn, int iLength ) { wcsncpy( out_pOut, pIn, iLength ); }
static int Compare( const wchar_t *pLhs, const wchar_t *pRhs ) { return wcscmp( pLhs, pRhs ); }
static int Length( const wchar_t *pValue ) { return (int) wcslen( pValue ); }
static const wchar_t *FindChar( const wchar_t *pStr, const wchar_t cSearch ) { return wcschr( pStr, cSearch ); }
static const wchar_t *EmptyString() { return L""; }
};
//-----------------------------------------------------------------------------
// Dirt-basic auto-release string class. Not intended for manipulation,
// can be stored in a container or forwarded as a functor parameter.
// Note the benefit over CUtlString: sizeof(CUtlConstString) == sizeof(char*).
// Also note: null char* pointers are treated identically to empty strings.
//-----------------------------------------------------------------------------
template < typename T = char >
class CUtlConstStringBase
{
public:
CUtlConstStringBase() : m_pString( NULL ) {}
CUtlConstStringBase( const T *pString ) : m_pString( NULL ) { Set( pString ); }
CUtlConstStringBase( const CUtlConstStringBase& src ) : m_pString( NULL ) { Set( src.m_pString ); }
~CUtlConstStringBase() { Set( NULL ); }
void Set( const T *pValue );
void Clear() { Set( NULL ); }
const T *Get() const { return m_pString ? m_pString : StringFuncs<T>::EmptyString(); }
operator const T*() const { return m_pString ? m_pString : StringFuncs<T>::EmptyString(); }
bool IsEmpty() const { return m_pString == NULL; } // Note: empty strings are never stored by Set
int Compare( const T *rhs ) const;
// Logical ops
bool operator<( const T *rhs ) const { return Compare( rhs ) < 0; }
bool operator==( const T *rhs ) const { return Compare( rhs ) == 0; }
bool operator!=( const T *rhs ) const { return Compare( rhs ) != 0; }
bool operator<( const CUtlConstStringBase &rhs ) const { return Compare( rhs.m_pString ) < 0; }
bool operator==( const CUtlConstStringBase &rhs ) const { return Compare( rhs.m_pString ) == 0; }
bool operator!=( const CUtlConstStringBase &rhs ) const { return Compare( rhs.m_pString ) != 0; }
// If these are not defined, CUtlConstString as rhs will auto-convert
// to const char* and do logical operations on the raw pointers. Ugh.
inline friend bool operator<( const T *lhs, const CUtlConstStringBase &rhs ) { return rhs.Compare( lhs ) > 0; }
inline friend bool operator==( const T *lhs, const CUtlConstStringBase &rhs ) { return rhs.Compare( lhs ) == 0; }
inline friend bool operator!=( const T *lhs, const CUtlConstStringBase &rhs ) { return rhs.Compare( lhs ) != 0; }
CUtlConstStringBase &operator=( const T *src ) { Set( src ); return *this; }
CUtlConstStringBase &operator=( const CUtlConstStringBase &src ) { Set( src.m_pString ); return *this; }
// Defining AltArgumentType_t is a hint to containers that they should
// implement Find/Insert/Remove functions that take const char* params.
typedef const T *AltArgumentType_t;
protected:
const T *m_pString;
};
template < typename T >
void CUtlConstStringBase<T>::Set( const T *pValue )
{
if ( pValue != m_pString )
{
free( ( void* ) m_pString );
m_pString = pValue && pValue[0] ? StringFuncs<T>::Duplicate( pValue ) : NULL;
}
}
template < typename T >
int CUtlConstStringBase<T>::Compare( const T *rhs ) const
{
if ( m_pString )
{
if ( rhs )
return StringFuncs<T>::Compare( m_pString, rhs );
else
return 1;
}
else
{
if ( rhs )
return -1;
else
return 0;
}
}
inline CUtlString CUtlString::operator+( const char *pOther )const
{
CUtlString s = *this;
s += pOther;
return s;
}
typedef CUtlConstStringBase<char> CUtlConstString;
typedef CUtlConstStringBase<wchar_t> CUtlConstWideString;
//-----------------------------------------------------------------------------
// Purpose: General purpose string class good for when it
// is rarely expected to be empty, and/or will undergo
// many modifications/appends.
//-----------------------------------------------------------------------------
class CUtlStringBuilder
{
public:
CUtlStringBuilder();
CUtlStringBuilder(const char *pchString);
CUtlStringBuilder(CUtlStringBuilder const &string);
explicit CUtlStringBuilder(size_t nPreallocateBytes);
//CUtlStringBuilder( const CUtlStringResult &strMoveSource );
//explicit CUtlStringBuilder( IFillStringFunctor& func );
~CUtlStringBuilder();
// operator=
CUtlStringBuilder &operator=(const CUtlStringBuilder &src);
CUtlStringBuilder &operator=(const char *pchString);
//CUtlStringBuilder &operator=( const CUtlStringResult &strMoveSource );
// operator==
bool operator==(CUtlStringBuilder const &src) const;
bool operator==(const char *pchString) const;
// operator!=
bool operator!=(CUtlStringBuilder const &src) const;
bool operator!=(const char *pchString) const;
// operator </>, performs case sensitive comparison
bool operator<(const CUtlStringBuilder &val) const;
bool operator<(const char *pchString) const;
bool operator>(const CUtlStringBuilder &val) const;
bool operator>(const char *pchString) const;
// operator+=
CUtlStringBuilder &operator+=(const char *rhs);
// is valid?
bool IsValid() const;
// gets the string
// never returns NULL, use IsValid() to see if it's never been set
const char *String() const;
const char *Get() const { return String(); }
operator const char *() const { return String(); }
// returns the string directly (could be NULL)
// useful for doing inline operations on the string
char *Access()
{
// aggressive warning that there has been a bad error;
// probably should not be retrieving the string by pointer.
Assert(!m_data.HasError());
if (!IsValid() || (Capacity() == 0))
return NULL;
return m_data.Access();
}
// return false if capacity can't be set. If false is returned,
// the error state is set.
bool EnsureCapacity(size_t nLength);
// append in-place, causing a re-allocation
void Append(const char *pchAddition);
void Append(const char *pchAddition, size_t cbLen);
void Append(const CUtlStringBuilder &str) { Append(str.String(), str.Length()); }
//void Append( IFillStringFunctor& func );
void AppendChar(char ch) { Append(&ch, 1); }
void AppendRepeat(char ch, int cCount);
// sets the string
void SetValue(const char *pchString);
void Set(const char *pchString);
void Clear() { m_data.Clear(); }
void SetPtr(char *pchString);
void SetPtr(char *pchString, size_t nLength);
void Swap(CUtlStringBuilder &src);
// If you want to take ownership of the ptr, you can use this. So for instance if you had
// a CUtlString you wanted to move it to a CUtlStringConst without making a copy, you
// could do: CUtlStringConst strConst( strUtl.Detach() );
// Also used for fast temporaries when a string that does not need to be retained is being
// returned from a func. ie: return (str.Detach());
// All strings in this file can take a CUtlStringResult as a constructor parm and will take ownership directly.
//CUtlStringResult Detach();
// Set directly and don't look for a null terminator in pValue.
// nChars is the string length. "abcd" nChars==3 would copy and null
// terminate "abc" in the string object.
void SetDirect(const char *pchString, size_t nChars);
// Get the length of the string in characters.
size_t Length() const;
bool IsEmpty() const;
size_t Capacity() const { return m_data.Capacity(); } // how much room is there to scribble
#if !defined(SWIG)
// SWIG fails on PRINTF_FORMAT_STRING
// Format like sprintf.
size_t Format(PRINTF_FORMAT_STRING const char *pFormat, ...) FMTFUNCTION(2, 3);
// format, then append what we crated in the format
size_t AppendFormat(PRINTF_FORMAT_STRING const char *pFormat, ...) FMTFUNCTION(2, 3);
#else
size_t Format(const char *pFormat, ...);
size_t AppendFormat(const char *pFormat, ...);
#endif
// replace a single character with another, returns hit count
size_t Replace(char chTarget, char chReplacement);
// replace a string with another string, returns hit count
// replacement string might be NULL or "" to remove target substring
size_t Replace(const char *pstrTarget, const char *pstrReplacement);
// ASCII-only case-insensitivity.
size_t ReplaceFastCaseless(const char *pstrTarget, const char *pstrReplacement);
// replace a sequence of characters at a given point with a new string
// replacement string might be NULL or "" to remove target substring
bool ReplaceAt(size_t nIndex, size_t nOldChars, const char *pNewStr, size_t nNewChars = SIZE_MAX);
ptrdiff_t IndexOf(const char *pstrTarget) const;
// remove whitespace from the string; anything that is isspace()
size_t RemoveWhitespace();
// trim whitepace from the beginning and end of the string
size_t TrimWhitespace();
// Allows setting the size to anything under the current
// capacity. Typically should not be used unless there was a specific
// reason to scribble on the string. Will not touch the string contents,
// but will append a NULL. Returns true if the length was changed.
bool SetLength(size_t nLen);
// Take responsibility for the string. May cause a heap alloc.
char *TakeOwnership(size_t *pnLen, size_t *pnCapacity);
// For operations that are long and/or complex - if something fails
// along the way, the error will be set and can be queried at the end.
// The string is undefined in the error state, but will likely hold the
// last value before the error occurred. The string is cleared
// if ClearError() is called. The error can be set be the user, and it
// will also be set if a dynamic allocation fails in string operations
// where it needs to grow the capacity.
void SetError() { m_data.SetError(true); }
void ClearError() { m_data.ClearError(); }
bool HasError() const { return m_data.HasError(); }
#ifdef DBGFLAG_VALIDATE
void Validate(CValidator &validator, const char *pchName); // validate our internal structures
#endif // DBGFLAG_VALIDATE
size_t VFormat(const char *pFormat, va_list args);
size_t VAppendFormat(const char *pFormat, va_list args);
void Truncate(size_t nChars);
// Access() With no assertion check - should only be used for tests
char *AccessNoAssert()
{
if (!IsValid())
return NULL;
return m_data.Access();
}
// SetError() With no assertion check - should only be used for tests
void SetErrorNoAssert() { m_data.SetError(false); }
size_t ReplaceCaseless(const char *pstrTarget, const char *pstrReplacement) {
return ReplaceFastCaseless(pstrTarget, pstrReplacement);
}
private:
size_t ReplaceInternal(const char *pstrTarget, const char *pstrReplacement, const char *pfnCompare(const char*, const char*));
operator bool() const { return IsValid(); }
// nChars is the number of characters you want, NOT including the null
char *PrepareBuffer(size_t nChars, bool bCopyOld = false, size_t nMinCapacity = 0)
{
char *pszString = NULL;
size_t nCapacity = m_data.Capacity();
if ((nChars <= nCapacity) && (nMinCapacity <= nCapacity))
{
// early out leaving it all alone, just update the length,
// even if it shortens an existing heap string to a width
// that would fit in the stack buffer.
pszString = m_data.SetLength(nChars);
// SetLength will have added the null. Pointer might
// be NULL if there is an error state and no buffer
Assert(!pszString || pszString[nChars] == '\0');
return pszString;
}
if (HasError())
return NULL;
// Need to actually adjust the capacity
return InternalPrepareBuffer(nChars, bCopyOld, Max(nChars, nMinCapacity));
}
char *InternalPrepareBuffer(size_t nChars, bool bCopyOld, size_t nMinCapacity);
template <typename T>
void Swap(T &p1, T &p2)
{
T t = p1;
p1 = p2;
p2 = t;
}
// correct for 32 or 64 bit
static const uint8 MAX_STACK_STRLEN = (sizeof(void*) == 4 ? 15 : 19);
enum
{
// Note: If it's ever desired to have the embedded string be larger than
// 63 (0x40-1), just make the sentinal 0xFF, and the error something (0xFF also
// is fine). Then shrink the scrap size by 1 and add a uint8 for the error state.
// The error byte is only valid if the heap is on (already have that restriction).
// and then embedded strings can get back to being up to 254. It's not done this
// way now just to make the tests for IsHeap()/HasError() faster since they
// are often both tested together the compiler can do nice bit test optimizations.
STRING_TYPE_SENTINEL = 0x80,
STRING_TYPE_ERROR = 0x40
}; // if Data.Stack.BytesLeft() or Data.Heap.sentinel == this value, data is in heap
union Data
{
struct _Heap
{ // 16 on 32 bit, sentinel == 0xff if Heap is the active union item
private:
char *m_pchString;
uint32 m_nLength;
uint32 m_nCapacity; // without trailing null; ie: m_pchString[m_nCapacity] = '\0' is not out of bounds
uint8 scrap[3];
uint8 sentinel;
public:
friend union Data;
friend char *CUtlStringBuilder::InternalPrepareBuffer(size_t, bool, size_t);
} Heap;
struct _Stack
{
private:
// last byte is doing a hack double duty. It holds how many bytes
// are left in the string; so when the string is 'full' it will be
// '0' and thus suffice as the terminating null. This is why
// we hold remaining chars instead of 'string length'
char m_szString[MAX_STACK_STRLEN + 1];
public:
uint8 BytesLeft() const { return (uint8)(m_szString[MAX_STACK_STRLEN]); }
void SetBytesLeft(char n) { m_szString[MAX_STACK_STRLEN] = n; }
friend char *CUtlStringBuilder::InternalPrepareBuffer(size_t, bool, size_t);
friend union Data;
} Stack;
// set to a clear state without looking at the current state
void Construct()
{
Stack.m_szString[0] = '\0';
Stack.SetBytesLeft(MAX_STACK_STRLEN);
}
// If we have heap allocated data, free it
void FreeHeap()
{
if (IsHeap() && Heap.m_pchString)
MemAlloc_Free(Heap.m_pchString);
}
// Back to a clean state, but retain the error state.
void Clear()
{
if (HasError())
return;
FreeHeap();
Heap.m_pchString = NULL;
Construct();
}
bool IsHeap() const { return ((Heap.sentinel & STRING_TYPE_SENTINEL) != 0); }
char *Access() { return IsHeap() ? Heap.m_pchString : Stack.m_szString; }
const char *String() const { return IsHeap() ? Heap.m_pchString : Stack.m_szString; }
size_t Length() const { return IsHeap() ? Heap.m_nLength : (MAX_STACK_STRLEN - Stack.BytesLeft()); }
bool IsEmpty() const
{
if (IsHeap())
return Heap.m_nLength == 0;
else
return Stack.BytesLeft() == MAX_STACK_STRLEN; // empty if all the bytes are available
}
size_t Capacity() const { return IsHeap() ? Heap.m_nCapacity : MAX_STACK_STRLEN; }
// Internally the code often needs the char * after setting the length, so
// just return it from here for conveniences.
char *SetLength(size_t nChars);
// Give the string away and set to an empty state
char *TakeOwnership(size_t &nLen, size_t &nCapacity)
{
MoveToHeap();
if (HasError())
{
nLen = 0;
nCapacity = 0;
return NULL;
}
nLen = Heap.m_nLength;
nCapacity = Heap.m_nCapacity;
char *psz = Heap.m_pchString;
Construct();
return psz;
}
void SetPtr(char *pchString, size_t nLength);
// Set the string to an error state
void SetError(bool bEnableAssert);
// clear the error state and reset the string
void ClearError();
// If string is in the heap and the error bit is set in the sentinel
// the error state is true.
bool HasError() const { return IsHeap() && ((Heap.sentinel & STRING_TYPE_ERROR) != 0); }
// If it's stack based, get it to the heap and return if it is
// successfully on the heap (or already was)
bool MoveToHeap();
private:
//-----------------------------------------------------------------------------
// Purpose: Needed facts for string class to work
//-----------------------------------------------------------------------------
void StaticAssertTests()
{
// If this fails when the heap sentinel and where the stack string stores its bytes left
// aren't aliases. This is needed so that regardless of how the 'sentinel' to mark
// that the string is on the heap is set, it is set as expected on both sides of the union.
COMPILE_TIME_ASSERT(offsetof(_Heap, sentinel) == (offsetof(_Stack, m_szString) + MAX_STACK_STRLEN));
// Lots of code assumes it can look at m_data.Stack.m_nBytesLeft for an empty string; which
// means that it will equal MAX_STACK_STRLEN. Therefor it must be a different value than
// the STRING_TYPE_SENTINEL which will be set if the string is in the heap.
COMPILE_TIME_ASSERT(MAX_STACK_STRLEN < STRING_TYPE_SENTINEL);
COMPILE_TIME_ASSERT(MAX_STACK_STRLEN < STRING_TYPE_ERROR);
// this is a no brainer, and I don't know anywhere in the world this isn't true,
// but this code does take this dependency.
COMPILE_TIME_ASSERT(0 == '\0');
}
};
private: // data
Data m_data;
};
//-----------------------------------------------------------------------------
// Purpose: constructor
//-----------------------------------------------------------------------------
inline CUtlStringBuilder::CUtlStringBuilder()
{
m_data.Construct();
}
//-----------------------------------------------------------------------------
// Purpose: constructor
//-----------------------------------------------------------------------------
inline CUtlStringBuilder::CUtlStringBuilder(size_t nPreallocateBytes)
{
if (nPreallocateBytes <= MAX_STACK_STRLEN)
{
m_data.Construct();
}
else
{
m_data.Construct();
PrepareBuffer(0, false, nPreallocateBytes);
}
}
//-----------------------------------------------------------------------------
// Purpose: constructor
//-----------------------------------------------------------------------------
inline CUtlStringBuilder::CUtlStringBuilder(const char *pchString)
{
m_data.Construct();
SetDirect(pchString, pchString ? strlen(pchString) : 0);
}
//-----------------------------------------------------------------------------
// Purpose: constructor
//-----------------------------------------------------------------------------
inline CUtlStringBuilder::CUtlStringBuilder(CUtlStringBuilder const &string)
{
m_data.Construct();
SetDirect(string.String(), string.Length());
// attempt the copy before checking for error. On the off chance there
// is data there that can be set, it will help with debugging.
if (string.HasError())
m_data.SetError(false);
}
//-----------------------------------------------------------------------------
// Purpose: destructor
//-----------------------------------------------------------------------------
inline CUtlStringBuilder::~CUtlStringBuilder()
{
m_data.FreeHeap();
}
//-----------------------------------------------------------------------------
// Purpose: Pre-Widen a string to an expected length
//-----------------------------------------------------------------------------
inline bool CUtlStringBuilder::EnsureCapacity(size_t nLength)
{
return PrepareBuffer(Length(), true, nLength) != NULL;
}
//-----------------------------------------------------------------------------
// Purpose: ask if the string has anything in it
//-----------------------------------------------------------------------------
inline bool CUtlStringBuilder::IsEmpty() const
{
return m_data.IsEmpty();
}
//-----------------------------------------------------------------------------
// Purpose: assignment
//-----------------------------------------------------------------------------
inline CUtlStringBuilder &CUtlStringBuilder::operator=(const char *pchString)
{
SetDirect(pchString, pchString ? strlen(pchString) : 0);
return *this;
}
//-----------------------------------------------------------------------------
// Purpose: assignment
//-----------------------------------------------------------------------------
inline CUtlStringBuilder &CUtlStringBuilder::operator=(CUtlStringBuilder const &src)
{
if (&src != this)
{
SetDirect(src.String(), src.Length());
// error propagate
if (src.HasError())
m_data.SetError(false);
}
return *this;
}
//-----------------------------------------------------------------------------
// Purpose: comparison
//-----------------------------------------------------------------------------
inline bool CUtlStringBuilder::operator==(CUtlStringBuilder const &src) const
{
return !Q_strcmp(String(), src.String());
}
//-----------------------------------------------------------------------------
// Purpose: comparison
//-----------------------------------------------------------------------------
inline bool CUtlStringBuilder::operator==(const char *pchString) const
{
return !Q_strcmp(String(), pchString);
}
//-----------------------------------------------------------------------------
// Purpose: comparison
//-----------------------------------------------------------------------------
inline bool CUtlStringBuilder::operator!=(CUtlStringBuilder const &src) const
{
return !(*this == src);
}
//-----------------------------------------------------------------------------
// Purpose: comparison
//-----------------------------------------------------------------------------
inline bool CUtlStringBuilder::operator!=(const char *pchString) const
{
return !(*this == pchString);
}
//-----------------------------------------------------------------------------
// Purpose: comparison
//-----------------------------------------------------------------------------
inline bool CUtlStringBuilder::operator<(CUtlStringBuilder const &val) const
{
return operator<(val.String());
}
//-----------------------------------------------------------------------------
// Purpose: comparison
//-----------------------------------------------------------------------------
inline bool CUtlStringBuilder::operator<(const char *pchString) const
{
return Q_strcmp(String(), pchString) < 0;
}
//-----------------------------------------------------------------------------
// Purpose: comparison
//-----------------------------------------------------------------------------
inline bool CUtlStringBuilder::operator>(CUtlStringBuilder const &val) const
{
return Q_strcmp(String(), val.String()) > 0;
}
//-----------------------------------------------------------------------------
// Purpose: comparison
//-----------------------------------------------------------------------------
inline bool CUtlStringBuilder::operator>(const char *pchString) const
{
return Q_strcmp(String(), pchString) > 0;
}
//-----------------------------------------------------------------------------
// Return a string with this string and rhs joined together.
inline CUtlStringBuilder& CUtlStringBuilder::operator+=(const char *rhs)
{
Append(rhs);
return *this;
}
//-----------------------------------------------------------------------------
// Purpose: returns true if the string is not null
//-----------------------------------------------------------------------------
inline bool CUtlStringBuilder::IsValid() const
{
return !HasError();
}
//-----------------------------------------------------------------------------
// Purpose: data accessor
//-----------------------------------------------------------------------------
inline const char *CUtlStringBuilder::String() const
{
const char *pszString = m_data.String();
if (pszString)
return pszString;
// pszString can be NULL in the error state. For const char*
// never return NULL.
return "";
}
//-----------------------------------------------------------------------------
// Purpose: Sets the string to be the new value, taking a copy of it
//-----------------------------------------------------------------------------
inline void CUtlStringBuilder::SetValue(const char *pchString)
{
size_t nLen = (pchString ? strlen(pchString) : 0);
SetDirect(pchString, nLen);
}
//-----------------------------------------------------------------------------
// Purpose: Set directly and don't look for a null terminator in pValue.
//-----------------------------------------------------------------------------
inline void CUtlStringBuilder::SetDirect(const char *pchSource, size_t nChars)
{
if (HasError())
return;
if (m_data.IsHeap() && Get() == pchSource)
return;
if (!pchSource || !nChars)
{
m_data.Clear();
return;
}
char *pszString = PrepareBuffer(nChars);
if (pszString)
{
memcpy(pszString, pchSource, nChars);
// PrepareBuffer already allocated space for the terminating null,
// and inserted it for us. Make sure we didn't clobber it.
// Also assign it anyways so we don't risk the caller having a buffer
// running into random bytes.
#ifdef _DEBUG
// Suppress a bogus noisy warning:
// warning C6385: Invalid data: accessing 'pszString', the readable size is 'nChars' bytes, but '1001' bytes might be read
ANALYZE_SUPPRESS(6385);
Assert(pszString[nChars] == '\0');
pszString[nChars] = '\0';
#endif
}
}
//-----------------------------------------------------------------------------
// Purpose: Sets the string to be the new value, taking a copy of it
//-----------------------------------------------------------------------------
inline void CUtlStringBuilder::Set(const char *pchString)
{
SetValue(pchString);
}
//-----------------------------------------------------------------------------
// Purpose: Sets the string to be the new value, taking ownership of the pointer
//-----------------------------------------------------------------------------
inline void CUtlStringBuilder::SetPtr(char *pchString)
{
size_t nLength = pchString ? strlen(pchString) : 0;
SetPtr(pchString, nLength);
}
//-----------------------------------------------------------------------------
// Purpose: Sets the string to be the new value, taking ownership of the pointer
// This API will clear the error state if it was set.
//-----------------------------------------------------------------------------
inline void CUtlStringBuilder::SetPtr(char *pchString, size_t nLength)
{
m_data.Clear();
if (!pchString || !nLength)
{
if (pchString)
MemAlloc_Free(pchString); // we don't hang onto empty strings.
return;
}
m_data.SetPtr(pchString, nLength);
}
//-----------------------------------------------------------------------------
// Purpose: return the conceptual 'strlen' of the string.
//-----------------------------------------------------------------------------
inline size_t CUtlStringBuilder::Length() const
{
return m_data.Length();
}
//-----------------------------------------------------------------------------
// Purpose: format something sprintf() style, and take it as the new value of this CUtlStringBuilder
//-----------------------------------------------------------------------------
inline size_t CUtlStringBuilder::Format(const char *pFormat, ...)
{
va_list args;
va_start(args, pFormat);
size_t nLen = VFormat(pFormat, args);
va_end(args);
return nLen;
}
int V_vscprintf(const char *format, va_list argptr);
//-----------------------------------------------------------------------------
// Purpose: Helper for Format() method
//-----------------------------------------------------------------------------
inline size_t CUtlStringBuilder::VFormat(const char *pFormat, va_list args)
{
if (HasError())
return 0;
int len = 0;
#ifdef _WIN32
// how much space will we need?
len = V_vscprintf(pFormat, args);
#else
// ISO spec defines the NULL/0 case as being valid and will return the
// needed length. Verified on PS3 as well. Ignore that bsd/linux/mac
// have vasprintf which will allocate a buffer. We'd rather have the
// self growing buffer management ourselves. Even the best implementations
// There does not seem to be a magic vasprintf that is significantly
// faster than 2 passes (some guess and get lucky).
// Scope ReuseArgs.
{
CReuseVaList ReuseArgs(args);
len = V_vsnprintf(NULL, 0, pFormat, ReuseArgs.m_ReuseList);
}
#endif
if (len > 0)
{
// get it
char *pszString = PrepareBuffer(len, true);
if (pszString)
len = V_vsnprintf(pszString, len + 1, pFormat, args);
else
len = 0;
}
Assert(len > 0 || HasError());
return len;
}
//-----------------------------------------------------------------------------
// format a string and append the result to the string we hold
//-----------------------------------------------------------------------------
inline size_t CUtlStringBuilder::AppendFormat(const char *pFormat, ...)
{
va_list args;
va_start(args, pFormat);
size_t nLen = VAppendFormat(pFormat, args);
va_end(args);
return nLen;
}
//-----------------------------------------------------------------------------
// Purpose: implementation helper for AppendFormat()
//-----------------------------------------------------------------------------
inline size_t CUtlStringBuilder::VAppendFormat(const char *pFormat, va_list args)
{
if (HasError())
return 0;
int len = 0;
#ifdef _WIN32
// how much space will we need?
len = _vscprintf(pFormat, args);
#else
// ISO spec defines the NULL/0 case as being valid and will return the
// needed length. Verified on PS3 as well. Ignore that bsd/linux/mac
// have vasprintf which will allocate a buffer. We'd rather have the
// self growing buffer management ourselves. Even the best implementations
// There does not seem to be a magic vasprintf that is significantly
// faster than 2 passes (some guess and get lucky).
// Scope ReuseArgs.
{
CReuseVaList ReuseArgs(args);
len = vsnprintf(NULL, 0, pFormat, ReuseArgs.m_ReuseList);
}
#endif
size_t nOldLen = Length();
if (len > 0)
{
// get it
char *pszString = PrepareBuffer(nOldLen + len, true);
if (pszString)
len = _vsnprintf(&pszString[nOldLen], len + 1, pFormat, args);
else
len = 0;
}
Assert(len > 0 || HasError());
return nOldLen + len;
}
//-----------------------------------------------------------------------------
// Purpose: concatenate the provided string to our current content
//-----------------------------------------------------------------------------
inline void CUtlStringBuilder::Append(const char *pchAddition)
{
if (pchAddition && pchAddition[0])
{
size_t cchLen = strlen(pchAddition);
Append(pchAddition, cchLen);
}
}
//-----------------------------------------------------------------------------
// Purpose: concatenate the provided string to our current content
// when the additional string length is known
//-----------------------------------------------------------------------------
inline void CUtlStringBuilder::Append(const char *pchAddition, size_t cbLen)
{
if (pchAddition && pchAddition[0] && cbLen)
{
if (IsEmpty())
{
SetDirect(pchAddition, cbLen);
}
else
{
size_t cbOld = Length();
char *pstrNew = PrepareBuffer(cbOld + cbLen, true);
// make sure we use raw memcpy to get intrinsic
if (pstrNew)
memcpy(pstrNew + cbOld, pchAddition, cbLen);
}
}
}
//-----------------------------------------------------------------------------
// Purpose: append a repeated series of a single character
//-----------------------------------------------------------------------------
inline void CUtlStringBuilder::AppendRepeat(char ch, int cCount)
{
size_t cbOld = Length();
char *pstrNew = PrepareBuffer(cbOld + cCount, true);
if (pstrNew)
memset(pstrNew + cbOld, ch, cCount);
}
//-----------------------------------------------------------------------------
// Purpose: Swaps string contents
//-----------------------------------------------------------------------------
inline void CUtlStringBuilder::Swap(CUtlStringBuilder &src)
{
// swapping m_data instead of '*this' prevents having to
// copy dynamic strings. Important that m_data doesn't know
// any lifetime rules about its members (ie: it should not have
// a destructor that frees the dynamic string pointer).
Swap(m_data, src.m_data);
}
//-----------------------------------------------------------------------------
// Purpose: replace all occurrences of one character with another
//-----------------------------------------------------------------------------
inline size_t CUtlStringBuilder::Replace(char chTarget, char chReplacement)
{
size_t cReplacements = 0;
if (!IsEmpty() && !HasError())
{
char *pszString = Access();
for (char *pstrWalker = pszString; *pstrWalker != 0; pstrWalker++)
{
if (*pstrWalker == chTarget)
{
*pstrWalker = chReplacement;
cReplacements++;
}
}
}
return cReplacements;
}
//-----------------------------------------------------------------------------
// replace a sequence of characters at a given point with a new string
// replacement string might be NULL or "" to remove target substring
//-----------------------------------------------------------------------------
inline bool CUtlStringBuilder::ReplaceAt(size_t nIndex, size_t nOldChars, const char *pNewStr, size_t nNewChars)
{
Assert(nIndex < Length() && nIndex + nOldChars <= Length());
if (nNewChars == SIZE_MAX)
{
nNewChars = pNewStr ? V_strlen(pNewStr) : 0;
}
size_t nOldLength = Length();
ptrdiff_t nDelta = nNewChars - nOldChars;
if (nDelta < 0)
{
char *pBuf = Access();
memmove(pBuf + nIndex + nNewChars, pBuf + nIndex + nOldChars, nOldLength - nIndex - nOldChars);
SetLength(nOldLength + nDelta);
}
else if (nDelta > 0)
{
char *pBuf = PrepareBuffer(nOldLength + nDelta, true);
if (!pBuf)
{
return false;
}
memmove(pBuf + nIndex + nNewChars, pBuf + nIndex + nOldChars, nOldLength - nIndex - nOldChars);
}
if (nNewChars)
{
memcpy(Access() + nIndex, pNewStr, nNewChars);
}
return true;
}
//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------
// Purpose: Truncates the string to the specified number of characters
//-----------------------------------------------------------------------------
inline void CUtlStringBuilder::Truncate(size_t nChars)
{
if (IsEmpty() || HasError())
return;
size_t nLen = Length();
if (nLen <= nChars)
return;
// we may be shortening enough to fit in the small buffer, but
// the external buffer is already allocated, so just keep using it.
m_data.SetLength(nChars);
}
//-----------------------------------------------------------------------------
// Data and memory validation
//-----------------------------------------------------------------------------
#ifdef DBGFLAG_VALIDATE
inline void CUtlStringBuilder::Validate(CValidator &validator, const char *pchName)
{
#ifdef _WIN32
validator.Push(typeid(*this).raw_name(), this, pchName);
#else
validator.Push(typeid(*this).name(), this, pchName);
#endif
if (m_data.IsHeap())
validator.ClaimMemory(Access());
validator.Pop();
}
#endif // DBGFLAG_VALIDATE
#endif // UTLSTRING_H