1 #include <libc/src/unistd.h>
2 #include <libsystem/syscall.h>
3 #include <libc/src/errno.h>
4 #include <libc/src/stdio.h>
5 #include <libc/src/stddef.h>
6 #include <libc/src/string.h>
7 #include <libc/src/fcntl.h>
8
9 /**
10 * @brief 关闭文件接口
11 *
12 * @param fd 文件描述符
13 * @return int
14 */
close(int fd)15 int close(int fd)
16 {
17 return syscall_invoke(SYS_CLOSE, fd, 0, 0, 0, 0, 0, 0, 0);
18 }
19
20 /**
21 * @brief 从文件读取数据的接口
22 *
23 * @param fd 文件描述符
24 * @param buf 缓冲区
25 * @param count 待读取数据的字节数
26 * @return ssize_t 成功读取的字节数
27 */
read(int fd,void * buf,size_t count)28 ssize_t read(int fd, void *buf, size_t count)
29 {
30 return (ssize_t)syscall_invoke(SYS_READ, fd, (uint64_t)buf, count, 0, 0, 0, 0, 0);
31 }
32
33 /**
34 * @brief 向文件写入数据的接口
35 *
36 * @param fd 文件描述符
37 * @param buf 缓冲区
38 * @param count 待写入数据的字节数
39 * @return ssize_t 成功写入的字节数
40 */
write(int fd,void const * buf,size_t count)41 ssize_t write(int fd, void const *buf, size_t count)
42 {
43 return (ssize_t)syscall_invoke(SYS_WRITE, fd, (uint64_t)buf, count, 0, 0, 0, 0, 0);
44 }
45
46 /**
47 * @brief 调整文件的访问位置
48 *
49 * @param fd 文件描述符号
50 * @param offset 偏移量
51 * @param whence 调整模式
52 * @return uint64_t 调整结束后的文件访问位置
53 */
lseek(int fd,off_t offset,int whence)54 off_t lseek(int fd, off_t offset, int whence)
55 {
56 return (off_t)syscall_invoke(SYS_LSEEK, fd, offset, whence, 0, 0, 0, 0, 0);
57 }
58
59 /**
60 * @brief fork当前进程
61 *
62 * @return pid_t
63 */
fork(void)64 pid_t fork(void)
65 {
66 return (pid_t)syscall_invoke(SYS_FORK, 0, 0, 0, 0, 0, 0, 0, 0);
67 }
68
69 /**
70 * @brief fork当前进程，但是与父进程共享VM、flags、fd
71 *
72 * @return pid_t
73 */
vfork(void)74 pid_t vfork(void)
75 {
76 return (pid_t)syscall_invoke(SYS_VFORK, 0, 0, 0, 0, 0, 0, 0, 0);
77 }
78
79 /**
80 * @brief 将堆内存调整为end_brk
81 *
82 * @param end_brk 新的堆区域的结束地址
83 * end_brk=-1 ===> 返回堆区域的起始地址
84 * end_brk=-2 ===> 返回堆区域的结束地址
85 * @return uint64_t 错误码
86 *
87 */
brk(uint64_t end_brk)88 uint64_t brk(uint64_t end_brk)
89 {
90 uint64_t x = (uint64_t)syscall_invoke(SYS_BRK, (uint64_t)end_brk, 0, 0, 0, 0, 0, 0, 0);
91 // printf("brk(): end_brk=%#018lx x=%#018lx", (uint64_t)end_brk, x);
92 return x;
93 }
94
95 /**
96 * @brief 将堆内存空间加上offset（注意，该系统调用只应在普通进程中调用，而不能是内核线程）
97 *
98 * @param increment offset偏移量
99 * @return uint64_t the previous program break
100 */
sbrk(int64_t increment)101 void *sbrk(int64_t increment)
102 {
103 void *retval = (void *)syscall_invoke(SYS_SBRK, (uint64_t)increment, 0, 0, 0, 0, 0, 0, 0);
104 if (retval == (void *)-ENOMEM)
105 return (void *)(-1);
106 else
107 {
108 errno = 0;
109 return (void *)retval;
110 }
111 }
112
113 /**
114 * @brief 切换当前工作目录
115 *
116 * @param dest_path 目标目录
117 * @return int64_t 成功：0,失败：负值（错误码）
118 */
chdir(char * dest_path)119 int64_t chdir(char *dest_path)
120 {
121 if (dest_path == NULL)
122 {
123 errno = -EFAULT;
124 return -1;
125 }
126 else
127 {
128 return syscall_invoke(SYS_CHDIR, (uint64_t)dest_path, 0, 0, 0, 0, 0, 0, 0);
129 }
130 }
131
132 /**
133 * @brief 执行新的程序
134 *
135 * @param path 文件路径
136 * @param argv 参数列表
137 * @return int
138 */
execv(const char * path,char * const argv[])139 int execv(const char *path, char *const argv[])
140 {
141 if (path == NULL)
142 {
143 errno = -ENOENT;
144 return -1;
145 }
146 int retval = syscall_invoke(SYS_EXECVE, (uint64_t)path, (uint64_t)argv, 0, 0, 0, 0, 0, 0);
147 if (retval != 0)
148 return -1;
149 else
150 return 0;
151 }
152
153 /**
154 * @brief 删除文件夹
155 *
156 * @param path 绝对路径
157 * @return int 错误码
158 */
rmdir(const char * path)159 int rmdir(const char *path)
160 {
161 return syscall_invoke(SYS_UNLINK_AT, 0, (uint64_t)path, AT_REMOVEDIR, 0, 0, 0, 0, 0);
162 }
163
164 /**
165 * @brief 删除文件
166 *
167 * @param path 绝对路径
168 * @return int
169 */
rm(const char * path)170 int rm(const char * path)
171 {
172 return syscall_invoke(SYS_UNLINK_AT, 0, (uint64_t)path, 0, 0, 0, 0, 0, 0);
173 }
174
175 /**
176 * @brief 交换n字节
177 * @param src 源地址
178 * @param dest 目的地址
179 * @param nbytes 交换字节数
180 */
swab(void * restrict src,void * restrict dest,ssize_t nbytes)181 void swab(void *restrict src, void *restrict dest, ssize_t nbytes)
182 {
183 unsigned char buf[32];
184 char *_src = src;
185 char *_dest = dest;
186 uint32_t transfer;
187 for (; nbytes > 0; nbytes -= transfer)
188 {
189 transfer = (nbytes > 32) ? 32 : nbytes;
190 memcpy(buf, _src, transfer);
191 memcpy(_src, _dest, transfer);
192 memcpy(_dest, buf, transfer);
193 _src += transfer;
194 _dest += transfer;
195 }
196 }