sal/rtl/strimp.cxx

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2 /*
   3  * This file is part of the LibreOffice project.
   4  *
   5  * This Source Code Form is subject to the terms of the Mozilla Public
   6  * License, v. 2.0. If a copy of the MPL was not distributed with this
   7  * file, You can obtain one at http://mozilla.org/MPL/2.0/.
   8  *
   9  * This file incorporates work covered by the following license notice:
  10  *
  11  *   Licensed to the Apache Software Foundation (ASF) under one or more
  12  *   contributor license agreements. See the NOTICE file distributed
  13  *   with this work for additional information regarding copyright
  14  *   ownership. The ASF licenses this file to you under the Apache
  15  *   License, Version 2.0 (the "License"); you may not use this file
  16  *   except in compliance with the License. You may obtain a copy of
  17  *   the License at http://www.apache.org/licenses/LICENSE-2.0 .
  18  */
  19 #include <sal/config.h>
  20
  21 #include <assert.h>
  22 #include <stdlib.h>
  23 #include <rtl/alloc.h>
  24 #include <rtl/ustring.h>
  25 #include <rtllifecycle.h>
  26
  27 #include "strimp.hxx"
  28 #include "alloc_impl.hxx"
  29 #include "alloc_arena.hxx"
  30
  31 /*
  32  * TODO: add a slower, more awful, but more space efficient
  33  * custom allocator for the pre-init phase. Existing slab
  34  * allocator's minimum alloc size is 24bytes, and by default
  35  * is 32 bytes.
  36  */
  37 static rtl_arena_type *pre_arena = nullptr;
  38
  39 rtl_allocateStringFn rtl_allocateString = malloc;
  40 rtl_freeStringFn rtl_freeString = free;
  41
  42 extern "C" {
  43 static void *pre_allocateStringFn(size_t n)
  44 {
  45     sal_Size size = RTL_MEMORY_ALIGN(n + 4, 4);
  46     char *addr = static_cast<char*>(rtl_arena_alloc(pre_arena, &size));
  47     assert(size>= 12);
  48     reinterpret_cast<sal_uInt32*>(addr)[0] = size - 12;
  49     return addr + 4;
  50 }
  51
  52 static void pre_freeStringFn(void *data)
  53 {
  54     char    *addr = static_cast<char*>(data) - 4;
  55     sal_uInt32 size = reinterpret_cast<sal_uInt32*>(addr)[0] + 12;
  56
  57     rtl_arena_free(pre_arena, addr, size);
  58 }
  59 } // extern "C"
  60
  61 static void mark_static(void *addr, sal_Size /* size */)
  62 {
  63     char *inner = static_cast<char*>(addr) + 4;
  64     rtl_uString *str = reinterpret_cast<rtl_uString *>(inner);
  65     str->refCount |= SAL_STRING_STATIC_FLAG;
  66 }
  67
  68 void SAL_CALL rtl_alloc_preInit (sal_Bool start) SAL_THROW_EXTERN_C()
  69 {
  70     if (start)
  71     {
  72         rtl_allocateString = pre_allocateStringFn;
  73         rtl_freeString = pre_freeStringFn;
  74         pre_arena = rtl_arena_create("pre-init strings", 4, 0,
  75                                      nullptr, rtl_arena_alloc,
  76                                      rtl_arena_free, 0);
  77
  78         // To be consistent (and to ensure the rtl_cache threads are started).
  79         ensureCacheSingleton();
  80     }
  81     else
  82     {
  83         rtl_arena_foreach(pre_arena, mark_static);
  84         rtl_allocateString = malloc;
  85         rtl_freeString = free;
  86
  87         // TODO: also re-initialize main allocator as well.
  88     }
  89 }
  90
  91 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */