#ifndef RUST_TASK_THREAD_H #define RUST_TASK_THREAD_H #include "context.h" #ifndef _WIN32 #include #else #include #endif struct rust_task_thread; struct rust_hashable_dict { UT_hash_handle hh; void* fields[0]; }; class rust_crate_cache { public: type_desc *get_type_desc(size_t size, size_t align, size_t n_descs, type_desc const **descs, uintptr_t n_obj_params); void** get_dict(size_t n_fields, void** dict); private: type_desc *type_descs; rust_hashable_dict *dicts; public: rust_task_thread *thread; size_t idx; rust_crate_cache(rust_task_thread *thread); ~rust_crate_cache(); void flush(); }; struct rust_task_thread : public kernel_owned, rust_thread { RUST_REFCOUNTED(rust_task_thread) // Fields known only by the runtime: rust_log _log; // NB: this is used to filter *runtime-originating* debug // logging, on a per-scheduler basis. It's not likely what // you want to expose to the user in terms of per-task // or per-module logging control. By default all schedulers // are set to debug-level logging here, and filtered by // runtime category using the pseudo-modules ::rt::foo. uint32_t log_lvl; rust_srv *srv; const char *const name; rust_task_list newborn_tasks; rust_task_list running_tasks; rust_task_list blocked_tasks; rust_task_list dead_tasks; rust_crate_cache cache; randctx rctx; rust_kernel *kernel; rust_scheduler *sched; int32_t list_index; const int id; lock_and_signal lock; size_t min_stack_size; #ifndef __WIN32__ pthread_attr_t attr; static pthread_key_t task_key; #else static DWORD task_key; #endif static bool tls_initialized; rust_env *env; context c_context; bool should_exit; // Only a pointer to 'name' is kept, so it must live as long as this // domain. rust_task_thread(rust_scheduler *sched, rust_srv *srv, int id); ~rust_task_thread(); void activate(rust_task *task); void log(rust_task *task, uint32_t level, char const *fmt, ...); rust_log & get_log(); void fail(); rust_crate_cache *get_cache(); size_t number_of_live_tasks(); void reap_dead_tasks(); rust_task *schedule_task(); void start_main_loop(); void log_state(); void kill_all_tasks(); rust_task *create_task(rust_task *spawner, const char *name, size_t init_stack_sz); virtual void run(); #ifdef __WIN32__ inline void win32_require(LPCTSTR fn, BOOL ok) { kernel->win32_require(fn, ok); } #endif void init_tls(); void place_task_in_tls(rust_task *task); static rust_task *get_task(); // Tells the scheduler to exit it's scheduling loop and thread void exit(); }; inline rust_log & rust_task_thread::get_log() { return _log; } // // Local Variables: // mode: C++ // fill-column: 78; // indent-tabs-mode: nil // c-basic-offset: 4 // buffer-file-coding-system: utf-8-unix // compile-command: "make -k -C $RBUILD 2>&1 | sed -e 's/\\/x\\//x:\\//g'"; // End: // #endif /* RUST_TASK_THREAD_H */