[ros-diffs] [cgutman] 47732: [DHCPCSVC] - Implement adapter discovery after startup (currently does not work because the iphlpapi code is not implemented)

Author: cgutman
Date: Thu Jun 10 04:38:04 2010
New Revision: 47732

URL: http://svn.reactos.org/svn/reactos?rev=47732&view=rev
Log:
[DHCPCSVC]
- Implement adapter discovery after startup (currently does not work because the iphlpapi
code is not implemented)

Modified:
    trunk/reactos/dll/win32/dhcpcsvc/dhcp/adapter.c
    trunk/reactos/dll/win32/dhcpcsvc/dhcp/dispatch.c
    trunk/reactos/dll/win32/dhcpcsvc/include/rosdhcp.h

Modified: trunk/reactos/dll/win32/dhcpcsvc/dhcp/adapter.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/dll/win32/dhcpcsvc/dhcp/ad…
==============================================================================
--- trunk/reactos/dll/win32/dhcpcsvc/dhcp/adapter.c [iso-8859-1] (original)
+++ trunk/reactos/dll/win32/dhcpcsvc/dhcp/adapter.c [iso-8859-1] Thu Jun 10 04:38:04 2010
@@ -227,141 +227,168 @@
 /*
  * XXX Figure out the way to bind a specific adapter to a socket.
  */
-BOOLEAN AdapterDiscover() {
+DWORD WINAPI AdapterDiscoveryThread(LPVOID Unused) {
     PMIB_IFTABLE Table = (PMIB_IFTABLE) malloc(sizeof(MIB_IFTABLE));
     DWORD Error, Size = sizeof(MIB_IFTABLE);
     PDHCP_ADAPTER Adapter = NULL;
     struct interface_info *ifi = NULL;
     int i;
-    BOOLEAN ret = TRUE;
-
-    DH_DbgPrint(MID_TRACE,("Getting Adapter List...\n"));
-
-    while( (Error = GetIfTable(Table, &Size, 0 )) ==
-           ERROR_INSUFFICIENT_BUFFER ) {
-        DH_DbgPrint(MID_TRACE,("Error %d, New Buffer Size: %d\n", Error,
Size));
-        free( Table );
-        Table = (PMIB_IFTABLE) malloc( Size );
-    }
-
-    if( Error != NO_ERROR ) {
-        ret = FALSE;
-        goto term;
-    }
-
-    DH_DbgPrint(MID_TRACE,("Got Adapter List (%d entries)\n",
Table->dwNumEntries));
-
-    for( i = Table->dwNumEntries - 1; i >= 0; i-- ) {
-        DH_DbgPrint(MID_TRACE,("Getting adapter %d attributes\n",
-                               Table->table[i].dwIndex));
-
-        if ((Adapter = AdapterFindByHardwareAddress(Table->table[i].bPhysAddr,
Table->table[i].dwPhysAddrLen)))
-        {
-            /* This is an existing adapter */
-            if (InterfaceConnected(Table->table[i])) {
-                /* We're still active so we stay in the list */
-                ifi = &Adapter->DhclientInfo;
-            } else {
-                /* We've lost our link so out we go */
-                RemoveEntryList(&Adapter->ListEntry);
-                free(Adapter);
+
+    /* FIXME: Kill this thread when the service is stopped */
+
+    do {
+       DH_DbgPrint(MID_TRACE,("Getting Adapter List...\n"));
+
+       while( (Error = GetIfTable(Table, &Size, 0 )) ==
+               ERROR_INSUFFICIENT_BUFFER ) {
+           DH_DbgPrint(MID_TRACE,("Error %d, New Buffer Size: %d\n", Error,
Size));
+           free( Table );
+           Table = (PMIB_IFTABLE) malloc( Size );
+       }
+
+       if( Error != NO_ERROR )
+           break;
+
+       DH_DbgPrint(MID_TRACE,("Got Adapter List (%d entries)\n",
Table->dwNumEntries));
+
+       for( i = Table->dwNumEntries - 1; i >= 0; i-- ) {
+            DH_DbgPrint(MID_TRACE,("Getting adapter %d attributes\n",
+                                   Table->table[i].dwIndex));
+
+            ApiLock();
+
+            if ((Adapter = AdapterFindByHardwareAddress(Table->table[i].bPhysAddr,
Table->table[i].dwPhysAddrLen)))
+            {
+                /* This is an existing adapter */
+                if (InterfaceConnected(Table->table[i])) {
+                    /* We're still active so we stay in the list */
+                    ifi = &Adapter->DhclientInfo;
+                } else {
+                    /* We've lost our link so out we go */
+                    RemoveEntryList(&Adapter->ListEntry);
+                    free(Adapter);
+                }
+
+                ApiUnlock();
+
+                continue;
             }
 
-            continue;
+            ApiUnlock();
+
+            Adapter = (DHCP_ADAPTER*) calloc( sizeof( DHCP_ADAPTER ) +
Table->table[i].dwMtu, 1 );
+
+            if( Adapter && Table->table[i].dwType == MIB_IF_TYPE_ETHERNET
&& InterfaceConnected(Table->table[i])) {
+                memcpy( &Adapter->IfMib, &Table->table[i],
+                        sizeof(Adapter->IfMib) );
+                Adapter->DhclientInfo.client = &Adapter->DhclientState;
+                Adapter->DhclientInfo.rbuf = Adapter->recv_buf;
+                Adapter->DhclientInfo.rbuf_max = Table->table[i].dwMtu;
+                Adapter->DhclientInfo.rbuf_len =
+                    Adapter->DhclientInfo.rbuf_offset = 0;
+                memcpy(Adapter->DhclientInfo.hw_address.haddr,
+                       Adapter->IfMib.bPhysAddr,
+                       Adapter->IfMib.dwPhysAddrLen);
+                Adapter->DhclientInfo.hw_address.hlen =
Adapter->IfMib.dwPhysAddrLen;
+
+                /* I'm not sure where else to set this, but
+                   some DHCP servers won't take a zero.
+                   We checked the hardware type earlier in
+                   the if statement. */
+                Adapter->DhclientInfo.hw_address.htype = HTYPE_ETHER;
+
+                if( DhcpSocket == INVALID_SOCKET ) {
+                    DhcpSocket =
+                        Adapter->DhclientInfo.rfdesc =
+                        Adapter->DhclientInfo.wfdesc =
+                        socket( AF_INET, SOCK_DGRAM, IPPROTO_UDP );
+
+                    if (DhcpSocket != INVALID_SOCKET) {
+                        Adapter->ListenAddr.sin_family = AF_INET;
+                        Adapter->ListenAddr.sin_port = htons(LOCAL_PORT);
+                        Adapter->BindStatus =
+                             (bind( Adapter->DhclientInfo.rfdesc,
+                                    (struct sockaddr *)&Adapter->ListenAddr,
+                                    sizeof(Adapter->ListenAddr) ) == 0) ?
+                             0 : WSAGetLastError();
+                    } else {
+                        error("socket() failed: %d\n", WSAGetLastError());
+                    }
+                } else {
+                    Adapter->DhclientInfo.rfdesc =
+                        Adapter->DhclientInfo.wfdesc = DhcpSocket;
+                }
+
+                Adapter->DhclientConfig.timeout = DHCP_PANIC_TIMEOUT;
+                Adapter->DhclientConfig.initial_interval = DHCP_DISCOVER_INTERVAL;
+                Adapter->DhclientConfig.retry_interval = DHCP_DISCOVER_INTERVAL;
+                Adapter->DhclientConfig.select_interval = 1;
+                Adapter->DhclientConfig.reboot_timeout = DHCP_REBOOT_TIMEOUT;
+                Adapter->DhclientConfig.backoff_cutoff = DHCP_BACKOFF_MAX;
+                Adapter->DhclientState.interval =
+                    Adapter->DhclientConfig.retry_interval;
+
+                if( PrepareAdapterForService( Adapter ) ) {
+                    Adapter->DhclientInfo.next = ifi;
+                    ifi = &Adapter->DhclientInfo;
+
+                    read_client_conf(&Adapter->DhclientInfo);
+
+                    if (Adapter->DhclientInfo.client->state == S_INIT)
+                    {
+                        add_protocol(Adapter->DhclientInfo.name,
+                                     Adapter->DhclientInfo.rfdesc,
+                                     got_one, &Adapter->DhclientInfo);
+
+	                state_init(&Adapter->DhclientInfo);
+                    }
+
+                    ApiLock();
+                    InsertTailList( &AdapterList, &Adapter->ListEntry );
+                    ApiUnlock();
+                } else { free( Adapter ); Adapter = 0; }
+            } else { free( Adapter ); Adapter = 0; }
+
+            if( !Adapter )
+                DH_DbgPrint(MID_TRACE,("Adapter %d was rejected\n",
+                                       Table->table[i].dwIndex));
         }
-
-        Adapter = (DHCP_ADAPTER*) calloc( sizeof( DHCP_ADAPTER ) +
Table->table[i].dwMtu, 1 );
-
-        if( Adapter && Table->table[i].dwType == MIB_IF_TYPE_ETHERNET
&& InterfaceConnected(Table->table[i])) {
-            memcpy( &Adapter->IfMib, &Table->table[i],
-                    sizeof(Adapter->IfMib) );
-            Adapter->DhclientInfo.client = &Adapter->DhclientState;
-            Adapter->DhclientInfo.rbuf = Adapter->recv_buf;
-            Adapter->DhclientInfo.rbuf_max = Table->table[i].dwMtu;
-            Adapter->DhclientInfo.rbuf_len =
-                Adapter->DhclientInfo.rbuf_offset = 0;
-            memcpy(Adapter->DhclientInfo.hw_address.haddr,
-                   Adapter->IfMib.bPhysAddr,
-                   Adapter->IfMib.dwPhysAddrLen);
-            Adapter->DhclientInfo.hw_address.hlen  =
-                Adapter->IfMib.dwPhysAddrLen;
-            /* I'm not sure where else to set this, but
-               some DHCP servers won't take a zero.
-               We checked the hardware type earlier in
-               the if statement. */
-            Adapter->DhclientInfo.hw_address.htype  =
-                HTYPE_ETHER;
-
-            if( DhcpSocket == INVALID_SOCKET ) {
-                DhcpSocket =
-                    Adapter->DhclientInfo.rfdesc =
-                    Adapter->DhclientInfo.wfdesc =
-                    socket( AF_INET, SOCK_DGRAM, IPPROTO_UDP );
-
-                if (DhcpSocket != INVALID_SOCKET) {
-                    Adapter->ListenAddr.sin_family = AF_INET;
-                    Adapter->ListenAddr.sin_port = htons(LOCAL_PORT);
-                    Adapter->BindStatus =
-                        (bind( Adapter->DhclientInfo.rfdesc,
-                               (struct sockaddr *)&Adapter->ListenAddr,
-                               sizeof(Adapter->ListenAddr) ) == 0) ?
-                        0 : WSAGetLastError();
-                } else {
-                    error("socket() failed: %d\n", WSAGetLastError());
-                }
-            } else {
-                Adapter->DhclientInfo.rfdesc =
-                    Adapter->DhclientInfo.wfdesc = DhcpSocket;
-            }
-
-            Adapter->DhclientConfig.timeout = DHCP_PANIC_TIMEOUT;
-            Adapter->DhclientConfig.initial_interval = DHCP_DISCOVER_INTERVAL;
-            Adapter->DhclientConfig.retry_interval = DHCP_DISCOVER_INTERVAL;
-            Adapter->DhclientConfig.select_interval = 1;
-            Adapter->DhclientConfig.reboot_timeout = DHCP_REBOOT_TIMEOUT;
-            Adapter->DhclientConfig.backoff_cutoff = DHCP_BACKOFF_MAX;
-            Adapter->DhclientState.interval =
-                Adapter->DhclientConfig.retry_interval;
-
-            if( PrepareAdapterForService( Adapter ) ) {
-                Adapter->DhclientInfo.next = ifi;
-                ifi = &Adapter->DhclientInfo;
-
-                read_client_conf(&Adapter->DhclientInfo);
-
-                if (Adapter->DhclientInfo.client->state == S_INIT)
-                {
-                    add_protocol(Adapter->DhclientInfo.name,
-                                 Adapter->DhclientInfo.rfdesc,
-                                 got_one, &Adapter->DhclientInfo);
-
-	            state_init(&Adapter->DhclientInfo);
-                }
-
-                InsertTailList( &AdapterList, &Adapter->ListEntry );
-            } else { free( Adapter ); Adapter = 0; }
-        } else { free( Adapter ); Adapter = 0; }
-
-        if( !Adapter )
-            DH_DbgPrint(MID_TRACE,("Adapter %d was rejected\n",
-                                   Table->table[i].dwIndex));
-    }
-
-    DH_DbgPrint(MID_TRACE,("done with AdapterInit\n"));
-
-term:
+    } while ((Error = NotifyAddrChange(NULL, NULL)) == NO_ERROR);
+
+    DbgPrint("DHCPCSVC: Adapter discovery thread is terminating! (Error:
%d)\n", Error);
+
     if( Table ) free( Table );
-    return ret;
+    return Error;
+}
+
+BOOLEAN StartAdapterDiscovery(VOID) {
+    HANDLE ThreadHandle;
+
+    ThreadHandle = CreateThread(NULL,
+                                0,
+                                AdapterDiscoveryThread,
+                                NULL,
+                                0,
+                                NULL);
+
+    if (ThreadHandle == NULL)
+        return FALSE;
+
+    CloseHandle(ThreadHandle);
+
+    return TRUE;
 }
 
 void AdapterStop() {
     PLIST_ENTRY ListEntry;
     PDHCP_ADAPTER Adapter;
+    ApiLock();
     while( !IsListEmpty( &AdapterList ) ) {
         ListEntry = (PLIST_ENTRY)RemoveHeadList( &AdapterList );
         Adapter = CONTAINING_RECORD( ListEntry, DHCP_ADAPTER, ListEntry );
         free( Adapter );
     }
+    ApiUnlock();
     WSACleanup();
 }
 

Modified: trunk/reactos/dll/win32/dhcpcsvc/dhcp/dispatch.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/dll/win32/dhcpcsvc/dhcp/di…
==============================================================================
--- trunk/reactos/dll/win32/dhcpcsvc/dhcp/dispatch.c [iso-8859-1] (original)
+++ trunk/reactos/dll/win32/dhcpcsvc/dhcp/dispatch.c [iso-8859-1] Thu Jun 10 04:38:04
2010
@@ -69,7 +69,7 @@
     time_t howlong, cur_time;
     struct timeval timeval;
 
-    if (!AdapterDiscover())
+    if (!StartAdapterDiscovery())
          return;
 
     ApiLock();

Modified: trunk/reactos/dll/win32/dhcpcsvc/include/rosdhcp.h
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/dll/win32/dhcpcsvc/include…
==============================================================================
--- trunk/reactos/dll/win32/dhcpcsvc/include/rosdhcp.h [iso-8859-1] (original)
+++ trunk/reactos/dll/win32/dhcpcsvc/include/rosdhcp.h [iso-8859-1] Thu Jun 10 04:38:04
2010
@@ -77,7 +77,7 @@
 #define srandom srand
 
 void AdapterInit(VOID);
-BOOLEAN AdapterDiscover(VOID);
+BOOLEAN StartAdapterDiscovery(VOID);
 void AdapterStop(VOID);
 extern PDHCP_ADAPTER AdapterGetFirst();
 extern PDHCP_ADAPTER AdapterGetNext(PDHCP_ADAPTER);