git.meshlink.io Git - catta/blob - src/iface-windows.c

   1 /***
   2   This file is part of catta.
   3
   4   catta is free software; you can redistribute it and/or modify it
   5   under the terms of the GNU Lesser General Public License as
   6   published by the Free Software Foundation; either version 2.1 of the
   7   License, or (at your option) any later version.
   8
   9   catta is distributed in the hope that it will be useful, but WITHOUT
  10   ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  11   or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
  12   Public License for more details.
  13
  14   You should have received a copy of the GNU Lesser General Public
  15   License along with catta; if not, write to the Free Software
  16   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  17   USA.
  18 ***/
  19
  20 #include "iface-windows.h"
  21 #include "iface.h"
  22
  23 #include <stdlib.h> // wcstombs
  24 #include <catta/malloc.h>
  25 #include <catta/log.h>
  26 #include <iphlpapi.h>
  27 #include "hashmap.h"
  28 #include "util.h"   // catta_format_mac_address
  29
  30
  31 // for the luid-to-idx hashmap
  32 static unsigned luid_hash(const void *data)
  33 {
  34     return ((NET_LUID *)data)->Info.NetLuidIndex;
  35 }
  36 static int luid_equal(const void *a, const void *b)
  37 {
  38     return (((NET_LUID *)a)->Value == ((NET_LUID *)b)->Value);
  39 }
  40
  41 static CattaIfIndex find_ifindex(CattaInterfaceMonitor *m, NET_LUID luid)
  42 {
  43     CattaIfIndex *pi = NULL;
  44     NET_LUID *key = NULL;
  45
  46     if((pi = catta_hashmap_lookup(m->osdep.idxmap, &luid)) == NULL) {
  47         // allocate memory for the hashmap key and value
  48         key = catta_malloc(sizeof(luid));
  49         pi = catta_malloc(sizeof(CattaIfIndex));
  50         if(!key || !pi)
  51             goto fail;
  52
  53         *key = luid;
  54
  55         // find an index for this luid
  56         *pi = m->osdep.nidx;
  57         if(*pi < 0)  // overflow
  58             goto fail;
  59
  60         // register the index
  61         if(catta_hashmap_replace(m->osdep.idxmap, key, pi) < 0)
  62             goto fail;
  63         m->osdep.nidx++;
  64     }
  65
  66     return *pi;
  67
  68 fail:
  69     catta_free(key);
  70     catta_free(pi);
  71     return -1;
  72 }
  73
  74 // integrate the information from an IP_ADAPTER_ADDRESSES structure
  75 // as returned by GetAdaptersAddresses into the CattaInterfaceMonitor
  76 static void ip_adapter_address(CattaInterfaceMonitor *m, IP_ADAPTER_ADDRESSES *p)
  77 {
  78     CattaIfIndex idx;
  79     CattaHwInterface *hw;
  80     size_t n;
  81
  82     // look up the interface index by LUID
  83     if((idx = find_ifindex(m, p->Luid)) < 0) {
  84         catta_log_error("could not allocate index ip_adapter_address");
  85         return;
  86     }
  87
  88     // find the CattaHwInterface by index or allocate a new one
  89     if((hw = catta_interface_monitor_get_hw_interface(m, idx)) == NULL) {
  90         if((hw = catta_hw_interface_new(m, idx)) == NULL) {
  91             catta_log_error("catta_hw_interface_new failed in ip_adapter_address");
  92             return;
  93         }
  94     }
  95
  96     // fill the CattaHwInterface struct with data
  97     hw->flags_ok =
  98         (p->OperStatus & IfOperStatusUp) &&
  99         !(p->IfType & IF_TYPE_SOFTWARE_LOOPBACK) &&
 100         !(p->Flags & IP_ADAPTER_NO_MULTICAST) &&
 101         (m->server->config.allow_point_to_point || !(p->IfType & IF_TYPE_PPP));
 102             // XXX what about IF_TYPE_TUNNEL?
 103
 104     n = wcstombs(NULL, p->FriendlyName, 0) + 1;
 105     catta_free(hw->name);
 106     hw->name = catta_new(char, n);
 107     wcstombs(hw->name, p->FriendlyName, n);
 108
 109     hw->mtu = p->Mtu;
 110
 111     hw->mac_address_size = p->PhysicalAddressLength;
 112     if (hw->mac_address_size > CATTA_MAC_ADDRESS_MAX)
 113       hw->mac_address_size = CATTA_MAC_ADDRESS_MAX;
 114     memcpy(hw->mac_address, p->PhysicalAddress, hw->mac_address_size);
 115
 116     // XXX process addresses
 117
 118     // XXX debugging, remove
 119    {
 120      char mac[256];
 121      catta_log_debug("======\n name: %s\n index:%d\n mtu:%d\n mac:%s\n flags_ok:%d\n======",
 122                     hw->name, hw->index,
 123                     hw->mtu,
 124                     catta_format_mac_address(mac, sizeof(mac), hw->mac_address, hw->mac_address_size),
 125                     hw->flags_ok);
 126    }
 127 }
 128
 129
 130 int catta_interface_monitor_init_osdep(CattaInterfaceMonitor *m)
 131 {
 132     m->osdep.nidx = 0;
 133     m->osdep.idxmap = catta_hashmap_new(luid_hash, luid_equal, catta_free, catta_free);
 134     if(m->osdep.idxmap == NULL) {
 135         catta_log_error("out of memory in catta_interface_monitor_init_osdep");
 136         return -1;
 137     }
 138
 139     // XXX register callbacks to get notified of interface/address changes
 140
 141     return 0;
 142 }
 143
 144 void catta_interface_monitor_free_osdep(CattaInterfaceMonitor *m)
 145 {
 146     catta_hashmap_free(m->osdep.idxmap);
 147 }
 148
 149 void catta_interface_monitor_sync(CattaInterfaceMonitor *m)
 150 {
 151     IP_ADAPTER_ADDRESSES *buf = NULL;
 152     IP_ADAPTER_ADDRESSES *p;
 153     ULONG bufsize = 15000;
 154     ULONG r;
 155
 156     // allocate a buffer and call GetAdaptersAddresses
 157     // retry with the correct size if the buffer was too small
 158     do {
 159         catta_free(buf);    // no-op on first iteration
 160         if((buf = catta_malloc(bufsize)) == NULL) {
 161             catta_log_error("malloc failed in catta_interface_monitor_sync");
 162             return;
 163         }
 164
 165         r = GetAdaptersAddresses(AF_UNSPEC, 0, NULL, buf, &bufsize);
 166     } while(r == ERROR_BUFFER_OVERFLOW);
 167
 168     if(r != NO_ERROR) {
 169         catta_log_error("GetAdaptersAddresses failed: %u", (unsigned int)r);
 170         return;
 171     }
 172
 173     // XXX remove interfaces for adapters that are no longer in the list
 174
 175     // create 'CattaInterface's for every adapter
 176     for(p=buf; p; p=p->Next)
 177         ip_adapter_address(m, p);
 178
 179     catta_free(buf);
 180
 181     m->list_complete = 1;
 182     catta_interface_monitor_check_relevant(m);
 183     catta_interface_monitor_update_rrs(m, 0);
 184     catta_log_info("Network interface enumeration completed.");
 185 }