fcntl.c
上传用户:lgb322
上传日期:2013-02-24
资源大小:30529k
文件大小:12k
源码类别:

嵌入式Linux

开发平台:

Unix_Linux

  1. /*
  2.  *  linux/fs/fcntl.c
  3.  *
  4.  *  Copyright (C) 1991, 1992  Linus Torvalds
  5.  */
  6. #include <linux/init.h>
  7. #include <linux/mm.h>
  8. #include <linux/file.h>
  9. #include <linux/dnotify.h>
  10. #include <linux/smp_lock.h>
  11. #include <linux/slab.h>
  12. #include <linux/iobuf.h>
  13. #include <asm/poll.h>
  14. #include <asm/siginfo.h>
  15. #include <asm/uaccess.h>
  16. extern int sock_fcntl (struct file *, unsigned int cmd, unsigned long arg);
  17. extern int fcntl_setlease(unsigned int fd, struct file *filp, long arg);
  18. extern int fcntl_getlease(struct file *filp);
  19. /* Expand files.  Return <0 on error; 0 nothing done; 1 files expanded,
  20.  * we may have blocked. 
  21.  *
  22.  * Should be called with the files->file_lock spinlock held for write.
  23.  */
  24. static int expand_files(struct files_struct *files, int nr)
  25. {
  26. int err, expand = 0;
  27. #ifdef FDSET_DEBUG
  28. printk (KERN_ERR __FUNCTION__ " %d: nr = %dn", current->pid, nr);
  29. #endif
  30. if (nr >= files->max_fdset) {
  31. expand = 1;
  32. if ((err = expand_fdset(files, nr)))
  33. goto out;
  34. }
  35. if (nr >= files->max_fds) {
  36. expand = 1;
  37. if ((err = expand_fd_array(files, nr)))
  38. goto out;
  39. }
  40. err = expand;
  41.  out:
  42. #ifdef FDSET_DEBUG
  43. if (err)
  44. printk (KERN_ERR __FUNCTION__ " %d: return %dn", current->pid, err);
  45. #endif
  46. return err;
  47. }
  48. /*
  49.  * locate_fd finds a free file descriptor in the open_fds fdset,
  50.  * expanding the fd arrays if necessary.  The files write lock will be
  51.  * held on exit to ensure that the fd can be entered atomically.
  52.  */
  53. static int locate_fd(struct files_struct *files, 
  54.     struct file *file, int orig_start)
  55. {
  56. unsigned int newfd;
  57. int error;
  58. int start;
  59. write_lock(&files->file_lock);
  60. repeat:
  61. /*
  62.  * Someone might have closed fd's in the range
  63.  * orig_start..files->next_fd
  64.  */
  65. start = orig_start;
  66. if (start < files->next_fd)
  67. start = files->next_fd;
  68. newfd = start;
  69. if (start < files->max_fdset) {
  70. newfd = find_next_zero_bit(files->open_fds->fds_bits,
  71. files->max_fdset, start);
  72. }
  73. error = -EMFILE;
  74. if (newfd >= current->rlim[RLIMIT_NOFILE].rlim_cur)
  75. goto out;
  76. error = expand_files(files, newfd);
  77. if (error < 0)
  78. goto out;
  79. /*
  80.  * If we needed to expand the fs array we
  81.  * might have blocked - try again.
  82.  */
  83. if (error)
  84. goto repeat;
  85. if (start <= files->next_fd)
  86. files->next_fd = newfd + 1;
  87. error = newfd;
  88. out:
  89. return error;
  90. }
  91. static inline void allocate_fd(struct files_struct *files, 
  92. struct file *file, int fd)
  93. {
  94. FD_SET(fd, files->open_fds);
  95. FD_CLR(fd, files->close_on_exec);
  96. write_unlock(&files->file_lock);
  97. fd_install(fd, file);
  98. }
  99. static int dupfd(struct file *file, int start)
  100. {
  101. struct files_struct * files = current->files;
  102. int ret;
  103. ret = locate_fd(files, file, start);
  104. if (ret < 0) 
  105. goto out_putf;
  106. allocate_fd(files, file, ret);
  107. return ret;
  108. out_putf:
  109. write_unlock(&files->file_lock);
  110. fput(file);
  111. return ret;
  112. }
  113. asmlinkage long sys_dup2(unsigned int oldfd, unsigned int newfd)
  114. {
  115. int err = -EBADF;
  116. struct file * file, *tofree;
  117. struct files_struct * files = current->files;
  118. write_lock(&files->file_lock);
  119. if (!(file = fcheck(oldfd)))
  120. goto out_unlock;
  121. err = newfd;
  122. if (newfd == oldfd)
  123. goto out_unlock;
  124. err = -EBADF;
  125. if (newfd >= current->rlim[RLIMIT_NOFILE].rlim_cur)
  126. goto out_unlock;
  127. get_file(file); /* We are now finished with oldfd */
  128. err = expand_files(files, newfd);
  129. if (err < 0)
  130. goto out_fput;
  131. /* To avoid races with open() and dup(), we will mark the fd as
  132.  * in-use in the open-file bitmap throughout the entire dup2()
  133.  * process.  This is quite safe: do_close() uses the fd array
  134.  * entry, not the bitmap, to decide what work needs to be
  135.  * done.  --sct */
  136. /* Doesn't work. open() might be there first. --AV */
  137. /* Yes. It's a race. In user space. Nothing sane to do */
  138. err = -EBUSY;
  139. tofree = files->fd[newfd];
  140. if (!tofree && FD_ISSET(newfd, files->open_fds))
  141. goto out_fput;
  142. files->fd[newfd] = file;
  143. FD_SET(newfd, files->open_fds);
  144. FD_CLR(newfd, files->close_on_exec);
  145. write_unlock(&files->file_lock);
  146. if (tofree)
  147. filp_close(tofree, files);
  148. err = newfd;
  149. out:
  150. return err;
  151. out_unlock:
  152. write_unlock(&files->file_lock);
  153. goto out;
  154. out_fput:
  155. write_unlock(&files->file_lock);
  156. fput(file);
  157. goto out;
  158. }
  159. asmlinkage long sys_dup(unsigned int fildes)
  160. {
  161. int ret = -EBADF;
  162. struct file * file = fget(fildes);
  163. if (file)
  164. ret = dupfd(file, 0);
  165. return ret;
  166. }
  167. #define SETFL_MASK (O_APPEND | O_NONBLOCK | O_NDELAY | FASYNC | O_DIRECT)
  168. static int setfl(int fd, struct file * filp, unsigned long arg)
  169. {
  170. struct inode * inode = filp->f_dentry->d_inode;
  171. int error;
  172. /*
  173.  * In the case of an append-only file, O_APPEND
  174.  * cannot be cleared
  175.  */
  176. if (!(arg & O_APPEND) && IS_APPEND(inode))
  177. return -EPERM;
  178. /* Did FASYNC state change? */
  179. if ((arg ^ filp->f_flags) & FASYNC) {
  180. if (filp->f_op && filp->f_op->fasync) {
  181. error = filp->f_op->fasync(fd, filp, (arg & FASYNC) != 0);
  182. if (error < 0)
  183. return error;
  184. }
  185. }
  186. if (arg & O_DIRECT) {
  187. /*
  188.  * alloc_kiovec() can sleep and we are only serialized by
  189.  * the big kernel lock here, so abuse the i_sem to serialize
  190.  * this case too. We of course wouldn't need to go deep down
  191.  * to the inode layer, we could stay at the file layer, but
  192.  * we don't want to pay for the memory of a semaphore in each
  193.  * file structure too and we use the inode semaphore that we just
  194.  * pay for anyways.
  195.  */
  196. error = 0;
  197. down(&inode->i_sem);
  198. if (!filp->f_iobuf)
  199. error = alloc_kiovec(1, &filp->f_iobuf);
  200. up(&inode->i_sem);
  201. if (error < 0)
  202. return error;
  203. }
  204. /* required for strict SunOS emulation */
  205. if (O_NONBLOCK != O_NDELAY)
  206.        if (arg & O_NDELAY)
  207.    arg |= O_NONBLOCK;
  208. filp->f_flags = (arg & SETFL_MASK) | (filp->f_flags & ~SETFL_MASK);
  209. return 0;
  210. }
  211. static long do_fcntl(unsigned int fd, unsigned int cmd,
  212.      unsigned long arg, struct file * filp)
  213. {
  214. long err = -EINVAL;
  215. switch (cmd) {
  216. case F_DUPFD:
  217. if (arg < NR_OPEN) {
  218. get_file(filp);
  219. err = dupfd(filp, arg);
  220. }
  221. break;
  222. case F_GETFD:
  223. err = get_close_on_exec(fd);
  224. break;
  225. case F_SETFD:
  226. err = 0;
  227. set_close_on_exec(fd, arg&1);
  228. break;
  229. case F_GETFL:
  230. err = filp->f_flags;
  231. break;
  232. case F_SETFL:
  233. lock_kernel();
  234. err = setfl(fd, filp, arg);
  235. unlock_kernel();
  236. break;
  237. case F_GETLK:
  238. err = fcntl_getlk(fd, (struct flock *) arg);
  239. break;
  240. case F_SETLK:
  241. case F_SETLKW:
  242. err = fcntl_setlk(fd, cmd, (struct flock *) arg);
  243. break;
  244. case F_GETOWN:
  245. /*
  246.  * XXX If f_owner is a process group, the
  247.  * negative return value will get converted
  248.  * into an error.  Oops.  If we keep the
  249.  * current syscall conventions, the only way
  250.  * to fix this will be in libc.
  251.  */
  252. err = filp->f_owner.pid;
  253. break;
  254. case F_SETOWN:
  255. lock_kernel();
  256. filp->f_owner.pid = arg;
  257. filp->f_owner.uid = current->uid;
  258. filp->f_owner.euid = current->euid;
  259. err = 0;
  260. if (S_ISSOCK (filp->f_dentry->d_inode->i_mode))
  261. err = sock_fcntl (filp, F_SETOWN, arg);
  262. unlock_kernel();
  263. break;
  264. case F_GETSIG:
  265. err = filp->f_owner.signum;
  266. break;
  267. case F_SETSIG:
  268. /* arg == 0 restores default behaviour. */
  269. if (arg < 0 || arg > _NSIG) {
  270. break;
  271. }
  272. err = 0;
  273. filp->f_owner.signum = arg;
  274. break;
  275. case F_GETLEASE:
  276. err = fcntl_getlease(filp);
  277. break;
  278. case F_SETLEASE:
  279. err = fcntl_setlease(fd, filp, arg);
  280. break;
  281. case F_NOTIFY:
  282. err = fcntl_dirnotify(fd, filp, arg);
  283. break;
  284. default:
  285. /* sockets need a few special fcntls. */
  286. err = -EINVAL;
  287. if (S_ISSOCK (filp->f_dentry->d_inode->i_mode))
  288. err = sock_fcntl (filp, cmd, arg);
  289. break;
  290. }
  291. return err;
  292. }
  293. asmlinkage long sys_fcntl(unsigned int fd, unsigned int cmd, unsigned long arg)
  294. {
  295. struct file * filp;
  296. long err = -EBADF;
  297. filp = fget(fd);
  298. if (!filp)
  299. goto out;
  300. err = do_fcntl(fd, cmd, arg, filp);
  301.   fput(filp);
  302. out:
  303. return err;
  304. }
  305. #if BITS_PER_LONG == 32
  306. asmlinkage long sys_fcntl64(unsigned int fd, unsigned int cmd, unsigned long arg)
  307. {
  308. struct file * filp;
  309. long err;
  310. err = -EBADF;
  311. filp = fget(fd);
  312. if (!filp)
  313. goto out;
  314. switch (cmd) {
  315. case F_GETLK64:
  316. err = fcntl_getlk64(fd, (struct flock64 *) arg);
  317. break;
  318. case F_SETLK64:
  319. err = fcntl_setlk64(fd, cmd, (struct flock64 *) arg);
  320. break;
  321. case F_SETLKW64:
  322. err = fcntl_setlk64(fd, cmd, (struct flock64 *) arg);
  323. break;
  324. default:
  325. err = do_fcntl(fd, cmd, arg, filp);
  326. break;
  327. }
  328. fput(filp);
  329. out:
  330. return err;
  331. }
  332. #endif
  333. /* Table to convert sigio signal codes into poll band bitmaps */
  334. static long band_table[NSIGPOLL] = {
  335. POLLIN | POLLRDNORM, /* POLL_IN */
  336. POLLOUT | POLLWRNORM | POLLWRBAND, /* POLL_OUT */
  337. POLLIN | POLLRDNORM | POLLMSG, /* POLL_MSG */
  338. POLLERR, /* POLL_ERR */
  339. POLLPRI | POLLRDBAND, /* POLL_PRI */
  340. POLLHUP | POLLERR /* POLL_HUP */
  341. };
  342. static void send_sigio_to_task(struct task_struct *p,
  343.        struct fown_struct *fown, 
  344.        int fd,
  345.        int reason)
  346. {
  347. if ((fown->euid != 0) &&
  348.     (fown->euid ^ p->suid) && (fown->euid ^ p->uid) &&
  349.     (fown->uid ^ p->suid) && (fown->uid ^ p->uid))
  350. return;
  351. switch (fown->signum) {
  352. siginfo_t si;
  353. default:
  354. /* Queue a rt signal with the appropriate fd as its
  355.    value.  We use SI_SIGIO as the source, not 
  356.    SI_KERNEL, since kernel signals always get 
  357.    delivered even if we can't queue.  Failure to
  358.    queue in this case _should_ be reported; we fall
  359.    back to SIGIO in that case. --sct */
  360. si.si_signo = fown->signum;
  361. si.si_errno = 0;
  362.         si.si_code  = reason & ~__SI_MASK;
  363. /* Make sure we are called with one of the POLL_*
  364.    reasons, otherwise we could leak kernel stack into
  365.    userspace.  */
  366. if ((reason & __SI_MASK) != __SI_POLL)
  367. BUG();
  368. if (reason - POLL_IN >= NSIGPOLL)
  369. si.si_band  = ~0L;
  370. else
  371. si.si_band = band_table[reason - POLL_IN];
  372. si.si_fd    = fd;
  373. if (!send_sig_info(fown->signum, &si, p))
  374. break;
  375. /* fall-through: fall back on the old plain SIGIO signal */
  376. case 0:
  377. send_sig(SIGIO, p, 1);
  378. }
  379. }
  380. void send_sigio(struct fown_struct *fown, int fd, int band)
  381. {
  382. struct task_struct * p;
  383. int   pid = fown->pid;
  384. read_lock(&tasklist_lock);
  385. if ( (pid > 0) && (p = find_task_by_pid(pid)) ) {
  386. send_sigio_to_task(p, fown, fd, band);
  387. goto out;
  388. }
  389. for_each_task(p) {
  390. int match = p->pid;
  391. if (pid < 0)
  392. match = -p->pgrp;
  393. if (pid != match)
  394. continue;
  395. send_sigio_to_task(p, fown, fd, band);
  396. }
  397. out:
  398. read_unlock(&tasklist_lock);
  399. }
  400. static rwlock_t fasync_lock = RW_LOCK_UNLOCKED;
  401. static kmem_cache_t *fasync_cache;
  402. /*
  403.  * fasync_helper() is used by some character device drivers (mainly mice)
  404.  * to set up the fasync queue. It returns negative on error, 0 if it did
  405.  * no changes and positive if it added/deleted the entry.
  406.  */
  407. int fasync_helper(int fd, struct file * filp, int on, struct fasync_struct **fapp)
  408. {
  409. struct fasync_struct *fa, **fp;
  410. struct fasync_struct *new = NULL;
  411. int result = 0;
  412. if (on) {
  413. new = kmem_cache_alloc(fasync_cache, SLAB_KERNEL);
  414. if (!new)
  415. return -ENOMEM;
  416. }
  417. write_lock_irq(&fasync_lock);
  418. for (fp = fapp; (fa = *fp) != NULL; fp = &fa->fa_next) {
  419. if (fa->fa_file == filp) {
  420. if(on) {
  421. fa->fa_fd = fd;
  422. kmem_cache_free(fasync_cache, new);
  423. } else {
  424. *fp = fa->fa_next;
  425. kmem_cache_free(fasync_cache, fa);
  426. result = 1;
  427. }
  428. goto out;
  429. }
  430. }
  431. if (on) {
  432. new->magic = FASYNC_MAGIC;
  433. new->fa_file = filp;
  434. new->fa_fd = fd;
  435. new->fa_next = *fapp;
  436. *fapp = new;
  437. result = 1;
  438. }
  439. out:
  440. write_unlock_irq(&fasync_lock);
  441. return result;
  442. }
  443. void __kill_fasync(struct fasync_struct *fa, int sig, int band)
  444. {
  445. while (fa) {
  446. struct fown_struct * fown;
  447. if (fa->magic != FASYNC_MAGIC) {
  448. printk(KERN_ERR "kill_fasync: bad magic number in "
  449.        "fasync_struct!n");
  450. return;
  451. }
  452. fown = &fa->fa_file->f_owner;
  453. /* Don't send SIGURG to processes which have not set a
  454.    queued signum: SIGURG has its own default signalling
  455.    mechanism. */
  456. if (fown->pid && !(sig == SIGURG && fown->signum == 0))
  457. send_sigio(fown, fa->fa_fd, band);
  458. fa = fa->fa_next;
  459. }
  460. }
  461. void kill_fasync(struct fasync_struct **fp, int sig, int band)
  462. {
  463. read_lock(&fasync_lock);
  464. __kill_fasync(*fp, sig, band);
  465. read_unlock(&fasync_lock);
  466. }
  467. static int __init fasync_init(void)
  468. {
  469. fasync_cache = kmem_cache_create("fasync cache",
  470. sizeof(struct fasync_struct), 0, 0, NULL, NULL);
  471. if (!fasync_cache)
  472. panic("cannot create fasync slab cache");
  473. return 0;
  474. }
  475. module_init(fasync_init)