fork.c 40 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653
  1. /*
  2. * linux/kernel/fork.c
  3. *
  4. * Copyright (C) 1991, 1992 Linus Torvalds
  5. */
  6. /*
  7. * 'fork.c' contains the help-routines for the 'fork' system call
  8. * (see also entry.S and others).
  9. * Fork is rather simple, once you get the hang of it, but the memory
  10. * management can be a bitch. See 'mm/memory.c': 'copy_page_range()'
  11. */
  12. #include <linux/slab.h>
  13. #include <linux/init.h>
  14. #include <linux/unistd.h>
  15. #include <linux/smp_lock.h>
  16. #include <linux/module.h>
  17. #include <linux/vmalloc.h>
  18. #include <linux/completion.h>
  19. #include <linux/namespace.h>
  20. #include <linux/personality.h>
  21. #include <linux/mempolicy.h>
  22. #include <linux/sem.h>
  23. #include <linux/file.h>
  24. #include <linux/key.h>
  25. #include <linux/binfmts.h>
  26. #include <linux/mman.h>
  27. #include <linux/fs.h>
  28. #include <linux/capability.h>
  29. #include <linux/cpu.h>
  30. #include <linux/cpuset.h>
  31. #include <linux/security.h>
  32. #include <linux/swap.h>
  33. #include <linux/syscalls.h>
  34. #include <linux/jiffies.h>
  35. #include <linux/futex.h>
  36. #include <linux/rcupdate.h>
  37. #include <linux/ptrace.h>
  38. #include <linux/mount.h>
  39. #include <linux/audit.h>
  40. #include <linux/profile.h>
  41. #include <linux/rmap.h>
  42. #include <linux/acct.h>
  43. #include <linux/cn_proc.h>
  44. #include <asm/pgtable.h>
  45. #include <asm/pgalloc.h>
  46. #include <asm/uaccess.h>
  47. #include <asm/mmu_context.h>
  48. #include <asm/cacheflush.h>
  49. #include <asm/tlbflush.h>
  50. /*
  51. * Protected counters by write_lock_irq(&tasklist_lock)
  52. */
  53. unsigned long total_forks; /* Handle normal Linux uptimes. */
  54. int nr_threads; /* The idle threads do not count.. */
  55. int max_threads; /* tunable limit on nr_threads */
  56. DEFINE_PER_CPU(unsigned long, process_counts) = 0;
  57. __cacheline_aligned DEFINE_RWLOCK(tasklist_lock); /* outer */
  58. EXPORT_SYMBOL(tasklist_lock);
  59. int nr_processes(void)
  60. {
  61. int cpu;
  62. int total = 0;
  63. for_each_online_cpu(cpu)
  64. total += per_cpu(process_counts, cpu);
  65. return total;
  66. }
  67. #ifndef __HAVE_ARCH_TASK_STRUCT_ALLOCATOR
  68. # define alloc_task_struct() kmem_cache_alloc(task_struct_cachep, GFP_KERNEL)
  69. # define free_task_struct(tsk) kmem_cache_free(task_struct_cachep, (tsk))
  70. static kmem_cache_t *task_struct_cachep;
  71. #endif
  72. /* SLAB cache for signal_struct structures (tsk->signal) */
  73. static kmem_cache_t *signal_cachep;
  74. /* SLAB cache for sighand_struct structures (tsk->sighand) */
  75. kmem_cache_t *sighand_cachep;
  76. /* SLAB cache for files_struct structures (tsk->files) */
  77. kmem_cache_t *files_cachep;
  78. /* SLAB cache for fs_struct structures (tsk->fs) */
  79. kmem_cache_t *fs_cachep;
  80. /* SLAB cache for vm_area_struct structures */
  81. kmem_cache_t *vm_area_cachep;
  82. /* SLAB cache for mm_struct structures (tsk->mm) */
  83. static kmem_cache_t *mm_cachep;
  84. void free_task(struct task_struct *tsk)
  85. {
  86. free_thread_info(tsk->thread_info);
  87. rt_mutex_debug_task_free(tsk);
  88. free_task_struct(tsk);
  89. }
  90. EXPORT_SYMBOL(free_task);
  91. void __put_task_struct(struct task_struct *tsk)
  92. {
  93. WARN_ON(!(tsk->exit_state & (EXIT_DEAD | EXIT_ZOMBIE)));
  94. WARN_ON(atomic_read(&tsk->usage));
  95. WARN_ON(tsk == current);
  96. security_task_free(tsk);
  97. free_uid(tsk->user);
  98. put_group_info(tsk->group_info);
  99. if (!profile_handoff_task(tsk))
  100. free_task(tsk);
  101. }
  102. void __init fork_init(unsigned long mempages)
  103. {
  104. #ifndef __HAVE_ARCH_TASK_STRUCT_ALLOCATOR
  105. #ifndef ARCH_MIN_TASKALIGN
  106. #define ARCH_MIN_TASKALIGN L1_CACHE_BYTES
  107. #endif
  108. /* create a slab on which task_structs can be allocated */
  109. task_struct_cachep =
  110. kmem_cache_create("task_struct", sizeof(struct task_struct),
  111. ARCH_MIN_TASKALIGN, SLAB_PANIC, NULL, NULL);
  112. #endif
  113. /*
  114. * The default maximum number of threads is set to a safe
  115. * value: the thread structures can take up at most half
  116. * of memory.
  117. */
  118. max_threads = mempages / (8 * THREAD_SIZE / PAGE_SIZE);
  119. /*
  120. * we need to allow at least 20 threads to boot a system
  121. */
  122. if(max_threads < 20)
  123. max_threads = 20;
  124. init_task.signal->rlim[RLIMIT_NPROC].rlim_cur = max_threads/2;
  125. init_task.signal->rlim[RLIMIT_NPROC].rlim_max = max_threads/2;
  126. init_task.signal->rlim[RLIMIT_SIGPENDING] =
  127. init_task.signal->rlim[RLIMIT_NPROC];
  128. }
  129. static struct task_struct *dup_task_struct(struct task_struct *orig)
  130. {
  131. struct task_struct *tsk;
  132. struct thread_info *ti;
  133. prepare_to_copy(orig);
  134. tsk = alloc_task_struct();
  135. if (!tsk)
  136. return NULL;
  137. ti = alloc_thread_info(tsk);
  138. if (!ti) {
  139. free_task_struct(tsk);
  140. return NULL;
  141. }
  142. *tsk = *orig;
  143. tsk->thread_info = ti;
  144. setup_thread_stack(tsk, orig);
  145. /* One for us, one for whoever does the "release_task()" (usually parent) */
  146. atomic_set(&tsk->usage,2);
  147. atomic_set(&tsk->fs_excl, 0);
  148. tsk->btrace_seq = 0;
  149. tsk->splice_pipe = NULL;
  150. return tsk;
  151. }
  152. #ifdef CONFIG_MMU
  153. static inline int dup_mmap(struct mm_struct *mm, struct mm_struct *oldmm)
  154. {
  155. struct vm_area_struct *mpnt, *tmp, **pprev;
  156. struct rb_node **rb_link, *rb_parent;
  157. int retval;
  158. unsigned long charge;
  159. struct mempolicy *pol;
  160. down_write(&oldmm->mmap_sem);
  161. flush_cache_mm(oldmm);
  162. down_write(&mm->mmap_sem);
  163. mm->locked_vm = 0;
  164. mm->mmap = NULL;
  165. mm->mmap_cache = NULL;
  166. mm->free_area_cache = oldmm->mmap_base;
  167. mm->cached_hole_size = ~0UL;
  168. mm->map_count = 0;
  169. cpus_clear(mm->cpu_vm_mask);
  170. mm->mm_rb = RB_ROOT;
  171. rb_link = &mm->mm_rb.rb_node;
  172. rb_parent = NULL;
  173. pprev = &mm->mmap;
  174. for (mpnt = oldmm->mmap; mpnt; mpnt = mpnt->vm_next) {
  175. struct file *file;
  176. if (mpnt->vm_flags & VM_DONTCOPY) {
  177. long pages = vma_pages(mpnt);
  178. mm->total_vm -= pages;
  179. vm_stat_account(mm, mpnt->vm_flags, mpnt->vm_file,
  180. -pages);
  181. continue;
  182. }
  183. charge = 0;
  184. if (mpnt->vm_flags & VM_ACCOUNT) {
  185. unsigned int len = (mpnt->vm_end - mpnt->vm_start) >> PAGE_SHIFT;
  186. if (security_vm_enough_memory(len))
  187. goto fail_nomem;
  188. charge = len;
  189. }
  190. tmp = kmem_cache_alloc(vm_area_cachep, SLAB_KERNEL);
  191. if (!tmp)
  192. goto fail_nomem;
  193. *tmp = *mpnt;
  194. pol = mpol_copy(vma_policy(mpnt));
  195. retval = PTR_ERR(pol);
  196. if (IS_ERR(pol))
  197. goto fail_nomem_policy;
  198. vma_set_policy(tmp, pol);
  199. tmp->vm_flags &= ~VM_LOCKED;
  200. tmp->vm_mm = mm;
  201. tmp->vm_next = NULL;
  202. anon_vma_link(tmp);
  203. file = tmp->vm_file;
  204. if (file) {
  205. struct inode *inode = file->f_dentry->d_inode;
  206. get_file(file);
  207. if (tmp->vm_flags & VM_DENYWRITE)
  208. atomic_dec(&inode->i_writecount);
  209. /* insert tmp into the share list, just after mpnt */
  210. spin_lock(&file->f_mapping->i_mmap_lock);
  211. tmp->vm_truncate_count = mpnt->vm_truncate_count;
  212. flush_dcache_mmap_lock(file->f_mapping);
  213. vma_prio_tree_add(tmp, mpnt);
  214. flush_dcache_mmap_unlock(file->f_mapping);
  215. spin_unlock(&file->f_mapping->i_mmap_lock);
  216. }
  217. /*
  218. * Link in the new vma and copy the page table entries.
  219. */
  220. *pprev = tmp;
  221. pprev = &tmp->vm_next;
  222. __vma_link_rb(mm, tmp, rb_link, rb_parent);
  223. rb_link = &tmp->vm_rb.rb_right;
  224. rb_parent = &tmp->vm_rb;
  225. mm->map_count++;
  226. retval = copy_page_range(mm, oldmm, mpnt);
  227. if (tmp->vm_ops && tmp->vm_ops->open)
  228. tmp->vm_ops->open(tmp);
  229. if (retval)
  230. goto out;
  231. }
  232. retval = 0;
  233. out:
  234. up_write(&mm->mmap_sem);
  235. flush_tlb_mm(oldmm);
  236. up_write(&oldmm->mmap_sem);
  237. return retval;
  238. fail_nomem_policy:
  239. kmem_cache_free(vm_area_cachep, tmp);
  240. fail_nomem:
  241. retval = -ENOMEM;
  242. vm_unacct_memory(charge);
  243. goto out;
  244. }
  245. static inline int mm_alloc_pgd(struct mm_struct * mm)
  246. {
  247. mm->pgd = pgd_alloc(mm);
  248. if (unlikely(!mm->pgd))
  249. return -ENOMEM;
  250. return 0;
  251. }
  252. static inline void mm_free_pgd(struct mm_struct * mm)
  253. {
  254. pgd_free(mm->pgd);
  255. }
  256. #else
  257. #define dup_mmap(mm, oldmm) (0)
  258. #define mm_alloc_pgd(mm) (0)
  259. #define mm_free_pgd(mm)
  260. #endif /* CONFIG_MMU */
  261. __cacheline_aligned_in_smp DEFINE_SPINLOCK(mmlist_lock);
  262. #define allocate_mm() (kmem_cache_alloc(mm_cachep, SLAB_KERNEL))
  263. #define free_mm(mm) (kmem_cache_free(mm_cachep, (mm)))
  264. #include <linux/init_task.h>
  265. static struct mm_struct * mm_init(struct mm_struct * mm)
  266. {
  267. atomic_set(&mm->mm_users, 1);
  268. atomic_set(&mm->mm_count, 1);
  269. init_rwsem(&mm->mmap_sem);
  270. INIT_LIST_HEAD(&mm->mmlist);
  271. mm->core_waiters = 0;
  272. mm->nr_ptes = 0;
  273. set_mm_counter(mm, file_rss, 0);
  274. set_mm_counter(mm, anon_rss, 0);
  275. spin_lock_init(&mm->page_table_lock);
  276. rwlock_init(&mm->ioctx_list_lock);
  277. mm->ioctx_list = NULL;
  278. mm->free_area_cache = TASK_UNMAPPED_BASE;
  279. mm->cached_hole_size = ~0UL;
  280. if (likely(!mm_alloc_pgd(mm))) {
  281. mm->def_flags = 0;
  282. return mm;
  283. }
  284. free_mm(mm);
  285. return NULL;
  286. }
  287. /*
  288. * Allocate and initialize an mm_struct.
  289. */
  290. struct mm_struct * mm_alloc(void)
  291. {
  292. struct mm_struct * mm;
  293. mm = allocate_mm();
  294. if (mm) {
  295. memset(mm, 0, sizeof(*mm));
  296. mm = mm_init(mm);
  297. }
  298. return mm;
  299. }
  300. /*
  301. * Called when the last reference to the mm
  302. * is dropped: either by a lazy thread or by
  303. * mmput. Free the page directory and the mm.
  304. */
  305. void fastcall __mmdrop(struct mm_struct *mm)
  306. {
  307. BUG_ON(mm == &init_mm);
  308. mm_free_pgd(mm);
  309. destroy_context(mm);
  310. free_mm(mm);
  311. }
  312. /*
  313. * Decrement the use count and release all resources for an mm.
  314. */
  315. void mmput(struct mm_struct *mm)
  316. {
  317. might_sleep();
  318. if (atomic_dec_and_test(&mm->mm_users)) {
  319. exit_aio(mm);
  320. exit_mmap(mm);
  321. if (!list_empty(&mm->mmlist)) {
  322. spin_lock(&mmlist_lock);
  323. list_del(&mm->mmlist);
  324. spin_unlock(&mmlist_lock);
  325. }
  326. put_swap_token(mm);
  327. mmdrop(mm);
  328. }
  329. }
  330. EXPORT_SYMBOL_GPL(mmput);
  331. /**
  332. * get_task_mm - acquire a reference to the task's mm
  333. *
  334. * Returns %NULL if the task has no mm. Checks PF_BORROWED_MM (meaning
  335. * this kernel workthread has transiently adopted a user mm with use_mm,
  336. * to do its AIO) is not set and if so returns a reference to it, after
  337. * bumping up the use count. User must release the mm via mmput()
  338. * after use. Typically used by /proc and ptrace.
  339. */
  340. struct mm_struct *get_task_mm(struct task_struct *task)
  341. {
  342. struct mm_struct *mm;
  343. task_lock(task);
  344. mm = task->mm;
  345. if (mm) {
  346. if (task->flags & PF_BORROWED_MM)
  347. mm = NULL;
  348. else
  349. atomic_inc(&mm->mm_users);
  350. }
  351. task_unlock(task);
  352. return mm;
  353. }
  354. EXPORT_SYMBOL_GPL(get_task_mm);
  355. /* Please note the differences between mmput and mm_release.
  356. * mmput is called whenever we stop holding onto a mm_struct,
  357. * error success whatever.
  358. *
  359. * mm_release is called after a mm_struct has been removed
  360. * from the current process.
  361. *
  362. * This difference is important for error handling, when we
  363. * only half set up a mm_struct for a new process and need to restore
  364. * the old one. Because we mmput the new mm_struct before
  365. * restoring the old one. . .
  366. * Eric Biederman 10 January 1998
  367. */
  368. void mm_release(struct task_struct *tsk, struct mm_struct *mm)
  369. {
  370. struct completion *vfork_done = tsk->vfork_done;
  371. /* Get rid of any cached register state */
  372. deactivate_mm(tsk, mm);
  373. /* notify parent sleeping on vfork() */
  374. if (vfork_done) {
  375. tsk->vfork_done = NULL;
  376. complete(vfork_done);
  377. }
  378. if (tsk->clear_child_tid && atomic_read(&mm->mm_users) > 1) {
  379. u32 __user * tidptr = tsk->clear_child_tid;
  380. tsk->clear_child_tid = NULL;
  381. /*
  382. * We don't check the error code - if userspace has
  383. * not set up a proper pointer then tough luck.
  384. */
  385. put_user(0, tidptr);
  386. sys_futex(tidptr, FUTEX_WAKE, 1, NULL, NULL, 0);
  387. }
  388. }
  389. /*
  390. * Allocate a new mm structure and copy contents from the
  391. * mm structure of the passed in task structure.
  392. */
  393. static struct mm_struct *dup_mm(struct task_struct *tsk)
  394. {
  395. struct mm_struct *mm, *oldmm = current->mm;
  396. int err;
  397. if (!oldmm)
  398. return NULL;
  399. mm = allocate_mm();
  400. if (!mm)
  401. goto fail_nomem;
  402. memcpy(mm, oldmm, sizeof(*mm));
  403. if (!mm_init(mm))
  404. goto fail_nomem;
  405. if (init_new_context(tsk, mm))
  406. goto fail_nocontext;
  407. err = dup_mmap(mm, oldmm);
  408. if (err)
  409. goto free_pt;
  410. mm->hiwater_rss = get_mm_rss(mm);
  411. mm->hiwater_vm = mm->total_vm;
  412. return mm;
  413. free_pt:
  414. mmput(mm);
  415. fail_nomem:
  416. return NULL;
  417. fail_nocontext:
  418. /*
  419. * If init_new_context() failed, we cannot use mmput() to free the mm
  420. * because it calls destroy_context()
  421. */
  422. mm_free_pgd(mm);
  423. free_mm(mm);
  424. return NULL;
  425. }
  426. static int copy_mm(unsigned long clone_flags, struct task_struct * tsk)
  427. {
  428. struct mm_struct * mm, *oldmm;
  429. int retval;
  430. tsk->min_flt = tsk->maj_flt = 0;
  431. tsk->nvcsw = tsk->nivcsw = 0;
  432. tsk->mm = NULL;
  433. tsk->active_mm = NULL;
  434. /*
  435. * Are we cloning a kernel thread?
  436. *
  437. * We need to steal a active VM for that..
  438. */
  439. oldmm = current->mm;
  440. if (!oldmm)
  441. return 0;
  442. if (clone_flags & CLONE_VM) {
  443. atomic_inc(&oldmm->mm_users);
  444. mm = oldmm;
  445. goto good_mm;
  446. }
  447. retval = -ENOMEM;
  448. mm = dup_mm(tsk);
  449. if (!mm)
  450. goto fail_nomem;
  451. good_mm:
  452. tsk->mm = mm;
  453. tsk->active_mm = mm;
  454. return 0;
  455. fail_nomem:
  456. return retval;
  457. }
  458. static inline struct fs_struct *__copy_fs_struct(struct fs_struct *old)
  459. {
  460. struct fs_struct *fs = kmem_cache_alloc(fs_cachep, GFP_KERNEL);
  461. /* We don't need to lock fs - think why ;-) */
  462. if (fs) {
  463. atomic_set(&fs->count, 1);
  464. rwlock_init(&fs->lock);
  465. fs->umask = old->umask;
  466. read_lock(&old->lock);
  467. fs->rootmnt = mntget(old->rootmnt);
  468. fs->root = dget(old->root);
  469. fs->pwdmnt = mntget(old->pwdmnt);
  470. fs->pwd = dget(old->pwd);
  471. if (old->altroot) {
  472. fs->altrootmnt = mntget(old->altrootmnt);
  473. fs->altroot = dget(old->altroot);
  474. } else {
  475. fs->altrootmnt = NULL;
  476. fs->altroot = NULL;
  477. }
  478. read_unlock(&old->lock);
  479. }
  480. return fs;
  481. }
  482. struct fs_struct *copy_fs_struct(struct fs_struct *old)
  483. {
  484. return __copy_fs_struct(old);
  485. }
  486. EXPORT_SYMBOL_GPL(copy_fs_struct);
  487. static inline int copy_fs(unsigned long clone_flags, struct task_struct * tsk)
  488. {
  489. if (clone_flags & CLONE_FS) {
  490. atomic_inc(&current->fs->count);
  491. return 0;
  492. }
  493. tsk->fs = __copy_fs_struct(current->fs);
  494. if (!tsk->fs)
  495. return -ENOMEM;
  496. return 0;
  497. }
  498. static int count_open_files(struct fdtable *fdt)
  499. {
  500. int size = fdt->max_fdset;
  501. int i;
  502. /* Find the last open fd */
  503. for (i = size/(8*sizeof(long)); i > 0; ) {
  504. if (fdt->open_fds->fds_bits[--i])
  505. break;
  506. }
  507. i = (i+1) * 8 * sizeof(long);
  508. return i;
  509. }
  510. static struct files_struct *alloc_files(void)
  511. {
  512. struct files_struct *newf;
  513. struct fdtable *fdt;
  514. newf = kmem_cache_alloc(files_cachep, SLAB_KERNEL);
  515. if (!newf)
  516. goto out;
  517. atomic_set(&newf->count, 1);
  518. spin_lock_init(&newf->file_lock);
  519. newf->next_fd = 0;
  520. fdt = &newf->fdtab;
  521. fdt->max_fds = NR_OPEN_DEFAULT;
  522. fdt->max_fdset = EMBEDDED_FD_SET_SIZE;
  523. fdt->close_on_exec = (fd_set *)&newf->close_on_exec_init;
  524. fdt->open_fds = (fd_set *)&newf->open_fds_init;
  525. fdt->fd = &newf->fd_array[0];
  526. INIT_RCU_HEAD(&fdt->rcu);
  527. fdt->free_files = NULL;
  528. fdt->next = NULL;
  529. rcu_assign_pointer(newf->fdt, fdt);
  530. out:
  531. return newf;
  532. }
  533. /*
  534. * Allocate a new files structure and copy contents from the
  535. * passed in files structure.
  536. * errorp will be valid only when the returned files_struct is NULL.
  537. */
  538. static struct files_struct *dup_fd(struct files_struct *oldf, int *errorp)
  539. {
  540. struct files_struct *newf;
  541. struct file **old_fds, **new_fds;
  542. int open_files, size, i, expand;
  543. struct fdtable *old_fdt, *new_fdt;
  544. *errorp = -ENOMEM;
  545. newf = alloc_files();
  546. if (!newf)
  547. goto out;
  548. spin_lock(&oldf->file_lock);
  549. old_fdt = files_fdtable(oldf);
  550. new_fdt = files_fdtable(newf);
  551. size = old_fdt->max_fdset;
  552. open_files = count_open_files(old_fdt);
  553. expand = 0;
  554. /*
  555. * Check whether we need to allocate a larger fd array or fd set.
  556. * Note: we're not a clone task, so the open count won't change.
  557. */
  558. if (open_files > new_fdt->max_fdset) {
  559. new_fdt->max_fdset = 0;
  560. expand = 1;
  561. }
  562. if (open_files > new_fdt->max_fds) {
  563. new_fdt->max_fds = 0;
  564. expand = 1;
  565. }
  566. /* if the old fdset gets grown now, we'll only copy up to "size" fds */
  567. if (expand) {
  568. spin_unlock(&oldf->file_lock);
  569. spin_lock(&newf->file_lock);
  570. *errorp = expand_files(newf, open_files-1);
  571. spin_unlock(&newf->file_lock);
  572. if (*errorp < 0)
  573. goto out_release;
  574. new_fdt = files_fdtable(newf);
  575. /*
  576. * Reacquire the oldf lock and a pointer to its fd table
  577. * who knows it may have a new bigger fd table. We need
  578. * the latest pointer.
  579. */
  580. spin_lock(&oldf->file_lock);
  581. old_fdt = files_fdtable(oldf);
  582. }
  583. old_fds = old_fdt->fd;
  584. new_fds = new_fdt->fd;
  585. memcpy(new_fdt->open_fds->fds_bits, old_fdt->open_fds->fds_bits, open_files/8);
  586. memcpy(new_fdt->close_on_exec->fds_bits, old_fdt->close_on_exec->fds_bits, open_files/8);
  587. for (i = open_files; i != 0; i--) {
  588. struct file *f = *old_fds++;
  589. if (f) {
  590. get_file(f);
  591. } else {
  592. /*
  593. * The fd may be claimed in the fd bitmap but not yet
  594. * instantiated in the files array if a sibling thread
  595. * is partway through open(). So make sure that this
  596. * fd is available to the new process.
  597. */
  598. FD_CLR(open_files - i, new_fdt->open_fds);
  599. }
  600. rcu_assign_pointer(*new_fds++, f);
  601. }
  602. spin_unlock(&oldf->file_lock);
  603. /* compute the remainder to be cleared */
  604. size = (new_fdt->max_fds - open_files) * sizeof(struct file *);
  605. /* This is long word aligned thus could use a optimized version */
  606. memset(new_fds, 0, size);
  607. if (new_fdt->max_fdset > open_files) {
  608. int left = (new_fdt->max_fdset-open_files)/8;
  609. int start = open_files / (8 * sizeof(unsigned long));
  610. memset(&new_fdt->open_fds->fds_bits[start], 0, left);
  611. memset(&new_fdt->close_on_exec->fds_bits[start], 0, left);
  612. }
  613. out:
  614. return newf;
  615. out_release:
  616. free_fdset (new_fdt->close_on_exec, new_fdt->max_fdset);
  617. free_fdset (new_fdt->open_fds, new_fdt->max_fdset);
  618. free_fd_array(new_fdt->fd, new_fdt->max_fds);
  619. kmem_cache_free(files_cachep, newf);
  620. return NULL;
  621. }
  622. static int copy_files(unsigned long clone_flags, struct task_struct * tsk)
  623. {
  624. struct files_struct *oldf, *newf;
  625. int error = 0;
  626. /*
  627. * A background process may not have any files ...
  628. */
  629. oldf = current->files;
  630. if (!oldf)
  631. goto out;
  632. if (clone_flags & CLONE_FILES) {
  633. atomic_inc(&oldf->count);
  634. goto out;
  635. }
  636. /*
  637. * Note: we may be using current for both targets (See exec.c)
  638. * This works because we cache current->files (old) as oldf. Don't
  639. * break this.
  640. */
  641. tsk->files = NULL;
  642. newf = dup_fd(oldf, &error);
  643. if (!newf)
  644. goto out;
  645. tsk->files = newf;
  646. error = 0;
  647. out:
  648. return error;
  649. }
  650. /*
  651. * Helper to unshare the files of the current task.
  652. * We don't want to expose copy_files internals to
  653. * the exec layer of the kernel.
  654. */
  655. int unshare_files(void)
  656. {
  657. struct files_struct *files = current->files;
  658. int rc;
  659. BUG_ON(!files);
  660. /* This can race but the race causes us to copy when we don't
  661. need to and drop the copy */
  662. if(atomic_read(&files->count) == 1)
  663. {
  664. atomic_inc(&files->count);
  665. return 0;
  666. }
  667. rc = copy_files(0, current);
  668. if(rc)
  669. current->files = files;
  670. return rc;
  671. }
  672. EXPORT_SYMBOL(unshare_files);
  673. static inline int copy_sighand(unsigned long clone_flags, struct task_struct * tsk)
  674. {
  675. struct sighand_struct *sig;
  676. if (clone_flags & (CLONE_SIGHAND | CLONE_THREAD)) {
  677. atomic_inc(&current->sighand->count);
  678. return 0;
  679. }
  680. sig = kmem_cache_alloc(sighand_cachep, GFP_KERNEL);
  681. rcu_assign_pointer(tsk->sighand, sig);
  682. if (!sig)
  683. return -ENOMEM;
  684. atomic_set(&sig->count, 1);
  685. memcpy(sig->action, current->sighand->action, sizeof(sig->action));
  686. return 0;
  687. }
  688. void __cleanup_sighand(struct sighand_struct *sighand)
  689. {
  690. if (atomic_dec_and_test(&sighand->count))
  691. kmem_cache_free(sighand_cachep, sighand);
  692. }
  693. static inline int copy_signal(unsigned long clone_flags, struct task_struct * tsk)
  694. {
  695. struct signal_struct *sig;
  696. int ret;
  697. if (clone_flags & CLONE_THREAD) {
  698. atomic_inc(&current->signal->count);
  699. atomic_inc(&current->signal->live);
  700. return 0;
  701. }
  702. sig = kmem_cache_alloc(signal_cachep, GFP_KERNEL);
  703. tsk->signal = sig;
  704. if (!sig)
  705. return -ENOMEM;
  706. ret = copy_thread_group_keys(tsk);
  707. if (ret < 0) {
  708. kmem_cache_free(signal_cachep, sig);
  709. return ret;
  710. }
  711. atomic_set(&sig->count, 1);
  712. atomic_set(&sig->live, 1);
  713. init_waitqueue_head(&sig->wait_chldexit);
  714. sig->flags = 0;
  715. sig->group_exit_code = 0;
  716. sig->group_exit_task = NULL;
  717. sig->group_stop_count = 0;
  718. sig->curr_target = NULL;
  719. init_sigpending(&sig->shared_pending);
  720. INIT_LIST_HEAD(&sig->posix_timers);
  721. hrtimer_init(&sig->real_timer, CLOCK_MONOTONIC, HRTIMER_REL);
  722. sig->it_real_incr.tv64 = 0;
  723. sig->real_timer.function = it_real_fn;
  724. sig->tsk = tsk;
  725. sig->it_virt_expires = cputime_zero;
  726. sig->it_virt_incr = cputime_zero;
  727. sig->it_prof_expires = cputime_zero;
  728. sig->it_prof_incr = cputime_zero;
  729. sig->leader = 0; /* session leadership doesn't inherit */
  730. sig->tty_old_pgrp = 0;
  731. sig->utime = sig->stime = sig->cutime = sig->cstime = cputime_zero;
  732. sig->nvcsw = sig->nivcsw = sig->cnvcsw = sig->cnivcsw = 0;
  733. sig->min_flt = sig->maj_flt = sig->cmin_flt = sig->cmaj_flt = 0;
  734. sig->sched_time = 0;
  735. INIT_LIST_HEAD(&sig->cpu_timers[0]);
  736. INIT_LIST_HEAD(&sig->cpu_timers[1]);
  737. INIT_LIST_HEAD(&sig->cpu_timers[2]);
  738. task_lock(current->group_leader);
  739. memcpy(sig->rlim, current->signal->rlim, sizeof sig->rlim);
  740. task_unlock(current->group_leader);
  741. if (sig->rlim[RLIMIT_CPU].rlim_cur != RLIM_INFINITY) {
  742. /*
  743. * New sole thread in the process gets an expiry time
  744. * of the whole CPU time limit.
  745. */
  746. tsk->it_prof_expires =
  747. secs_to_cputime(sig->rlim[RLIMIT_CPU].rlim_cur);
  748. }
  749. acct_init_pacct(&sig->pacct);
  750. return 0;
  751. }
  752. void __cleanup_signal(struct signal_struct *sig)
  753. {
  754. exit_thread_group_keys(sig);
  755. kmem_cache_free(signal_cachep, sig);
  756. }
  757. static inline void cleanup_signal(struct task_struct *tsk)
  758. {
  759. struct signal_struct *sig = tsk->signal;
  760. atomic_dec(&sig->live);
  761. if (atomic_dec_and_test(&sig->count))
  762. __cleanup_signal(sig);
  763. }
  764. static inline void copy_flags(unsigned long clone_flags, struct task_struct *p)
  765. {
  766. unsigned long new_flags = p->flags;
  767. new_flags &= ~(PF_SUPERPRIV | PF_NOFREEZE);
  768. new_flags |= PF_FORKNOEXEC;
  769. if (!(clone_flags & CLONE_PTRACE))
  770. p->ptrace = 0;
  771. p->flags = new_flags;
  772. }
  773. asmlinkage long sys_set_tid_address(int __user *tidptr)
  774. {
  775. current->clear_child_tid = tidptr;
  776. return current->pid;
  777. }
  778. static inline void rt_mutex_init_task(struct task_struct *p)
  779. {
  780. #ifdef CONFIG_RT_MUTEXES
  781. spin_lock_init(&p->pi_lock);
  782. plist_head_init(&p->pi_waiters, &p->pi_lock);
  783. p->pi_blocked_on = NULL;
  784. #endif
  785. }
  786. /*
  787. * This creates a new process as a copy of the old one,
  788. * but does not actually start it yet.
  789. *
  790. * It copies the registers, and all the appropriate
  791. * parts of the process environment (as per the clone
  792. * flags). The actual kick-off is left to the caller.
  793. */
  794. static task_t *copy_process(unsigned long clone_flags,
  795. unsigned long stack_start,
  796. struct pt_regs *regs,
  797. unsigned long stack_size,
  798. int __user *parent_tidptr,
  799. int __user *child_tidptr,
  800. int pid)
  801. {
  802. int retval;
  803. struct task_struct *p = NULL;
  804. if ((clone_flags & (CLONE_NEWNS|CLONE_FS)) == (CLONE_NEWNS|CLONE_FS))
  805. return ERR_PTR(-EINVAL);
  806. /*
  807. * Thread groups must share signals as well, and detached threads
  808. * can only be started up within the thread group.
  809. */
  810. if ((clone_flags & CLONE_THREAD) && !(clone_flags & CLONE_SIGHAND))
  811. return ERR_PTR(-EINVAL);
  812. /*
  813. * Shared signal handlers imply shared VM. By way of the above,
  814. * thread groups also imply shared VM. Blocking this case allows
  815. * for various simplifications in other code.
  816. */
  817. if ((clone_flags & CLONE_SIGHAND) && !(clone_flags & CLONE_VM))
  818. return ERR_PTR(-EINVAL);
  819. retval = security_task_create(clone_flags);
  820. if (retval)
  821. goto fork_out;
  822. retval = -ENOMEM;
  823. p = dup_task_struct(current);
  824. if (!p)
  825. goto fork_out;
  826. retval = -EAGAIN;
  827. if (atomic_read(&p->user->processes) >=
  828. p->signal->rlim[RLIMIT_NPROC].rlim_cur) {
  829. if (!capable(CAP_SYS_ADMIN) && !capable(CAP_SYS_RESOURCE) &&
  830. p->user != &root_user)
  831. goto bad_fork_free;
  832. }
  833. atomic_inc(&p->user->__count);
  834. atomic_inc(&p->user->processes);
  835. get_group_info(p->group_info);
  836. /*
  837. * If multiple threads are within copy_process(), then this check
  838. * triggers too late. This doesn't hurt, the check is only there
  839. * to stop root fork bombs.
  840. */
  841. if (nr_threads >= max_threads)
  842. goto bad_fork_cleanup_count;
  843. if (!try_module_get(task_thread_info(p)->exec_domain->module))
  844. goto bad_fork_cleanup_count;
  845. if (p->binfmt && !try_module_get(p->binfmt->module))
  846. goto bad_fork_cleanup_put_domain;
  847. p->did_exec = 0;
  848. copy_flags(clone_flags, p);
  849. p->pid = pid;
  850. retval = -EFAULT;
  851. if (clone_flags & CLONE_PARENT_SETTID)
  852. if (put_user(p->pid, parent_tidptr))
  853. goto bad_fork_cleanup;
  854. INIT_LIST_HEAD(&p->children);
  855. INIT_LIST_HEAD(&p->sibling);
  856. p->vfork_done = NULL;
  857. spin_lock_init(&p->alloc_lock);
  858. clear_tsk_thread_flag(p, TIF_SIGPENDING);
  859. init_sigpending(&p->pending);
  860. p->utime = cputime_zero;
  861. p->stime = cputime_zero;
  862. p->sched_time = 0;
  863. p->rchar = 0; /* I/O counter: bytes read */
  864. p->wchar = 0; /* I/O counter: bytes written */
  865. p->syscr = 0; /* I/O counter: read syscalls */
  866. p->syscw = 0; /* I/O counter: write syscalls */
  867. acct_clear_integrals(p);
  868. p->it_virt_expires = cputime_zero;
  869. p->it_prof_expires = cputime_zero;
  870. p->it_sched_expires = 0;
  871. INIT_LIST_HEAD(&p->cpu_timers[0]);
  872. INIT_LIST_HEAD(&p->cpu_timers[1]);
  873. INIT_LIST_HEAD(&p->cpu_timers[2]);
  874. p->lock_depth = -1; /* -1 = no lock */
  875. do_posix_clock_monotonic_gettime(&p->start_time);
  876. p->security = NULL;
  877. p->io_context = NULL;
  878. p->io_wait = NULL;
  879. p->audit_context = NULL;
  880. cpuset_fork(p);
  881. #ifdef CONFIG_NUMA
  882. p->mempolicy = mpol_copy(p->mempolicy);
  883. if (IS_ERR(p->mempolicy)) {
  884. retval = PTR_ERR(p->mempolicy);
  885. p->mempolicy = NULL;
  886. goto bad_fork_cleanup_cpuset;
  887. }
  888. mpol_fix_fork_child_flag(p);
  889. #endif
  890. rt_mutex_init_task(p);
  891. #ifdef CONFIG_DEBUG_MUTEXES
  892. p->blocked_on = NULL; /* not blocked yet */
  893. #endif
  894. p->tgid = p->pid;
  895. if (clone_flags & CLONE_THREAD)
  896. p->tgid = current->tgid;
  897. if ((retval = security_task_alloc(p)))
  898. goto bad_fork_cleanup_policy;
  899. if ((retval = audit_alloc(p)))
  900. goto bad_fork_cleanup_security;
  901. /* copy all the process information */
  902. if ((retval = copy_semundo(clone_flags, p)))
  903. goto bad_fork_cleanup_audit;
  904. if ((retval = copy_files(clone_flags, p)))
  905. goto bad_fork_cleanup_semundo;
  906. if ((retval = copy_fs(clone_flags, p)))
  907. goto bad_fork_cleanup_files;
  908. if ((retval = copy_sighand(clone_flags, p)))
  909. goto bad_fork_cleanup_fs;
  910. if ((retval = copy_signal(clone_flags, p)))
  911. goto bad_fork_cleanup_sighand;
  912. if ((retval = copy_mm(clone_flags, p)))
  913. goto bad_fork_cleanup_signal;
  914. if ((retval = copy_keys(clone_flags, p)))
  915. goto bad_fork_cleanup_mm;
  916. if ((retval = copy_namespace(clone_flags, p)))
  917. goto bad_fork_cleanup_keys;
  918. retval = copy_thread(0, clone_flags, stack_start, stack_size, p, regs);
  919. if (retval)
  920. goto bad_fork_cleanup_namespace;
  921. p->set_child_tid = (clone_flags & CLONE_CHILD_SETTID) ? child_tidptr : NULL;
  922. /*
  923. * Clear TID on mm_release()?
  924. */
  925. p->clear_child_tid = (clone_flags & CLONE_CHILD_CLEARTID) ? child_tidptr: NULL;
  926. p->robust_list = NULL;
  927. #ifdef CONFIG_COMPAT
  928. p->compat_robust_list = NULL;
  929. #endif
  930. INIT_LIST_HEAD(&p->pi_state_list);
  931. p->pi_state_cache = NULL;
  932. /*
  933. * sigaltstack should be cleared when sharing the same VM
  934. */
  935. if ((clone_flags & (CLONE_VM|CLONE_VFORK)) == CLONE_VM)
  936. p->sas_ss_sp = p->sas_ss_size = 0;
  937. /*
  938. * Syscall tracing should be turned off in the child regardless
  939. * of CLONE_PTRACE.
  940. */
  941. clear_tsk_thread_flag(p, TIF_SYSCALL_TRACE);
  942. #ifdef TIF_SYSCALL_EMU
  943. clear_tsk_thread_flag(p, TIF_SYSCALL_EMU);
  944. #endif
  945. /* Our parent execution domain becomes current domain
  946. These must match for thread signalling to apply */
  947. p->parent_exec_id = p->self_exec_id;
  948. /* ok, now we should be set up.. */
  949. p->exit_signal = (clone_flags & CLONE_THREAD) ? -1 : (clone_flags & CSIGNAL);
  950. p->pdeath_signal = 0;
  951. p->exit_state = 0;
  952. /*
  953. * Ok, make it visible to the rest of the system.
  954. * We dont wake it up yet.
  955. */
  956. p->group_leader = p;
  957. INIT_LIST_HEAD(&p->thread_group);
  958. INIT_LIST_HEAD(&p->ptrace_children);
  959. INIT_LIST_HEAD(&p->ptrace_list);
  960. /* Perform scheduler related setup. Assign this task to a CPU. */
  961. sched_fork(p, clone_flags);
  962. /* Need tasklist lock for parent etc handling! */
  963. write_lock_irq(&tasklist_lock);
  964. /*
  965. * The task hasn't been attached yet, so its cpus_allowed mask will
  966. * not be changed, nor will its assigned CPU.
  967. *
  968. * The cpus_allowed mask of the parent may have changed after it was
  969. * copied first time - so re-copy it here, then check the child's CPU
  970. * to ensure it is on a valid CPU (and if not, just force it back to
  971. * parent's CPU). This avoids alot of nasty races.
  972. */
  973. p->cpus_allowed = current->cpus_allowed;
  974. if (unlikely(!cpu_isset(task_cpu(p), p->cpus_allowed) ||
  975. !cpu_online(task_cpu(p))))
  976. set_task_cpu(p, smp_processor_id());
  977. /* CLONE_PARENT re-uses the old parent */
  978. if (clone_flags & (CLONE_PARENT|CLONE_THREAD))
  979. p->real_parent = current->real_parent;
  980. else
  981. p->real_parent = current;
  982. p->parent = p->real_parent;
  983. spin_lock(&current->sighand->siglock);
  984. /*
  985. * Process group and session signals need to be delivered to just the
  986. * parent before the fork or both the parent and the child after the
  987. * fork. Restart if a signal comes in before we add the new process to
  988. * it's process group.
  989. * A fatal signal pending means that current will exit, so the new
  990. * thread can't slip out of an OOM kill (or normal SIGKILL).
  991. */
  992. recalc_sigpending();
  993. if (signal_pending(current)) {
  994. spin_unlock(&current->sighand->siglock);
  995. write_unlock_irq(&tasklist_lock);
  996. retval = -ERESTARTNOINTR;
  997. goto bad_fork_cleanup_namespace;
  998. }
  999. if (clone_flags & CLONE_THREAD) {
  1000. p->group_leader = current->group_leader;
  1001. list_add_tail_rcu(&p->thread_group, &p->group_leader->thread_group);
  1002. if (!cputime_eq(current->signal->it_virt_expires,
  1003. cputime_zero) ||
  1004. !cputime_eq(current->signal->it_prof_expires,
  1005. cputime_zero) ||
  1006. current->signal->rlim[RLIMIT_CPU].rlim_cur != RLIM_INFINITY ||
  1007. !list_empty(&current->signal->cpu_timers[0]) ||
  1008. !list_empty(&current->signal->cpu_timers[1]) ||
  1009. !list_empty(&current->signal->cpu_timers[2])) {
  1010. /*
  1011. * Have child wake up on its first tick to check
  1012. * for process CPU timers.
  1013. */
  1014. p->it_prof_expires = jiffies_to_cputime(1);
  1015. }
  1016. }
  1017. /*
  1018. * inherit ioprio
  1019. */
  1020. p->ioprio = current->ioprio;
  1021. if (likely(p->pid)) {
  1022. add_parent(p);
  1023. if (unlikely(p->ptrace & PT_PTRACED))
  1024. __ptrace_link(p, current->parent);
  1025. if (thread_group_leader(p)) {
  1026. p->signal->tty = current->signal->tty;
  1027. p->signal->pgrp = process_group(current);
  1028. p->signal->session = current->signal->session;
  1029. attach_pid(p, PIDTYPE_PGID, process_group(p));
  1030. attach_pid(p, PIDTYPE_SID, p->signal->session);
  1031. list_add_tail_rcu(&p->tasks, &init_task.tasks);
  1032. __get_cpu_var(process_counts)++;
  1033. }
  1034. attach_pid(p, PIDTYPE_PID, p->pid);
  1035. nr_threads++;
  1036. }
  1037. total_forks++;
  1038. spin_unlock(&current->sighand->siglock);
  1039. write_unlock_irq(&tasklist_lock);
  1040. proc_fork_connector(p);
  1041. return p;
  1042. bad_fork_cleanup_namespace:
  1043. exit_namespace(p);
  1044. bad_fork_cleanup_keys:
  1045. exit_keys(p);
  1046. bad_fork_cleanup_mm:
  1047. if (p->mm)
  1048. mmput(p->mm);
  1049. bad_fork_cleanup_signal:
  1050. cleanup_signal(p);
  1051. bad_fork_cleanup_sighand:
  1052. __cleanup_sighand(p->sighand);
  1053. bad_fork_cleanup_fs:
  1054. exit_fs(p); /* blocking */
  1055. bad_fork_cleanup_files:
  1056. exit_files(p); /* blocking */
  1057. bad_fork_cleanup_semundo:
  1058. exit_sem(p);
  1059. bad_fork_cleanup_audit:
  1060. audit_free(p);
  1061. bad_fork_cleanup_security:
  1062. security_task_free(p);
  1063. bad_fork_cleanup_policy:
  1064. #ifdef CONFIG_NUMA
  1065. mpol_free(p->mempolicy);
  1066. bad_fork_cleanup_cpuset:
  1067. #endif
  1068. cpuset_exit(p);
  1069. bad_fork_cleanup:
  1070. if (p->binfmt)
  1071. module_put(p->binfmt->module);
  1072. bad_fork_cleanup_put_domain:
  1073. module_put(task_thread_info(p)->exec_domain->module);
  1074. bad_fork_cleanup_count:
  1075. put_group_info(p->group_info);
  1076. atomic_dec(&p->user->processes);
  1077. free_uid(p->user);
  1078. bad_fork_free:
  1079. free_task(p);
  1080. fork_out:
  1081. return ERR_PTR(retval);
  1082. }
  1083. struct pt_regs * __devinit __attribute__((weak)) idle_regs(struct pt_regs *regs)
  1084. {
  1085. memset(regs, 0, sizeof(struct pt_regs));
  1086. return regs;
  1087. }
  1088. task_t * __devinit fork_idle(int cpu)
  1089. {
  1090. task_t *task;
  1091. struct pt_regs regs;
  1092. task = copy_process(CLONE_VM, 0, idle_regs(&regs), 0, NULL, NULL, 0);
  1093. if (!task)
  1094. return ERR_PTR(-ENOMEM);
  1095. init_idle(task, cpu);
  1096. return task;
  1097. }
  1098. static inline int fork_traceflag (unsigned clone_flags)
  1099. {
  1100. if (clone_flags & CLONE_UNTRACED)
  1101. return 0;
  1102. else if (clone_flags & CLONE_VFORK) {
  1103. if (current->ptrace & PT_TRACE_VFORK)
  1104. return PTRACE_EVENT_VFORK;
  1105. } else if ((clone_flags & CSIGNAL) != SIGCHLD) {
  1106. if (current->ptrace & PT_TRACE_CLONE)
  1107. return PTRACE_EVENT_CLONE;
  1108. } else if (current->ptrace & PT_TRACE_FORK)
  1109. return PTRACE_EVENT_FORK;
  1110. return 0;
  1111. }
  1112. /*
  1113. * Ok, this is the main fork-routine.
  1114. *
  1115. * It copies the process, and if successful kick-starts
  1116. * it and waits for it to finish using the VM if required.
  1117. */
  1118. long do_fork(unsigned long clone_flags,
  1119. unsigned long stack_start,
  1120. struct pt_regs *regs,
  1121. unsigned long stack_size,
  1122. int __user *parent_tidptr,
  1123. int __user *child_tidptr)
  1124. {
  1125. struct task_struct *p;
  1126. int trace = 0;
  1127. struct pid *pid = alloc_pid();
  1128. long nr;
  1129. if (!pid)
  1130. return -EAGAIN;
  1131. nr = pid->nr;
  1132. if (unlikely(current->ptrace)) {
  1133. trace = fork_traceflag (clone_flags);
  1134. if (trace)
  1135. clone_flags |= CLONE_PTRACE;
  1136. }
  1137. p = copy_process(clone_flags, stack_start, regs, stack_size, parent_tidptr, child_tidptr, nr);
  1138. /*
  1139. * Do this prior waking up the new thread - the thread pointer
  1140. * might get invalid after that point, if the thread exits quickly.
  1141. */
  1142. if (!IS_ERR(p)) {
  1143. struct completion vfork;
  1144. if (clone_flags & CLONE_VFORK) {
  1145. p->vfork_done = &vfork;
  1146. init_completion(&vfork);
  1147. }
  1148. if ((p->ptrace & PT_PTRACED) || (clone_flags & CLONE_STOPPED)) {
  1149. /*
  1150. * We'll start up with an immediate SIGSTOP.
  1151. */
  1152. sigaddset(&p->pending.signal, SIGSTOP);
  1153. set_tsk_thread_flag(p, TIF_SIGPENDING);
  1154. }
  1155. if (!(clone_flags & CLONE_STOPPED))
  1156. wake_up_new_task(p, clone_flags);
  1157. else
  1158. p->state = TASK_STOPPED;
  1159. if (unlikely (trace)) {
  1160. current->ptrace_message = nr;
  1161. ptrace_notify ((trace << 8) | SIGTRAP);
  1162. }
  1163. if (clone_flags & CLONE_VFORK) {
  1164. wait_for_completion(&vfork);
  1165. if (unlikely (current->ptrace & PT_TRACE_VFORK_DONE))
  1166. ptrace_notify ((PTRACE_EVENT_VFORK_DONE << 8) | SIGTRAP);
  1167. }
  1168. } else {
  1169. free_pid(pid);
  1170. nr = PTR_ERR(p);
  1171. }
  1172. return nr;
  1173. }
  1174. #ifndef ARCH_MIN_MMSTRUCT_ALIGN
  1175. #define ARCH_MIN_MMSTRUCT_ALIGN 0
  1176. #endif
  1177. static void sighand_ctor(void *data, kmem_cache_t *cachep, unsigned long flags)
  1178. {
  1179. struct sighand_struct *sighand = data;
  1180. if ((flags & (SLAB_CTOR_VERIFY | SLAB_CTOR_CONSTRUCTOR)) ==
  1181. SLAB_CTOR_CONSTRUCTOR)
  1182. spin_lock_init(&sighand->siglock);
  1183. }
  1184. void __init proc_caches_init(void)
  1185. {
  1186. sighand_cachep = kmem_cache_create("sighand_cache",
  1187. sizeof(struct sighand_struct), 0,
  1188. SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_DESTROY_BY_RCU,
  1189. sighand_ctor, NULL);
  1190. signal_cachep = kmem_cache_create("signal_cache",
  1191. sizeof(struct signal_struct), 0,
  1192. SLAB_HWCACHE_ALIGN|SLAB_PANIC, NULL, NULL);
  1193. files_cachep = kmem_cache_create("files_cache",
  1194. sizeof(struct files_struct), 0,
  1195. SLAB_HWCACHE_ALIGN|SLAB_PANIC, NULL, NULL);
  1196. fs_cachep = kmem_cache_create("fs_cache",
  1197. sizeof(struct fs_struct), 0,
  1198. SLAB_HWCACHE_ALIGN|SLAB_PANIC, NULL, NULL);
  1199. vm_area_cachep = kmem_cache_create("vm_area_struct",
  1200. sizeof(struct vm_area_struct), 0,
  1201. SLAB_PANIC, NULL, NULL);
  1202. mm_cachep = kmem_cache_create("mm_struct",
  1203. sizeof(struct mm_struct), ARCH_MIN_MMSTRUCT_ALIGN,
  1204. SLAB_HWCACHE_ALIGN|SLAB_PANIC, NULL, NULL);
  1205. }
  1206. /*
  1207. * Check constraints on flags passed to the unshare system call and
  1208. * force unsharing of additional process context as appropriate.
  1209. */
  1210. static inline void check_unshare_flags(unsigned long *flags_ptr)
  1211. {
  1212. /*
  1213. * If unsharing a thread from a thread group, must also
  1214. * unshare vm.
  1215. */
  1216. if (*flags_ptr & CLONE_THREAD)
  1217. *flags_ptr |= CLONE_VM;
  1218. /*
  1219. * If unsharing vm, must also unshare signal handlers.
  1220. */
  1221. if (*flags_ptr & CLONE_VM)
  1222. *flags_ptr |= CLONE_SIGHAND;
  1223. /*
  1224. * If unsharing signal handlers and the task was created
  1225. * using CLONE_THREAD, then must unshare the thread
  1226. */
  1227. if ((*flags_ptr & CLONE_SIGHAND) &&
  1228. (atomic_read(&current->signal->count) > 1))
  1229. *flags_ptr |= CLONE_THREAD;
  1230. /*
  1231. * If unsharing namespace, must also unshare filesystem information.
  1232. */
  1233. if (*flags_ptr & CLONE_NEWNS)
  1234. *flags_ptr |= CLONE_FS;
  1235. }
  1236. /*
  1237. * Unsharing of tasks created with CLONE_THREAD is not supported yet
  1238. */
  1239. static int unshare_thread(unsigned long unshare_flags)
  1240. {
  1241. if (unshare_flags & CLONE_THREAD)
  1242. return -EINVAL;
  1243. return 0;
  1244. }
  1245. /*
  1246. * Unshare the filesystem structure if it is being shared
  1247. */
  1248. static int unshare_fs(unsigned long unshare_flags, struct fs_struct **new_fsp)
  1249. {
  1250. struct fs_struct *fs = current->fs;
  1251. if ((unshare_flags & CLONE_FS) &&
  1252. (fs && atomic_read(&fs->count) > 1)) {
  1253. *new_fsp = __copy_fs_struct(current->fs);
  1254. if (!*new_fsp)
  1255. return -ENOMEM;
  1256. }
  1257. return 0;
  1258. }
  1259. /*
  1260. * Unshare the namespace structure if it is being shared
  1261. */
  1262. static int unshare_namespace(unsigned long unshare_flags, struct namespace **new_nsp, struct fs_struct *new_fs)
  1263. {
  1264. struct namespace *ns = current->namespace;
  1265. if ((unshare_flags & CLONE_NEWNS) &&
  1266. (ns && atomic_read(&ns->count) > 1)) {
  1267. if (!capable(CAP_SYS_ADMIN))
  1268. return -EPERM;
  1269. *new_nsp = dup_namespace(current, new_fs ? new_fs : current->fs);
  1270. if (!*new_nsp)
  1271. return -ENOMEM;
  1272. }
  1273. return 0;
  1274. }
  1275. /*
  1276. * Unsharing of sighand for tasks created with CLONE_SIGHAND is not
  1277. * supported yet
  1278. */
  1279. static int unshare_sighand(unsigned long unshare_flags, struct sighand_struct **new_sighp)
  1280. {
  1281. struct sighand_struct *sigh = current->sighand;
  1282. if ((unshare_flags & CLONE_SIGHAND) &&
  1283. (sigh && atomic_read(&sigh->count) > 1))
  1284. return -EINVAL;
  1285. else
  1286. return 0;
  1287. }
  1288. /*
  1289. * Unshare vm if it is being shared
  1290. */
  1291. static int unshare_vm(unsigned long unshare_flags, struct mm_struct **new_mmp)
  1292. {
  1293. struct mm_struct *mm = current->mm;
  1294. if ((unshare_flags & CLONE_VM) &&
  1295. (mm && atomic_read(&mm->mm_users) > 1)) {
  1296. return -EINVAL;
  1297. }
  1298. return 0;
  1299. }
  1300. /*
  1301. * Unshare file descriptor table if it is being shared
  1302. */
  1303. static int unshare_fd(unsigned long unshare_flags, struct files_struct **new_fdp)
  1304. {
  1305. struct files_struct *fd = current->files;
  1306. int error = 0;
  1307. if ((unshare_flags & CLONE_FILES) &&
  1308. (fd && atomic_read(&fd->count) > 1)) {
  1309. *new_fdp = dup_fd(fd, &error);
  1310. if (!*new_fdp)
  1311. return error;
  1312. }
  1313. return 0;
  1314. }
  1315. /*
  1316. * Unsharing of semundo for tasks created with CLONE_SYSVSEM is not
  1317. * supported yet
  1318. */
  1319. static int unshare_semundo(unsigned long unshare_flags, struct sem_undo_list **new_ulistp)
  1320. {
  1321. if (unshare_flags & CLONE_SYSVSEM)
  1322. return -EINVAL;
  1323. return 0;
  1324. }
  1325. /*
  1326. * unshare allows a process to 'unshare' part of the process
  1327. * context which was originally shared using clone. copy_*
  1328. * functions used by do_fork() cannot be used here directly
  1329. * because they modify an inactive task_struct that is being
  1330. * constructed. Here we are modifying the current, active,
  1331. * task_struct.
  1332. */
  1333. asmlinkage long sys_unshare(unsigned long unshare_flags)
  1334. {
  1335. int err = 0;
  1336. struct fs_struct *fs, *new_fs = NULL;
  1337. struct namespace *ns, *new_ns = NULL;
  1338. struct sighand_struct *sigh, *new_sigh = NULL;
  1339. struct mm_struct *mm, *new_mm = NULL, *active_mm = NULL;
  1340. struct files_struct *fd, *new_fd = NULL;
  1341. struct sem_undo_list *new_ulist = NULL;
  1342. check_unshare_flags(&unshare_flags);
  1343. /* Return -EINVAL for all unsupported flags */
  1344. err = -EINVAL;
  1345. if (unshare_flags & ~(CLONE_THREAD|CLONE_FS|CLONE_NEWNS|CLONE_SIGHAND|
  1346. CLONE_VM|CLONE_FILES|CLONE_SYSVSEM))
  1347. goto bad_unshare_out;
  1348. if ((err = unshare_thread(unshare_flags)))
  1349. goto bad_unshare_out;
  1350. if ((err = unshare_fs(unshare_flags, &new_fs)))
  1351. goto bad_unshare_cleanup_thread;
  1352. if ((err = unshare_namespace(unshare_flags, &new_ns, new_fs)))
  1353. goto bad_unshare_cleanup_fs;
  1354. if ((err = unshare_sighand(unshare_flags, &new_sigh)))
  1355. goto bad_unshare_cleanup_ns;
  1356. if ((err = unshare_vm(unshare_flags, &new_mm)))
  1357. goto bad_unshare_cleanup_sigh;
  1358. if ((err = unshare_fd(unshare_flags, &new_fd)))
  1359. goto bad_unshare_cleanup_vm;
  1360. if ((err = unshare_semundo(unshare_flags, &new_ulist)))
  1361. goto bad_unshare_cleanup_fd;
  1362. if (new_fs || new_ns || new_sigh || new_mm || new_fd || new_ulist) {
  1363. task_lock(current);
  1364. if (new_fs) {
  1365. fs = current->fs;
  1366. current->fs = new_fs;
  1367. new_fs = fs;
  1368. }
  1369. if (new_ns) {
  1370. ns = current->namespace;
  1371. current->namespace = new_ns;
  1372. new_ns = ns;
  1373. }
  1374. if (new_sigh) {
  1375. sigh = current->sighand;
  1376. rcu_assign_pointer(current->sighand, new_sigh);
  1377. new_sigh = sigh;
  1378. }
  1379. if (new_mm) {
  1380. mm = current->mm;
  1381. active_mm = current->active_mm;
  1382. current->mm = new_mm;
  1383. current->active_mm = new_mm;
  1384. activate_mm(active_mm, new_mm);
  1385. new_mm = mm;
  1386. }
  1387. if (new_fd) {
  1388. fd = current->files;
  1389. current->files = new_fd;
  1390. new_fd = fd;
  1391. }
  1392. task_unlock(current);
  1393. }
  1394. bad_unshare_cleanup_fd:
  1395. if (new_fd)
  1396. put_files_struct(new_fd);
  1397. bad_unshare_cleanup_vm:
  1398. if (new_mm)
  1399. mmput(new_mm);
  1400. bad_unshare_cleanup_sigh:
  1401. if (new_sigh)
  1402. if (atomic_dec_and_test(&new_sigh->count))
  1403. kmem_cache_free(sighand_cachep, new_sigh);
  1404. bad_unshare_cleanup_ns:
  1405. if (new_ns)
  1406. put_namespace(new_ns);
  1407. bad_unshare_cleanup_fs:
  1408. if (new_fs)
  1409. put_fs_struct(new_fs);
  1410. bad_unshare_cleanup_thread:
  1411. bad_unshare_out:
  1412. return err;
  1413. }