Add logic to assign IP address in apps/examples/wlan

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

1 files changed, 103 insertions, 1 deletions

diff --git a/apps/examples/wlan/wlan_main.c b/apps/examples/wlan/wlan_main.c
index 652d8269c..9993e8deb 100755
--- a/apps/examples/wlan/wlan_main.c
+++ b/apps/examples/wlan/wlan_main.c
@@ -42,15 +42,34 @@
 
 #include <sys/types.h>
 #include <sys/stat.h>
+#include <sys/ioctl.h>
 
+#include <stdint.h>
 #include <stdio.h>
+#include <string.h>
 #include <unistd.h>
 #include <fcntl.h>
+#include <time.h>
 #include <sched.h>
 #include <errno.h>
+#include <debug.h>
 
 #include <nuttx/usb/usbhost.h>
 
+#include <net/if.h>
+#include <net/uip/uip.h>
+#include <net/uip/uip-arp.h>
+
+#include <apps/netutils/uiplib.h>
+
+/* DHCPC may be used in conjunction with any other feature (or not) */
+
+#ifdef CONFIG_EXAMPLE_WLAN_DHCPC
+#  include <arpa/inet.h>
+#  include <apps/netutils/resolv.h>
+#  include <apps/netutils/dhcpc.h>
+#endif
+
 /****************************************************************************
  * Definitions
  ****************************************************************************/
@@ -112,6 +131,13 @@ static struct usbhost_driver_s *g_drvr;
 
 static int wlan_waiter(int argc, char *argv[])
 {
+#if defined(CONFIG_EXAMPLE_WLAN_DHCPC) || defined(CONFIG_EXAMPLE_WLAN_NOMAC)
+  uint8_t mac[IFHWADDRLEN];
+#endif
+  struct in_addr addr;
+#ifdef CONFIG_EXAMPLE_WLAN_DHCPC
+  void *handle;
+#endif
   bool connected = false;
   int ret;
 
@@ -132,7 +158,83 @@ static int wlan_waiter(int argc, char *argv[])
         {
           /* Yes.. enumerate the newly connected device */
 
-          (void)DRVR_ENUMERATE(g_drvr);
+          ret = DRVR_ENUMERATE(g_drvr);
+
+          /* If the enumeration was successful, then bring up the interface */
+
+          /* Many embedded network interfaces must have a software assigned
+           * MAC
+           */
+
+#ifdef CONFIG_EXAMPLE_WLAN_NOMAC
+          mac[0] = 0x00;
+          mac[1] = 0xe0;
+          mac[2] = 0xb0;
+          mac[3] = 0x0b;
+          mac[4] = 0xba;
+          mac[5] = 0xbe;
+          uip_setmacaddr("eth0", mac);
+#endif
+
+          /* Set up our host address */
+
+#ifdef CONFIG_EXAMPLE_WLAN_DHCPC
+          addr.s_addr = 0;
+#else
+          addr.s_addr = HTONL(CONFIG_EXAMPLE_WLAN_IPADDR);
+#endif
+          uip_sethostaddr("eth0", &addr);
+
+          /* Set up the default router address */
+
+          addr.s_addr = HTONL(CONFIG_EXAMPLE_WLAN_DRIPADDR);
+          uip_setdraddr("eth0", &addr);
+
+          /* Setup the subnet mask */
+
+          addr.s_addr = HTONL(CONFIG_EXAMPLE_WLAN_NETMASK);
+          uip_setnetmask("eth0", &addr);
+
+#ifdef CONFIG_EXAMPLE_WLAN_DHCPC
+          /* Set up the resolver */
+
+          resolv_init();
+
+          /* Get the MAC address of the NIC */
+
+          uip_getmacaddr("eth0", mac);
+
+          /* Set up the DHCPC modules */
+
+          handle = dhcpc_open(&mac, IFHWADDRLEN);
+
+          /* Get an IP address.  Note:  there is no logic here for renewing
+           * the address in this example.  The address should be renewed in
+           * ds.lease_time/2 seconds.
+           */
+
+          printf("Getting IP address\n");
+          if (handle)
+            {
+              struct dhcpc_state ds;
+              (void)dhcpc_request(handle, &ds);
+              uip_sethostaddr("eth1", &ds.ipaddr);
+              if (ds.netmask.s_addr != 0)
+                {
+                  uip_setnetmask("eth0", &ds.netmask);
+                }
+              if (ds.default_router.s_addr != 0)
+                {
+                  uip_setdraddr("eth0", &ds.default_router);
+                }
+              if (ds.dnsaddr.s_addr != 0)
+                {
+                  resolv_conf(&ds.dnsaddr);
+                }
+              dhcpc_close(handle);
+              printf("IP: %s\n", inet_ntoa(ds.ipaddr));
+            }
+#endif
         }
     }