System.cpp 9.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345
  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. #endif
  78. ErrorOr<void> sigaction(int signal, struct sigaction const* action, struct sigaction* old_action)
  79. {
  80. if (::sigaction(signal, action, old_action) < 0)
  81. return Error::from_syscall("sigaction"sv, -errno);
  82. return {};
  83. }
  84. ErrorOr<struct stat> fstat(int fd)
  85. {
  86. struct stat st = {};
  87. if (::fstat(fd, &st) < 0)
  88. return Error::from_syscall("fstat"sv, -errno);
  89. return st;
  90. }
  91. ErrorOr<int> fcntl(int fd, int command, ...)
  92. {
  93. va_list ap;
  94. va_start(ap, command);
  95. u32 extra_arg = va_arg(ap, u32);
  96. int rc = ::fcntl(fd, command, extra_arg);
  97. va_end(ap);
  98. if (rc < 0)
  99. return Error::from_syscall("fcntl"sv, -errno);
  100. return rc;
  101. }
  102. 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)
  103. {
  104. #ifdef __serenity__
  105. Syscall::SC_mmap_params params { (uintptr_t)address, size, alignment, protection, flags, fd, offset, { name.characters_without_null_termination(), name.length() } };
  106. ptrdiff_t rc = syscall(SC_mmap, &params);
  107. if (rc < 0 && rc > -EMAXERRNO)
  108. return Error::from_syscall("mmap"sv, rc);
  109. return reinterpret_cast<void*>(rc);
  110. #else
  111. // NOTE: Regular POSIX mmap() doesn't support custom alignment requests.
  112. VERIFY(!alignment);
  113. auto* ptr = ::mmap(address, size, protection, flags, fd, offset);
  114. if (ptr == MAP_FAILED)
  115. return Error::from_syscall("mmap"sv, -errno);
  116. return ptr;
  117. #endif
  118. }
  119. ErrorOr<void> munmap(void* address, size_t size)
  120. {
  121. if (::munmap(address, size) < 0)
  122. return Error::from_syscall("munmap"sv, -errno);
  123. return {};
  124. }
  125. ErrorOr<int> open(StringView path, int options, ...)
  126. {
  127. if (!path.characters_without_null_termination())
  128. return Error::from_syscall("open"sv, -EFAULT);
  129. va_list ap;
  130. va_start(ap, options);
  131. auto mode = (mode_t)va_arg(ap, unsigned);
  132. va_end(ap);
  133. #ifdef __serenity__
  134. Syscall::SC_open_params params { AT_FDCWD, { path.characters_without_null_termination(), path.length() }, options, mode };
  135. int rc = syscall(SC_open, &params);
  136. HANDLE_SYSCALL_RETURN_VALUE("open"sv, rc, rc);
  137. #else
  138. // NOTE: We have to ensure that the path is null-terminated.
  139. String path_string = path;
  140. int rc = ::open(path_string.characters(), options, mode);
  141. if (rc < 0)
  142. return Error::from_syscall("open"sv, -errno);
  143. return rc;
  144. #endif
  145. }
  146. ErrorOr<void> close(int fd)
  147. {
  148. if (::close(fd) < 0)
  149. return Error::from_syscall("close"sv, -errno);
  150. return {};
  151. }
  152. ErrorOr<void> ftruncate(int fd, off_t length)
  153. {
  154. if (::ftruncate(fd, length) < 0)
  155. return Error::from_syscall("ftruncate"sv, -errno);
  156. return {};
  157. }
  158. ErrorOr<struct stat> stat(StringView path)
  159. {
  160. if (!path.characters_without_null_termination())
  161. return Error::from_syscall("stat"sv, -EFAULT);
  162. struct stat st = {};
  163. #ifdef __serenity__
  164. Syscall::SC_stat_params params { { path.characters_without_null_termination(), path.length() }, &st, AT_FDCWD, true };
  165. int rc = syscall(SC_stat, &params);
  166. HANDLE_SYSCALL_RETURN_VALUE("stat"sv, rc, st);
  167. #else
  168. String path_string = path;
  169. if (::stat(path_string.characters(), &st) < 0)
  170. return Error::from_syscall("stat"sv, -errno);
  171. return st;
  172. #endif
  173. }
  174. ErrorOr<struct stat> lstat(StringView path)
  175. {
  176. if (!path.characters_without_null_termination())
  177. return Error::from_syscall("lstat"sv, -EFAULT);
  178. struct stat st = {};
  179. #ifdef __serenity__
  180. Syscall::SC_stat_params params { { path.characters_without_null_termination(), path.length() }, &st, AT_FDCWD, false };
  181. int rc = syscall(SC_stat, &params);
  182. HANDLE_SYSCALL_RETURN_VALUE("lstat"sv, rc, st);
  183. #else
  184. String path_string = path;
  185. if (::stat(path_string.characters(), &st) < 0)
  186. return Error::from_syscall("lstat"sv, -errno);
  187. return st;
  188. #endif
  189. }
  190. ErrorOr<ssize_t> read(int fd, Bytes buffer)
  191. {
  192. ssize_t rc = ::read(fd, buffer.data(), buffer.size());
  193. if (rc < 0)
  194. return Error::from_syscall("read"sv, -errno);
  195. return rc;
  196. }
  197. ErrorOr<ssize_t> write(int fd, ReadonlyBytes buffer)
  198. {
  199. ssize_t rc = ::write(fd, buffer.data(), buffer.size());
  200. if (rc < 0)
  201. return Error::from_syscall("write"sv, -errno);
  202. return rc;
  203. }
  204. ErrorOr<void> kill(pid_t pid, int signal)
  205. {
  206. if (::kill(pid, signal) < 0)
  207. return Error::from_syscall("kill"sv, -errno);
  208. return {};
  209. }
  210. ErrorOr<int> dup(int source_fd)
  211. {
  212. int fd = ::dup(source_fd);
  213. if (fd < 0)
  214. return Error::from_syscall("dup"sv, -errno);
  215. return fd;
  216. }
  217. ErrorOr<int> dup2(int source_fd, int destination_fd)
  218. {
  219. int fd = ::dup2(source_fd, destination_fd);
  220. if (fd < 0)
  221. return Error::from_syscall("dup2"sv, -errno);
  222. return fd;
  223. }
  224. ErrorOr<String> ptsname(int fd)
  225. {
  226. auto* name = ::ptsname(fd);
  227. if (!name)
  228. return Error::from_syscall("ptsname"sv, -errno);
  229. return String(name);
  230. }
  231. ErrorOr<String> gethostname()
  232. {
  233. char hostname[256];
  234. int rc = ::gethostname(hostname, sizeof(hostname));
  235. if (rc < 0)
  236. return Error::from_syscall("gethostname"sv, -errno);
  237. return String(&hostname[0]);
  238. }
  239. ErrorOr<void> ioctl(int fd, unsigned request, ...)
  240. {
  241. va_list ap;
  242. va_start(ap, request);
  243. FlatPtr arg = va_arg(ap, FlatPtr);
  244. va_end(ap);
  245. if (::ioctl(fd, request, arg) < 0)
  246. return Error::from_syscall("ioctl"sv, -errno);
  247. return {};
  248. }
  249. ErrorOr<struct termios> tcgetattr(int fd)
  250. {
  251. struct termios ios = {};
  252. if (::tcgetattr(fd, &ios) < 0)
  253. return Error::from_syscall("tcgetattr"sv, -errno);
  254. return ios;
  255. }
  256. ErrorOr<void> tcsetattr(int fd, int optional_actions, struct termios const& ios)
  257. {
  258. if (::tcsetattr(fd, optional_actions, &ios) < 0)
  259. return Error::from_syscall("tcsetattr"sv, -errno);
  260. return {};
  261. }
  262. ErrorOr<void> chmod(StringView pathname, mode_t mode)
  263. {
  264. if (!pathname.characters_without_null_termination())
  265. return Error::from_syscall("chmod"sv, -EFAULT);
  266. #ifdef __serenity__
  267. int rc = syscall(SC_chmod, pathname.characters_without_null_termination(), pathname.length(), mode);
  268. HANDLE_SYSCALL_RETURN_VALUE("chmod"sv, rc, {});
  269. #else
  270. String path = pathname;
  271. if (::chmod(path.characters(), mode) < 0)
  272. return Error::from_syscall("chmod"sv, -errno);
  273. return {};
  274. #endif
  275. }
  276. ErrorOr<void> chown(StringView pathname, uid_t uid, gid_t gid)
  277. {
  278. if (!pathname.characters_without_null_termination())
  279. return Error::from_syscall("chown"sv, -EFAULT);
  280. #ifdef __serenity__
  281. Syscall::SC_chown_params params = { { pathname.characters_without_null_termination(), pathname.length() }, uid, gid };
  282. int rc = syscall(SC_chown, &params);
  283. HANDLE_SYSCALL_RETURN_VALUE("chown"sv, rc, {});
  284. #else
  285. String path = pathname;
  286. if (::chown(path.characters(), uid, gid) < 0)
  287. return Error::from_syscall("chown"sv, -errno);
  288. return {};
  289. #endif
  290. }
  291. ErrorOr<struct passwd> getpwnam(StringView name)
  292. {
  293. ::setpwent();
  294. if (errno)
  295. return Error::from_syscall("getpwnam"sv, -errno);
  296. while (auto* pw = ::getpwent()) {
  297. if (errno)
  298. return Error::from_syscall("getpwnam"sv, -errno);
  299. if (pw->pw_name == name)
  300. return *pw;
  301. }
  302. if (errno)
  303. return Error::from_syscall("getpwnam"sv, -errno);
  304. else
  305. return Error::from_string_literal("getpwnam: Unknown username"sv);
  306. }
  307. }