refactor code

This commit is contained in:
CismonX 2020-06-16 13:09:44 +08:00
parent e9c4b35e81
commit 86a54e6faf
No known key found for this signature in database
GPG Key ID: 315D6652268C5007
2 changed files with 22 additions and 13 deletions

View File

@ -17,7 +17,6 @@
* along with this program. If not, see <https://www.gnu.org/licenses/>. * along with this program. If not, see <https://www.gnu.org/licenses/>.
*/ */
#include "vm_stack.h"
#include "vm_pool.h" #include "vm_pool.h"
#include "logging.h" #include "logging.h"

View File

@ -22,7 +22,6 @@
#include "common.h" #include "common.h"
#include "vm_defs.h" #include "vm_defs.h"
#include "logging.h"
#include <stdint.h> #include <stdint.h>
#include <stdbool.h> #include <stdbool.h>
@ -55,15 +54,26 @@ struct u6a_vm_pool_ctx {
const char* err_stage; const char* err_stage;
}; };
// Forward declarations
struct u6a_vm_stack*
u6a_vm_stack_dup(struct u6a_vm_stack_ctx* ctx, struct u6a_vm_stack* vs);
void
u6a_vm_stack_discard(struct u6a_vm_stack_ctx* ctx, struct u6a_vm_stack* vs);
void
u6a_err_vm_pool_oom(const char* stage);
static inline void static inline void
free_stack_push(struct u6a_vm_pool_ctx* ctx, struct u6a_vm_var_fn fn) { u6a_free_stack_push_(struct u6a_vm_pool_ctx* ctx, struct u6a_vm_var_fn fn) {
if (fn.token.fn & U6A_VM_FN_REF) { if (fn.token.fn & U6A_VM_FN_REF) {
ctx->fstack[++ctx->fstack_top] = ctx->active_pool->elems + fn.ref; ctx->fstack[++ctx->fstack_top] = ctx->active_pool->elems + fn.ref;
} }
} }
static inline struct u6a_vm_pool_elem* static inline struct u6a_vm_pool_elem*
free_stack_pop(struct u6a_vm_pool_ctx* ctx) { u6a_free_stack_pop_(struct u6a_vm_pool_ctx* ctx) {
if (ctx->fstack_top == UINT32_MAX) { if (ctx->fstack_top == UINT32_MAX) {
return NULL; return NULL;
} }
@ -71,7 +81,7 @@ free_stack_pop(struct u6a_vm_pool_ctx* ctx) {
} }
static inline struct u6a_vm_pool_elem* static inline struct u6a_vm_pool_elem*
vm_pool_elem_alloc(struct u6a_vm_pool_ctx* ctx) { u6a_vm_pool_elem_alloc_(struct u6a_vm_pool_ctx* ctx) {
struct u6a_vm_pool* pool = ctx->active_pool; struct u6a_vm_pool* pool = ctx->active_pool;
struct u6a_vm_pool_elem_ptrs* holes = ctx->holes; struct u6a_vm_pool_elem_ptrs* holes = ctx->holes;
struct u6a_vm_pool_elem* new_elem; struct u6a_vm_pool_elem* new_elem;
@ -89,8 +99,8 @@ vm_pool_elem_alloc(struct u6a_vm_pool_ctx* ctx) {
} }
static inline struct u6a_vm_pool_elem* static inline struct u6a_vm_pool_elem*
vm_pool_elem_dup(struct u6a_vm_pool_ctx* ctx, struct u6a_vm_pool_elem* elem) { u6a_vm_pool_elem_dup_(struct u6a_vm_pool_ctx* ctx, struct u6a_vm_pool_elem* elem) {
struct u6a_vm_pool_elem* new_elem = vm_pool_elem_alloc(ctx); struct u6a_vm_pool_elem* new_elem = u6a_vm_pool_elem_alloc_(ctx);
if (UNLIKELY(new_elem == NULL)) { if (UNLIKELY(new_elem == NULL)) {
return NULL; return NULL;
} }
@ -103,7 +113,7 @@ u6a_vm_pool_init(struct u6a_vm_pool_ctx* ctx, uint32_t pool_len, uint32_t ins_le
static inline uint32_t static inline uint32_t
u6a_vm_pool_alloc1(struct u6a_vm_pool_ctx* ctx, struct u6a_vm_var_fn v1) { u6a_vm_pool_alloc1(struct u6a_vm_pool_ctx* ctx, struct u6a_vm_var_fn v1) {
struct u6a_vm_pool_elem* elem = vm_pool_elem_alloc(ctx); struct u6a_vm_pool_elem* elem = u6a_vm_pool_elem_alloc_(ctx);
if (UNLIKELY(elem == NULL)) { if (UNLIKELY(elem == NULL)) {
return UINT32_MAX; return UINT32_MAX;
} }
@ -114,7 +124,7 @@ u6a_vm_pool_alloc1(struct u6a_vm_pool_ctx* ctx, struct u6a_vm_var_fn v1) {
static inline uint32_t static inline uint32_t
u6a_vm_pool_alloc2(struct u6a_vm_pool_ctx* ctx, struct u6a_vm_var_fn v1, struct u6a_vm_var_fn v2) { u6a_vm_pool_alloc2(struct u6a_vm_pool_ctx* ctx, struct u6a_vm_var_fn v1, struct u6a_vm_var_fn v2) {
struct u6a_vm_pool_elem* elem = vm_pool_elem_alloc(ctx); struct u6a_vm_pool_elem* elem = u6a_vm_pool_elem_alloc_(ctx);
if (UNLIKELY(elem == NULL)) { if (UNLIKELY(elem == NULL)) {
return UINT32_MAX; return UINT32_MAX;
} }
@ -125,7 +135,7 @@ u6a_vm_pool_alloc2(struct u6a_vm_pool_ctx* ctx, struct u6a_vm_var_fn v1, struct
static inline uint32_t static inline uint32_t
u6a_vm_pool_alloc2_ptr(struct u6a_vm_pool_ctx* ctx, void* v1, void* v2) { u6a_vm_pool_alloc2_ptr(struct u6a_vm_pool_ctx* ctx, void* v1, void* v2) {
struct u6a_vm_pool_elem* elem = vm_pool_elem_alloc(ctx); struct u6a_vm_pool_elem* elem = u6a_vm_pool_elem_alloc_(ctx);
if (UNLIKELY(elem == NULL)) { if (UNLIKELY(elem == NULL)) {
return UINT32_MAX; return UINT32_MAX;
} }
@ -172,11 +182,11 @@ u6a_vm_pool_free(struct u6a_vm_pool_ctx* ctx, uint32_t offset) {
// Continuation destroyed before used // Continuation destroyed before used
u6a_vm_stack_discard(ctx->stack_ctx, elem->values.v1.ptr); u6a_vm_stack_discard(ctx->stack_ctx, elem->values.v1.ptr);
} else { } else {
free_stack_push(ctx, elem->values.v2.fn); u6a_free_stack_push_(ctx, elem->values.v2.fn);
free_stack_push(ctx, elem->values.v1.fn); u6a_free_stack_push_(ctx, elem->values.v1.fn);
} }
} }
} while ((elem = free_stack_pop(ctx))); } while ((elem = u6a_free_stack_pop_(ctx)));
} }
void void