| /**************************************************************************** |
| * sched/group_setuptaskfiles.c |
| * |
| * Copyright (C) 2007-2008, 2010, 2012-2013 Gregory Nutt. All rights reserved. |
| * Author: Gregory Nutt <gnutt@nuttx.org> |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions |
| * are met: |
| * |
| * 1. Redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer. |
| * 2. Redistributions in binary form must reproduce the above copyright |
| * notice, this list of conditions and the following disclaimer in |
| * the documentation and/or other materials provided with the |
| * distribution. |
| * 3. Neither the name NuttX nor the names of its contributors may be |
| * used to endorse or promote products derived from this software |
| * without specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS |
| * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE |
| * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, |
| * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, |
| * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS |
| * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED |
| * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT |
| * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN |
| * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
| * POSSIBILITY OF SUCH DAMAGE. |
| * |
| ****************************************************************************/ |
| |
| /**************************************************************************** |
| * Included Files |
| ****************************************************************************/ |
| |
| #include <nuttx/config.h> |
| |
| #include <sched.h> |
| #include <errno.h> |
| |
| #include <nuttx/fs/fs.h> |
| #include <nuttx/net/net.h> |
| |
| #include "os_internal.h" |
| #include "group_internal.h" |
| |
| #if CONFIG_NFILE_DESCRIPTORS > 0 || CONFIG_NSOCKET_DESCRIPTORS > 0 |
| |
| /**************************************************************************** |
| * Pre-processor Definitions |
| ****************************************************************************/ |
| |
| /* Determine how many file descriptors to clone. If CONFIG_FDCLONE_DISABLE |
| * is set, no file descriptors will be cloned. If CONFIG_FDCLONE_STDIO is |
| * set, only the first three descriptors (stdin, stdout, and stderr) will |
| * be cloned. Otherwise all file descriptors will be cloned. |
| */ |
| |
| #if defined(CONFIG_FDCLONE_STDIO) && CONFIG_NFILE_DESCRIPTORS > 3 |
| # define NFDS_TOCLONE 3 |
| #else |
| # define NFDS_TOCLONE CONFIG_NFILE_DESCRIPTORS |
| #endif |
| |
| /**************************************************************************** |
| * Private Functions |
| ****************************************************************************/ |
| |
| /**************************************************************************** |
| * Name: sched_dupfiles |
| * |
| * Description: |
| * Duplicate parent task's file descriptors. |
| * |
| * Input Parameters: |
| * tcb - tcb of the new task. |
| * |
| * Return Value: |
| * None |
| * |
| ****************************************************************************/ |
| |
| #if CONFIG_NFILE_DESCRIPTORS > 0 && !defined(CONFIG_FDCLONE_DISABLE) |
| static inline void sched_dupfiles(FAR struct task_tcb_s *tcb) |
| { |
| /* The parent task is the one at the head of the ready-to-run list */ |
| |
| FAR struct tcb_s *rtcb = (FAR struct tcb_s*)g_readytorun.head; |
| FAR struct file *parent; |
| FAR struct file *child; |
| int i; |
| |
| DEBUGASSERT(tcb && tcb->cmn.group && rtcb->group); |
| |
| /* Duplicate the file descriptors. This will be either all of the |
| * file descriptors or just the first three (stdin, stdout, and stderr) |
| * if CONFIG_FDCLONE_STDIO is defined. NFSDS_TOCLONE is set |
| * accordingly above. |
| */ |
| |
| /* Get pointers to the parent and child task file lists */ |
| |
| parent = rtcb->group->tg_filelist.fl_files; |
| child = tcb->cmn.group->tg_filelist.fl_files; |
| |
| /* Check each file in the parent file list */ |
| |
| for (i = 0; i < NFDS_TOCLONE; i++) |
| { |
| /* Check if this file is opened by the parent. We can tell if |
| * if the file is open because it contain a reference to a non-NULL |
| * i-node structure. |
| */ |
| |
| if (parent[i].f_inode) |
| { |
| /* Yes... duplicate it for the child */ |
| |
| (void)files_dup(&parent[i], &child[i]); |
| } |
| } |
| } |
| #else /* CONFIG_NFILE_DESCRIPTORS && !CONFIG_FDCLONE_DISABLE */ |
| # define sched_dupfiles(tcb) |
| #endif /* CONFIG_NFILE_DESCRIPTORS && !CONFIG_FDCLONE_DISABLE */ |
| |
| /**************************************************************************** |
| * Name: sched_dupsockets |
| * |
| * Description: |
| * Duplicate the parent task's socket descriptors. |
| * |
| * Input Parameters: |
| * tcb - tcb of the new task. |
| * |
| * Return Value: |
| * None |
| * |
| ****************************************************************************/ |
| |
| #if CONFIG_NSOCKET_DESCRIPTORS > 0 && !defined(CONFIG_SDCLONE_DISABLE) |
| static inline void sched_dupsockets(FAR struct task_tcb_s *tcb) |
| { |
| /* The parent task is the one at the head of the ready-to-run list */ |
| |
| FAR struct tcb_s *rtcb = (FAR struct tcb_s*)g_readytorun.head; |
| FAR struct socket *parent; |
| FAR struct socket *child; |
| int i; |
| |
| /* Duplicate the socket descriptors of all sockets opened by the parent |
| * task. |
| */ |
| |
| DEBUGASSERT(tcb && tcb->cmn.group && rtcb->group); |
| |
| /* Get pointers to the parent and child task socket lists */ |
| |
| parent = rtcb->group->tg_socketlist.sl_sockets; |
| child = tcb->cmn.group->tg_socketlist.sl_sockets; |
| |
| /* Check each socket in the parent socket list */ |
| |
| for (i = 0; i < CONFIG_NSOCKET_DESCRIPTORS; i++) |
| { |
| /* Check if this parent socket is allocated. We can tell if the |
| * socket is allocated because it will have a positive, non-zero |
| * reference count. |
| */ |
| |
| if (parent[i].s_crefs > 0) |
| { |
| /* Yes... duplicate it for the child */ |
| |
| (void)net_clone(&parent[i], &child[i]); |
| } |
| } |
| } |
| #else /* CONFIG_NSOCKET_DESCRIPTORS && !CONFIG_SDCLONE_DISABLE */ |
| # define sched_dupsockets(tcb) |
| #endif /* CONFIG_NSOCKET_DESCRIPTORS && !CONFIG_SDCLONE_DISABLE */ |
| |
| /**************************************************************************** |
| * Public Functions |
| ****************************************************************************/ |
| |
| /**************************************************************************** |
| * Name: group_setuptaskfiles |
| * |
| * Description: |
| * Configure a newly allocated TCB so that it will inherit |
| * file descriptors and streams from the parent task. |
| * |
| * Parameters: |
| * tcb - tcb of the new task. |
| * |
| * Return Value: |
| * Zero (OK) is returned on success; A negated errno value is returned on |
| * failure. |
| * |
| * Assumptions: |
| * |
| ****************************************************************************/ |
| |
| int group_setuptaskfiles(FAR struct task_tcb_s *tcb) |
| { |
| #if CONFIG_NFILE_DESCRIPTORS > 0 || CONFIG_NSOCKET_DESCRIPTORS > 0 |
| FAR struct task_group_s *group = tcb->cmn.group; |
| |
| DEBUGASSERT(group); |
| #endif |
| #ifndef CONFIG_DISABLE_PTHREAD |
| DEBUGASSERT((tcb->cmn.flags & TCB_FLAG_TTYPE_MASK) != TCB_FLAG_TTYPE_PTHREAD); |
| #endif |
| |
| #if CONFIG_NFILE_DESCRIPTORS > 0 |
| /* Initialize file descriptors for the TCB */ |
| |
| files_initlist(&group->tg_filelist); |
| #endif |
| |
| #if CONFIG_NSOCKET_DESCRIPTORS > 0 |
| /* Allocate socket descriptors for the TCB */ |
| |
| net_initlist(&group->tg_socketlist); |
| #endif |
| |
| /* Duplicate the parent task's file descriptors */ |
| |
| sched_dupfiles(tcb); |
| |
| /* Duplicate the parent task's socket descriptors */ |
| |
| sched_dupsockets(tcb); |
| |
| /* Allocate file/socket streams for the new TCB */ |
| |
| #if CONFIG_NFILE_STREAMS > 0 |
| return group_setupstreams(tcb); |
| #else |
| return OK; |
| #endif |
| } |
| |
| #endif /* CONFIG_NFILE_DESCRIPTORS > 0 || CONFIG_NSOCKET_DESCRIPTORS > 0 */ |
| |