dll/win32/dhcpcsvc/dhcp/adapter.c

   1 #include "rosdhcp.h"
   2
   3 SOCKET DhcpSocket = INVALID_SOCKET;
   4 static LIST_ENTRY AdapterList;
   5 static WSADATA wsd;
   6
   7 PCHAR *GetSubkeyNames( PCHAR MainKeyName, PCHAR Append ) {
   8     int i = 0;
   9     DWORD Error;
  10     HKEY MainKey;
  11     PCHAR *Out, OutKeyName;
  12     DWORD CharTotal = 0, AppendLen = 1 + strlen(Append);
  13     DWORD MaxSubKeyLen = 0, MaxSubKeys = 0;
  14
  15     Error = RegOpenKey( HKEY_LOCAL_MACHINE, MainKeyName, &MainKey );
  16
  17     if( Error ) return NULL;
  18
  19     Error = RegQueryInfoKey
  20         ( MainKey,
  21           NULL, NULL, NULL,
  22           &MaxSubKeys, &MaxSubKeyLen,
  23           NULL, NULL, NULL, NULL, NULL, NULL );
  24
  25     DH_DbgPrint(MID_TRACE,("MaxSubKeys: %d, MaxSubKeyLen %d\n",
  26                            MaxSubKeys, MaxSubKeyLen));
  27
  28     CharTotal = (sizeof(PCHAR) + MaxSubKeyLen + AppendLen) * (MaxSubKeys + 1);
  29
  30     DH_DbgPrint(MID_TRACE,("AppendLen: %d, CharTotal: %d\n",
  31                            AppendLen, CharTotal));
  32
  33     Out = (CHAR**) malloc( CharTotal );
  34     OutKeyName = ((PCHAR)&Out[MaxSubKeys+1]);
  35
  36     if( !Out ) { RegCloseKey( MainKey ); return NULL; }
  37
  38     i = 0;
  39     do {
  40         Out[i] = OutKeyName;
  41         Error = RegEnumKey( MainKey, i, OutKeyName, MaxSubKeyLen );
  42         if( !Error ) {
  43             strcat( OutKeyName, Append );
  44             DH_DbgPrint(MID_TRACE,("[%d]: %s\n", i, OutKeyName));
  45             OutKeyName += strlen(OutKeyName) + 1;
  46             i++;
  47         } else Out[i] = 0;
  48     } while( Error == ERROR_SUCCESS );
  49
  50     RegCloseKey( MainKey );
  51
  52     return Out;
  53 }
  54
  55 PCHAR RegReadString( HKEY Root, PCHAR Subkey, PCHAR Value ) {
  56     PCHAR SubOut = NULL;
  57     DWORD SubOutLen = 0, Error = 0;
  58     HKEY  ValueKey = NULL;
  59
  60     DH_DbgPrint(MID_TRACE,("Looking in %x:%s:%s\n", Root, Subkey, Value ));
  61
  62     if( Subkey && strlen(Subkey) ) {
  63         if( RegOpenKey( Root, Subkey, &ValueKey ) != ERROR_SUCCESS )
  64             goto regerror;
  65     } else ValueKey = Root;
  66
  67     DH_DbgPrint(MID_TRACE,("Got Key %x\n", ValueKey));
  68
  69     if( (Error = RegQueryValueEx( ValueKey, Value, NULL, NULL,
  70                                   (LPBYTE)SubOut, &SubOutLen )) != ERROR_SUCCESS )
  71         goto regerror;
  72
  73     DH_DbgPrint(MID_TRACE,("Value %s has size %d\n", Value, SubOutLen));
  74
  75     if( !(SubOut = (CHAR*) malloc(SubOutLen)) )
  76         goto regerror;
  77
  78     if( (Error = RegQueryValueEx( ValueKey, Value, NULL, NULL,
  79                                   (LPBYTE)SubOut, &SubOutLen )) != ERROR_SUCCESS )
  80         goto regerror;
  81
  82     DH_DbgPrint(MID_TRACE,("Value %s is %s\n", Value, SubOut));
  83
  84     goto cleanup;
  85
  86 regerror:
  87     if( SubOut ) { free( SubOut ); SubOut = NULL; }
  88 cleanup:
  89     if( ValueKey && ValueKey != Root ) {
  90         DH_DbgPrint(MID_TRACE,("Closing key %x\n", ValueKey));
  91         RegCloseKey( ValueKey );
  92     }
  93
  94     DH_DbgPrint(MID_TRACE,("Returning %x with error %d\n", SubOut, Error));
  95
  96     return SubOut;
  97 }
  98
  99 HKEY FindAdapterKey( PDHCP_ADAPTER Adapter ) {
 100     int i = 0;
 101     PCHAR EnumKeyName =
 102         "SYSTEM\\CurrentControlSet\\Control\\Class\\"
 103         "{4D36E972-E325-11CE-BFC1-08002BE10318}";
 104     PCHAR TargetKeyNameStart =
 105         "SYSTEM\\CurrentControlSet\\Services\\Tcpip\\Parameters\\Interfaces\\";
 106     PCHAR TargetKeyName = NULL;
 107     PCHAR *EnumKeysLinkage = GetSubkeyNames( EnumKeyName, "\\Linkage" );
 108     PCHAR *EnumKeysTop     = GetSubkeyNames( EnumKeyName, "" );
 109     PCHAR RootDevice = NULL;
 110     HKEY EnumKey, OutKey = NULL;
 111     DWORD Error = ERROR_SUCCESS;
 112
 113     if( !EnumKeysLinkage || !EnumKeysTop ) goto cleanup;
 114
 115     Error = RegOpenKey( HKEY_LOCAL_MACHINE, EnumKeyName, &EnumKey );
 116
 117     if( Error ) goto cleanup;
 118
 119     for( i = 0; EnumKeysLinkage[i]; i++ ) {
 120         RootDevice = RegReadString
 121             ( EnumKey, EnumKeysLinkage[i], "RootDevice" );
 122
 123         if( RootDevice &&
 124             !strcmp( RootDevice, Adapter->DhclientInfo.name ) ) {
 125             TargetKeyName =
 126                 (CHAR*) malloc( strlen( TargetKeyNameStart ) +
 127                         strlen( RootDevice ) + 1);
 128             if( !TargetKeyName ) goto cleanup;
 129             sprintf( TargetKeyName, "%s%s",
 130                      TargetKeyNameStart, RootDevice );
 131             Error = RegCreateKeyExA( HKEY_LOCAL_MACHINE, TargetKeyName, 0, NULL, 0, KEY_READ, NULL, &OutKey, NULL );
 132             break;
 133         } else {
 134             free( RootDevice ); RootDevice = 0;
 135         }
 136     }
 137
 138 cleanup:
 139     if( RootDevice ) free( RootDevice );
 140     if( EnumKeysLinkage ) free( EnumKeysLinkage );
 141     if( EnumKeysTop ) free( EnumKeysTop );
 142     if( TargetKeyName ) free( TargetKeyName );
 143
 144     return OutKey;
 145 }
 146
 147 BOOL PrepareAdapterForService( PDHCP_ADAPTER Adapter ) {
 148     HKEY AdapterKey;
 149     DWORD Error = ERROR_SUCCESS, DhcpEnabled, Length = sizeof(DWORD);
 150
 151     Adapter->DhclientState.config = &Adapter->DhclientConfig;
 152     strncpy(Adapter->DhclientInfo.name, (char*)Adapter->IfMib.bDescr,
 153             sizeof(Adapter->DhclientInfo.name));
 154
 155     AdapterKey = FindAdapterKey( Adapter );
 156     if( AdapterKey )
 157     {
 158         Error = RegQueryValueEx(AdapterKey, "EnableDHCP", NULL, NULL, (LPBYTE)&DhcpEnabled, &Length);
 159
 160         if (Error != ERROR_SUCCESS || Length != sizeof(DWORD))
 161             DhcpEnabled = 1;
 162
 163         CloseHandle(AdapterKey);
 164     }
 165     else
 166     {
 167         /* DHCP enabled by default */
 168         DhcpEnabled = 1;
 169     }
 170
 171     if( !DhcpEnabled ) {
 172         /* Non-automatic case */
 173         DbgPrint("DHCPCSVC: Adapter Name: [%s] (static)\n", Adapter->DhclientInfo.name);
 174
 175         Adapter->DhclientState.state = S_STATIC;
 176     } else {
 177         /* Automatic case */
 178         DbgPrint("DHCPCSVC: Adapter Name: [%s] (dynamic)\n", Adapter->DhclientInfo.name);
 179
 180         Adapter->DhclientInfo.client->state = S_INIT;
 181     }
 182
 183     return TRUE;
 184 }
 185
 186 void AdapterInit() {
 187     WSAStartup(0x0101,&wsd);
 188
 189     InitializeListHead( &AdapterList );
 190 }
 191
 192 int
 193 InterfaceConnected(const MIB_IFROW* IfEntry)
 194 {
 195     if (IfEntry->dwOperStatus == IF_OPER_STATUS_CONNECTED ||
 196         IfEntry->dwOperStatus == IF_OPER_STATUS_OPERATIONAL)
 197         return 1;
 198
 199     DH_DbgPrint(MID_TRACE,("Interface %d is down\n", IfEntry->dwIndex));
 200     return 0;
 201 }
 202
 203 BOOL
 204 IsReconnectHackNeeded(PDHCP_ADAPTER Adapter, const MIB_IFROW* IfEntry)
 205 {
 206     struct protocol *proto;
 207     PIP_ADAPTER_INFO AdapterInfo, Orig;
 208     DWORD Size, Ret;
 209     char *ZeroAddress = "0.0.0.0";
 210
 211     proto = find_protocol_by_adapter(&Adapter->DhclientInfo);
 212
 213     if (!proto)
 214         return FALSE;
 215
 216     if (Adapter->DhclientInfo.client->state != S_BOUND)
 217         return FALSE;
 218
 219     ApiUnlock();
 220
 221     Orig = AdapterInfo = HeapAlloc(GetProcessHeap(), 0, sizeof(IP_ADAPTER_INFO));
 222     Size = sizeof(IP_ADAPTER_INFO);
 223     if (!AdapterInfo)
 224     {
 225         ApiLock();
 226         return FALSE;
 227     }
 228
 229     Ret = GetAdaptersInfo(AdapterInfo, &Size);
 230     if (Ret == ERROR_BUFFER_OVERFLOW)
 231     {
 232         HeapFree(GetProcessHeap(), 0, AdapterInfo);
 233         AdapterInfo = HeapAlloc(GetProcessHeap(), 0, Size);
 234         if (!AdapterInfo)
 235         {
 236             ApiLock();
 237             return FALSE;
 238         }
 239
 240         if (GetAdaptersInfo(AdapterInfo, &Size) != NO_ERROR)
 241         {
 242             ApiLock();
 243             return FALSE;
 244         }
 245
 246         Orig = AdapterInfo;
 247         for (; AdapterInfo != NULL; AdapterInfo = AdapterInfo->Next)
 248         {
 249             if (AdapterInfo->Index == IfEntry->dwIndex)
 250                 break;
 251         }
 252
 253         if (AdapterInfo == NULL)
 254         {
 255             HeapFree(GetProcessHeap(), 0, Orig);
 256             ApiLock();
 257             return FALSE;
 258         }
 259     }
 260     else if (Ret != NO_ERROR)
 261     {
 262         HeapFree(GetProcessHeap(), 0, Orig);
 263         ApiLock();
 264         return FALSE;
 265     }
 266
 267     if (!strcmp(AdapterInfo->IpAddressList.IpAddress.String, ZeroAddress))
 268     {
 269         HeapFree(GetProcessHeap(), 0, Orig);
 270         ApiLock();
 271         return TRUE;
 272     }
 273     else
 274     {
 275         HeapFree(GetProcessHeap(), 0, Orig);
 276         ApiLock();
 277         return FALSE;
 278     }
 279 }
 280
 281 /*
 282  * XXX Figure out the way to bind a specific adapter to a socket.
 283  */
 284 DWORD WINAPI AdapterDiscoveryThread(LPVOID Context) {
 285     PMIB_IFTABLE Table = (PMIB_IFTABLE) malloc(sizeof(MIB_IFTABLE));
 286     DWORD Error, Size = sizeof(MIB_IFTABLE);
 287     PDHCP_ADAPTER Adapter = NULL;
 288     HANDLE AdapterStateChangedEvent = (HANDLE)Context;
 289     struct interface_info *ifi = NULL;
 290     struct protocol *proto;
 291     int i, AdapterCount = 0, Broadcast;
 292
 293     /* FIXME: Kill this thread when the service is stopped */
 294
 295     do {
 296        DH_DbgPrint(MID_TRACE,("Getting Adapter List...\n"));
 297
 298        while( (Error = GetIfTable(Table, &Size, 0 )) ==
 299                ERROR_INSUFFICIENT_BUFFER ) {
 300            DH_DbgPrint(MID_TRACE,("Error %d, New Buffer Size: %d\n", Error, Size));
 301            free( Table );
 302            Table = (PMIB_IFTABLE) malloc( Size );
 303        }
 304
 305        if( Error != NO_ERROR )
 306        {
 307            /* HACK: We are waiting until TCP/IP starts */
 308            Sleep(2000);
 309            continue;
 310        }
 311
 312        DH_DbgPrint(MID_TRACE,("Got Adapter List (%d entries)\n", Table->dwNumEntries));
 313
 314        for( i = Table->dwNumEntries - 1; i >= 0; i-- ) {
 315             DH_DbgPrint(MID_TRACE,("Getting adapter %d attributes\n",
 316                                    Table->table[i].dwIndex));
 317
 318             ApiLock();
 319
 320             if ((Adapter = AdapterFindByHardwareAddress(Table->table[i].bPhysAddr, Table->table[i].dwPhysAddrLen)))
 321             {
 322                 proto = find_protocol_by_adapter(&Adapter->DhclientInfo);
 323
 324                 /* This is an existing adapter */
 325                 if (InterfaceConnected(&Table->table[i])) {
 326                     /* We're still active so we stay in the list */
 327                     ifi = &Adapter->DhclientInfo;
 328
 329                     /* This is a hack because IP helper API sucks */
 330                     if (IsReconnectHackNeeded(Adapter, &Table->table[i]))
 331                     {
 332                         /* This handles a disconnect/reconnect */
 333
 334                         remove_protocol(proto);
 335                         Adapter->DhclientInfo.client->state = S_INIT;
 336
 337                         /* These are already invalid since the media state change */
 338                         Adapter->RouterMib.dwForwardNextHop = 0;
 339                         Adapter->NteContext = 0;
 340
 341                         add_protocol(Adapter->DhclientInfo.name,
 342                                      Adapter->DhclientInfo.rfdesc,
 343                                      got_one, &Adapter->DhclientInfo);
 344                         state_init(&Adapter->DhclientInfo);
 345
 346                         SetEvent(AdapterStateChangedEvent);
 347                     }
 348
 349                 } else {
 350                     if (proto)
 351                         remove_protocol(proto);
 352
 353                     /* We've lost our link so out we go */
 354                     RemoveEntryList(&Adapter->ListEntry);
 355                     free(Adapter);
 356                 }
 357
 358                 ApiUnlock();
 359
 360                 continue;
 361             }
 362
 363             ApiUnlock();
 364
 365             Adapter = (DHCP_ADAPTER*) calloc( sizeof( DHCP_ADAPTER ) + Table->table[i].dwMtu, 1 );
 366
 367             if( Adapter && Table->table[i].dwType == MIB_IF_TYPE_ETHERNET && InterfaceConnected(&Table->table[i])) {
 368                 memcpy( &Adapter->IfMib, &Table->table[i],
 369                         sizeof(Adapter->IfMib) );
 370                 Adapter->DhclientInfo.client = &Adapter->DhclientState;
 371                 Adapter->DhclientInfo.rbuf = Adapter->recv_buf;
 372                 Adapter->DhclientInfo.rbuf_max = Table->table[i].dwMtu;
 373                 Adapter->DhclientInfo.rbuf_len =
 374                     Adapter->DhclientInfo.rbuf_offset = 0;
 375                 memcpy(Adapter->DhclientInfo.hw_address.haddr,
 376                        Adapter->IfMib.bPhysAddr,
 377                        Adapter->IfMib.dwPhysAddrLen);
 378                 Adapter->DhclientInfo.hw_address.hlen = Adapter->IfMib.dwPhysAddrLen;
 379
 380                 /* I'm not sure where else to set this, but
 381                    some DHCP servers won't take a zero.
 382                    We checked the hardware type earlier in
 383                    the if statement. */
 384                 Adapter->DhclientInfo.hw_address.htype = HTYPE_ETHER;
 385
 386                 if( DhcpSocket == INVALID_SOCKET ) {
 387                     DhcpSocket =
 388                         Adapter->DhclientInfo.rfdesc =
 389                         Adapter->DhclientInfo.wfdesc =
 390                         socket( AF_INET, SOCK_DGRAM, IPPROTO_UDP );
 391
 392                     if (DhcpSocket != INVALID_SOCKET) {
 393
 394                                                 /* Allow broadcast on this socket */
 395                                                 Broadcast = 1;
 396                                                 setsockopt(DhcpSocket,
 397                                                                    SOL_SOCKET,
 398                                                                    SO_BROADCAST,
 399                                                                    (const char *)&Broadcast,
 400                                                                    sizeof(Broadcast));
 401
 402                         Adapter->ListenAddr.sin_family = AF_INET;
 403                         Adapter->ListenAddr.sin_port = htons(LOCAL_PORT);
 404                         Adapter->BindStatus =
 405                              (bind( Adapter->DhclientInfo.rfdesc,
 406                                     (struct sockaddr *)&Adapter->ListenAddr,
 407                                     sizeof(Adapter->ListenAddr) ) == 0) ?
 408                              0 : WSAGetLastError();
 409                     } else {
 410                         error("socket() failed: %d\n", WSAGetLastError());
 411                     }
 412                 } else {
 413                     Adapter->DhclientInfo.rfdesc =
 414                         Adapter->DhclientInfo.wfdesc = DhcpSocket;
 415                 }
 416
 417                 Adapter->DhclientConfig.timeout = DHCP_PANIC_TIMEOUT;
 418                 Adapter->DhclientConfig.initial_interval = DHCP_DISCOVER_INTERVAL;
 419                 Adapter->DhclientConfig.retry_interval = DHCP_DISCOVER_INTERVAL;
 420                 Adapter->DhclientConfig.select_interval = 1;
 421                 Adapter->DhclientConfig.reboot_timeout = DHCP_REBOOT_TIMEOUT;
 422                 Adapter->DhclientConfig.backoff_cutoff = DHCP_BACKOFF_MAX;
 423                 Adapter->DhclientState.interval =
 424                     Adapter->DhclientConfig.retry_interval;
 425
 426                 if( PrepareAdapterForService( Adapter ) ) {
 427                     Adapter->DhclientInfo.next = ifi;
 428                     ifi = &Adapter->DhclientInfo;
 429
 430                     read_client_conf(&Adapter->DhclientInfo);
 431
 432                     if (Adapter->DhclientInfo.client->state == S_INIT)
 433                     {
 434                         add_protocol(Adapter->DhclientInfo.name,
 435                                      Adapter->DhclientInfo.rfdesc,
 436                                      got_one, &Adapter->DhclientInfo);
 437
 438                         state_init(&Adapter->DhclientInfo);
 439                     }
 440
 441                     ApiLock();
 442                     InsertTailList( &AdapterList, &Adapter->ListEntry );
 443                     AdapterCount++;
 444                     SetEvent(AdapterStateChangedEvent);
 445                     ApiUnlock();
 446                 } else { free( Adapter ); Adapter = 0; }
 447             } else { free( Adapter ); Adapter = 0; }
 448
 449             if( !Adapter )
 450                 DH_DbgPrint(MID_TRACE,("Adapter %d was rejected\n",
 451                                        Table->table[i].dwIndex));
 452         }
 453 #if 0
 454         Error = NotifyAddrChange(NULL, NULL);
 455         if (Error != NO_ERROR)
 456             break;
 457 #else
 458         Sleep(3000);
 459 #endif
 460     } while (TRUE);
 461
 462     DbgPrint("DHCPCSVC: Adapter discovery thread is terminating! (Error: %d)\n", Error);
 463
 464     if( Table ) free( Table );
 465     return Error;
 466 }
 467
 468 HANDLE StartAdapterDiscovery(VOID) {
 469     HANDLE ThreadHandle, EventHandle;
 470
 471     EventHandle = CreateEvent(NULL,
 472                               FALSE,
 473                               FALSE,
 474                               NULL);
 475
 476     ThreadHandle = CreateThread(NULL,
 477                                 0,
 478                                 AdapterDiscoveryThread,
 479                                 (LPVOID)EventHandle,
 480                                 0,
 481                                 NULL);
 482
 483     if (ThreadHandle == NULL)
 484         return NULL;
 485
 486     CloseHandle(ThreadHandle);
 487
 488     return EventHandle;
 489 }
 490
 491 void AdapterStop() {
 492     PLIST_ENTRY ListEntry;
 493     PDHCP_ADAPTER Adapter;
 494     ApiLock();
 495     while( !IsListEmpty( &AdapterList ) ) {
 496         ListEntry = (PLIST_ENTRY)RemoveHeadList( &AdapterList );
 497         Adapter = CONTAINING_RECORD( ListEntry, DHCP_ADAPTER, ListEntry );
 498         free( Adapter );
 499     }
 500     ApiUnlock();
 501     WSACleanup();
 502 }
 503
 504 PDHCP_ADAPTER AdapterFindIndex( unsigned int indx ) {
 505     PDHCP_ADAPTER Adapter;
 506     PLIST_ENTRY ListEntry;
 507
 508     for( ListEntry = AdapterList.Flink;
 509          ListEntry != &AdapterList;
 510          ListEntry = ListEntry->Flink ) {
 511         Adapter = CONTAINING_RECORD( ListEntry, DHCP_ADAPTER, ListEntry );
 512         if( Adapter->IfMib.dwIndex == indx ) return Adapter;
 513     }
 514
 515     return NULL;
 516 }
 517
 518 PDHCP_ADAPTER AdapterFindName( const WCHAR *name ) {
 519     PDHCP_ADAPTER Adapter;
 520     PLIST_ENTRY ListEntry;
 521
 522     for( ListEntry = AdapterList.Flink;
 523          ListEntry != &AdapterList;
 524          ListEntry = ListEntry->Flink ) {
 525         Adapter = CONTAINING_RECORD( ListEntry, DHCP_ADAPTER, ListEntry );
 526         if( !wcsicmp( Adapter->IfMib.wszName, name ) ) return Adapter;
 527     }
 528
 529     return NULL;
 530 }
 531
 532 PDHCP_ADAPTER AdapterFindInfo( struct interface_info *ip ) {
 533     PDHCP_ADAPTER Adapter;
 534     PLIST_ENTRY ListEntry;
 535
 536     for( ListEntry = AdapterList.Flink;
 537          ListEntry != &AdapterList;
 538          ListEntry = ListEntry->Flink ) {
 539         Adapter = CONTAINING_RECORD( ListEntry, DHCP_ADAPTER, ListEntry );
 540         if( ip == &Adapter->DhclientInfo ) return Adapter;
 541     }
 542
 543     return NULL;
 544 }
 545
 546 PDHCP_ADAPTER AdapterFindByHardwareAddress( u_int8_t haddr[16], u_int8_t hlen ) {
 547     PDHCP_ADAPTER Adapter;
 548     PLIST_ENTRY ListEntry;
 549
 550     for(ListEntry = AdapterList.Flink;
 551         ListEntry != &AdapterList;
 552         ListEntry = ListEntry->Flink) {
 553        Adapter = CONTAINING_RECORD( ListEntry, DHCP_ADAPTER, ListEntry );
 554        if (Adapter->DhclientInfo.hw_address.hlen == hlen &&
 555            !memcmp(Adapter->DhclientInfo.hw_address.haddr,
 556                   haddr,
 557                   hlen)) return Adapter;
 558     }
 559
 560     return NULL;
 561 }
 562
 563 PDHCP_ADAPTER AdapterGetFirst() {
 564     if( IsListEmpty( &AdapterList ) ) return NULL; else {
 565         return CONTAINING_RECORD
 566             ( AdapterList.Flink, DHCP_ADAPTER, ListEntry );
 567     }
 568 }
 569
 570 PDHCP_ADAPTER AdapterGetNext( PDHCP_ADAPTER This )
 571 {
 572     if( This->ListEntry.Flink == &AdapterList ) return NULL;
 573     return CONTAINING_RECORD
 574         ( This->ListEntry.Flink, DHCP_ADAPTER, ListEntry );
 575 }
 576
 577 void if_register_send(struct interface_info *ip) {
 578
 579 }
 580
 581 void if_register_receive(struct interface_info *ip) {
 582 }