strbuf.hxx

Go to the documentation of this file.

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/*
 * This file is part of the LibreOffice project.
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 *
 * This file incorporates work covered by the following license notice:
 *
 *   Licensed to the Apache Software Foundation (ASF) under one or more
 *   contributor license agreements. See the NOTICE file distributed
 *   with this work for additional information regarding copyright
 *   ownership. The ASF licenses this file to you under the Apache
 *   License, Version 2.0 (the "License"); you may not use this file
 *   except in compliance with the License. You may obtain a copy of
 *   the License at http://www.apache.org/licenses/LICENSE-2.0 .
 */

/*
 * This file is part of LibreOffice published API.
 */

#pragma once

#include "sal/config.h"

#include <cassert>
#include <cstring>
#include <limits>

#include "rtl/strbuf.h"
#include "rtl/string.hxx"
#include "rtl/stringutils.hxx"

#ifdef LIBO_INTERNAL_ONLY // "RTL_FAST_STRING"
#include "rtl/stringconcat.hxx"
#include <string_view>
#include <type_traits>
#endif

#ifdef RTL_STRING_UNITTEST
extern bool rtl_string_unittest_const_literal;
extern bool rtl_string_unittest_const_literal_function;
#endif

// The unittest uses slightly different code to help check that the proper
// calls are made. The class is put into a different namespace to make
// sure the compiler generates a different (if generating also non-inline)
// copy of the function and does not merge them together. The class
// is "brought" into the proper rtl namespace by a typedef below.
#ifdef RTL_STRING_UNITTEST
#define rtl rtlunittest
#endif

namespace rtl
{

#ifdef RTL_STRING_UNITTEST
#undef rtl
// helper macro to make functions appear more readable
#define RTL_STRING_CONST_FUNCTION rtl_string_unittest_const_literal_function = true;
#else
#define RTL_STRING_CONST_FUNCTION
#endif

class SAL_WARN_UNUSED OStringBuffer
{
public:
    OStringBuffer()
        : pData(NULL)
        , nCapacity( 16 )
    {
        rtl_string_new_WithLength( &pData, nCapacity );
    }

    OStringBuffer( const OStringBuffer & value )
        : pData(NULL)
        , nCapacity( value.nCapacity )
    {
        rtl_stringbuffer_newFromStringBuffer( &pData, value.nCapacity, value.pData );
    }

    explicit OStringBuffer(sal_Int32 length)
        : pData(NULL)
        , nCapacity( length )
    {
        rtl_string_new_WithLength( &pData, length );
    }
#if defined LIBO_INTERNAL_ONLY
    template<typename T>
    explicit OStringBuffer(T length, std::enable_if_t<std::is_integral_v<T>, int> = 0)
        : OStringBuffer(static_cast<sal_Int32>(length))
    {
        assert(
            length >= 0
            && static_cast<std::make_unsigned_t<T>>(length)
                <= static_cast<std::make_unsigned_t<sal_Int32>>(
                    std::numeric_limits<sal_Int32>::max()));
    }
    // avoid (obvious) bugs
    explicit OStringBuffer(bool) = delete;
    explicit OStringBuffer(char) = delete;
    explicit OStringBuffer(wchar_t) = delete;
#if !(defined _MSC_VER && _MSC_VER >= 1930 && _MSC_VER <= 1939 && defined _MANAGED)
    explicit OStringBuffer(char8_t) = delete;
#endif
    explicit OStringBuffer(char16_t) = delete;
    explicit OStringBuffer(char32_t) = delete;
#endif

#if defined LIBO_INTERNAL_ONLY
    OStringBuffer(std::string_view sv)
        : pData(nullptr)
        , nCapacity( sv.length() + 16 )
    {
        if (sv.size() > sal_uInt32(std::numeric_limits<sal_Int32>::max())) {
            throw std::bad_alloc();
        }
        rtl_stringbuffer_newFromStr_WithLength( &pData, sv.data(), sv.length() );
    }
#else
    OStringBuffer(const OString& value)
        : pData(NULL)
        , nCapacity( value.getLength() + 16 )
    {
        rtl_stringbuffer_newFromStr_WithLength( &pData, value.getStr(), value.getLength() );
    }
#endif

    template< typename T >
    OStringBuffer( const T& value, typename libreoffice_internal::CharPtrDetector< T, libreoffice_internal::Dummy >::Type = libreoffice_internal::Dummy())
        : pData(NULL)
    {
        sal_Int32 length = rtl_str_getLength( value );
        nCapacity = length + 16;
        rtl_stringbuffer_newFromStr_WithLength( &pData, value, length );
    }

    template< typename T >
    OStringBuffer( T& value, typename libreoffice_internal::NonConstCharArrayDetector< T, libreoffice_internal::Dummy >::Type = libreoffice_internal::Dummy())
        : pData(NULL)
    {
        sal_Int32 length = rtl_str_getLength( value );
        nCapacity = length + 16;
        rtl_stringbuffer_newFromStr_WithLength( &pData, value, length );
    }

#if __cplusplus > 202002L // C++23 P2266R3 "Simpler implicit move"
    template< typename T >
    OStringBuffer( T&& value, typename libreoffice_internal::NonConstCharArrayDetector< T, libreoffice_internal::Dummy >::Type = libreoffice_internal::Dummy())
        : pData(NULL)
    {
        sal_Int32 length = rtl_str_getLength( value );
        nCapacity = length + 16;
        rtl_stringbuffer_newFromStr_WithLength( &pData, value, length );
    }
#endif

    template< typename T >
    OStringBuffer( T& literal, typename libreoffice_internal::ConstCharArrayDetector< T, libreoffice_internal::Dummy >::Type = libreoffice_internal::Dummy())
        : pData(NULL)
        , nCapacity( libreoffice_internal::ConstCharArrayDetector<T>::length + 16 )
    {
        assert(
            libreoffice_internal::ConstCharArrayDetector<T>::isValid(literal));
        rtl_string_newFromLiteral(
            &pData,
            libreoffice_internal::ConstCharArrayDetector<T>::toPointer(literal),
            libreoffice_internal::ConstCharArrayDetector<T>::length, 16);
#ifdef RTL_STRING_UNITTEST
        rtl_string_unittest_const_literal = true;
#endif
    }

    OStringBuffer(const char * value, sal_Int32 length)
        : pData(NULL)
        , nCapacity( length + 16 )
    {
        rtl_stringbuffer_newFromStr_WithLength( &pData, value, length );
    }

#ifdef LIBO_INTERNAL_ONLY // "RTL_FAST_STRING"

    template< typename T1, typename T2 >
    OStringBuffer( OStringConcat< T1, T2 >&& c )
    {
        const sal_Int32 l = c.length();
        nCapacity = l + 16;
        pData = rtl_string_alloc( nCapacity );
        char* end = c.addData( pData->buffer );
        *end = '\0';
        pData->length = l;
    }

    template< std::size_t N >
    OStringBuffer( OStringNumber< N >&& n )
        : OStringBuffer( n.buf, n.length)
    {}
#endif

#if defined LIBO_INTERNAL_ONLY
    operator std::string_view() const { return {getStr(), sal_uInt32(getLength())}; }
#endif

    OStringBuffer& operator = ( const OStringBuffer& value )
    {
        if (this != &value)
        {
            rtl_stringbuffer_newFromStringBuffer(&pData,
                                                  value.nCapacity,
                                                  value.pData);
            nCapacity = value.nCapacity;
        }
        return *this;
    }

#if defined LIBO_INTERNAL_ONLY
    OStringBuffer & operator =(std::string_view string) {
        sal_Int32 n = string.length();
        if (n >= nCapacity) {
            ensureCapacity(n + 16); //TODO: check for overflow
        }
        std::memcpy(pData->buffer, string.data(), n);
        pData->buffer[n] = '\0';
        pData->length = n;
        return *this;
    }
#else
    OStringBuffer & operator =(OString const & string) {
        sal_Int32 n = string.getLength();
        if (n >= nCapacity) {
            ensureCapacity(n + 16); //TODO: check for overflow
        }
        std::memcpy(pData->buffer, string.pData->buffer, n + 1);
        pData->length = n;
        return *this;
    }
#endif

    template<typename T>
    typename
        libreoffice_internal::ConstCharArrayDetector<T, OStringBuffer &>::Type
    operator =(T & literal) {
        assert(
            libreoffice_internal::ConstCharArrayDetector<T>::isValid(literal));
        sal_Int32 const n
            = libreoffice_internal::ConstCharArrayDetector<T>::length;
        if (n >= nCapacity) {
            ensureCapacity(n + 16); //TODO: check for overflow
        }
        std::memcpy(
            pData->buffer,
            libreoffice_internal::ConstCharArrayDetector<T>::toPointer(literal),
            n + 1);
        pData->length = n;
        return *this;
    }

#if defined LIBO_INTERNAL_ONLY

    template<typename T1, typename T2>
    OStringBuffer & operator =(OStringConcat<T1, T2> && concat) {
        sal_Int32 const n = concat.length();
        if (n >= nCapacity) {
            ensureCapacity(n + 16); //TODO: check for overflow
        }
        *concat.addData(pData->buffer) = 0;
        pData->length = n;
        return *this;
    }

    template<std::size_t N>
    OStringBuffer & operator =(OStringNumber<N> && n)
    {
        return operator =(std::string_view(n));
    }
#endif

    ~OStringBuffer()
    {
        rtl_string_release( pData );
    }

    SAL_WARN_UNUSED_RESULT OString makeStringAndClear()
    {
        OString aRet( pData );
        rtl_string_new(&pData);
        nCapacity = 0;
        return aRet;
    }

    sal_Int32 getLength() const
    {
        return pData->length;
    }

    bool isEmpty() const
    {
        return pData->length == 0;
    }

    sal_Int32 getCapacity() const
    {
        return nCapacity;
    }

    void ensureCapacity(sal_Int32 minimumCapacity)
    {
        rtl_stringbuffer_ensureCapacity( &pData, &nCapacity, minimumCapacity );
    }

    void setLength(sal_Int32 newLength)
    {
        assert(newLength >= 0);
        // Avoid modifications if pData points to const empty string:
        if( newLength != pData->length )
        {
            if( newLength > nCapacity )
                rtl_stringbuffer_ensureCapacity(&pData, &nCapacity, newLength);
            else
                pData->buffer[newLength] = '\0';
            pData->length = newLength;
        }
    }

    SAL_DEPRECATED("use rtl::OStringBuffer::operator [] instead")
    char charAt( sal_Int32 index )
    {
        assert(index >= 0 && index < pData->length);
        return pData->buffer[ index ];
    }

    SAL_DEPRECATED("use rtl::OStringBuffer::operator [] instead")
    OStringBuffer & setCharAt(sal_Int32 index, char ch)
    {
        assert(index >= 0 && index < pData->length);
        pData->buffer[ index ] = ch;
        return *this;
    }

    const char* getStr() const SAL_RETURNS_NONNULL { return pData->buffer; }

    char & operator [](sal_Int32 index)
    {
        assert(index >= 0 && index < pData->length);
        return pData->buffer[index];
    }

    OString toString() const
    {
        return OString(pData->buffer, pData->length);
    }

#if !defined LIBO_INTERNAL_ONLY

    OStringBuffer & append(const OString &str)
    {
        return insert(getLength(), str);
    }
#endif

    template< typename T >
    typename libreoffice_internal::CharPtrDetector< T, OStringBuffer& >::Type append( const T& str )
    {
        return insert(getLength(), str);
    }

    template< typename T >
    typename libreoffice_internal::NonConstCharArrayDetector< T, OStringBuffer& >::Type append( T& str )
    {
        return insert(getLength(), str);
    }

    template< typename T >
    typename libreoffice_internal::ConstCharArrayDetector< T, OStringBuffer& >::Type append( T& literal )
    {
        return insert(getLength(), literal);
    }

    OStringBuffer & append( const char * str, sal_Int32 len)
    {
        return insert(getLength(), str, len);
    }

#ifdef LIBO_INTERNAL_ONLY // "RTL_FAST_STRING"

    template< typename T1, typename T2 >
    OStringBuffer& append( OStringConcat< T1, T2 >&& c )
    {
        sal_Int32 l = c.length();
        if (l != 0)
            c.addData(appendUninitialized(l));
        return *this;
    }

    OStringBuffer& append( std::string_view s )
    {
        return insert(getLength(), s);
    }

#endif

    OStringBuffer & append(sal_Bool b)
    {
        return insert(getLength(), b);
    }

    OStringBuffer & append(bool b)
    {
        return insert(getLength(), b);
    }

    // Pointer can be automatically converted to bool, which is unwanted here.
    // Explicitly delete all pointer append() overloads to prevent this
    // (except for char* overload, which is handled elsewhere).
    template< typename T >
    typename libreoffice_internal::Enable< void,
        !libreoffice_internal::CharPtrDetector< T* >::ok >::Type
        append( T* ) SAL_DELETED_FUNCTION;

    OStringBuffer & append(char c)
    {
        return insert(getLength(), c);
    }

    OStringBuffer & append(sal_Int32 i, sal_Int16 radix = 10 )
    {
        return insert(getLength(), i, radix);
    }

    OStringBuffer & append(sal_Int64 l, sal_Int16 radix = 10 )
    {
        return insert(getLength(), l, radix);
    }

    OStringBuffer & append(float f)
    {
        return insert(getLength(), f);
    }

    OStringBuffer & append(double d)
    {
        return insert(getLength(), d);
    }

    char * appendUninitialized(sal_Int32 length) SAL_RETURNS_NONNULL {
        sal_Int32 n = getLength();
        rtl_stringbuffer_insert(&pData, &nCapacity, n, NULL, length);
        return pData->buffer + n;
    }

#if defined LIBO_INTERNAL_ONLY
    OStringBuffer & insert(sal_Int32 offset, std::string_view str)
    {
        return insert( offset, str.data(), str.length() );
    }
#else
    OStringBuffer & insert(sal_Int32 offset, const OString & str)
    {
        return insert( offset, str.getStr(), str.getLength() );
    }
#endif

    template< typename T >
    typename libreoffice_internal::CharPtrDetector< T, OStringBuffer& >::Type insert( sal_Int32 offset, const T& str )
    {
        return insert( offset, str, rtl_str_getLength( str ) );
    }

    template< typename T >
    typename libreoffice_internal::NonConstCharArrayDetector< T, OStringBuffer& >::Type insert( sal_Int32 offset, T& str )
    {
        return insert( offset, str, rtl_str_getLength( str ) );
    }

    template< typename T >
    typename libreoffice_internal::ConstCharArrayDetector< T, OStringBuffer& >::Type insert( sal_Int32 offset, T& literal )
    {
        RTL_STRING_CONST_FUNCTION
        assert(
            libreoffice_internal::ConstCharArrayDetector<T>::isValid(literal));
        return insert(
            offset,
            libreoffice_internal::ConstCharArrayDetector<T>::toPointer(literal),
            libreoffice_internal::ConstCharArrayDetector<T>::length);
    }

    OStringBuffer & insert( sal_Int32 offset, const char * str, sal_Int32 len)
    {
        assert( len == 0 || str != NULL ); // cannot assert that in rtl_stringbuffer_insert
        rtl_stringbuffer_insert( &pData, &nCapacity, offset, str, len );
        return *this;
    }

    OStringBuffer & insert(sal_Int32 offset, sal_Bool b)
    {
        char sz[RTL_STR_MAX_VALUEOFBOOLEAN];
        return insert( offset, sz, rtl_str_valueOfBoolean( sz, b ) );
    }

    OStringBuffer & insert(sal_Int32 offset, bool b)
    {
        char sz[RTL_STR_MAX_VALUEOFBOOLEAN];
        return insert( offset, sz, rtl_str_valueOfBoolean( sz, b ) );
    }

    OStringBuffer & insert(sal_Int32 offset, char c)
    {
        return insert( offset, &c, 1 );
    }

    OStringBuffer & insert(sal_Int32 offset, sal_Int32 i, sal_Int16 radix = 10 )
    {
        char sz[RTL_STR_MAX_VALUEOFINT32];
        return insert( offset, sz, rtl_str_valueOfInt32( sz, i, radix ) );
    }

    OStringBuffer & insert(sal_Int32 offset, sal_Int64 l, sal_Int16 radix = 10 )
    {
        char sz[RTL_STR_MAX_VALUEOFINT64];
        return insert( offset, sz, rtl_str_valueOfInt64( sz, l, radix ) );
    }

    OStringBuffer & insert(sal_Int32 offset, float f)
    {
        // Same as rtl::str::valueOfFP, used for rtl_str_valueOfFloat
        rtl_math_doubleToString(&pData, &nCapacity, offset, f, rtl_math_StringFormat_G,
                                RTL_STR_MAX_VALUEOFFLOAT - SAL_N_ELEMENTS("-x.E-xxx") + 1, '.',
                                NULL, 0, true);
        return *this;
    }

    OStringBuffer & insert(sal_Int32 offset, double d)
    {
        // Same as rtl::str::valueOfFP, used for rtl_str_valueOfDouble
        rtl_math_doubleToString(&pData, &nCapacity, offset, d, rtl_math_StringFormat_G,
                                RTL_STR_MAX_VALUEOFDOUBLE - SAL_N_ELEMENTS("-x.E-xxx") + 1, '.',
                                NULL, 0, true);
        return *this;
    }

    OStringBuffer & remove( sal_Int32 start, sal_Int32 len )
    {
        rtl_stringbuffer_remove( &pData, start, len );
        return *this;
    }

    void accessInternals(
        rtl_String *** pInternalData, sal_Int32 ** pInternalCapacity)
    {
        *pInternalData = &pData;
        *pInternalCapacity = &nCapacity;
    }

private:
    rtl_String * pData;

    sal_Int32       nCapacity;
};

#if defined LIBO_INTERNAL_ONLY
template<> struct ToStringHelper<OStringBuffer> {
    static std::size_t length(OStringBuffer const & s) { return s.getLength(); }

    char * operator()(char * buffer, OStringBuffer const & s) const SAL_RETURNS_NONNULL
    { return addDataHelper(buffer, s.getStr(), s.getLength()); }
};
#endif

}

#ifdef RTL_STRING_UNITTEST
namespace rtl
{
typedef rtlunittest::OStringBuffer OStringBuffer;
}
#undef RTL_STRING_CONST_FUNCTION
#endif

#if defined LIBO_INTERNAL_ONLY && !defined RTL_STRING_UNITTEST
using ::rtl::OStringBuffer;
#endif

/* vim:set shiftwidth=4 softtabstop=4 expandtab: */