426 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
		
		
			
		
	
	
			426 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
|  | /*
 | ||
|  |  * Copyright (c) 2006-2021, RT-Thread Development Team | ||
|  |  * | ||
|  |  * SPDX-License-Identifier: Apache-2.0 | ||
|  |  * | ||
|  |  * Change Logs: | ||
|  |  * Date           Author       Notes | ||
|  |  * 2018-06-29     heyuanjie    first version | ||
|  |  * 2019-10-12     Jesven       Add MMU and userspace support | ||
|  |  * 2020-10-08     Bernard      Architecture and code cleanup | ||
|  |  * 2021-08-26     linzhenxing  add lwp_setcwd\lwp_getcwd | ||
|  |  * 2023-11-17     xqyjlj       add process group and session support | ||
|  |  * 2023-12-02     Shell        Add macro to create lwp status and | ||
|  |  *                             fix dead lock problem on pgrp | ||
|  |  */ | ||
|  | 
 | ||
|  | /*
 | ||
|  |  * RT-Thread light-weight process | ||
|  |  */ | ||
|  | #ifndef __LWP_H__
 | ||
|  | #define __LWP_H__
 | ||
|  | 
 | ||
|  | #include <stdint.h>
 | ||
|  | 
 | ||
|  | #include <rthw.h>
 | ||
|  | #include <rtthread.h>
 | ||
|  | #include <dfs.h>
 | ||
|  | 
 | ||
|  | #include "lwp_arch.h"
 | ||
|  | #include "lwp_pid.h"
 | ||
|  | #include "lwp_ipc.h"
 | ||
|  | #include "lwp_signal.h"
 | ||
|  | #include "lwp_syscall.h"
 | ||
|  | #include "lwp_avl.h"
 | ||
|  | #include "mm_aspace.h"
 | ||
|  | 
 | ||
|  | #ifdef RT_USING_MUSLLIBC
 | ||
|  | #include "libc_musl.h"
 | ||
|  | #endif /* RT_USING_MUSLLIBC */
 | ||
|  | 
 | ||
|  | #ifdef ARCH_MM_MMU
 | ||
|  | #include "lwp_shm.h"
 | ||
|  | #include <locale.h>
 | ||
|  | #include "mmu.h"
 | ||
|  | #include "page.h"
 | ||
|  | #else
 | ||
|  | #include "lwp_mpu.h"
 | ||
|  | #endif /* ARCH_MM_MMU */
 | ||
|  | 
 | ||
|  | #ifdef RT_USING_MUSLLIBC
 | ||
|  | #include <locale.h>
 | ||
|  | #endif /* RT_USING_MUSLLIBC */
 | ||
|  | 
 | ||
|  | #ifdef __cplusplus
 | ||
|  | extern "C" { | ||
|  | #endif
 | ||
|  | 
 | ||
|  | #define LWP_MAGIC           0x5A
 | ||
|  | 
 | ||
|  | #define LWP_TYPE_FIX_ADDR   0x01
 | ||
|  | #define LWP_TYPE_DYN_ADDR   0x02
 | ||
|  | 
 | ||
|  | #define LWP_ARG_MAX         8
 | ||
|  | 
 | ||
|  | struct rt_lwp_objs | ||
|  | { | ||
|  |     rt_aspace_t source; | ||
|  |     struct rt_mem_obj mem_obj; | ||
|  | }; | ||
|  | 
 | ||
|  | struct rt_lwp_notify | ||
|  | { | ||
|  |     void (*notify)(rt_wqueue_t *signalfd_queue, int signo); | ||
|  |     rt_wqueue_t *signalfd_queue; | ||
|  |     rt_slist_t list_node; | ||
|  | }; | ||
|  | 
 | ||
|  | struct lwp_tty; | ||
|  | 
 | ||
|  | #ifdef RT_USING_MUSLLIBC
 | ||
|  | #define LWP_COREDUMP_FLAG                         0x80
 | ||
|  | #define LWP_CREATE_STAT_EXIT(exit_code)           (((exit_code)&0xff) << 8)
 | ||
|  | #define LWP_CREATE_STAT_SIGNALED(signo, coredump) (((signo) & 0x7f) | (coredump ? LWP_COREDUMP_FLAG : 0))
 | ||
|  | #define LWP_CREATE_STAT_STOPPED(signo)            (LWP_CREATE_STAT_EXIT(signo) | 0x7f)
 | ||
|  | #define LWP_CREATE_STAT_CONTINUED                 (0xffff)
 | ||
|  | #else
 | ||
|  | #error "No compatible lwp set status provided for this libc"
 | ||
|  | #endif
 | ||
|  | 
 | ||
|  | typedef struct rt_lwp *rt_lwp_t; | ||
|  | typedef struct rt_session *rt_session_t; | ||
|  | typedef struct rt_processgroup *rt_processgroup_t; | ||
|  | 
 | ||
|  | struct rt_session { | ||
|  |     struct rt_object    object; | ||
|  |     rt_lwp_t            leader; | ||
|  |     rt_list_t           processgroup; | ||
|  |     pid_t               sid; | ||
|  |     pid_t               foreground_pgid; | ||
|  |     struct rt_mutex     mutex; | ||
|  |     struct lwp_tty      *ctty; | ||
|  | }; | ||
|  | 
 | ||
|  | struct rt_processgroup { | ||
|  |     struct rt_object    object; | ||
|  |     rt_lwp_t            leader; | ||
|  |     rt_list_t           process; | ||
|  |     rt_list_t           pgrp_list_node; | ||
|  |     pid_t               pgid; | ||
|  |     pid_t               sid; | ||
|  |     struct rt_session   *session; | ||
|  |     struct rt_mutex     mutex; | ||
|  | 
 | ||
|  |     rt_atomic_t         ref; | ||
|  | 
 | ||
|  |     /* flags on process group */ | ||
|  |     unsigned int is_orphaned:1; | ||
|  | }; | ||
|  | 
 | ||
|  | struct rt_lwp | ||
|  | { | ||
|  | #ifdef ARCH_MM_MMU
 | ||
|  |     size_t end_heap; | ||
|  |     rt_aspace_t aspace; | ||
|  | #else
 | ||
|  | #ifdef ARCH_MM_MPU
 | ||
|  |     struct rt_mpu_info mpu_info; | ||
|  | #endif /* ARCH_MM_MPU */
 | ||
|  | #endif /* ARCH_MM_MMU */
 | ||
|  | 
 | ||
|  | #ifdef RT_USING_SMP
 | ||
|  |     int bind_cpu; | ||
|  | #endif
 | ||
|  | 
 | ||
|  |     uint8_t lwp_type; | ||
|  |     uint8_t reserv[3]; | ||
|  | 
 | ||
|  |     /* flags */ | ||
|  |     unsigned int terminated:1; | ||
|  |     unsigned int background:1; | ||
|  |     unsigned int term_ctrlterm:1;  /* have control terminal? */ | ||
|  |     unsigned int did_exec:1;       /* Whether exec has been performed */ | ||
|  |     unsigned int jobctl_stopped:1; /* job control: current proc is stopped */ | ||
|  |     unsigned int wait_reap_stp:1;  /* job control: has wait event for parent */ | ||
|  |     unsigned int sig_protected:1;  /* signal: protected proc cannot be killed or stopped */ | ||
|  | 
 | ||
|  |     struct rt_lwp *parent;          /* parent process */ | ||
|  |     struct rt_lwp *first_child;     /* first child process */ | ||
|  |     struct rt_lwp *sibling;         /* sibling(child) process */ | ||
|  | 
 | ||
|  |     struct rt_wqueue waitpid_waiters; | ||
|  |     lwp_status_t lwp_status; | ||
|  | 
 | ||
|  |     void *text_entry; | ||
|  |     uint32_t text_size; | ||
|  |     void *data_entry; | ||
|  |     uint32_t data_size; | ||
|  | 
 | ||
|  |     rt_atomic_t ref; | ||
|  |     void *args; | ||
|  |     uint32_t args_length; | ||
|  |     pid_t pid; | ||
|  |     pid_t sid;                      /* session ID */ | ||
|  |     pid_t pgid;                     /* process group ID */ | ||
|  |     struct rt_processgroup *pgrp; | ||
|  |     rt_list_t pgrp_node;            /* process group node */ | ||
|  |     rt_list_t t_grp;                /* thread group */ | ||
|  |     rt_list_t timer;                /* POSIX timer object binding to a process */ | ||
|  | 
 | ||
|  |     struct dfs_fdtable fdt; | ||
|  |     char cmd[RT_NAME_MAX]; | ||
|  |     char *exe_file;                 /* process file path */ | ||
|  | 
 | ||
|  |     /* POSIX signal */ | ||
|  |     struct lwp_signal signal; | ||
|  | 
 | ||
|  |     struct lwp_avl_struct *object_root; | ||
|  |     struct rt_mutex object_mutex; | ||
|  |     struct rt_user_context user_ctx; | ||
|  | 
 | ||
|  |     struct rt_wqueue wait_queue; /* for console */ | ||
|  |     struct tty_struct *tty; /* NULL if no tty */ | ||
|  | 
 | ||
|  |     struct lwp_avl_struct *address_search_head; /* for addressed object fast search */ | ||
|  |     char working_directory[DFS_PATH_MAX]; | ||
|  | 
 | ||
|  |     int debug; | ||
|  |     rt_uint32_t bak_first_inst; /* backup of first instruction */ | ||
|  | 
 | ||
|  |     struct rt_mutex lwp_lock; | ||
|  | 
 | ||
|  |     rt_slist_t signalfd_notify_head; | ||
|  | 
 | ||
|  | #ifdef LWP_ENABLE_ASID
 | ||
|  |     uint64_t generation; | ||
|  |     unsigned int asid; | ||
|  | #endif
 | ||
|  |     struct rusage rt_rusage; | ||
|  | }; | ||
|  | 
 | ||
|  | 
 | ||
|  | struct rt_lwp *lwp_self(void); | ||
|  | rt_err_t lwp_children_register(struct rt_lwp *parent, struct rt_lwp *child); | ||
|  | rt_err_t lwp_children_unregister(struct rt_lwp *parent, struct rt_lwp *child); | ||
|  | 
 | ||
|  | enum lwp_exit_request_type | ||
|  | { | ||
|  |     LWP_EXIT_REQUEST_NONE = 0, | ||
|  |     LWP_EXIT_REQUEST_TRIGGERED, | ||
|  |     LWP_EXIT_REQUEST_IN_PROCESS, | ||
|  | }; | ||
|  | struct termios *get_old_termios(void); | ||
|  | void lwp_setcwd(char *buf); | ||
|  | char *lwp_getcwd(void); | ||
|  | int  lwp_check_exit_request(void); | ||
|  | void lwp_terminate(struct rt_lwp *lwp); | ||
|  | 
 | ||
|  | int lwp_tid_init(void); | ||
|  | int lwp_tid_get(void); | ||
|  | void lwp_tid_put(int tid); | ||
|  | 
 | ||
|  | /**
 | ||
|  |  * @brief Automatically get a thread and increase a reference count | ||
|  |  * | ||
|  |  * @param tid queried thread ID | ||
|  |  * @return rt_thread_t | ||
|  |  */ | ||
|  | rt_thread_t lwp_tid_get_thread_and_inc_ref(int tid); | ||
|  | rt_thread_t lwp_tid_get_thread_raw(int tid); | ||
|  | /**
 | ||
|  |  * @brief Decrease a reference count | ||
|  |  * | ||
|  |  * @param thread target thread | ||
|  |  */ | ||
|  | void lwp_tid_dec_ref(rt_thread_t thread); | ||
|  | 
 | ||
|  | void lwp_tid_set_thread(int tid, rt_thread_t thread); | ||
|  | 
 | ||
|  | int lwp_execve(char *filename, int debug, int argc, char **argv, char **envp); | ||
|  | 
 | ||
|  | /*create by lwp_setsid.c*/ | ||
|  | int setsid(void); | ||
|  | #ifdef ARCH_MM_MMU
 | ||
|  | void lwp_aspace_switch(struct rt_thread *thread); | ||
|  | #endif
 | ||
|  | void lwp_user_setting_save(rt_thread_t thread); | ||
|  | void lwp_user_setting_restore(rt_thread_t thread); | ||
|  | 
 | ||
|  | void lwp_uthread_ctx_save(void *ctx); | ||
|  | void lwp_uthread_ctx_restore(void); | ||
|  | 
 | ||
|  | int lwp_setaffinity(pid_t pid, int cpu); | ||
|  | 
 | ||
|  | pid_t exec(char *filename, int debug, int argc, char **argv); | ||
|  | 
 | ||
|  | /* ctime lwp API */ | ||
|  | int timer_list_free(rt_list_t *timer_list); | ||
|  | 
 | ||
|  | rt_err_t lwp_futex_init(void); | ||
|  | rt_err_t lwp_futex(struct rt_lwp *lwp, int *uaddr, int op, int val, | ||
|  |                    const struct timespec *timeout, int *uaddr2, int val3); | ||
|  | 
 | ||
|  | /* processgroup api */ | ||
|  | rt_inline pid_t lwp_pgid_get_bypgrp(rt_processgroup_t group) | ||
|  | { | ||
|  |     return group ? group->pgid : 0; | ||
|  | } | ||
|  | 
 | ||
|  | rt_inline pid_t lwp_pgid_get_byprocess(rt_lwp_t process) | ||
|  | { | ||
|  |     return process ? process->pgid : 0; | ||
|  | } | ||
|  | rt_processgroup_t lwp_pgrp_find(pid_t pgid); | ||
|  | void lwp_pgrp_dec_ref(rt_processgroup_t pgrp); | ||
|  | rt_processgroup_t lwp_pgrp_find_and_inc_ref(pid_t pgid); | ||
|  | rt_processgroup_t lwp_pgrp_create(rt_lwp_t leader); | ||
|  | int lwp_pgrp_delete(rt_processgroup_t group); | ||
|  | 
 | ||
|  | /**
 | ||
|  |  * Note: all the pgrp with process operation must be called in the context where | ||
|  |  * process lock is taken. This is protect us from a possible dead lock condition | ||
|  |  * | ||
|  |  * The order is mandatory in the case: | ||
|  |  *      PGRP_LOCK(pgrp); | ||
|  |  *      LWP_LOCK(p); | ||
|  |  *      ... bussiness logic | ||
|  |  *      LWP_UNLOCK(p); | ||
|  |  *      PGRP_UNLOCK(pgrp); | ||
|  |  */ | ||
|  | int lwp_pgrp_insert(rt_processgroup_t group, rt_lwp_t process); | ||
|  | int lwp_pgrp_remove(rt_processgroup_t group, rt_lwp_t process); | ||
|  | int lwp_pgrp_move(rt_processgroup_t group, rt_lwp_t process); | ||
|  | 
 | ||
|  | int lwp_pgrp_update_children_info(rt_processgroup_t group, pid_t sid, pid_t pgid); | ||
|  | 
 | ||
|  | /* session api */ | ||
|  | rt_inline pid_t lwp_sid_get_bysession(rt_session_t session) | ||
|  | { | ||
|  |     return session ? session->sid : 0; | ||
|  | } | ||
|  | 
 | ||
|  | rt_inline pid_t lwp_sid_get_bypgrp(rt_processgroup_t group) | ||
|  | { | ||
|  |     return group ? group->sid : 0; | ||
|  | } | ||
|  | 
 | ||
|  | rt_inline pid_t lwp_sid_get_byprocess(rt_lwp_t process) | ||
|  | { | ||
|  |     return process ? process->sid : 0; | ||
|  | } | ||
|  | 
 | ||
|  | rt_session_t lwp_session_find(pid_t sid); | ||
|  | rt_session_t lwp_session_create(struct rt_lwp *leader); | ||
|  | int lwp_session_delete(rt_session_t session); | ||
|  | 
 | ||
|  | /**
 | ||
|  |  * Note: all the session operation must be called in the context where | ||
|  |  * process lock is taken. This is protect us from a possible dead lock condition | ||
|  |  * | ||
|  |  * The order is mandatory in the case: | ||
|  |  *      PGRP_LOCK(pgrp); | ||
|  |  *      LWP_LOCK(p); | ||
|  |  *      ... bussiness logic | ||
|  |  *      LWP_UNLOCK(p); | ||
|  |  *      PGRP_UNLOCK(pgrp); | ||
|  |  */ | ||
|  | int lwp_session_insert(rt_session_t session, rt_processgroup_t group); | ||
|  | int lwp_session_remove(rt_session_t session, rt_processgroup_t group); | ||
|  | int lwp_session_move(rt_session_t session, rt_processgroup_t group); | ||
|  | int lwp_session_update_children_info(rt_session_t session, pid_t sid); | ||
|  | int lwp_session_set_foreground(rt_session_t session, pid_t pgid); | ||
|  | 
 | ||
|  | /* complete the job control related bussiness on process exit */ | ||
|  | void lwp_jobctrl_on_exit(struct rt_lwp *lwp); | ||
|  | 
 | ||
|  | #ifdef __cplusplus
 | ||
|  | } | ||
|  | #endif
 | ||
|  | 
 | ||
|  | #ifndef AUX_ARRAY_ITEMS_NR
 | ||
|  | #define AUX_ARRAY_ITEMS_NR 32
 | ||
|  | #endif
 | ||
|  | 
 | ||
|  | /* aux key */ | ||
|  | #define AT_NULL 0
 | ||
|  | #define AT_IGNORE 1
 | ||
|  | #define AT_EXECFD 2
 | ||
|  | #define AT_PHDR 3
 | ||
|  | #define AT_PHENT 4
 | ||
|  | #define AT_PHNUM 5
 | ||
|  | #define AT_PAGESZ 6
 | ||
|  | #define AT_BASE 7
 | ||
|  | #define AT_FLAGS 8
 | ||
|  | #define AT_ENTRY 9
 | ||
|  | #define AT_NOTELF 10
 | ||
|  | #define AT_UID 11
 | ||
|  | #define AT_EUID 12
 | ||
|  | #define AT_GID 13
 | ||
|  | #define AT_EGID 14
 | ||
|  | #define AT_CLKTCK 17
 | ||
|  | #define AT_PLATFORM 15
 | ||
|  | #define AT_HWCAP 16
 | ||
|  | #define AT_FPUCW 18
 | ||
|  | #define AT_DCACHEBSIZE 19
 | ||
|  | #define AT_ICACHEBSIZE 20
 | ||
|  | #define AT_UCACHEBSIZE 21
 | ||
|  | #define AT_IGNOREPPC 22
 | ||
|  | #define AT_SECURE 23
 | ||
|  | #define AT_BASE_PLATFORM 24
 | ||
|  | #define AT_RANDOM 25
 | ||
|  | #define AT_HWCAP2 26
 | ||
|  | #define AT_EXECFN 31
 | ||
|  | 
 | ||
|  | struct process_aux_item | ||
|  | { | ||
|  |     size_t key; | ||
|  |     size_t value; | ||
|  | }; | ||
|  | 
 | ||
|  | struct process_aux | ||
|  | { | ||
|  |     struct process_aux_item item[AUX_ARRAY_ITEMS_NR]; | ||
|  | }; | ||
|  | 
 | ||
|  | struct lwp_args_info | ||
|  | { | ||
|  |     char **argv; | ||
|  |     char **envp; | ||
|  |     int argc; | ||
|  |     int envc; | ||
|  |     int size; | ||
|  | }; | ||
|  | 
 | ||
|  | struct dbg_ops_t | ||
|  | { | ||
|  |     int (*dbg)(int argc, char **argv); | ||
|  |     uint32_t (*arch_get_ins)(void); | ||
|  |     void (*arch_activate_step)(void); | ||
|  |     void (*arch_deactivate_step)(void); | ||
|  |     int (*check_debug_event)(struct rt_hw_exp_stack *regs, unsigned long esr); | ||
|  |     rt_channel_t (*gdb_get_server_channel)(void); | ||
|  |     int (*gdb_get_step_type)(void); | ||
|  |     void (*lwp_check_debug_attach_req)(void *pc); | ||
|  |     int (*lwp_check_debug_suspend)(void); | ||
|  | }; | ||
|  | extern struct dbg_ops_t *rt_dbg_ops; | ||
|  | 
 | ||
|  | int dbg_thread_in_debug(void); | ||
|  | void dbg_register(struct dbg_ops_t *dbg_ops); | ||
|  | 
 | ||
|  | uint32_t dbg_get_ins(void); | ||
|  | void dbg_activate_step(void); | ||
|  | void dbg_deactivate_step(void); | ||
|  | int dbg_check_event(struct rt_hw_exp_stack *regs, unsigned long arg); | ||
|  | rt_channel_t gdb_server_channel(void); | ||
|  | int dbg_step_type(void); | ||
|  | void dbg_attach_req(void *pc); | ||
|  | int dbg_check_suspend(void); | ||
|  | void rt_hw_set_process_id(int pid); | ||
|  | void lwp_futex_exit_robust_list(rt_thread_t thread); | ||
|  | 
 | ||
|  | /* backtrace service */ | ||
|  | rt_err_t lwp_backtrace_frame(rt_thread_t uthread, struct rt_hw_backtrace_frame *frame); | ||
|  | 
 | ||
|  | #endif
 |