Basic setup of network select

git-svn-id: svn://svn.code.sf.net/p/nuttx/code/trunk@1278 42af7a65-404d-4744-a932-0658087f49c3

1 files changed, 372 insertions, 0 deletions

diff --git a/nuttx/examples/poll/net_listener.c b/nuttx/examples/poll/net_listener.c
new file mode 100644
index 000000000..d511bfd44
--- /dev/null
+++ b/nuttx/examples/poll/net_listener.c
@@ -0,0 +1,372 @@
+/****************************************************************************
+ * examples/poll/net_listener.c
+ *
+ *   Copyright (C) 2008 Gregory Nutt. All rights reserved.
+ *   Author: Gregory Nutt <spudmonkey@racsa.co.cr>
+ *
+ * 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 <sys/types.h>
+#include <sys/stat.h>
+#include <sys/select.h>
+#include <sys/socket.h>
+#include <sys/ioctl.h>
+
+#include <stdio.h>
+#include <unistd.h>
+#include <string.h>
+#include <pthread.h>
+#include <errno.h>
+#include <debug.h>
+
+#include "poll_internal.h"
+
+/****************************************************************************
+ * Definitions
+ ****************************************************************************/
+
+#define IOBUFFER_SIZE 80
+
+/****************************************************************************
+ * Private Types
+ ****************************************************************************/
+
+struct net_listener_s
+{
+  struct sockaddr_in addr;
+  fd_set          master;
+  fd_set          working;
+  char            buffer[IOBUFFER_SIZE];
+  int             listensd;
+  int             mxsd;
+};
+
+/****************************************************************************
+ * Private Data
+ ****************************************************************************/
+
+/****************************************************************************
+ * Private Functions
+ ****************************************************************************/
+
+/****************************************************************************
+ * Name: net_closeclient
+ ****************************************************************************/
+
+static boolean net_closeclient(struct net_listener_s *nls, int sd)
+{
+  message("net_listener: Closing host side connection sd=%d\n", sd);
+  close(sd);
+  FD_CLR(sd, &nls->master);
+
+  /* If we just closed the max SD, then search downward for the next biggest SD. */
+
+  while (FD_ISSET(nls->mxsd, &nls->master) == FALSE)
+    {
+      nls->mxsd -= 1;
+    }
+  return TRUE;
+}
+
+/****************************************************************************
+ * Name: net_incomingdata
+ ****************************************************************************/
+
+static inline boolean net_incomingdata(struct net_listener_s *nls, int sd)
+{
+  int nbytes;
+  int ret;
+
+  /* Read data from the socket */
+
+#ifdef FIONBIO
+  for (;;)
+#endif
+    {
+      message("net_listener: Read data from sd=%d\n", sd);
+      ret = recv(sd, nls->buffer, IOBUFFER_SIZE, 0);
+      if (ret < 0)
+        {
+          if (errno != EINTR)
+            {
+              message("net_listener: recv failed sd=%d: %d\n", sd, errno);
+              if (errno != EAGAIN)
+                {
+                  net_closeclient(nls, sd);
+                  return FALSE;
+                }
+            }
+        }
+      else if (ret == 0)
+        {
+          message("net_listener: Client connection lost sd=%d\n", sd);
+          net_closeclient(nls, sd);
+          return FALSE;
+        }
+      else
+        {
+          nls->buffer[ret]='\0';
+          message("poll_listener: Read '%s' (%d bytes)\n", sd, nls->buffer, ret);
+
+          /* Echo the data back to the client */
+
+          for (nbytes = ret; nbytes > 0; )
+            {
+              ret = send(sd, nls->buffer, nbytes, 0);
+              if (ret < 0)
+                {
+                  if (errno != EINTR)
+                    {
+                       message("net_listener: Send faile sd=%d: \n", sd, errno);
+                       net_closeclient(nls, sd);
+                       return FALSE;
+                    }
+                }
+              else
+                {
+                  nbytes += ret;
+                }
+            }
+        }
+    }
+}
+
+/****************************************************************************
+ * Name: net_connection
+ ****************************************************************************/
+
+static inline boolean net_connection(struct net_listener_s *nls)
+{
+  int sd;
+
+  /* Loop until all connections have been processed */
+
+#ifdef FIONBIO
+  for (;;)
+#endif
+    {
+      message("net_listener: Accepting accepting connection on sd=%d\n", nls->listensd);
+      sd = accept(nls->listensd, NULL, NULL);
+      if (sd < 0)
+        {
+          message("net_listener: accept failed: %d\n", sd);
+          if (errno != EINTR)
+            {
+              return FALSE;
+            }
+        }
+      else
+        {
+          /* Add the new connection to the master set */
+
+          FD_SET(sd, &nls->master);
+          if (sd > nls->mxsd)
+            {
+               nls->mxsd = sd;
+            }
+          return TRUE;
+        }
+    }
+}
+
+/****************************************************************************
+ * Name: net_mksocket
+ ****************************************************************************/
+
+static inline boolean net_mksocket(struct net_listener_s *nls)
+{
+  int value;
+  int ret;
+
+  /* Create a listening socket */
+
+  message("net_listener: Initializing listener socket\n");
+  nls->listensd = socket(AF_INET, SOCK_STREAM, 0);
+  if (nls->listensd < 0)
+    {
+      message("net_listener: socket failed: %d\n", errno);
+      return FALSE;
+    }
+
+  /* Configure the socket */
+
+  value = 1;
+  ret = setsockopt(nls->listensd, SOL_SOCKET,  SO_REUSEADDR, (char*)&value, sizeof(int));
+  if (ret < 0)
+    {
+      message("net_listener: setsockopt failed: %d\n", errno);
+      close(nls->listensd);
+      return FALSE;
+    }
+
+  /* Set the socket to non-blocking */
+
+#ifdef FIONBIO
+  ret = ioctl(nls->listensd, FIONBIO, (char *)&value);
+  if (ret < 0)
+    {
+      message("net_listener: ioctl failed: %d\n", errno);
+      close(nls->listensd);
+      return FALSE;
+    }
+#endif
+
+  /* Bind the socket */
+
+  memset(&nls->addr, 0, sizeof(struct sockaddr_in));
+  nls->addr.sin_family      = AF_INET;
+  nls->addr.sin_addr.s_addr = htonl(INADDR_ANY);
+  nls->addr.sin_port        = htons(LISTENER_PORT);
+  ret = bind(nls->listensd, (struct sockaddr *)&nls->addr, sizeof(struct sockaddr_in));
+  if (ret < 0)
+    {
+      message("net_listener: bind failed: %d\n", errno);
+      close(nls->listensd);
+      return FALSE;
+    }
+
+  /* Mark the socket as a listener */
+
+  ret = listen(nls->listensd, 32);
+  if (ret < 0)
+    {
+      message("net_listener: bind failed: %d\n", errno);
+      close(nls->listensd);
+      return FALSE;
+    }
+
+  return TRUE;
+}
+
+/****************************************************************************
+ * Public Functions
+ ****************************************************************************/
+
+/****************************************************************************
+ * Name: net_listener
+ ****************************************************************************/
+
+void *net_listener(pthread_addr_t pvarg)
+{
+  struct net_listener_s nls;
+  struct timeval timeout;
+  int nsds;
+  int ret;
+  int i;
+
+  /* Set up a listening socket */
+
+  memset(&nls, 0, sizeof(struct net_listener_s));
+  if (!net_mksocket(&nls))
+    {
+       return (void*)1;
+    }
+
+  /* Initialize the 'master' file descriptor set */
+
+  FD_ZERO(&nls.master);
+  nls.mxsd = nls.listensd;
+  FD_SET(nls.listensd, &nls.master);
+
+  /* Set up a 3 second timeout */
+
+  timeout.tv_sec  = NET_LISTENER_DELAY;
+  timeout.tv_usec = 0;
+
+  /* Loop waiting for incoming connections or for incoming data
+   * on any of the connect sockets.
+   */
+
+  for (;;)
+    {
+      /* Wait on select */
+
+      message("net_listener: Calling select(), listener sd=%d\n", nls.listensd);
+      memcpy(&nls.working, &nls.master, sizeof(fd_set));
+      ret = select(nls.mxsd + 1, &nls.working, NULL, NULL, &timeout);
+      if (ret < 0)
+        {
+          message("net_listener: select failed: %d\n", errno);
+          break;
+        }
+
+      /* Check for timeout */
+
+      if (ret == 0)
+        {
+          message("net_listener: Timeout\n");
+          continue;
+        }
+
+      /* Find which descriptors caused the wakeup */
+
+      nsds = ret;
+      for (i = 0; i <= nls.mxsd && nsds > 0; i++)
+        {
+          /* Is this descriptor ready? */
+
+          if (FD_ISSET(i, &nls.working))
+            {
+              /* Yes, is it our listener? */
+
+              message("net_listener: Activity on sd=%d\n", i);
+
+              nsds--;
+              if (i == nls.listensd)
+                {
+                  (void)net_connection(&nls);
+                }
+              else
+                {
+                  net_incomingdata(&nls, i);
+                }
+            }
+        }
+    }
+
+  /* Cleanup */
+
+#if 0 /* Don't get here */
+   for (i = 0; i <= nls.mxsd; +i++)
+    {
+      if (FD_ISSET(i, &nls.master))
+        {
+          close(i);
+        }
+    }
+#endif
+  return NULL;  /* Keeps some compilers from complaining */
+}