process.c 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260
  1. /*
  2. * Copyright (C) 2002 - 2007 Jeff Dike (jdike@{addtoit,linux.intel}.com)
  3. * Licensed under the GPL
  4. */
  5. #include <stdio.h>
  6. #include <unistd.h>
  7. #include <errno.h>
  8. #include <signal.h>
  9. #include <fcntl.h>
  10. #include <sys/mman.h>
  11. #include <sys/ptrace.h>
  12. #include <sys/wait.h>
  13. #include <asm/unistd.h>
  14. #include "init.h"
  15. #include "longjmp.h"
  16. #include "os.h"
  17. #include "skas_ptrace.h"
  18. #define ARBITRARY_ADDR -1
  19. #define FAILURE_PID -1
  20. #define STAT_PATH_LEN sizeof("/proc/#######/stat\0")
  21. #define COMM_SCANF "%*[^)])"
  22. unsigned long os_process_pc(int pid)
  23. {
  24. char proc_stat[STAT_PATH_LEN], buf[256];
  25. unsigned long pc = ARBITRARY_ADDR;
  26. int fd, err;
  27. sprintf(proc_stat, "/proc/%d/stat", pid);
  28. fd = open(proc_stat, O_RDONLY, 0);
  29. if (fd < 0) {
  30. printk(UM_KERN_ERR "os_process_pc - couldn't open '%s', "
  31. "errno = %d\n", proc_stat, errno);
  32. goto out;
  33. }
  34. CATCH_EINTR(err = read(fd, buf, sizeof(buf)));
  35. if (err < 0) {
  36. printk(UM_KERN_ERR "os_process_pc - couldn't read '%s', "
  37. "err = %d\n", proc_stat, errno);
  38. goto out_close;
  39. }
  40. os_close_file(fd);
  41. pc = ARBITRARY_ADDR;
  42. if (sscanf(buf, "%*d " COMM_SCANF " %*c %*d %*d %*d %*d %*d %*d %*d "
  43. "%*d %*d %*d %*d %*d %*d %*d %*d %*d %*d %*d %*d %*d %*d "
  44. "%*d %*d %*d %*d %*d %lu", &pc) != 1)
  45. printk(UM_KERN_ERR "os_process_pc - couldn't find pc in '%s'\n",
  46. buf);
  47. out_close:
  48. close(fd);
  49. out:
  50. return pc;
  51. }
  52. int os_process_parent(int pid)
  53. {
  54. char stat[STAT_PATH_LEN];
  55. char data[256];
  56. int parent = FAILURE_PID, n, fd;
  57. if (pid == -1)
  58. return parent;
  59. snprintf(stat, sizeof(stat), "/proc/%d/stat", pid);
  60. fd = open(stat, O_RDONLY, 0);
  61. if (fd < 0) {
  62. printk(UM_KERN_ERR "Couldn't open '%s', errno = %d\n", stat,
  63. errno);
  64. return parent;
  65. }
  66. CATCH_EINTR(n = read(fd, data, sizeof(data)));
  67. close(fd);
  68. if (n < 0) {
  69. printk(UM_KERN_ERR "Couldn't read '%s', errno = %d\n", stat,
  70. errno);
  71. return parent;
  72. }
  73. parent = FAILURE_PID;
  74. n = sscanf(data, "%*d " COMM_SCANF " %*c %d", &parent);
  75. if (n != 1)
  76. printk(UM_KERN_ERR "Failed to scan '%s'\n", data);
  77. return parent;
  78. }
  79. void os_stop_process(int pid)
  80. {
  81. kill(pid, SIGSTOP);
  82. }
  83. void os_kill_process(int pid, int reap_child)
  84. {
  85. kill(pid, SIGKILL);
  86. if (reap_child)
  87. CATCH_EINTR(waitpid(pid, NULL, __WALL));
  88. }
  89. /* This is here uniquely to have access to the userspace errno, i.e. the one
  90. * used by ptrace in case of error.
  91. */
  92. long os_ptrace_ldt(long pid, long addr, long data)
  93. {
  94. int ret;
  95. ret = ptrace(PTRACE_LDT, pid, addr, data);
  96. if (ret < 0)
  97. return -errno;
  98. return ret;
  99. }
  100. /* Kill off a ptraced child by all means available. kill it normally first,
  101. * then PTRACE_KILL it, then PTRACE_CONT it in case it's in a run state from
  102. * which it can't exit directly.
  103. */
  104. void os_kill_ptraced_process(int pid, int reap_child)
  105. {
  106. kill(pid, SIGKILL);
  107. ptrace(PTRACE_KILL, pid);
  108. ptrace(PTRACE_CONT, pid);
  109. if (reap_child)
  110. CATCH_EINTR(waitpid(pid, NULL, __WALL));
  111. }
  112. /* Don't use the glibc version, which caches the result in TLS. It misses some
  113. * syscalls, and also breaks with clone(), which does not unshare the TLS.
  114. */
  115. int os_getpid(void)
  116. {
  117. return syscall(__NR_getpid);
  118. }
  119. int os_getpgrp(void)
  120. {
  121. return getpgrp();
  122. }
  123. int os_map_memory(void *virt, int fd, unsigned long long off, unsigned long len,
  124. int r, int w, int x)
  125. {
  126. void *loc;
  127. int prot;
  128. prot = (r ? PROT_READ : 0) | (w ? PROT_WRITE : 0) |
  129. (x ? PROT_EXEC : 0);
  130. loc = mmap64((void *) virt, len, prot, MAP_SHARED | MAP_FIXED,
  131. fd, off);
  132. if (loc == MAP_FAILED)
  133. return -errno;
  134. return 0;
  135. }
  136. int os_protect_memory(void *addr, unsigned long len, int r, int w, int x)
  137. {
  138. int prot = ((r ? PROT_READ : 0) | (w ? PROT_WRITE : 0) |
  139. (x ? PROT_EXEC : 0));
  140. if (mprotect(addr, len, prot) < 0)
  141. return -errno;
  142. return 0;
  143. }
  144. int os_unmap_memory(void *addr, int len)
  145. {
  146. int err;
  147. err = munmap(addr, len);
  148. if (err < 0)
  149. return -errno;
  150. return 0;
  151. }
  152. #ifndef MADV_REMOVE
  153. #define MADV_REMOVE KERNEL_MADV_REMOVE
  154. #endif
  155. int os_drop_memory(void *addr, int length)
  156. {
  157. int err;
  158. err = madvise(addr, length, MADV_REMOVE);
  159. if (err < 0)
  160. err = -errno;
  161. return err;
  162. }
  163. int __init can_drop_memory(void)
  164. {
  165. void *addr;
  166. int fd, ok = 0;
  167. printk(UM_KERN_INFO "Checking host MADV_REMOVE support...");
  168. fd = create_mem_file(UM_KERN_PAGE_SIZE);
  169. if (fd < 0) {
  170. printk(UM_KERN_ERR "Creating test memory file failed, "
  171. "err = %d\n", -fd);
  172. goto out;
  173. }
  174. addr = mmap64(NULL, UM_KERN_PAGE_SIZE, PROT_READ | PROT_WRITE,
  175. MAP_SHARED, fd, 0);
  176. if (addr == MAP_FAILED) {
  177. printk(UM_KERN_ERR "Mapping test memory file failed, "
  178. "err = %d\n", -errno);
  179. goto out_close;
  180. }
  181. if (madvise(addr, UM_KERN_PAGE_SIZE, MADV_REMOVE) != 0) {
  182. printk(UM_KERN_ERR "MADV_REMOVE failed, err = %d\n", -errno);
  183. goto out_unmap;
  184. }
  185. printk(UM_KERN_CONT "OK\n");
  186. ok = 1;
  187. out_unmap:
  188. munmap(addr, UM_KERN_PAGE_SIZE);
  189. out_close:
  190. close(fd);
  191. out:
  192. return ok;
  193. }
  194. void init_new_thread_signals(void)
  195. {
  196. set_handler(SIGSEGV);
  197. set_handler(SIGTRAP);
  198. set_handler(SIGFPE);
  199. set_handler(SIGILL);
  200. set_handler(SIGBUS);
  201. signal(SIGHUP, SIG_IGN);
  202. set_handler(SIGIO);
  203. signal(SIGWINCH, SIG_IGN);
  204. signal(SIGTERM, SIG_DFL);
  205. }
  206. int run_kernel_thread(int (*fn)(void *), void *arg, jmp_buf **jmp_ptr)
  207. {
  208. jmp_buf buf;
  209. int n;
  210. *jmp_ptr = &buf;
  211. n = UML_SETJMP(&buf);
  212. if (n != 0)
  213. return n;
  214. (*fn)(arg);
  215. return 0;
  216. }