process.h

Go to the documentation of this file.

 
#ifndef __PROCESS_H__
#define __PROCESS_H__
 
#if defined(__GNUC__) || defined(__clang__)
#ifndef __cplusplus
#include <stdbool.h>
#endif
#endif
#include <stdint.h>
#include <Fs/vfs.h>
#include <Cred/group.h>
#include <Cred/user.h>
#include <Cap/capability.h>
 
#ifdef ARCH_ARM64
#include <Hal/AA64/sched.h>
#endif
 
#ifdef ARCH_X64
#include <Hal/x86_64_sched.h>
#include <Sync/mutex.h>
#endif
#include <list.h>
 
#define PROCESS_USER_STACK_SZ 512*1024
 
#define MAX_THREADS_PER_PROCESS  60
#define FILE_DESC_PER_PROCESS 60
 
#define PROCESS_STATE_NOT_READY  (1<<0)
#define PROCESS_STATE_READY      (1<<1)
#define PROCESS_STATE_SUSPENDED  (1<<2)
 
/* when a process exits, it's marked
 * as died */
#define PROCESS_STATE_DIED       (1<<3) 
 
/* when a process's parents dies, whitout
 * terminating its childs, childs are 
 * marked as zombies
 */
#define PROCESS_STATE_ZOMBIE     (1<<4)
#define PROCESS_STATE_ORPHAN     (1<<5)
 
/* process state busy wait is set when
 * any locks set its as waiting 
 */
#define PROCESS_STATE_BUSY_WAIT  (1<<6)
 
/* Exit codes for processes */
#define PROCESS_EXIT_SUCCESS  0
#define PROCESS_EXIT_FAILURE  1
 
/* process types */
#define PROCESS_TYPE_GENERAL      (0<<1)
#define PROCESS_TYPE_SYSTEM       (1<<1)
#define PROCESS_TYPE_BACKGROUND   (1<<2)
#define PROCESS_TYPE_NON_KILLABLE (1<<3)
 
#define PROCESS_BREAK_ADDRESS   0x0000003000000000
#ifdef ARCH_X64
#define PROCESS_MMAP_ADDRESS    0x00000000C0000000
#define PROCESS_SHM_ADDRESS     0x0000000080000000
#elif ARCH_ARM64
#define PROCESS_MMAP_ADDRESS    0x0000000C00000000
#define PROCESS_SHM_ADDRESS     0x0000000800000000
#endif
 
typedef void(*entry) (void*);
 
typedef struct _au_proc_cred_ {
    uint8_t caps;
    UID_NUM uid;
    GID_NUM gid;
    GID_NUM sgid[AURORA_MAX_GROUPS];
    uint16_t num_sgid;
}AuProcCredentials;
 
typedef struct _sys_proc_list_ {
    int proc_id;
    char name[16];
    uint64_t total_runtime_us;
    uint64_t window_runtime_us;
    uint32_t num_threads;
    uint32_t num_file_opened;
    uint32_t cpu_usage;
}AuProcessList;
 
 
//#pragma pack(push,1)
typedef struct _au_proc_ {
    char name[16];
    int proc_id;
    uint8_t state;
    uint8_t type_flags;
    
    /* process image related stuff */
    uint64_t  *cr3;
    uint64_t _image_size_;
    uint64_t _image_base_;
    uint64_t _main_stack_;
    size_t _user_stack_index_;
    size_t _kstack_index_;
    uint64_t _envp_block_;
    uint64_t total_runtime_us;
    uint64_t window_runtime_us;
    uint32_t cpu_usage;
    uint64_t prev_sample_runtime_us;
    uint64_t prev_sample_time_us;
#ifdef ARCH_X64
    /* threading section */
    AuThread* main_thread;
#elif ARCH_ARM64
    AA64Thread* main_thread;
#endif // Maybe RISC-V?
 
    uint8_t num_thread;
    entry entry_point;
 
#ifdef ARCH_X64
    AuThread* threads[MAX_THREADS_PER_PROCESS];
#elif ARCH_ARM64
    AA64Thread* threads[MAX_THREADS_PER_PROCESS];
#endif // Maybe RISC-V?
 
    /* file descriptors */
    AuVFSNode* fds[FILE_DESC_PER_PROCESS];
    /* capability table -- parallel to fds[]*/
    AuCapability caps[FILE_DESC_PER_PROCESS];
    /*loader related data*/
    AuVFSNode *file;
    AuVFSNode *fsys;
 
    /* memory account */
    list_t* vmareas;
    list_t* shmmaps;
    list_t* waitlist;
    size_t shm_break;
    size_t proc_mem_heap;
    size_t proc_heapmem_len;
    size_t proc_mmap_len;
 
    AuProcCredentials creds;
 
    /* data structure */
    struct _au_proc_ *next;
    struct _au_proc_ *prev;
}AuProcess;
 
 
//#pragma pack(pop)
/*
* AuAddProcess -- adds process to kernel data structure
* @param root -- pointer to the root process
* @param proc -- process to add
*/
extern void AuAddProcess(AuProcess* parent, AuProcess *proc);
 
/*
* AuRemoveProcess -- removes a process from the process
* data structure
* @param parent -- pointer to the parent process
* @param proc -- process to remove
*/
extern void AuRemoveProcess(AuProcess* parent, AuProcess* proc);
 
 
/*
* Allocate kernel stack
* @param cr3 -- root page map level, it should be
* converted to linear virtual address
*/
extern uint64_t CreateKernelStack(AuProcess* proc, uint64_t *cr3);
/*
* KernelStackFree -- frees up an allocated stack
* @param proc -- Pointer to process
* @param ptr -- Starting address of the stack
* @param cr3 -- page root level mapping
*/
extern void KernelStackFree(AuProcess* proc, void* ptr, uint64_t *cr3);
 
/*
* AuProcessFindPID -- finds a process by its pid
* @param pid -- process id to find
*/
extern AuProcess* AuProcessFindPID(int pid);
 
#ifdef ARCH_X64
/*
* AuProcessFindThread -- finds a process by its
* main thread
* @param thread -- pointer to  main thread
*/
extern AuProcess *AuProcessFindThread(AuThread* thread);
 
/*
* AuProcessFindSubThread -- find a process from its
* sub threads which contain a pointer to its process
* slot
* @param thread -- Pointer to sub thread
*/
extern AuProcess* AuProcessFindSubThread(AuThread* thread);
#elif ARCH_ARM64
/*
* AuProcessFindThread -- finds a process by its
* main thread
* @param thread -- pointer to  main thread
*/
extern AuProcess* AuProcessFindThread(AA64Thread* thread);
 
/*
* AuProcessFindSubThread -- find a process from its
* sub threads which contain a pointer to its process
* slot
* @param thread -- Pointer to sub thread
*/
extern AuProcess* AuProcessFindSubThread(AA64Thread* thread);
 
#endif
 
/*
* CreateUserStack -- creates new user stack
* @param proc -- Pointer to process
* @param cr3 -- pointer to the address space where to
* map
*/
extern uint64_t* CreateUserStack(AuProcess* proc,uint64_t* cr3);
 
/*
* AuStartRootProc -- starts the very first process
* of aurora system
*/
extern void AuStartRootProc();
 
/*
* AuGetRootProcess -- returns the root process
*/
extern AuProcess* AuGetRootProcess();
 
/*
* AuCreateProcessSlot -- creates a blank process slot
* @param parent -- pointer to the parent process
*/
extern AuProcess* AuCreateProcessSlot(AuProcess* parent, char* name);
 
/*
* AuProcessFork -- fork a parent process
* @param parent -- pointer to parent process
*/
extern AuProcess* AuProcessFork(AuProcess* parent);
 
 
/* AuProcessExit -- marks a process
* as killable
* @param proc -- process to exit
*/
extern void AuProcessExit(AuProcess* proc, bool schedulable);
 
/*
* AuProcessGetFileDesc -- returns a empty file descriptor
* from process slot
* @param proc -- pointer to process slot
*/
extern int AuProcessGetFileDesc(AuProcess* proc);
 
/*
* AuProcessWaitForTermination -- waits for termination
* of child processes
* @param proc -- pointer to process who needs to
* wait for termination
* @param pid -- pid of the process, if -1 then any child
* process
*/
extern int AuProcessWaitForTermination(AuProcess *proc, int pid);
 
#ifdef ARCH_X64
extern AuMutex* AuProcessGetMutex();
#endif
 
#ifdef ARCH_ARM64
extern uint64_t AuCreateSubKernelStack(AuProcess* proc, uint64_t* pml);
/*
 * @brief CreateSubUserStack -- creates new user stack
 * @param proc -- Pointer to process slot
 * @param cr3 -- pointer to the address space where to
 * map
 */
extern uint64_t* CreateSubUserStack(AuProcess* proc, uint64_t* cr3);
#endif
extern int AuCreateUserthread(AuProcess* proc, void(*entry) (), char *name);
 
extern int AuProcGetNumProcessCount();
 
extern int AuProcessFetch(AuProcessList* list, int num_proc_count);
 
#endif