Commit Graph

1002 Commits

Author SHA1 Message Date
Will Brown
e6bf8570d0 [Rust] add support for optionally compiling libc functions (#763)
* add SNMALLOC_RUST_LIBC_API option
2025-04-03 13:56:41 +01:00
Jeff Sooknarine
59987ca908 Header-only build fix for WIN32_LEAN_AND_MEAN redefinition (#764)
When using snmalloc as header-only library in a project that globally defines `WIN32_LEAN_AND_MEAN`, there is a compile error for macro redefinition.

Guard the pal_windows.h define.
2025-04-01 21:16:55 +00:00
Matthew Parkinson
dff1057db2 Refactor representation of thread local state. (#751)
* Lift checking for init to ThreadAlloc

The check init code was tightly integrated into LocalAllocator.  This commit pull that code out into ThreadAlloc, and passes a template parameter into the remaining LocalAllocator to perform the relevant TLS manipulations.  This removes some of the awkward layering around register_clean_up.

* Reduce size of test due to failures.

Fully disable lotsofthreads test

Need to investigate if the test is unreliable, or we have actually
regressed perf.  A quick mimalloc-bench didn't show any regressions.

* Simplify message queue initialisation

This introduces one additional branch on when processing a batch of messages, but it is likely to only be hit when a lot of messages are processed.

* Patch Domestication test.

* Refactor CoreAlloc/LocalAlloc

This combines the notion of CoreAlloc, LocalAlloc and LocalCache into a single class.  Previously, these were separated so that a more complex structure would be stored directly in the TLS.  This however, proved to be bad for compatibility if the allocator is part of the libc implementation.

This commit collapses all the stages of the allocator into a single class. This simplifies the sequencing and overall is a nice reduction in complexity.

* Re-enable lots of threads test.

* Reenable concept using alternative lazy checking for concepts.

* Self code review
2025-03-21 15:13:32 +00:00
Matthew Parkinson
3348bf9e56 Fix test when using GWP-Asan (#758) 2025-03-17 15:37:54 +00:00
Matthew Parkinson
a106a2e69d Refactor check_bounds (#756)
This changes the shape of check_bounds to take a continuation to call if the bounds check succeeds.  This is designed to allow for easily wrapping existing code with a bounds check, e.g.

```
void* memcpy(void* dest, const void* src, size_t n) {
  return check_bounds(dest, n, [&] {
    return memcpy_impl(dest, src, n);
  });
}
```
2025-03-14 16:09:09 +00:00
Trithek
dfda7a8442 Fixed pagemap reservations for a fixed-region use-case using the default PAL (#757)
* Made reserve_range do the correct thing when a fixed range configuration was used

* Corrected the check for out-of-bounds
2025-03-13 10:20:43 +00:00
Matthew Parkinson
ccc03ce0fc Protection against fork (#735)
If a thread forks, while another thread is holding an snmalloc lock, then the allocator could stop working.

This patch attempts to protect against the cases of this. There is one case that is not covered. If a fork occurs during the very first allocation. This can result in the installation of the fork handler racing with the fork, and all bets are off.
2025-03-04 13:44:59 +00:00
Matthew Parkinson
c2e22ccb79 Prevent internal errno setting escaping to the client. (#754) 2025-02-28 11:12:22 +00:00
Matthew Parkinson
06df9dd9e1 Update realloc(p,0) semantics. (#753)
* Factor out small sizeclass check

* Update realloc(p,0) semantics

This commit changes the behaviour of realloc(p,0) to be free(p) if p!=nullptr, and malloc(0) if p== nullptr.
2025-02-27 15:20:31 +00:00
Matthew Parkinson
6622dc584e Bug fixes (#752)
* Fix overflow by alignment

* Bug fix: Ensure bytes_free is the total

The bytes freed was not added to the total, but
overrode it.  This meant it never fired. This
commit fixes that.
2025-02-27 08:52:50 +00:00
Matthew Parkinson
5f7baef755 Refactor: Remove unused features and functions, and move most allocator operations to a global namespace. (#750)
* Factor out explicit Config type

Instead of using snmalloc::Alloc::Config, expose snmalloc::Config, which is then used to derive the allocator type.

* Move globalalloc to front end.

* Remove unneed template parameter from global snmalloc functions.

* Remove SNMALLOC_PASS_THROUGH

VeronaRT now has an abstraction layer which can easily replace the allocator.
Having such a complex integration still in snmalloc does not make sense.

* Take some global functions off of local alloc.

* Drop comparison overloads on atomic Capptr.

Performing a comparison on two atomic ptr is a complex operation, and should not be implicit.  The memory model order and such things needs to be considered by the caller.

* Remove function_ref and use templates

The implementation prefers to use templates over the function_ref.  This now only exists in the Pal for a currently unused feature.

* Removing function_ref reduces stl needs.

* Remove use of __is_convertible to support older g++

* Inline function that is only used once.

* Remove unused function

* Restrict ThreadAlloc usage to globalalloc

This commit introduces various inline functions on snmalloc:: that perform allocation/deallocation using the thread local allocator.

They remove all usage from a particular test.

* Move cheri checks to own file.

* Refactor is_owned checks.

* Move alloc_size and check_size to globalalloc.

* Minor simplification of dealloc path

* Fix up is_owned to take a config

* Improve usage of scoped allocator.

* Handle Config_ in globalalloc.
2025-02-22 19:53:27 +00:00
Matthew Parkinson
6d50141e35 build: Fix nothrow include. (#749) 2025-02-14 12:50:18 +00:00
Neil Monday
ef4748289d Windows: Remove exception handler for pagemap on unloading (#746)
Co-authored-by: Neil Monday <neil.monday@amd.com>
2025-02-12 15:38:54 +00:00
Schrodinger ZHU Yifan
16b96245f6 Enable a seconary allocator support (e.g. GWP-Asan) (#737) 2025-02-05 09:03:11 +00:00
Matthew Parkinson
32495fd42d Multiple Pals could return spuriously from wake on address (#739)
* Add stress test benchmark

Co-authored-by: Alexander Nadeau <wareya@gmail.com>

* Add defensive code against spurious wakeup

This commit checks that wait_on_address has not returned spuriously.

* pal: spurious wake up.

The code in the Pal for wake on address was incorrectly assuming the operation returning success meant it had actually changed.  The specification allows for spurious wake ups.

This change makes the Pals recheck for a change.

---------

Co-authored-by: Alexander Nadeau <wareya@gmail.com>
2025-02-04 18:44:02 +00:00
Jean-Michaël Celerier
aa07749f46 pal_windows.h: missing <cerrno> for ENOMEM on MinGW (#733)
Co-authored-by: Matthew Parkinson <mjp41@users.noreply.github.com>
2025-01-21 09:25:11 +00:00
Schrodinger ZHU Yifan
229c6d04fe remove extra puts definitions (#731)
LGTM, thanks.
2025-01-16 10:31:31 +00:00
Matthew Parkinson
0fbc325a0a Fix merge issue for out of data CI run. (#730) 2025-01-10 08:37:33 +00:00
Matthew Parkinson
9e632047de Add Windows Support for Lazy Commit of page map (#727)
The Pagemap can potentially be accessed where there is no direct
knowledge of the location.  On platforms with lazy commit this can
be handled by the OS faulting in a zero page.  Here we modify the
Windows PAL to add an exception handler that will commit pages
in the read only range, i.e. the page map.  We still require explicit
notify_using to be able to modify the pagemap.
2025-01-09 17:57:52 +00:00
Schrodinger ZHU Yifan
0111a410a2 [proxy](7/n) self vendor array and algorithm headers (#722)
* [proxy](7/n) proxy array and algorithm headers

* [wip] address CRs

* undo comment

* format

* fix build

* simplify macro dispatching
2025-01-09 13:49:22 +00:00
Schrodinger ZHU Yifan
0027f02396 [proxy](6/n) self-vendor utility headers (#721) 2025-01-09 10:25:35 +00:00
Schrodinger ZHU Yifan
e874617d8e [proxy](5/n) use c headers instead of c++ headers (#720) 2025-01-09 09:45:24 +00:00
Neil Monday
6a3c5750bd Changing DEBUG constexpr to Debug to avoid collision with any #define DEBUG that may exist in other projects. (#729)
Co-authored-by: Neil Monday <neil.monday@amd.com>
2025-01-08 17:13:26 +00:00
Schrodinger ZHU Yifan
d1c65c2852 [proxy](4/n) use customised placement-new (#719)
* [proxy](4/n) use customized placement-new

* fix
2025-01-08 06:38:23 +00:00
Matthew Parkinson
046c5ac766 Limit remote batch size (#724)
When processing a remote batch, the system will process every single message that was available at the start of processing.
This can lead to a long pause time if there have been a considerable number of frees to this thread.

This commit introduces a new mechanism to only process messages up to a limit of 1MiB. The limit is configurable using CMake.

Choosing too small a limit can cause freeing to never catch up with the incoming messages.
2025-01-07 14:29:34 +00:00
Matthew Parkinson
35c5c6038c Remove dependencies from Singleton and FlagLock (#725)
* Move get_tid() to ds_core.

* Rewrite Singleton to only use Aal.

* Move data structure to be usable from the Pal.
2025-01-07 14:28:57 +00:00
Schrodinger ZHU Yifan
8c6474d05a [proxy](3/n) remove chrono dependency for windows (#718)
* [proxy](3/n) remove chrono dependency for windows

* minor fixes
2025-01-07 13:40:57 +00:00
Schrodinger ZHU Yifan
2cc74eac10 [proxy](2/n) provide proxy layers for type traits (#717)
* [proxy](2/n) provide proxy layers for type traits

* fix after rebasing
2025-01-06 17:07:49 +00:00
Schrodinger ZHU Yifan
feff2e8151 [proxy](1/n) start self-vendored STL with atomic (#715)
* [proxy] start self-vendored STL

* address CR
2025-01-03 07:49:32 +00:00
David CARLIER
fb29a5e085 PALSolaris AlignedAllocation feature implementation attempt. (#723) 2025-01-02 14:11:58 +00:00
David CARLIER
515b97d996 PALOpenBSD implement WaitOnAddress feature attempt. (#716) 2025-01-02 09:07:46 +00:00
Schrodinger ZHU Yifan
141cdd9361 [ds_core] simplify MessageBuilder construction (#713)
* [ds_core] simplify MessageBuilder construction

* remove special handling of zero case

* select latest xcode
2024-12-18 15:18:04 +00:00
David CARLIER
6952683692 memcpy test fix build warning on openbsd. (#714)
* memcpy test fix build warning on openbsd.

* enabling TEST_LIMITED for openbsd too.
2024-12-18 12:42:26 +00:00
Neil Monday
57c88efccb The _msize function has a non-const void* parameter in the msvc CRT. (#708)
Co-authored-by: Neil Monday <neil.monday@amd.com>
2024-12-17 11:37:44 +00:00
Schrodinger ZHU Yifan
cbc3018e1a remove string_view STL dependency (#703)
* remove string_view STL dependency

* fix

* inline string literal handling directly

* simplify dispatch
2024-12-15 07:46:49 +00:00
Schrodinger ZHU Yifan
1250ee002f [ds_core] remove TrivialInitAtomic (#711) 2024-12-14 07:10:06 +00:00
Schrodinger ZHU Yifan
cd91793def avoid chrono inclusion when clock_gettime is available (#695)
* avoid chrono inclusion when clock_gettime is available

* fix build
2024-11-26 21:04:45 +00:00
Matthew Parkinson
69e280c331 Minor update to clangformat file. (#694) 2024-11-22 12:40:16 +00:00
David CARLIER
fd560d472b Revert "custom memmove implementation proposal. (#593)" (#692)
* Revert "custom memmove implementation proposal. (#593)"

This reverts commit 01885f5a04.

* disable memmove fuzzing
2024-11-20 10:36:42 +00:00
David CARLIER
01885f5a04 custom memmove implementation proposal. (#593)
mostly like memcpy with optional bound checking but
capable of handling overlapping cases thus using
reverse copy instead.
2024-11-16 07:39:38 +00:00
Schrodinger ZHU Yifan
f7fe702f77 implement polite waiting (#685)
* implement polite waiting

* Update src/snmalloc/pal/pal_linux.h

Co-authored-by: Matthew Parkinson <mjp41@users.noreply.github.com>

* fix build issues

* fix more build issues

* support _umtx_op for freebsd

* unify waiting style

* fix

* address CR

Co-authored-by: Matthew Parkinson <mjp41@users.noreply.github.com>

* support macos

* static dispatch os APIs for apple

* make wait_on_address configurable via cmake

* undo extra include

* fix macos build

* fix clang-tidy build

---------

Co-authored-by: Matthew Parkinson <mjp41@users.noreply.github.com>
2024-11-16 07:33:11 +00:00
Matthew Parkinson
fe3fed4414 Remove unneeded template. (#687) 2024-11-12 19:56:14 +00:00
David CARLIER
43ad730c29 PAL::haiku finally supports getentropy. (#684)
Also fixing the build with PAL::posix assuming the intent is
to avoid using std namespace for max but rather using the in-house
implementation.
2024-10-31 11:54:27 +00:00
Matthew Parkinson
97b7675670 Remove some unneeded headers (#680)
* Removed unneeded headers

This removes some unneeded headers from the headers.

* Remove use of std::string

This stack allocates and copies a c-string to replace the calls to std::string.
2024-10-06 09:14:56 +01:00
Matthew Parkinson
c77076983d Add documentation for the combining lock (#683)
This adds some documentation to make the combining lock easier to understand.
This is working towards documenting the changes for the 0.7 release.
2024-10-05 07:31:00 +01:00
David CARLIER
f1df3d43bb override malloc, aligned_alloc calling libc::aligned_alloc (#681) 2024-09-28 22:02:40 +01:00
Matthew Parkinson
ab4fe84804 Provide option to CMake for Page Size (#664)
* Pickup page size from unistd.h

This uses the PAGESIZE constant from the unistd.h on POSIX.
This should make the code more resilient to being compiled on platforms with
different page sizes.

* Allow pagesize to come from cmake.

* Update src/snmalloc/pal/pal_posix.h

Co-authored-by: Nathaniel Filardo <105816689+nwf-msr@users.noreply.github.com>

---------

Co-authored-by: Nathaniel Filardo <105816689+nwf-msr@users.noreply.github.com>
2024-09-25 11:27:31 +01:00
Nathaniel Wesley Filardo
fb776da909 WIP: BatchIt (#677)
* Rename dealloc_local_object_slower to _meta

Unlike its brethren, `dealloc_local_object` and
`dealloc_local_object_slow`, the `dealloc_local_object_slower` method
does not take a pointer to free space.  Make this slightly more apparent
by renaming it and adding some commentary to both definition and call
site.

* corealloc: get meta in dealloc_local_object

Make both _fast() and _slow() arms take the meta as an argument; _meta()
already did.

* Introduce RemoteMessage structure

Plumb its use around remoteallocator and remotecache

* NFC: Plumb metadata to remotecache dealloc

* Initial steps in batched remote messages

This prepares the recipient to process a batched message.

* Initial dealloc-side batching machinery

Exercise recipient machinery by having the senders collect adjacent frees to
the same slab into a batch.

* Match free batch keying to slab freelist keying

* freelist: add append_segment

* SlabMetadata: machinery for returning multiple objects

This might involve multiple (I think at most two, at the moment) transitions in
the slab lifecycle state machine.  Towards that end, return indicators to the
caller that the slow path must be taken and how many objects of the original
set have not yet been counted as returned.

* corealloc: operate ring-at-a-time on remote queues

* RemoteCache associative cache of rings

* RemoteCache: N-set caching

* Initial CHERI support for free rings

* Matt's fix for slow-path codegen

* Try: remotecache: don't store allocator IDs

We can, as Matt so kindly reminds me, go get them from the pagemap.  Since we
need this value only when closing a ring, the read from over there is probably
not very onerous.  (We could also get the slab pointer from an object in the
ring, but we need that whenever inserting into the cache, so it's probably more
sensible to store that locally?)

* Make BatchIt optional

Move ring set bits and associativity knobs to allocconfig and expose them via
CMake.  If associtivity is zero, use non-batched implementations of the
`RemoteMessage` and `RemoteDeallocCacheBatching` classes.

By default, kick BatchIt on when we have enough room in the minimum allocation
size to do it.  Exactly how much space is enough is a function of which
mitigations we have enabled and whether or not we are compiling with C++20.

This commit reverts the change to `MIN_ALLOC_SIZE` made in "Introduce
RemoteMessage structure" now that we have multiple types, and zies, of
remote messages to choose from.

* RemoteDeallocCacheBatching: store metas as address

There's no need for a full pointer here, it'd just make the structure larger on
CHERI.

* NFC: plumb entropy from LocalAlloc to BatchIt

* BatchIt random eviction

In order not to thwart `mitigations(random_preserve)` too much, if it's on in
combination with BatchIt, roll the dice every time we append to a batch to
decide if we should stochastically evict this batch.  By increasing the number
of batches, we allow the recipient allocator increased opportunity to randomly
stripe batches across the two `freelist::Builder` segments associated with each
slab.

---------

Co-authored-by: Nathaniel Wesley Filardo <nfilardo@microsoft.com>
Co-authored-by: Matthew Parkinson <mattpark@microsoft.com>
2024-09-23 19:18:09 +01:00
Nathaniel Wesley Filardo
416fd39f6a gcc UAF warning in test/perf/singlethread -malloc
When building test/perf/singlethread to use the system allocator, gcc
(Debian 14.2.0-3) correctly sees that we were using the value of a
pointer after it had been passed to the privileged free(), which is UB.

Flip the check and dealloc, so that we query the set of pointers we're
tracking first, using the pointer while the allocation is still live.
2024-09-21 14:48:04 +00:00
Nathaniel Wesley Filardo
19259095c6 Further gcc -Werror=array-bounds fix
In test/perf/startup, gcc (Debian 14.2.0-3) seems to get confused about
the size of the counters vector as the code was written.  Rewrite the
code to pass the same value (`std::thread::hardware_concurrency()`, but
in a local) to both `counters.resize()` and the `ParallelTest` ctor.
2024-09-21 10:32:28 +00:00