84 lines
2.7 KiB
C++
84 lines
2.7 KiB
C++
// 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
|
|
//
|
|
// Unless required by applicable law or agreed to in writing,
|
|
// software distributed under the License is distributed on an
|
|
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
// KIND, either express or implied. See the License for the
|
|
// specific language governing permissions and limitations
|
|
// under the License.
|
|
|
|
// bthread - A M:N threading library to make applications more concurrent.
|
|
|
|
// Date: Sun Sep 7 22:37:39 CST 2014
|
|
|
|
#ifndef BTHREAD_ALLOCATE_STACK_H
|
|
#define BTHREAD_ALLOCATE_STACK_H
|
|
|
|
#include <assert.h>
|
|
#include <gflags/gflags.h> // DECLARE_int32
|
|
#include "bthread/types.h"
|
|
#include "bthread/context.h" // bthread_fcontext_t
|
|
#include "butil/object_pool.h"
|
|
|
|
namespace bthread {
|
|
|
|
struct StackStorage {
|
|
int stacksize;
|
|
int guardsize;
|
|
// Assume stack grows upwards.
|
|
// http://www.boost.org/doc/libs/1_55_0/libs/context/doc/html/context/stack.html
|
|
void* bottom;
|
|
unsigned valgrind_stack_id;
|
|
|
|
// Clears all members.
|
|
void zeroize() {
|
|
stacksize = 0;
|
|
guardsize = 0;
|
|
bottom = NULL;
|
|
valgrind_stack_id = 0;
|
|
}
|
|
};
|
|
|
|
// Allocate a piece of stack.
|
|
int allocate_stack_storage(StackStorage* s, int stacksize, int guardsize);
|
|
// Deallocate a piece of stack. Parameters MUST be returned or set by the
|
|
// corresponding allocate_stack_storage() otherwise behavior is undefined.
|
|
void deallocate_stack_storage(StackStorage* s);
|
|
|
|
enum StackType {
|
|
STACK_TYPE_MAIN = 0,
|
|
STACK_TYPE_PTHREAD = BTHREAD_STACKTYPE_PTHREAD,
|
|
STACK_TYPE_SMALL = BTHREAD_STACKTYPE_SMALL,
|
|
STACK_TYPE_NORMAL = BTHREAD_STACKTYPE_NORMAL,
|
|
STACK_TYPE_LARGE = BTHREAD_STACKTYPE_LARGE
|
|
};
|
|
|
|
struct ContextualStack {
|
|
bthread_fcontext_t context;
|
|
StackType stacktype;
|
|
StackStorage storage;
|
|
};
|
|
|
|
// Get a stack in the `type' and run `entry' at the first time that the
|
|
// stack is jumped.
|
|
ContextualStack* get_stack(StackType type, void (*entry)(intptr_t));
|
|
// Recycle a stack. NULL does nothing.
|
|
void return_stack(ContextualStack*);
|
|
// Jump from stack `from' to stack `to'. `from' must be the stack of callsite
|
|
// (to save contexts before jumping)
|
|
void jump_stack(ContextualStack* from, ContextualStack* to);
|
|
|
|
} // namespace bthread
|
|
|
|
#include "bthread/stack_inl.h"
|
|
|
|
#endif // BTHREAD_ALLOCATE_STACK_H
|