wasm-micro-runtime/core/iwasm/libraries/lib-wasi-threads/tid_allocator.h

/*
 * Copyright (C) 2023 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 */

#ifndef _TID_ALLOCATOR_H
#define _TID_ALLOCATOR_H

#include "platform_common.h"

#ifdef __cplusplus
extern "C" {
#endif

#define TID_ALLOCATOR_INIT_SIZE CLUSTER_MAX_THREAD_NUM
enum {
    /* Keep it in sync with
       https://github.com/WebAssembly/wasi-threads#design-choice-thread-ids */
    TID_MIN = 1,
    TID_MAX = 0x1FFFFFFF
}; // Reserved TIDs (WASI specification)

/* Stack data structure to track available thread identifiers */
typedef struct {
    int32 *ids;  // Array used to store the stack
    uint32 size; // Stack capacity
    uint32 pos;  // Index of the element after the stack top
} TidAllocator;

bool
tid_allocator_init(TidAllocator *tid_allocator);

void
tid_allocator_deinit(TidAllocator *tid_allocator);

int32
tid_allocator_get_tid(TidAllocator *tid_allocator);

void
tid_allocator_release_tid(TidAllocator *tid_allocator, int32 thread_id);

#ifdef __cplusplus
}
#endif

#endif /* _TID_ALLOCATOR_H */