System.cpp 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434
  1. /*
  2. * Copyright (c) 2021, Andreas Kling <kling@serenityos.org>
  3. * Copyright (c) 2021, Kenneth Myhra <kennethmyhra@gmail.com>
  4. *
  5. * SPDX-License-Identifier: BSD-2-Clause
  6. */
  7. #include <AK/String.h>
  8. #include <LibCore/System.h>
  9. #include <LibSystem/syscall.h>
  10. #include <fcntl.h>
  11. #include <stdarg.h>
  12. #include <sys/ioctl.h>
  13. #include <sys/mman.h>
  14. #include <sys/ptrace.h>
  15. #include <sys/socket.h>
  16. #include <termios.h>
  17. #include <unistd.h>
  18. #define HANDLE_SYSCALL_RETURN_VALUE(syscall_name, rc, success_value) \
  19. if ((rc) < 0) { \
  20. return Error::from_syscall(syscall_name, rc); \
  21. } \
  22. return success_value;
  23. namespace Core::System {
  24. #ifdef __serenity__
  25. ErrorOr<void> pledge(StringView promises, StringView execpromises)
  26. {
  27. Syscall::SC_pledge_params params {
  28. { promises.characters_without_null_termination(), promises.length() },
  29. { execpromises.characters_without_null_termination(), execpromises.length() },
  30. };
  31. int rc = syscall(SC_pledge, &params);
  32. HANDLE_SYSCALL_RETURN_VALUE("pledge"sv, rc, {});
  33. }
  34. ErrorOr<void> unveil(StringView path, StringView permissions)
  35. {
  36. Syscall::SC_unveil_params params {
  37. { path.characters_without_null_termination(), path.length() },
  38. { permissions.characters_without_null_termination(), permissions.length() },
  39. };
  40. int rc = syscall(SC_unveil, &params);
  41. HANDLE_SYSCALL_RETURN_VALUE("unveil"sv, rc, {});
  42. }
  43. ErrorOr<Array<int, 2>> pipe2(int flags)
  44. {
  45. Array<int, 2> fds;
  46. if (::pipe2(fds.data(), flags) < 0)
  47. return Error::from_syscall("pipe2"sv, -errno);
  48. return fds;
  49. }
  50. ErrorOr<void> sendfd(int sockfd, int fd)
  51. {
  52. if (::sendfd(sockfd, fd) < 0)
  53. return Error::from_syscall("sendfd"sv, -errno);
  54. return {};
  55. }
  56. ErrorOr<int> recvfd(int sockfd, int options)
  57. {
  58. auto fd = ::recvfd(sockfd, options);
  59. if (fd < 0)
  60. return Error::from_syscall("recvfd"sv, -errno);
  61. return fd;
  62. }
  63. ErrorOr<void> ptrace_peekbuf(pid_t tid, void const* tracee_addr, Bytes destination_buf)
  64. {
  65. Syscall::SC_ptrace_buf_params buf_params {
  66. { destination_buf.data(), destination_buf.size() }
  67. };
  68. Syscall::SC_ptrace_params params {
  69. PT_PEEKBUF,
  70. tid,
  71. const_cast<void*>(tracee_addr),
  72. (FlatPtr)&buf_params,
  73. };
  74. int rc = syscall(SC_ptrace, &params);
  75. HANDLE_SYSCALL_RETURN_VALUE("ptrace_peekbuf", rc, {});
  76. }
  77. ErrorOr<void> setgroups(Span<gid_t const> gids)
  78. {
  79. if (::setgroups(gids.size(), gids.data()) < 0)
  80. return Error::from_syscall("setgroups"sv, -errno);
  81. return {};
  82. }
  83. #endif
  84. ErrorOr<void> sigaction(int signal, struct sigaction const* action, struct sigaction* old_action)
  85. {
  86. if (::sigaction(signal, action, old_action) < 0)
  87. return Error::from_syscall("sigaction"sv, -errno);
  88. return {};
  89. }
  90. ErrorOr<struct stat> fstat(int fd)
  91. {
  92. struct stat st = {};
  93. if (::fstat(fd, &st) < 0)
  94. return Error::from_syscall("fstat"sv, -errno);
  95. return st;
  96. }
  97. ErrorOr<int> fcntl(int fd, int command, ...)
  98. {
  99. va_list ap;
  100. va_start(ap, command);
  101. u32 extra_arg = va_arg(ap, u32);
  102. int rc = ::fcntl(fd, command, extra_arg);
  103. va_end(ap);
  104. if (rc < 0)
  105. return Error::from_syscall("fcntl"sv, -errno);
  106. return rc;
  107. }
  108. ErrorOr<void*> mmap(void* address, size_t size, int protection, int flags, int fd, off_t offset, [[maybe_unused]] size_t alignment, [[maybe_unused]] StringView name)
  109. {
  110. #ifdef __serenity__
  111. Syscall::SC_mmap_params params { (uintptr_t)address, size, alignment, protection, flags, fd, offset, { name.characters_without_null_termination(), name.length() } };
  112. ptrdiff_t rc = syscall(SC_mmap, &params);
  113. if (rc < 0 && rc > -EMAXERRNO)
  114. return Error::from_syscall("mmap"sv, rc);
  115. return reinterpret_cast<void*>(rc);
  116. #else
  117. // NOTE: Regular POSIX mmap() doesn't support custom alignment requests.
  118. VERIFY(!alignment);
  119. auto* ptr = ::mmap(address, size, protection, flags, fd, offset);
  120. if (ptr == MAP_FAILED)
  121. return Error::from_syscall("mmap"sv, -errno);
  122. return ptr;
  123. #endif
  124. }
  125. ErrorOr<void> munmap(void* address, size_t size)
  126. {
  127. if (::munmap(address, size) < 0)
  128. return Error::from_syscall("munmap"sv, -errno);
  129. return {};
  130. }
  131. ErrorOr<int> open(StringView path, int options, ...)
  132. {
  133. if (!path.characters_without_null_termination())
  134. return Error::from_syscall("open"sv, -EFAULT);
  135. va_list ap;
  136. va_start(ap, options);
  137. auto mode = (mode_t)va_arg(ap, unsigned);
  138. va_end(ap);
  139. #ifdef __serenity__
  140. Syscall::SC_open_params params { AT_FDCWD, { path.characters_without_null_termination(), path.length() }, options, mode };
  141. int rc = syscall(SC_open, &params);
  142. HANDLE_SYSCALL_RETURN_VALUE("open"sv, rc, rc);
  143. #else
  144. // NOTE: We have to ensure that the path is null-terminated.
  145. String path_string = path;
  146. int rc = ::open(path_string.characters(), options, mode);
  147. if (rc < 0)
  148. return Error::from_syscall("open"sv, -errno);
  149. return rc;
  150. #endif
  151. }
  152. ErrorOr<void> close(int fd)
  153. {
  154. if (::close(fd) < 0)
  155. return Error::from_syscall("close"sv, -errno);
  156. return {};
  157. }
  158. ErrorOr<void> ftruncate(int fd, off_t length)
  159. {
  160. if (::ftruncate(fd, length) < 0)
  161. return Error::from_syscall("ftruncate"sv, -errno);
  162. return {};
  163. }
  164. ErrorOr<struct stat> stat(StringView path)
  165. {
  166. if (!path.characters_without_null_termination())
  167. return Error::from_syscall("stat"sv, -EFAULT);
  168. struct stat st = {};
  169. #ifdef __serenity__
  170. Syscall::SC_stat_params params { { path.characters_without_null_termination(), path.length() }, &st, AT_FDCWD, true };
  171. int rc = syscall(SC_stat, &params);
  172. HANDLE_SYSCALL_RETURN_VALUE("stat"sv, rc, st);
  173. #else
  174. String path_string = path;
  175. if (::stat(path_string.characters(), &st) < 0)
  176. return Error::from_syscall("stat"sv, -errno);
  177. return st;
  178. #endif
  179. }
  180. ErrorOr<struct stat> lstat(StringView path)
  181. {
  182. if (!path.characters_without_null_termination())
  183. return Error::from_syscall("lstat"sv, -EFAULT);
  184. struct stat st = {};
  185. #ifdef __serenity__
  186. Syscall::SC_stat_params params { { path.characters_without_null_termination(), path.length() }, &st, AT_FDCWD, false };
  187. int rc = syscall(SC_stat, &params);
  188. HANDLE_SYSCALL_RETURN_VALUE("lstat"sv, rc, st);
  189. #else
  190. String path_string = path;
  191. if (::stat(path_string.characters(), &st) < 0)
  192. return Error::from_syscall("lstat"sv, -errno);
  193. return st;
  194. #endif
  195. }
  196. ErrorOr<ssize_t> read(int fd, Bytes buffer)
  197. {
  198. ssize_t rc = ::read(fd, buffer.data(), buffer.size());
  199. if (rc < 0)
  200. return Error::from_syscall("read"sv, -errno);
  201. return rc;
  202. }
  203. ErrorOr<ssize_t> write(int fd, ReadonlyBytes buffer)
  204. {
  205. ssize_t rc = ::write(fd, buffer.data(), buffer.size());
  206. if (rc < 0)
  207. return Error::from_syscall("write"sv, -errno);
  208. return rc;
  209. }
  210. ErrorOr<void> kill(pid_t pid, int signal)
  211. {
  212. if (::kill(pid, signal) < 0)
  213. return Error::from_syscall("kill"sv, -errno);
  214. return {};
  215. }
  216. ErrorOr<int> dup(int source_fd)
  217. {
  218. int fd = ::dup(source_fd);
  219. if (fd < 0)
  220. return Error::from_syscall("dup"sv, -errno);
  221. return fd;
  222. }
  223. ErrorOr<int> dup2(int source_fd, int destination_fd)
  224. {
  225. int fd = ::dup2(source_fd, destination_fd);
  226. if (fd < 0)
  227. return Error::from_syscall("dup2"sv, -errno);
  228. return fd;
  229. }
  230. ErrorOr<String> ptsname(int fd)
  231. {
  232. auto* name = ::ptsname(fd);
  233. if (!name)
  234. return Error::from_syscall("ptsname"sv, -errno);
  235. return String(name);
  236. }
  237. ErrorOr<String> gethostname()
  238. {
  239. char hostname[256];
  240. int rc = ::gethostname(hostname, sizeof(hostname));
  241. if (rc < 0)
  242. return Error::from_syscall("gethostname"sv, -errno);
  243. return String(&hostname[0]);
  244. }
  245. ErrorOr<void> ioctl(int fd, unsigned request, ...)
  246. {
  247. va_list ap;
  248. va_start(ap, request);
  249. FlatPtr arg = va_arg(ap, FlatPtr);
  250. va_end(ap);
  251. if (::ioctl(fd, request, arg) < 0)
  252. return Error::from_syscall("ioctl"sv, -errno);
  253. return {};
  254. }
  255. ErrorOr<struct termios> tcgetattr(int fd)
  256. {
  257. struct termios ios = {};
  258. if (::tcgetattr(fd, &ios) < 0)
  259. return Error::from_syscall("tcgetattr"sv, -errno);
  260. return ios;
  261. }
  262. ErrorOr<void> tcsetattr(int fd, int optional_actions, struct termios const& ios)
  263. {
  264. if (::tcsetattr(fd, optional_actions, &ios) < 0)
  265. return Error::from_syscall("tcsetattr"sv, -errno);
  266. return {};
  267. }
  268. ErrorOr<void> chmod(StringView pathname, mode_t mode)
  269. {
  270. if (!pathname.characters_without_null_termination())
  271. return Error::from_syscall("chmod"sv, -EFAULT);
  272. #ifdef __serenity__
  273. int rc = syscall(SC_chmod, pathname.characters_without_null_termination(), pathname.length(), mode);
  274. HANDLE_SYSCALL_RETURN_VALUE("chmod"sv, rc, {});
  275. #else
  276. String path = pathname;
  277. if (::chmod(path.characters(), mode) < 0)
  278. return Error::from_syscall("chmod"sv, -errno);
  279. return {};
  280. #endif
  281. }
  282. ErrorOr<void> chown(StringView pathname, uid_t uid, gid_t gid)
  283. {
  284. if (!pathname.characters_without_null_termination())
  285. return Error::from_syscall("chown"sv, -EFAULT);
  286. #ifdef __serenity__
  287. Syscall::SC_chown_params params = { { pathname.characters_without_null_termination(), pathname.length() }, uid, gid };
  288. int rc = syscall(SC_chown, &params);
  289. HANDLE_SYSCALL_RETURN_VALUE("chown"sv, rc, {});
  290. #else
  291. String path = pathname;
  292. if (::chown(path.characters(), uid, gid) < 0)
  293. return Error::from_syscall("chown"sv, -errno);
  294. return {};
  295. #endif
  296. }
  297. ErrorOr<struct passwd> getpwnam(StringView name)
  298. {
  299. ::setpwent();
  300. if (errno)
  301. return Error::from_syscall("getpwnam"sv, -errno);
  302. while (auto* pw = ::getpwent()) {
  303. if (errno)
  304. return Error::from_syscall("getpwnam"sv, -errno);
  305. if (pw->pw_name == name)
  306. return *pw;
  307. }
  308. if (errno)
  309. return Error::from_syscall("getpwnam"sv, -errno);
  310. else
  311. return Error::from_string_literal("getpwnam: Unknown username"sv);
  312. }
  313. ErrorOr<struct group> getgrnam(StringView name)
  314. {
  315. ::setgrent();
  316. if (errno)
  317. return Error::from_syscall("getgrnam"sv, -errno);
  318. while (auto* gr = ::getgrent()) {
  319. if (errno)
  320. return Error::from_syscall("getgrnam"sv, -errno);
  321. if (gr->gr_name == name)
  322. return *gr;
  323. }
  324. if (errno)
  325. return Error::from_syscall("getgrnam"sv, -errno);
  326. else
  327. return Error::from_string_literal("getgrnam: Unknown username"sv);
  328. }
  329. ErrorOr<void> clock_settime(clockid_t clock_id, struct timespec* ts)
  330. {
  331. #ifdef __serenity__
  332. int rc = syscall(SC_clock_settime, clock_id, ts);
  333. HANDLE_SYSCALL_RETURN_VALUE("clocksettime"sv, rc, {});
  334. #else
  335. if (::clock_settime(clock_id, ts) < 0)
  336. return Error::from_syscall("clocksettime"sv, -errno);
  337. return {};
  338. #endif
  339. }
  340. ErrorOr<pid_t> posix_spawnp(StringView const path, posix_spawn_file_actions_t* const file_actions, posix_spawnattr_t* const attr, char* const arguments[], char* const envp[])
  341. {
  342. pid_t child_pid;
  343. if ((errno = ::posix_spawnp(&child_pid, path.to_string().characters(), file_actions, attr, arguments, envp)))
  344. return Error::from_syscall("posix_spawnp"sv, -errno);
  345. return child_pid;
  346. }
  347. ErrorOr<pid_t> waitpid(pid_t waitee, int* wstatus, int options)
  348. {
  349. pid_t pid = ::waitpid(waitee, wstatus, options);
  350. if (pid < 0)
  351. return Error::from_syscall("waitpid"sv, -errno);
  352. return pid;
  353. }
  354. ErrorOr<void> setuid(uid_t uid)
  355. {
  356. if (::setuid(uid) < 0)
  357. return Error::from_syscall("setuid"sv, -errno);
  358. return {};
  359. }
  360. ErrorOr<void> seteuid(uid_t uid)
  361. {
  362. if (::setuid(uid) < 0)
  363. return Error::from_syscall("seteuid"sv, -errno);
  364. return {};
  365. }
  366. ErrorOr<void> setgid(gid_t gid)
  367. {
  368. if (::setgid(gid) < 0)
  369. return Error::from_syscall("setgid"sv, -errno);
  370. return {};
  371. }
  372. ErrorOr<void> setegid(gid_t gid)
  373. {
  374. if (::setgid(gid) < 0)
  375. return Error::from_syscall("setegid"sv, -errno);
  376. return {};
  377. }
  378. ErrorOr<bool> isatty(int fd)
  379. {
  380. int rc = ::isatty(fd);
  381. if (rc < 0)
  382. return Error::from_syscall("isatty"sv, -errno);
  383. return rc == 1;
  384. }
  385. }